2c1c405ec7e8f5ffe1182e33168a3fb068197075
[openssl.git] / crypto / bio / bss_sock.c
1 /* crypto/bio/bss_sock.c */
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 #ifndef OPENSSL_NO_SOCK
60
61 #include <stdio.h>
62 #include <errno.h>
63 #define USE_SOCKETS
64 #include "cryptlib.h"
65 #include <openssl/bio.h>
66
67 #ifdef WATT32
68 #define sock_write SockWrite  /* Watt-32 uses same names */
69 #define sock_read  SockRead
70 #define sock_puts  SockPuts
71 #endif
72
73 static int sock_write(BIO *h, const char *buf, int num);
74 static int sock_read(BIO *h, char *buf, int size);
75 static int sock_puts(BIO *h, const char *str);
76 static long sock_ctrl(BIO *h, int cmd, long arg1, void *arg2);
77 static int sock_new(BIO *h);
78 static int sock_free(BIO *data);
79 int BIO_sock_should_retry(int s);
80
81 static BIO_METHOD methods_sockp=
82         {
83         BIO_TYPE_SOCKET,
84         "socket",
85         sock_write,
86         sock_read,
87         sock_puts,
88         NULL, /* sock_gets, */
89         sock_ctrl,
90         sock_new,
91         sock_free,
92         NULL,
93         };
94
95 BIO_METHOD *BIO_s_socket(void)
96         {
97         return(&methods_sockp);
98         }
99
100 BIO *BIO_new_socket(int fd, int close_flag)
101         {
102         BIO *ret;
103
104         ret=BIO_new(BIO_s_socket());
105         if (ret == NULL) return(NULL);
106         BIO_set_fd(ret,fd,close_flag);
107         return(ret);
108         }
109
110 static int sock_new(BIO *bi)
111         {
112         bi->init=0;
113         bi->num=0;
114         bi->ptr=NULL;
115         bi->flags=0;
116         return(1);
117         }
118
119 static int sock_free(BIO *a)
120         {
121         if (a == NULL) return(0);
122         if (a->shutdown)
123                 {
124                 if (a->init)
125                         {
126                         SHUTDOWN2(a->num);
127                         }
128                 a->init=0;
129                 a->flags=0;
130                 }
131         return(1);
132         }
133         
134 static int sock_read(BIO *b, char *out, int outl)
135         {
136         int ret=0;
137
138         if (out != NULL)
139                 {
140                 clear_socket_error();
141                 ret=readsocket(b->num,out,outl);
142                 BIO_clear_retry_flags(b);
143                 if (ret <= 0)
144                         {
145                         if (BIO_sock_should_retry(ret))
146                                 BIO_set_retry_read(b);
147                         }
148                 }
149         return(ret);
150         }
151
152 static int sock_write(BIO *b, const char *in, int inl)
153         {
154         int ret;
155         
156         clear_socket_error();
157         ret=writesocket(b->num,in,inl);
158         BIO_clear_retry_flags(b);
159         if (ret <= 0)
160                 {
161                 if (BIO_sock_should_retry(ret))
162                         BIO_set_retry_write(b);
163                 }
164         return(ret);
165         }
166
167 static long sock_ctrl(BIO *b, int cmd, long num, void *ptr)
168         {
169         long ret=1;
170         int *ip;
171
172         switch (cmd)
173                 {
174         case BIO_CTRL_RESET:
175                 num=0;
176         case BIO_C_FILE_SEEK:
177                 ret=0;
178                 break;
179         case BIO_C_FILE_TELL:
180         case BIO_CTRL_INFO:
181                 ret=0;
182                 break;
183         case BIO_C_SET_FD:
184                 sock_free(b);
185                 b->num= *((int *)ptr);
186                 b->shutdown=(int)num;
187                 b->init=1;
188                 break;
189         case BIO_C_GET_FD:
190                 if (b->init)
191                         {
192                         ip=(int *)ptr;
193                         if (ip != NULL) *ip=b->num;
194                         ret=b->num;
195                         }
196                 else
197                         ret= -1;
198                 break;
199         case BIO_CTRL_GET_CLOSE:
200                 ret=b->shutdown;
201                 break;
202         case BIO_CTRL_SET_CLOSE:
203                 b->shutdown=(int)num;
204                 break;
205         case BIO_CTRL_PENDING:
206         case BIO_CTRL_WPENDING:
207                 ret=0;
208                 break;
209         case BIO_CTRL_DUP:
210         case BIO_CTRL_FLUSH:
211                 ret=1;
212                 break;
213         default:
214                 ret=0;
215                 break;
216                 }
217         return(ret);
218         }
219
220 static int sock_puts(BIO *bp, const char *str)
221         {
222         int n,ret;
223
224         n=strlen(str);
225         ret=sock_write(bp,str,n);
226         return(ret);
227         }
228
229 int BIO_sock_should_retry(int i)
230         {
231         int err;
232
233         if ((i == 0) || (i == -1))
234                 {
235                 err=get_last_socket_error();
236
237 #if defined(OPENSSL_SYS_WINDOWS) && 0 /* more microsoft stupidity? perhaps not? Ben 4/1/99 */
238                 if ((i == -1) && (err == 0))
239                         return(1);
240 #endif
241
242                 return(BIO_sock_non_fatal_error(err));
243                 }
244         return(0);
245         }
246
247 int BIO_sock_non_fatal_error(int err)
248         {
249         switch (err)
250                 {
251 #if defined(OPENSSL_SYS_WINDOWS)
252 # if defined(WSAEWOULDBLOCK)
253         case WSAEWOULDBLOCK:
254 # endif
255
256 # if 0 /* This appears to always be an error */
257 #  if defined(WSAENOTCONN)
258         case WSAENOTCONN:
259 #  endif
260 # endif
261 #endif
262
263 #ifdef EWOULDBLOCK
264 # ifdef WSAEWOULDBLOCK
265 #  if WSAEWOULDBLOCK != EWOULDBLOCK
266         case EWOULDBLOCK:
267 #  endif
268 # else
269         case EWOULDBLOCK:
270 # endif
271 #endif
272
273 #if defined(ENOTCONN)
274         case ENOTCONN:
275 #endif
276
277 #ifdef EINTR
278         case EINTR:
279 #endif
280
281 #ifdef EAGAIN
282 #if EWOULDBLOCK != EAGAIN
283         case EAGAIN:
284 # endif
285 #endif
286
287 #ifdef EPROTO
288         case EPROTO:
289 #endif
290
291 #ifdef EINPROGRESS
292         case EINPROGRESS:
293 #endif
294
295 #ifdef EALREADY
296         case EALREADY:
297 #endif
298                 return(1);
299                 /* break; */
300         default:
301                 break;
302                 }
303         return(0);
304         }
305 #endif