sctp_sys_calls.c revision 209760
1/*	$KAME: sctp_sys_calls.c,v 1.9 2004/08/17 06:08:53 itojun Exp $ */
2
3/*
4 * Copyright (C) 2002-2007 Cisco Systems Inc,
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the project nor the names of its contributors
16 *    may be used to endorse or promote products derived from this software
17 *    without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31#include <sys/cdefs.h>
32__FBSDID("$FreeBSD: head/lib/libc/net/sctp_sys_calls.c 209760 2010-07-07 11:19:06Z rrs $");
33#include <stdio.h>
34#include <string.h>
35#include <errno.h>
36#include <stdlib.h>
37#include <unistd.h>
38#include <sys/types.h>
39#include <sys/socket.h>
40#include <sys/errno.h>
41#include <sys/syscall.h>
42#include <sys/uio.h>
43#include <netinet/in.h>
44#include <arpa/inet.h>
45#include <netinet/sctp_uio.h>
46#include <netinet/sctp.h>
47
48#include <net/if_dl.h>
49
50#ifndef IN6_IS_ADDR_V4MAPPED
51#define IN6_IS_ADDR_V4MAPPED(a)		      \
52	((*(const uint32_t *)(const void *)(&(a)->s6_addr[0]) == 0) &&	\
53	 (*(const uint32_t *)(const void *)(&(a)->s6_addr[4]) == 0) &&	\
54	 (*(const uint32_t *)(const void *)(&(a)->s6_addr[8]) == ntohl(0x0000ffff)))
55#endif
56
57
58#define SCTP_CONTROL_VEC_SIZE_SND   8192
59#define SCTP_CONTROL_VEC_SIZE_RCV  16384
60#define SCTP_STACK_BUF_SIZE         2048
61#define SCTP_SMALL_IOVEC_SIZE          2
62
63#ifdef SCTP_DEBUG_PRINT_ADDRESS
64
65#define SCTP_STRING_BUF_SZ 256
66
67static void
68SCTPPrintAnAddress(struct sockaddr *a)
69{
70	char stringToPrint[SCTP_STRING_BUF_SZ];
71	u_short prt;
72	char *srcaddr, *txt;
73
74	if (a == NULL) {
75		printf("NULL\n");
76		return;
77	}
78	if (a->sa_family == AF_INET) {
79		srcaddr = (char *)&((struct sockaddr_in *)a)->sin_addr;
80		txt = "IPv4 Address: ";
81		prt = ntohs(((struct sockaddr_in *)a)->sin_port);
82	} else if (a->sa_family == AF_INET6) {
83		srcaddr = (char *)&((struct sockaddr_in6 *)a)->sin6_addr;
84		prt = ntohs(((struct sockaddr_in6 *)a)->sin6_port);
85		txt = "IPv6 Address: ";
86	} else if (a->sa_family == AF_LINK) {
87		int i;
88		char tbuf[SCTP_STRING_BUF_SZ];
89		u_char adbuf[SCTP_STRING_BUF_SZ];
90		struct sockaddr_dl *dl;
91
92		dl = (struct sockaddr_dl *)a;
93		strncpy(tbuf, dl->sdl_data, dl->sdl_nlen);
94		tbuf[dl->sdl_nlen] = 0;
95		printf("Intf:%s (len:%d)Interface index:%d type:%x(%d) ll-len:%d ",
96		    tbuf,
97		    dl->sdl_nlen,
98		    dl->sdl_index,
99		    dl->sdl_type,
100		    dl->sdl_type,
101		    dl->sdl_alen
102		    );
103		memcpy(adbuf, LLADDR(dl), dl->sdl_alen);
104		for (i = 0; i < dl->sdl_alen; i++) {
105			printf("%2.2x", adbuf[i]);
106			if (i < (dl->sdl_alen - 1))
107				printf(":");
108		}
109		printf("\n");
110		return;
111	} else {
112		return;
113	}
114	if (inet_ntop(a->sa_family, srcaddr, stringToPrint, sizeof(stringToPrint))) {
115		if (a->sa_family == AF_INET6) {
116			printf("%s%s:%d scope:%d\n",
117			    txt, stringToPrint, prt,
118			    ((struct sockaddr_in6 *)a)->sin6_scope_id);
119		} else {
120			printf("%s%s:%d\n", txt, stringToPrint, prt);
121		}
122
123	} else {
124		printf("%s unprintable?\n", txt);
125	}
126}
127
128#endif				/* SCTP_DEBUG_PRINT_ADDRESS */
129
130static void
131in6_sin6_2_sin(struct sockaddr_in *sin, struct sockaddr_in6 *sin6)
132{
133	bzero(sin, sizeof(*sin));
134	sin->sin_len = sizeof(struct sockaddr_in);
135	sin->sin_family = AF_INET;
136	sin->sin_port = sin6->sin6_port;
137	sin->sin_addr.s_addr = sin6->sin6_addr.__u6_addr.__u6_addr32[3];
138}
139
140int
141sctp_getaddrlen(sa_family_t family)
142{
143	int error, sd;
144	socklen_t siz;
145	struct sctp_assoc_value av;
146
147	av.assoc_value = family;
148	siz = sizeof(av);
149#if defined(AF_INET)
150	sd = socket(AF_INET, SOCK_SEQPACKET, IPPROTO_SCTP);
151#elif defined(AF_INET6)
152	sd = socket(AF_INET6, SOCK_SEQPACKET, IPPROTO_SCTP);
153#endif
154	if (sd == -1) {
155		return (-1);
156	}
157	error = getsockopt(sd, IPPROTO_SCTP, SCTP_GET_ADDR_LEN, &av, &siz);
158	close(sd);
159	if (error == 0) {
160		return ((int)av.assoc_value);
161	} else {
162		return (-1);
163	}
164}
165
166int
167sctp_connectx(int sd, const struct sockaddr *addrs, int addrcnt,
168    sctp_assoc_t * id)
169{
170	char buf[SCTP_STACK_BUF_SIZE];
171	int i, ret, cnt, *aa;
172	char *cpto;
173	const struct sockaddr *at;
174	sctp_assoc_t *p_id;
175	size_t len = sizeof(int);
176
177	/* validate the address count and list */
178	if ((addrs == NULL) || (addrcnt <= 0)) {
179		errno = EINVAL;
180		return (-1);
181	}
182	at = addrs;
183	cnt = 0;
184	cpto = ((caddr_t)buf + sizeof(int));
185	/* validate all the addresses and get the size */
186	for (i = 0; i < addrcnt; i++) {
187		if (at->sa_family == AF_INET) {
188			if (at->sa_len != sizeof(struct sockaddr_in)) {
189				errno = EINVAL;
190				return (-1);
191			}
192			memcpy(cpto, at, at->sa_len);
193			cpto = ((caddr_t)cpto + at->sa_len);
194			len += at->sa_len;
195		} else if (at->sa_family == AF_INET6) {
196			if (at->sa_len != sizeof(struct sockaddr_in6)) {
197				errno = EINVAL;
198				return (-1);
199			}
200			if (IN6_IS_ADDR_V4MAPPED(&((struct sockaddr_in6 *)at)->sin6_addr)) {
201				len += sizeof(struct sockaddr_in);
202				in6_sin6_2_sin((struct sockaddr_in *)cpto, (struct sockaddr_in6 *)at);
203				cpto = ((caddr_t)cpto + sizeof(struct sockaddr_in));
204				len += sizeof(struct sockaddr_in);
205			} else {
206				memcpy(cpto, at, at->sa_len);
207				cpto = ((caddr_t)cpto + at->sa_len);
208				len += at->sa_len;
209			}
210		} else {
211			errno = EINVAL;
212			return (-1);
213		}
214		if (len > (sizeof(buf) - sizeof(int))) {
215			/* Never enough memory */
216			errno = E2BIG;
217			return (-1);
218		}
219		at = (struct sockaddr *)((caddr_t)at + at->sa_len);
220		cnt++;
221	}
222	/* do we have any? */
223	if (cnt == 0) {
224		errno = EINVAL;
225		return (-1);
226	}
227	aa = (int *)buf;
228	*aa = cnt;
229	ret = setsockopt(sd, IPPROTO_SCTP, SCTP_CONNECT_X, (void *)buf,
230	    (socklen_t) len);
231	if ((ret == 0) && id) {
232		p_id = (sctp_assoc_t *) buf;
233		*id = *p_id;
234	}
235	return (ret);
236}
237
238int
239sctp_bindx(int sd, struct sockaddr *addrs, int addrcnt, int flags)
240{
241	struct sctp_getaddresses *gaddrs;
242	struct sockaddr *sa;
243	struct sockaddr_in *sin;
244	struct sockaddr_in6 *sin6;
245	int i, sz, argsz;
246	uint16_t sport = 0;
247
248	/* validate the flags */
249	if ((flags != SCTP_BINDX_ADD_ADDR) &&
250	    (flags != SCTP_BINDX_REM_ADDR)) {
251		errno = EFAULT;
252		return (-1);
253	}
254	/* validate the address count and list */
255	if ((addrcnt <= 0) || (addrs == NULL)) {
256		errno = EINVAL;
257		return (-1);
258	}
259	argsz = (sizeof(struct sockaddr_storage) +
260	    sizeof(struct sctp_getaddresses));
261	gaddrs = (struct sctp_getaddresses *)calloc(1, argsz);
262	if (gaddrs == NULL) {
263		errno = ENOMEM;
264		return (-1);
265	}
266	/* First pre-screen the addresses */
267	sa = addrs;
268	for (i = 0; i < addrcnt; i++) {
269		sz = sa->sa_len;
270		if (sa->sa_family == AF_INET) {
271			if (sa->sa_len != sizeof(struct sockaddr_in))
272				goto out_error;
273			sin = (struct sockaddr_in *)sa;
274			if (sin->sin_port) {
275				/* non-zero port, check or save */
276				if (sport) {
277					/* Check against our port */
278					if (sport != sin->sin_port) {
279						goto out_error;
280					}
281				} else {
282					/* save off the port */
283					sport = sin->sin_port;
284				}
285			}
286		} else if (sa->sa_family == AF_INET6) {
287			if (sa->sa_len != sizeof(struct sockaddr_in6))
288				goto out_error;
289			sin6 = (struct sockaddr_in6 *)sa;
290			if (sin6->sin6_port) {
291				/* non-zero port, check or save */
292				if (sport) {
293					/* Check against our port */
294					if (sport != sin6->sin6_port) {
295						goto out_error;
296					}
297				} else {
298					/* save off the port */
299					sport = sin6->sin6_port;
300				}
301			}
302		} else {
303			/* invalid address family specified */
304			goto out_error;
305		}
306
307		sa = (struct sockaddr *)((caddr_t)sa + sz);
308	}
309	sa = addrs;
310	/*
311	 * Now if there was a port mentioned, assure that the first address
312	 * has that port to make sure it fails or succeeds correctly.
313	 */
314	if (sport) {
315		sin = (struct sockaddr_in *)sa;
316		sin->sin_port = sport;
317	}
318	for (i = 0; i < addrcnt; i++) {
319		sz = sa->sa_len;
320		if (sa->sa_family == AF_INET) {
321			if (sa->sa_len != sizeof(struct sockaddr_in))
322				goto out_error;
323		} else if (sa->sa_family == AF_INET6) {
324			if (sa->sa_len != sizeof(struct sockaddr_in6))
325				goto out_error;
326		} else {
327			/* invalid address family specified */
328	out_error:
329			free(gaddrs);
330			errno = EINVAL;
331			return (-1);
332		}
333		memset(gaddrs, 0, argsz);
334		gaddrs->sget_assoc_id = 0;
335		memcpy(gaddrs->addr, sa, sz);
336		if (setsockopt(sd, IPPROTO_SCTP, flags, gaddrs,
337		    (socklen_t) argsz) != 0) {
338			free(gaddrs);
339			return (-1);
340		}
341		sa = (struct sockaddr *)((caddr_t)sa + sz);
342	}
343	free(gaddrs);
344	return (0);
345}
346
347
348int
349sctp_opt_info(int sd, sctp_assoc_t id, int opt, void *arg, socklen_t * size)
350{
351	if (arg == NULL) {
352		errno = EINVAL;
353		return (-1);
354	}
355	switch (opt) {
356	case SCTP_RTOINFO:
357		((struct sctp_rtoinfo *)arg)->srto_assoc_id = id;
358		break;
359	case SCTP_ASSOCINFO:
360		((struct sctp_assocparams *)arg)->sasoc_assoc_id = id;
361		break;
362	case SCTP_DEFAULT_SEND_PARAM:
363		((struct sctp_assocparams *)arg)->sasoc_assoc_id = id;
364		break;
365	case SCTP_SET_PEER_PRIMARY_ADDR:
366		((struct sctp_setpeerprim *)arg)->sspp_assoc_id = id;
367		break;
368	case SCTP_PRIMARY_ADDR:
369		((struct sctp_setprim *)arg)->ssp_assoc_id = id;
370		break;
371	case SCTP_PEER_ADDR_PARAMS:
372		((struct sctp_paddrparams *)arg)->spp_assoc_id = id;
373		break;
374	case SCTP_MAXSEG:
375		((struct sctp_assoc_value *)arg)->assoc_id = id;
376		break;
377	case SCTP_AUTH_KEY:
378		((struct sctp_authkey *)arg)->sca_assoc_id = id;
379		break;
380	case SCTP_AUTH_ACTIVE_KEY:
381		((struct sctp_authkeyid *)arg)->scact_assoc_id = id;
382		break;
383	case SCTP_DELAYED_SACK:
384		((struct sctp_sack_info *)arg)->sack_assoc_id = id;
385		break;
386	case SCTP_CONTEXT:
387		((struct sctp_assoc_value *)arg)->assoc_id = id;
388		break;
389	case SCTP_STATUS:
390		((struct sctp_status *)arg)->sstat_assoc_id = id;
391		break;
392	case SCTP_GET_PEER_ADDR_INFO:
393		((struct sctp_paddrinfo *)arg)->spinfo_assoc_id = id;
394		break;
395	case SCTP_PEER_AUTH_CHUNKS:
396		((struct sctp_authchunks *)arg)->gauth_assoc_id = id;
397		break;
398	case SCTP_LOCAL_AUTH_CHUNKS:
399		((struct sctp_authchunks *)arg)->gauth_assoc_id = id;
400		break;
401	default:
402		break;
403	}
404	return (getsockopt(sd, IPPROTO_SCTP, opt, arg, size));
405}
406
407int
408sctp_getpaddrs(int sd, sctp_assoc_t id, struct sockaddr **raddrs)
409{
410	struct sctp_getaddresses *addrs;
411	struct sockaddr *sa;
412	struct sockaddr *re;
413	sctp_assoc_t asoc;
414	caddr_t lim;
415	socklen_t siz;
416	int cnt;
417
418	if (raddrs == NULL) {
419		errno = EFAULT;
420		return (-1);
421	}
422	asoc = id;
423	siz = sizeof(sctp_assoc_t);
424	if (getsockopt(sd, IPPROTO_SCTP, SCTP_GET_REMOTE_ADDR_SIZE,
425	    &asoc, &siz) != 0) {
426		return (-1);
427	}
428	/* size required is returned in 'asoc' */
429	siz = (size_t)asoc;
430	siz += sizeof(struct sctp_getaddresses);
431	addrs = calloc(1, siz);
432	if (addrs == NULL) {
433		return (-1);
434	}
435	addrs->sget_assoc_id = id;
436	/* Now lets get the array of addresses */
437	if (getsockopt(sd, IPPROTO_SCTP, SCTP_GET_PEER_ADDRESSES,
438	    addrs, &siz) != 0) {
439		free(addrs);
440		return (-1);
441	}
442	re = (struct sockaddr *)&addrs->addr[0];
443	*raddrs = re;
444	cnt = 0;
445	sa = (struct sockaddr *)&addrs->addr[0];
446	lim = (caddr_t)addrs + siz;
447	while (((caddr_t)sa < lim) && (sa->sa_len > 0)) {
448		sa = (struct sockaddr *)((caddr_t)sa + sa->sa_len);
449		cnt++;
450	}
451	return (cnt);
452}
453
454void
455sctp_freepaddrs(struct sockaddr *addrs)
456{
457	/* Take away the hidden association id */
458	void *fr_addr;
459
460	fr_addr = (void *)((caddr_t)addrs - sizeof(sctp_assoc_t));
461	/* Now free it */
462	free(fr_addr);
463}
464
465int
466sctp_getladdrs(int sd, sctp_assoc_t id, struct sockaddr **raddrs)
467{
468	struct sctp_getaddresses *addrs;
469	struct sockaddr *re;
470	caddr_t lim;
471	struct sockaddr *sa;
472	int size_of_addresses;
473	socklen_t siz;
474	int cnt;
475
476	if (raddrs == NULL) {
477		errno = EFAULT;
478		return (-1);
479	}
480	size_of_addresses = 0;
481	siz = sizeof(int);
482	if (getsockopt(sd, IPPROTO_SCTP, SCTP_GET_LOCAL_ADDR_SIZE,
483	    &size_of_addresses, &siz) != 0) {
484		errno = ENOMEM;
485		return (-1);
486	}
487	if (size_of_addresses == 0) {
488		errno = ENOTCONN;
489		return (-1);
490	}
491	siz = size_of_addresses + sizeof(struct sockaddr_storage);
492	siz += sizeof(struct sctp_getaddresses);
493	addrs = calloc(1, siz);
494	if (addrs == NULL) {
495		errno = ENOMEM;
496		return (-1);
497	}
498	addrs->sget_assoc_id = id;
499	/* Now lets get the array of addresses */
500	if (getsockopt(sd, IPPROTO_SCTP, SCTP_GET_LOCAL_ADDRESSES, addrs,
501	    &siz) != 0) {
502		free(addrs);
503		errno = ENOMEM;
504		return (-1);
505	}
506	re = (struct sockaddr *)&addrs->addr[0];
507	*raddrs = re;
508	cnt = 0;
509	sa = (struct sockaddr *)&addrs->addr[0];
510	lim = (caddr_t)addrs + siz;
511	while (((caddr_t)sa < lim) && (sa->sa_len > 0)) {
512		sa = (struct sockaddr *)((caddr_t)sa + sa->sa_len);
513		cnt++;
514	}
515	return (cnt);
516}
517
518void
519sctp_freeladdrs(struct sockaddr *addrs)
520{
521	/* Take away the hidden association id */
522	void *fr_addr;
523
524	fr_addr = (void *)((caddr_t)addrs - sizeof(sctp_assoc_t));
525	/* Now free it */
526	free(fr_addr);
527}
528
529
530ssize_t
531sctp_sendmsg(int s,
532    const void *data,
533    size_t len,
534    const struct sockaddr *to,
535    socklen_t tolen,
536    uint32_t ppid,
537    uint32_t flags,
538    uint16_t stream_no,
539    uint32_t timetolive,
540    uint32_t context)
541{
542#ifdef SYS_sctp_generic_sendmsg
543	struct sctp_sndrcvinfo sinfo;
544
545	sinfo.sinfo_ppid = ppid;
546	sinfo.sinfo_flags = flags;
547	sinfo.sinfo_stream = stream_no;
548	sinfo.sinfo_timetolive = timetolive;
549	sinfo.sinfo_context = context;
550	sinfo.sinfo_assoc_id = 0;
551	return (syscall(SYS_sctp_generic_sendmsg, s,
552	    data, len, to, tolen, &sinfo, 0));
553#else
554
555	ssize_t sz;
556	struct msghdr msg;
557	struct sctp_sndrcvinfo *s_info;
558	struct iovec iov[SCTP_SMALL_IOVEC_SIZE];
559	char controlVector[SCTP_CONTROL_VEC_SIZE_RCV];
560	struct cmsghdr *cmsg;
561	struct sockaddr *who = NULL;
562	union {
563		struct sockaddr_in in;
564		struct sockaddr_in6 in6;
565	}     addr;
566
567/*
568  fprintf(io, "sctp_sendmsg(sd:%d, data:%x, len:%d, to:%x, tolen:%d, ppid:%x, flags:%x str:%d ttl:%d ctx:%x\n",
569  s,
570  (u_int)data,
571  (int)len,
572  (u_int)to,
573  (int)tolen,
574  ppid, flags,
575  (int)stream_no,
576  (int)timetolive,
577  (u_int)context);
578  fflush(io);
579*/
580	if ((tolen > 0) && ((to == NULL) || (tolen < sizeof(struct sockaddr)))) {
581		errno = EINVAL;
582		return -1;
583	}
584	if (to && (tolen > 0)) {
585		if (to->sa_family == AF_INET) {
586			if (tolen != sizeof(struct sockaddr_in)) {
587				errno = EINVAL;
588				return -1;
589			}
590			if ((to->sa_len > 0) && (to->sa_len != sizeof(struct sockaddr_in))) {
591				errno = EINVAL;
592				return -1;
593			}
594			memcpy(&addr, to, sizeof(struct sockaddr_in));
595			addr.in.sin_len = sizeof(struct sockaddr_in);
596		} else if (to->sa_family == AF_INET6) {
597			if (tolen != sizeof(struct sockaddr_in6)) {
598				errno = EINVAL;
599				return -1;
600			}
601			if ((to->sa_len > 0) && (to->sa_len != sizeof(struct sockaddr_in6))) {
602				errno = EINVAL;
603				return -1;
604			}
605			memcpy(&addr, to, sizeof(struct sockaddr_in6));
606			addr.in6.sin6_len = sizeof(struct sockaddr_in6);
607		} else {
608			errno = EAFNOSUPPORT;
609			return -1;
610		}
611		who = (struct sockaddr *)&addr;
612	}
613	iov[0].iov_base = (char *)data;
614	iov[0].iov_len = len;
615	iov[1].iov_base = NULL;
616	iov[1].iov_len = 0;
617
618	if (who) {
619		msg.msg_name = (caddr_t)who;
620		msg.msg_namelen = who->sa_len;
621	} else {
622		msg.msg_name = (caddr_t)NULL;
623		msg.msg_namelen = 0;
624	}
625	msg.msg_iov = iov;
626	msg.msg_iovlen = 1;
627	msg.msg_control = (caddr_t)controlVector;
628
629	cmsg = (struct cmsghdr *)controlVector;
630
631	cmsg->cmsg_level = IPPROTO_SCTP;
632	cmsg->cmsg_type = SCTP_SNDRCV;
633	cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndrcvinfo));
634	s_info = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
635
636	s_info->sinfo_stream = stream_no;
637	s_info->sinfo_ssn = 0;
638	s_info->sinfo_flags = flags;
639	s_info->sinfo_ppid = ppid;
640	s_info->sinfo_context = context;
641	s_info->sinfo_assoc_id = 0;
642	s_info->sinfo_timetolive = timetolive;
643	errno = 0;
644	msg.msg_controllen = cmsg->cmsg_len;
645	sz = sendmsg(s, &msg, 0);
646	return (sz);
647#endif
648}
649
650
651sctp_assoc_t
652sctp_getassocid(int sd, struct sockaddr *sa)
653{
654	struct sctp_paddrinfo sp;
655	socklen_t siz;
656
657	/* First get the assoc id */
658	siz = sizeof(sp);
659	memset(&sp, 0, sizeof(sp));
660	memcpy((caddr_t)&sp.spinfo_address, sa, sa->sa_len);
661	errno = 0;
662	if (getsockopt(sd, IPPROTO_SCTP,
663	    SCTP_GET_PEER_ADDR_INFO, &sp, &siz) != 0) {
664		return ((sctp_assoc_t) 0);
665	}
666	/* We depend on the fact that 0 can never be returned */
667	return (sp.spinfo_assoc_id);
668}
669
670ssize_t
671sctp_send(int sd, const void *data, size_t len,
672    const struct sctp_sndrcvinfo *sinfo,
673    int flags)
674{
675
676#ifdef SYS_sctp_generic_sendmsg
677	struct sockaddr *to = NULL;
678
679	return (syscall(SYS_sctp_generic_sendmsg, sd,
680	    data, len, to, 0, sinfo, flags));
681#else
682	ssize_t sz;
683	struct msghdr msg;
684	struct iovec iov[SCTP_SMALL_IOVEC_SIZE];
685	struct sctp_sndrcvinfo *s_info;
686	char controlVector[SCTP_CONTROL_VEC_SIZE_SND];
687	struct cmsghdr *cmsg;
688
689	if (sinfo == NULL) {
690		errno = EINVAL;
691		return (-1);
692	}
693	iov[0].iov_base = (char *)data;
694	iov[0].iov_len = len;
695	iov[1].iov_base = NULL;
696	iov[1].iov_len = 0;
697
698	msg.msg_name = 0;
699	msg.msg_namelen = 0;
700	msg.msg_iov = iov;
701	msg.msg_iovlen = 1;
702	msg.msg_control = (caddr_t)controlVector;
703
704	cmsg = (struct cmsghdr *)controlVector;
705
706	cmsg->cmsg_level = IPPROTO_SCTP;
707	cmsg->cmsg_type = SCTP_SNDRCV;
708	cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndrcvinfo));
709	s_info = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
710	/* copy in the data */
711	*s_info = *sinfo;
712	errno = 0;
713	msg.msg_controllen = cmsg->cmsg_len;
714	sz = sendmsg(sd, &msg, flags);
715	return (sz);
716#endif
717}
718
719
720
721ssize_t
722sctp_sendx(int sd, const void *msg, size_t msg_len,
723    struct sockaddr *addrs, int addrcnt,
724    struct sctp_sndrcvinfo *sinfo,
725    int flags)
726{
727	struct sctp_sndrcvinfo __sinfo;
728	ssize_t ret;
729	int i, cnt, *aa, saved_errno;
730	char *buf;
731	int add_len, len, no_end_cx = 0;
732	struct sockaddr *at;
733
734	if (addrs == NULL) {
735		errno = EINVAL;
736		return (-1);
737	}
738#ifdef SYS_sctp_generic_sendmsg
739	if (addrcnt < SCTP_SMALL_IOVEC_SIZE) {
740		socklen_t l;
741
742		/*
743		 * Quick way, we don't need to do a connectx so lets use the
744		 * syscall directly.
745		 */
746		l = addrs->sa_len;
747		return (syscall(SYS_sctp_generic_sendmsg, sd,
748		    msg, msg_len, addrs, l, sinfo, flags));
749	}
750#endif
751
752	len = sizeof(int);
753	at = addrs;
754	cnt = 0;
755	/* validate all the addresses and get the size */
756	for (i = 0; i < addrcnt; i++) {
757		if (at->sa_family == AF_INET) {
758			add_len = sizeof(struct sockaddr_in);
759		} else if (at->sa_family == AF_INET6) {
760			add_len = sizeof(struct sockaddr_in6);
761		} else {
762			errno = EINVAL;
763			return (-1);
764		}
765		len += add_len;
766		at = (struct sockaddr *)((caddr_t)at + add_len);
767		cnt++;
768	}
769	/* do we have any? */
770	if (cnt == 0) {
771		errno = EINVAL;
772		return (-1);
773	}
774	buf = malloc(len);
775	if (buf == NULL) {
776		return (-1);
777	}
778	aa = (int *)buf;
779	*aa = cnt;
780	aa++;
781	memcpy((caddr_t)aa, addrs, (len - sizeof(int)));
782	ret = setsockopt(sd, IPPROTO_SCTP, SCTP_CONNECT_X_DELAYED, (void *)buf,
783	    (socklen_t) len);
784
785	free(buf);
786	if (ret != 0) {
787		if (errno == EALREADY) {
788			no_end_cx = 1;
789			goto continue_send;
790		}
791		return (ret);
792	}
793continue_send:
794	if (sinfo == NULL) {
795		sinfo = &__sinfo;
796		memset(&__sinfo, 0, sizeof(__sinfo));
797	}
798	sinfo->sinfo_assoc_id = sctp_getassocid(sd, addrs);
799	if (sinfo->sinfo_assoc_id == 0) {
800		printf("Huh, can't get associd? TSNH!\n");
801		(void)setsockopt(sd, IPPROTO_SCTP, SCTP_CONNECT_X_COMPLETE, (void *)addrs,
802		    (socklen_t) addrs->sa_len);
803		errno = ENOENT;
804		return (-1);
805	}
806	ret = sctp_send(sd, msg, msg_len, sinfo, flags);
807	saved_errno = errno;
808	if (no_end_cx == 0)
809		(void)setsockopt(sd, IPPROTO_SCTP, SCTP_CONNECT_X_COMPLETE, (void *)addrs,
810		    (socklen_t) addrs->sa_len);
811
812	errno = saved_errno;
813	return (ret);
814}
815
816ssize_t
817sctp_sendmsgx(int sd,
818    const void *msg,
819    size_t len,
820    struct sockaddr *addrs,
821    int addrcnt,
822    uint32_t ppid,
823    uint32_t flags,
824    uint16_t stream_no,
825    uint32_t timetolive,
826    uint32_t context)
827{
828	struct sctp_sndrcvinfo sinfo;
829
830	memset((void *)&sinfo, 0, sizeof(struct sctp_sndrcvinfo));
831	sinfo.sinfo_ppid = ppid;
832	sinfo.sinfo_flags = flags;
833	sinfo.sinfo_ssn = stream_no;
834	sinfo.sinfo_timetolive = timetolive;
835	sinfo.sinfo_context = context;
836	return sctp_sendx(sd, msg, len, addrs, addrcnt, &sinfo, 0);
837}
838
839ssize_t
840sctp_recvmsg(int s,
841    void *dbuf,
842    size_t len,
843    struct sockaddr *from,
844    socklen_t * fromlen,
845    struct sctp_sndrcvinfo *sinfo,
846    int *msg_flags)
847{
848#ifdef SYS_sctp_generic_recvmsg
849	struct iovec iov[SCTP_SMALL_IOVEC_SIZE];
850
851	iov[0].iov_base = dbuf;
852	iov[0].iov_len = len;
853	return (syscall(SYS_sctp_generic_recvmsg, s,
854	    iov, 1, from, fromlen, sinfo, msg_flags));
855#else
856	struct sctp_sndrcvinfo *s_info;
857	ssize_t sz;
858	int sinfo_found = 0;
859	struct msghdr msg;
860	struct iovec iov[SCTP_SMALL_IOVEC_SIZE];
861	char controlVector[SCTP_CONTROL_VEC_SIZE_RCV];
862	struct cmsghdr *cmsg;
863
864	if (msg_flags == NULL) {
865		errno = EINVAL;
866		return (-1);
867	}
868	msg.msg_flags = 0;
869	iov[0].iov_base = dbuf;
870	iov[0].iov_len = len;
871	iov[1].iov_base = NULL;
872	iov[1].iov_len = 0;
873	msg.msg_name = (caddr_t)from;
874	if (fromlen == NULL)
875		msg.msg_namelen = 0;
876	else
877		msg.msg_namelen = *fromlen;
878	msg.msg_iov = iov;
879	msg.msg_iovlen = 1;
880	msg.msg_control = (caddr_t)controlVector;
881	msg.msg_controllen = sizeof(controlVector);
882	errno = 0;
883	sz = recvmsg(s, &msg, *msg_flags);
884	if (sz <= 0)
885		return (sz);
886
887	s_info = NULL;
888	len = sz;
889	*msg_flags = msg.msg_flags;
890	if (sinfo)
891		sinfo->sinfo_assoc_id = 0;
892
893	if ((msg.msg_controllen) && sinfo) {
894		/*
895		 * parse through and see if we find the sctp_sndrcvinfo (if
896		 * the user wants it).
897		 */
898		cmsg = (struct cmsghdr *)controlVector;
899		while (cmsg) {
900			if ((cmsg->cmsg_len == 0) || (cmsg->cmsg_len > msg.msg_controllen)) {
901				break;
902			}
903			if (cmsg->cmsg_level == IPPROTO_SCTP) {
904				if (cmsg->cmsg_type == SCTP_SNDRCV) {
905					/* Got it */
906					s_info = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
907					/* Copy it to the user */
908					if (sinfo)
909						*sinfo = *s_info;
910					sinfo_found = 1;
911					break;
912				} else if (cmsg->cmsg_type == SCTP_EXTRCV) {
913					/*
914					 * Got it, presumably the user has
915					 * asked for this extra info, so the
916					 * structure holds more room :-D
917					 */
918					s_info = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
919					/* Copy it to the user */
920					if (sinfo) {
921						memcpy(sinfo, s_info, sizeof(struct sctp_extrcvinfo));
922					}
923					sinfo_found = 1;
924					break;
925
926				}
927			}
928			cmsg = CMSG_NXTHDR(&msg, cmsg);
929		}
930	}
931	return (sz);
932#endif
933}
934
935
936#if defined(HAVE_SCTP_PEELOFF_SOCKOPT)
937#include <netinet/sctp_peeloff.h>
938
939int
940sctp_peeloff(int sd, sctp_assoc_t assoc_id)
941{
942	struct sctp_peeloff_opt peeloff;
943	int error;
944	socklen_t optlen;
945
946	/* set in the socket option params */
947	memset(&peeloff, 0, sizeof(peeloff));
948	peeloff.s = sd;
949	peeloff.assoc_id = assoc_id;
950	optlen = sizeof(peeloff);
951	error = getsockopt(sd, IPPROTO_SCTP, SCTP_PEELOFF, (void *)&peeloff,
952	    &optlen);
953	if (error) {
954		errno = error;
955		return (-1);
956	} else {
957		return (peeloff.new_sd);
958	}
959}
960
961#endif
962
963#if !defined(SYS_sctp_peeloff) && !defined(HAVE_SCTP_PEELOFF_SOCKOPT)
964
965int
966sctp_peeloff(int sd, sctp_assoc_t assoc_id)
967{
968	/* NOT supported, return invalid sd */
969	errno = ENOTSUP;
970	return (-1);
971}
972
973#endif
974#if defined(SYS_sctp_peeloff) && !defined(HAVE_SCTP_PEELOFF_SOCKOPT)
975int
976sctp_peeloff(int sd, sctp_assoc_t assoc_id)
977{
978	return (syscall(SYS_sctp_peeloff, sd, assoc_id));
979}
980
981#endif
982
983
984#undef SCTP_CONTROL_VEC_SIZE_SND
985#undef SCTP_CONTROL_VEC_SIZE_RCV
986#undef SCTP_STACK_BUF_SIZE
987#undef SCTP_SMALL_IOVEC_SIZE
988