crypto/bio/b_sock.c

   1 /*
   2  * Copyright 1995-2018 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the Apache License 2.0 (the "License").  You may not use
   5  * this file except in compliance with the License.  You can obtain a copy
   6  * in the file LICENSE in the source distribution or at
   7  * https://www.openssl.org/source/license.html
   8  */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <errno.h>
  13 #include "bio_local.h"
  14 #ifndef OPENSSL_NO_SOCK
  15 # define SOCKET_PROTOCOL IPPROTO_TCP
  16 # ifdef SO_MAXCONN
  17 #  define MAX_LISTEN  SO_MAXCONN
  18 # elif defined(SOMAXCONN)
  19 #  define MAX_LISTEN  SOMAXCONN
  20 # else
  21 #  define MAX_LISTEN  32
  22 # endif
  23 # if defined(OPENSSL_SYS_WINDOWS)
  24 static int wsa_init_done = 0;
  25 # endif
  26
  27 # ifndef OPENSSL_NO_DEPRECATED_1_1_0
  28 int BIO_get_host_ip(const char *str, unsigned char *ip)
  29 {
  30     BIO_ADDRINFO *res = NULL;
  31     int ret = 0;
  32
  33     if (BIO_sock_init() != 1)
  34         return 0;               /* don't generate another error code here */
  35
  36     if (BIO_lookup(str, NULL, BIO_LOOKUP_CLIENT, AF_INET, SOCK_STREAM, &res)) {
  37         size_t l;
  38
  39         if (BIO_ADDRINFO_family(res) != AF_INET) {
  40             BIOerr(BIO_F_BIO_GET_HOST_IP,
  41                    BIO_R_GETHOSTBYNAME_ADDR_IS_NOT_AF_INET);
  42         } else if (BIO_ADDR_rawaddress(BIO_ADDRINFO_address(res), NULL, &l)) {
  43             /*
  44              * Because only AF_INET addresses will reach this far, we can assert
  45              * that l should be 4
  46              */
  47             if (ossl_assert(l == 4))
  48                 ret = BIO_ADDR_rawaddress(BIO_ADDRINFO_address(res), ip, &l);
  49         }
  50         BIO_ADDRINFO_free(res);
  51     } else {
  52         ERR_add_error_data(2, "host=", str);
  53     }
  54
  55     return ret;
  56 }
  57
  58 int BIO_get_port(const char *str, unsigned short *port_ptr)
  59 {
  60     BIO_ADDRINFO *res = NULL;
  61     int ret = 0;
  62
  63     if (str == NULL) {
  64         BIOerr(BIO_F_BIO_GET_PORT, BIO_R_NO_PORT_DEFINED);
  65         return 0;
  66     }
  67
  68     if (BIO_sock_init() != 1)
  69         return 0;               /* don't generate another error code here */
  70
  71     if (BIO_lookup(NULL, str, BIO_LOOKUP_CLIENT, AF_INET, SOCK_STREAM, &res)) {
  72         if (BIO_ADDRINFO_family(res) != AF_INET) {
  73             BIOerr(BIO_F_BIO_GET_PORT,
  74                    BIO_R_ADDRINFO_ADDR_IS_NOT_AF_INET);
  75         } else {
  76             *port_ptr = ntohs(BIO_ADDR_rawport(BIO_ADDRINFO_address(res)));
  77             ret = 1;
  78         }
  79         BIO_ADDRINFO_free(res);
  80     } else {
  81         ERR_add_error_data(2, "host=", str);
  82     }
  83
  84     return ret;
  85 }
  86 # endif
  87
  88 int BIO_sock_error(int sock)
  89 {
  90     int j = 0, i;
  91     socklen_t size = sizeof(j);
  92
  93     /*
  94      * Note: under Windows the third parameter is of type (char *) whereas
  95      * under other systems it is (void *) if you don't have a cast it will
  96      * choke the compiler: if you do have a cast then you can either go for
  97      * (char *) or (void *).
  98      */
  99     i = getsockopt(sock, SOL_SOCKET, SO_ERROR, (void *)&j, &size);
 100     if (i < 0)
 101         return get_last_socket_error();
 102     else
 103         return j;
 104 }
 105
 106 # ifndef OPENSSL_NO_DEPRECATED_1_1_0
 107 struct hostent *BIO_gethostbyname(const char *name)
 108 {
 109     /*
 110      * Caching gethostbyname() results forever is wrong, so we have to let
 111      * the true gethostbyname() worry about this
 112      */
 113     return gethostbyname(name);
 114 }
 115 # endif
 116
 117 int BIO_sock_init(void)
 118 {
 119 # ifdef OPENSSL_SYS_WINDOWS
 120     static struct WSAData wsa_state;
 121
 122     if (!wsa_init_done) {
 123         wsa_init_done = 1;
 124         memset(&wsa_state, 0, sizeof(wsa_state));
 125         /*
 126          * Not making wsa_state available to the rest of the code is formally
 127          * wrong. But the structures we use are [believed to be] invariable
 128          * among Winsock DLLs, while API availability is [expected to be]
 129          * probed at run-time with DSO_global_lookup.
 130          */
 131         if (WSAStartup(0x0202, &wsa_state) != 0) {
 132             ERR_raise_data(ERR_LIB_SYS, get_last_socket_error(),
 133                            "calling wsastartup()");
 134             BIOerr(BIO_F_BIO_SOCK_INIT, BIO_R_WSASTARTUP);
 135             return -1;
 136         }
 137     }
 138 # endif                         /* OPENSSL_SYS_WINDOWS */
 139 # ifdef WATT32
 140     extern int _watt_do_exit;
 141     _watt_do_exit = 0;          /* don't make sock_init() call exit() */
 142     if (sock_init())
 143         return -1;
 144 # endif
 145
 146     return 1;
 147 }
 148
 149 void bio_sock_cleanup_int(void)
 150 {
 151 # ifdef OPENSSL_SYS_WINDOWS
 152     if (wsa_init_done) {
 153         wsa_init_done = 0;
 154         WSACleanup();
 155     }
 156 # endif
 157 }
 158
 159 int BIO_socket_ioctl(int fd, long type, void *arg)
 160 {
 161     int i;
 162
 163 #  ifdef __DJGPP__
 164     i = ioctlsocket(fd, type, (char *)arg);
 165 #  else
 166 #   if defined(OPENSSL_SYS_VMS)
 167     /*-
 168      * 2011-02-18 SMS.
 169      * VMS ioctl() can't tolerate a 64-bit "void *arg", but we
 170      * observe that all the consumers pass in an "unsigned long *",
 171      * so we arrange a local copy with a short pointer, and use
 172      * that, instead.
 173      */
 174 #    if __INITIAL_POINTER_SIZE == 64
 175 #     define ARG arg_32p
 176 #     pragma pointer_size save
 177 #     pragma pointer_size 32
 178     unsigned long arg_32;
 179     unsigned long *arg_32p;
 180 #     pragma pointer_size restore
 181     arg_32p = &arg_32;
 182     arg_32 = *((unsigned long *)arg);
 183 #    else                       /* __INITIAL_POINTER_SIZE == 64 */
 184 #     define ARG arg
 185 #    endif                      /* __INITIAL_POINTER_SIZE == 64 [else] */
 186 #   else                        /* defined(OPENSSL_SYS_VMS) */
 187 #    define ARG arg
 188 #   endif                       /* defined(OPENSSL_SYS_VMS) [else] */
 189
 190     i = ioctlsocket(fd, type, ARG);
 191 #  endif                        /* __DJGPP__ */
 192     if (i < 0)
 193         ERR_raise_data(ERR_LIB_SYS, get_last_socket_error(),
 194                        "calling ioctlsocket()");
 195     return i;
 196 }
 197
 198 # ifndef OPENSSL_NO_DEPRECATED_1_1_0
 199 int BIO_get_accept_socket(char *host, int bind_mode)
 200 {
 201     int s = INVALID_SOCKET;
 202     char *h = NULL, *p = NULL;
 203     BIO_ADDRINFO *res = NULL;
 204
 205     if (!BIO_parse_hostserv(host, &h, &p, BIO_PARSE_PRIO_SERV))
 206         return INVALID_SOCKET;
 207
 208     if (BIO_sock_init() != 1)
 209         return INVALID_SOCKET;
 210
 211     if (BIO_lookup(h, p, BIO_LOOKUP_SERVER, AF_UNSPEC, SOCK_STREAM, &res) != 0)
 212         goto err;
 213
 214     if ((s = BIO_socket(BIO_ADDRINFO_family(res), BIO_ADDRINFO_socktype(res),
 215                         BIO_ADDRINFO_protocol(res), 0)) == INVALID_SOCKET) {
 216         s = INVALID_SOCKET;
 217         goto err;
 218     }
 219
 220     if (!BIO_listen(s, BIO_ADDRINFO_address(res),
 221                     bind_mode ? BIO_SOCK_REUSEADDR : 0)) {
 222         BIO_closesocket(s);
 223         s = INVALID_SOCKET;
 224     }
 225
 226  err:
 227     BIO_ADDRINFO_free(res);
 228     OPENSSL_free(h);
 229     OPENSSL_free(p);
 230
 231     return s;
 232 }
 233
 234 int BIO_accept(int sock, char **ip_port)
 235 {
 236     BIO_ADDR res;
 237     int ret = -1;
 238
 239     ret = BIO_accept_ex(sock, &res, 0);
 240     if (ret == (int)INVALID_SOCKET) {
 241         if (BIO_sock_should_retry(ret)) {
 242             ret = -2;
 243             goto end;
 244         }
 245         ERR_raise_data(ERR_LIB_SYS, get_last_socket_error(),
 246                        "calling accept()");
 247         BIOerr(BIO_F_BIO_ACCEPT, BIO_R_ACCEPT_ERROR);
 248         goto end;
 249     }
 250
 251     if (ip_port != NULL) {
 252         char *host = BIO_ADDR_hostname_string(&res, 1);
 253         char *port = BIO_ADDR_service_string(&res, 1);
 254         if (host != NULL && port != NULL)
 255             *ip_port = OPENSSL_zalloc(strlen(host) + strlen(port) + 2);
 256         else
 257             *ip_port = NULL;
 258
 259         if (*ip_port == NULL) {
 260             BIOerr(BIO_F_BIO_ACCEPT, ERR_R_MALLOC_FAILURE);
 261             BIO_closesocket(ret);
 262             ret = (int)INVALID_SOCKET;
 263         } else {
 264             strcpy(*ip_port, host);
 265             strcat(*ip_port, ":");
 266             strcat(*ip_port, port);
 267         }
 268         OPENSSL_free(host);
 269         OPENSSL_free(port);
 270     }
 271
 272  end:
 273     return ret;
 274 }
 275 # endif
 276
 277 int BIO_set_tcp_ndelay(int s, int on)
 278 {
 279     int ret = 0;
 280 # if defined(TCP_NODELAY) && (defined(IPPROTO_TCP) || defined(SOL_TCP))
 281     int opt;
 282
 283 #  ifdef SOL_TCP
 284     opt = SOL_TCP;
 285 #  else
 286 #   ifdef IPPROTO_TCP
 287     opt = IPPROTO_TCP;
 288 #   endif
 289 #  endif
 290
 291     ret = setsockopt(s, opt, TCP_NODELAY, (char *)&on, sizeof(on));
 292 # endif
 293     return (ret == 0);
 294 }
 295
 296 int BIO_socket_nbio(int s, int mode)
 297 {
 298     int ret = -1;
 299     int l;
 300
 301     l = mode;
 302 # ifdef FIONBIO
 303     l = mode;
 304
 305     ret = BIO_socket_ioctl(s, FIONBIO, &l);
 306 # elif defined(F_GETFL) && defined(F_SETFL) && (defined(O_NONBLOCK) || defined(FNDELAY))
 307     /* make sure this call always pushes an error level; BIO_socket_ioctl() does so, so we do too. */
 308
 309     l = fcntl(s, F_GETFL, 0);
 310     if (l == -1) {
 311         ERR_raise_data(ERR_LIB_SYS, get_last_sys_error(),
 312                        "calling fcntl()");
 313         ret = -1;
 314     } else {
 315 #  if defined(O_NONBLOCK)
 316         l &= ~O_NONBLOCK;
 317 #  else
 318         l &= ~FNDELAY; /* BSD4.x */
 319 #  endif
 320         if (mode) {
 321 #  if defined(O_NONBLOCK)
 322             l |= O_NONBLOCK;
 323 #  else
 324             l |= FNDELAY; /* BSD4.x */
 325 #  endif
 326         }
 327         ret = fcntl(s, F_SETFL, l);
 328
 329         if (ret < 0) {
 330             ERR_raise_data(ERR_LIB_SYS, get_last_sys_error(),
 331                            "calling fcntl()");
 332         }
 333     }
 334 # else
 335     /* make sure this call always pushes an error level; BIO_socket_ioctl() does so, so we do too. */
 336     BIOerr(BIO_F_BIO_SOCKET_NBIO, ERR_R_PASSED_INVALID_ARGUMENT);
 337 # endif
 338
 339     return (ret == 0);
 340 }
 341
 342 int BIO_sock_info(int sock,
 343                   enum BIO_sock_info_type type, union BIO_sock_info_u *info)
 344 {
 345     switch (type) {
 346     case BIO_SOCK_INFO_ADDRESS:
 347         {
 348             socklen_t addr_len;
 349             int ret = 0;
 350             addr_len = sizeof(*info->addr);
 351             ret = getsockname(sock, BIO_ADDR_sockaddr_noconst(info->addr),
 352                               &addr_len);
 353             if (ret == -1) {
 354                 ERR_raise_data(ERR_LIB_SYS, get_last_socket_error(),
 355                                "calling getsockname()");
 356                 BIOerr(BIO_F_BIO_SOCK_INFO, BIO_R_GETSOCKNAME_ERROR);
 357                 return 0;
 358             }
 359             if ((size_t)addr_len > sizeof(*info->addr)) {
 360                 BIOerr(BIO_F_BIO_SOCK_INFO, BIO_R_GETSOCKNAME_TRUNCATED_ADDRESS);
 361                 return 0;
 362             }
 363         }
 364         break;
 365     default:
 366         BIOerr(BIO_F_BIO_SOCK_INFO, BIO_R_UNKNOWN_INFO_TYPE);
 367         return 0;
 368     }
 369     return 1;
 370 }
 371
 372 #endif