granicus.if.org Git - strace/blob - tests/netlink_unix_diag.c

   1 /*
   2  * This file is part of net-yy-unix strace test.
   3  *
   4  * Copyright (c) 2014-2016 Dmitry V. Levin <ldv@altlinux.org>
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. The name of the author may not be used to endorse or promote products
  16  *    derived from this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  19  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  20  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  21  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  22  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  23  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #include "tests.h"
  31 #include <assert.h>
  32 #include <errno.h>
  33 #include <stddef.h>
  34 #include <stdint.h>
  35 #include <string.h>
  36 #include <unistd.h>
  37 #include <sys/socket.h>
  38 #include <sys/un.h>
  39 #include <linux/netlink.h>
  40 #include <linux/sock_diag.h>
  41 #include <linux/unix_diag.h>
  42
  43 #if !defined NETLINK_SOCK_DIAG && defined NETLINK_INET_DIAG
  44 # define NETLINK_SOCK_DIAG NETLINK_INET_DIAG
  45 #endif
  46
  47 static void
  48 send_query(const int fd)
  49 {
  50         struct sockaddr_nl nladdr = {
  51                 .nl_family = AF_NETLINK
  52         };
  53         struct {
  54                 struct nlmsghdr nlh;
  55                 struct unix_diag_req udr;
  56         } req = {
  57                 .nlh = {
  58                         .nlmsg_len = sizeof(req),
  59                         .nlmsg_type = SOCK_DIAG_BY_FAMILY,
  60                         .nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP
  61                 },
  62                 .udr = {
  63                         .sdiag_family = AF_UNIX,
  64                         .udiag_states = -1,
  65                         .udiag_show = UDIAG_SHOW_NAME | UDIAG_SHOW_PEER
  66                 }
  67         };
  68         struct iovec iov = {
  69                 .iov_base = &req,
  70                 .iov_len = sizeof(req)
  71         };
  72         struct msghdr msg = {
  73                 .msg_name = (void *) &nladdr,
  74                 .msg_namelen = sizeof(nladdr),
  75                 .msg_iov = &iov,
  76                 .msg_iovlen = 1
  77         };
  78
  79         if (sendmsg(fd, &msg, 0) <= 0)
  80                 perror_msg_and_skip("sendmsg");
  81 }
  82
  83 static void
  84 check_responses(const int fd)
  85 {
  86         static union {
  87                 struct nlmsghdr hdr;
  88                 long buf[8192 / sizeof(long)];
  89         } hdr_buf;
  90
  91         struct sockaddr_nl nladdr = {
  92                 .nl_family = AF_NETLINK
  93         };
  94         struct iovec iov = {
  95                 .iov_base = hdr_buf.buf,
  96                 .iov_len = sizeof(hdr_buf.buf)
  97         };
  98         struct msghdr msg = {
  99                 .msg_name = (void *) &nladdr,
 100                 .msg_namelen = sizeof(nladdr),
 101                 .msg_iov = &iov,
 102                 .msg_iovlen = 1
 103         };
 104
 105         ssize_t ret = recvmsg(fd, &msg, 0);
 106         if (ret <= 0)
 107                 perror_msg_and_skip("recvmsg");
 108
 109         struct nlmsghdr *h = &hdr_buf.hdr;
 110         if (!NLMSG_OK(h, ret))
 111                 error_msg_and_skip("!NLMSG_OK");
 112         if (h->nlmsg_type == NLMSG_ERROR) {
 113                 const struct nlmsgerr *err = NLMSG_DATA(h);
 114                 if (h->nlmsg_len < NLMSG_LENGTH(sizeof(*err)))
 115                         error_msg_and_skip("NLMSG_ERROR");
 116                 errno = -err->error;
 117                 perror_msg_and_skip("NLMSG_ERROR");
 118         }
 119         if (h->nlmsg_type != SOCK_DIAG_BY_FAMILY)
 120                 error_msg_and_skip("unexpected nlmsg_type %u",
 121                                    (unsigned) h->nlmsg_type);
 122
 123         const struct unix_diag_msg *diag = NLMSG_DATA(h);
 124         if (h->nlmsg_len < NLMSG_LENGTH(sizeof(*diag)))
 125                 error_msg_and_skip("short response");
 126 }
 127
 128 #define SUN_PATH "netlink_unix_diag_socket"
 129 int main(void)
 130 {
 131         struct sockaddr_un addr = {
 132                 .sun_family = AF_UNIX,
 133                 .sun_path = SUN_PATH
 134         };
 135         socklen_t len = offsetof(struct sockaddr_un, sun_path) + sizeof(SUN_PATH);
 136
 137         close(0);
 138         close(1);
 139
 140         (void) unlink(SUN_PATH);
 141         if (socket(AF_UNIX, SOCK_STREAM, 0))
 142                 perror_msg_and_skip("socket AF_UNIX");
 143         if (bind(0, (struct sockaddr *) &addr, len))
 144                 perror_msg_and_skip("bind");
 145         if (listen(0, 5))
 146                 perror_msg_and_skip("listen");
 147
 148         assert(unlink(SUN_PATH) == 0);
 149
 150         if (socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG) != 1)
 151                 perror_msg_and_skip("socket AF_NETLINK");
 152
 153         send_query(1);
 154         check_responses(1);
 155         return 0;
 156 }