diff options
Diffstat (limited to 'src/sp')
| -rw-r--r-- | src/sp/CMakeLists.txt | 1 | ||||
| -rw-r--r-- | src/sp/device_test.c | 170 |
2 files changed, 171 insertions, 0 deletions
diff --git a/src/sp/CMakeLists.txt b/src/sp/CMakeLists.txt index 128aac15..7ccaf051 100644 --- a/src/sp/CMakeLists.txt +++ b/src/sp/CMakeLists.txt @@ -19,4 +19,5 @@ nng_sources( ) nng_test(pipe_test) +nng_test(device_test) nng_test(multistress_test) diff --git a/src/sp/device_test.c b/src/sp/device_test.c new file mode 100644 index 00000000..6d2ff1fb --- /dev/null +++ b/src/sp/device_test.c @@ -0,0 +1,170 @@ +// +// Copyright 2024 Staysail Systems, Inc. <info@staysail.tech> +// Copyright 2018 Capitar IT Group BV <info@capitar.com> +// +// This software is supplied under the terms of the MIT License, a +// copy of which should be located in the distribution where this +// file was obtained (LICENSE.txt). A copy of the license may also be +// found online at https://opensource.org/licenses/MIT. +// + +#include <string.h> + +#include <nng/nng.h> +#include <nng/protocol/pair0/pair.h> +#include <nng/protocol/pair1/pair.h> + +#include <nuts.h> + +struct dev_data { + nng_socket s1; + nng_socket s2; +}; + +void +dodev(void *arg) +{ + struct dev_data *d = arg; + + nng_device(d->s1, d->s2); +} + +#define SECOND(x) ((x) *1000) + +void +test_device_not_cooked(void) +{ + nng_socket cooked; + nng_socket raw; + NUTS_PASS(nng_pair1_open(&cooked)); + NUTS_PASS(nng_pair1_open_raw(&raw)); + NUTS_FAIL(nng_device(cooked, cooked), NNG_EINVAL); + NUTS_FAIL(nng_device(raw, cooked), NNG_EINVAL); + NUTS_FAIL(nng_device(cooked, raw), NNG_EINVAL); + NUTS_CLOSE(cooked); + NUTS_CLOSE(raw); +} + +void +test_device_incompatible(void) +{ + nng_socket s1; + nng_socket s2; + NUTS_PASS(nng_pair0_open_raw(&s1)); + NUTS_PASS(nng_pair1_open_raw(&s2)); + NUTS_FAIL(nng_device(s1, s2), NNG_EINVAL); + NUTS_CLOSE(s1); + NUTS_CLOSE(s2); +} +void +test_device_forward(void) +{ + nng_thread *thr; + struct dev_data d; + nng_duration tmo = SECOND(1); + nng_socket e1, e2; + + // will be a pair variant + NUTS_PASS(nng_pair1_open_raw(&d.s1)); + NUTS_PASS(nng_pair1_open_raw(&d.s2)); + NUTS_PASS(nng_pair1_open(&e1)); + NUTS_PASS(nng_pair1_open(&e2)); + NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_RECVTIMEO, tmo)); + NUTS_PASS(nng_socket_set_ms(e2, NNG_OPT_RECVTIMEO, tmo)); + NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_SENDTIMEO, tmo)); + NUTS_PASS(nng_socket_set_ms(e2, NNG_OPT_SENDTIMEO, tmo)); + + NUTS_MARRY(e1, d.s1); + NUTS_MARRY(e2, d.s2); + NUTS_PASS(nng_thread_create(&thr, dodev, &d)); + nng_thread_set_name(thr, "device thread"); + + NUTS_SEND(e1, "ping"); + NUTS_RECV(e2, "ping"); + NUTS_SEND(e2, "pong"); + NUTS_RECV(e1, "pong"); + + NUTS_CLOSE(e1); + NUTS_CLOSE(e2); + NUTS_CLOSE(d.s1); + NUTS_CLOSE(d.s2); + nng_thread_destroy(thr); +} + +void +test_device_reflect(void) +{ + nng_thread *thr; + struct dev_data d; + nng_duration tmo = SECOND(1); + nng_socket e1; + + // will be a pair variant + NUTS_PASS(nng_pair1_open_raw(&d.s1)); + d.s2 = d.s1; + NUTS_PASS(nng_pair1_open(&e1)); + NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_RECVTIMEO, tmo)); + NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_SENDTIMEO, tmo)); + + NUTS_MARRY(e1, d.s1); + NUTS_PASS(nng_thread_create(&thr, dodev, &d)); + nng_thread_set_name(thr, "device thread"); + + NUTS_SEND(e1, "ping"); + NUTS_RECV(e1, "ping"); + NUTS_SEND(e1, "pong"); + NUTS_RECV(e1, "pong"); + + NUTS_CLOSE(e1); + NUTS_CLOSE(d.s1); + nng_thread_destroy(thr); +} + +void +test_device_aio(void) +{ + struct dev_data d; + nng_duration tmo = SECOND(1); + nng_socket e1, e2; + nng_aio *aio; + + // will be a pair variant + NUTS_PASS(nng_pair1_open_raw(&d.s1)); + NUTS_PASS(nng_pair1_open_raw(&d.s2)); + NUTS_PASS(nng_pair1_open(&e1)); + NUTS_PASS(nng_pair1_open(&e2)); + NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_RECVTIMEO, tmo)); + NUTS_PASS(nng_socket_set_ms(e2, NNG_OPT_RECVTIMEO, tmo)); + NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_SENDTIMEO, tmo)); + NUTS_PASS(nng_socket_set_ms(e2, NNG_OPT_SENDTIMEO, tmo)); + + NUTS_MARRY(e1, d.s1); + NUTS_MARRY(e2, d.s2); + // cancellation of this aio is how we stop it + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + nng_device_aio(aio, d.s1, d.s2); + + NUTS_SEND(e1, "ping"); + NUTS_RECV(e2, "ping"); + NUTS_SEND(e2, "pong"); + NUTS_RECV(e1, "pong"); + + nng_aio_cancel(aio); + nng_aio_wait(aio); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + nng_aio_free(aio); + + NUTS_CLOSE(e1); + NUTS_CLOSE(e2); + NUTS_CLOSE(d.s1); + NUTS_CLOSE(d.s2); +} + +NUTS_TESTS = { + { "device not cooked", test_device_not_cooked }, + { "device incompatible", test_device_incompatible }, + { "device forward", test_device_forward }, + { "device reflect", test_device_reflect }, + { "device aio", test_device_aio }, + { NULL, NULL }, +}; |