From f70c0b0bf0447fd5db784f8f6c41226d5c5f7846 Mon Sep 17 00:00:00 2001 From: Garrett D'Amore Date: Tue, 5 Nov 2024 22:49:59 -0800 Subject: websocket: fix for setting user headers This was a crasher bug (found by test suite). --- src/supplemental/websocket/websocket_test.c | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'src/supplemental/websocket/websocket_test.c') diff --git a/src/supplemental/websocket/websocket_test.c b/src/supplemental/websocket/websocket_test.c index f648f740..781ca1d8 100644 --- a/src/supplemental/websocket/websocket_test.c +++ b/src/supplemental/websocket/websocket_test.c @@ -146,6 +146,12 @@ test_websocket_conn_props(void) NUTS_PASS(nng_stream_listener_listen(l)); NUTS_PASS(nng_stream_dialer_alloc(&d, uri)); + NUTS_PASS(nng_stream_dialer_set_string( + d, NNG_OPT_WS_REQUEST_HEADER "NNG-Req", "True")); + + NUTS_PASS(nng_stream_listener_set_string( + l, NNG_OPT_WS_RESPONSE_HEADER "NNG-Rep", "True")); + nng_stream_dialer_dial(d, daio); nng_stream_listener_accept(l, laio); @@ -180,6 +186,19 @@ test_websocket_conn_props(void) NUTS_FAIL( nng_stream_get_size(c1, NNG_OPT_TCP_NODELAY, &sz), NNG_EBADTYPE); + NUTS_FAIL(nng_stream_get_string( + c1, NNG_OPT_WS_REQUEST_HEADER "No-Such-Header", &str), + NNG_ENOENT); + NUTS_PASS(nng_stream_get_string( + c1, NNG_OPT_WS_REQUEST_HEADER "NNG-Req", &str)); + NUTS_MATCH(str, "True"); + nng_strfree(str); + + NUTS_PASS(nng_stream_get_string( + c2, NNG_OPT_WS_RESPONSE_HEADER "NNG-Rep", &str)); + NUTS_MATCH(str, "True"); + nng_strfree(str); + NUTS_PASS(nng_stream_get_string( c1, NNG_OPT_WS_REQUEST_HEADER "Sec-WebSocket-Version", &str)); NUTS_TRUE(str != NULL); -- cgit v1.2.3-70-g09d2