diff options
Diffstat (limited to 'lib/libnv/tests/nvlist_send_recv_test.c')
| -rw-r--r-- | lib/libnv/tests/nvlist_send_recv_test.c | 749 |
1 files changed, 749 insertions, 0 deletions
diff --git a/lib/libnv/tests/nvlist_send_recv_test.c b/lib/libnv/tests/nvlist_send_recv_test.c new file mode 100644 index 000000000000..79297dfe2043 --- /dev/null +++ b/lib/libnv/tests/nvlist_send_recv_test.c @@ -0,0 +1,749 @@ +/*- + * Copyright (c) 2013 The FreeBSD Foundation + * + * This software was developed by Pawel Jakub Dawidek under sponsorship from + * the FreeBSD Foundation. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +#include <sys/param.h> +#include <sys/socket.h> +#include <sys/sysctl.h> +#include <sys/wait.h> +#include <sys/nv.h> + +#include <stdlib.h> +#include <err.h> +#include <errno.h> +#include <fcntl.h> +#include <paths.h> +#include <stdio.h> +#include <string.h> +#include <unistd.h> + +#include <atf-c.h> + +#include <nv_impl.h> +#include <msgio.h> + +#define ALPHABET "abcdefghijklmnopqrstuvwxyz" +#define fd_is_valid(fd) (fcntl((fd), F_GETFL) != -1 || errno != EBADF) + +static void +send_nvlist_child(int sock) +{ + nvlist_t *nvl; + nvlist_t *empty; + int pfd[2]; + + nvl = nvlist_create(0); + empty = nvlist_create(0); + + nvlist_add_bool(nvl, "nvlist/bool/true", true); + nvlist_add_bool(nvl, "nvlist/bool/false", false); + nvlist_add_number(nvl, "nvlist/number/0", 0); + nvlist_add_number(nvl, "nvlist/number/1", 1); + nvlist_add_number(nvl, "nvlist/number/-1", -1); + nvlist_add_number(nvl, "nvlist/number/UINT64_MAX", UINT64_MAX); + nvlist_add_number(nvl, "nvlist/number/INT64_MIN", INT64_MIN); + nvlist_add_number(nvl, "nvlist/number/INT64_MAX", INT64_MAX); + nvlist_add_string(nvl, "nvlist/string/", ""); + nvlist_add_string(nvl, "nvlist/string/x", "x"); + nvlist_add_string(nvl, "nvlist/string/" ALPHABET, ALPHABET); + + nvlist_add_descriptor(nvl, "nvlist/descriptor/STDERR_FILENO", + STDERR_FILENO); + if (pipe(pfd) == -1) + err(EXIT_FAILURE, "pipe"); + if (write(pfd[1], "test", 4) != 4) + err(EXIT_FAILURE, "write"); + close(pfd[1]); + nvlist_add_descriptor(nvl, "nvlist/descriptor/pipe_rd", pfd[0]); + close(pfd[0]); + + nvlist_add_binary(nvl, "nvlist/binary/x", "x", 1); + nvlist_add_binary(nvl, "nvlist/binary/" ALPHABET, ALPHABET, + sizeof(ALPHABET)); + nvlist_move_nvlist(nvl, "nvlist/nvlist/empty", empty); + nvlist_add_nvlist(nvl, "nvlist/nvlist", nvl); + + nvlist_send(sock, nvl); + + nvlist_destroy(nvl); +} + +static void +send_nvlist_parent(int sock) +{ + nvlist_t *nvl; + const nvlist_t *cnvl, *empty; + const char *name, *cname; + void *cookie, *ccookie; + int type, ctype, fd; + size_t size; + char buf[4]; + + nvl = nvlist_recv(sock, 0); + ATF_REQUIRE(nvlist_error(nvl) == 0); + if (nvlist_error(nvl) != 0) + err(1, "nvlist_recv() failed"); + + cookie = NULL; + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_BOOL); + ATF_REQUIRE(strcmp(name, "nvlist/bool/true") == 0); + ATF_REQUIRE(nvlist_get_bool(nvl, name) == true); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_BOOL); + ATF_REQUIRE(strcmp(name, "nvlist/bool/false") == 0); + ATF_REQUIRE(nvlist_get_bool(nvl, name) == false); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(name, "nvlist/number/0") == 0); + ATF_REQUIRE(nvlist_get_number(nvl, name) == 0); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(name, "nvlist/number/1") == 0); + ATF_REQUIRE(nvlist_get_number(nvl, name) == 1); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(name, "nvlist/number/-1") == 0); + ATF_REQUIRE((int)nvlist_get_number(nvl, name) == -1); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(name, "nvlist/number/UINT64_MAX") == 0); + ATF_REQUIRE(nvlist_get_number(nvl, name) == UINT64_MAX); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(name, "nvlist/number/INT64_MIN") == 0); + ATF_REQUIRE((int64_t)nvlist_get_number(nvl, name) == INT64_MIN); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(name, "nvlist/number/INT64_MAX") == 0); + ATF_REQUIRE((int64_t)nvlist_get_number(nvl, name) == INT64_MAX); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_STRING); + ATF_REQUIRE(strcmp(name, "nvlist/string/") == 0); + ATF_REQUIRE(strcmp(nvlist_get_string(nvl, name), "") == 0); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_STRING); + ATF_REQUIRE(strcmp(name, "nvlist/string/x") == 0); + ATF_REQUIRE(strcmp(nvlist_get_string(nvl, name), "x") == 0); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_STRING); + ATF_REQUIRE(strcmp(name, "nvlist/string/" ALPHABET) == 0); + ATF_REQUIRE(strcmp(nvlist_get_string(nvl, name), ALPHABET) == 0); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_DESCRIPTOR); + ATF_REQUIRE(strcmp(name, "nvlist/descriptor/STDERR_FILENO") == 0); + ATF_REQUIRE(fd_is_valid(nvlist_get_descriptor(nvl, name))); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_DESCRIPTOR); + ATF_REQUIRE(strcmp(name, "nvlist/descriptor/pipe_rd") == 0); + fd = nvlist_get_descriptor(nvl, name); + ATF_REQUIRE(fd_is_valid(fd)); + ATF_REQUIRE(read(fd, buf, sizeof(buf)) == 4); + ATF_REQUIRE(strncmp(buf, "test", sizeof(buf)) == 0); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_BINARY); + ATF_REQUIRE(strcmp(name, "nvlist/binary/x") == 0); + ATF_REQUIRE(memcmp(nvlist_get_binary(nvl, name, NULL), "x", 1) == 0); + ATF_REQUIRE(memcmp(nvlist_get_binary(nvl, name, &size), "x", 1) == 0); + ATF_REQUIRE(size == 1); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_BINARY); + ATF_REQUIRE(strcmp(name, "nvlist/binary/" ALPHABET) == 0); + ATF_REQUIRE(memcmp(nvlist_get_binary(nvl, name, NULL), ALPHABET, + sizeof(ALPHABET)) == 0); + ATF_REQUIRE(memcmp(nvlist_get_binary(nvl, name, &size), ALPHABET, + sizeof(ALPHABET)) == 0); + ATF_REQUIRE(size == sizeof(ALPHABET)); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_NVLIST); + ATF_REQUIRE(strcmp(name, "nvlist/nvlist/empty") == 0); + cnvl = nvlist_get_nvlist(nvl, name); + ATF_REQUIRE(nvlist_empty(cnvl)); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name != NULL); + ATF_REQUIRE(type == NV_TYPE_NVLIST); + ATF_REQUIRE(strcmp(name, "nvlist/nvlist") == 0); + cnvl = nvlist_get_nvlist(nvl, name); + + ccookie = NULL; + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_BOOL); + ATF_REQUIRE(strcmp(cname, "nvlist/bool/true") == 0); + ATF_REQUIRE(nvlist_get_bool(cnvl, cname) == true); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_BOOL); + ATF_REQUIRE(strcmp(cname, "nvlist/bool/false") == 0); + ATF_REQUIRE(nvlist_get_bool(cnvl, cname) == false); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(cname, "nvlist/number/0") == 0); + ATF_REQUIRE(nvlist_get_number(cnvl, cname) == 0); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(cname, "nvlist/number/1") == 0); + ATF_REQUIRE(nvlist_get_number(cnvl, cname) == 1); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(cname, "nvlist/number/-1") == 0); + ATF_REQUIRE((int)nvlist_get_number(cnvl, cname) == -1); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(cname, "nvlist/number/UINT64_MAX") == 0); + ATF_REQUIRE(nvlist_get_number(cnvl, cname) == UINT64_MAX); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(cname, "nvlist/number/INT64_MIN") == 0); + ATF_REQUIRE((int64_t)nvlist_get_number(cnvl, cname) == INT64_MIN); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_NUMBER); + ATF_REQUIRE(strcmp(cname, "nvlist/number/INT64_MAX") == 0); + ATF_REQUIRE((int64_t)nvlist_get_number(cnvl, cname) == INT64_MAX); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_STRING); + ATF_REQUIRE(strcmp(cname, "nvlist/string/") == 0); + ATF_REQUIRE(strcmp(nvlist_get_string(cnvl, cname), "") == 0); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_STRING); + ATF_REQUIRE(strcmp(cname, "nvlist/string/x") == 0); + ATF_REQUIRE(strcmp(nvlist_get_string(cnvl, cname), "x") == 0); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_STRING); + ATF_REQUIRE(strcmp(cname, "nvlist/string/" ALPHABET) == 0); + ATF_REQUIRE(strcmp(nvlist_get_string(cnvl, cname), ALPHABET) == 0); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_DESCRIPTOR); + ATF_REQUIRE(strcmp(cname, "nvlist/descriptor/STDERR_FILENO") == 0); + ATF_REQUIRE(fd_is_valid(nvlist_get_descriptor(cnvl, cname))); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_DESCRIPTOR); + ATF_REQUIRE(strcmp(cname, "nvlist/descriptor/pipe_rd") == 0); + ATF_REQUIRE(fd_is_valid(nvlist_get_descriptor(cnvl, cname))); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_BINARY); + ATF_REQUIRE(strcmp(cname, "nvlist/binary/x") == 0); + ATF_REQUIRE(memcmp(nvlist_get_binary(cnvl, cname, NULL), "x", 1) == 0); + ATF_REQUIRE(memcmp(nvlist_get_binary(cnvl, cname, &size), "x", 1) == 0); + ATF_REQUIRE(size == 1); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_BINARY); + ATF_REQUIRE(strcmp(cname, "nvlist/binary/" ALPHABET) == 0); + ATF_REQUIRE(memcmp(nvlist_get_binary(cnvl, cname, NULL), ALPHABET, + sizeof(ALPHABET)) == 0); + ATF_REQUIRE(memcmp(nvlist_get_binary(cnvl, cname, &size), ALPHABET, + sizeof(ALPHABET)) == 0); + ATF_REQUIRE(size == sizeof(ALPHABET)); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname != NULL); + ATF_REQUIRE(ctype == NV_TYPE_NVLIST); + ATF_REQUIRE(strcmp(cname, "nvlist/nvlist/empty") == 0); + empty = nvlist_get_nvlist(cnvl, cname); + ATF_REQUIRE(nvlist_empty(empty)); + + cname = nvlist_next(cnvl, &ctype, &ccookie); + ATF_REQUIRE(cname == NULL); + + name = nvlist_next(nvl, &type, &cookie); + ATF_REQUIRE(name == NULL); + + nvlist_destroy(nvl); +} + +static void +nvlist_send_recv__send_nvlist(short sotype) +{ + int socks[2], status; + pid_t pid; + + ATF_REQUIRE(socketpair(PF_UNIX, sotype, 0, socks) == 0); + + pid = fork(); + ATF_REQUIRE(pid >= 0); + if (pid == 0) { + /* Child. */ + (void)close(socks[0]); + send_nvlist_child(socks[1]); + _exit(0); + } + + (void)close(socks[1]); + send_nvlist_parent(socks[0]); + + ATF_REQUIRE(waitpid(pid, &status, 0) == pid); + ATF_REQUIRE(status == 0); +} + +static void +nvlist_send_recv__send_closed_fd(short sotype) +{ + nvlist_t *nvl; + int socks[2]; + + ATF_REQUIRE(socketpair(PF_UNIX, sotype, 0, socks) == 0); + + nvl = nvlist_create(0); + ATF_REQUIRE(nvl != NULL); + nvlist_add_descriptor(nvl, "fd", 12345); + ATF_REQUIRE(nvlist_error(nvl) == EBADF); + + ATF_REQUIRE_ERRNO(EBADF, nvlist_send(socks[1], nvl) != 0); +} + +static int +nopenfds(void) +{ + size_t len; + int error, mib[4], n; + + mib[0] = CTL_KERN; + mib[1] = KERN_PROC; + mib[2] = KERN_PROC_NFDS; + mib[3] = 0; + + len = sizeof(n); + error = sysctl(mib, nitems(mib), &n, &len, NULL, 0); + if (error != 0) + return (-1); + return (n); +} + +#define NFDS 512 + +static void +send_many_fds_child(int sock) +{ + char name[16]; + nvlist_t *nvl; + int anfds, bnfds, fd, i, j; + + fd = open(_PATH_DEVNULL, O_RDONLY); + ATF_REQUIRE(fd >= 0); + + for (i = 1; i < NFDS; i++) { + nvl = nvlist_create(0); + bnfds = nopenfds(); + if (bnfds == -1) + err(EXIT_FAILURE, "sysctl"); + + for (j = 0; j < i; j++) { + snprintf(name, sizeof(name), "fd%d", j); + nvlist_add_descriptor(nvl, name, fd); + } + nvlist_send(sock, nvl); + nvlist_destroy(nvl); + + anfds = nopenfds(); + if (anfds == -1) + err(EXIT_FAILURE, "sysctl"); + if (anfds != bnfds) + errx(EXIT_FAILURE, "fd count mismatch"); + } +} + +static void +nvlist_send_recv__send_many_fds(short sotype) +{ + char name[16]; + nvlist_t *nvl; + int anfds, bnfds, fd, i, j, socks[2], status; + pid_t pid; + + ATF_REQUIRE(socketpair(PF_UNIX, sotype, 0, socks) == 0); + + pid = fork(); + ATF_REQUIRE(pid >= 0); + if (pid == 0) { + /* Child. */ + (void)close(socks[0]); + send_many_fds_child(socks[1]); + _exit(0); + } + + (void)close(socks[1]); + + for (i = 1; i < NFDS; i++) { + bnfds = nopenfds(); + ATF_REQUIRE(bnfds != -1); + + nvl = nvlist_recv(socks[0], 0); + ATF_REQUIRE(nvl != NULL); + for (j = 0; j < i; j++) { + snprintf(name, sizeof(name), "fd%d", j); + fd = nvlist_take_descriptor(nvl, name); + ATF_REQUIRE(close(fd) == 0); + } + nvlist_destroy(nvl); + + anfds = nopenfds(); + ATF_REQUIRE(anfds != -1); + ATF_REQUIRE(anfds == bnfds); + } + + ATF_REQUIRE(waitpid(pid, &status, 0) == pid); + ATF_REQUIRE(status == 0); +} + +/* + * This test needs to tune the following sysctl's: + * net.local.dgram.maxdgram + * net.local.dgram.recvspace + */ +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__send_many_fds__dgram); +ATF_TC_BODY(nvlist_send_recv__send_many_fds__dgram, tc) +{ + u_long maxdgram, recvspace, temp_maxdgram, temp_recvspace; + size_t len; + int error; + + atf_tc_skip("https://bugs.freebsd.org/260891"); + + /* size of the largest datagram to send */ + temp_maxdgram = 16772; + len = sizeof(maxdgram); + error = sysctlbyname("net.local.dgram.maxdgram", &maxdgram, + &len, &temp_maxdgram, sizeof(temp_maxdgram)); + if (error != 0) + atf_tc_skip("cannot set net.local.dgram.maxdgram: %s", strerror(errno)); + + /* + * The receive queue fills up quicker than it's being emptied, + * bump it to a sufficiently large enough value, 1M. + */ + temp_recvspace = 1048576; + len = sizeof(recvspace); + error = sysctlbyname("net.local.dgram.recvspace", &recvspace, + &len, &temp_recvspace, sizeof(temp_recvspace)); + if (error != 0) + atf_tc_skip("cannot set net.local.dgram.recvspace: %s", strerror(errno)); + + nvlist_send_recv__send_many_fds(SOCK_DGRAM); + + /* restore original values */ + error = sysctlbyname("net.local.dgram.maxdgram", NULL, NULL, &maxdgram, sizeof(maxdgram)); + if (error != 0) + warn("failed to restore net.local.dgram.maxdgram"); + + error = sysctlbyname("net.local.dgram.recvspace", NULL, NULL, &recvspace, sizeof(recvspace)); + if (error != 0) + warn("failed to restore net.local.dgram.recvspace"); +} + +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__send_many_fds__stream); +ATF_TC_BODY(nvlist_send_recv__send_many_fds__stream, tc) +{ + nvlist_send_recv__send_many_fds(SOCK_STREAM); +} + +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__send_nvlist__dgram); +ATF_TC_BODY(nvlist_send_recv__send_nvlist__dgram, tc) +{ + nvlist_send_recv__send_nvlist(SOCK_DGRAM); +} + +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__send_nvlist__stream); +ATF_TC_BODY(nvlist_send_recv__send_nvlist__stream, tc) +{ + nvlist_send_recv__send_nvlist(SOCK_STREAM); +} + +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__send_closed_fd__dgram); +ATF_TC_BODY(nvlist_send_recv__send_closed_fd__dgram, tc) +{ + nvlist_send_recv__send_closed_fd(SOCK_DGRAM); +} + +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__send_closed_fd__stream); +ATF_TC_BODY(nvlist_send_recv__send_closed_fd__stream, tc) +{ + nvlist_send_recv__send_closed_fd(SOCK_STREAM); +} + +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__overflow_header_size); +ATF_TC_BODY(nvlist_send_recv__overflow_header_size, tc) +{ + nvlist_t *nvl; + void *packed; + size_t packed_size; + struct nvlist_header *header; + int fd, socks[2], status; + pid_t pid; + +#ifdef NO_ASAN + atf_tc_skip("This test requires ASAN"); +#endif + + ATF_REQUIRE_EQ(socketpair(PF_UNIX, SOCK_STREAM, 0, socks), 0); + + pid = fork(); + ATF_REQUIRE(pid >= 0); + + if (pid == 0) { + /* Child. */ + fd = socks[0]; + close(socks[1]); + + nvl = nvlist_create(0); + ATF_REQUIRE(nvl != NULL); + ATF_REQUIRE(nvlist_empty(nvl)); + + packed = nvlist_pack(nvl, &packed_size); + ATF_REQUIRE(packed != NULL); + ATF_REQUIRE(packed_size >= sizeof(struct nvlist_header)); + + header = (struct nvlist_header *)packed; + header->nvlh_size = SIZE_MAX - sizeof(struct nvlist_header) + 2; + + ATF_REQUIRE_EQ(write(fd, packed, packed_size), + (ssize_t)sizeof(struct nvlist_header)); + + nvlist_destroy(nvl); + free(packed); + + exit(0); + } else { + /* Parent */ + fd = socks[1]; + close(socks[0]); + + errno = 0; + nvl = nvlist_recv(fd, 0); + ATF_REQUIRE(nvl == NULL); + + /* + * Make sure it has failed on EINVAL, and not on + * errors returned by malloc or recv. + */ + ATF_REQUIRE(errno == EINVAL); + + ATF_REQUIRE(waitpid(pid, &status, 0) == pid); + ATF_REQUIRE(status == 0); + close(fd); + } +} + +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__invalid_fd_size); +ATF_TC_BODY(nvlist_send_recv__invalid_fd_size, tc) +{ + nvlist_t *nvl; + void *packed; + size_t packed_size; + struct nvlist_header *header; + int fd, socks[2], status; + pid_t pid; + + ATF_REQUIRE_EQ(socketpair(PF_UNIX, SOCK_STREAM, 0, socks), 0); + + pid = fork(); + ATF_REQUIRE(pid >= 0); + + if (pid == 0) { + /* Child. */ + fd = socks[0]; + close(socks[1]); + + nvl = nvlist_create(0); + ATF_REQUIRE(nvl != NULL); + ATF_REQUIRE(nvlist_empty(nvl)); + + nvlist_add_string(nvl, "nvl/string", "test"); + ATF_REQUIRE_EQ(nvlist_error(nvl), 0); + + packed = nvlist_pack(nvl, &packed_size); + ATF_REQUIRE(packed != NULL); + ATF_REQUIRE(packed_size >= sizeof(struct nvlist_header)); + + header = (struct nvlist_header *)packed; + header->nvlh_descriptors = 0x20; + + ATF_REQUIRE_EQ(write(fd, packed, packed_size), + (ssize_t)packed_size); + + nvlist_destroy(nvl); + free(packed); + + exit(0); + } else { + /* Parent */ + fd = socks[1]; + close(socks[0]); + + nvl = nvlist_recv(fd, 0); + ATF_REQUIRE(nvl == NULL); + + ATF_REQUIRE(waitpid(pid, &status, 0) == pid); + ATF_REQUIRE(status == 0); + } + + close(fd); +} + +ATF_TC_WITHOUT_HEAD(nvlist_send_recv__overflow_fd_size); +ATF_TC_BODY(nvlist_send_recv__overflow_fd_size, tc) +{ + nvlist_t *nvl; + void *packed; + size_t packed_size; + struct nvlist_header *header; + int fd, socks[2], fds[1], status; + pid_t pid; + + ATF_REQUIRE_EQ(socketpair(PF_UNIX, SOCK_STREAM, 0, socks), 0); + + pid = fork(); + ATF_REQUIRE(pid >= 0); + + if (pid == 0) { + /* Child. */ + fd = socks[0]; + close(socks[1]); + + nvl = nvlist_create(0); + ATF_REQUIRE(nvl != NULL); + ATF_REQUIRE(nvlist_empty(nvl)); + + nvlist_add_string(nvl, "nvl/string", "test"); + ATF_REQUIRE_EQ(nvlist_error(nvl), 0); + + packed = nvlist_pack(nvl, &packed_size); + ATF_REQUIRE(packed != NULL); + ATF_REQUIRE(packed_size >= sizeof(struct nvlist_header)); + + header = (struct nvlist_header *)packed; + header->nvlh_descriptors = 0x4000000000000002; + + ATF_REQUIRE_EQ(write(fd, packed, packed_size), + (ssize_t)packed_size); + + fds[0] = dup(STDERR_FILENO); + ATF_REQUIRE(fds[0] >= 0); + ATF_REQUIRE_EQ(fd_send(fd, fds, 1), 0); + + nvlist_destroy(nvl); + free(packed); + + close(fds[0]); + close(fd); + + exit(0); + } else { + /* Parent */ + fd = socks[1]; + close(socks[0]); + + nvl = nvlist_recv(fd, 0); + ATF_REQUIRE(nvl == NULL); + + /* Make sure that fd was not parsed by nvlist */ + ATF_REQUIRE(fd_recv(fd, fds, 1) == 0); + + ATF_REQUIRE(waitpid(pid, &status, 0) == pid); + ATF_REQUIRE(status == 0); + + close(fds[0]); + close(fd); + } +} + +ATF_TP_ADD_TCS(tp) +{ + + ATF_TP_ADD_TC(tp, nvlist_send_recv__send_nvlist__dgram); + ATF_TP_ADD_TC(tp, nvlist_send_recv__send_nvlist__stream); + ATF_TP_ADD_TC(tp, nvlist_send_recv__send_closed_fd__dgram); + ATF_TP_ADD_TC(tp, nvlist_send_recv__send_closed_fd__stream); + ATF_TP_ADD_TC(tp, nvlist_send_recv__send_many_fds__dgram); + ATF_TP_ADD_TC(tp, nvlist_send_recv__send_many_fds__stream); + + ATF_TP_ADD_TC(tp, nvlist_send_recv__overflow_header_size); + ATF_TP_ADD_TC(tp, nvlist_send_recv__invalid_fd_size); + ATF_TP_ADD_TC(tp, nvlist_send_recv__overflow_fd_size); + + return (atf_no_error()); +} |