crypto/bio/bss_sock.c

   1 /*
   2  * Copyright 1995-2016 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the OpenSSL license (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 <errno.h>
  12 #include "bio_lcl.h"
  13 #include "internal/cryptlib.h"
  14
  15 #ifndef OPENSSL_NO_SOCK
  16
  17 # include <openssl/bio.h>
  18
  19 # ifdef WATT32
  20 /* Watt-32 uses same names */
  21 #  undef sock_write
  22 #  undef sock_read
  23 #  undef sock_puts
  24 #  define sock_write SockWrite
  25 #  define sock_read  SockRead
  26 #  define sock_puts  SockPuts
  27 # endif
  28
  29 static int sock_write(BIO *h, const char *buf, int num);
  30 static int sock_read(BIO *h, char *buf, int size);
  31 static int sock_puts(BIO *h, const char *str);
  32 static long sock_ctrl(BIO *h, int cmd, long arg1, void *arg2);
  33 static int sock_new(BIO *h);
  34 static int sock_free(BIO *data);
  35 int BIO_sock_should_retry(int s);
  36
  37 static const BIO_METHOD methods_sockp = {
  38     BIO_TYPE_SOCKET,
  39     "socket",
  40     /* TODO: Convert to new style write function */
  41     bwrite_conv,
  42     sock_write,
  43     /* TODO: Convert to new style read function */
  44     bread_conv,
  45     sock_read,
  46     sock_puts,
  47     NULL,                       /* sock_gets, */
  48     sock_ctrl,
  49     sock_new,
  50     sock_free,
  51     NULL,
  52 };
  53
  54 const BIO_METHOD *BIO_s_socket(void)
  55 {
  56     return &methods_sockp;
  57 }
  58
  59 BIO *BIO_new_socket(int fd, int close_flag)
  60 {
  61     BIO *ret;
  62
  63     ret = BIO_new(BIO_s_socket());
  64     if (ret == NULL)
  65         return NULL;
  66     BIO_set_fd(ret, fd, close_flag);
  67     return ret;
  68 }
  69
  70 static int sock_new(BIO *bi)
  71 {
  72     bi->init = 0;
  73     bi->num = 0;
  74     bi->ptr = NULL;
  75     bi->flags = 0;
  76     return 1;
  77 }
  78
  79 static int sock_free(BIO *a)
  80 {
  81     if (a == NULL)
  82         return 0;
  83     if (a->shutdown) {
  84         if (a->init) {
  85             BIO_closesocket(a->num);
  86         }
  87         a->init = 0;
  88         a->flags = 0;
  89     }
  90     return 1;
  91 }
  92
  93 static int sock_read(BIO *b, char *out, int outl)
  94 {
  95     int ret = 0;
  96
  97     if (out != NULL) {
  98         clear_socket_error();
  99         ret = readsocket(b->num, out, outl);
 100         BIO_clear_retry_flags(b);
 101         if (ret <= 0) {
 102             if (BIO_sock_should_retry(ret))
 103                 BIO_set_retry_read(b);
 104         }
 105     }
 106     return ret;
 107 }
 108
 109 static int sock_write(BIO *b, const char *in, int inl)
 110 {
 111     int ret;
 112
 113     clear_socket_error();
 114     ret = writesocket(b->num, in, inl);
 115     BIO_clear_retry_flags(b);
 116     if (ret <= 0) {
 117         if (BIO_sock_should_retry(ret))
 118             BIO_set_retry_write(b);
 119     }
 120     return ret;
 121 }
 122
 123 static long sock_ctrl(BIO *b, int cmd, long num, void *ptr)
 124 {
 125     long ret = 1;
 126     int *ip;
 127
 128     switch (cmd) {
 129     case BIO_C_SET_FD:
 130         sock_free(b);
 131         b->num = *((int *)ptr);
 132         b->shutdown = (int)num;
 133         b->init = 1;
 134         break;
 135     case BIO_C_GET_FD:
 136         if (b->init) {
 137             ip = (int *)ptr;
 138             if (ip != NULL)
 139                 *ip = b->num;
 140             ret = b->num;
 141         } else
 142             ret = -1;
 143         break;
 144     case BIO_CTRL_GET_CLOSE:
 145         ret = b->shutdown;
 146         break;
 147     case BIO_CTRL_SET_CLOSE:
 148         b->shutdown = (int)num;
 149         break;
 150     case BIO_CTRL_DUP:
 151     case BIO_CTRL_FLUSH:
 152         ret = 1;
 153         break;
 154     default:
 155         ret = 0;
 156         break;
 157     }
 158     return ret;
 159 }
 160
 161 static int sock_puts(BIO *bp, const char *str)
 162 {
 163     int n, ret;
 164
 165     n = strlen(str);
 166     ret = sock_write(bp, str, n);
 167     return ret;
 168 }
 169
 170 int BIO_sock_should_retry(int i)
 171 {
 172     int err;
 173
 174     if ((i == 0) || (i == -1)) {
 175         err = get_last_socket_error();
 176
 177         return BIO_sock_non_fatal_error(err);
 178     }
 179     return 0;
 180 }
 181
 182 int BIO_sock_non_fatal_error(int err)
 183 {
 184     switch (err) {
 185 # if defined(OPENSSL_SYS_WINDOWS)
 186 #  if defined(WSAEWOULDBLOCK)
 187     case WSAEWOULDBLOCK:
 188 #  endif
 189 # endif
 190
 191 # ifdef EWOULDBLOCK
 192 #  ifdef WSAEWOULDBLOCK
 193 #   if WSAEWOULDBLOCK != EWOULDBLOCK
 194     case EWOULDBLOCK:
 195 #   endif
 196 #  else
 197     case EWOULDBLOCK:
 198 #  endif
 199 # endif
 200
 201 # if defined(ENOTCONN)
 202     case ENOTCONN:
 203 # endif
 204
 205 # ifdef EINTR
 206     case EINTR:
 207 # endif
 208
 209 # ifdef EAGAIN
 210 #  if EWOULDBLOCK != EAGAIN
 211     case EAGAIN:
 212 #  endif
 213 # endif
 214
 215 # ifdef EPROTO
 216     case EPROTO:
 217 # endif
 218
 219 # ifdef EINPROGRESS
 220     case EINPROGRESS:
 221 # endif
 222
 223 # ifdef EALREADY
 224     case EALREADY:
 225 # endif
 226         return 1;
 227     default:
 228         break;
 229     }
 230     return 0;
 231 }
 232
 233 #endif                          /* #ifndef OPENSSL_NO_SOCK */