]> git.ipfire.org Git - thirdparty/openssl.git/blob - crypto/bio/bss_acpt.c
Rename some BUF_xxx to OPENSSL_xxx
[thirdparty/openssl.git] / crypto / bio / bss_acpt.c
1 /* crypto/bio/bss_acpt.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 "internal/cryptlib.h"
63 #include <openssl/bio.h>
64
65 #ifndef OPENSSL_NO_SOCK
66
67 # if (defined(OPENSSL_SYS_VMS) && __VMS_VER < 70000000)
68 /* FIONBIO used as a switch to enable ioctl, and that isn't in VMS < 7.0 */
69 # undef FIONBIO
70 # endif
71
72 typedef struct bio_accept_st {
73 int state;
74 char *param_addr;
75 int accept_sock;
76 int accept_nbio;
77 char *addr;
78 int nbio;
79 /*
80 * If 0, it means normal, if 1, do a connect on bind failure, and if
81 * there is no-one listening, bind with SO_REUSEADDR. If 2, always use
82 * SO_REUSEADDR.
83 */
84 int bind_mode;
85 BIO *bio_chain;
86 } BIO_ACCEPT;
87
88 static int acpt_write(BIO *h, const char *buf, int num);
89 static int acpt_read(BIO *h, char *buf, int size);
90 static int acpt_puts(BIO *h, const char *str);
91 static long acpt_ctrl(BIO *h, int cmd, long arg1, void *arg2);
92 static int acpt_new(BIO *h);
93 static int acpt_free(BIO *data);
94 static int acpt_state(BIO *b, BIO_ACCEPT *c);
95 static void acpt_close_socket(BIO *data);
96 static BIO_ACCEPT *BIO_ACCEPT_new(void);
97 static void BIO_ACCEPT_free(BIO_ACCEPT *a);
98
99 # define ACPT_S_BEFORE 1
100 # define ACPT_S_GET_ACCEPT_SOCKET 2
101 # define ACPT_S_OK 3
102
103 static BIO_METHOD methods_acceptp = {
104 BIO_TYPE_ACCEPT,
105 "socket accept",
106 acpt_write,
107 acpt_read,
108 acpt_puts,
109 NULL, /* connect_gets, */
110 acpt_ctrl,
111 acpt_new,
112 acpt_free,
113 NULL,
114 };
115
116 BIO_METHOD *BIO_s_accept(void)
117 {
118 return (&methods_acceptp);
119 }
120
121 static int acpt_new(BIO *bi)
122 {
123 BIO_ACCEPT *ba;
124
125 bi->init = 0;
126 bi->num = (int)INVALID_SOCKET;
127 bi->flags = 0;
128 if ((ba = BIO_ACCEPT_new()) == NULL)
129 return (0);
130 bi->ptr = (char *)ba;
131 ba->state = ACPT_S_BEFORE;
132 bi->shutdown = 1;
133 return (1);
134 }
135
136 static BIO_ACCEPT *BIO_ACCEPT_new(void)
137 {
138 BIO_ACCEPT *ret;
139
140 if ((ret = OPENSSL_zalloc(sizeof(*ret))) == NULL)
141 return (NULL);
142 ret->accept_sock = (int)INVALID_SOCKET;
143 ret->bind_mode = BIO_BIND_NORMAL;
144 return (ret);
145 }
146
147 static void BIO_ACCEPT_free(BIO_ACCEPT *a)
148 {
149 if (a == NULL)
150 return;
151
152 OPENSSL_free(a->param_addr);
153 OPENSSL_free(a->addr);
154 BIO_free(a->bio_chain);
155 OPENSSL_free(a);
156 }
157
158 static void acpt_close_socket(BIO *bio)
159 {
160 BIO_ACCEPT *c;
161
162 c = (BIO_ACCEPT *)bio->ptr;
163 if (c->accept_sock != (int)INVALID_SOCKET) {
164 shutdown(c->accept_sock, 2);
165 closesocket(c->accept_sock);
166 c->accept_sock = (int)INVALID_SOCKET;
167 bio->num = (int)INVALID_SOCKET;
168 }
169 }
170
171 static int acpt_free(BIO *a)
172 {
173 BIO_ACCEPT *data;
174
175 if (a == NULL)
176 return (0);
177 data = (BIO_ACCEPT *)a->ptr;
178
179 if (a->shutdown) {
180 acpt_close_socket(a);
181 BIO_ACCEPT_free(data);
182 a->ptr = NULL;
183 a->flags = 0;
184 a->init = 0;
185 }
186 return (1);
187 }
188
189 static int acpt_state(BIO *b, BIO_ACCEPT *c)
190 {
191 BIO *bio = NULL, *dbio;
192 int s = -1;
193 int i;
194
195 again:
196 switch (c->state) {
197 case ACPT_S_BEFORE:
198 if (c->param_addr == NULL) {
199 BIOerr(BIO_F_ACPT_STATE, BIO_R_NO_ACCEPT_PORT_SPECIFIED);
200 return (-1);
201 }
202 s = BIO_get_accept_socket(c->param_addr, c->bind_mode);
203 if (s == (int)INVALID_SOCKET)
204 return (-1);
205
206 if (c->accept_nbio) {
207 if (!BIO_socket_nbio(s, 1)) {
208 closesocket(s);
209 BIOerr(BIO_F_ACPT_STATE,
210 BIO_R_ERROR_SETTING_NBIO_ON_ACCEPT_SOCKET);
211 return (-1);
212 }
213 }
214 c->accept_sock = s;
215 b->num = s;
216 c->state = ACPT_S_GET_ACCEPT_SOCKET;
217 return (1);
218 /* break; */
219 case ACPT_S_GET_ACCEPT_SOCKET:
220 if (b->next_bio != NULL) {
221 c->state = ACPT_S_OK;
222 goto again;
223 }
224 BIO_clear_retry_flags(b);
225 b->retry_reason = 0;
226 i = BIO_accept(c->accept_sock, &(c->addr));
227
228 /* -2 return means we should retry */
229 if (i == -2) {
230 BIO_set_retry_special(b);
231 b->retry_reason = BIO_RR_ACCEPT;
232 return -1;
233 }
234
235 if (i < 0)
236 return (i);
237
238 bio = BIO_new_socket(i, BIO_CLOSE);
239 if (bio == NULL)
240 goto err;
241
242 BIO_set_callback(bio, BIO_get_callback(b));
243 BIO_set_callback_arg(bio, BIO_get_callback_arg(b));
244
245 if (c->nbio) {
246 if (!BIO_socket_nbio(i, 1)) {
247 BIOerr(BIO_F_ACPT_STATE,
248 BIO_R_ERROR_SETTING_NBIO_ON_ACCEPTED_SOCKET);
249 goto err;
250 }
251 }
252
253 /*
254 * If the accept BIO has an bio_chain, we dup it and put the new
255 * socket at the end.
256 */
257 if (c->bio_chain != NULL) {
258 if ((dbio = BIO_dup_chain(c->bio_chain)) == NULL)
259 goto err;
260 if (!BIO_push(dbio, bio))
261 goto err;
262 bio = dbio;
263 }
264 if (BIO_push(b, bio) == NULL)
265 goto err;
266
267 c->state = ACPT_S_OK;
268 return (1);
269 err:
270 if (bio != NULL)
271 BIO_free(bio);
272 else if (s >= 0)
273 closesocket(s);
274 return (0);
275 /* break; */
276 case ACPT_S_OK:
277 if (b->next_bio == NULL) {
278 c->state = ACPT_S_GET_ACCEPT_SOCKET;
279 goto again;
280 }
281 return (1);
282 /* break; */
283 default:
284 return (0);
285 /* break; */
286 }
287
288 }
289
290 static int acpt_read(BIO *b, char *out, int outl)
291 {
292 int ret = 0;
293 BIO_ACCEPT *data;
294
295 BIO_clear_retry_flags(b);
296 data = (BIO_ACCEPT *)b->ptr;
297
298 while (b->next_bio == NULL) {
299 ret = acpt_state(b, data);
300 if (ret <= 0)
301 return (ret);
302 }
303
304 ret = BIO_read(b->next_bio, out, outl);
305 BIO_copy_next_retry(b);
306 return (ret);
307 }
308
309 static int acpt_write(BIO *b, const char *in, int inl)
310 {
311 int ret;
312 BIO_ACCEPT *data;
313
314 BIO_clear_retry_flags(b);
315 data = (BIO_ACCEPT *)b->ptr;
316
317 while (b->next_bio == NULL) {
318 ret = acpt_state(b, data);
319 if (ret <= 0)
320 return (ret);
321 }
322
323 ret = BIO_write(b->next_bio, in, inl);
324 BIO_copy_next_retry(b);
325 return (ret);
326 }
327
328 static long acpt_ctrl(BIO *b, int cmd, long num, void *ptr)
329 {
330 int *ip;
331 long ret = 1;
332 BIO_ACCEPT *data;
333 char **pp;
334
335 data = (BIO_ACCEPT *)b->ptr;
336
337 switch (cmd) {
338 case BIO_CTRL_RESET:
339 ret = 0;
340 data->state = ACPT_S_BEFORE;
341 acpt_close_socket(b);
342 b->flags = 0;
343 break;
344 case BIO_C_DO_STATE_MACHINE:
345 /* use this one to start the connection */
346 ret = (long)acpt_state(b, data);
347 break;
348 case BIO_C_SET_ACCEPT:
349 if (ptr != NULL) {
350 if (num == 0) {
351 b->init = 1;
352 OPENSSL_free(data->param_addr);
353 data->param_addr = OPENSSL_strdup(ptr);
354 } else if (num == 1) {
355 data->accept_nbio = (ptr != NULL);
356 } else if (num == 2) {
357 BIO_free(data->bio_chain);
358 data->bio_chain = (BIO *)ptr;
359 }
360 }
361 break;
362 case BIO_C_SET_NBIO:
363 data->nbio = (int)num;
364 break;
365 case BIO_C_SET_FD:
366 b->init = 1;
367 b->num = *((int *)ptr);
368 data->accept_sock = b->num;
369 data->state = ACPT_S_GET_ACCEPT_SOCKET;
370 b->shutdown = (int)num;
371 b->init = 1;
372 break;
373 case BIO_C_GET_FD:
374 if (b->init) {
375 ip = (int *)ptr;
376 if (ip != NULL)
377 *ip = data->accept_sock;
378 ret = data->accept_sock;
379 } else
380 ret = -1;
381 break;
382 case BIO_C_GET_ACCEPT:
383 if (b->init) {
384 if (ptr != NULL) {
385 pp = (char **)ptr;
386 *pp = data->param_addr;
387 } else
388 ret = -1;
389 } else
390 ret = -1;
391 break;
392 case BIO_CTRL_GET_CLOSE:
393 ret = b->shutdown;
394 break;
395 case BIO_CTRL_SET_CLOSE:
396 b->shutdown = (int)num;
397 break;
398 case BIO_CTRL_PENDING:
399 case BIO_CTRL_WPENDING:
400 ret = 0;
401 break;
402 case BIO_CTRL_FLUSH:
403 break;
404 case BIO_C_SET_BIND_MODE:
405 data->bind_mode = (int)num;
406 break;
407 case BIO_C_GET_BIND_MODE:
408 ret = (long)data->bind_mode;
409 break;
410 case BIO_CTRL_DUP:
411 /*- dbio=(BIO *)ptr;
412 if (data->param_port) EAY EAY
413 BIO_set_port(dbio,data->param_port);
414 if (data->param_hostname)
415 BIO_set_hostname(dbio,data->param_hostname);
416 BIO_set_nbio(dbio,data->nbio); */
417 break;
418
419 default:
420 ret = 0;
421 break;
422 }
423 return (ret);
424 }
425
426 static int acpt_puts(BIO *bp, const char *str)
427 {
428 int n, ret;
429
430 n = strlen(str);
431 ret = acpt_write(bp, str, n);
432 return (ret);
433 }
434
435 BIO *BIO_new_accept(const char *str)
436 {
437 BIO *ret;
438
439 ret = BIO_new(BIO_s_accept());
440 if (ret == NULL)
441 return (NULL);
442 if (BIO_set_accept_port(ret, str))
443 return (ret);
444 BIO_free(ret);
445 return (NULL);
446 }
447
448 #endif