lib/libcrypto/bio/bss_sock.c

   1 /* $OpenBSD: bss_sock.c,v 1.23 2014/07/11 08:44:47 jsing Exp $ */
   2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
   3  * All rights reserved.
   4  *
   5  * This package is an SSL implementation written
   6  * by Eric Young (eay@cryptsoft.com).
   7  * The implementation was written so as to conform with Netscapes SSL.
   8  *
   9  * This library is free for commercial and non-commercial use as long as
  10  * the following conditions are aheared to.  The following conditions
  11  * apply to all code found in this distribution, be it the RC4, RSA,
  12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
  13  * included with this distribution is covered by the same copyright terms
  14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  15  *
  16  * Copyright remains Eric Young's, and as such any Copyright notices in
  17  * the code are not to be removed.
  18  * If this package is used in a product, Eric Young should be given attribution
  19  * as the author of the parts of the library used.
  20  * This can be in the form of a textual message at program startup or
  21  * in documentation (online or textual) provided with the package.
  22  *
  23  * Redistribution and use in source and binary forms, with or without
  24  * modification, are permitted provided that the following conditions
  25  * are met:
  26  * 1. Redistributions of source code must retain the copyright
  27  *    notice, this list of conditions and the following disclaimer.
  28  * 2. Redistributions in binary form must reproduce the above copyright
  29  *    notice, this list of conditions and the following disclaimer in the
  30  *    documentation and/or other materials provided with the distribution.
  31  * 3. All advertising materials mentioning features or use of this software
  32  *    must display the following acknowledgement:
  33  *    "This product includes cryptographic software written by
  34  *     Eric Young (eay@cryptsoft.com)"
  35  *    The word 'cryptographic' can be left out if the rouines from the library
  36  *    being used are not cryptographic related :-).
  37  * 4. If you include any Windows specific code (or a derivative thereof) from
  38  *    the apps directory (application code) you must include an acknowledgement:
  39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  40  *
  41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  51  * SUCH DAMAGE.
  52  *
  53  * The licence and distribution terms for any publically available version or
  54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
  55  * copied and put under another distribution licence
  56  * [including the GNU Public Licence.]
  57  */
  58
  59 #include <sys/socket.h>
  60
  61 #include <errno.h>
  62 #include <stdio.h>
  63 #include <string.h>
  64 #include <unistd.h>
  65
  66 #include <openssl/bio.h>
  67
  68 static int sock_write(BIO *h, const char *buf, int num);
  69 static int sock_read(BIO *h, char *buf, int size);
  70 static int sock_puts(BIO *h, const char *str);
  71 static long sock_ctrl(BIO *h, int cmd, long arg1, void *arg2);
  72 static int sock_new(BIO *h);
  73 static int sock_free(BIO *data);
  74 int BIO_sock_should_retry(int s);
  75
  76 static BIO_METHOD methods_sockp = {
  77         .type = BIO_TYPE_SOCKET,
  78         .name = "socket",
  79         .bwrite = sock_write,
  80         .bread = sock_read,
  81         .bputs = sock_puts,
  82         .ctrl = sock_ctrl,
  83         .create = sock_new,
  84         .destroy = sock_free
  85 };
  86
  87 BIO_METHOD *
  88 BIO_s_socket(void)
  89 {
  90         return (&methods_sockp);
  91 }
  92
  93 BIO *
  94 BIO_new_socket(int fd, int close_flag)
  95 {
  96         BIO *ret;
  97
  98         ret = BIO_new(BIO_s_socket());
  99         if (ret == NULL)
 100                 return (NULL);
 101         BIO_set_fd(ret, fd, close_flag);
 102         return (ret);
 103 }
 104
 105 static int
 106 sock_new(BIO *bi)
 107 {
 108         bi->init = 0;
 109         bi->num = 0;
 110         bi->ptr = NULL;
 111         bi->flags = 0;
 112         return (1);
 113 }
 114
 115 static int
 116 sock_free(BIO *a)
 117 {
 118         if (a == NULL)
 119                 return (0);
 120         if (a->shutdown) {
 121                 if (a->init) {
 122                         shutdown(a->num, SHUT_RDWR);
 123                         close(a->num);
 124                 }
 125                 a->init = 0;
 126                 a->flags = 0;
 127         }
 128         return (1);
 129 }
 130
 131 static int
 132 sock_read(BIO *b, char *out, int outl)
 133 {
 134         int ret = 0;
 135
 136         if (out != NULL) {
 137                 errno = 0;
 138                 ret = read(b->num, out, outl);
 139                 BIO_clear_retry_flags(b);
 140                 if (ret <= 0) {
 141                         if (BIO_sock_should_retry(ret))
 142                                 BIO_set_retry_read(b);
 143                 }
 144         }
 145         return (ret);
 146 }
 147
 148 static int
 149 sock_write(BIO *b, const char *in, int inl)
 150 {
 151         int ret;
 152
 153         errno = 0;
 154         ret = write(b->num, in, inl);
 155         BIO_clear_retry_flags(b);
 156         if (ret <= 0) {
 157                 if (BIO_sock_should_retry(ret))
 158                         BIO_set_retry_write(b);
 159         }
 160         return (ret);
 161 }
 162
 163 static long
 164 sock_ctrl(BIO *b, int cmd, long num, void *ptr)
 165 {
 166         long ret = 1;
 167         int *ip;
 168
 169         switch (cmd) {
 170         case BIO_C_SET_FD:
 171                 sock_free(b);
 172                 b->num= *((int *)ptr);
 173                 b->shutdown = (int)num;
 174                 b->init = 1;
 175                 break;
 176         case BIO_C_GET_FD:
 177                 if (b->init) {
 178                         ip = (int *)ptr;
 179                         if (ip != NULL)
 180                                 *ip = b->num;
 181                         ret = b->num;
 182                 } else
 183                         ret = -1;
 184                 break;
 185         case BIO_CTRL_GET_CLOSE:
 186                 ret = b->shutdown;
 187                 break;
 188         case BIO_CTRL_SET_CLOSE:
 189                 b->shutdown = (int)num;
 190                 break;
 191         case BIO_CTRL_DUP:
 192         case BIO_CTRL_FLUSH:
 193                 ret = 1;
 194                 break;
 195         default:
 196                 ret = 0;
 197                 break;
 198         }
 199         return (ret);
 200 }
 201
 202 static int
 203 sock_puts(BIO *bp, const char *str)
 204 {
 205         int n, ret;
 206
 207         n = strlen(str);
 208         ret = sock_write(bp, str, n);
 209         return (ret);
 210 }
 211
 212 int
 213 BIO_sock_should_retry(int i)
 214 {
 215         int err;
 216
 217         if ((i == 0) || (i == -1)) {
 218                 err = errno;
 219                 return (BIO_sock_non_fatal_error(err));
 220         }
 221         return (0);
 222 }
 223
 224 int
 225 BIO_sock_non_fatal_error(int err)
 226 {
 227         switch (err) {
 228         case ENOTCONN:
 229         case EINTR:
 230         case EAGAIN:
 231         case EINPROGRESS:
 232         case EALREADY:
 233                 return (1);
 234         default:
 235                 break;
 236         }
 237         return (0);
 238 }
 239