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