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 */
308