]>
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> |
e37f390b | 37 | #ifdef HAVE_POLL_H |
e8e24c80 | 38 | #include <poll.h> |
e37f390b | 39 | #endif |
00146caa | 40 | #include <string.h> |
31652869 | 41 | #include <stdarg.h> |
028094f4 | 42 | |
1853d1ef | 43 | #include "log.h" |
44 | #include "monitor_fdpass.h" | |
45 | ||
8de7aaab | 46 | int |
ca75d7de | 47 | mm_send_fd(int sock, int fd) |
1853d1ef | 48 | { |
1b06e75e | 49 | #if defined(HAVE_SENDMSG) && (defined(HAVE_ACCRIGHTS_IN_MSGHDR) || defined(HAVE_CONTROL_IN_MSGHDR)) |
1853d1ef | 50 | struct msghdr msg; |
637f9177 | 51 | #ifndef HAVE_ACCRIGHTS_IN_MSGHDR |
0c9a63f1 | 52 | union { |
53 | struct cmsghdr hdr; | |
d1978bb4 | 54 | char buf[CMSG_SPACE(sizeof(int))]; |
55 | } cmsgbuf; | |
dc90b259 | 56 | struct cmsghdr *cmsg; |
57 | #endif | |
d2aa725a | 58 | struct iovec vec; |
59 | char ch = '\0'; | |
60 | ssize_t n; | |
e8e24c80 | 61 | struct pollfd pfd; |
1853d1ef | 62 | |
63 | memset(&msg, 0, sizeof(msg)); | |
637f9177 | 64 | #ifdef HAVE_ACCRIGHTS_IN_MSGHDR |
dc90b259 | 65 | msg.msg_accrights = (caddr_t)&fd; |
66 | msg.msg_accrightslen = sizeof(fd); | |
67 | #else | |
d1978bb4 | 68 | msg.msg_control = (caddr_t)&cmsgbuf.buf; |
cab36661 | 69 | msg.msg_controllen = sizeof(cmsgbuf.buf); |
1853d1ef | 70 | cmsg = CMSG_FIRSTHDR(&msg); |
71 | cmsg->cmsg_len = CMSG_LEN(sizeof(int)); | |
72 | cmsg->cmsg_level = SOL_SOCKET; | |
73 | cmsg->cmsg_type = SCM_RIGHTS; | |
74 | *(int *)CMSG_DATA(cmsg) = fd; | |
dc90b259 | 75 | #endif |
1853d1ef | 76 | |
77 | vec.iov_base = &ch; | |
78 | vec.iov_len = 1; | |
79 | msg.msg_iov = &vec; | |
80 | msg.msg_iovlen = 1; | |
81 | ||
e8e24c80 | 82 | pfd.fd = sock; |
83 | pfd.events = POLLOUT; | |
84 | while ((n = sendmsg(sock, &msg, 0)) == -1 && | |
85 | (errno == EAGAIN || errno == EINTR)) { | |
8e10da10 | 86 | debug3("%s: sendmsg(%d): %s", __func__, fd, strerror(errno)); |
e8e24c80 | 87 | (void)poll(&pfd, 1, -1); |
88 | } | |
8e10da10 | 89 | if (n == -1) { |
8de7aaab | 90 | error("%s: sendmsg(%d): %s", __func__, fd, |
d4355079 | 91 | strerror(errno)); |
8de7aaab | 92 | return -1; |
93 | } | |
94 | ||
95 | if (n != 1) { | |
96 | error("%s: sendmsg: expected sent 1 got %ld", | |
63b5f1a1 | 97 | __func__, (long)n); |
8de7aaab | 98 | return -1; |
99 | } | |
100 | return 0; | |
84071420 | 101 | #else |
8de7aaab | 102 | error("%s: file descriptor passing not supported", __func__); |
103 | return -1; | |
84071420 | 104 | #endif |
1853d1ef | 105 | } |
106 | ||
107 | int | |
ca75d7de | 108 | mm_receive_fd(int sock) |
1853d1ef | 109 | { |
1b06e75e | 110 | #if defined(HAVE_RECVMSG) && (defined(HAVE_ACCRIGHTS_IN_MSGHDR) || defined(HAVE_CONTROL_IN_MSGHDR)) |
1853d1ef | 111 | struct msghdr msg; |
637f9177 | 112 | #ifndef HAVE_ACCRIGHTS_IN_MSGHDR |
0c9a63f1 | 113 | union { |
0c9a63f1 | 114 | struct cmsghdr hdr; |
d1978bb4 | 115 | char buf[CMSG_SPACE(sizeof(int))]; |
116 | } cmsgbuf; | |
dc90b259 | 117 | struct cmsghdr *cmsg; |
dc90b259 | 118 | #endif |
d2aa725a | 119 | struct iovec vec; |
120 | ssize_t n; | |
121 | char ch; | |
122 | int fd; | |
e8e24c80 | 123 | struct pollfd pfd; |
1853d1ef | 124 | |
125 | memset(&msg, 0, sizeof(msg)); | |
126 | vec.iov_base = &ch; | |
127 | vec.iov_len = 1; | |
128 | msg.msg_iov = &vec; | |
129 | msg.msg_iovlen = 1; | |
637f9177 | 130 | #ifdef HAVE_ACCRIGHTS_IN_MSGHDR |
dc90b259 | 131 | msg.msg_accrights = (caddr_t)&fd; |
132 | msg.msg_accrightslen = sizeof(fd); | |
133 | #else | |
d1978bb4 | 134 | msg.msg_control = &cmsgbuf.buf; |
cab36661 | 135 | msg.msg_controllen = sizeof(cmsgbuf.buf); |
dc90b259 | 136 | #endif |
1853d1ef | 137 | |
e8e24c80 | 138 | pfd.fd = sock; |
139 | pfd.events = POLLIN; | |
140 | while ((n = recvmsg(sock, &msg, 0)) == -1 && | |
141 | (errno == EAGAIN || errno == EINTR)) { | |
8e10da10 | 142 | debug3("%s: recvmsg: %s", __func__, strerror(errno)); |
e8e24c80 | 143 | (void)poll(&pfd, 1, -1); |
144 | } | |
8e10da10 | 145 | if (n == -1) { |
8de7aaab | 146 | error("%s: recvmsg: %s", __func__, strerror(errno)); |
147 | return -1; | |
148 | } | |
d2aa725a | 149 | |
8de7aaab | 150 | if (n != 1) { |
151 | error("%s: recvmsg: expected received 1 got %ld", | |
63b5f1a1 | 152 | __func__, (long)n); |
8de7aaab | 153 | return -1; |
154 | } | |
1853d1ef | 155 | |
637f9177 | 156 | #ifdef HAVE_ACCRIGHTS_IN_MSGHDR |
8de7aaab | 157 | if (msg.msg_accrightslen != sizeof(fd)) { |
158 | error("%s: no fd", __func__); | |
159 | return -1; | |
160 | } | |
dc90b259 | 161 | #else |
1853d1ef | 162 | cmsg = CMSG_FIRSTHDR(&msg); |
8de7aaab | 163 | if (cmsg == NULL) { |
164 | error("%s: no message header", __func__); | |
165 | return -1; | |
166 | } | |
d2aa725a | 167 | |
bf7c1e6c | 168 | #ifndef BROKEN_CMSG_TYPE |
8de7aaab | 169 | if (cmsg->cmsg_type != SCM_RIGHTS) { |
170 | error("%s: expected type %d got %d", __func__, | |
1853d1ef | 171 | SCM_RIGHTS, cmsg->cmsg_type); |
8de7aaab | 172 | return -1; |
173 | } | |
bf7c1e6c | 174 | #endif |
d4355079 | 175 | fd = (*(int *)CMSG_DATA(cmsg)); |
dc90b259 | 176 | #endif |
d4355079 | 177 | return fd; |
84071420 | 178 | #else |
8de7aaab | 179 | error("%s: file descriptor passing not supported", __func__); |
180 | return -1; | |
84071420 | 181 | #endif |
1853d1ef | 182 | } |