• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 #ifdef OPENSSL_SYS_LINUX
74 #define IP_MTU      14 /* linux is lame */
75 #endif
76 
77 #ifdef WATT32
78 #define sock_write SockWrite  /* Watt-32 uses same names */
79 #define sock_read  SockRead
80 #define sock_puts  SockPuts
81 #endif
82 
83 static int dgram_write(BIO *h, const char *buf, int num);
84 static int dgram_read(BIO *h, char *buf, int size);
85 static int dgram_puts(BIO *h, const char *str);
86 static long dgram_ctrl(BIO *h, int cmd, long arg1, void *arg2);
87 static int dgram_new(BIO *h);
88 static int dgram_free(BIO *data);
89 static int dgram_clear(BIO *bio);
90 
91 static int BIO_dgram_should_retry(int s);
92 
93 static void get_current_time(struct timeval *t);
94 
95 static BIO_METHOD methods_dgramp=
96 	{
97 	BIO_TYPE_DGRAM,
98 	"datagram socket",
99 	dgram_write,
100 	dgram_read,
101 	dgram_puts,
102 	NULL, /* dgram_gets, */
103 	dgram_ctrl,
104 	dgram_new,
105 	dgram_free,
106 	NULL,
107 	};
108 
109 typedef struct bio_dgram_data_st
110 	{
111 	union {
112 		struct sockaddr sa;
113 		struct sockaddr_in sa_in;
114 #if OPENSSL_USE_IPV6
115 		struct sockaddr_in6 sa_in6;
116 #endif
117 	} peer;
118 	unsigned int connected;
119 	unsigned int _errno;
120 	unsigned int mtu;
121 	struct timeval next_timeout;
122 	struct timeval socket_timeout;
123 	} bio_dgram_data;
124 
BIO_s_datagram(void)125 BIO_METHOD *BIO_s_datagram(void)
126 	{
127 	return(&methods_dgramp);
128 	}
129 
BIO_new_dgram(int fd,int close_flag)130 BIO *BIO_new_dgram(int fd, int close_flag)
131 	{
132 	BIO *ret;
133 
134 	ret=BIO_new(BIO_s_datagram());
135 	if (ret == NULL) return(NULL);
136 	BIO_set_fd(ret,fd,close_flag);
137 	return(ret);
138 	}
139 
dgram_new(BIO * bi)140 static int dgram_new(BIO *bi)
141 	{
142 	bio_dgram_data *data = NULL;
143 
144 	bi->init=0;
145 	bi->num=0;
146 	data = OPENSSL_malloc(sizeof(bio_dgram_data));
147 	if (data == NULL)
148 		return 0;
149 	memset(data, 0x00, sizeof(bio_dgram_data));
150     bi->ptr = data;
151 
152 	bi->flags=0;
153 	return(1);
154 	}
155 
dgram_free(BIO * a)156 static int dgram_free(BIO *a)
157 	{
158 	bio_dgram_data *data;
159 
160 	if (a == NULL) return(0);
161 	if ( ! dgram_clear(a))
162 		return 0;
163 
164 	data = (bio_dgram_data *)a->ptr;
165 	if(data != NULL) OPENSSL_free(data);
166 
167 	return(1);
168 	}
169 
dgram_clear(BIO * a)170 static int dgram_clear(BIO *a)
171 	{
172 	if (a == NULL) return(0);
173 	if (a->shutdown)
174 		{
175 		if (a->init)
176 			{
177 			SHUTDOWN2(a->num);
178 			}
179 		a->init=0;
180 		a->flags=0;
181 		}
182 	return(1);
183 	}
184 
dgram_adjust_rcv_timeout(BIO * b)185 static void dgram_adjust_rcv_timeout(BIO *b)
186 	{
187 #if defined(SO_RCVTIMEO)
188 	bio_dgram_data *data = (bio_dgram_data *)b->ptr;
189 	int sz = sizeof(int);
190 
191 	/* Is a timer active? */
192 	if (data->next_timeout.tv_sec > 0 || data->next_timeout.tv_usec > 0)
193 		{
194 		struct timeval timenow, timeleft;
195 
196 		/* Read current socket timeout */
197 #ifdef OPENSSL_SYS_WINDOWS
198 		int timeout;
199 		if (getsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
200 					   (void*)&timeout, &sz) < 0)
201 			{ perror("getsockopt"); }
202 		else
203 			{
204 			data->socket_timeout.tv_sec = timeout / 1000;
205 			data->socket_timeout.tv_usec = (timeout % 1000) * 1000;
206 			}
207 #else
208 		if ( getsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
209 						&(data->socket_timeout), (void *)&sz) < 0)
210 			{ perror("getsockopt"); }
211 #endif
212 
213 		/* Get current time */
214 		get_current_time(&timenow);
215 
216 		/* Calculate time left until timer expires */
217 		memcpy(&timeleft, &(data->next_timeout), sizeof(struct timeval));
218 		timeleft.tv_sec -= timenow.tv_sec;
219 		timeleft.tv_usec -= timenow.tv_usec;
220 		if (timeleft.tv_usec < 0)
221 			{
222 			timeleft.tv_sec--;
223 			timeleft.tv_usec += 1000000;
224 			}
225 
226 		if (timeleft.tv_sec < 0)
227 			{
228 			timeleft.tv_sec = 0;
229 			timeleft.tv_usec = 1;
230 			}
231 
232 		/* Adjust socket timeout if next handhake message timer
233 		 * will expire earlier.
234 		 */
235 		if ((data->socket_timeout.tv_sec == 0 && data->socket_timeout.tv_usec == 0) ||
236 			(data->socket_timeout.tv_sec > timeleft.tv_sec) ||
237 			(data->socket_timeout.tv_sec == timeleft.tv_sec &&
238 			 data->socket_timeout.tv_usec >= timeleft.tv_usec))
239 			{
240 #ifdef OPENSSL_SYS_WINDOWS
241 			timeout = timeleft.tv_sec * 1000 + timeleft.tv_usec / 1000;
242 			if (setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
243 						   (void*)&timeout, sizeof(timeout)) < 0)
244 				{ perror("setsockopt"); }
245 #else
246 			if ( setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO, &timeleft,
247 							sizeof(struct timeval)) < 0)
248 				{ perror("setsockopt"); }
249 #endif
250 			}
251 		}
252 #endif
253 	}
254 
dgram_reset_rcv_timeout(BIO * b)255 static void dgram_reset_rcv_timeout(BIO *b)
256 	{
257 #if defined(SO_RCVTIMEO)
258 	bio_dgram_data *data = (bio_dgram_data *)b->ptr;
259 
260 	/* Is a timer active? */
261 	if (data->next_timeout.tv_sec > 0 || data->next_timeout.tv_usec > 0)
262 		{
263 #ifdef OPENSSL_SYS_WINDOWS
264 		int timeout = data->socket_timeout.tv_sec * 1000 +
265 					  data->socket_timeout.tv_usec / 1000;
266 		if (setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
267 					   (void*)&timeout, sizeof(timeout)) < 0)
268 			{ perror("setsockopt"); }
269 #else
270 		if ( setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO, &(data->socket_timeout),
271 						sizeof(struct timeval)) < 0)
272 			{ perror("setsockopt"); }
273 #endif
274 		}
275 #endif
276 	}
277 
dgram_read(BIO * b,char * out,int outl)278 static int dgram_read(BIO *b, char *out, int outl)
279 	{
280 	int ret=0;
281 	bio_dgram_data *data = (bio_dgram_data *)b->ptr;
282 
283 	struct	{
284 	/*
285 	 * See commentary in b_sock.c. <appro>
286 	 */
287 	union	{ size_t s; int i; } len;
288 	union	{
289 		struct sockaddr sa;
290 		struct sockaddr_in sa_in;
291 #if OPENSSL_USE_IPV6
292 		struct sockaddr_in6 sa_in6;
293 #endif
294 		} peer;
295 	} sa;
296 
297 	sa.len.s=0;
298 	sa.len.i=sizeof(sa.peer);
299 
300 	if (out != NULL)
301 		{
302 		clear_socket_error();
303 		memset(&sa.peer, 0x00, sizeof(sa.peer));
304 		dgram_adjust_rcv_timeout(b);
305 		ret=recvfrom(b->num,out,outl,0,&sa.peer.sa,(void *)&sa.len);
306 		if (sizeof(sa.len.i)!=sizeof(sa.len.s) && sa.len.i==0)
307 			{
308 			OPENSSL_assert(sa.len.s<=sizeof(sa.peer));
309 			sa.len.i = (int)sa.len.s;
310 			}
311 
312 		if ( ! data->connected  && ret >= 0)
313 			BIO_ctrl(b, BIO_CTRL_DGRAM_SET_PEER, 0, &sa.peer);
314 
315 		BIO_clear_retry_flags(b);
316 		if (ret < 0)
317 			{
318 			if (BIO_dgram_should_retry(ret))
319 				{
320 				BIO_set_retry_read(b);
321 				data->_errno = get_last_socket_error();
322 				}
323 			}
324 
325 		dgram_reset_rcv_timeout(b);
326 		}
327 	return(ret);
328 	}
329 
dgram_write(BIO * b,const char * in,int inl)330 static int dgram_write(BIO *b, const char *in, int inl)
331 	{
332 	int ret;
333 	bio_dgram_data *data = (bio_dgram_data *)b->ptr;
334 	clear_socket_error();
335 
336 	if ( data->connected )
337 		ret=writesocket(b->num,in,inl);
338 	else
339 		{
340 		int peerlen = sizeof(data->peer);
341 
342 		if (data->peer.sa.sa_family == AF_INET)
343 			peerlen = sizeof(data->peer.sa_in);
344 #if OPENSSL_USE_IPV6
345 		else if (data->peer.sa.sa_family == AF_INET6)
346 			peerlen = sizeof(data->peer.sa_in6);
347 #endif
348 #if defined(NETWARE_CLIB) && defined(NETWARE_BSDSOCK)
349 		ret=sendto(b->num, (char *)in, inl, 0, &data->peer.sa, peerlen);
350 #else
351 		ret=sendto(b->num, in, inl, 0, &data->peer.sa, peerlen);
352 #endif
353 		}
354 
355 	BIO_clear_retry_flags(b);
356 	if (ret <= 0)
357 		{
358 		if (BIO_dgram_should_retry(ret))
359 			{
360 			BIO_set_retry_write(b);
361 			data->_errno = get_last_socket_error();
362 
363 #if 0 /* higher layers are responsible for querying MTU, if necessary */
364 			if ( data->_errno == EMSGSIZE)
365 				/* retrieve the new MTU */
366 				BIO_ctrl(b, BIO_CTRL_DGRAM_QUERY_MTU, 0, NULL);
367 #endif
368 			}
369 		}
370 	return(ret);
371 	}
372 
dgram_ctrl(BIO * b,int cmd,long num,void * ptr)373 static long dgram_ctrl(BIO *b, int cmd, long num, void *ptr)
374 	{
375 	long ret=1;
376 	int *ip;
377 	struct sockaddr *to = NULL;
378 	bio_dgram_data *data = NULL;
379 #if defined(IP_MTU_DISCOVER) || defined(IP_MTU)
380 	long sockopt_val = 0;
381 	unsigned int sockopt_len = 0;
382 #endif
383 #ifdef OPENSSL_SYS_LINUX
384 	socklen_t addr_len;
385 	union	{
386 		struct sockaddr	sa;
387 		struct sockaddr_in s4;
388 #if OPENSSL_USE_IPV6
389 		struct sockaddr_in6 s6;
390 #endif
391 		} addr;
392 #endif
393 
394 	data = (bio_dgram_data *)b->ptr;
395 
396 	switch (cmd)
397 		{
398 	case BIO_CTRL_RESET:
399 		num=0;
400 	case BIO_C_FILE_SEEK:
401 		ret=0;
402 		break;
403 	case BIO_C_FILE_TELL:
404 	case BIO_CTRL_INFO:
405 		ret=0;
406 		break;
407 	case BIO_C_SET_FD:
408 		dgram_clear(b);
409 		b->num= *((int *)ptr);
410 		b->shutdown=(int)num;
411 		b->init=1;
412 		break;
413 	case BIO_C_GET_FD:
414 		if (b->init)
415 			{
416 			ip=(int *)ptr;
417 			if (ip != NULL) *ip=b->num;
418 			ret=b->num;
419 			}
420 		else
421 			ret= -1;
422 		break;
423 	case BIO_CTRL_GET_CLOSE:
424 		ret=b->shutdown;
425 		break;
426 	case BIO_CTRL_SET_CLOSE:
427 		b->shutdown=(int)num;
428 		break;
429 	case BIO_CTRL_PENDING:
430 	case BIO_CTRL_WPENDING:
431 		ret=0;
432 		break;
433 	case BIO_CTRL_DUP:
434 	case BIO_CTRL_FLUSH:
435 		ret=1;
436 		break;
437 	case BIO_CTRL_DGRAM_CONNECT:
438 		to = (struct sockaddr *)ptr;
439 #if 0
440 		if (connect(b->num, to, sizeof(struct sockaddr)) < 0)
441 			{ perror("connect"); ret = 0; }
442 		else
443 			{
444 #endif
445 			switch (to->sa_family)
446 				{
447 				case AF_INET:
448 					memcpy(&data->peer,to,sizeof(data->peer.sa_in));
449 					break;
450 #if OPENSSL_USE_IPV6
451 				case AF_INET6:
452 					memcpy(&data->peer,to,sizeof(data->peer.sa_in6));
453 					break;
454 #endif
455 				default:
456 					memcpy(&data->peer,to,sizeof(data->peer.sa));
457 					break;
458 				}
459 #if 0
460 			}
461 #endif
462 		break;
463 		/* (Linux)kernel sets DF bit on outgoing IP packets */
464 	case BIO_CTRL_DGRAM_MTU_DISCOVER:
465 #ifdef OPENSSL_SYS_LINUX
466 		addr_len = (socklen_t)sizeof(addr);
467 		memset((void *)&addr, 0, sizeof(addr));
468 		if (getsockname(b->num, &addr.sa, &addr_len) < 0)
469 			{
470 			ret = 0;
471 			break;
472 			}
473 		sockopt_len = sizeof(sockopt_val);
474 		switch (addr.sa.sa_family)
475 			{
476 		case AF_INET:
477 			sockopt_val = IP_PMTUDISC_DO;
478 			if ((ret = setsockopt(b->num, IPPROTO_IP, IP_MTU_DISCOVER,
479 				&sockopt_val, sizeof(sockopt_val))) < 0)
480 				perror("setsockopt");
481 			break;
482 #if OPENSSL_USE_IPV6 && defined(IPV6_MTU_DISCOVER)
483 		case AF_INET6:
484 			sockopt_val = IPV6_PMTUDISC_DO;
485 			if ((ret = setsockopt(b->num, IPPROTO_IPV6, IPV6_MTU_DISCOVER,
486 				&sockopt_val, sizeof(sockopt_val))) < 0)
487 				perror("setsockopt");
488 			break;
489 #endif
490 		default:
491 			ret = -1;
492 			break;
493 			}
494 		ret = -1;
495 #else
496 		break;
497 #endif
498 	case BIO_CTRL_DGRAM_QUERY_MTU:
499 #ifdef OPENSSL_SYS_LINUX
500 		addr_len = (socklen_t)sizeof(addr);
501 		memset((void *)&addr, 0, sizeof(addr));
502 		if (getsockname(b->num, &addr.sa, &addr_len) < 0)
503 			{
504 			ret = 0;
505 			break;
506 			}
507 		sockopt_len = sizeof(sockopt_val);
508 		switch (addr.sa.sa_family)
509 			{
510 		case AF_INET:
511 			if ((ret = getsockopt(b->num, IPPROTO_IP, IP_MTU, (void *)&sockopt_val,
512 				&sockopt_len)) < 0 || sockopt_val < 0)
513 				{
514 				ret = 0;
515 				}
516 			else
517 				{
518 				/* we assume that the transport protocol is UDP and no
519 				 * IP options are used.
520 				 */
521 				data->mtu = sockopt_val - 8 - 20;
522 				ret = data->mtu;
523 				}
524 			break;
525 #if OPENSSL_USE_IPV6 && defined(IPV6_MTU)
526 		case AF_INET6:
527 			if ((ret = getsockopt(b->num, IPPROTO_IPV6, IPV6_MTU, (void *)&sockopt_val,
528 				&sockopt_len)) < 0 || sockopt_val < 0)
529 				{
530 				ret = 0;
531 				}
532 			else
533 				{
534 				/* we assume that the transport protocol is UDP and no
535 				 * IPV6 options are used.
536 				 */
537 				data->mtu = sockopt_val - 8 - 40;
538 				ret = data->mtu;
539 				}
540 			break;
541 #endif
542 		default:
543 			ret = 0;
544 			break;
545 			}
546 #else
547 		ret = 0;
548 #endif
549 		break;
550 	case BIO_CTRL_DGRAM_GET_MTU:
551 		return data->mtu;
552 		break;
553 	case BIO_CTRL_DGRAM_SET_MTU:
554 		data->mtu = num;
555 		ret = num;
556 		break;
557 	case BIO_CTRL_DGRAM_SET_CONNECTED:
558 		to = (struct sockaddr *)ptr;
559 
560 		if ( to != NULL)
561 			{
562 			data->connected = 1;
563 			switch (to->sa_family)
564 				{
565 				case AF_INET:
566 					memcpy(&data->peer,to,sizeof(data->peer.sa_in));
567 					break;
568 #if OPENSSL_USE_IPV6
569 				case AF_INET6:
570 					memcpy(&data->peer,to,sizeof(data->peer.sa_in6));
571 					break;
572 #endif
573 				default:
574 					memcpy(&data->peer,to,sizeof(data->peer.sa));
575 					break;
576 				}
577 			}
578 		else
579 			{
580 			data->connected = 0;
581 			memset(&(data->peer), 0x00, sizeof(data->peer));
582 			}
583 		break;
584 	case BIO_CTRL_DGRAM_GET_PEER:
585 		switch (data->peer.sa.sa_family)
586 			{
587 			case AF_INET:
588 				ret=sizeof(data->peer.sa_in);
589 				break;
590 #if OPENSSL_USE_IPV6
591 			case AF_INET6:
592 				ret=sizeof(data->peer.sa_in6);
593 				break;
594 #endif
595 			default:
596 				ret=sizeof(data->peer.sa);
597 				break;
598 			}
599 		if (num==0 || num>ret)
600 			num=ret;
601 		memcpy(ptr,&data->peer,(ret=num));
602 		break;
603 	case BIO_CTRL_DGRAM_SET_PEER:
604 		to = (struct sockaddr *) ptr;
605 		switch (to->sa_family)
606 			{
607 			case AF_INET:
608 				memcpy(&data->peer,to,sizeof(data->peer.sa_in));
609 				break;
610 #if OPENSSL_USE_IPV6
611 			case AF_INET6:
612 				memcpy(&data->peer,to,sizeof(data->peer.sa_in6));
613 				break;
614 #endif
615 			default:
616 				memcpy(&data->peer,to,sizeof(data->peer.sa));
617 				break;
618 			}
619 		break;
620 	case BIO_CTRL_DGRAM_SET_NEXT_TIMEOUT:
621 		memcpy(&(data->next_timeout), ptr, sizeof(struct timeval));
622 		break;
623 #if defined(SO_RCVTIMEO)
624 	case BIO_CTRL_DGRAM_SET_RECV_TIMEOUT:
625 #ifdef OPENSSL_SYS_WINDOWS
626 		{
627 		struct timeval *tv = (struct timeval *)ptr;
628 		int timeout = tv->tv_sec * 1000 + tv->tv_usec/1000;
629 		if (setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
630 			(void*)&timeout, sizeof(timeout)) < 0)
631 			{ perror("setsockopt"); ret = -1; }
632 		}
633 #else
634 		if ( setsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO, ptr,
635 			sizeof(struct timeval)) < 0)
636 			{ perror("setsockopt");	ret = -1; }
637 #endif
638 		break;
639 	case BIO_CTRL_DGRAM_GET_RECV_TIMEOUT:
640 #ifdef OPENSSL_SYS_WINDOWS
641 		{
642 		int timeout, sz = sizeof(timeout);
643 		struct timeval *tv = (struct timeval *)ptr;
644 		if (getsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
645 			(void*)&timeout, &sz) < 0)
646 			{ perror("getsockopt"); ret = -1; }
647 		else
648 			{
649 			tv->tv_sec = timeout / 1000;
650 			tv->tv_usec = (timeout % 1000) * 1000;
651 			ret = sizeof(*tv);
652 			}
653 		}
654 #else
655 		if ( getsockopt(b->num, SOL_SOCKET, SO_RCVTIMEO,
656 			ptr, (void *)&ret) < 0)
657 			{ perror("getsockopt"); ret = -1; }
658 #endif
659 		break;
660 #endif
661 #if defined(SO_SNDTIMEO)
662 	case BIO_CTRL_DGRAM_SET_SEND_TIMEOUT:
663 #ifdef OPENSSL_SYS_WINDOWS
664 		{
665 		struct timeval *tv = (struct timeval *)ptr;
666 		int timeout = tv->tv_sec * 1000 + tv->tv_usec/1000;
667 		if (setsockopt(b->num, SOL_SOCKET, SO_SNDTIMEO,
668 			(void*)&timeout, sizeof(timeout)) < 0)
669 			{ perror("setsockopt"); ret = -1; }
670 		}
671 #else
672 		if ( setsockopt(b->num, SOL_SOCKET, SO_SNDTIMEO, ptr,
673 			sizeof(struct timeval)) < 0)
674 			{ perror("setsockopt");	ret = -1; }
675 #endif
676 		break;
677 	case BIO_CTRL_DGRAM_GET_SEND_TIMEOUT:
678 #ifdef OPENSSL_SYS_WINDOWS
679 		{
680 		int timeout, sz = sizeof(timeout);
681 		struct timeval *tv = (struct timeval *)ptr;
682 		if (getsockopt(b->num, SOL_SOCKET, SO_SNDTIMEO,
683 			(void*)&timeout, &sz) < 0)
684 			{ perror("getsockopt"); ret = -1; }
685 		else
686 			{
687 			tv->tv_sec = timeout / 1000;
688 			tv->tv_usec = (timeout % 1000) * 1000;
689 			ret = sizeof(*tv);
690 			}
691 		}
692 #else
693 		if ( getsockopt(b->num, SOL_SOCKET, SO_SNDTIMEO,
694 			ptr, (void *)&ret) < 0)
695 			{ perror("getsockopt"); ret = -1; }
696 #endif
697 		break;
698 #endif
699 	case BIO_CTRL_DGRAM_GET_SEND_TIMER_EXP:
700 		/* fall-through */
701 	case BIO_CTRL_DGRAM_GET_RECV_TIMER_EXP:
702 #ifdef OPENSSL_SYS_WINDOWS
703 		if ( data->_errno == WSAETIMEDOUT)
704 #else
705 		if ( data->_errno == EAGAIN)
706 #endif
707 			{
708 			ret = 1;
709 			data->_errno = 0;
710 			}
711 		else
712 			ret = 0;
713 		break;
714 #ifdef EMSGSIZE
715 	case BIO_CTRL_DGRAM_MTU_EXCEEDED:
716 		if ( data->_errno == EMSGSIZE)
717 			{
718 			ret = 1;
719 			data->_errno = 0;
720 			}
721 		else
722 			ret = 0;
723 		break;
724 #endif
725 	default:
726 		ret=0;
727 		break;
728 		}
729 	return(ret);
730 	}
731 
dgram_puts(BIO * bp,const char * str)732 static int dgram_puts(BIO *bp, const char *str)
733 	{
734 	int n,ret;
735 
736 	n=strlen(str);
737 	ret=dgram_write(bp,str,n);
738 	return(ret);
739 	}
740 
BIO_dgram_should_retry(int i)741 static int BIO_dgram_should_retry(int i)
742 	{
743 	int err;
744 
745 	if ((i == 0) || (i == -1))
746 		{
747 		err=get_last_socket_error();
748 
749 #if defined(OPENSSL_SYS_WINDOWS)
750 	/* If the socket return value (i) is -1
751 	 * and err is unexpectedly 0 at this point,
752 	 * the error code was overwritten by
753 	 * another system call before this error
754 	 * handling is called.
755 	 */
756 #endif
757 
758 		return(BIO_dgram_non_fatal_error(err));
759 		}
760 	return(0);
761 	}
762 
BIO_dgram_non_fatal_error(int err)763 int BIO_dgram_non_fatal_error(int err)
764 	{
765 	switch (err)
766 		{
767 #if defined(OPENSSL_SYS_WINDOWS)
768 # if defined(WSAEWOULDBLOCK)
769 	case WSAEWOULDBLOCK:
770 # endif
771 
772 # if 0 /* This appears to always be an error */
773 #  if defined(WSAENOTCONN)
774 	case WSAENOTCONN:
775 #  endif
776 # endif
777 #endif
778 
779 #ifdef EWOULDBLOCK
780 # ifdef WSAEWOULDBLOCK
781 #  if WSAEWOULDBLOCK != EWOULDBLOCK
782 	case EWOULDBLOCK:
783 #  endif
784 # else
785 	case EWOULDBLOCK:
786 # endif
787 #endif
788 
789 #ifdef EINTR
790 	case EINTR:
791 #endif
792 
793 #ifdef EAGAIN
794 #if EWOULDBLOCK != EAGAIN
795 	case EAGAIN:
796 # endif
797 #endif
798 
799 #ifdef EPROTO
800 	case EPROTO:
801 #endif
802 
803 #ifdef EINPROGRESS
804 	case EINPROGRESS:
805 #endif
806 
807 #ifdef EALREADY
808 	case EALREADY:
809 #endif
810 
811 		return(1);
812 		/* break; */
813 	default:
814 		break;
815 		}
816 	return(0);
817 	}
818 
get_current_time(struct timeval * t)819 static void get_current_time(struct timeval *t)
820 	{
821 #ifdef OPENSSL_SYS_WIN32
822 	struct _timeb tb;
823 	_ftime(&tb);
824 	t->tv_sec = (long)tb.time;
825 	t->tv_usec = (long)tb.millitm * 1000;
826 #elif defined(OPENSSL_SYS_VMS)
827 	struct timeb tb;
828 	ftime(&tb);
829 	t->tv_sec = (long)tb.time;
830 	t->tv_usec = (long)tb.millitm * 1000;
831 #else
832 	gettimeofday(t, NULL);
833 #endif
834 	}
835 
836 #endif
837