Fix incorrect OPENSSL_assert() usage.
[openssl.git] / crypto / bio / bss_dgram.c
1 /* crypto/bio/bio_dgram.c */
2 /* 
3  * DTLS implementation written by Nagendra Modadugu
4  * (nagendra@cs.stanford.edu) for the OpenSSL project 2005.  
5  */
6 /* ====================================================================
7  * Copyright (c) 1999-2005 The OpenSSL Project.  All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  *
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer. 
15  *
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in
18  *    the documentation and/or other materials provided with the
19  *    distribution.
20  *
21  * 3. All advertising materials mentioning features or use of this
22  *    software must display the following acknowledgment:
23  *    "This product includes software developed by the OpenSSL Project
24  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
25  *
26  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
27  *    endorse or promote products derived from this software without
28  *    prior written permission. For written permission, please contact
29  *    openssl-core@OpenSSL.org.
30  *
31  * 5. Products derived from this software may not be called "OpenSSL"
32  *    nor may "OpenSSL" appear in their names without prior written
33  *    permission of the OpenSSL Project.
34  *
35  * 6. Redistributions of any form whatsoever must retain the following
36  *    acknowledgment:
37  *    "This product includes software developed by the OpenSSL Project
38  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
39  *
40  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
41  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
42  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
43  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
44  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
45  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
46  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
47  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
49  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
50  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
51  * OF THE POSSIBILITY OF SUCH DAMAGE.
52  * ====================================================================
53  *
54  * This product includes cryptographic software written by Eric Young
55  * (eay@cryptsoft.com).  This product includes software written by Tim
56  * Hudson (tjh@cryptsoft.com).
57  *
58  */
59
60
61 #include <stdio.h>
62 #include <errno.h>
63 #define USE_SOCKETS
64 #include "cryptlib.h"
65
66 #include <openssl/bio.h>
67 #ifndef OPENSSL_NO_DGRAM
68
69 #if defined(OPENSSL_SYS_WIN32) || defined(OPENSSL_SYS_VMS)
70 #include <sys/timeb.h>
71 #endif
72
73 #ifndef OPENSSL_NO_SCTP
74 #include <netinet/sctp.h>
75 #include <fcntl.h>
76 #define OPENSSL_SCTP_DATA_CHUNK_TYPE            0x00
77 #define OPENSSL_SCTP_FORWARD_CUM_TSN_CHUNK_TYPE 0xc0
78 #endif
79
80 #if defined(OPENSSL_SYS_LINUX) && !defined(IP_MTU)
81 #define IP_MTU      14 /* linux is lame */
82 #endif
83
84 #if defined(__FreeBSD__) && defined(IN6_IS_ADDR_V4MAPPED)
85 /* Standard definition causes type-punning problems. */
86 #undef IN6_IS_ADDR_V4MAPPED
87 #define s6_addr32 __u6_addr.__u6_addr32
88 #define IN6_IS_ADDR_V4MAPPED(a)               \
89         (((a)->s6_addr32[0] == 0) &&          \
90          ((a)->s6_addr32[1] == 0) &&          \
91          ((a)->s6_addr32[2] == htonl(0x0000ffff)))
92 #endif
93
94 #ifdef WATT32
95 #define sock_write SockWrite  /* Watt-32 uses same names */
96 #define sock_read  SockRead
97 #define sock_puts  SockPuts
98 #endif
99
100 static int dgram_write(BIO *h, const char *buf, int num);
101 static int dgram_read(BIO *h, char *buf, int size);
102 static int dgram_puts(BIO *h, const char *str);
103 static long dgram_ctrl(BIO *h, int cmd, long arg1, void *arg2);
104 static int dgram_new(BIO *h);
105 static int dgram_free(BIO *data);
106 static int dgram_clear(BIO *bio);
107
108 #ifndef OPENSSL_NO_SCTP
109 static int dgram_sctp_write(BIO *h, const char *buf, int num);
110 static int dgram_sctp_read(BIO *h, char *buf, int size);
111 static int dgram_sctp_puts(BIO *h, const char *str);
112 static long dgram_sctp_ctrl(BIO *h, int cmd, long arg1, void *arg2);
113 static int dgram_sctp_new(BIO *h);
114 static int dgram_sctp_free(BIO *data);
115 #ifdef SCTP_AUTHENTICATION_EVENT
116 static void dgram_sctp_handle_auth_free_key_event(BIO *b, union sctp_notification *snp);
117 #endif
118 #endif
119
120 static int BIO_dgram_should_retry(int s);
121
122 static void get_current_time(struct timeval *t);
123
124 static BIO_METHOD methods_dgramp=
125         {
126         BIO_TYPE_DGRAM,
127         "datagram socket",
128         dgram_write,
129         dgram_read,
130         dgram_puts,
131         NULL, /* dgram_gets, */
132         dgram_ctrl,
133         dgram_new,
134         dgram_free,
135         NULL,
136         };
137
138 #ifndef OPENSSL_NO_SCTP
139 static BIO_METHOD methods_dgramp_sctp=
140         {
141         BIO_TYPE_DGRAM_SCTP,
142         "datagram sctp socket",
143         dgram_sctp_write,
144         dgram_sctp_read,
145         dgram_sctp_puts,
146         NULL, /* dgram_gets, */
147         dgram_sctp_ctrl,
148         dgram_sctp_new,
149         dgram_sctp_free,
150         NULL,
151         };
152 #endif
153
154 typedef struct bio_dgram_data_st
155         {
156         union {
157                 struct sockaddr sa;
158                 struct sockaddr_in sa_in;
159 #if OPENSSL_USE_IPV6
160                 struct sockaddr_in6 sa_in6;
161 #endif
162         } peer;
163         unsigned int connected;
164         unsigned int _errno;
165         unsigned int mtu;
166         struct timeval next_timeout;
167         struct timeval socket_timeout;
168         } bio_dgram_data;
169
170 #ifndef OPENSSL_NO_SCTP
171 typedef struct bio_dgram_sctp_save_message_st
172         {
173         BIO *bio;
174         char *data;
175         int length;
176         } bio_dgram_sctp_save_message;
177
178 typedef struct bio_dgram_sctp_data_st
179         {
180         union {
181                 struct sockaddr sa;
182                 struct sockaddr_in sa_in;
183 #if OPENSSL_USE_IPV6
184                 struct sockaddr_in6 sa_in6;
185 #endif
186         } peer;
187         unsigned int connected;
188         unsigned int _errno;
189         unsigned int mtu;
190         struct bio_dgram_sctp_sndinfo sndinfo;
191         struct bio_dgram_sctp_rcvinfo rcvinfo;
192         struct bio_dgram_sctp_prinfo prinfo;
193         void (*handle_notifications)(BIO *bio, void *context, void *buf);
194         void* notification_context;
195         int in_handshake;
196         int ccs_rcvd;
197         int ccs_sent;
198         int save_shutdown;
199         int peer_auth_tested;
200         bio_dgram_sctp_save_message saved_message;
201         } bio_dgram_sctp_data;
202 #endif
203
204 BIO_METHOD *BIO_s_datagram(void)
205         {
206         return(&methods_dgramp);
207         }
208
209 BIO *BIO_new_dgram(int fd, int close_flag)
210         {
211         BIO *ret;
212
213         ret=BIO_new(BIO_s_datagram());
214         if (ret == NULL) return(NULL);
215         BIO_set_fd(ret,fd,close_flag);
216         return(ret);
217         }
218
219 static int dgram_new(BIO *bi)
220         {
221         bio_dgram_data *data = NULL;
222
223         bi->init=0;
224         bi->num=0;
225         data = OPENSSL_malloc(sizeof(bio_dgram_data));
226         if (data == NULL)
227                 return 0;
228         memset(data, 0x00, sizeof(bio_dgram_data));
229     bi->ptr = data;
230
231         bi->flags=0;
232         return(1);
233         }
234
235 static int dgram_free(BIO *a)
236         {
237         bio_dgram_data *data;
238
239         if (a == NULL) return(0);
240         if ( ! dgram_clear(a))
241                 return 0;
242
243         data = (bio_dgram_data *)a->ptr;
244         if(data != NULL) OPENSSL_free(data);
245
246         return(1);
247         }
248
249 static int dgram_clear(BIO *a)
250         {
251         if (a == NULL) return(0);
252         if (a->shutdown)
253                 {
254                 if (a->init)
255                         {
256                         SHUTDOWN2(a->num);
257                         }
258                 a->init=0;
259                 a->flags=0;
260                 }
261         return(1);
262         }
263
264 static void dgram_adjust_rcv_timeout(BIO *b)
265         {
266 #if defined(SO_RCVTIMEO)
267         bio_dgram_data *data = (bio_dgram_data *)b->ptr;
268         union { size_t s; int i; } sz = {0};
269
270         /* Is a timer active? */
271         if (data->next_timeout.tv_sec > 0 || data->next_timeout.tv_usec > 0)
272                 {
273                 struct timeval timenow, timeleft;
274
275                 /* Read current socket timeout */
276 #ifdef OPENSSL_SYS_WINDOWS
277                 int timeout;
278
279                 sz.i = sizeof(timeout);
280                 if (getsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
281                                            (void*)&timeout, &sz.i) < 0)
282                         { perror("getsockopt"); }
283                 else
284                         {
285                         data->socket_timeout.tv_sec = timeout / 1000;
286                         data->socket_timeout.tv_usec = (timeout % 1000) * 1000;
287                         }
288 #else
289                 sz.i = sizeof(data->socket_timeout);
290                 if ( getsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO, 
291                                                 &(data->socket_timeout), (void *)&sz) < 0)
292                         { perror("getsockopt"); }
293                 else if (sizeof(sz.s)!=sizeof(sz.i) && sz.i==0)
294                         OPENSSL_assert(sz.s<=sizeof(data->socket_timeout));
295 #endif
296
297                 /* Get current time */
298                 get_current_time(&timenow);
299
300                 /* Calculate time left until timer expires */
301                 memcpy(&timeleft, &(data->next_timeout), sizeof(struct timeval));
302                 timeleft.tv_sec -= timenow.tv_sec;
303                 timeleft.tv_usec -= timenow.tv_usec;
304                 if (timeleft.tv_usec < 0)
305                         {
306                         timeleft.tv_sec--;
307                         timeleft.tv_usec += 1000000;
308                         }
309
310                 if (timeleft.tv_sec < 0)
311                         {
312                         timeleft.tv_sec = 0;
313                         timeleft.tv_usec = 1;
314                         }
315
316                 /* Adjust socket timeout if next handhake message timer
317                  * will expire earlier.
318                  */
319                 if ((data->socket_timeout.tv_sec == 0 && data->socket_timeout.tv_usec == 0) ||
320                         (data->socket_timeout.tv_sec > timeleft.tv_sec) ||
321                         (data->socket_timeout.tv_sec == timeleft.tv_sec &&
322                          data->socket_timeout.tv_usec >= timeleft.tv_usec))
323                         {
324 #ifdef OPENSSL_SYS_WINDOWS
325                         timeout = timeleft.tv_sec * 1000 + timeleft.tv_usec / 1000;
326                         if (setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
327                                                    (void*)&timeout, sizeof(timeout)) < 0)
328                                 { perror("setsockopt"); }
329 #else
330                         if ( setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO, &timeleft,
331                                                         sizeof(struct timeval)) < 0)
332                                 { perror("setsockopt"); }
333 #endif
334                         }
335                 }
336 #endif
337         }
338
339 static void dgram_reset_rcv_timeout(BIO *b)
340         {
341 #if defined(SO_RCVTIMEO)
342         bio_dgram_data *data = (bio_dgram_data *)b->ptr;
343
344         /* Is a timer active? */
345         if (data->next_timeout.tv_sec > 0 || data->next_timeout.tv_usec > 0)
346                 {
347 #ifdef OPENSSL_SYS_WINDOWS
348                 int timeout = data->socket_timeout.tv_sec * 1000 +
349                                           data->socket_timeout.tv_usec / 1000;
350                 if (setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
351                                            (void*)&timeout, sizeof(timeout)) < 0)
352                         { perror("setsockopt"); }
353 #else
354                 if ( setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO, &(data->socket_timeout),
355                                                 sizeof(struct timeval)) < 0)
356                         { perror("setsockopt"); }
357 #endif
358                 }
359 #endif
360         }
361
362 static int dgram_read(BIO *b, char *out, int outl)
363         {
364         int ret=0;
365         bio_dgram_data *data = (bio_dgram_data *)b->ptr;
366
367         struct  {
368         /*
369          * See commentary in b_sock.c. <appro>
370          */
371         union   { size_t s; int i; } len;
372         union   {
373                 struct sockaddr sa;
374                 struct sockaddr_in sa_in;
375 #if OPENSSL_USE_IPV6
376                 struct sockaddr_in6 sa_in6;
377 #endif
378                 } peer;
379         } sa;
380
381         sa.len.s=0;
382         sa.len.i=sizeof(sa.peer);
383
384         if (out != NULL)
385                 {
386                 clear_socket_error();
387                 memset(&sa.peer, 0x00, sizeof(sa.peer));
388                 dgram_adjust_rcv_timeout(b);
389                 ret=recvfrom(b->num,out,outl,0,&sa.peer.sa,(void *)&sa.len);
390                 if (sizeof(sa.len.i)!=sizeof(sa.len.s) && sa.len.i==0)
391                         {
392                         OPENSSL_assert(sa.len.s<=sizeof(sa.peer));
393                         sa.len.i = (int)sa.len.s;
394                         }
395
396                 if ( ! data->connected  && ret >= 0)
397                         BIO_ctrl(b, BIO_CTRL_DGRAM_SET_PEER, 0, &sa.peer);
398
399                 BIO_clear_retry_flags(b);
400                 if (ret < 0)
401                         {
402                         if (BIO_dgram_should_retry(ret))
403                                 {
404                                 BIO_set_retry_read(b);
405                                 data->_errno = get_last_socket_error();
406                                 }
407                         }
408
409                 dgram_reset_rcv_timeout(b);
410                 }
411         return(ret);
412         }
413
414 static int dgram_write(BIO *b, const char *in, int inl)
415         {
416         int ret;
417         bio_dgram_data *data = (bio_dgram_data *)b->ptr;
418         clear_socket_error();
419
420         if ( data->connected )
421                 ret=writesocket(b->num,in,inl);
422         else
423                 {
424                 int peerlen = sizeof(data->peer);
425
426                 if (data->peer.sa.sa_family == AF_INET)
427                         peerlen = sizeof(data->peer.sa_in);
428 #if OPENSSL_USE_IPV6
429                 else if (data->peer.sa.sa_family == AF_INET6)
430                         peerlen = sizeof(data->peer.sa_in6);
431 #endif
432 #if defined(NETWARE_CLIB) && defined(NETWARE_BSDSOCK)
433                 ret=sendto(b->num, (char *)in, inl, 0, &data->peer.sa, peerlen);
434 #else
435                 ret=sendto(b->num, in, inl, 0, &data->peer.sa, peerlen);
436 #endif
437                 }
438
439         BIO_clear_retry_flags(b);
440         if (ret <= 0)
441                 {
442                 if (BIO_dgram_should_retry(ret))
443                         {
444                         BIO_set_retry_write(b);  
445                         data->_errno = get_last_socket_error();
446
447 #if 0 /* higher layers are responsible for querying MTU, if necessary */
448                         if ( data->_errno == EMSGSIZE)
449                                 /* retrieve the new MTU */
450                                 BIO_ctrl(b, BIO_CTRL_DGRAM_QUERY_MTU, 0, NULL);
451 #endif
452                         }
453                 }
454         return(ret);
455         }
456
457 static long dgram_get_mtu_overhead(bio_dgram_data *data)
458         {
459         long ret;
460
461         switch (data->peer.sa.sa_family)
462                 {
463                 case AF_INET:
464                         /* Assume this is UDP - 20 bytes for IP, 8 bytes for UDP */
465                         ret = 28;
466                         break;
467 #if OPENSSL_USE_IPV6
468                 case AF_INET6:
469 #ifdef IN6_IS_ADDR_V4MAPPED
470                         if (IN6_IS_ADDR_V4MAPPED(&data->peer.sa_in6.sin6_addr))
471                                 /* Assume this is UDP - 20 bytes for IP, 8 bytes for UDP */
472                                 ret = 28;
473                         else
474 #endif
475                                 /* Assume this is UDP - 40 bytes for IP, 8 bytes for UDP */
476                                 ret = 48;
477                         break;
478 #endif
479                 default:
480                         /* We don't know. Go with the historical default */
481                         ret = 28;
482                         break;
483                 }
484         return ret;
485         }
486
487 static long dgram_ctrl(BIO *b, int cmd, long num, void *ptr)
488         {
489         long ret=1;
490         int *ip;
491         struct sockaddr *to = NULL;
492         bio_dgram_data *data = NULL;
493 #if defined(OPENSSL_SYS_LINUX) && (defined(IP_MTU_DISCOVER) || defined(IP_MTU))
494         int sockopt_val = 0;
495         socklen_t sockopt_len;  /* assume that system supporting IP_MTU is
496                                  * modern enough to define socklen_t */
497         socklen_t addr_len;
498         union   {
499                 struct sockaddr sa;
500                 struct sockaddr_in s4;
501 #if OPENSSL_USE_IPV6
502                 struct sockaddr_in6 s6;
503 #endif
504                 } addr;
505 #endif
506
507         data = (bio_dgram_data *)b->ptr;
508
509         switch (cmd)
510                 {
511         case BIO_CTRL_RESET:
512                 num=0;
513         case BIO_C_FILE_SEEK:
514                 ret=0;
515                 break;
516         case BIO_C_FILE_TELL:
517         case BIO_CTRL_INFO:
518                 ret=0;
519                 break;
520         case BIO_C_SET_FD:
521                 dgram_clear(b);
522                 b->num= *((int *)ptr);
523                 b->shutdown=(int)num;
524                 b->init=1;
525                 break;
526         case BIO_C_GET_FD:
527                 if (b->init)
528                         {
529                         ip=(int *)ptr;
530                         if (ip != NULL) *ip=b->num;
531                         ret=b->num;
532                         }
533                 else
534                         ret= -1;
535                 break;
536         case BIO_CTRL_GET_CLOSE:
537                 ret=b->shutdown;
538                 break;
539         case BIO_CTRL_SET_CLOSE:
540                 b->shutdown=(int)num;
541                 break;
542         case BIO_CTRL_PENDING:
543         case BIO_CTRL_WPENDING:
544                 ret=0;
545                 break;
546         case BIO_CTRL_DUP:
547         case BIO_CTRL_FLUSH:
548                 ret=1;
549                 break;
550         case BIO_CTRL_DGRAM_CONNECT:
551                 to = (struct sockaddr *)ptr;
552 #if 0
553                 if (connect(b->num, to, sizeof(struct sockaddr)) < 0)
554                         { perror("connect"); ret = 0; }
555                 else
556                         {
557 #endif
558                         switch (to->sa_family)
559                                 {
560                                 case AF_INET:
561                                         memcpy(&data->peer,to,sizeof(data->peer.sa_in));
562                                         break;
563 #if OPENSSL_USE_IPV6
564                                 case AF_INET6:
565                                         memcpy(&data->peer,to,sizeof(data->peer.sa_in6));
566                                         break;
567 #endif
568                                 default:
569                                         memcpy(&data->peer,to,sizeof(data->peer.sa));
570                                         break;
571                                 }
572 #if 0
573                         }
574 #endif
575                 break;
576                 /* (Linux)kernel sets DF bit on outgoing IP packets */
577         case BIO_CTRL_DGRAM_MTU_DISCOVER:
578 #if defined(OPENSSL_SYS_LINUX) && defined(IP_MTU_DISCOVER) && defined(IP_PMTUDISC_DO)
579                 addr_len = (socklen_t)sizeof(addr);
580                 memset((void *)&addr, 0, sizeof(addr));
581                 if (getsockname(b->num, &addr.sa, &addr_len) < 0)
582                         {
583                         ret = 0;
584                         break;
585                         }
586                 switch (addr.sa.sa_family)
587                         {
588                 case AF_INET:
589                         sockopt_val = IP_PMTUDISC_DO;
590                         if ((ret = setsockopt(b->num, IPPROTO_IP, IP_MTU_DISCOVER,
591                                 &sockopt_val, sizeof(sockopt_val))) < 0)
592                                 perror("setsockopt");
593                         break;
594 #if OPENSSL_USE_IPV6 && defined(IPV6_MTU_DISCOVER) && defined(IPV6_PMTUDISC_DO)
595                 case AF_INET6:
596                         sockopt_val = IPV6_PMTUDISC_DO;
597                         if ((ret = setsockopt(b->num, IPPROTO_IPV6, IPV6_MTU_DISCOVER,
598                                 &sockopt_val, sizeof(sockopt_val))) < 0)
599                                 perror("setsockopt");
600                         break;
601 #endif
602                 default:
603                         ret = -1;
604                         break;
605                         }
606                 ret = -1;
607 #else
608                 break;
609 #endif
610         case BIO_CTRL_DGRAM_QUERY_MTU:
611 #if defined(OPENSSL_SYS_LINUX) && defined(IP_MTU)
612                 addr_len = (socklen_t)sizeof(addr);
613                 memset((void *)&addr, 0, sizeof(addr));
614                 if (getsockname(b->num, &addr.sa, &addr_len) < 0)
615                         {
616                         ret = 0;
617                         break;
618                         }
619                 sockopt_len = sizeof(sockopt_val);
620                 switch (addr.sa.sa_family)
621                         {
622                 case AF_INET:
623                         if ((ret = getsockopt(b->num, IPPROTO_IP, IP_MTU, (void *)&sockopt_val,
624                                 &sockopt_len)) < 0 || sockopt_val < 0)
625                                 {
626                                 ret = 0;
627                                 }
628                         else
629                                 {
630                                 /* we assume that the transport protocol is UDP and no
631                                  * IP options are used.
632                                  */
633                                 data->mtu = sockopt_val - 8 - 20;
634                                 ret = data->mtu;
635                                 }
636                         break;
637 #if OPENSSL_USE_IPV6 && defined(IPV6_MTU)
638                 case AF_INET6:
639                         if ((ret = getsockopt(b->num, IPPROTO_IPV6, IPV6_MTU, (void *)&sockopt_val,
640                                 &sockopt_len)) < 0 || sockopt_val < 0)
641                                 {
642                                 ret = 0;
643                                 }
644                         else
645                                 {
646                                 /* we assume that the transport protocol is UDP and no
647                                  * IPV6 options are used.
648                                  */
649                                 data->mtu = sockopt_val - 8 - 40;
650                                 ret = data->mtu;
651                                 }
652                         break;
653 #endif
654                 default:
655                         ret = 0;
656                         break;
657                         }
658 #else
659                 ret = 0;
660 #endif
661                 break;
662         case BIO_CTRL_DGRAM_GET_FALLBACK_MTU:
663                 ret = -dgram_get_mtu_overhead(data);
664                 switch (data->peer.sa.sa_family)
665                         {
666                         case AF_INET:
667                                 ret += 576;
668                                 break;
669 #if OPENSSL_USE_IPV6
670                         case AF_INET6:
671 #ifdef IN6_IS_ADDR_V4MAPPED
672                                 if (IN6_IS_ADDR_V4MAPPED(&data->peer.sa_in6.sin6_addr))
673                                         ret += 576;
674                                 else
675 #endif
676                                         ret += 1280;
677                                 break;
678 #endif
679                         default:
680                                 ret += 576;
681                                 break;
682                         }
683                 break;
684         case BIO_CTRL_DGRAM_GET_MTU:
685                 return data->mtu;
686                 break;
687         case BIO_CTRL_DGRAM_SET_MTU:
688                 data->mtu = num;
689                 ret = num;
690                 break;
691         case BIO_CTRL_DGRAM_SET_CONNECTED:
692                 to = (struct sockaddr *)ptr;
693
694                 if ( to != NULL)
695                         {
696                         data->connected = 1;
697                         switch (to->sa_family)
698                                 {
699                                 case AF_INET:
700                                         memcpy(&data->peer,to,sizeof(data->peer.sa_in));
701                                         break;
702 #if OPENSSL_USE_IPV6
703                                 case AF_INET6:
704                                         memcpy(&data->peer,to,sizeof(data->peer.sa_in6));
705                                         break;
706 #endif
707                                 default:
708                                         memcpy(&data->peer,to,sizeof(data->peer.sa));
709                                         break;
710                                 }
711                         }
712                 else
713                         {
714                         data->connected = 0;
715                         memset(&(data->peer), 0x00, sizeof(data->peer));
716                         }
717                 break;
718         case BIO_CTRL_DGRAM_GET_PEER:
719                 switch (data->peer.sa.sa_family)
720                         {
721                         case AF_INET:
722                                 ret=sizeof(data->peer.sa_in);
723                                 break;
724 #if OPENSSL_USE_IPV6
725                         case AF_INET6:
726                                 ret=sizeof(data->peer.sa_in6);
727                                 break;
728 #endif
729                         default:
730                                 ret=sizeof(data->peer.sa);
731                                 break;
732                         }
733                 if (num==0 || num>ret)
734                         num=ret;
735                 memcpy(ptr,&data->peer,(ret=num));
736                 break;
737         case BIO_CTRL_DGRAM_SET_PEER:
738                 to = (struct sockaddr *) ptr;
739                 switch (to->sa_family)
740                         {
741                         case AF_INET:
742                                 memcpy(&data->peer,to,sizeof(data->peer.sa_in));
743                                 break;
744 #if OPENSSL_USE_IPV6
745                         case AF_INET6:
746                                 memcpy(&data->peer,to,sizeof(data->peer.sa_in6));
747                                 break;
748 #endif
749                         default:
750                                 memcpy(&data->peer,to,sizeof(data->peer.sa));
751                                 break;
752                         }
753                 break;
754         case BIO_CTRL_DGRAM_SET_NEXT_TIMEOUT:
755                 memcpy(&(data->next_timeout), ptr, sizeof(struct timeval));
756                 break;
757 #if defined(SO_RCVTIMEO)
758         case BIO_CTRL_DGRAM_SET_RECV_TIMEOUT:
759 #ifdef OPENSSL_SYS_WINDOWS
760                 {
761                 struct timeval *tv = (struct timeval *)ptr;
762                 int timeout = tv->tv_sec * 1000 + tv->tv_usec/1000;
763                 if (setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
764                         (void*)&timeout, sizeof(timeout)) < 0)
765                         { perror("setsockopt"); ret = -1; }
766                 }
767 #else
768                 if ( setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO, ptr,
769                         sizeof(struct timeval)) < 0)
770                         { perror("setsockopt"); ret = -1; }
771 #endif
772                 break;
773         case BIO_CTRL_DGRAM_GET_RECV_TIMEOUT:
774                 {
775                 union { size_t s; int i; } sz = {0};
776 #ifdef OPENSSL_SYS_WINDOWS
777                 int timeout;
778                 struct timeval *tv = (struct timeval *)ptr;
779
780                 sz.i = sizeof(timeout);
781                 if (getsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
782                         (void*)&timeout, &sz.i) < 0)
783                         { perror("getsockopt"); ret = -1; }
784                 else
785                         {
786                         tv->tv_sec = timeout / 1000;
787                         tv->tv_usec = (timeout % 1000) * 1000;
788                         ret = sizeof(*tv);
789                         }
790 #else
791                 sz.i = sizeof(struct timeval);
792                 if ( getsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO, 
793                         ptr, (void *)&sz) < 0)
794                         { perror("getsockopt"); ret = -1; }
795                 else if (sizeof(sz.s)!=sizeof(sz.i) && sz.i==0)
796                         {
797                         OPENSSL_assert(sz.s<=sizeof(struct timeval));
798                         ret = (int)sz.s;
799                         }
800                 else
801                         ret = sz.i;
802 #endif
803                 }
804                 break;
805 #endif
806 #if defined(SO_SNDTIMEO)
807         case BIO_CTRL_DGRAM_SET_SEND_TIMEOUT:
808 #ifdef OPENSSL_SYS_WINDOWS
809                 {
810                 struct timeval *tv = (struct timeval *)ptr;
811                 int timeout = tv->tv_sec * 1000 + tv->tv_usec/1000;
812                 if (setsockopt(b->num, SOL_SOCKET, SO_SNDTIMEO,
813                         (void*)&timeout, sizeof(timeout)) < 0)
814                         { perror("setsockopt"); ret = -1; }
815                 }
816 #else
817                 if ( setsockopt(b->num, SOL_SOCKET, SO_SNDTIMEO, ptr,
818                         sizeof(struct timeval)) < 0)
819                         { perror("setsockopt"); ret = -1; }
820 #endif
821                 break;
822         case BIO_CTRL_DGRAM_GET_SEND_TIMEOUT:
823                 {
824                 union { size_t s; int i; } sz = {0};
825 #ifdef OPENSSL_SYS_WINDOWS
826                 int timeout;
827                 struct timeval *tv = (struct timeval *)ptr;
828
829                 sz.i = sizeof(timeout);
830                 if (getsockopt(b->num, SOL_SOCKET, SO_SNDTIMEO,
831                         (void*)&timeout, &sz.i) < 0)
832                         { perror("getsockopt"); ret = -1; }
833                 else
834                         {
835                         tv->tv_sec = timeout / 1000;
836                         tv->tv_usec = (timeout % 1000) * 1000;
837                         ret = sizeof(*tv);
838                         }
839 #else
840                 sz.i = sizeof(struct timeval);
841                 if ( getsockopt(b->num, SOL_SOCKET, SO_SNDTIMEO, 
842                         ptr, (void *)&sz) < 0)
843                         { perror("getsockopt"); ret = -1; }
844                 else if (sizeof(sz.s)!=sizeof(sz.i) && sz.i==0)
845                         {
846                         OPENSSL_assert(sz.s<=sizeof(struct timeval));
847                         ret = (int)sz.s;
848                         }
849                 else
850                         ret = sz.i;
851 #endif
852                 }
853                 break;
854 #endif
855         case BIO_CTRL_DGRAM_GET_SEND_TIMER_EXP:
856                 /* fall-through */
857         case BIO_CTRL_DGRAM_GET_RECV_TIMER_EXP:
858 #ifdef OPENSSL_SYS_WINDOWS
859                 if ( data->_errno == WSAETIMEDOUT)
860 #else
861                 if ( data->_errno == EAGAIN)
862 #endif
863                         {
864                         ret = 1;
865                         data->_errno = 0;
866                         }
867                 else
868                         ret = 0;
869                 break;
870 #ifdef EMSGSIZE
871         case BIO_CTRL_DGRAM_MTU_EXCEEDED:
872                 if ( data->_errno == EMSGSIZE)
873                         {
874                         ret = 1;
875                         data->_errno = 0;
876                         }
877                 else
878                         ret = 0;
879                 break;
880 #endif
881         case BIO_CTRL_DGRAM_GET_MTU_OVERHEAD:
882                 ret = dgram_get_mtu_overhead(data);
883                 break;
884         default:
885                 ret=0;
886                 break;
887                 }
888         return(ret);
889         }
890
891 static int dgram_puts(BIO *bp, const char *str)
892         {
893         int n,ret;
894
895         n=strlen(str);
896         ret=dgram_write(bp,str,n);
897         return(ret);
898         }
899
900 #ifndef OPENSSL_NO_SCTP
901 BIO_METHOD *BIO_s_datagram_sctp(void)
902         {
903         return(&methods_dgramp_sctp);
904         }
905
906 BIO *BIO_new_dgram_sctp(int fd, int close_flag)
907         {
908         BIO *bio;
909         int ret, optval = 20000;
910         int auth_data = 0, auth_forward = 0;
911         unsigned char *p;
912         struct sctp_authchunk auth;
913         struct sctp_authchunks *authchunks;
914         socklen_t sockopt_len;
915 #ifdef SCTP_AUTHENTICATION_EVENT
916 #ifdef SCTP_EVENT
917         struct sctp_event event;
918 #else
919         struct sctp_event_subscribe event;
920 #endif
921 #endif
922
923         bio=BIO_new(BIO_s_datagram_sctp());
924         if (bio == NULL) return(NULL);
925         BIO_set_fd(bio,fd,close_flag);
926
927         /* Activate SCTP-AUTH for DATA and FORWARD-TSN chunks */
928         auth.sauth_chunk = OPENSSL_SCTP_DATA_CHUNK_TYPE;
929         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_AUTH_CHUNK, &auth, sizeof(struct sctp_authchunk));
930         if (ret < 0)
931                 {
932                 BIO_vfree(bio);
933                 return(NULL);
934                 }
935         auth.sauth_chunk = OPENSSL_SCTP_FORWARD_CUM_TSN_CHUNK_TYPE;
936         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_AUTH_CHUNK, &auth, sizeof(struct sctp_authchunk));
937         if (ret < 0)
938                 {
939                 BIO_vfree(bio);
940                 return(NULL);
941                 }
942
943         /* Test if activation was successful. When using accept(),
944          * SCTP-AUTH has to be activated for the listening socket
945          * already, otherwise the connected socket won't use it. */
946         sockopt_len = (socklen_t)(sizeof(sctp_assoc_t) + 256 * sizeof(uint8_t));
947         authchunks = OPENSSL_malloc(sockopt_len);
948         memset(authchunks, 0, sizeof(sockopt_len));
949         ret = getsockopt(fd, IPPROTO_SCTP, SCTP_LOCAL_AUTH_CHUNKS, authchunks, &sockopt_len);
950
951         if (ret < 0)
952                 {
953                 OPENSSL_free(authchunks);
954                 BIO_vfree(bio);
955                 return(NULL);
956                 }
957
958         for (p = (unsigned char*) authchunks->gauth_chunks;
959              p < (unsigned char*) authchunks + sockopt_len;
960              p += sizeof(uint8_t))
961                 {
962                 if (*p == OPENSSL_SCTP_DATA_CHUNK_TYPE) auth_data = 1;
963                 if (*p == OPENSSL_SCTP_FORWARD_CUM_TSN_CHUNK_TYPE) auth_forward = 1;
964                 }
965                 
966         OPENSSL_free(authchunks);
967
968         OPENSSL_assert(auth_data);
969         OPENSSL_assert(auth_forward);
970
971 #ifdef SCTP_AUTHENTICATION_EVENT
972 #ifdef SCTP_EVENT
973         memset(&event, 0, sizeof(struct sctp_event));
974         event.se_assoc_id = 0;
975         event.se_type = SCTP_AUTHENTICATION_EVENT;
976         event.se_on = 1;
977         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(struct sctp_event));
978         if (ret < 0)
979                 {
980                 BIO_vfree(bio);
981                 return(NULL);
982                 }
983 #else
984         sockopt_len = (socklen_t) sizeof(struct sctp_event_subscribe);
985         ret = getsockopt(fd, IPPROTO_SCTP, SCTP_EVENTS, &event, &sockopt_len);
986         if (ret < 0)
987                 {
988                 BIO_vfree(bio);
989                 return(NULL);
990                 }
991
992         event.sctp_authentication_event = 1;
993
994         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_EVENTS, &event, sizeof(struct sctp_event_subscribe));
995         if (ret < 0)
996                 {
997                 BIO_vfree(bio);
998                 return(NULL);
999                 }
1000 #endif
1001 #endif
1002
1003         /* Disable partial delivery by setting the min size
1004          * larger than the max record size of 2^14 + 2048 + 13
1005          */
1006         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_PARTIAL_DELIVERY_POINT, &optval, sizeof(optval));
1007         if (ret < 0)
1008                 {
1009                 BIO_vfree(bio);
1010                 return(NULL);
1011                 }
1012
1013         return(bio);
1014         }
1015
1016 int BIO_dgram_is_sctp(BIO *bio)
1017         {
1018         return (BIO_method_type(bio) == BIO_TYPE_DGRAM_SCTP);
1019         }
1020
1021 static int dgram_sctp_new(BIO *bi)
1022         {
1023         bio_dgram_sctp_data *data = NULL;
1024
1025         bi->init=0;
1026         bi->num=0;
1027         data = OPENSSL_malloc(sizeof(bio_dgram_sctp_data));
1028         if (data == NULL)
1029                 return 0;
1030         memset(data, 0x00, sizeof(bio_dgram_sctp_data));
1031 #ifdef SCTP_PR_SCTP_NONE
1032         data->prinfo.pr_policy = SCTP_PR_SCTP_NONE;
1033 #endif
1034     bi->ptr = data;
1035
1036         bi->flags=0;
1037         return(1);
1038         }
1039
1040 static int dgram_sctp_free(BIO *a)
1041         {
1042         bio_dgram_sctp_data *data;
1043
1044         if (a == NULL) return(0);
1045         if ( ! dgram_clear(a))
1046                 return 0;
1047
1048         data = (bio_dgram_sctp_data *)a->ptr;
1049         if(data != NULL) OPENSSL_free(data);
1050
1051         return(1);
1052         }
1053
1054 #ifdef SCTP_AUTHENTICATION_EVENT
1055 void dgram_sctp_handle_auth_free_key_event(BIO *b, union sctp_notification *snp)
1056         {
1057         int ret;
1058         struct sctp_authkey_event* authkeyevent = &snp->sn_auth_event;
1059
1060         if (authkeyevent->auth_indication == SCTP_AUTH_FREE_KEY)
1061                 {
1062                 struct sctp_authkeyid authkeyid;
1063
1064                 /* delete key */
1065                 authkeyid.scact_keynumber = authkeyevent->auth_keynumber;
1066                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_DELETE_KEY,
1067                       &authkeyid, sizeof(struct sctp_authkeyid));
1068                 }
1069         }
1070 #endif
1071
1072 static int dgram_sctp_read(BIO *b, char *out, int outl)
1073         {
1074         int ret = 0, n = 0, i, optval;
1075         socklen_t optlen;
1076         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *)b->ptr;
1077         union sctp_notification *snp;
1078         struct msghdr msg;
1079         struct iovec iov;
1080         struct cmsghdr *cmsg;
1081         char cmsgbuf[512];
1082
1083         if (out != NULL)
1084                 {
1085                 clear_socket_error();
1086
1087                 do
1088                         {
1089                         memset(&data->rcvinfo, 0x00, sizeof(struct bio_dgram_sctp_rcvinfo));
1090                         iov.iov_base = out;
1091                         iov.iov_len = outl;
1092                         msg.msg_name = NULL;
1093                         msg.msg_namelen = 0;
1094                         msg.msg_iov = &iov;
1095                         msg.msg_iovlen = 1;
1096                         msg.msg_control = cmsgbuf;
1097                         msg.msg_controllen = 512;
1098                         msg.msg_flags = 0;
1099                         n = recvmsg(b->num, &msg, 0);
1100
1101                         if (n <= 0)
1102                                 {
1103                                 if (n < 0)
1104                                         ret = n;
1105                                 break;
1106                                 }
1107
1108                         if (msg.msg_controllen > 0)
1109                                 {
1110                                 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg))
1111                                         {
1112                                         if (cmsg->cmsg_level != IPPROTO_SCTP)
1113                                                 continue;
1114 #ifdef SCTP_RCVINFO
1115                                         if (cmsg->cmsg_type == SCTP_RCVINFO)
1116                                                 {
1117                                                 struct sctp_rcvinfo *rcvinfo;
1118
1119                                                 rcvinfo = (struct sctp_rcvinfo *)CMSG_DATA(cmsg);
1120                                                 data->rcvinfo.rcv_sid = rcvinfo->rcv_sid;
1121                                                 data->rcvinfo.rcv_ssn = rcvinfo->rcv_ssn;
1122                                                 data->rcvinfo.rcv_flags = rcvinfo->rcv_flags;
1123                                                 data->rcvinfo.rcv_ppid = rcvinfo->rcv_ppid;
1124                                                 data->rcvinfo.rcv_tsn = rcvinfo->rcv_tsn;
1125                                                 data->rcvinfo.rcv_cumtsn = rcvinfo->rcv_cumtsn;
1126                                                 data->rcvinfo.rcv_context = rcvinfo->rcv_context;
1127                                                 }
1128 #endif
1129 #ifdef SCTP_SNDRCV
1130                                         if (cmsg->cmsg_type == SCTP_SNDRCV)
1131                                                 {
1132                                                 struct sctp_sndrcvinfo *sndrcvinfo;
1133
1134                                                 sndrcvinfo = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
1135                                                 data->rcvinfo.rcv_sid = sndrcvinfo->sinfo_stream;
1136                                                 data->rcvinfo.rcv_ssn = sndrcvinfo->sinfo_ssn;
1137                                                 data->rcvinfo.rcv_flags = sndrcvinfo->sinfo_flags;
1138                                                 data->rcvinfo.rcv_ppid = sndrcvinfo->sinfo_ppid;
1139                                                 data->rcvinfo.rcv_tsn = sndrcvinfo->sinfo_tsn;
1140                                                 data->rcvinfo.rcv_cumtsn = sndrcvinfo->sinfo_cumtsn;
1141                                                 data->rcvinfo.rcv_context = sndrcvinfo->sinfo_context;
1142                                                 }
1143 #endif
1144                                         }
1145                                 }
1146
1147                         if (msg.msg_flags & MSG_NOTIFICATION)
1148                                 {
1149                                 snp = (union sctp_notification*) out;
1150                                 if (snp->sn_header.sn_type == SCTP_SENDER_DRY_EVENT)
1151                                         {
1152 #ifdef SCTP_EVENT
1153                                         struct sctp_event event;
1154 #else
1155                                         struct sctp_event_subscribe event;
1156                                         socklen_t eventsize;
1157 #endif
1158                                         /* If a message has been delayed until the socket
1159                                          * is dry, it can be sent now.
1160                                          */
1161                                         if (data->saved_message.length > 0)
1162                                                 {
1163                                                 dgram_sctp_write(data->saved_message.bio, data->saved_message.data,
1164                                                                  data->saved_message.length);
1165                                                 OPENSSL_free(data->saved_message.data);
1166                                                 data->saved_message.length = 0;
1167                                                 }
1168
1169                                         /* disable sender dry event */
1170 #ifdef SCTP_EVENT
1171                                         memset(&event, 0, sizeof(struct sctp_event));
1172                                         event.se_assoc_id = 0;
1173                                         event.se_type = SCTP_SENDER_DRY_EVENT;
1174                                         event.se_on = 0;
1175                                         i = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(struct sctp_event));
1176                                         if (i < 0)
1177                                                 {
1178                                                 ret = i;
1179                                                 break;
1180                                                 }
1181 #else
1182                                         eventsize = sizeof(struct sctp_event_subscribe);
1183                                         i = getsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, &eventsize);
1184                                         if (i < 0)
1185                                                 {
1186                                                 ret = i;
1187                                                 break;
1188                                                 }
1189
1190                                         event.sctp_sender_dry_event = 0;
1191
1192                                         i = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, sizeof(struct sctp_event_subscribe));
1193                                         if (i < 0)
1194                                                 {
1195                                                 ret = i;
1196                                                 break;
1197                                                 }
1198 #endif
1199                                         }
1200
1201 #ifdef SCTP_AUTHENTICATION_EVENT
1202                                 if (snp->sn_header.sn_type == SCTP_AUTHENTICATION_EVENT)
1203                                         dgram_sctp_handle_auth_free_key_event(b, snp);
1204 #endif
1205
1206                                 if (data->handle_notifications != NULL)
1207                                         data->handle_notifications(b, data->notification_context, (void*) out);
1208
1209                                 memset(out, 0, outl);
1210                                 }
1211                         else
1212                                 ret += n;
1213                         }
1214                 while ((msg.msg_flags & MSG_NOTIFICATION) && (msg.msg_flags & MSG_EOR) && (ret < outl));
1215
1216                 if (ret > 0 && !(msg.msg_flags & MSG_EOR))
1217                         {
1218                         /* Partial message read, this should never happen! */
1219
1220                         /* The buffer was too small, this means the peer sent
1221                          * a message that was larger than allowed. */
1222                         if (ret == outl)
1223                                 return -1;
1224
1225                         /* Test if socket buffer can handle max record
1226                          * size (2^14 + 2048 + 13)
1227                          */
1228                         optlen = (socklen_t) sizeof(int);
1229                         ret = getsockopt(b->num, SOL_SOCKET, SO_RCVBUF, &optval, &optlen);
1230                         if (ret >= 0)
1231                                 OPENSSL_assert(optval >= 18445);
1232
1233                         /* Test if SCTP doesn't partially deliver below
1234                          * max record size (2^14 + 2048 + 13)
1235                          */
1236                         optlen = (socklen_t) sizeof(int);
1237                         ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_PARTIAL_DELIVERY_POINT,
1238                                          &optval, &optlen);
1239                         if (ret >= 0)
1240                                 OPENSSL_assert(optval >= 18445);
1241
1242                         /* Partially delivered notification??? Probably a bug.... */
1243                         OPENSSL_assert(!(msg.msg_flags & MSG_NOTIFICATION));
1244
1245                         /* Everything seems ok till now, so it's most likely
1246                          * a message dropped by PR-SCTP.
1247                          */
1248                         memset(out, 0, outl);
1249                         BIO_set_retry_read(b);
1250                         return -1;
1251                         }
1252
1253                 BIO_clear_retry_flags(b);
1254                 if (ret < 0)
1255                         {
1256                         if (BIO_dgram_should_retry(ret))
1257                                 {
1258                                 BIO_set_retry_read(b);
1259                                 data->_errno = get_last_socket_error();
1260                                 }
1261                         }
1262
1263                 /* Test if peer uses SCTP-AUTH before continuing */
1264                 if (!data->peer_auth_tested)
1265                         {
1266                         int ii, auth_data = 0, auth_forward = 0;
1267                         unsigned char *p;
1268                         struct sctp_authchunks *authchunks;
1269
1270                         optlen = (socklen_t)(sizeof(sctp_assoc_t) + 256 * sizeof(uint8_t));
1271                         authchunks = OPENSSL_malloc(optlen);
1272                         memset(authchunks, 0, sizeof(optlen));
1273                         ii = getsockopt(b->num, IPPROTO_SCTP, SCTP_PEER_AUTH_CHUNKS, authchunks, &optlen);
1274
1275                         if (ii >= 0)
1276                                 for (p = (unsigned char*) authchunks->gauth_chunks;
1277                                      p < (unsigned char*) authchunks + optlen;
1278                                      p += sizeof(uint8_t))
1279                                         {
1280                                         if (*p == OPENSSL_SCTP_DATA_CHUNK_TYPE) auth_data = 1;
1281                                         if (*p == OPENSSL_SCTP_FORWARD_CUM_TSN_CHUNK_TYPE) auth_forward = 1;
1282                                         }
1283
1284                         OPENSSL_free(authchunks);
1285
1286                         if (!auth_data || !auth_forward)
1287                                 {
1288                                 BIOerr(BIO_F_DGRAM_SCTP_READ,BIO_R_CONNECT_ERROR);
1289                                 return -1;
1290                                 }
1291
1292                         data->peer_auth_tested = 1;
1293                         }
1294                 }
1295         return(ret);
1296         }
1297
1298 static int dgram_sctp_write(BIO *b, const char *in, int inl)
1299         {
1300         int ret;
1301         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *)b->ptr;
1302         struct bio_dgram_sctp_sndinfo *sinfo = &(data->sndinfo);
1303         struct bio_dgram_sctp_prinfo *pinfo = &(data->prinfo);
1304         struct bio_dgram_sctp_sndinfo handshake_sinfo;
1305         struct iovec iov[1];
1306         struct msghdr msg;
1307         struct cmsghdr *cmsg;
1308 #if defined(SCTP_SNDINFO) && defined(SCTP_PRINFO)
1309         char cmsgbuf[CMSG_SPACE(sizeof(struct sctp_sndinfo)) + CMSG_SPACE(sizeof(struct sctp_prinfo))];
1310         struct sctp_sndinfo *sndinfo;
1311         struct sctp_prinfo *prinfo;
1312 #else
1313         char cmsgbuf[CMSG_SPACE(sizeof(struct sctp_sndrcvinfo))];
1314         struct sctp_sndrcvinfo *sndrcvinfo;
1315 #endif
1316
1317         clear_socket_error();
1318
1319         /* If we're send anything else than application data,
1320          * disable all user parameters and flags.
1321          */
1322         if (in[0] != 23) {
1323                 memset(&handshake_sinfo, 0x00, sizeof(struct bio_dgram_sctp_sndinfo));
1324 #ifdef SCTP_SACK_IMMEDIATELY
1325                 handshake_sinfo.snd_flags = SCTP_SACK_IMMEDIATELY;
1326 #endif
1327                 sinfo = &handshake_sinfo;
1328         }
1329
1330         /* If we have to send a shutdown alert message and the
1331          * socket is not dry yet, we have to save it and send it
1332          * as soon as the socket gets dry.
1333          */
1334         if (data->save_shutdown && !BIO_dgram_sctp_wait_for_dry(b))
1335         {
1336                 data->saved_message.bio = b;
1337                 data->saved_message.length = inl;
1338                 data->saved_message.data = OPENSSL_malloc(inl);
1339                 memcpy(data->saved_message.data, in, inl);
1340                 return inl;
1341         }
1342
1343         iov[0].iov_base = (char *)in;
1344         iov[0].iov_len = inl;
1345         msg.msg_name = NULL;
1346         msg.msg_namelen = 0;
1347         msg.msg_iov = iov;
1348         msg.msg_iovlen = 1;
1349         msg.msg_control = (caddr_t)cmsgbuf;
1350         msg.msg_controllen = 0;
1351         msg.msg_flags = 0;
1352 #if defined(SCTP_SNDINFO) && defined(SCTP_PRINFO)
1353         cmsg = (struct cmsghdr *)cmsgbuf;
1354         cmsg->cmsg_level = IPPROTO_SCTP;
1355         cmsg->cmsg_type = SCTP_SNDINFO;
1356         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndinfo));
1357         sndinfo = (struct sctp_sndinfo *)CMSG_DATA(cmsg);
1358         memset(sndinfo, 0, sizeof(struct sctp_sndinfo));
1359         sndinfo->snd_sid = sinfo->snd_sid;
1360         sndinfo->snd_flags = sinfo->snd_flags;
1361         sndinfo->snd_ppid = sinfo->snd_ppid;
1362         sndinfo->snd_context = sinfo->snd_context;
1363         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_sndinfo));
1364
1365         cmsg = (struct cmsghdr *)&cmsgbuf[CMSG_SPACE(sizeof(struct sctp_sndinfo))];
1366         cmsg->cmsg_level = IPPROTO_SCTP;
1367         cmsg->cmsg_type = SCTP_PRINFO;
1368         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_prinfo));
1369         prinfo = (struct sctp_prinfo *)CMSG_DATA(cmsg);
1370         memset(prinfo, 0, sizeof(struct sctp_prinfo));
1371         prinfo->pr_policy = pinfo->pr_policy;
1372         prinfo->pr_value = pinfo->pr_value;
1373         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_prinfo));
1374 #else
1375         cmsg = (struct cmsghdr *)cmsgbuf;
1376         cmsg->cmsg_level = IPPROTO_SCTP;
1377         cmsg->cmsg_type = SCTP_SNDRCV;
1378         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndrcvinfo));
1379         sndrcvinfo = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
1380         memset(sndrcvinfo, 0, sizeof(struct sctp_sndrcvinfo));
1381         sndrcvinfo->sinfo_stream = sinfo->snd_sid;
1382         sndrcvinfo->sinfo_flags = sinfo->snd_flags;
1383 #ifdef __FreeBSD__
1384         sndrcvinfo->sinfo_flags |= pinfo->pr_policy;
1385 #endif
1386         sndrcvinfo->sinfo_ppid = sinfo->snd_ppid;
1387         sndrcvinfo->sinfo_context = sinfo->snd_context;
1388         sndrcvinfo->sinfo_timetolive = pinfo->pr_value;
1389         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_sndrcvinfo));
1390 #endif
1391
1392         ret = sendmsg(b->num, &msg, 0);
1393
1394         BIO_clear_retry_flags(b);
1395         if (ret <= 0)
1396                 {
1397                 if (BIO_dgram_should_retry(ret))
1398                         {
1399                         BIO_set_retry_write(b);  
1400                         data->_errno = get_last_socket_error();
1401                         }
1402                 }
1403         return(ret);
1404         }
1405
1406 static long dgram_sctp_ctrl(BIO *b, int cmd, long num, void *ptr)
1407         {
1408         long ret=1;
1409         bio_dgram_sctp_data *data = NULL;
1410         socklen_t sockopt_len = 0;
1411         struct sctp_authkeyid authkeyid;
1412         struct sctp_authkey *authkey = NULL;
1413
1414         data = (bio_dgram_sctp_data *)b->ptr;
1415
1416         switch (cmd)
1417                 {
1418         case BIO_CTRL_DGRAM_QUERY_MTU:
1419                 /* Set to maximum (2^14)
1420                  * and ignore user input to enable transport
1421                  * protocol fragmentation.
1422                  * Returns always 2^14.
1423                  */
1424                 data->mtu = 16384;
1425                 ret = data->mtu;
1426                 break;
1427         case BIO_CTRL_DGRAM_SET_MTU:
1428                 /* Set to maximum (2^14)
1429                  * and ignore input to enable transport
1430                  * protocol fragmentation.
1431                  * Returns always 2^14.
1432                  */
1433                 data->mtu = 16384;
1434                 ret = data->mtu;
1435                 break;
1436         case BIO_CTRL_DGRAM_SET_CONNECTED:
1437         case BIO_CTRL_DGRAM_CONNECT:
1438                 /* Returns always -1. */
1439                 ret = -1;
1440                 break;
1441         case BIO_CTRL_DGRAM_SET_NEXT_TIMEOUT:
1442                 /* SCTP doesn't need the DTLS timer
1443                  * Returns always 1.
1444                  */
1445                 break;
1446         case BIO_CTRL_DGRAM_GET_MTU_OVERHEAD:
1447                 /* We allow transport protocol fragmentation so this is irrelevant */
1448                 ret = 0;
1449                 break;
1450         case BIO_CTRL_DGRAM_SCTP_SET_IN_HANDSHAKE:
1451                 if (num > 0)
1452                         data->in_handshake = 1;
1453                 else
1454                         data->in_handshake = 0;
1455
1456                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_NODELAY, &data->in_handshake, sizeof(int));
1457                 break;
1458         case BIO_CTRL_DGRAM_SCTP_ADD_AUTH_KEY:
1459                 /* New shared key for SCTP AUTH.
1460                  * Returns 0 on success, -1 otherwise.
1461                  */
1462
1463                 /* Get active key */
1464                 sockopt_len = sizeof(struct sctp_authkeyid);
1465                 ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY, &authkeyid, &sockopt_len);
1466                 if (ret < 0) break;
1467
1468                 /* Add new key */
1469                 sockopt_len = sizeof(struct sctp_authkey) + 64 * sizeof(uint8_t);
1470                 authkey = OPENSSL_malloc(sockopt_len);
1471                 if (authkey == NULL)
1472                         {
1473                         ret = -1;
1474                         break;
1475                         }
1476                 memset(authkey, 0x00, sockopt_len);
1477                 authkey->sca_keynumber = authkeyid.scact_keynumber + 1;
1478 #ifndef __FreeBSD__
1479                 /* This field is missing in FreeBSD 8.2 and earlier,
1480                  * and FreeBSD 8.3 and higher work without it.
1481                  */
1482                 authkey->sca_keylength = 64;
1483 #endif
1484                 memcpy(&authkey->sca_key[0], ptr, 64 * sizeof(uint8_t));
1485
1486                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_KEY, authkey, sockopt_len);
1487                 OPENSSL_free(authkey);
1488                 authkey = NULL;
1489                 if (ret < 0) break;
1490
1491                 /* Reset active key */
1492                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY,
1493                       &authkeyid, sizeof(struct sctp_authkeyid));
1494                 if (ret < 0) break;
1495
1496                 break;
1497         case BIO_CTRL_DGRAM_SCTP_NEXT_AUTH_KEY:
1498                 /* Returns 0 on success, -1 otherwise. */
1499
1500                 /* Get active key */
1501                 sockopt_len = sizeof(struct sctp_authkeyid);
1502                 ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY, &authkeyid, &sockopt_len);
1503                 if (ret < 0) break;
1504
1505                 /* Set active key */
1506                 authkeyid.scact_keynumber = authkeyid.scact_keynumber + 1;
1507                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY,
1508                       &authkeyid, sizeof(struct sctp_authkeyid));
1509                 if (ret < 0) break;
1510
1511                 /* CCS has been sent, so remember that and fall through
1512                  * to check if we need to deactivate an old key
1513                  */
1514                 data->ccs_sent = 1;
1515
1516         case BIO_CTRL_DGRAM_SCTP_AUTH_CCS_RCVD:
1517                 /* Returns 0 on success, -1 otherwise. */
1518
1519                 /* Has this command really been called or is this just a fall-through? */
1520                 if (cmd == BIO_CTRL_DGRAM_SCTP_AUTH_CCS_RCVD)
1521                         data->ccs_rcvd = 1;
1522
1523                 /* CSS has been both, received and sent, so deactivate an old key */
1524                 if (data->ccs_rcvd == 1 && data->ccs_sent == 1)
1525                         {
1526                         /* Get active key */
1527                         sockopt_len = sizeof(struct sctp_authkeyid);
1528                         ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY, &authkeyid, &sockopt_len);
1529                         if (ret < 0) break;
1530
1531                         /* Deactivate key or delete second last key if
1532                          * SCTP_AUTHENTICATION_EVENT is not available.
1533                          */
1534                         authkeyid.scact_keynumber = authkeyid.scact_keynumber - 1;
1535 #ifdef SCTP_AUTH_DEACTIVATE_KEY
1536                         sockopt_len = sizeof(struct sctp_authkeyid);
1537                         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_DEACTIVATE_KEY,
1538                               &authkeyid, sockopt_len);
1539                         if (ret < 0) break;
1540 #endif
1541 #ifndef SCTP_AUTHENTICATION_EVENT
1542                         if (authkeyid.scact_keynumber > 0)
1543                                 {
1544                                 authkeyid.scact_keynumber = authkeyid.scact_keynumber - 1;
1545                                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_DELETE_KEY,
1546                                           &authkeyid, sizeof(struct sctp_authkeyid));
1547                                 if (ret < 0) break;
1548                                 }
1549 #endif
1550
1551                         data->ccs_rcvd = 0;
1552                         data->ccs_sent = 0;
1553                         }
1554                 break;
1555         case BIO_CTRL_DGRAM_SCTP_GET_SNDINFO:
1556                 /* Returns the size of the copied struct. */
1557                 if (num > (long) sizeof(struct bio_dgram_sctp_sndinfo))
1558                         num = sizeof(struct bio_dgram_sctp_sndinfo);
1559
1560                 memcpy(ptr, &(data->sndinfo), num);
1561                 ret = num;
1562                 break;
1563         case BIO_CTRL_DGRAM_SCTP_SET_SNDINFO:
1564                 /* Returns the size of the copied struct. */
1565                 if (num > (long) sizeof(struct bio_dgram_sctp_sndinfo))
1566                         num = sizeof(struct bio_dgram_sctp_sndinfo);
1567
1568                 memcpy(&(data->sndinfo), ptr, num);
1569                 break;
1570         case BIO_CTRL_DGRAM_SCTP_GET_RCVINFO:
1571                 /* Returns the size of the copied struct. */
1572                 if (num > (long) sizeof(struct bio_dgram_sctp_rcvinfo))
1573                         num = sizeof(struct bio_dgram_sctp_rcvinfo);
1574
1575                 memcpy(ptr, &data->rcvinfo, num);
1576
1577                 ret = num;
1578                 break;
1579         case BIO_CTRL_DGRAM_SCTP_SET_RCVINFO:
1580                 /* Returns the size of the copied struct. */
1581                 if (num > (long) sizeof(struct bio_dgram_sctp_rcvinfo))
1582                         num = sizeof(struct bio_dgram_sctp_rcvinfo);
1583
1584                 memcpy(&(data->rcvinfo), ptr, num);
1585                 break;
1586         case BIO_CTRL_DGRAM_SCTP_GET_PRINFO:
1587                 /* Returns the size of the copied struct. */
1588                 if (num > (long) sizeof(struct bio_dgram_sctp_prinfo))
1589                         num = sizeof(struct bio_dgram_sctp_prinfo);
1590
1591                 memcpy(ptr, &(data->prinfo), num);
1592                 ret = num;
1593                 break;
1594         case BIO_CTRL_DGRAM_SCTP_SET_PRINFO:
1595                 /* Returns the size of the copied struct. */
1596                 if (num > (long) sizeof(struct bio_dgram_sctp_prinfo))
1597                         num = sizeof(struct bio_dgram_sctp_prinfo);
1598
1599                 memcpy(&(data->prinfo), ptr, num);
1600                 break;
1601         case BIO_CTRL_DGRAM_SCTP_SAVE_SHUTDOWN:
1602                 /* Returns always 1. */
1603                 if (num > 0)
1604                         data->save_shutdown = 1;
1605                 else
1606                         data->save_shutdown = 0;
1607                 break;
1608
1609         default:
1610                 /* Pass to default ctrl function to
1611                  * process SCTP unspecific commands
1612                  */
1613                 ret=dgram_ctrl(b, cmd, num, ptr);
1614                 break;
1615                 }
1616         return(ret);
1617         }
1618
1619 int BIO_dgram_sctp_notification_cb(BIO *b,
1620                                    void (*handle_notifications)(BIO *bio, void *context, void *buf),
1621                                    void *context)
1622         {
1623         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *) b->ptr;
1624
1625         if (handle_notifications != NULL)
1626                 {
1627                 data->handle_notifications = handle_notifications;
1628                 data->notification_context = context;
1629                 }
1630         else
1631                 return -1;
1632
1633         return 0;
1634         }
1635
1636 int BIO_dgram_sctp_wait_for_dry(BIO *b)
1637 {
1638         int is_dry = 0;
1639         int n, sockflags, ret;
1640         union sctp_notification snp;
1641         struct msghdr msg;
1642         struct iovec iov;
1643 #ifdef SCTP_EVENT
1644         struct sctp_event event;
1645 #else
1646         struct sctp_event_subscribe event;
1647         socklen_t eventsize;
1648 #endif
1649         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *)b->ptr;
1650
1651         /* set sender dry event */
1652 #ifdef SCTP_EVENT
1653         memset(&event, 0, sizeof(struct sctp_event));
1654         event.se_assoc_id = 0;
1655         event.se_type = SCTP_SENDER_DRY_EVENT;
1656         event.se_on = 1;
1657         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(struct sctp_event));
1658 #else
1659         eventsize = sizeof(struct sctp_event_subscribe);
1660         ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, &eventsize);
1661         if (ret < 0)
1662                 return -1;
1663         
1664         event.sctp_sender_dry_event = 1;
1665         
1666         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, sizeof(struct sctp_event_subscribe));
1667 #endif
1668         if (ret < 0)
1669                 return -1;
1670
1671         /* peek for notification */
1672         memset(&snp, 0x00, sizeof(union sctp_notification));
1673         iov.iov_base = (char *)&snp;
1674         iov.iov_len = sizeof(union sctp_notification);
1675         msg.msg_name = NULL;
1676         msg.msg_namelen = 0;
1677         msg.msg_iov = &iov;
1678         msg.msg_iovlen = 1;
1679         msg.msg_control = NULL;
1680         msg.msg_controllen = 0;
1681         msg.msg_flags = 0;
1682
1683         n = recvmsg(b->num, &msg, MSG_PEEK);
1684         if (n <= 0)
1685                 {
1686                 if ((n < 0) && (get_last_socket_error() != EAGAIN) && (get_last_socket_error() != EWOULDBLOCK))
1687                         return -1;
1688                 else
1689                         return 0;
1690                 }
1691
1692         /* if we find a notification, process it and try again if necessary */
1693         while (msg.msg_flags & MSG_NOTIFICATION)
1694                 {
1695                 memset(&snp, 0x00, sizeof(union sctp_notification));
1696                 iov.iov_base = (char *)&snp;
1697                 iov.iov_len = sizeof(union sctp_notification);
1698                 msg.msg_name = NULL;
1699                 msg.msg_namelen = 0;
1700                 msg.msg_iov = &iov;
1701                 msg.msg_iovlen = 1;
1702                 msg.msg_control = NULL;
1703                 msg.msg_controllen = 0;
1704                 msg.msg_flags = 0;
1705
1706                 n = recvmsg(b->num, &msg, 0);
1707                 if (n <= 0)
1708                         {
1709                         if ((n < 0) && (get_last_socket_error() != EAGAIN) && (get_last_socket_error() != EWOULDBLOCK))
1710                                 return -1;
1711                         else
1712                                 return is_dry;
1713                         }
1714                 
1715                 if (snp.sn_header.sn_type == SCTP_SENDER_DRY_EVENT)
1716                         {
1717                         is_dry = 1;
1718
1719                         /* disable sender dry event */
1720 #ifdef SCTP_EVENT
1721                         memset(&event, 0, sizeof(struct sctp_event));
1722                         event.se_assoc_id = 0;
1723                         event.se_type = SCTP_SENDER_DRY_EVENT;
1724                         event.se_on = 0;
1725                         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(struct sctp_event));
1726 #else
1727                         eventsize = (socklen_t) sizeof(struct sctp_event_subscribe);
1728                         ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, &eventsize);
1729                         if (ret < 0)
1730                                 return -1;
1731
1732                         event.sctp_sender_dry_event = 0;
1733
1734                         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, sizeof(struct sctp_event_subscribe));
1735 #endif
1736                         if (ret < 0)
1737                                 return -1;
1738                         }
1739
1740 #ifdef SCTP_AUTHENTICATION_EVENT
1741                 if (snp.sn_header.sn_type == SCTP_AUTHENTICATION_EVENT)
1742                         dgram_sctp_handle_auth_free_key_event(b, &snp);
1743 #endif
1744
1745                 if (data->handle_notifications != NULL)
1746                         data->handle_notifications(b, data->notification_context, (void*) &snp);
1747
1748                 /* found notification, peek again */
1749                 memset(&snp, 0x00, sizeof(union sctp_notification));
1750                 iov.iov_base = (char *)&snp;
1751                 iov.iov_len = sizeof(union sctp_notification);
1752                 msg.msg_name = NULL;
1753                 msg.msg_namelen = 0;
1754                 msg.msg_iov = &iov;
1755                 msg.msg_iovlen = 1;
1756                 msg.msg_control = NULL;
1757                 msg.msg_controllen = 0;
1758                 msg.msg_flags = 0;
1759
1760                 /* if we have seen the dry already, don't wait */
1761                 if (is_dry)
1762                         {
1763                         sockflags = fcntl(b->num, F_GETFL, 0);
1764                         fcntl(b->num, F_SETFL, O_NONBLOCK);
1765                         }
1766
1767                 n = recvmsg(b->num, &msg, MSG_PEEK);
1768
1769                 if (is_dry)
1770                         {
1771                         fcntl(b->num, F_SETFL, sockflags);
1772                         }
1773
1774                 if (n <= 0)
1775                         {
1776                         if ((n < 0) && (get_last_socket_error() != EAGAIN) && (get_last_socket_error() != EWOULDBLOCK))
1777                                 return -1;
1778                         else
1779                                 return is_dry;
1780                         }
1781                 }
1782
1783         /* read anything else */
1784         return is_dry;
1785 }
1786
1787 int BIO_dgram_sctp_msg_waiting(BIO *b)
1788         {
1789         int n, sockflags;
1790         union sctp_notification snp;
1791         struct msghdr msg;
1792         struct iovec iov;
1793         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *)b->ptr;
1794
1795         /* Check if there are any messages waiting to be read */
1796         do
1797                 {
1798                 memset(&snp, 0x00, sizeof(union sctp_notification));
1799                 iov.iov_base = (char *)&snp;
1800                 iov.iov_len = sizeof(union sctp_notification);
1801                 msg.msg_name = NULL;
1802                 msg.msg_namelen = 0;
1803                 msg.msg_iov = &iov;
1804                 msg.msg_iovlen = 1;
1805                 msg.msg_control = NULL;
1806                 msg.msg_controllen = 0;
1807                 msg.msg_flags = 0;
1808
1809                 sockflags = fcntl(b->num, F_GETFL, 0);
1810                 fcntl(b->num, F_SETFL, O_NONBLOCK);
1811                 n = recvmsg(b->num, &msg, MSG_PEEK);
1812                 fcntl(b->num, F_SETFL, sockflags);
1813
1814                 /* if notification, process and try again */
1815                 if (n > 0 && (msg.msg_flags & MSG_NOTIFICATION))
1816                         {
1817 #ifdef SCTP_AUTHENTICATION_EVENT
1818                         if (snp.sn_header.sn_type == SCTP_AUTHENTICATION_EVENT)
1819                                 dgram_sctp_handle_auth_free_key_event(b, &snp);
1820 #endif
1821
1822                         memset(&snp, 0x00, sizeof(union sctp_notification));
1823                         iov.iov_base = (char *)&snp;
1824                         iov.iov_len = sizeof(union sctp_notification);
1825                         msg.msg_name = NULL;
1826                         msg.msg_namelen = 0;
1827                         msg.msg_iov = &iov;
1828                         msg.msg_iovlen = 1;
1829                         msg.msg_control = NULL;
1830                         msg.msg_controllen = 0;
1831                         msg.msg_flags = 0;
1832                         n = recvmsg(b->num, &msg, 0);
1833
1834                         if (data->handle_notifications != NULL)
1835                                 data->handle_notifications(b, data->notification_context, (void*) &snp);
1836                         }
1837
1838                 } while (n > 0 && (msg.msg_flags & MSG_NOTIFICATION));
1839
1840         /* Return 1 if there is a message to be read, return 0 otherwise. */
1841         if (n > 0)
1842                 return 1;
1843         else
1844                 return 0;
1845         }
1846
1847 static int dgram_sctp_puts(BIO *bp, const char *str)
1848         {
1849         int n,ret;
1850
1851         n=strlen(str);
1852         ret=dgram_sctp_write(bp,str,n);
1853         return(ret);
1854         }
1855 #endif
1856
1857 static int BIO_dgram_should_retry(int i)
1858         {
1859         int err;
1860
1861         if ((i == 0) || (i == -1))
1862                 {
1863                 err=get_last_socket_error();
1864
1865 #if defined(OPENSSL_SYS_WINDOWS)
1866         /* If the socket return value (i) is -1
1867          * and err is unexpectedly 0 at this point,
1868          * the error code was overwritten by
1869          * another system call before this error
1870          * handling is called.
1871          */
1872 #endif
1873
1874                 return(BIO_dgram_non_fatal_error(err));
1875                 }
1876         return(0);
1877         }
1878
1879 int BIO_dgram_non_fatal_error(int err)
1880         {
1881         switch (err)
1882                 {
1883 #if defined(OPENSSL_SYS_WINDOWS)
1884 # if defined(WSAEWOULDBLOCK)
1885         case WSAEWOULDBLOCK:
1886 # endif
1887
1888 # if 0 /* This appears to always be an error */
1889 #  if defined(WSAENOTCONN)
1890         case WSAENOTCONN:
1891 #  endif
1892 # endif
1893 #endif
1894
1895 #ifdef EWOULDBLOCK
1896 # ifdef WSAEWOULDBLOCK
1897 #  if WSAEWOULDBLOCK != EWOULDBLOCK
1898         case EWOULDBLOCK:
1899 #  endif
1900 # else
1901         case EWOULDBLOCK:
1902 # endif
1903 #endif
1904
1905 #ifdef EINTR
1906         case EINTR:
1907 #endif
1908
1909 #ifdef EAGAIN
1910 #if EWOULDBLOCK != EAGAIN
1911         case EAGAIN:
1912 # endif
1913 #endif
1914
1915 #ifdef EPROTO
1916         case EPROTO:
1917 #endif
1918
1919 #ifdef EINPROGRESS
1920         case EINPROGRESS:
1921 #endif
1922
1923 #ifdef EALREADY
1924         case EALREADY:
1925 #endif
1926
1927                 return(1);
1928                 /* break; */
1929         default:
1930                 break;
1931                 }
1932         return(0);
1933         }
1934
1935 static void get_current_time(struct timeval *t)
1936         {
1937 #ifdef OPENSSL_SYS_WIN32
1938         struct _timeb tb;
1939         _ftime(&tb);
1940         t->tv_sec = (long)tb.time;
1941         t->tv_usec = (long)tb.millitm * 1000;
1942 #elif defined(OPENSSL_SYS_VMS)
1943         struct timeb tb;
1944         ftime(&tb);
1945         t->tv_sec = (long)tb.time;
1946         t->tv_usec = (long)tb.millitm * 1000;
1947 #else
1948         gettimeofday(t, NULL);
1949 #endif
1950         }
1951
1952 #endif