tools/regression/sockets/unix_cmsg/t_cmsgcred.c

   1 /*-
   2  * Copyright (c) 2005 Andrey Simonenko
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  */
  26
  27 #include <sys/cdefs.h>
  28 __FBSDID("$FreeBSD$");
  29
  30 #include <sys/types.h>
  31 #include <sys/socket.h>
  32 #include <sys/un.h>
  33 #include <inttypes.h>
  34 #include <stdarg.h>
  35 #include <stdbool.h>
  36 #include <stdlib.h>
  37
  38 #include "uc_common.h"
  39 #include "t_generic.h"
  40 #include "t_cmsgcred.h"
  41
  42 int
  43 t_cmsgcred_client(int fd)
  44 {
  45         struct msghdr msghdr;
  46         struct iovec iov[1];
  47         void *cmsg_data;
  48         size_t cmsg_size;
  49         int rv;
  50
  51         if (uc_sync_recv() < 0)
  52                 return (-2);
  53
  54         rv = -2;
  55
  56         cmsg_size = CMSG_SPACE(sizeof(struct cmsgcred));
  57         cmsg_data = malloc(cmsg_size);
  58         if (cmsg_data == NULL) {
  59                 uc_logmsg("malloc");
  60                 goto done;
  61         }
  62         uc_msghdr_init_client(&msghdr, iov, cmsg_data, cmsg_size,
  63             SCM_CREDS, sizeof(struct cmsgcred));
  64
  65         if (uc_socket_connect(fd) < 0)
  66                 goto done;
  67
  68         if (uc_message_sendn(fd, &msghdr) < 0)
  69                 goto done;
  70
  71         rv = 0;
  72 done:
  73         free(cmsg_data);
  74         return (rv);
  75 }
  76
  77 static int
  78 t_cmsgcred_server(int fd1)
  79 {
  80         struct msghdr msghdr;
  81         struct iovec iov[1];
  82         struct cmsghdr *cmsghdr;
  83         void *cmsg_data;
  84         size_t cmsg_size;
  85         u_int i;
  86         int fd2, rv;
  87
  88         if (uc_sync_send() < 0)
  89                 return (-2);
  90
  91         fd2 = -1;
  92         rv = -2;
  93
  94         cmsg_size = CMSG_SPACE(sizeof(struct cmsgcred));
  95         cmsg_data = malloc(cmsg_size);
  96         if (cmsg_data == NULL) {
  97                 uc_logmsg("malloc");
  98                 goto done;
  99         }
 100
 101         if (uc_cfg.sock_type == SOCK_STREAM) {
 102                 fd2 = uc_socket_accept(fd1);
 103                 if (fd2 < 0)
 104                         goto done;
 105         } else
 106                 fd2 = fd1;
 107
 108         rv = -1;
 109         for (i = 1; i <= uc_cfg.ipc_msg.msg_num; ++i) {
 110                 uc_dbgmsg("message #%u", i);
 111
 112                 uc_msghdr_init_server(&msghdr, iov, cmsg_data, cmsg_size);
 113                 if (uc_message_recv(fd2, &msghdr) < 0) {
 114                         rv = -2;
 115                         break;
 116                 }
 117
 118                 if (uc_check_msghdr(&msghdr, sizeof(*cmsghdr)) < 0)
 119                         break;
 120
 121                 cmsghdr = CMSG_FIRSTHDR(&msghdr);
 122                 if (uc_check_scm_creds_cmsgcred(cmsghdr) < 0)
 123                         break;
 124         }
 125         if (i > uc_cfg.ipc_msg.msg_num)
 126                 rv = 0;
 127 done:
 128         free(cmsg_data);
 129         if (uc_cfg.sock_type == SOCK_STREAM && fd2 >= 0)
 130                 if (uc_socket_close(fd2) < 0)
 131                         rv = -2;
 132         return (rv);
 133 }
 134
 135 int
 136 t_cmsgcred(void)
 137 {
 138         return (t_generic(t_cmsgcred_client, t_cmsgcred_server));
 139 }