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