]> CyberLeo.Net >> Repos - FreeBSD/stable/10.git/blob - lib/libc/net/sctp_sys_calls.c
MFC r269858:
[FreeBSD/stable/10.git] / lib / libc / net / sctp_sys_calls.c
1 /*-
2  * Copyright (c) 2001-2007, by Cisco Systems, Inc. All rights reserved.
3  * Copyright (c) 2008-2012, by Randall Stewart. All rights reserved.
4  * Copyright (c) 2008-2012, by Michael Tuexen. 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 are met:
8  *
9  * a) Redistributions of source code must retain the above copyright notice,
10  *    this list of conditions and the following disclaimer.
11  *
12  * b) Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in
14  *    the documentation and/or other materials provided with the distribution.
15  *
16  * c) Neither the name of Cisco Systems, Inc. nor the names of its
17  *    contributors may be used to endorse or promote products derived
18  *    from this software without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
22  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
24  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
30  * THE POSSIBILITY OF SUCH DAMAGE.
31  */
32
33 #include <sys/cdefs.h>
34 __FBSDID("$FreeBSD$");
35
36 #include <stdio.h>
37 #include <string.h>
38 #include <errno.h>
39 #include <stdlib.h>
40 #include <unistd.h>
41 #include <sys/types.h>
42 #include <sys/socket.h>
43 #include <sys/errno.h>
44 #include <sys/syscall.h>
45 #include <sys/uio.h>
46 #include <netinet/in.h>
47 #include <arpa/inet.h>
48 #include <netinet/sctp_uio.h>
49 #include <netinet/sctp.h>
50
51 #ifndef IN6_IS_ADDR_V4MAPPED
52 #define IN6_IS_ADDR_V4MAPPED(a)               \
53         ((*(const uint32_t *)(const void *)(&(a)->s6_addr[0]) == 0) &&  \
54          (*(const uint32_t *)(const void *)(&(a)->s6_addr[4]) == 0) &&  \
55          (*(const uint32_t *)(const void *)(&(a)->s6_addr[8]) == ntohl(0x0000ffff)))
56 #endif
57
58 #define SCTP_CONTROL_VEC_SIZE_RCV  16384
59
60
61 static void
62 in6_sin6_2_sin(struct sockaddr_in *sin, struct sockaddr_in6 *sin6)
63 {
64         bzero(sin, sizeof(*sin));
65         sin->sin_len = sizeof(struct sockaddr_in);
66         sin->sin_family = AF_INET;
67         sin->sin_port = sin6->sin6_port;
68         sin->sin_addr.s_addr = sin6->sin6_addr.__u6_addr.__u6_addr32[3];
69 }
70
71 int
72 sctp_getaddrlen(sa_family_t family)
73 {
74         int ret, sd;
75         socklen_t siz;
76         struct sctp_assoc_value av;
77
78         av.assoc_value = family;
79         siz = sizeof(av);
80 #if defined(AF_INET)
81         sd = socket(AF_INET, SOCK_SEQPACKET, IPPROTO_SCTP);
82 #elif defined(AF_INET6)
83         sd = socket(AF_INET6, SOCK_SEQPACKET, IPPROTO_SCTP);
84 #else
85         sd = -1;
86 #endif
87         if (sd == -1) {
88                 return (-1);
89         }
90         ret = getsockopt(sd, IPPROTO_SCTP, SCTP_GET_ADDR_LEN, &av, &siz);
91         close(sd);
92         if (ret == 0) {
93                 return ((int)av.assoc_value);
94         } else {
95                 return (-1);
96         }
97 }
98
99 int
100 sctp_connectx(int sd, const struct sockaddr *addrs, int addrcnt,
101     sctp_assoc_t * id)
102 {
103         char *buf;
104         int i, ret, *aa;
105         char *cpto;
106         const struct sockaddr *at;
107         size_t len;
108
109         /* validate the address count and list */
110         if ((addrs == NULL) || (addrcnt <= 0)) {
111                 errno = EINVAL;
112                 return (-1);
113         }
114         if ((buf = malloc(sizeof(int) + (size_t)addrcnt * sizeof(struct sockaddr_in6))) == NULL) {
115                 errno = E2BIG;
116                 return (-1);
117         }
118         len = sizeof(int);
119         at = addrs;
120         cpto = buf + sizeof(int);
121         /* validate all the addresses and get the size */
122         for (i = 0; i < addrcnt; i++) {
123                 switch (at->sa_family) {
124                 case AF_INET:
125                         if (at->sa_len != sizeof(struct sockaddr_in)) {
126                                 free(buf);
127                                 errno = EINVAL;
128                                 return (-1);
129                         }
130                         memcpy(cpto, at, sizeof(struct sockaddr_in));
131                         cpto = ((caddr_t)cpto + sizeof(struct sockaddr_in));
132                         len += sizeof(struct sockaddr_in);
133                         break;
134                 case AF_INET6:
135                         if (at->sa_len != sizeof(struct sockaddr_in6)) {
136                                 free(buf);
137                                 errno = EINVAL;
138                                 return (-1);
139                         }
140                         if (IN6_IS_ADDR_V4MAPPED(&((struct sockaddr_in6 *)at)->sin6_addr)) {
141                                 in6_sin6_2_sin((struct sockaddr_in *)cpto, (struct sockaddr_in6 *)at);
142                                 cpto = ((caddr_t)cpto + sizeof(struct sockaddr_in));
143                                 len += sizeof(struct sockaddr_in);
144                         } else {
145                                 memcpy(cpto, at, sizeof(struct sockaddr_in6));
146                                 cpto = ((caddr_t)cpto + sizeof(struct sockaddr_in6));
147                                 len += sizeof(struct sockaddr_in6);
148                         }
149                         break;
150                 default:
151                         free(buf);
152                         errno = EINVAL;
153                         return (-1);
154                 }
155                 at = (struct sockaddr *)((caddr_t)at + at->sa_len);
156         }
157         aa = (int *)buf;
158         *aa = addrcnt;
159         ret = setsockopt(sd, IPPROTO_SCTP, SCTP_CONNECT_X, (void *)buf,
160             (socklen_t) len);
161         if ((ret == 0) && (id != NULL)) {
162                 *id = *(sctp_assoc_t *) buf;
163         }
164         free(buf);
165         return (ret);
166 }
167
168 int
169 sctp_bindx(int sd, struct sockaddr *addrs, int addrcnt, int flags)
170 {
171         struct sctp_getaddresses *gaddrs;
172         struct sockaddr *sa;
173         struct sockaddr_in *sin;
174         struct sockaddr_in6 *sin6;
175         int i;
176         size_t argsz;
177         uint16_t sport = 0;
178
179         /* validate the flags */
180         if ((flags != SCTP_BINDX_ADD_ADDR) &&
181             (flags != SCTP_BINDX_REM_ADDR)) {
182                 errno = EFAULT;
183                 return (-1);
184         }
185         /* validate the address count and list */
186         if ((addrcnt <= 0) || (addrs == NULL)) {
187                 errno = EINVAL;
188                 return (-1);
189         }
190         /* First pre-screen the addresses */
191         sa = addrs;
192         for (i = 0; i < addrcnt; i++) {
193                 switch (sa->sa_family) {
194                 case AF_INET:
195                         if (sa->sa_len != sizeof(struct sockaddr_in)) {
196                                 errno = EINVAL;
197                                 return (-1);
198                         }
199                         sin = (struct sockaddr_in *)sa;
200                         if (sin->sin_port) {
201                                 /* non-zero port, check or save */
202                                 if (sport) {
203                                         /* Check against our port */
204                                         if (sport != sin->sin_port) {
205                                                 errno = EINVAL;
206                                                 return (-1);
207                                         }
208                                 } else {
209                                         /* save off the port */
210                                         sport = sin->sin_port;
211                                 }
212                         }
213                         break;
214                 case AF_INET6:
215                         if (sa->sa_len != sizeof(struct sockaddr_in6)) {
216                                 errno = EINVAL;
217                                 return (-1);
218                         }
219                         sin6 = (struct sockaddr_in6 *)sa;
220                         if (sin6->sin6_port) {
221                                 /* non-zero port, check or save */
222                                 if (sport) {
223                                         /* Check against our port */
224                                         if (sport != sin6->sin6_port) {
225                                                 errno = EINVAL;
226                                                 return (-1);
227                                         }
228                                 } else {
229                                         /* save off the port */
230                                         sport = sin6->sin6_port;
231                                 }
232                         }
233                         break;
234                 default:
235                         /* Invalid address family specified. */
236                         errno = EAFNOSUPPORT;
237                         return (-1);
238                 }
239                 sa = (struct sockaddr *)((caddr_t)sa + sa->sa_len);
240         }
241         argsz = sizeof(struct sctp_getaddresses) +
242             sizeof(struct sockaddr_storage);
243         if ((gaddrs = (struct sctp_getaddresses *)malloc(argsz)) == NULL) {
244                 errno = ENOMEM;
245                 return (-1);
246         }
247         sa = addrs;
248         for (i = 0; i < addrcnt; i++) {
249                 memset(gaddrs, 0, argsz);
250                 gaddrs->sget_assoc_id = 0;
251                 memcpy(gaddrs->addr, sa, sa->sa_len);
252                 /*
253                  * Now, if there was a port mentioned, assure that the first
254                  * address has that port to make sure it fails or succeeds
255                  * correctly.
256                  */
257                 if ((i == 0) && (sport != 0)) {
258                         switch (gaddrs->addr->sa_family) {
259                         case AF_INET:
260                                 sin = (struct sockaddr_in *)gaddrs->addr;
261                                 sin->sin_port = sport;
262                                 break;
263                         case AF_INET6:
264                                 sin6 = (struct sockaddr_in6 *)gaddrs->addr;
265                                 sin6->sin6_port = sport;
266                                 break;
267                         }
268                 }
269                 if (setsockopt(sd, IPPROTO_SCTP, flags, gaddrs,
270                     (socklen_t) argsz) != 0) {
271                         free(gaddrs);
272                         return (-1);
273                 }
274                 sa = (struct sockaddr *)((caddr_t)sa + sa->sa_len);
275         }
276         free(gaddrs);
277         return (0);
278 }
279
280 int
281 sctp_opt_info(int sd, sctp_assoc_t id, int opt, void *arg, socklen_t * size)
282 {
283         if (arg == NULL) {
284                 errno = EINVAL;
285                 return (-1);
286         }
287         if ((id == SCTP_CURRENT_ASSOC) ||
288             (id == SCTP_ALL_ASSOC)) {
289                 errno = EINVAL;
290                 return (-1);
291         }
292         switch (opt) {
293         case SCTP_RTOINFO:
294                 ((struct sctp_rtoinfo *)arg)->srto_assoc_id = id;
295                 break;
296         case SCTP_ASSOCINFO:
297                 ((struct sctp_assocparams *)arg)->sasoc_assoc_id = id;
298                 break;
299         case SCTP_DEFAULT_SEND_PARAM:
300                 ((struct sctp_assocparams *)arg)->sasoc_assoc_id = id;
301                 break;
302         case SCTP_PRIMARY_ADDR:
303                 ((struct sctp_setprim *)arg)->ssp_assoc_id = id;
304                 break;
305         case SCTP_PEER_ADDR_PARAMS:
306                 ((struct sctp_paddrparams *)arg)->spp_assoc_id = id;
307                 break;
308         case SCTP_MAXSEG:
309                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
310                 break;
311         case SCTP_AUTH_KEY:
312                 ((struct sctp_authkey *)arg)->sca_assoc_id = id;
313                 break;
314         case SCTP_AUTH_ACTIVE_KEY:
315                 ((struct sctp_authkeyid *)arg)->scact_assoc_id = id;
316                 break;
317         case SCTP_DELAYED_SACK:
318                 ((struct sctp_sack_info *)arg)->sack_assoc_id = id;
319                 break;
320         case SCTP_CONTEXT:
321                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
322                 break;
323         case SCTP_STATUS:
324                 ((struct sctp_status *)arg)->sstat_assoc_id = id;
325                 break;
326         case SCTP_GET_PEER_ADDR_INFO:
327                 ((struct sctp_paddrinfo *)arg)->spinfo_assoc_id = id;
328                 break;
329         case SCTP_PEER_AUTH_CHUNKS:
330                 ((struct sctp_authchunks *)arg)->gauth_assoc_id = id;
331                 break;
332         case SCTP_LOCAL_AUTH_CHUNKS:
333                 ((struct sctp_authchunks *)arg)->gauth_assoc_id = id;
334                 break;
335         case SCTP_TIMEOUTS:
336                 ((struct sctp_timeouts *)arg)->stimo_assoc_id = id;
337                 break;
338         case SCTP_EVENT:
339                 ((struct sctp_event *)arg)->se_assoc_id = id;
340                 break;
341         case SCTP_DEFAULT_SNDINFO:
342                 ((struct sctp_sndinfo *)arg)->snd_assoc_id = id;
343                 break;
344         case SCTP_DEFAULT_PRINFO:
345                 ((struct sctp_default_prinfo *)arg)->pr_assoc_id = id;
346                 break;
347         case SCTP_PEER_ADDR_THLDS:
348                 ((struct sctp_paddrthlds *)arg)->spt_assoc_id = id;
349                 break;
350         case SCTP_REMOTE_UDP_ENCAPS_PORT:
351                 ((struct sctp_udpencaps *)arg)->sue_assoc_id = id;
352                 break;
353         case SCTP_ECN_SUPPORTED:
354                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
355                 break;
356         case SCTP_PR_SUPPORTED:
357                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
358                 break;
359         case SCTP_AUTH_SUPPORTED:
360                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
361                 break;
362         case SCTP_ASCONF_SUPPORTED:
363                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
364                 break;
365         case SCTP_RECONFIG_SUPPORTED:
366                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
367                 break;
368         case SCTP_NRSACK_SUPPORTED:
369                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
370                 break;
371         case SCTP_PKTDROP_SUPPORTED:
372                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
373                 break;
374         case SCTP_MAX_BURST:
375                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
376                 break;
377         case SCTP_ENABLE_STREAM_RESET:
378                 ((struct sctp_assoc_value *)arg)->assoc_id = id;
379                 break;
380         default:
381                 break;
382         }
383         return (getsockopt(sd, IPPROTO_SCTP, opt, arg, size));
384 }
385
386 int
387 sctp_getpaddrs(int sd, sctp_assoc_t id, struct sockaddr **raddrs)
388 {
389         struct sctp_getaddresses *addrs;
390         struct sockaddr *sa;
391         sctp_assoc_t asoc;
392         caddr_t lim;
393         socklen_t opt_len;
394         int cnt;
395
396         if (raddrs == NULL) {
397                 errno = EFAULT;
398                 return (-1);
399         }
400         asoc = id;
401         opt_len = (socklen_t) sizeof(sctp_assoc_t);
402         if (getsockopt(sd, IPPROTO_SCTP, SCTP_GET_REMOTE_ADDR_SIZE,
403             &asoc, &opt_len) != 0) {
404                 return (-1);
405         }
406         /* size required is returned in 'asoc' */
407         opt_len = (socklen_t) ((size_t)asoc + sizeof(struct sctp_getaddresses));
408         addrs = calloc(1, (size_t)opt_len);
409         if (addrs == NULL) {
410                 errno = ENOMEM;
411                 return (-1);
412         }
413         addrs->sget_assoc_id = id;
414         /* Now lets get the array of addresses */
415         if (getsockopt(sd, IPPROTO_SCTP, SCTP_GET_PEER_ADDRESSES,
416             addrs, &opt_len) != 0) {
417                 free(addrs);
418                 return (-1);
419         }
420         *raddrs = (struct sockaddr *)&addrs->addr[0];
421         cnt = 0;
422         sa = (struct sockaddr *)&addrs->addr[0];
423         lim = (caddr_t)addrs + opt_len;
424         while (((caddr_t)sa < lim) && (sa->sa_len > 0)) {
425                 sa = (struct sockaddr *)((caddr_t)sa + sa->sa_len);
426                 cnt++;
427         }
428         return (cnt);
429 }
430
431 void
432 sctp_freepaddrs(struct sockaddr *addrs)
433 {
434         void *fr_addr;
435
436         /* Take away the hidden association id */
437         fr_addr = (void *)((caddr_t)addrs - sizeof(sctp_assoc_t));
438         /* Now free it */
439         free(fr_addr);
440 }
441
442 int
443 sctp_getladdrs(int sd, sctp_assoc_t id, struct sockaddr **raddrs)
444 {
445         struct sctp_getaddresses *addrs;
446         caddr_t lim;
447         struct sockaddr *sa;
448         size_t size_of_addresses;
449         socklen_t opt_len;
450         int cnt;
451
452         if (raddrs == NULL) {
453                 errno = EFAULT;
454                 return (-1);
455         }
456         size_of_addresses = 0;
457         opt_len = (socklen_t) sizeof(int);
458         if (getsockopt(sd, IPPROTO_SCTP, SCTP_GET_LOCAL_ADDR_SIZE,
459             &size_of_addresses, &opt_len) != 0) {
460                 errno = ENOMEM;
461                 return (-1);
462         }
463         if (size_of_addresses == 0) {
464                 errno = ENOTCONN;
465                 return (-1);
466         }
467         opt_len = (socklen_t) (size_of_addresses +
468             sizeof(struct sockaddr_storage) +
469             sizeof(struct sctp_getaddresses));
470         addrs = calloc(1, (size_t)opt_len);
471         if (addrs == NULL) {
472                 errno = ENOMEM;
473                 return (-1);
474         }
475         addrs->sget_assoc_id = id;
476         /* Now lets get the array of addresses */
477         if (getsockopt(sd, IPPROTO_SCTP, SCTP_GET_LOCAL_ADDRESSES, addrs,
478             &opt_len) != 0) {
479                 free(addrs);
480                 errno = ENOMEM;
481                 return (-1);
482         }
483         *raddrs = (struct sockaddr *)&addrs->addr[0];
484         cnt = 0;
485         sa = (struct sockaddr *)&addrs->addr[0];
486         lim = (caddr_t)addrs + opt_len;
487         while (((caddr_t)sa < lim) && (sa->sa_len > 0)) {
488                 sa = (struct sockaddr *)((caddr_t)sa + sa->sa_len);
489                 cnt++;
490         }
491         return (cnt);
492 }
493
494 void
495 sctp_freeladdrs(struct sockaddr *addrs)
496 {
497         void *fr_addr;
498
499         /* Take away the hidden association id */
500         fr_addr = (void *)((caddr_t)addrs - sizeof(sctp_assoc_t));
501         /* Now free it */
502         free(fr_addr);
503 }
504
505 ssize_t
506 sctp_sendmsg(int s,
507     const void *data,
508     size_t len,
509     const struct sockaddr *to,
510     socklen_t tolen,
511     uint32_t ppid,
512     uint32_t flags,
513     uint16_t stream_no,
514     uint32_t timetolive,
515     uint32_t context)
516 {
517 #ifdef SYS_sctp_generic_sendmsg
518         struct sctp_sndrcvinfo sinfo;
519
520         memset(&sinfo, 0, sizeof(struct sctp_sndrcvinfo));
521         sinfo.sinfo_ppid = ppid;
522         sinfo.sinfo_flags = flags;
523         sinfo.sinfo_stream = stream_no;
524         sinfo.sinfo_timetolive = timetolive;
525         sinfo.sinfo_context = context;
526         sinfo.sinfo_assoc_id = 0;
527         return (syscall(SYS_sctp_generic_sendmsg, s,
528             data, len, to, tolen, &sinfo, 0));
529 #else
530         struct msghdr msg;
531         struct sctp_sndrcvinfo *sinfo;
532         struct iovec iov;
533         char cmsgbuf[CMSG_SPACE(sizeof(struct sctp_sndrcvinfo))];
534         struct cmsghdr *cmsg;
535         struct sockaddr *who = NULL;
536         union {
537                 struct sockaddr_in in;
538                 struct sockaddr_in6 in6;
539         }     addr;
540
541         if ((tolen > 0) &&
542             ((to == NULL) || (tolen < sizeof(struct sockaddr)))) {
543                 errno = EINVAL;
544                 return (-1);
545         }
546         if ((to != NULL) && (tolen > 0)) {
547                 switch (to->sa_family) {
548                 case AF_INET:
549                         if (tolen != sizeof(struct sockaddr_in)) {
550                                 errno = EINVAL;
551                                 return (-1);
552                         }
553                         if ((to->sa_len > 0) &&
554                             (to->sa_len != sizeof(struct sockaddr_in))) {
555                                 errno = EINVAL;
556                                 return (-1);
557                         }
558                         memcpy(&addr, to, sizeof(struct sockaddr_in));
559                         addr.in.sin_len = sizeof(struct sockaddr_in);
560                         break;
561                 case AF_INET6:
562                         if (tolen != sizeof(struct sockaddr_in6)) {
563                                 errno = EINVAL;
564                                 return (-1);
565                         }
566                         if ((to->sa_len > 0) &&
567                             (to->sa_len != sizeof(struct sockaddr_in6))) {
568                                 errno = EINVAL;
569                                 return (-1);
570                         }
571                         memcpy(&addr, to, sizeof(struct sockaddr_in6));
572                         addr.in6.sin6_len = sizeof(struct sockaddr_in6);
573                         break;
574                 default:
575                         errno = EAFNOSUPPORT;
576                         return (-1);
577                 }
578                 who = (struct sockaddr *)&addr;
579         }
580         iov.iov_base = (char *)data;
581         iov.iov_len = len;
582
583         if (who) {
584                 msg.msg_name = (caddr_t)who;
585                 msg.msg_namelen = who->sa_len;
586         } else {
587                 msg.msg_name = (caddr_t)NULL;
588                 msg.msg_namelen = 0;
589         }
590         msg.msg_iov = &iov;
591         msg.msg_iovlen = 1;
592         msg.msg_control = cmsgbuf;
593         msg.msg_controllen = CMSG_SPACE(sizeof(struct sctp_sndrcvinfo));
594         cmsg = (struct cmsghdr *)cmsgbuf;
595         cmsg->cmsg_level = IPPROTO_SCTP;
596         cmsg->cmsg_type = SCTP_SNDRCV;
597         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndrcvinfo));
598         sinfo = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
599         memset(sinfo, 0, sizeof(struct sctp_sndrcvinfo));
600         sinfo->sinfo_stream = stream_no;
601         sinfo->sinfo_ssn = 0;
602         sinfo->sinfo_flags = flags;
603         sinfo->sinfo_ppid = ppid;
604         sinfo->sinfo_context = context;
605         sinfo->sinfo_assoc_id = 0;
606         sinfo->sinfo_timetolive = timetolive;
607         return (sendmsg(s, &msg, 0));
608 #endif
609 }
610
611
612 sctp_assoc_t
613 sctp_getassocid(int sd, struct sockaddr *sa)
614 {
615         struct sctp_paddrinfo sp;
616         socklen_t siz;
617
618         /* First get the assoc id */
619         siz = sizeof(sp);
620         memset(&sp, 0, sizeof(sp));
621         memcpy((caddr_t)&sp.spinfo_address, sa, sa->sa_len);
622         if (getsockopt(sd, IPPROTO_SCTP,
623             SCTP_GET_PEER_ADDR_INFO, &sp, &siz) != 0) {
624                 /* We depend on the fact that 0 can never be returned */
625                 return ((sctp_assoc_t) 0);
626         }
627         return (sp.spinfo_assoc_id);
628 }
629
630 ssize_t
631 sctp_send(int sd, const void *data, size_t len,
632     const struct sctp_sndrcvinfo *sinfo,
633     int flags)
634 {
635
636 #ifdef SYS_sctp_generic_sendmsg
637         struct sockaddr *to = NULL;
638
639         return (syscall(SYS_sctp_generic_sendmsg, sd,
640             data, len, to, 0, sinfo, flags));
641 #else
642         struct msghdr msg;
643         struct iovec iov;
644         char cmsgbuf[CMSG_SPACE(sizeof(struct sctp_sndrcvinfo))];
645         struct cmsghdr *cmsg;
646
647         if (sinfo == NULL) {
648                 errno = EINVAL;
649                 return (-1);
650         }
651         iov.iov_base = (char *)data;
652         iov.iov_len = len;
653
654         msg.msg_name = NULL;
655         msg.msg_namelen = 0;
656         msg.msg_iov = &iov;
657         msg.msg_iovlen = 1;
658         msg.msg_control = cmsgbuf;
659         msg.msg_controllen = CMSG_SPACE(sizeof(struct sctp_sndrcvinfo));
660         cmsg = (struct cmsghdr *)cmsgbuf;
661         cmsg->cmsg_level = IPPROTO_SCTP;
662         cmsg->cmsg_type = SCTP_SNDRCV;
663         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndrcvinfo));
664         memcpy(CMSG_DATA(cmsg), sinfo, sizeof(struct sctp_sndrcvinfo));
665         return (sendmsg(sd, &msg, flags));
666 #endif
667 }
668
669
670
671 ssize_t
672 sctp_sendx(int sd, const void *msg, size_t msg_len,
673     struct sockaddr *addrs, int addrcnt,
674     struct sctp_sndrcvinfo *sinfo,
675     int flags)
676 {
677         struct sctp_sndrcvinfo __sinfo;
678         ssize_t ret;
679         int i, cnt, *aa, saved_errno;
680         char *buf;
681         int no_end_cx = 0;
682         size_t len, add_len;
683         struct sockaddr *at;
684
685         if (addrs == NULL) {
686                 errno = EINVAL;
687                 return (-1);
688         }
689 #ifdef SYS_sctp_generic_sendmsg
690         if (addrcnt == 1) {
691                 socklen_t l;
692
693                 /*
694                  * Quick way, we don't need to do a connectx so lets use the
695                  * syscall directly.
696                  */
697                 l = addrs->sa_len;
698                 return (syscall(SYS_sctp_generic_sendmsg, sd,
699                     msg, msg_len, addrs, l, sinfo, flags));
700         }
701 #endif
702
703         len = sizeof(int);
704         at = addrs;
705         cnt = 0;
706         /* validate all the addresses and get the size */
707         for (i = 0; i < addrcnt; i++) {
708                 if (at->sa_family == AF_INET) {
709                         add_len = sizeof(struct sockaddr_in);
710                 } else if (at->sa_family == AF_INET6) {
711                         add_len = sizeof(struct sockaddr_in6);
712                 } else {
713                         errno = EINVAL;
714                         return (-1);
715                 }
716                 len += add_len;
717                 at = (struct sockaddr *)((caddr_t)at + add_len);
718                 cnt++;
719         }
720         /* do we have any? */
721         if (cnt == 0) {
722                 errno = EINVAL;
723                 return (-1);
724         }
725         buf = malloc(len);
726         if (buf == NULL) {
727                 errno = ENOMEM;
728                 return (-1);
729         }
730         aa = (int *)buf;
731         *aa = cnt;
732         aa++;
733         memcpy((caddr_t)aa, addrs, (size_t)(len - sizeof(int)));
734         ret = setsockopt(sd, IPPROTO_SCTP, SCTP_CONNECT_X_DELAYED, (void *)buf,
735             (socklen_t) len);
736
737         free(buf);
738         if (ret != 0) {
739                 if (errno == EALREADY) {
740                         no_end_cx = 1;
741                         goto continue_send;
742                 }
743                 return (ret);
744         }
745 continue_send:
746         if (sinfo == NULL) {
747                 sinfo = &__sinfo;
748                 memset(&__sinfo, 0, sizeof(__sinfo));
749         }
750         sinfo->sinfo_assoc_id = sctp_getassocid(sd, addrs);
751         if (sinfo->sinfo_assoc_id == 0) {
752                 (void)setsockopt(sd, IPPROTO_SCTP, SCTP_CONNECT_X_COMPLETE, (void *)addrs,
753                     (socklen_t) addrs->sa_len);
754                 errno = ENOENT;
755                 return (-1);
756         }
757         ret = sctp_send(sd, msg, msg_len, sinfo, flags);
758         saved_errno = errno;
759         if (no_end_cx == 0)
760                 (void)setsockopt(sd, IPPROTO_SCTP, SCTP_CONNECT_X_COMPLETE, (void *)addrs,
761                     (socklen_t) addrs->sa_len);
762
763         errno = saved_errno;
764         return (ret);
765 }
766
767 ssize_t
768 sctp_sendmsgx(int sd,
769     const void *msg,
770     size_t len,
771     struct sockaddr *addrs,
772     int addrcnt,
773     uint32_t ppid,
774     uint32_t flags,
775     uint16_t stream_no,
776     uint32_t timetolive,
777     uint32_t context)
778 {
779         struct sctp_sndrcvinfo sinfo;
780
781         memset((void *)&sinfo, 0, sizeof(struct sctp_sndrcvinfo));
782         sinfo.sinfo_ppid = ppid;
783         sinfo.sinfo_flags = flags;
784         sinfo.sinfo_ssn = stream_no;
785         sinfo.sinfo_timetolive = timetolive;
786         sinfo.sinfo_context = context;
787         return (sctp_sendx(sd, msg, len, addrs, addrcnt, &sinfo, 0));
788 }
789
790 ssize_t
791 sctp_recvmsg(int s,
792     void *dbuf,
793     size_t len,
794     struct sockaddr *from,
795     socklen_t * fromlen,
796     struct sctp_sndrcvinfo *sinfo,
797     int *msg_flags)
798 {
799 #ifdef SYS_sctp_generic_recvmsg
800         struct iovec iov;
801
802         iov.iov_base = dbuf;
803         iov.iov_len = len;
804         return (syscall(SYS_sctp_generic_recvmsg, s,
805             &iov, 1, from, fromlen, sinfo, msg_flags));
806 #else
807         ssize_t sz;
808         struct msghdr msg;
809         struct iovec iov;
810         char cmsgbuf[SCTP_CONTROL_VEC_SIZE_RCV];
811         struct cmsghdr *cmsg;
812
813         if (msg_flags == NULL) {
814                 errno = EINVAL;
815                 return (-1);
816         }
817         msg.msg_flags = 0;
818         iov.iov_base = dbuf;
819         iov.iov_len = len;
820         msg.msg_name = (caddr_t)from;
821         if (fromlen == NULL)
822                 msg.msg_namelen = 0;
823         else
824                 msg.msg_namelen = *fromlen;
825         msg.msg_iov = &iov;
826         msg.msg_iovlen = 1;
827         msg.msg_control = cmsgbuf;
828         msg.msg_controllen = sizeof(cmsgbuf);
829         sz = recvmsg(s, &msg, *msg_flags);
830         *msg_flags = msg.msg_flags;
831         if (sz <= 0) {
832                 return (sz);
833         }
834         if (sinfo) {
835                 sinfo->sinfo_assoc_id = 0;
836         }
837         if ((msg.msg_controllen > 0) && (sinfo != NULL)) {
838                 /*
839                  * parse through and see if we find the sctp_sndrcvinfo (if
840                  * the user wants it).
841                  */
842                 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
843                         if (cmsg->cmsg_level != IPPROTO_SCTP) {
844                                 continue;
845                         }
846                         if (cmsg->cmsg_type == SCTP_SNDRCV) {
847                                 memcpy(sinfo, CMSG_DATA(cmsg), sizeof(struct sctp_sndrcvinfo));
848                                 break;
849                         }
850                         if (cmsg->cmsg_type == SCTP_EXTRCV) {
851                                 /*
852                                  * Let's hope that the user provided enough
853                                  * enough memory. At least he asked for more
854                                  * information.
855                                  */
856                                 memcpy(sinfo, CMSG_DATA(cmsg), sizeof(struct sctp_extrcvinfo));
857                                 break;
858                         }
859                 }
860         }
861         return (sz);
862 #endif
863 }
864
865 ssize_t 
866 sctp_recvv(int sd,
867     const struct iovec *iov,
868     int iovlen,
869     struct sockaddr *from,
870     socklen_t * fromlen,
871     void *info,
872     socklen_t * infolen,
873     unsigned int *infotype,
874     int *flags)
875 {
876         char cmsgbuf[SCTP_CONTROL_VEC_SIZE_RCV];
877         struct msghdr msg;
878         struct cmsghdr *cmsg;
879         ssize_t ret;
880         struct sctp_rcvinfo *rcvinfo;
881         struct sctp_nxtinfo *nxtinfo;
882
883         if (((info != NULL) && (infolen == NULL)) |
884             ((info == NULL) && (infolen != NULL) && (*infolen != 0)) ||
885             ((info != NULL) && (infotype == NULL))) {
886                 errno = EINVAL;
887                 return (-1);
888         }
889         if (infotype) {
890                 *infotype = SCTP_RECVV_NOINFO;
891         }
892         msg.msg_name = from;
893         if (fromlen == NULL) {
894                 msg.msg_namelen = 0;
895         } else {
896                 msg.msg_namelen = *fromlen;
897         }
898         msg.msg_iov = (struct iovec *)iov;
899         msg.msg_iovlen = iovlen;
900         msg.msg_control = cmsgbuf;
901         msg.msg_controllen = sizeof(cmsgbuf);
902         ret = recvmsg(sd, &msg, *flags);
903         *flags = msg.msg_flags;
904         if ((ret > 0) &&
905             (msg.msg_controllen > 0) &&
906             (infotype != NULL) &&
907             (infolen != NULL) &&
908             (*infolen > 0)) {
909                 rcvinfo = NULL;
910                 nxtinfo = NULL;
911                 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
912                         if (cmsg->cmsg_level != IPPROTO_SCTP) {
913                                 continue;
914                         }
915                         if (cmsg->cmsg_type == SCTP_RCVINFO) {
916                                 rcvinfo = (struct sctp_rcvinfo *)CMSG_DATA(cmsg);
917                                 if (nxtinfo != NULL) {
918                                         break;
919                                 } else {
920                                         continue;
921                                 }
922                         }
923                         if (cmsg->cmsg_type == SCTP_NXTINFO) {
924                                 nxtinfo = (struct sctp_nxtinfo *)CMSG_DATA(cmsg);
925                                 if (rcvinfo != NULL) {
926                                         break;
927                                 } else {
928                                         continue;
929                                 }
930                         }
931                 }
932                 if (rcvinfo != NULL) {
933                         if ((nxtinfo != NULL) && (*infolen >= sizeof(struct sctp_recvv_rn))) {
934                                 struct sctp_recvv_rn *rn_info;
935
936                                 rn_info = (struct sctp_recvv_rn *)info;
937                                 rn_info->recvv_rcvinfo = *rcvinfo;
938                                 rn_info->recvv_nxtinfo = *nxtinfo;
939                                 *infolen = (socklen_t) sizeof(struct sctp_recvv_rn);
940                                 *infotype = SCTP_RECVV_RN;
941                         } else if (*infolen >= sizeof(struct sctp_rcvinfo)) {
942                                 memcpy(info, rcvinfo, sizeof(struct sctp_rcvinfo));
943                                 *infolen = (socklen_t) sizeof(struct sctp_rcvinfo);
944                                 *infotype = SCTP_RECVV_RCVINFO;
945                         }
946                 } else if (nxtinfo != NULL) {
947                         if (*infolen >= sizeof(struct sctp_nxtinfo)) {
948                                 memcpy(info, nxtinfo, sizeof(struct sctp_nxtinfo));
949                                 *infolen = (socklen_t) sizeof(struct sctp_nxtinfo);
950                                 *infotype = SCTP_RECVV_NXTINFO;
951                         }
952                 }
953         }
954         return (ret);
955 }
956
957 ssize_t
958 sctp_sendv(int sd,
959     const struct iovec *iov, int iovcnt,
960     struct sockaddr *addrs, int addrcnt,
961     void *info, socklen_t infolen, unsigned int infotype,
962     int flags)
963 {
964         ssize_t ret;
965         int i;
966         socklen_t addr_len;
967         struct msghdr msg;
968         in_port_t port;
969         struct sctp_sendv_spa *spa_info;
970         struct cmsghdr *cmsg;
971         char *cmsgbuf;
972         struct sockaddr *addr;
973         struct sockaddr_in *addr_in;
974         struct sockaddr_in6 *addr_in6;
975
976         if ((addrcnt < 0) ||
977             (iovcnt < 0) ||
978             ((addrs == NULL) && (addrcnt > 0)) ||
979             ((addrs != NULL) && (addrcnt == 0)) ||
980             ((iov == NULL) && (iovcnt > 0)) ||
981             ((iov != NULL) && (iovcnt == 0))) {
982                 errno = EINVAL;
983                 return (-1);
984         }
985         cmsgbuf = malloc(CMSG_SPACE(sizeof(struct sctp_sndinfo)) +
986             CMSG_SPACE(sizeof(struct sctp_prinfo)) +
987             CMSG_SPACE(sizeof(struct sctp_authinfo)) +
988             (size_t)addrcnt * CMSG_SPACE(sizeof(struct in6_addr)));
989         if (cmsgbuf == NULL) {
990                 errno = ENOMEM;
991                 return (-1);
992         }
993         msg.msg_control = cmsgbuf;
994         msg.msg_controllen = 0;
995         cmsg = (struct cmsghdr *)cmsgbuf;
996         switch (infotype) {
997         case SCTP_SENDV_NOINFO:
998                 if ((infolen != 0) || (info != NULL)) {
999                         free(cmsgbuf);
1000                         errno = EINVAL;
1001                         return (-1);
1002                 }
1003                 break;
1004         case SCTP_SENDV_SNDINFO:
1005                 if ((info == NULL) || (infolen < sizeof(struct sctp_sndinfo))) {
1006                         free(cmsgbuf);
1007                         errno = EINVAL;
1008                         return (-1);
1009                 }
1010                 cmsg->cmsg_level = IPPROTO_SCTP;
1011                 cmsg->cmsg_type = SCTP_SNDINFO;
1012                 cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndinfo));
1013                 memcpy(CMSG_DATA(cmsg), info, sizeof(struct sctp_sndinfo));
1014                 msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_sndinfo));
1015                 cmsg = (struct cmsghdr *)((caddr_t)cmsg + CMSG_SPACE(sizeof(struct sctp_sndinfo)));
1016                 break;
1017         case SCTP_SENDV_PRINFO:
1018                 if ((info == NULL) || (infolen < sizeof(struct sctp_prinfo))) {
1019                         free(cmsgbuf);
1020                         errno = EINVAL;
1021                         return (-1);
1022                 }
1023                 cmsg->cmsg_level = IPPROTO_SCTP;
1024                 cmsg->cmsg_type = SCTP_PRINFO;
1025                 cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_prinfo));
1026                 memcpy(CMSG_DATA(cmsg), info, sizeof(struct sctp_prinfo));
1027                 msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_prinfo));
1028                 cmsg = (struct cmsghdr *)((caddr_t)cmsg + CMSG_SPACE(sizeof(struct sctp_prinfo)));
1029                 break;
1030         case SCTP_SENDV_AUTHINFO:
1031                 if ((info == NULL) || (infolen < sizeof(struct sctp_authinfo))) {
1032                         free(cmsgbuf);
1033                         errno = EINVAL;
1034                         return (-1);
1035                 }
1036                 cmsg->cmsg_level = IPPROTO_SCTP;
1037                 cmsg->cmsg_type = SCTP_AUTHINFO;
1038                 cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_authinfo));
1039                 memcpy(CMSG_DATA(cmsg), info, sizeof(struct sctp_authinfo));
1040                 msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_authinfo));
1041                 cmsg = (struct cmsghdr *)((caddr_t)cmsg + CMSG_SPACE(sizeof(struct sctp_authinfo)));
1042                 break;
1043         case SCTP_SENDV_SPA:
1044                 if ((info == NULL) || (infolen < sizeof(struct sctp_sendv_spa))) {
1045                         free(cmsgbuf);
1046                         errno = EINVAL;
1047                         return (-1);
1048                 }
1049                 spa_info = (struct sctp_sendv_spa *)info;
1050                 if (spa_info->sendv_flags & SCTP_SEND_SNDINFO_VALID) {
1051                         cmsg->cmsg_level = IPPROTO_SCTP;
1052                         cmsg->cmsg_type = SCTP_SNDINFO;
1053                         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndinfo));
1054                         memcpy(CMSG_DATA(cmsg), &spa_info->sendv_sndinfo, sizeof(struct sctp_sndinfo));
1055                         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_sndinfo));
1056                         cmsg = (struct cmsghdr *)((caddr_t)cmsg + CMSG_SPACE(sizeof(struct sctp_sndinfo)));
1057                 }
1058                 if (spa_info->sendv_flags & SCTP_SEND_PRINFO_VALID) {
1059                         cmsg->cmsg_level = IPPROTO_SCTP;
1060                         cmsg->cmsg_type = SCTP_PRINFO;
1061                         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_prinfo));
1062                         memcpy(CMSG_DATA(cmsg), &spa_info->sendv_prinfo, sizeof(struct sctp_prinfo));
1063                         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_prinfo));
1064                         cmsg = (struct cmsghdr *)((caddr_t)cmsg + CMSG_SPACE(sizeof(struct sctp_prinfo)));
1065                 }
1066                 if (spa_info->sendv_flags & SCTP_SEND_AUTHINFO_VALID) {
1067                         cmsg->cmsg_level = IPPROTO_SCTP;
1068                         cmsg->cmsg_type = SCTP_AUTHINFO;
1069                         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_authinfo));
1070                         memcpy(CMSG_DATA(cmsg), &spa_info->sendv_authinfo, sizeof(struct sctp_authinfo));
1071                         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_authinfo));
1072                         cmsg = (struct cmsghdr *)((caddr_t)cmsg + CMSG_SPACE(sizeof(struct sctp_authinfo)));
1073                 }
1074                 break;
1075         default:
1076                 free(cmsgbuf);
1077                 errno = EINVAL;
1078                 return (-1);
1079         }
1080         addr = addrs;
1081         msg.msg_name = NULL;
1082         msg.msg_namelen = 0;
1083
1084         for (i = 0; i < addrcnt; i++) {
1085                 switch (addr->sa_family) {
1086                 case AF_INET:
1087                         addr_len = (socklen_t) sizeof(struct sockaddr_in);
1088                         addr_in = (struct sockaddr_in *)addr;
1089                         if (addr_in->sin_len != addr_len) {
1090                                 free(cmsgbuf);
1091                                 errno = EINVAL;
1092                                 return (-1);
1093                         }
1094                         if (i == 0) {
1095                                 port = addr_in->sin_port;
1096                         } else {
1097                                 if (port == addr_in->sin_port) {
1098                                         cmsg->cmsg_level = IPPROTO_SCTP;
1099                                         cmsg->cmsg_type = SCTP_DSTADDRV4;
1100                                         cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_addr));
1101                                         memcpy(CMSG_DATA(cmsg), &addr_in->sin_addr, sizeof(struct in_addr));
1102                                         msg.msg_controllen += CMSG_SPACE(sizeof(struct in_addr));
1103                                         cmsg = (struct cmsghdr *)((caddr_t)cmsg + CMSG_SPACE(sizeof(struct in_addr)));
1104                                 } else {
1105                                         free(cmsgbuf);
1106                                         errno = EINVAL;
1107                                         return (-1);
1108                                 }
1109                         }
1110                         break;
1111                 case AF_INET6:
1112                         addr_len = (socklen_t) sizeof(struct sockaddr_in6);
1113                         addr_in6 = (struct sockaddr_in6 *)addr;
1114                         if (addr_in6->sin6_len != addr_len) {
1115                                 free(cmsgbuf);
1116                                 errno = EINVAL;
1117                                 return (-1);
1118                         }
1119                         if (i == 0) {
1120                                 port = addr_in6->sin6_port;
1121                         } else {
1122                                 if (port == addr_in6->sin6_port) {
1123                                         cmsg->cmsg_level = IPPROTO_SCTP;
1124                                         cmsg->cmsg_type = SCTP_DSTADDRV6;
1125                                         cmsg->cmsg_len = CMSG_LEN(sizeof(struct in6_addr));
1126                                         memcpy(CMSG_DATA(cmsg), &addr_in6->sin6_addr, sizeof(struct in6_addr));
1127                                         msg.msg_controllen += CMSG_SPACE(sizeof(struct in6_addr));
1128                                         cmsg = (struct cmsghdr *)((caddr_t)cmsg + CMSG_SPACE(sizeof(struct in6_addr)));
1129                                 } else {
1130                                         free(cmsgbuf);
1131                                         errno = EINVAL;
1132                                         return (-1);
1133                                 }
1134                         }
1135                         break;
1136                 default:
1137                         free(cmsgbuf);
1138                         errno = EINVAL;
1139                         return (-1);
1140                 }
1141                 if (i == 0) {
1142                         msg.msg_name = addr;
1143                         msg.msg_namelen = addr_len;
1144                 }
1145                 addr = (struct sockaddr *)((caddr_t)addr + addr_len);
1146         }
1147         if (msg.msg_controllen == 0) {
1148                 msg.msg_control = NULL;
1149         }
1150         msg.msg_iov = (struct iovec *)iov;
1151         msg.msg_iovlen = iovcnt;
1152         msg.msg_flags = 0;
1153         ret = sendmsg(sd, &msg, flags);
1154         free(cmsgbuf);
1155         return (ret);
1156 }
1157
1158
1159 #if !defined(SYS_sctp_peeloff) && !defined(HAVE_SCTP_PEELOFF_SOCKOPT)
1160
1161 int
1162 sctp_peeloff(int sd, sctp_assoc_t assoc_id)
1163 {
1164         /* NOT supported, return invalid sd */
1165         errno = ENOTSUP;
1166         return (-1);
1167 }
1168
1169 #endif
1170 #if defined(SYS_sctp_peeloff) && !defined(HAVE_SCTP_PEELOFF_SOCKOPT)
1171 int
1172 sctp_peeloff(int sd, sctp_assoc_t assoc_id)
1173 {
1174         return (syscall(SYS_sctp_peeloff, sd, assoc_id));
1175 }
1176
1177 #endif
1178
1179 #undef SCTP_CONTROL_VEC_SIZE_RCV