tests: extend TEST_NETLINK_OBJECT macro

[strace] / tests / scm_rights.c

diff --git a/tests/scm_rights.c b/tests/scm_rights.c
index c41444fa953e15ed78bc06447f5a44772ef0498b..632cf91dfc4d732370d564b1d435490c0b9b5323 100644 (file)
--- a/tests/scm_rights.c
+++ b/tests/scm_rights.c
@@ -1,5 +1,6 @@
 /*
- * Copyright (c) 2014-2015 Dmitry V. Levin <ldv@altlinux.org>
+ * Copyright (c) 2014-2016 Dmitry V. Levin <ldv@altlinux.org>
+ * Copyright (c) 2014-2017 The strace developers.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -25,86 +26,81 @@
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
+#include "tests.h"
 #include <assert.h>
-#include <string.h>
+#include <fcntl.h>
 #include <stdlib.h>
+#include <string.h>
 #include <unistd.h>
-#include <errno.h>
-#include <fcntl.h>
 #include <sys/socket.h>
-#include <sys/wait.h>
 
 int main(int ac, const char **av)
 {
-       int i;
-       int data = 0;
-       struct iovec iov = {
-               .iov_base = &data,
-               .iov_len = sizeof(iov)
-       };
+       assert(ac > 0);
+       int fds[ac];
 
-       while ((i = open("/dev/null", O_RDWR)) < 3)
+       static const char sample[] =
+               "\xf1\xf2\xf3\xf4\xf5\xf6\xf7\xf8\xf9\xfa\xfb\xfc\xfd\xfe\xff";
+       const unsigned int data_size = sizeof(sample) - 1;
+       void *data = tail_alloc(data_size);
+       memcpy(data, sample, data_size);
+
+       TAIL_ALLOC_OBJECT_CONST_PTR(struct iovec, iov);
+       iov->iov_base = data;
+       iov->iov_len = data_size;
+
+       TAIL_ALLOC_OBJECT_CONST_PTR(struct msghdr, mh);
+       memset(mh, 0, sizeof(*mh));
+       mh->msg_iov = iov;
+       mh->msg_iovlen = 1;
+
+       int i;
+       while ((i = open("/dev/null", O_RDWR)) <= ac + 2)
                assert(i >= 0);
-       (void) close(3);
+       while (i > 2)
+               assert(close(i--) == 0);
+       assert(close(0) == 0);
 
        int sv[2];
-       assert(socketpair(AF_UNIX, SOCK_STREAM, 0, sv) == 0);
+       if (socketpair(AF_UNIX, SOCK_STREAM, 0, sv))
+               perror_msg_and_skip("socketpair");
        int one = 1;
-       assert(setsockopt(sv[0], SOL_SOCKET, SO_PASSCRED, &one, sizeof(one)) == 0);
-
-       pid_t pid = fork();
-       assert(pid >= 0);
-
-       if (pid) {
-               assert(close(sv[0]) == 0);
-               assert(dup2(sv[1], 1) == 1);
-               assert(close(sv[1]) == 0);
-
-               int fds[ac];
-               assert((fds[0] = open("/dev/null", O_RDWR)) == 3);
-               for (i = 1; i < ac; ++i)
-                       assert((fds[i] = open(av[i], O_RDONLY)) == i + 3);
-
-               union {
-                       struct cmsghdr cmsg;
-                       char buf[CMSG_LEN(sizeof(fds))];
-               } control;
-
-               control.cmsg.cmsg_level = SOL_SOCKET;
-               control.cmsg.cmsg_type = SCM_RIGHTS;
-               control.cmsg.cmsg_len = CMSG_LEN(sizeof(fds));
-               memcpy(CMSG_DATA(&control.cmsg), fds, sizeof(fds));
-
-               struct msghdr mh = {
-                       .msg_iov = &iov,
-                       .msg_iovlen = 1,
-                       .msg_control = &control,
-                       .msg_controllen = sizeof(control)
-               };
-
-               assert(sendmsg(1, &mh, 0) == sizeof(iov));
-               assert(close(1) == 0);
-
-                int status;
-               assert(waitpid(pid, &status, 0) == pid);
-               assert(status == 0);
-       } else {
-               assert(close(sv[1]) == 0);
-               assert(dup2(sv[0], 0) == 0);
-               assert(close(sv[0]) == 0);
-
-               struct cmsghdr control[4 + ac * sizeof(int) / sizeof(struct cmsghdr)];
-
-               struct msghdr mh = {
-                       .msg_iov = &iov,
-                       .msg_iovlen = 1,
-                       .msg_control = control,
-                       .msg_controllen = sizeof(control)
-               };
-
-               assert(recvmsg(0, &mh, 0) == sizeof(iov));
-               assert(close(0) == 0);
+       if (setsockopt(sv[0], SOL_SOCKET, SO_PASSCRED, &one, sizeof(one)))
+               perror_msg_and_skip("setsockopt");
+
+       assert((fds[0] = open("/dev/null", O_RDWR)) == 4);
+       for (i = 1; i < ac; ++i)
+               assert((fds[i] = open(av[i], O_RDONLY)) == i + 4);
+
+       unsigned int cmsg_size = CMSG_SPACE(sizeof(fds));
+       struct cmsghdr *cmsg = tail_alloc(cmsg_size);
+       memset(cmsg, 0, cmsg_size);
+       cmsg->cmsg_level = SOL_SOCKET;
+       cmsg->cmsg_type = SCM_RIGHTS;
+       cmsg->cmsg_len = CMSG_LEN(sizeof(fds));
+       memcpy(CMSG_DATA(cmsg), fds, sizeof(fds));
+
+       mh->msg_control = cmsg;
+       mh->msg_controllen = cmsg_size;
+
+       assert(sendmsg(sv[1], mh, 0) == (int) data_size);
+
+       assert(close(sv[1]) == 0);
+       assert(open("/dev/null", O_RDWR) == sv[1]);
+
+       for (i = 0; i < ac; ++i) {
+               assert(close(fds[i]) == 0);
+               fds[i] = 0;
        }
 
+       cmsg_size += CMSG_SPACE(sizeof(struct ucred));
+       cmsg = tail_alloc(cmsg_size);
+       memset(cmsg, 0, cmsg_size);
+       mh->msg_control = cmsg;
+       mh->msg_controllen = cmsg_size;
+
+       assert(recvmsg(0, mh, 0) == (int) data_size);
+       assert(close(0) == 0);
+
        return 0;
 }