1*d3273b5bSchristos /* $NetBSD: recvmsg.c,v 1.2 2017/01/28 21:31:50 christos Exp $ */
2ca1c9b0cSelric
3ca1c9b0cSelric /*
4ca1c9b0cSelric * Copyright (c) 1995, 1996, 1997, 1998, 1999 Kungliga Tekniska Högskolan
5ca1c9b0cSelric * (Royal Institute of Technology, Stockholm, Sweden).
6ca1c9b0cSelric * All rights reserved.
7ca1c9b0cSelric *
8ca1c9b0cSelric * Redistribution and use in source and binary forms, with or without
9ca1c9b0cSelric * modification, are permitted provided that the following conditions
10ca1c9b0cSelric * are met:
11ca1c9b0cSelric *
12ca1c9b0cSelric * 1. Redistributions of source code must retain the above copyright
13ca1c9b0cSelric * notice, this list of conditions and the following disclaimer.
14ca1c9b0cSelric *
15ca1c9b0cSelric * 2. Redistributions in binary form must reproduce the above copyright
16ca1c9b0cSelric * notice, this list of conditions and the following disclaimer in the
17ca1c9b0cSelric * documentation and/or other materials provided with the distribution.
18ca1c9b0cSelric *
19ca1c9b0cSelric * 3. Neither the name of the Institute nor the names of its contributors
20ca1c9b0cSelric * may be used to endorse or promote products derived from this software
21ca1c9b0cSelric * without specific prior written permission.
22ca1c9b0cSelric *
23ca1c9b0cSelric * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
24ca1c9b0cSelric * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25ca1c9b0cSelric * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26ca1c9b0cSelric * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
27ca1c9b0cSelric * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28ca1c9b0cSelric * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29ca1c9b0cSelric * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30ca1c9b0cSelric * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31ca1c9b0cSelric * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32ca1c9b0cSelric * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33ca1c9b0cSelric * SUCH DAMAGE.
34ca1c9b0cSelric */
35ca1c9b0cSelric
36ca1c9b0cSelric #include <config.h>
37ca1c9b0cSelric
38ca1c9b0cSelric #include <krb5/roken.h>
39ca1c9b0cSelric
40ca1c9b0cSelric ROKEN_LIB_FUNCTION ssize_t ROKEN_LIB_CALL
recvmsg(int s,struct msghdr * msg,int flags)41ca1c9b0cSelric recvmsg(int s, struct msghdr *msg, int flags)
42ca1c9b0cSelric {
43ca1c9b0cSelric ssize_t ret, nb;
44ca1c9b0cSelric size_t tot = 0;
45ca1c9b0cSelric int i;
46ca1c9b0cSelric char *buf, *p;
47ca1c9b0cSelric struct iovec *iov = msg->msg_iov;
48ca1c9b0cSelric
49ca1c9b0cSelric for(i = 0; i < msg->msg_iovlen; ++i)
50ca1c9b0cSelric tot += iov[i].iov_len;
51ca1c9b0cSelric buf = malloc(tot);
52ca1c9b0cSelric if (tot != 0 && buf == NULL) {
53ca1c9b0cSelric errno = ENOMEM;
54ca1c9b0cSelric return -1;
55ca1c9b0cSelric }
56ca1c9b0cSelric nb = ret = recvfrom (s, buf, tot, flags, msg->msg_name, &msg->msg_namelen);
57ca1c9b0cSelric p = buf;
58ca1c9b0cSelric while (nb > 0) {
59ca1c9b0cSelric ssize_t cnt = min(nb, iov->iov_len);
60ca1c9b0cSelric
61ca1c9b0cSelric memcpy (iov->iov_base, p, cnt);
62ca1c9b0cSelric p += cnt;
63ca1c9b0cSelric nb -= cnt;
64ca1c9b0cSelric ++iov;
65ca1c9b0cSelric }
66ca1c9b0cSelric free(buf);
67ca1c9b0cSelric return ret;
68ca1c9b0cSelric }
69