There are a number of instances throughout the code where the constant 28 is
[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         OPENSSL_assert(ret >= 0);
931         auth.sauth_chunk = OPENSSL_SCTP_FORWARD_CUM_TSN_CHUNK_TYPE;
932         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_AUTH_CHUNK, &auth, sizeof(struct sctp_authchunk));
933         OPENSSL_assert(ret >= 0);
934
935         /* Test if activation was successful. When using accept(),
936          * SCTP-AUTH has to be activated for the listening socket
937          * already, otherwise the connected socket won't use it. */
938         sockopt_len = (socklen_t)(sizeof(sctp_assoc_t) + 256 * sizeof(uint8_t));
939         authchunks = OPENSSL_malloc(sockopt_len);
940         memset(authchunks, 0, sizeof(sockopt_len));
941         ret = getsockopt(fd, IPPROTO_SCTP, SCTP_LOCAL_AUTH_CHUNKS, authchunks, &sockopt_len);
942         OPENSSL_assert(ret >= 0);
943
944         for (p = (unsigned char*) authchunks->gauth_chunks;
945              p < (unsigned char*) authchunks + sockopt_len;
946              p += sizeof(uint8_t))
947                 {
948                 if (*p == OPENSSL_SCTP_DATA_CHUNK_TYPE) auth_data = 1;
949                 if (*p == OPENSSL_SCTP_FORWARD_CUM_TSN_CHUNK_TYPE) auth_forward = 1;
950                 }
951                 
952         OPENSSL_free(authchunks);
953
954         OPENSSL_assert(auth_data);
955         OPENSSL_assert(auth_forward);
956
957 #ifdef SCTP_AUTHENTICATION_EVENT
958 #ifdef SCTP_EVENT
959         memset(&event, 0, sizeof(struct sctp_event));
960         event.se_assoc_id = 0;
961         event.se_type = SCTP_AUTHENTICATION_EVENT;
962         event.se_on = 1;
963         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(struct sctp_event));
964         OPENSSL_assert(ret >= 0);
965 #else
966         sockopt_len = (socklen_t) sizeof(struct sctp_event_subscribe);
967         ret = getsockopt(fd, IPPROTO_SCTP, SCTP_EVENTS, &event, &sockopt_len);
968         OPENSSL_assert(ret >= 0);
969
970         event.sctp_authentication_event = 1;
971
972         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_EVENTS, &event, sizeof(struct sctp_event_subscribe));
973         OPENSSL_assert(ret >= 0);
974 #endif
975 #endif
976
977         /* Disable partial delivery by setting the min size
978          * larger than the max record size of 2^14 + 2048 + 13
979          */
980         ret = setsockopt(fd, IPPROTO_SCTP, SCTP_PARTIAL_DELIVERY_POINT, &optval, sizeof(optval));
981         OPENSSL_assert(ret >= 0);
982
983         return(bio);
984         }
985
986 int BIO_dgram_is_sctp(BIO *bio)
987         {
988         return (BIO_method_type(bio) == BIO_TYPE_DGRAM_SCTP);
989         }
990
991 static int dgram_sctp_new(BIO *bi)
992         {
993         bio_dgram_sctp_data *data = NULL;
994
995         bi->init=0;
996         bi->num=0;
997         data = OPENSSL_malloc(sizeof(bio_dgram_sctp_data));
998         if (data == NULL)
999                 return 0;
1000         memset(data, 0x00, sizeof(bio_dgram_sctp_data));
1001 #ifdef SCTP_PR_SCTP_NONE
1002         data->prinfo.pr_policy = SCTP_PR_SCTP_NONE;
1003 #endif
1004     bi->ptr = data;
1005
1006         bi->flags=0;
1007         return(1);
1008         }
1009
1010 static int dgram_sctp_free(BIO *a)
1011         {
1012         bio_dgram_sctp_data *data;
1013
1014         if (a == NULL) return(0);
1015         if ( ! dgram_clear(a))
1016                 return 0;
1017
1018         data = (bio_dgram_sctp_data *)a->ptr;
1019         if(data != NULL) OPENSSL_free(data);
1020
1021         return(1);
1022         }
1023
1024 #ifdef SCTP_AUTHENTICATION_EVENT
1025 void dgram_sctp_handle_auth_free_key_event(BIO *b, union sctp_notification *snp)
1026         {
1027         int ret;
1028         struct sctp_authkey_event* authkeyevent = &snp->sn_auth_event;
1029
1030         if (authkeyevent->auth_indication == SCTP_AUTH_FREE_KEY)
1031                 {
1032                 struct sctp_authkeyid authkeyid;
1033
1034                 /* delete key */
1035                 authkeyid.scact_keynumber = authkeyevent->auth_keynumber;
1036                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_DELETE_KEY,
1037                       &authkeyid, sizeof(struct sctp_authkeyid));
1038                 }
1039         }
1040 #endif
1041
1042 static int dgram_sctp_read(BIO *b, char *out, int outl)
1043         {
1044         int ret = 0, n = 0, i, optval;
1045         socklen_t optlen;
1046         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *)b->ptr;
1047         union sctp_notification *snp;
1048         struct msghdr msg;
1049         struct iovec iov;
1050         struct cmsghdr *cmsg;
1051         char cmsgbuf[512];
1052
1053         if (out != NULL)
1054                 {
1055                 clear_socket_error();
1056
1057                 do
1058                         {
1059                         memset(&data->rcvinfo, 0x00, sizeof(struct bio_dgram_sctp_rcvinfo));
1060                         iov.iov_base = out;
1061                         iov.iov_len = outl;
1062                         msg.msg_name = NULL;
1063                         msg.msg_namelen = 0;
1064                         msg.msg_iov = &iov;
1065                         msg.msg_iovlen = 1;
1066                         msg.msg_control = cmsgbuf;
1067                         msg.msg_controllen = 512;
1068                         msg.msg_flags = 0;
1069                         n = recvmsg(b->num, &msg, 0);
1070
1071                         if (n <= 0)
1072                                 {
1073                                 if (n < 0)
1074                                         ret = n;
1075                                 break;
1076                                 }
1077
1078                         if (msg.msg_controllen > 0)
1079                                 {
1080                                 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg))
1081                                         {
1082                                         if (cmsg->cmsg_level != IPPROTO_SCTP)
1083                                                 continue;
1084 #ifdef SCTP_RCVINFO
1085                                         if (cmsg->cmsg_type == SCTP_RCVINFO)
1086                                                 {
1087                                                 struct sctp_rcvinfo *rcvinfo;
1088
1089                                                 rcvinfo = (struct sctp_rcvinfo *)CMSG_DATA(cmsg);
1090                                                 data->rcvinfo.rcv_sid = rcvinfo->rcv_sid;
1091                                                 data->rcvinfo.rcv_ssn = rcvinfo->rcv_ssn;
1092                                                 data->rcvinfo.rcv_flags = rcvinfo->rcv_flags;
1093                                                 data->rcvinfo.rcv_ppid = rcvinfo->rcv_ppid;
1094                                                 data->rcvinfo.rcv_tsn = rcvinfo->rcv_tsn;
1095                                                 data->rcvinfo.rcv_cumtsn = rcvinfo->rcv_cumtsn;
1096                                                 data->rcvinfo.rcv_context = rcvinfo->rcv_context;
1097                                                 }
1098 #endif
1099 #ifdef SCTP_SNDRCV
1100                                         if (cmsg->cmsg_type == SCTP_SNDRCV)
1101                                                 {
1102                                                 struct sctp_sndrcvinfo *sndrcvinfo;
1103
1104                                                 sndrcvinfo = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
1105                                                 data->rcvinfo.rcv_sid = sndrcvinfo->sinfo_stream;
1106                                                 data->rcvinfo.rcv_ssn = sndrcvinfo->sinfo_ssn;
1107                                                 data->rcvinfo.rcv_flags = sndrcvinfo->sinfo_flags;
1108                                                 data->rcvinfo.rcv_ppid = sndrcvinfo->sinfo_ppid;
1109                                                 data->rcvinfo.rcv_tsn = sndrcvinfo->sinfo_tsn;
1110                                                 data->rcvinfo.rcv_cumtsn = sndrcvinfo->sinfo_cumtsn;
1111                                                 data->rcvinfo.rcv_context = sndrcvinfo->sinfo_context;
1112                                                 }
1113 #endif
1114                                         }
1115                                 }
1116
1117                         if (msg.msg_flags & MSG_NOTIFICATION)
1118                                 {
1119                                 snp = (union sctp_notification*) out;
1120                                 if (snp->sn_header.sn_type == SCTP_SENDER_DRY_EVENT)
1121                                         {
1122 #ifdef SCTP_EVENT
1123                                         struct sctp_event event;
1124 #else
1125                                         struct sctp_event_subscribe event;
1126                                         socklen_t eventsize;
1127 #endif
1128                                         /* If a message has been delayed until the socket
1129                                          * is dry, it can be sent now.
1130                                          */
1131                                         if (data->saved_message.length > 0)
1132                                                 {
1133                                                 dgram_sctp_write(data->saved_message.bio, data->saved_message.data,
1134                                                                  data->saved_message.length);
1135                                                 OPENSSL_free(data->saved_message.data);
1136                                                 data->saved_message.length = 0;
1137                                                 }
1138
1139                                         /* disable sender dry event */
1140 #ifdef SCTP_EVENT
1141                                         memset(&event, 0, sizeof(struct sctp_event));
1142                                         event.se_assoc_id = 0;
1143                                         event.se_type = SCTP_SENDER_DRY_EVENT;
1144                                         event.se_on = 0;
1145                                         i = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(struct sctp_event));
1146                                         OPENSSL_assert(i >= 0);
1147 #else
1148                                         eventsize = sizeof(struct sctp_event_subscribe);
1149                                         i = getsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, &eventsize);
1150                                         OPENSSL_assert(i >= 0);
1151
1152                                         event.sctp_sender_dry_event = 0;
1153
1154                                         i = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, sizeof(struct sctp_event_subscribe));
1155                                         OPENSSL_assert(i >= 0);
1156 #endif
1157                                         }
1158
1159 #ifdef SCTP_AUTHENTICATION_EVENT
1160                                 if (snp->sn_header.sn_type == SCTP_AUTHENTICATION_EVENT)
1161                                         dgram_sctp_handle_auth_free_key_event(b, snp);
1162 #endif
1163
1164                                 if (data->handle_notifications != NULL)
1165                                         data->handle_notifications(b, data->notification_context, (void*) out);
1166
1167                                 memset(out, 0, outl);
1168                                 }
1169                         else
1170                                 ret += n;
1171                         }
1172                 while ((msg.msg_flags & MSG_NOTIFICATION) && (msg.msg_flags & MSG_EOR) && (ret < outl));
1173
1174                 if (ret > 0 && !(msg.msg_flags & MSG_EOR))
1175                         {
1176                         /* Partial message read, this should never happen! */
1177
1178                         /* The buffer was too small, this means the peer sent
1179                          * a message that was larger than allowed. */
1180                         if (ret == outl)
1181                                 return -1;
1182
1183                         /* Test if socket buffer can handle max record
1184                          * size (2^14 + 2048 + 13)
1185                          */
1186                         optlen = (socklen_t) sizeof(int);
1187                         ret = getsockopt(b->num, SOL_SOCKET, SO_RCVBUF, &optval, &optlen);
1188                         OPENSSL_assert(ret >= 0);
1189                         OPENSSL_assert(optval >= 18445);
1190
1191                         /* Test if SCTP doesn't partially deliver below
1192                          * max record size (2^14 + 2048 + 13)
1193                          */
1194                         optlen = (socklen_t) sizeof(int);
1195                         ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_PARTIAL_DELIVERY_POINT,
1196                                          &optval, &optlen);
1197                         OPENSSL_assert(ret >= 0);
1198                         OPENSSL_assert(optval >= 18445);
1199
1200                         /* Partially delivered notification??? Probably a bug.... */
1201                         OPENSSL_assert(!(msg.msg_flags & MSG_NOTIFICATION));
1202
1203                         /* Everything seems ok till now, so it's most likely
1204                          * a message dropped by PR-SCTP.
1205                          */
1206                         memset(out, 0, outl);
1207                         BIO_set_retry_read(b);
1208                         return -1;
1209                         }
1210
1211                 BIO_clear_retry_flags(b);
1212                 if (ret < 0)
1213                         {
1214                         if (BIO_dgram_should_retry(ret))
1215                                 {
1216                                 BIO_set_retry_read(b);
1217                                 data->_errno = get_last_socket_error();
1218                                 }
1219                         }
1220
1221                 /* Test if peer uses SCTP-AUTH before continuing */
1222                 if (!data->peer_auth_tested)
1223                         {
1224                         int ii, auth_data = 0, auth_forward = 0;
1225                         unsigned char *p;
1226                         struct sctp_authchunks *authchunks;
1227
1228                         optlen = (socklen_t)(sizeof(sctp_assoc_t) + 256 * sizeof(uint8_t));
1229                         authchunks = OPENSSL_malloc(optlen);
1230                         memset(authchunks, 0, sizeof(optlen));
1231                         ii = getsockopt(b->num, IPPROTO_SCTP, SCTP_PEER_AUTH_CHUNKS, authchunks, &optlen);
1232                         OPENSSL_assert(ii >= 0);
1233
1234                         for (p = (unsigned char*) authchunks->gauth_chunks;
1235                                  p < (unsigned char*) authchunks + optlen;
1236                                  p += sizeof(uint8_t))
1237                                 {
1238                                 if (*p == OPENSSL_SCTP_DATA_CHUNK_TYPE) auth_data = 1;
1239                                 if (*p == OPENSSL_SCTP_FORWARD_CUM_TSN_CHUNK_TYPE) auth_forward = 1;
1240                                 }
1241
1242                         OPENSSL_free(authchunks);
1243
1244                         if (!auth_data || !auth_forward)
1245                                 {
1246                                 BIOerr(BIO_F_DGRAM_SCTP_READ,BIO_R_CONNECT_ERROR);
1247                                 return -1;
1248                                 }
1249
1250                         data->peer_auth_tested = 1;
1251                         }
1252                 }
1253         return(ret);
1254         }
1255
1256 static int dgram_sctp_write(BIO *b, const char *in, int inl)
1257         {
1258         int ret;
1259         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *)b->ptr;
1260         struct bio_dgram_sctp_sndinfo *sinfo = &(data->sndinfo);
1261         struct bio_dgram_sctp_prinfo *pinfo = &(data->prinfo);
1262         struct bio_dgram_sctp_sndinfo handshake_sinfo;
1263         struct iovec iov[1];
1264         struct msghdr msg;
1265         struct cmsghdr *cmsg;
1266 #if defined(SCTP_SNDINFO) && defined(SCTP_PRINFO)
1267         char cmsgbuf[CMSG_SPACE(sizeof(struct sctp_sndinfo)) + CMSG_SPACE(sizeof(struct sctp_prinfo))];
1268         struct sctp_sndinfo *sndinfo;
1269         struct sctp_prinfo *prinfo;
1270 #else
1271         char cmsgbuf[CMSG_SPACE(sizeof(struct sctp_sndrcvinfo))];
1272         struct sctp_sndrcvinfo *sndrcvinfo;
1273 #endif
1274
1275         clear_socket_error();
1276
1277         /* If we're send anything else than application data,
1278          * disable all user parameters and flags.
1279          */
1280         if (in[0] != 23) {
1281                 memset(&handshake_sinfo, 0x00, sizeof(struct bio_dgram_sctp_sndinfo));
1282 #ifdef SCTP_SACK_IMMEDIATELY
1283                 handshake_sinfo.snd_flags = SCTP_SACK_IMMEDIATELY;
1284 #endif
1285                 sinfo = &handshake_sinfo;
1286         }
1287
1288         /* If we have to send a shutdown alert message and the
1289          * socket is not dry yet, we have to save it and send it
1290          * as soon as the socket gets dry.
1291          */
1292         if (data->save_shutdown && !BIO_dgram_sctp_wait_for_dry(b))
1293         {
1294                 data->saved_message.bio = b;
1295                 data->saved_message.length = inl;
1296                 data->saved_message.data = OPENSSL_malloc(inl);
1297                 memcpy(data->saved_message.data, in, inl);
1298                 return inl;
1299         }
1300
1301         iov[0].iov_base = (char *)in;
1302         iov[0].iov_len = inl;
1303         msg.msg_name = NULL;
1304         msg.msg_namelen = 0;
1305         msg.msg_iov = iov;
1306         msg.msg_iovlen = 1;
1307         msg.msg_control = (caddr_t)cmsgbuf;
1308         msg.msg_controllen = 0;
1309         msg.msg_flags = 0;
1310 #if defined(SCTP_SNDINFO) && defined(SCTP_PRINFO)
1311         cmsg = (struct cmsghdr *)cmsgbuf;
1312         cmsg->cmsg_level = IPPROTO_SCTP;
1313         cmsg->cmsg_type = SCTP_SNDINFO;
1314         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndinfo));
1315         sndinfo = (struct sctp_sndinfo *)CMSG_DATA(cmsg);
1316         memset(sndinfo, 0, sizeof(struct sctp_sndinfo));
1317         sndinfo->snd_sid = sinfo->snd_sid;
1318         sndinfo->snd_flags = sinfo->snd_flags;
1319         sndinfo->snd_ppid = sinfo->snd_ppid;
1320         sndinfo->snd_context = sinfo->snd_context;
1321         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_sndinfo));
1322
1323         cmsg = (struct cmsghdr *)&cmsgbuf[CMSG_SPACE(sizeof(struct sctp_sndinfo))];
1324         cmsg->cmsg_level = IPPROTO_SCTP;
1325         cmsg->cmsg_type = SCTP_PRINFO;
1326         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_prinfo));
1327         prinfo = (struct sctp_prinfo *)CMSG_DATA(cmsg);
1328         memset(prinfo, 0, sizeof(struct sctp_prinfo));
1329         prinfo->pr_policy = pinfo->pr_policy;
1330         prinfo->pr_value = pinfo->pr_value;
1331         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_prinfo));
1332 #else
1333         cmsg = (struct cmsghdr *)cmsgbuf;
1334         cmsg->cmsg_level = IPPROTO_SCTP;
1335         cmsg->cmsg_type = SCTP_SNDRCV;
1336         cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndrcvinfo));
1337         sndrcvinfo = (struct sctp_sndrcvinfo *)CMSG_DATA(cmsg);
1338         memset(sndrcvinfo, 0, sizeof(struct sctp_sndrcvinfo));
1339         sndrcvinfo->sinfo_stream = sinfo->snd_sid;
1340         sndrcvinfo->sinfo_flags = sinfo->snd_flags;
1341 #ifdef __FreeBSD__
1342         sndrcvinfo->sinfo_flags |= pinfo->pr_policy;
1343 #endif
1344         sndrcvinfo->sinfo_ppid = sinfo->snd_ppid;
1345         sndrcvinfo->sinfo_context = sinfo->snd_context;
1346         sndrcvinfo->sinfo_timetolive = pinfo->pr_value;
1347         msg.msg_controllen += CMSG_SPACE(sizeof(struct sctp_sndrcvinfo));
1348 #endif
1349
1350         ret = sendmsg(b->num, &msg, 0);
1351
1352         BIO_clear_retry_flags(b);
1353         if (ret <= 0)
1354                 {
1355                 if (BIO_dgram_should_retry(ret))
1356                         {
1357                         BIO_set_retry_write(b);  
1358                         data->_errno = get_last_socket_error();
1359                         }
1360                 }
1361         return(ret);
1362         }
1363
1364 static long dgram_sctp_ctrl(BIO *b, int cmd, long num, void *ptr)
1365         {
1366         long ret=1;
1367         bio_dgram_sctp_data *data = NULL;
1368         socklen_t sockopt_len = 0;
1369         struct sctp_authkeyid authkeyid;
1370         struct sctp_authkey *authkey = NULL;
1371
1372         data = (bio_dgram_sctp_data *)b->ptr;
1373
1374         switch (cmd)
1375                 {
1376         case BIO_CTRL_DGRAM_QUERY_MTU:
1377                 /* Set to maximum (2^14)
1378                  * and ignore user input to enable transport
1379                  * protocol fragmentation.
1380                  * Returns always 2^14.
1381                  */
1382                 data->mtu = 16384;
1383                 ret = data->mtu;
1384                 break;
1385         case BIO_CTRL_DGRAM_SET_MTU:
1386                 /* Set to maximum (2^14)
1387                  * and ignore input to enable transport
1388                  * protocol fragmentation.
1389                  * Returns always 2^14.
1390                  */
1391                 data->mtu = 16384;
1392                 ret = data->mtu;
1393                 break;
1394         case BIO_CTRL_DGRAM_SET_CONNECTED:
1395         case BIO_CTRL_DGRAM_CONNECT:
1396                 /* Returns always -1. */
1397                 ret = -1;
1398                 break;
1399         case BIO_CTRL_DGRAM_SET_NEXT_TIMEOUT:
1400                 /* SCTP doesn't need the DTLS timer
1401                  * Returns always 1.
1402                  */
1403                 break;
1404         case BIO_CTRL_DGRAM_GET_MTU_OVERHEAD:
1405                 /* We allow transport protocol fragmentation so this is irrelevant */
1406                 ret = 0;
1407                 break;
1408         case BIO_CTRL_DGRAM_SCTP_SET_IN_HANDSHAKE:
1409                 if (num > 0)
1410                         data->in_handshake = 1;
1411                 else
1412                         data->in_handshake = 0;
1413
1414                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_NODELAY, &data->in_handshake, sizeof(int));
1415                 break;
1416         case BIO_CTRL_DGRAM_SCTP_ADD_AUTH_KEY:
1417                 /* New shared key for SCTP AUTH.
1418                  * Returns 0 on success, -1 otherwise.
1419                  */
1420
1421                 /* Get active key */
1422                 sockopt_len = sizeof(struct sctp_authkeyid);
1423                 ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY, &authkeyid, &sockopt_len);
1424                 if (ret < 0) break;
1425
1426                 /* Add new key */
1427                 sockopt_len = sizeof(struct sctp_authkey) + 64 * sizeof(uint8_t);
1428                 authkey = OPENSSL_malloc(sockopt_len);
1429                 if (authkey == NULL)
1430                         {
1431                         ret = -1;
1432                         break;
1433                         }
1434                 memset(authkey, 0x00, sockopt_len);
1435                 authkey->sca_keynumber = authkeyid.scact_keynumber + 1;
1436 #ifndef __FreeBSD__
1437                 /* This field is missing in FreeBSD 8.2 and earlier,
1438                  * and FreeBSD 8.3 and higher work without it.
1439                  */
1440                 authkey->sca_keylength = 64;
1441 #endif
1442                 memcpy(&authkey->sca_key[0], ptr, 64 * sizeof(uint8_t));
1443
1444                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_KEY, authkey, sockopt_len);
1445                 OPENSSL_free(authkey);
1446                 authkey = NULL;
1447                 if (ret < 0) break;
1448
1449                 /* Reset active key */
1450                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY,
1451                       &authkeyid, sizeof(struct sctp_authkeyid));
1452                 if (ret < 0) break;
1453
1454                 break;
1455         case BIO_CTRL_DGRAM_SCTP_NEXT_AUTH_KEY:
1456                 /* Returns 0 on success, -1 otherwise. */
1457
1458                 /* Get active key */
1459                 sockopt_len = sizeof(struct sctp_authkeyid);
1460                 ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY, &authkeyid, &sockopt_len);
1461                 if (ret < 0) break;
1462
1463                 /* Set active key */
1464                 authkeyid.scact_keynumber = authkeyid.scact_keynumber + 1;
1465                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY,
1466                       &authkeyid, sizeof(struct sctp_authkeyid));
1467                 if (ret < 0) break;
1468
1469                 /* CCS has been sent, so remember that and fall through
1470                  * to check if we need to deactivate an old key
1471                  */
1472                 data->ccs_sent = 1;
1473
1474         case BIO_CTRL_DGRAM_SCTP_AUTH_CCS_RCVD:
1475                 /* Returns 0 on success, -1 otherwise. */
1476
1477                 /* Has this command really been called or is this just a fall-through? */
1478                 if (cmd == BIO_CTRL_DGRAM_SCTP_AUTH_CCS_RCVD)
1479                         data->ccs_rcvd = 1;
1480
1481                 /* CSS has been both, received and sent, so deactivate an old key */
1482                 if (data->ccs_rcvd == 1 && data->ccs_sent == 1)
1483                         {
1484                         /* Get active key */
1485                         sockopt_len = sizeof(struct sctp_authkeyid);
1486                         ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_ACTIVE_KEY, &authkeyid, &sockopt_len);
1487                         if (ret < 0) break;
1488
1489                         /* Deactivate key or delete second last key if
1490                          * SCTP_AUTHENTICATION_EVENT is not available.
1491                          */
1492                         authkeyid.scact_keynumber = authkeyid.scact_keynumber - 1;
1493 #ifdef SCTP_AUTH_DEACTIVATE_KEY
1494                         sockopt_len = sizeof(struct sctp_authkeyid);
1495                         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_DEACTIVATE_KEY,
1496                               &authkeyid, sockopt_len);
1497                         if (ret < 0) break;
1498 #endif
1499 #ifndef SCTP_AUTHENTICATION_EVENT
1500                         if (authkeyid.scact_keynumber > 0)
1501                                 {
1502                                 authkeyid.scact_keynumber = authkeyid.scact_keynumber - 1;
1503                                 ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_AUTH_DELETE_KEY,
1504                                           &authkeyid, sizeof(struct sctp_authkeyid));
1505                                 if (ret < 0) break;
1506                                 }
1507 #endif
1508
1509                         data->ccs_rcvd = 0;
1510                         data->ccs_sent = 0;
1511                         }
1512                 break;
1513         case BIO_CTRL_DGRAM_SCTP_GET_SNDINFO:
1514                 /* Returns the size of the copied struct. */
1515                 if (num > (long) sizeof(struct bio_dgram_sctp_sndinfo))
1516                         num = sizeof(struct bio_dgram_sctp_sndinfo);
1517
1518                 memcpy(ptr, &(data->sndinfo), num);
1519                 ret = num;
1520                 break;
1521         case BIO_CTRL_DGRAM_SCTP_SET_SNDINFO:
1522                 /* Returns the size of the copied struct. */
1523                 if (num > (long) sizeof(struct bio_dgram_sctp_sndinfo))
1524                         num = sizeof(struct bio_dgram_sctp_sndinfo);
1525
1526                 memcpy(&(data->sndinfo), ptr, num);
1527                 break;
1528         case BIO_CTRL_DGRAM_SCTP_GET_RCVINFO:
1529                 /* Returns the size of the copied struct. */
1530                 if (num > (long) sizeof(struct bio_dgram_sctp_rcvinfo))
1531                         num = sizeof(struct bio_dgram_sctp_rcvinfo);
1532
1533                 memcpy(ptr, &data->rcvinfo, num);
1534
1535                 ret = num;
1536                 break;
1537         case BIO_CTRL_DGRAM_SCTP_SET_RCVINFO:
1538                 /* Returns the size of the copied struct. */
1539                 if (num > (long) sizeof(struct bio_dgram_sctp_rcvinfo))
1540                         num = sizeof(struct bio_dgram_sctp_rcvinfo);
1541
1542                 memcpy(&(data->rcvinfo), ptr, num);
1543                 break;
1544         case BIO_CTRL_DGRAM_SCTP_GET_PRINFO:
1545                 /* Returns the size of the copied struct. */
1546                 if (num > (long) sizeof(struct bio_dgram_sctp_prinfo))
1547                         num = sizeof(struct bio_dgram_sctp_prinfo);
1548
1549                 memcpy(ptr, &(data->prinfo), num);
1550                 ret = num;
1551                 break;
1552         case BIO_CTRL_DGRAM_SCTP_SET_PRINFO:
1553                 /* Returns the size of the copied struct. */
1554                 if (num > (long) sizeof(struct bio_dgram_sctp_prinfo))
1555                         num = sizeof(struct bio_dgram_sctp_prinfo);
1556
1557                 memcpy(&(data->prinfo), ptr, num);
1558                 break;
1559         case BIO_CTRL_DGRAM_SCTP_SAVE_SHUTDOWN:
1560                 /* Returns always 1. */
1561                 if (num > 0)
1562                         data->save_shutdown = 1;
1563                 else
1564                         data->save_shutdown = 0;
1565                 break;
1566
1567         default:
1568                 /* Pass to default ctrl function to
1569                  * process SCTP unspecific commands
1570                  */
1571                 ret=dgram_ctrl(b, cmd, num, ptr);
1572                 break;
1573                 }
1574         return(ret);
1575         }
1576
1577 int BIO_dgram_sctp_notification_cb(BIO *b,
1578                                    void (*handle_notifications)(BIO *bio, void *context, void *buf),
1579                                    void *context)
1580         {
1581         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *) b->ptr;
1582
1583         if (handle_notifications != NULL)
1584                 {
1585                 data->handle_notifications = handle_notifications;
1586                 data->notification_context = context;
1587                 }
1588         else
1589                 return -1;
1590
1591         return 0;
1592         }
1593
1594 int BIO_dgram_sctp_wait_for_dry(BIO *b)
1595 {
1596         int is_dry = 0;
1597         int n, sockflags, ret;
1598         union sctp_notification snp;
1599         struct msghdr msg;
1600         struct iovec iov;
1601 #ifdef SCTP_EVENT
1602         struct sctp_event event;
1603 #else
1604         struct sctp_event_subscribe event;
1605         socklen_t eventsize;
1606 #endif
1607         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *)b->ptr;
1608
1609         /* set sender dry event */
1610 #ifdef SCTP_EVENT
1611         memset(&event, 0, sizeof(struct sctp_event));
1612         event.se_assoc_id = 0;
1613         event.se_type = SCTP_SENDER_DRY_EVENT;
1614         event.se_on = 1;
1615         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(struct sctp_event));
1616 #else
1617         eventsize = sizeof(struct sctp_event_subscribe);
1618         ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, &eventsize);
1619         if (ret < 0)
1620                 return -1;
1621         
1622         event.sctp_sender_dry_event = 1;
1623         
1624         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, sizeof(struct sctp_event_subscribe));
1625 #endif
1626         if (ret < 0)
1627                 return -1;
1628
1629         /* peek for notification */
1630         memset(&snp, 0x00, sizeof(union sctp_notification));
1631         iov.iov_base = (char *)&snp;
1632         iov.iov_len = sizeof(union sctp_notification);
1633         msg.msg_name = NULL;
1634         msg.msg_namelen = 0;
1635         msg.msg_iov = &iov;
1636         msg.msg_iovlen = 1;
1637         msg.msg_control = NULL;
1638         msg.msg_controllen = 0;
1639         msg.msg_flags = 0;
1640
1641         n = recvmsg(b->num, &msg, MSG_PEEK);
1642         if (n <= 0)
1643                 {
1644                 if ((n < 0) && (get_last_socket_error() != EAGAIN) && (get_last_socket_error() != EWOULDBLOCK))
1645                         return -1;
1646                 else
1647                         return 0;
1648                 }
1649
1650         /* if we find a notification, process it and try again if necessary */
1651         while (msg.msg_flags & MSG_NOTIFICATION)
1652                 {
1653                 memset(&snp, 0x00, sizeof(union sctp_notification));
1654                 iov.iov_base = (char *)&snp;
1655                 iov.iov_len = sizeof(union sctp_notification);
1656                 msg.msg_name = NULL;
1657                 msg.msg_namelen = 0;
1658                 msg.msg_iov = &iov;
1659                 msg.msg_iovlen = 1;
1660                 msg.msg_control = NULL;
1661                 msg.msg_controllen = 0;
1662                 msg.msg_flags = 0;
1663
1664                 n = recvmsg(b->num, &msg, 0);
1665                 if (n <= 0)
1666                         {
1667                         if ((n < 0) && (get_last_socket_error() != EAGAIN) && (get_last_socket_error() != EWOULDBLOCK))
1668                                 return -1;
1669                         else
1670                                 return is_dry;
1671                         }
1672                 
1673                 if (snp.sn_header.sn_type == SCTP_SENDER_DRY_EVENT)
1674                         {
1675                         is_dry = 1;
1676
1677                         /* disable sender dry event */
1678 #ifdef SCTP_EVENT
1679                         memset(&event, 0, sizeof(struct sctp_event));
1680                         event.se_assoc_id = 0;
1681                         event.se_type = SCTP_SENDER_DRY_EVENT;
1682                         event.se_on = 0;
1683                         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(struct sctp_event));
1684 #else
1685                         eventsize = (socklen_t) sizeof(struct sctp_event_subscribe);
1686                         ret = getsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, &eventsize);
1687                         if (ret < 0)
1688                                 return -1;
1689
1690                         event.sctp_sender_dry_event = 0;
1691
1692                         ret = setsockopt(b->num, IPPROTO_SCTP, SCTP_EVENTS, &event, sizeof(struct sctp_event_subscribe));
1693 #endif
1694                         if (ret < 0)
1695                                 return -1;
1696                         }
1697
1698 #ifdef SCTP_AUTHENTICATION_EVENT
1699                 if (snp.sn_header.sn_type == SCTP_AUTHENTICATION_EVENT)
1700                         dgram_sctp_handle_auth_free_key_event(b, &snp);
1701 #endif
1702
1703                 if (data->handle_notifications != NULL)
1704                         data->handle_notifications(b, data->notification_context, (void*) &snp);
1705
1706                 /* found notification, peek again */
1707                 memset(&snp, 0x00, sizeof(union sctp_notification));
1708                 iov.iov_base = (char *)&snp;
1709                 iov.iov_len = sizeof(union sctp_notification);
1710                 msg.msg_name = NULL;
1711                 msg.msg_namelen = 0;
1712                 msg.msg_iov = &iov;
1713                 msg.msg_iovlen = 1;
1714                 msg.msg_control = NULL;
1715                 msg.msg_controllen = 0;
1716                 msg.msg_flags = 0;
1717
1718                 /* if we have seen the dry already, don't wait */
1719                 if (is_dry)
1720                         {
1721                         sockflags = fcntl(b->num, F_GETFL, 0);
1722                         fcntl(b->num, F_SETFL, O_NONBLOCK);
1723                         }
1724
1725                 n = recvmsg(b->num, &msg, MSG_PEEK);
1726
1727                 if (is_dry)
1728                         {
1729                         fcntl(b->num, F_SETFL, sockflags);
1730                         }
1731
1732                 if (n <= 0)
1733                         {
1734                         if ((n < 0) && (get_last_socket_error() != EAGAIN) && (get_last_socket_error() != EWOULDBLOCK))
1735                                 return -1;
1736                         else
1737                                 return is_dry;
1738                         }
1739                 }
1740
1741         /* read anything else */
1742         return is_dry;
1743 }
1744
1745 int BIO_dgram_sctp_msg_waiting(BIO *b)
1746         {
1747         int n, sockflags;
1748         union sctp_notification snp;
1749         struct msghdr msg;
1750         struct iovec iov;
1751         bio_dgram_sctp_data *data = (bio_dgram_sctp_data *)b->ptr;
1752
1753         /* Check if there are any messages waiting to be read */
1754         do
1755                 {
1756                 memset(&snp, 0x00, sizeof(union sctp_notification));
1757                 iov.iov_base = (char *)&snp;
1758                 iov.iov_len = sizeof(union sctp_notification);
1759                 msg.msg_name = NULL;
1760                 msg.msg_namelen = 0;
1761                 msg.msg_iov = &iov;
1762                 msg.msg_iovlen = 1;
1763                 msg.msg_control = NULL;
1764                 msg.msg_controllen = 0;
1765                 msg.msg_flags = 0;
1766
1767                 sockflags = fcntl(b->num, F_GETFL, 0);
1768                 fcntl(b->num, F_SETFL, O_NONBLOCK);
1769                 n = recvmsg(b->num, &msg, MSG_PEEK);
1770                 fcntl(b->num, F_SETFL, sockflags);
1771
1772                 /* if notification, process and try again */
1773                 if (n > 0 && (msg.msg_flags & MSG_NOTIFICATION))
1774                         {
1775 #ifdef SCTP_AUTHENTICATION_EVENT
1776                         if (snp.sn_header.sn_type == SCTP_AUTHENTICATION_EVENT)
1777                                 dgram_sctp_handle_auth_free_key_event(b, &snp);
1778 #endif
1779
1780                         memset(&snp, 0x00, sizeof(union sctp_notification));
1781                         iov.iov_base = (char *)&snp;
1782                         iov.iov_len = sizeof(union sctp_notification);
1783                         msg.msg_name = NULL;
1784                         msg.msg_namelen = 0;
1785                         msg.msg_iov = &iov;
1786                         msg.msg_iovlen = 1;
1787                         msg.msg_control = NULL;
1788                         msg.msg_controllen = 0;
1789                         msg.msg_flags = 0;
1790                         n = recvmsg(b->num, &msg, 0);
1791
1792                         if (data->handle_notifications != NULL)
1793                                 data->handle_notifications(b, data->notification_context, (void*) &snp);
1794                         }
1795
1796                 } while (n > 0 && (msg.msg_flags & MSG_NOTIFICATION));
1797
1798         /* Return 1 if there is a message to be read, return 0 otherwise. */
1799         if (n > 0)
1800                 return 1;
1801         else
1802                 return 0;
1803         }
1804
1805 static int dgram_sctp_puts(BIO *bp, const char *str)
1806         {
1807         int n,ret;
1808
1809         n=strlen(str);
1810         ret=dgram_sctp_write(bp,str,n);
1811         return(ret);
1812         }
1813 #endif
1814
1815 static int BIO_dgram_should_retry(int i)
1816         {
1817         int err;
1818
1819         if ((i == 0) || (i == -1))
1820                 {
1821                 err=get_last_socket_error();
1822
1823 #if defined(OPENSSL_SYS_WINDOWS)
1824         /* If the socket return value (i) is -1
1825          * and err is unexpectedly 0 at this point,
1826          * the error code was overwritten by
1827          * another system call before this error
1828          * handling is called.
1829          */
1830 #endif
1831
1832                 return(BIO_dgram_non_fatal_error(err));
1833                 }
1834         return(0);
1835         }
1836
1837 int BIO_dgram_non_fatal_error(int err)
1838         {
1839         switch (err)
1840                 {
1841 #if defined(OPENSSL_SYS_WINDOWS)
1842 # if defined(WSAEWOULDBLOCK)
1843         case WSAEWOULDBLOCK:
1844 # endif
1845
1846 # if 0 /* This appears to always be an error */
1847 #  if defined(WSAENOTCONN)
1848         case WSAENOTCONN:
1849 #  endif
1850 # endif
1851 #endif
1852
1853 #ifdef EWOULDBLOCK
1854 # ifdef WSAEWOULDBLOCK
1855 #  if WSAEWOULDBLOCK != EWOULDBLOCK
1856         case EWOULDBLOCK:
1857 #  endif
1858 # else
1859         case EWOULDBLOCK:
1860 # endif
1861 #endif
1862
1863 #ifdef EINTR
1864         case EINTR:
1865 #endif
1866
1867 #ifdef EAGAIN
1868 #if EWOULDBLOCK != EAGAIN
1869         case EAGAIN:
1870 # endif
1871 #endif
1872
1873 #ifdef EPROTO
1874         case EPROTO:
1875 #endif
1876
1877 #ifdef EINPROGRESS
1878         case EINPROGRESS:
1879 #endif
1880
1881 #ifdef EALREADY
1882         case EALREADY:
1883 #endif
1884
1885                 return(1);
1886                 /* break; */
1887         default:
1888                 break;
1889                 }
1890         return(0);
1891         }
1892
1893 static void get_current_time(struct timeval *t)
1894         {
1895 #ifdef OPENSSL_SYS_WIN32
1896         struct _timeb tb;
1897         _ftime(&tb);
1898         t->tv_sec = (long)tb.time;
1899         t->tv_usec = (long)tb.millitm * 1000;
1900 #elif defined(OPENSSL_SYS_VMS)
1901         struct timeb tb;
1902         ftime(&tb);
1903         t->tv_sec = (long)tb.time;
1904         t->tv_usec = (long)tb.millitm * 1000;
1905 #else
1906         gettimeofday(t, NULL);
1907 #endif
1908         }
1909
1910 #endif