// // Copyright 2017 Garrett D'Amore // // 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 "convey.h" #include "core/nng_impl.h" #include "nng.h" #include Main({ Test("Socket Operations", { Convey("We are able to open a PAIR socket", { int rv; nng_socket sock; So(nng_pair_open(&sock) == 0); Reset({ nng_close(sock); }); Convey("And we can shut it down", { rv = nng_shutdown(sock); So(rv == 0); rv = nng_shutdown(sock); So(rv == NNG_ECLOSED); }); Convey("It's type is still proto", { So(nng_protocol(sock) == NNG_PROTO_PAIR); }); Convey("Recv with no pipes times out correctly", { nng_msg *msg = NULL; int64_t when = 100000; uint64_t now; now = nni_clock(); rv = nng_setopt(sock, NNG_OPT_RCVTIMEO, &when, sizeof(when)); So(rv == 0); rv = nng_recvmsg(sock, &msg, 0); So(rv == NNG_ETIMEDOUT); So(msg == NULL); So(nni_clock() >= (now + when)); So(nni_clock() < (now + (when * 2))); }); Convey("Recv nonblock with no pipes gives EAGAIN", { nng_msg *msg = NULL; rv = nng_recvmsg(sock, &msg, NNG_FLAG_NONBLOCK); So(rv == NNG_EAGAIN); So(msg == NULL); }); Convey("Send with no pipes times out correctly", { nng_msg *msg = NULL; int64_t when = 100000; uint64_t now; // We cheat to get access to the core's clock. So(nng_msg_alloc(&msg, 0) == 0); So(msg != NULL); now = nni_clock(); rv = nng_setopt(sock, NNG_OPT_SNDTIMEO, &when, sizeof(when)); So(rv == 0); rv = nng_sendmsg(sock, msg, 0); So(rv == NNG_ETIMEDOUT); So(nni_clock() >= (now + when)); So(nni_clock() < (now + (when * 2))); nng_msg_free(msg); }); Convey("We can set and get options", { int64_t when = 1234; int64_t check = 0; size_t sz; rv = nng_setopt(sock, NNG_OPT_SNDTIMEO, &when, sizeof(when)); So(rv == 0); sz = sizeof(check); Convey("Short size is not copied", { sz = 0; rv = nng_getopt(sock, NNG_OPT_SNDTIMEO, &check, &sz); So(rv == 0); So(sz == sizeof(check)); So(check == 0); }); Convey("Correct size is copied", { sz = sizeof(check); rv = nng_getopt(sock, NNG_OPT_SNDTIMEO, &check, &sz); So(rv == 0); So(sz == sizeof(check)); So(check == 1234); }); Convey("Short size buf is not copied", { int len = 5; sz = 0; So(nng_getopt(sock, NNG_OPT_RCVBUF, &len, &sz) == 0); So(len == 5); }); Convey("Insane buffer size fails", { int len = 1024 * 1024; sz = sizeof(len); So(nng_setopt(sock, NNG_OPT_RCVBUF, &len, sz) == NNG_EINVAL); }); Convey("Negative timeout fails", { when = -5; sz = sizeof(when); So(nng_setopt(sock, NNG_OPT_RCVTIMEO, &when, sz) == NNG_EINVAL); }); Convey("Short timeout fails", { when = 0; sz = sizeof(when) - 1; So(nng_setopt(sock, NNG_OPT_RCVTIMEO, &when, sz) == NNG_EINVAL); }); Convey("Bogus raw fails", { int raw = 42; sz = sizeof(raw); raw = 42; So(nng_setopt(sock, NNG_OPT_RAW, &raw, sz) == NNG_EINVAL); raw = -42; So(nng_setopt(sock, NNG_OPT_RAW, &raw, sz) == NNG_EINVAL); raw = 0; So(nng_setopt(sock, NNG_OPT_RAW, &raw, sz) == 0); }); Convey("Unsupported options fail", { char *crap = "crap"; So(nng_setopt(sock, NNG_OPT_SUBSCRIBE, crap, strlen(crap)) == NNG_ENOTSUP); }); Convey("Bogus sizes fail", { size_t rmax; rmax = 6550; So(nng_setopt(sock, NNG_OPT_RCVMAXSZ, &rmax, sizeof(rmax)) == 0); rmax = 0; sz = sizeof(rmax); So(nng_getopt(sock, NNG_OPT_RCVMAXSZ, &rmax, &sz) == 0); So(rmax == 6550); rmax = 102400; So(nng_setopt(sock, NNG_OPT_RCVMAXSZ, &rmax, 1) == NNG_EINVAL); So(nng_getopt(sock, NNG_OPT_RCVMAXSZ, &rmax, &sz) == 0); So(rmax == 6550); if (sizeof(size_t) == 8) { rmax = 0x10000; rmax <<= 30; So(nng_setopt(sock, NNG_OPT_RCVMAXSZ, &rmax, sizeof(rmax)) == NNG_EINVAL); So(nng_getopt(sock, NNG_OPT_RCVMAXSZ, &rmax, &sz) == 0); So(rmax == 6550); } }); }); Convey("Bogus URLs not supported", { Convey("Dialing fails properly", { rv = nng_dial(sock, "bogus://somewhere", NULL, 0); So(rv == NNG_ENOTSUP); }); Convey("Listening fails properly", { rv = nng_listen(sock, "bogus://elsewhere", NULL, 0); So(rv == NNG_ENOTSUP); }); }); Convey("Dialing synch can get refused", { rv = nng_dial(sock, "inproc://notthere", NULL, NNG_FLAG_SYNCH); So(rv == NNG_ECONNREFUSED); }); Convey("Listening works", { rv = nng_listen(sock, "inproc://here", NULL, NNG_FLAG_SYNCH); So(rv == 0); Convey("Second listen fails ADDRINUSE", { rv = nng_listen(sock, "inproc://here", NULL, NNG_FLAG_SYNCH); So(rv == NNG_EADDRINUSE); }); Convey("We can connect to it", { nng_socket sock2; So(nng_pair_open(&sock2) == 0); Reset({ nng_close(sock2); }); rv = nng_dial(sock2, "inproc://here", NULL, NNG_FLAG_SYNCH); So(rv == 0); nng_close(sock2); }); }); Convey("We can send and receive messages", { nng_socket sock2; int len = 1; size_t sz; uint64_t second = 3000000; char * buf; So(nng_pair_open(&sock2) == 0); Reset({ nng_close(sock2); }); So(nng_setopt(sock, NNG_OPT_RCVBUF, &len, sizeof(len)) == 0); sz = sizeof(len); So(nng_getopt( sock, NNG_OPT_RCVBUF, &len, &sz) == 0); So(len == 1); sz = 0; So(nng_setopt(sock, NNG_OPT_SNDBUF, &len, sizeof(len)) == 0); So(nng_setopt(sock2, NNG_OPT_SNDBUF, &len, sizeof(len)) == 0); So(nng_setopt(sock, NNG_OPT_SNDTIMEO, &second, sizeof(second)) == 0); So(nng_setopt(sock, NNG_OPT_RCVTIMEO, &second, sizeof(second)) == 0); So(nng_setopt(sock2, NNG_OPT_SNDTIMEO, &second, sizeof(second)) == 0); So(nng_setopt(sock2, NNG_OPT_RCVTIMEO, &second, sizeof(second)) == 0); So(nng_listen(sock, "inproc://test1", NULL, NNG_FLAG_SYNCH) == 0); So(nng_dial(sock2, "inproc://test1", NULL, NNG_FLAG_SYNCH) == 0); So(nng_send(sock, "abc", 4, 0) == 0); So(nng_recv( sock2, &buf, &sz, NNG_FLAG_ALLOC) == 0); So(buf != NULL); So(sz == 4); So(memcmp(buf, "abc", 4) == 0); nng_free(buf, sz); }); }); }); })