]>
Commit | Line | Data |
---|---|---|
e8e24c80 | 1 | /* $OpenBSD: monitor_fdpass.c,v 1.19 2010/01/12 00:58:25 djm Exp $ */ |
1853d1ef | 2 | /* |
3 | * Copyright 2001 Niels Provos <provos@citi.umich.edu> | |
4 | * All rights reserved. | |
5 | * | |
6 | * Redistribution and use in source and binary forms, with or without | |
7 | * modification, are permitted provided that the following conditions | |
8 | * are met: | |
9 | * 1. Redistributions of source code must retain the above copyright | |
10 | * notice, this list of conditions and the following disclaimer. | |
11 | * 2. Redistributions in binary form must reproduce the above copyright | |
12 | * notice, this list of conditions and the following disclaimer in the | |
13 | * documentation and/or other materials provided with the distribution. | |
14 | * | |
15 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR | |
16 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES | |
17 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. | |
18 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, | |
19 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT | |
20 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | |
21 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | |
22 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
23 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF | |
24 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
25 | */ | |
26 | ||
27 | #include "includes.h" | |
1853d1ef | 28 | |
5b04a8bf | 29 | #include <sys/types.h> |
30 | #include <sys/socket.h> | |
1853d1ef | 31 | #include <sys/uio.h> |
8153fef1 | 32 | #ifdef HAVE_SYS_UN_H |
33 | #include <sys/un.h> | |
34 | #endif | |
1853d1ef | 35 | |
028094f4 | 36 | #include <errno.h> |
e8e24c80 | 37 | #include <poll.h> |
00146caa | 38 | #include <string.h> |
31652869 | 39 | #include <stdarg.h> |
028094f4 | 40 | |
1853d1ef | 41 | #include "log.h" |
42 | #include "monitor_fdpass.h" | |
43 | ||
8de7aaab | 44 | int |
ca75d7de | 45 | mm_send_fd(int sock, int fd) |
1853d1ef | 46 | { |
1b06e75e | 47 | #if defined(HAVE_SENDMSG) && (defined(HAVE_ACCRIGHTS_IN_MSGHDR) || defined(HAVE_CONTROL_IN_MSGHDR)) |
1853d1ef | 48 | struct msghdr msg; |
637f9177 | 49 | #ifndef HAVE_ACCRIGHTS_IN_MSGHDR |
0c9a63f1 | 50 | union { |
51 | struct cmsghdr hdr; | |
d1978bb4 | 52 | char buf[CMSG_SPACE(sizeof(int))]; |
53 | } cmsgbuf; | |
dc90b259 | 54 | struct cmsghdr *cmsg; |
55 | #endif | |
d2aa725a | 56 | struct iovec vec; |
57 | char ch = '\0'; | |
58 | ssize_t n; | |
e8e24c80 | 59 | struct pollfd pfd; |
1853d1ef | 60 | |
61 | memset(&msg, 0, sizeof(msg)); | |
637f9177 | 62 | #ifdef HAVE_ACCRIGHTS_IN_MSGHDR |
dc90b259 | 63 | msg.msg_accrights = (caddr_t)&fd; |
64 | msg.msg_accrightslen = sizeof(fd); | |
65 | #else | |
d1978bb4 | 66 | msg.msg_control = (caddr_t)&cmsgbuf.buf; |
cab36661 | 67 | msg.msg_controllen = sizeof(cmsgbuf.buf); |
1853d1ef | 68 | cmsg = CMSG_FIRSTHDR(&msg); |
69 | cmsg->cmsg_len = CMSG_LEN(sizeof(int)); | |
70 | cmsg->cmsg_level = SOL_SOCKET; | |
71 | cmsg->cmsg_type = SCM_RIGHTS; | |
72 | *(int *)CMSG_DATA(cmsg) = fd; | |
dc90b259 | 73 | #endif |
1853d1ef | 74 | |
75 | vec.iov_base = &ch; | |
76 | vec.iov_len = 1; | |
77 | msg.msg_iov = &vec; | |
78 | msg.msg_iovlen = 1; | |
79 | ||
e8e24c80 | 80 | pfd.fd = sock; |
81 | pfd.events = POLLOUT; | |
82 | while ((n = sendmsg(sock, &msg, 0)) == -1 && | |
83 | (errno == EAGAIN || errno == EINTR)) { | |
8e10da10 | 84 | debug3("%s: sendmsg(%d): %s", __func__, fd, strerror(errno)); |
e8e24c80 | 85 | (void)poll(&pfd, 1, -1); |
86 | } | |
8e10da10 | 87 | if (n == -1) { |
8de7aaab | 88 | error("%s: sendmsg(%d): %s", __func__, fd, |
d4355079 | 89 | strerror(errno)); |
8de7aaab | 90 | return -1; |
91 | } | |
92 | ||
93 | if (n != 1) { | |
94 | error("%s: sendmsg: expected sent 1 got %ld", | |
63b5f1a1 | 95 | __func__, (long)n); |
8de7aaab | 96 | return -1; |
97 | } | |
98 | return 0; | |
84071420 | 99 | #else |
8de7aaab | 100 | error("%s: file descriptor passing not supported", __func__); |
101 | return -1; | |
84071420 | 102 | #endif |
1853d1ef | 103 | } |
104 | ||
105 | int | |
ca75d7de | 106 | mm_receive_fd(int sock) |
1853d1ef | 107 | { |
1b06e75e | 108 | #if defined(HAVE_RECVMSG) && (defined(HAVE_ACCRIGHTS_IN_MSGHDR) || defined(HAVE_CONTROL_IN_MSGHDR)) |
1853d1ef | 109 | struct msghdr msg; |
637f9177 | 110 | #ifndef HAVE_ACCRIGHTS_IN_MSGHDR |
0c9a63f1 | 111 | union { |
0c9a63f1 | 112 | struct cmsghdr hdr; |
d1978bb4 | 113 | char buf[CMSG_SPACE(sizeof(int))]; |
114 | } cmsgbuf; | |
dc90b259 | 115 | struct cmsghdr *cmsg; |
dc90b259 | 116 | #endif |
d2aa725a | 117 | struct iovec vec; |
118 | ssize_t n; | |
119 | char ch; | |
120 | int fd; | |
e8e24c80 | 121 | struct pollfd pfd; |
1853d1ef | 122 | |
123 | memset(&msg, 0, sizeof(msg)); | |
124 | vec.iov_base = &ch; | |
125 | vec.iov_len = 1; | |
126 | msg.msg_iov = &vec; | |
127 | msg.msg_iovlen = 1; | |
637f9177 | 128 | #ifdef HAVE_ACCRIGHTS_IN_MSGHDR |
dc90b259 | 129 | msg.msg_accrights = (caddr_t)&fd; |
130 | msg.msg_accrightslen = sizeof(fd); | |
131 | #else | |
d1978bb4 | 132 | msg.msg_control = &cmsgbuf.buf; |
cab36661 | 133 | msg.msg_controllen = sizeof(cmsgbuf.buf); |
dc90b259 | 134 | #endif |
1853d1ef | 135 | |
e8e24c80 | 136 | pfd.fd = sock; |
137 | pfd.events = POLLIN; | |
138 | while ((n = recvmsg(sock, &msg, 0)) == -1 && | |
139 | (errno == EAGAIN || errno == EINTR)) { | |
8e10da10 | 140 | debug3("%s: recvmsg: %s", __func__, strerror(errno)); |
e8e24c80 | 141 | (void)poll(&pfd, 1, -1); |
142 | } | |
8e10da10 | 143 | if (n == -1) { |
8de7aaab | 144 | error("%s: recvmsg: %s", __func__, strerror(errno)); |
145 | return -1; | |
146 | } | |
d2aa725a | 147 | |
8de7aaab | 148 | if (n != 1) { |
149 | error("%s: recvmsg: expected received 1 got %ld", | |
63b5f1a1 | 150 | __func__, (long)n); |
8de7aaab | 151 | return -1; |
152 | } | |
1853d1ef | 153 | |
637f9177 | 154 | #ifdef HAVE_ACCRIGHTS_IN_MSGHDR |
8de7aaab | 155 | if (msg.msg_accrightslen != sizeof(fd)) { |
156 | error("%s: no fd", __func__); | |
157 | return -1; | |
158 | } | |
dc90b259 | 159 | #else |
1853d1ef | 160 | cmsg = CMSG_FIRSTHDR(&msg); |
8de7aaab | 161 | if (cmsg == NULL) { |
162 | error("%s: no message header", __func__); | |
163 | return -1; | |
164 | } | |
d2aa725a | 165 | |
bf7c1e6c | 166 | #ifndef BROKEN_CMSG_TYPE |
8de7aaab | 167 | if (cmsg->cmsg_type != SCM_RIGHTS) { |
168 | error("%s: expected type %d got %d", __func__, | |
1853d1ef | 169 | SCM_RIGHTS, cmsg->cmsg_type); |
8de7aaab | 170 | return -1; |
171 | } | |
bf7c1e6c | 172 | #endif |
d4355079 | 173 | fd = (*(int *)CMSG_DATA(cmsg)); |
dc90b259 | 174 | #endif |
d4355079 | 175 | return fd; |
84071420 | 176 | #else |
8de7aaab | 177 | error("%s: file descriptor passing not supported", __func__); |
178 | return -1; | |
84071420 | 179 | #endif |
1853d1ef | 180 | } |