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