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 #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
 125 BIO_METHOD *BIO_s_datagram(void)
 126         {
 127         return(&methods_dgramp);
 128         }
 129
 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
 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
 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
 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
 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
 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
 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
 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
 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
 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
 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
 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
 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