2 * Unix SMB/CIFS implementation.
3 * Copyright (C) Volker Lendecke 2014
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 3 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
20 #include "lib/util/msghdr.h"
21 #include "lib/util/iov_buf.h"
22 #include <sys/socket.h>
24 #if defined(HAVE_STRUCT_MSGHDR_MSG_CONTROL)
26 ssize_t msghdr_prep_fds(struct msghdr *msg, uint8_t *buf, size_t bufsize,
27 const int *fds, size_t num_fds)
29 size_t fds_size = sizeof(int) * MIN(num_fds, INT8_MAX);
30 size_t cmsg_len = CMSG_LEN(fds_size);
31 size_t cmsg_space = CMSG_SPACE(fds_size);
37 msg->msg_control = NULL;
38 msg->msg_controllen = 0;
42 if (num_fds > INT8_MAX) {
45 if ((msg == NULL) || (cmsg_space > bufsize)) {
49 msg->msg_control = buf;
50 msg->msg_controllen = cmsg_space;
52 cmsg = CMSG_FIRSTHDR(msg);
53 cmsg->cmsg_level = SOL_SOCKET;
54 cmsg->cmsg_type = SCM_RIGHTS;
55 cmsg->cmsg_len = cmsg_len;
56 fdptr = CMSG_DATA(cmsg);
57 memcpy(fdptr, fds, fds_size);
58 msg->msg_controllen = cmsg->cmsg_len;
63 size_t msghdr_prep_recv_fds(struct msghdr *msg, uint8_t *buf, size_t bufsize,
66 size_t ret = CMSG_SPACE(sizeof(int) * num_fds);
73 msg->msg_control = buf;
74 msg->msg_controllen = ret;
76 msg->msg_control = NULL;
77 msg->msg_controllen = 0;
83 size_t msghdr_extract_fds(struct msghdr *msg, int *fds, size_t fds_size)
88 for(cmsg = CMSG_FIRSTHDR(msg);
90 cmsg = CMSG_NXTHDR(msg, cmsg))
92 if ((cmsg->cmsg_type == SCM_RIGHTS) &&
93 (cmsg->cmsg_level == SOL_SOCKET)) {
102 num_fds = (cmsg->cmsg_len - CMSG_LEN(0)) / sizeof(int);
104 if ((num_fds != 0) && (fds != NULL) && (fds_size >= num_fds)) {
105 memcpy(fds, CMSG_DATA(cmsg), num_fds * sizeof(int));
111 #elif defined(HAVE_STRUCT_MSGHDR_MSG_ACCRIGHTS)
113 ssize_t msghdr_prep_fds(struct msghdr *msg, uint8_t *buf, size_t bufsize,
114 const int *fds, size_t num_fds)
118 if (num_fds > INT8_MAX) {
122 needed = sizeof(int) * num_fds;
124 if ((msg == NULL) || (needed > bufsize)) {
128 memcpy(buf, fds, needed);
130 msg->msg_accrights = (caddr_t) buf;
131 msg->msg_accrightslen = needed;
136 size_t msghdr_prep_recv_fds(struct msghdr *msg, uint8_t *buf, size_t bufsize,
139 size_t ret = num_fds * sizeof(int);
147 msg->msg_accrights = (caddr_t) buf;
148 msg->msg_accrightslen = ret;
150 msg->msg_accrights = NULL;
151 msg->msg_accrightslen = 0;
157 size_t msghdr_extract_fds(struct msghdr *msg, int *fds, size_t fds_size)
159 size_t num_fds = msg->msg_accrightslen / sizeof(int);
161 if ((fds != 0) && (num_fds <= fds_size)) {
162 memcpy(fds, msg->msg_accrights, msg->msg_accrightslen);
170 ssize_t msghdr_prep_fds(struct msghdr *msg, uint8_t *buf, size_t bufsize,
171 const int *fds, size_t num_fds)
176 size_t msghdr_prep_recv_fds(struct msghdr *msg, uint8_t *buf, size_t bufsize,
182 size_t msghdr_extract_fds(struct msghdr *msg, int *fds, size_t fds_size)
191 struct sockaddr_storage addr;
196 ssize_t msghdr_copy(struct msghdr_buf *msg, size_t msgsize,
197 const void *addr, socklen_t addrlen,
198 const struct iovec *iov, int iovcnt,
199 const int *fds, size_t num_fds)
202 size_t iov_len, needed, bufsize;
204 bufsize = (msgsize > offsetof(struct msghdr_buf, buf)) ?
205 msgsize - offsetof(struct msghdr_buf, buf) : 0;
207 fd_len = msghdr_prep_fds(&msg->msg, msg->buf, bufsize, fds, num_fds);
213 if (bufsize >= fd_len) {
222 if (addrlen > sizeof(struct sockaddr_storage)) {
226 memcpy(&msg->addr, addr, addrlen);
227 msg->msg.msg_name = &msg->addr;
228 msg->msg.msg_namelen = addrlen;
230 msg->msg.msg_name = NULL;
231 msg->msg.msg_namelen = 0;
234 msg->iov.iov_base = msg->buf + fd_len;
235 msg->iov.iov_len = iov_buf(
236 iov, iovcnt, msg->iov.iov_base, bufsize);
237 iov_len = msg->iov.iov_len;
239 msg->msg.msg_iov = &msg->iov;
240 msg->msg.msg_iovlen = 1;
242 iov_len = iov_buflen(iov, iovcnt);
245 needed = offsetof(struct msghdr_buf, buf) + fd_len;
246 if (needed < fd_len) {
250 if (needed < iov_len) {
257 struct msghdr *msghdr_buf_msghdr(struct msghdr_buf *msg)