comstack/tcpip.c

   1 /*
   2  * Copyright (c) 1995-2001, Index Data
   3  * See the file LICENSE for details.
   4  *
   5  * $Log: tcpip.c,v $
   6  * Revision 1.38  2001-03-21 12:43:36  adam
   7  * Implemented cs_create_host. Better error reporting for SSL comstack.
   8  *
   9  * Revision 1.37  2001/03/08 20:18:55  adam
  10  * Added cs_set_blocking. Patch from Matthew Carey.
  11  *
  12  * Revision 1.36  2001/02/21 13:46:53  adam
  13  * C++ fixes.
  14  *
  15  * Revision 1.35  2000/11/27 15:17:40  adam
  16  * Using SSLeay_add_all_algorithms instead of OpenSSL_add_all_algorithms.
  17  *
  18  * Revision 1.34  2000/11/23 10:58:32  adam
  19  * SSL comstack support. Separate POSIX thread support library.
  20  *
  21  * Revision 1.33  2000/09/04 08:27:11  adam
  22  * Work on error handling for tcpip_accept.
  23  *
  24  * Revision 1.32  1999/11/30 13:47:11  adam
  25  * Improved installation. Moved header files to include/yaz.
  26  *
  27  * Revision 1.31  1999/04/29 07:31:23  adam
  28  * Changed tcpip_strtoaddr_ex so that only part 'till '/' is considered
  29  * part of hostname.
  30  *
  31  * Revision 1.30  1999/04/20 09:56:48  adam
  32  * Added 'name' paramter to encoder/decoder routines (typedef Odr_fun).
  33  * Modified all encoders/decoders to reflect this change.
  34  *
  35  * Revision 1.29  1999/04/16 14:45:55  adam
  36  * Added interface for tcpd wrapper for access control.
  37  *
  38  * Revision 1.28  1999/03/31 11:11:14  adam
  39  * Function getprotobyname only called once. Minor change in tcpip_get
  40  * to handle multi-threaded conditions.
  41  *
  42  * Revision 1.27  1999/02/02 13:57:31  adam
  43  * Uses preprocessor define WIN32 instead of WINDOWS to build code
  44  * for Microsoft WIN32.
  45  *
  46  * Revision 1.26  1999/01/08 11:23:14  adam
  47  * Added const modifier to some of the BER/ODR encoding routines.
  48  *
  49  * Revision 1.25  1998/07/07 15:49:23  adam
  50  * Added braces to avoid warning.
  51  *
  52  * Revision 1.24  1998/06/29 07:59:17  adam
  53  * Minor fix.
  54  *
  55  * Revision 1.23  1998/06/23 15:37:50  adam
  56  * Added type cast to prevent warning.
  57  *
  58  * Revision 1.22  1998/06/22 11:32:36  adam
  59  * Added 'conditional cs_listen' feature.
  60  *
  61  * Revision 1.21  1998/05/20 09:55:32  adam
  62  * Function tcpip_get treats EINPROGRESS error in the same way as
  63  * EWOULDBLOCK. EINPROGRESS shouldn't be returned - but it is on
  64  * Solaris in some cases.
  65  *
  66  * Revision 1.20  1998/05/18 10:10:40  adam
  67  * Minor change to avoid C++ warning.
  68  *
  69  * Revision 1.19  1998/02/11 11:53:33  adam
  70  * Changed code so that it compiles as C++.
  71  *
  72  * Revision 1.18  1997/09/29 07:15:25  adam
  73  * Changed use of setsockopt to avoid warnings on MSVC.
  74  *
  75  * Revision 1.17  1997/09/17 12:10:30  adam
  76  * YAZ version 1.4.
  77  *
  78  * Revision 1.16  1997/09/01 08:49:14  adam
  79  * New windows NT/95 port using MSV5.0. Minor changes only.
  80  *
  81  * Revision 1.15  1997/05/14 06:53:33  adam
  82  * C++ support.
  83  *
  84  * Revision 1.14  1997/05/01 15:06:32  adam
  85  * Moved WINSOCK init. code to tcpip_init routine.
  86  *
  87  * Revision 1.13  1996/11/01 08:45:18  adam
  88  * Bug fix: used close on MS-Windows. Fixed to closesocket.
  89  *
  90  * Revision 1.12  1996/07/06 19:58:30  quinn
  91  * System headerfiles gathered in yconfig
  92  *
  93  * Revision 1.11  1996/02/23  10:00:39  quinn
  94  * WAIS Work
  95  *
  96  * Revision 1.10  1996/02/20  12:52:11  quinn
  97  * WAIS protocol support.
  98  *
  99  * Revision 1.9  1996/02/10  12:23:11  quinn
 100  * Enablie inetd operations fro TCP/IP stack
 101  *
 102  * Revision 1.8  1995/11/01  13:54:27  quinn
 103  * Minor adjustments
 104  *
 105  * Revision 1.7  1995/10/30  12:41:16  quinn
 106  * Added hostname lookup for server.
 107  *
 108  * Revision 1.6  1995/09/29  17:12:00  quinn
 109  * Smallish
 110  *
 111  * Revision 1.5  1995/09/29  17:01:48  quinn
 112  * More Windows work
 113  *
 114  * Revision 1.4  1995/09/28  10:12:26  quinn
 115  * Windows-support changes
 116  *
 117  * Revision 1.3  1995/09/27  15:02:45  quinn
 118  * Modified function heads & prototypes.
 119  *
 120  * Revision 1.2  1995/06/15  12:30:06  quinn
 121  * Added @ as hostname alias for INADDR ANY.
 122  *
 123  * Revision 1.1  1995/06/14  09:58:20  quinn
 124  * Renamed yazlib to comstack.
 125  *
 126  * Revision 1.20  1995/05/16  08:51:16  quinn
 127  * License, documentation, and memory fixes
 128  *
 129  * Revision 1.19  1995/04/10  10:24:08  quinn
 130  * Some bug-fixes.
 131  *
 132  * Revision 1.18  1995/03/30  13:29:27  quinn
 133  * Added REUSEADDR in tcpip_bind
 134  *
 135  * Revision 1.17  1995/03/27  08:36:10  quinn
 136  * Some work on nonblocking operation in xmosi.c and rfct.c.
 137  * Added protocol parameter to cs_create()
 138  *
 139  * Revision 1.16  1995/03/21  15:53:41  quinn
 140  * Added rcvconnect
 141  *
 142  * Revision 1.15  1995/03/21  12:31:27  quinn
 143  * Added check for EINPROGRESS on connect.
 144  *
 145  * Revision 1.14  1995/03/20  09:47:21  quinn
 146  * Added server-side support to xmosi.c
 147  * Fixed possible problems in rfct
 148  * Other little mods
 149  *
 150  * Revision 1.13  1995/03/15  16:15:13  adam
 151  * Removed p_write.
 152  *
 153  * Revision 1.12  1995/03/15  15:36:27  quinn
 154  * Mods to support nonblocking I/O
 155  *
 156  * Revision 1.11  1995/03/15  08:37:57  quinn
 157  * Now we're pretty much set for nonblocking I/O.
 158  *
 159  * Revision 1.10  1995/03/14  17:00:07  quinn
 160  * Bug-fixes - added tracing info to tcpip.c
 161  *
 162  * Revision 1.9  1995/03/14  10:28:42  quinn
 163  * Adding server-side support to tcpip.c and fixing bugs in nonblocking I/O
 164  *
 165  * Revision 1.8  1995/03/10  14:22:50  quinn
 166  * Removed debug output.
 167  *
 168  * Revision 1.7  1995/03/10  11:44:59  quinn
 169  * Fixes and debugging
 170  *
 171  * Revision 1.6  1995/03/07  10:26:55  quinn
 172  * Initialized type field in the comstacks.
 173  *
 174  * Revision 1.5  1995/02/14  20:40:07  quinn
 175  * Various stuff.
 176  *
 177  * Revision 1.4  1995/02/14  11:54:49  quinn
 178  * Beginning to add full CCL.
 179  *
 180  * Revision 1.3  1995/02/10  18:58:10  quinn
 181  * Fixed tcpip_get (formerly tcpip_read).
 182  * Turned tst (cli) into a proper, event-driven thingy.
 183  *
 184  * Revision 1.2  1995/02/10  15:55:47  quinn
 185  * Small things.
 186  *
 187  * Revision 1.1  1995/02/09  15:51:52  quinn
 188  * Works better now.
 189  *
 190  */
 191
 192 #include <stdio.h>
 193 #include <string.h>
 194 #include <stdlib.h>
 195 #ifndef WIN32
 196 #include <unistd.h>
 197 #endif
 198 #include <errno.h>
 199 #include <fcntl.h>
 200 #if HAVE_OPENSSL_SSL_H
 201 #include <openssl/ssl.h>
 202 #include <openssl/err.h>
 203 #endif
 204
 205 #include <yaz/comstack.h>
 206 #include <yaz/tcpip.h>
 207 #include <yaz/log.h>
 208
 209 /* Chas added the following, so we get the definition of completeBER */
 210 #include <yaz/odr.h>
 211
 212 int tcpip_close(COMSTACK h);
 213 int tcpip_put(COMSTACK h, char *buf, int size);
 214 int tcpip_get(COMSTACK h, char **buf, int *bufsize);
 215 int tcpip_connect(COMSTACK h, void *address);
 216 int tcpip_more(COMSTACK h);
 217 int tcpip_rcvconnect(COMSTACK h);
 218 int tcpip_bind(COMSTACK h, void *address, int mode);
 219 int tcpip_listen(COMSTACK h, char *raddr, int *addrlen,
 220                  int (*check_ip)(void *cd, const char *a, int len, int type),
 221                  void *cd);
 222 int static tcpip_set_blocking(COMSTACK p, int blocking);
 223
 224 #if HAVE_OPENSSL_SSL_H
 225 int ssl_get(COMSTACK h, char **buf, int *bufsize);
 226 int ssl_put(COMSTACK h, char *buf, int size);
 227 #endif
 228
 229 COMSTACK tcpip_accept(COMSTACK h);
 230 char *tcpip_addrstr(COMSTACK h);
 231 void *tcpip_straddr(COMSTACK h, const char *str);
 232
 233 #if 0
 234 #define TRC(x) x
 235 #else
 236 #define TRC(X)
 237 #endif
 238
 239 /* this state is used for both SSL and straight TCP/IP */
 240 typedef struct tcpip_state
 241 {
 242     char *altbuf; /* alternate buffer for surplus data */
 243     int altsize;  /* size as xmalloced */
 244     int altlen;   /* length of data or 0 if none */
 245
 246     int written;  /* -1 if we aren't writing */
 247     int towrite;  /* to verify against user input */
 248     int (*complete)(const unsigned char *buf, int len); /* length/comple. */
 249     struct sockaddr_in addr;  /* returned by cs_straddr */
 250     char buf[128]; /* returned by cs_addrstr */
 251 #if HAVE_OPENSSL_SSL_H
 252     SSL_CTX *ctx;
 253     SSL_CTX *ctx_alloc;
 254     SSL *ssl;
 255 #endif
 256 } tcpip_state;
 257
 258 #ifdef WIN32
 259 static int tcpip_init (void)
 260 {
 261     static int initialized = 0;
 262     if (!initialized)
 263     {
 264         WORD requested;
 265         WSADATA wd;
 266
 267         requested = MAKEWORD(1, 1);
 268         if (WSAStartup(requested, &wd))
 269             return 0;
 270         initialized = 1;
 271     }
 272     return 1;
 273 }
 274 #else
 275 static int proto_number = 0;
 276
 277 static int tcpip_init (void)
 278 {
 279     struct protoent *proto;
 280     /* only call getprotobyname once, in case it allocates memory */
 281     if (!(proto = getprotobyname("tcp")))
 282         return 0;
 283     proto_number = proto->p_proto;
 284     return 1;
 285 }
 286 #endif
 287
 288 /*
 289  * This function is always called through the cs_create() macro.
 290  * s >= 0: socket has already been established for us.
 291  */
 292 COMSTACK tcpip_type(int s, int blocking, int protocol, void *vp)
 293 {
 294     COMSTACK p;
 295     tcpip_state *state;
 296     int new_socket;
 297 #ifdef WIN32
 298     unsigned long tru = 1;
 299 #endif
 300
 301     if (!tcpip_init ())
 302         return 0;
 303     if (s < 0)
 304     {
 305 #ifdef WIN32
 306         if ((s = socket(AF_INET, SOCK_STREAM, 0)) < 0)
 307             return 0;
 308 #else
 309         if ((s = socket(AF_INET, SOCK_STREAM, proto_number)) < 0)
 310             return 0;
 311 #endif
 312         new_socket = 1;
 313     }
 314     else
 315         new_socket = 0;
 316     if (!(p = (struct comstack *)xmalloc(sizeof(struct comstack))))
 317         return 0;
 318     if (!(state = (struct tcpip_state *)(p->cprivate =
 319                                          xmalloc(sizeof(tcpip_state)))))
 320         return 0;
 321
 322 #ifdef WIN32
 323     if (!(p->blocking = blocking) && ioctlsocket(s, FIONBIO, &tru) < 0)
 324 #else
 325     if (!(p->blocking = blocking) && fcntl(s, F_SETFL, O_NONBLOCK) < 0)
 326 #endif
 327         return 0;
 328
 329     p->io_pending = 0;
 330     p->iofile = s;
 331     p->type = tcpip_type;
 332     p->protocol = (enum oid_proto) protocol;
 333
 334     p->f_connect = tcpip_connect;
 335     p->f_rcvconnect = tcpip_rcvconnect;
 336     p->f_get = tcpip_get;
 337     p->f_put = tcpip_put;
 338     p->f_close = tcpip_close;
 339     p->f_more = tcpip_more;
 340     p->f_bind = tcpip_bind;
 341     p->f_listen = tcpip_listen;
 342     p->f_accept = tcpip_accept;
 343     p->f_addrstr = tcpip_addrstr;
 344     p->f_straddr = tcpip_straddr;
 345     p->f_set_blocking = tcpip_set_blocking;
 346
 347     p->state = new_socket ? CS_UNBND : CS_IDLE; /* state of line */
 348     p->event = CS_NONE;
 349     p->cerrno = 0;
 350     p->stackerr = 0;
 351
 352 #if HAVE_OPENSSL_SSL_H
 353     state->ctx = state->ctx_alloc = 0;
 354     state->ssl = 0;
 355 #endif
 356
 357     state->altbuf = 0;
 358     state->altsize = state->altlen = 0;
 359     state->towrite = state->written = -1;
 360     if (protocol == PROTO_WAIS)
 361         state->complete = completeWAIS;
 362     else
 363         state->complete = completeBER;
 364
 365     p->timeout = COMSTACK_DEFAULT_TIMEOUT;
 366     TRC(fprintf(stderr, "Created new TCPIP comstack\n"));
 367
 368     return p;
 369 }
 370
 371 #if HAVE_OPENSSL_SSL_H
 372
 373 COMSTACK ssl_type(int s, int blocking, int protocol, void *vp)
 374 {
 375     tcpip_state *state;
 376     COMSTACK p;
 377     yaz_log(LOG_LOG, "ssl_type begin");
 378
 379     p = tcpip_type (s, blocking, protocol, 0);
 380     if (!p)
 381         return 0;
 382     p->f_get = ssl_get;
 383     p->f_put = ssl_put;
 384     p->type = ssl_type;
 385     state = (tcpip_state *) p->cprivate;
 386     if (vp)
 387         state->ctx = vp;
 388     else
 389     {
 390         SSL_load_error_strings();
 391         SSLeay_add_all_algorithms();
 392
 393         state->ctx = state->ctx_alloc = SSL_CTX_new (SSLv23_method());
 394         if (!state->ctx)
 395         {
 396             tcpip_close(p);
 397             return 0;
 398         }
 399     }
 400     /* note: we don't handle already opened socket in SSL mode - yet */
 401     yaz_log(LOG_LOG, "ssl_type end");
 402     return p;
 403 }
 404 #endif
 405
 406 int tcpip_strtoaddr_ex(const char *str, struct sockaddr_in *add)
 407 {
 408     struct hostent *hp;
 409     char *p, buf[512];
 410     short int port = 210;
 411     unsigned tmpadd;
 412
 413     if (!tcpip_init ())
 414         return 0;
 415     TRC(fprintf(stderr, "tcpip_strtoaddress: %s\n", str ? str : "NULL"));
 416     add->sin_family = AF_INET;
 417     strncpy(buf, str, 511);
 418     buf[511] = 0;
 419     if ((p = strchr(buf, '/')))
 420         *p = 0;
 421     if ((p = strchr(buf, ':')))
 422     {
 423         *p = 0;
 424         port = atoi(p + 1);
 425     }
 426     add->sin_port = htons(port);
 427     if (!strcmp("@", buf))
 428         add->sin_addr.s_addr = INADDR_ANY;
 429     else if ((hp = gethostbyname(buf)))
 430         memcpy(&add->sin_addr.s_addr, *hp->h_addr_list,
 431                sizeof(struct in_addr));
 432     else if ((tmpadd = (unsigned) inet_addr(buf)) != 0)
 433         memcpy(&add->sin_addr.s_addr, &tmpadd, sizeof(struct in_addr));
 434     else
 435         return 0;
 436     return 1;
 437 }
 438
 439 void *tcpip_straddr(COMSTACK h, const char *str)
 440 {
 441     tcpip_state *sp = (tcpip_state *)h->cprivate;
 442
 443     if (!tcpip_strtoaddr_ex (str, &sp->addr))
 444         return 0;
 445     return &sp->addr;
 446 }
 447
 448 struct sockaddr_in *tcpip_strtoaddr(const char *str)
 449 {
 450     static struct sockaddr_in add;
 451
 452     if (!tcpip_strtoaddr_ex (str, &add))
 453         return 0;
 454     return &add;
 455 }
 456
 457 int tcpip_more(COMSTACK h)
 458 {
 459     tcpip_state *sp = (tcpip_state *)h->cprivate;
 460
 461     return sp->altlen && (*sp->complete)((unsigned char *) sp->altbuf,
 462         sp->altlen);
 463 }
 464
 465 /*
 466  * connect(2) will block (sometimes) - nothing we can do short of doing
 467  * weird things like spawning subprocesses or threading or some weird junk
 468  * like that.
 469  */
 470 int tcpip_connect(COMSTACK h, void *address)
 471 {
 472     struct sockaddr_in *add = (struct sockaddr_in *)address;
 473 #if HAVE_OPENSSL_SSL_H
 474         tcpip_state *sp = (tcpip_state *)h->cprivate;
 475 #endif
 476     int r;
 477
 478     TRC(fprintf(stderr, "tcpip_connect\n"));
 479     h->io_pending = 0;
 480     if (h->state == CS_UNBND)
 481     {
 482         r = connect(h->iofile, (struct sockaddr *) add, sizeof(*add));
 483         if (r < 0)
 484         {
 485 #ifdef WIN32
 486             if (WSAGetLastError() == WSAEWOULDBLOCK)
 487             {
 488                 h->state = CS_CONNECTING;
 489                 h->io_pending = CS_WANT_WRITE;
 490                 return 1;
 491             }
 492 #else
 493             if (errno == EINPROGRESS)
 494             {
 495                 h->state = CS_CONNECTING;
 496                 h->io_pending = CS_WANT_WRITE|CS_WANT_READ;
 497                 return 1;
 498             }
 499 #endif
 500             h->cerrno = CSYSERR;
 501             return -1;
 502         }
 503         h->state = CS_CONNECTING;
 504     }
 505     if (h->state != CS_CONNECTING)
 506     {
 507         h->cerrno = CSOUTSTATE;
 508         return -1;
 509     }
 510 #if HAVE_OPENSSL_SSL_H
 511     if (sp->ctx)
 512     {
 513         int res;
 514
 515         if (!sp->ssl)
 516         {
 517             sp->ssl = SSL_new (sp->ctx);
 518             SSL_set_fd (sp->ssl, h->iofile);
 519         }
 520         res = SSL_connect (sp->ssl);
 521         if (res <= 0)
 522         {
 523             int err = SSL_get_error(sp->ssl, res);
 524             if (err == SSL_ERROR_WANT_READ)
 525             {
 526                 yaz_log (LOG_LOG, "SSL_connect. want_read");
 527                 h->io_pending = CS_WANT_READ;
 528                 return 1;
 529             }
 530             if (err == SSL_ERROR_WANT_WRITE)
 531             {
 532                 yaz_log (LOG_LOG, "SSL_connect. want_write");
 533                 h->io_pending = CS_WANT_WRITE;
 534                 return 1;
 535             }
 536             h->cerrno = CSERRORSSL;
 537             return -1;
 538         }
 539     }
 540 #endif
 541     h->state = CS_DATAXFER;
 542     return 0;
 543 }
 544
 545 /*
 546  * nop
 547  */
 548 int tcpip_rcvconnect(COMSTACK h)
 549 {
 550     TRC(fprintf(stderr, "tcpip_rcvconnect\n"));
 551     return 0;
 552 }
 553
 554 #define CERTF "ztest.pem"
 555 #define KEYF "ztest.pem"
 556
 557 int tcpip_bind(COMSTACK h, void *address, int mode)
 558 {
 559     struct sockaddr *addr = (struct sockaddr *)address;
 560 #ifdef WIN32
 561     BOOL one = 1;
 562 #else
 563     unsigned long one = 1;
 564 #endif
 565
 566 #if HAVE_OPENSSL_SSL_H
 567     tcpip_state *sp = (tcpip_state *)h->cprivate;
 568     if (sp->ctx)
 569     {
 570         if (sp->ctx_alloc)
 571         {
 572             int res;
 573             res = SSL_CTX_use_certificate_file (sp->ctx, CERTF,
 574                                                 SSL_FILETYPE_PEM);
 575             if (res <= 0)
 576             {
 577                 ERR_print_errors_fp(stderr);
 578                 exit (2);
 579             }
 580             res = SSL_CTX_use_PrivateKey_file (sp->ctx, KEYF,
 581                                                SSL_FILETYPE_PEM);
 582             if (res <= 0)
 583             {
 584                 ERR_print_errors_fp(stderr);
 585                 exit (3);
 586             }
 587             res = SSL_CTX_check_private_key (sp->ctx);
 588             if (res <= 0)
 589             {
 590                 ERR_print_errors_fp(stderr);
 591                 exit(5);
 592             }
 593         }
 594         TRC (fprintf (stderr, "ssl_bind\n"));
 595     }
 596     else
 597     {
 598         TRC (fprintf (stderr, "tcpip_bind\n"));
 599     }
 600 #else
 601     TRC (fprintf (stderr, "tcpip_bind\n"));
 602 #endif
 603     if (setsockopt(h->iofile, SOL_SOCKET, SO_REUSEADDR, (char*)
 604         &one, sizeof(one)) < 0)
 605     {
 606         h->cerrno = CSYSERR;
 607         return -1;
 608     }
 609     if (bind(h->iofile, addr, sizeof(struct sockaddr_in)) < 0)
 610     {
 611         h->cerrno = CSYSERR;
 612         return -1;
 613     }
 614     if (mode == CS_SERVER && listen(h->iofile, 3) < 0)
 615     {
 616         h->cerrno = CSYSERR;
 617         return -1;
 618     }
 619     h->state = CS_IDLE;
 620     return 0;
 621 }
 622
 623 int tcpip_listen(COMSTACK h, char *raddr, int *addrlen,
 624                  int (*check_ip)(void *cd, const char *a, int len, int t),
 625                  void *cd)
 626 {
 627     struct sockaddr_in addr;
 628 #ifdef __cplusplus
 629     socklen_t len = sizeof(addr);
 630 #else
 631     int len = sizeof(addr);
 632 #endif
 633
 634     TRC(fprintf(stderr, "tcpip_listen pid=%d\n", getpid()));
 635     if (h->state != CS_IDLE)
 636     {
 637         h->cerrno = CSOUTSTATE;
 638         return -1;
 639     }
 640     h->newfd = accept(h->iofile, (struct sockaddr*)&addr, &len);
 641     if (h->newfd < 0)
 642     {
 643         if (
 644 #ifdef WIN32
 645             WSAGetLastError() == WSAEWOULDBLOCK
 646 #else
 647             errno == EWOULDBLOCK
 648 #endif
 649             )
 650             h->cerrno = CSNODATA;
 651         else
 652             h->cerrno = CSYSERR;
 653         return -1;
 654     }
 655     if (addrlen && (size_t) (*addrlen) >= sizeof(struct sockaddr_in))
 656         memcpy(raddr, &addr, *addrlen = sizeof(struct sockaddr_in));
 657     else if (addrlen)
 658         *addrlen = 0;
 659     if (check_ip && (*check_ip)(cd, (const char *) &addr,
 660         sizeof(addr), AF_INET))
 661     {
 662         h->cerrno = CSDENY;
 663 #ifdef WIN32
 664         closesocket(h->newfd);
 665 #else
 666         close(h->newfd);
 667 #endif
 668         h->newfd = -1;
 669         return -1;
 670     }
 671     h->state = CS_INCON;
 672     return 0;
 673 }
 674
 675 COMSTACK tcpip_accept(COMSTACK h)
 676 {
 677     COMSTACK cnew;
 678     tcpip_state *state, *st = (tcpip_state *)h->cprivate;
 679 #ifdef WIN32
 680     unsigned long tru = 1;
 681 #endif
 682
 683     TRC(fprintf(stderr, "tcpip_accept\n"));
 684     if (h->state == CS_INCON)
 685     {
 686         if (!(cnew = (COMSTACK)xmalloc(sizeof(*cnew))))
 687         {
 688             h->cerrno = CSYSERR;
 689 #ifdef WIN32
 690             closesocket(h->newfd);
 691 #else
 692             close(h->newfd);
 693 #endif
 694             h->newfd = -1;
 695             return 0;
 696         }
 697         memcpy(cnew, h, sizeof(*h));
 698         cnew->iofile = h->newfd;
 699         cnew->io_pending = 0;
 700         if (!(state = (tcpip_state *)
 701               (cnew->cprivate = xmalloc(sizeof(tcpip_state)))))
 702         {
 703             h->cerrno = CSYSERR;
 704             if (h->newfd != -1)
 705             {
 706 #ifdef WIN32
 707                 closesocket(h->newfd);
 708 #else
 709                 close(h->newfd);
 710 #endif
 711                 h->newfd = -1;
 712             }
 713             return 0;
 714         }
 715         if (!cnew->blocking &&
 716 #ifdef WIN32
 717             (ioctlsocket(cnew->iofile, FIONBIO, &tru) < 0)
 718 #else
 719             (!cnew->blocking && fcntl(cnew->iofile, F_SETFL, O_NONBLOCK) < 0)
 720 #endif
 721             )
 722         {
 723             h->cerrno = CSYSERR;
 724             if (h->newfd != -1)
 725             {
 726 #ifdef WIN32
 727                 closesocket(h->newfd);
 728 #else
 729                 close(h->newfd);
 730 #endif
 731                 h->newfd = -1;
 732             }
 733             xfree (cnew);
 734             xfree (state);
 735             return 0;
 736         }
 737         h->newfd = -1;
 738         state->altbuf = 0;
 739         state->altsize = state->altlen = 0;
 740         state->towrite = state->written = -1;
 741         state->complete = st->complete;
 742         cnew->state = CS_ACCEPT;
 743         h->state = CS_IDLE;
 744
 745 #if HAVE_OPENSSL_SSL_H
 746         state->ctx = st->ctx;
 747         state->ctx_alloc = 0;
 748         state->ssl = st->ssl;
 749         if (state->ctx)
 750         {
 751             state->ssl = SSL_new (state->ctx);
 752             SSL_set_fd (state->ssl, cnew->iofile);
 753         }
 754 #endif
 755         h = cnew;
 756     }
 757     if (h->state == CS_ACCEPT)
 758     {
 759 #if HAVE_OPENSSL_SSL_H
 760         tcpip_state *state = (tcpip_state *)h->cprivate;
 761         if (state->ctx)
 762         {
 763             int res = SSL_accept (state->ssl);
 764             TRC(fprintf(stderr, "SSL_accept\n"));
 765             if (res <= 0)
 766             {
 767                 int err = SSL_get_error(state->ssl, res);
 768                 if (err == SSL_ERROR_WANT_READ)
 769                 {
 770                     h->io_pending = CS_WANT_READ;
 771                     yaz_log (LOG_LOG, "SSL_accept. want_read");
 772                     return h;
 773                 }
 774                 if (err == SSL_ERROR_WANT_WRITE)
 775                 {
 776                     h->io_pending = CS_WANT_WRITE;
 777                     yaz_log (LOG_LOG, "SSL_accept. want_write");
 778                     return h;
 779                 }
 780                 cs_close (h);
 781                 return 0;
 782             }
 783         }
 784 #endif
 785     }
 786     else
 787     {
 788         h->cerrno = CSOUTSTATE;
 789         return 0;
 790     }
 791     h->io_pending = 0;
 792     h->state = CS_DATAXFER;
 793     return h;
 794 }
 795
 796 #define CS_TCPIP_BUFCHUNK 4096
 797
 798 /*
 799  * Return: -1 error, >1 good, len of buffer, ==1 incomplete buffer,
 800  * 0=connection closed.
 801  */
 802 int tcpip_get(COMSTACK h, char **buf, int *bufsize)
 803 {
 804     tcpip_state *sp = (tcpip_state *)h->cprivate;
 805     char *tmpc;
 806     int tmpi, berlen, rest, req, tomove;
 807     int hasread = 0, res;
 808
 809     TRC(fprintf(stderr, "tcpip_get: bufsize=%d\n", *bufsize));
 810     if (sp->altlen) /* switch buffers */
 811     {
 812         TRC(fprintf(stderr, "  %d bytes in altbuf (0x%x)\n", sp->altlen,
 813             (unsigned) sp->altbuf));
 814         tmpc = *buf;
 815         tmpi = *bufsize;
 816         *buf = sp->altbuf;
 817         *bufsize = sp->altsize;
 818         hasread = sp->altlen;
 819         sp->altlen = 0;
 820         sp->altbuf = tmpc;
 821         sp->altsize = tmpi;
 822     }
 823     h->io_pending = 0;
 824     while (!(berlen = (*sp->complete)((unsigned char *)*buf, hasread)))
 825     {
 826         if (!*bufsize)
 827         {
 828             if (!(*buf = (char *)xmalloc(*bufsize = CS_TCPIP_BUFCHUNK)))
 829                 return -1;
 830         }
 831         else if (*bufsize - hasread < CS_TCPIP_BUFCHUNK)
 832             if (!(*buf =(char *)xrealloc(*buf, *bufsize *= 2)))
 833                 return -1;
 834         res = recv(h->iofile, *buf + hasread, CS_TCPIP_BUFCHUNK, 0);
 835         TRC(fprintf(stderr, "  recv res=%d, hasread=%d\n", res, hasread));
 836         if (res < 0)
 837         {
 838 #ifdef WIN32
 839             if (WSAGetLastError() == WSAEWOULDBLOCK)
 840             {
 841                 h->io_pending = CS_WANT_READ;
 842                 break;
 843             }
 844             else
 845                 return -1;
 846 #else
 847             if (errno == EWOULDBLOCK
 848 #ifdef EINPROGRESS
 849                 || errno == EINPROGRESS
 850 #endif
 851                 )
 852             {
 853                 h->io_pending = CS_WANT_READ;
 854                 break;
 855             }
 856             else if (errno == 0)
 857                 continue;
 858             else
 859                 return -1;
 860 #endif
 861         }
 862         else if (!res)
 863             return 0;
 864         hasread += res;
 865     }
 866     TRC (fprintf (stderr, "  Out of read loop with hasread=%d, berlen=%d\n",
 867                   hasread, berlen));
 868     /* move surplus buffer (or everything if we didn't get a BER rec.) */
 869     if (hasread > berlen)
 870     {
 871         tomove = req = hasread - berlen;
 872         rest = tomove % CS_TCPIP_BUFCHUNK;
 873         if (rest)
 874             req += CS_TCPIP_BUFCHUNK - rest;
 875         if (!sp->altbuf)
 876         {
 877             if (!(sp->altbuf = (char *)xmalloc(sp->altsize = req)))
 878                 return -1;
 879         } else if (sp->altsize < req)
 880             if (!(sp->altbuf =(char *)xrealloc(sp->altbuf, sp->altsize = req)))
 881                 return -1;
 882         TRC(fprintf(stderr, "  Moving %d bytes to altbuf(0x%x)\n", tomove,
 883             (unsigned) sp->altbuf));
 884         memcpy(sp->altbuf, *buf + berlen, sp->altlen = tomove);
 885     }
 886     if (berlen < CS_TCPIP_BUFCHUNK - 1)
 887         *(*buf + berlen) = '\0';
 888     return berlen ? berlen : 1;
 889 }
 890
 891
 892 #if HAVE_OPENSSL_SSL_H
 893 /*
 894  * Return: -1 error, >1 good, len of buffer, ==1 incomplete buffer,
 895  * 0=connection closed.
 896  */
 897 int ssl_get(COMSTACK h, char **buf, int *bufsize)
 898 {
 899     tcpip_state *sp = (tcpip_state *)h->cprivate;
 900     char *tmpc;
 901     int tmpi, berlen, rest, req, tomove;
 902     int hasread = 0, res;
 903
 904     TRC(fprintf(stderr, "ssl_get: bufsize=%d\n", *bufsize));
 905     if (sp->altlen) /* switch buffers */
 906     {
 907         TRC(fprintf(stderr, "  %d bytes in altbuf (0x%x)\n", sp->altlen,
 908             (unsigned) sp->altbuf));
 909         tmpc = *buf;
 910         tmpi = *bufsize;
 911         *buf = sp->altbuf;
 912         *bufsize = sp->altsize;
 913         hasread = sp->altlen;
 914         sp->altlen = 0;
 915         sp->altbuf = tmpc;
 916         sp->altsize = tmpi;
 917     }
 918     h->io_pending = 0;
 919     while (!(berlen = (*sp->complete)((unsigned char *)*buf, hasread)))
 920     {
 921         if (!*bufsize)
 922         {
 923             if (!(*buf = (char *)xmalloc(*bufsize = CS_TCPIP_BUFCHUNK)))
 924                 return -1;
 925         }
 926         else if (*bufsize - hasread < CS_TCPIP_BUFCHUNK)
 927             if (!(*buf =(char *)xrealloc(*buf, *bufsize *= 2)))
 928                 return -1;
 929         res = SSL_read (sp->ssl, *buf + hasread, CS_TCPIP_BUFCHUNK);
 930         TRC(fprintf(stderr, "  SSL_read res=%d, hasread=%d\n", res, hasread));
 931         if (res <= 0)
 932         {
 933             int ssl_err = SSL_get_error(sp->ssl, res);
 934             if (ssl_err == SSL_ERROR_WANT_READ)
 935             {
 936                 h->io_pending = CS_WANT_READ;
 937                 yaz_log (LOG_LOG, "SSL_read. want_read");
 938                 break;
 939             }
 940             if (ssl_err == SSL_ERROR_WANT_WRITE)
 941             {
 942                 h->io_pending = CS_WANT_WRITE;
 943                 yaz_log (LOG_LOG, "SSL_read. want_write");
 944                 break;
 945             }
 946             if (res == 0)
 947                 return 0;
 948             h->cerrno = CSERRORSSL;
 949             return -1;
 950         }
 951         hasread += res;
 952     }
 953     TRC (fprintf (stderr, "  Out of read loop with hasread=%d, berlen=%d\n",
 954         hasread, berlen));
 955     /* move surplus buffer (or everything if we didn't get a BER rec.) */
 956     if (hasread > berlen)
 957     {
 958         tomove = req = hasread - berlen;
 959         rest = tomove % CS_TCPIP_BUFCHUNK;
 960         if (rest)
 961             req += CS_TCPIP_BUFCHUNK - rest;
 962         if (!sp->altbuf)
 963         {
 964             if (!(sp->altbuf = (char *)xmalloc(sp->altsize = req)))
 965                 return -1;
 966         } else if (sp->altsize < req)
 967             if (!(sp->altbuf =(char *)xrealloc(sp->altbuf, sp->altsize = req)))
 968                 return -1;
 969         TRC(fprintf(stderr, "  Moving %d bytes to altbuf(0x%x)\n", tomove,
 970             (unsigned) sp->altbuf));
 971         memcpy(sp->altbuf, *buf + berlen, sp->altlen = tomove);
 972     }
 973     if (berlen < CS_TCPIP_BUFCHUNK - 1)
 974         *(*buf + berlen) = '\0';
 975     return berlen ? berlen : 1;
 976 }
 977 #endif
 978
 979 /*
 980  * Returns 1, 0 or -1
 981  * In nonblocking mode, you must call again with same buffer while
 982  * return value is 1.
 983  */
 984 int tcpip_put(COMSTACK h, char *buf, int size)
 985 {
 986     int res;
 987     struct tcpip_state *state = (struct tcpip_state *)h->cprivate;
 988
 989     TRC(fprintf(stderr, "tcpip_put: size=%d\n", size));
 990     h->io_pending = 0;
 991     if (state->towrite < 0)
 992     {
 993         state->towrite = size;
 994         state->written = 0;
 995     }
 996     else if (state->towrite != size)
 997     {
 998         h->cerrno = CSWRONGBUF;
 999         return -1;
1000     }
1001     while (state->towrite > state->written)
1002     {
1003         if ((res = send(h->iofile, buf + state->written, size -
1004                         state->written, 0)) < 0)
1005         {
1006             if (
1007 #ifdef WIN32
1008                 WSAGetLastError() == WSAEWOULDBLOCK
1009 #else
1010                 errno == EAGAIN
1011 #endif
1012                 )
1013             {
1014                 TRC(fprintf(stderr, "  Flow control stop\n"));
1015                 h->io_pending = CS_WANT_WRITE;
1016                 return 1;
1017             }
1018             h->cerrno = CSYSERR;
1019             return -1;
1020         }
1021         state->written += res;
1022         TRC(fprintf(stderr, "  Wrote %d, written=%d, nbytes=%d\n",
1023                     res, state->written, size));
1024     }
1025     state->towrite = state->written = -1;
1026     TRC(fprintf(stderr, "  Ok\n"));
1027     return 0;
1028 }
1029
1030
1031 #if HAVE_OPENSSL_SSL_H
1032 /*
1033  * Returns 1, 0 or -1
1034  * In nonblocking mode, you must call again with same buffer while
1035  * return value is 1.
1036  */
1037 int ssl_put(COMSTACK h, char *buf, int size)
1038 {
1039     int res;
1040     struct tcpip_state *state = (struct tcpip_state *)h->cprivate;
1041
1042     TRC(fprintf(stderr, "ssl_put: size=%d\n", size));
1043     h->io_pending = 0;
1044     if (state->towrite < 0)
1045     {
1046         state->towrite = size;
1047         state->written = 0;
1048     }
1049     else if (state->towrite != size)
1050     {
1051         h->cerrno = CSWRONGBUF;
1052         return -1;
1053     }
1054     while (state->towrite > state->written)
1055     {
1056         res = SSL_write (state->ssl, buf + state->written,
1057                          size - state->written);
1058         if (res <= 0)
1059         {
1060             int ssl_err = SSL_get_error(state->ssl, res);
1061             if (ssl_err == SSL_ERROR_WANT_READ)
1062             {
1063                 h->io_pending = CS_WANT_READ;
1064                 yaz_log (LOG_LOG, "SSL_write. want_read");
1065                 return 1;
1066             }
1067             if (ssl_err == SSL_ERROR_WANT_WRITE)
1068             {
1069                 h->io_pending = CS_WANT_WRITE;
1070                 yaz_log (LOG_LOG, "SSL_write. want_write");
1071                 return 1;
1072             }
1073             h->cerrno = CSERRORSSL;
1074             return -1;
1075         }
1076         state->written += res;
1077         TRC(fprintf(stderr, "  Wrote %d, written=%d, nbytes=%d\n",
1078                     res, state->written, size));
1079     }
1080     state->towrite = state->written = -1;
1081     TRC(fprintf(stderr, "  Ok\n"));
1082     return 0;
1083 }
1084 #endif
1085
1086 int tcpip_close(COMSTACK h)
1087 {
1088     tcpip_state *sp = (struct tcpip_state *)h->cprivate;
1089
1090     TRC(fprintf(stderr, "tcpip_close\n"));
1091     if (h->iofile != -1)
1092     {
1093 #if HAVE_OPENSSL_SSL_H
1094         if (sp->ssl)
1095         {
1096             SSL_shutdown (sp->ssl);
1097         }
1098 #endif
1099 #ifdef WIN32
1100         closesocket(h->iofile);
1101 #else
1102         close(h->iofile);
1103 #endif
1104     }
1105     if (sp->altbuf)
1106         xfree(sp->altbuf);
1107 #if HAVE_OPENSSL_SSL_H
1108     if (sp->ssl)
1109     {
1110         TRC (fprintf(stderr, "SSL_free\n"));
1111         SSL_free (sp->ssl);
1112     }
1113     sp->ssl = 0;
1114     if (sp->ctx_alloc)
1115         SSL_CTX_free (sp->ctx_alloc);
1116 #endif
1117     xfree(sp);
1118     xfree(h);
1119     return 0;
1120 }
1121
1122 char *tcpip_addrstr(COMSTACK h)
1123 {
1124     struct sockaddr_in addr;
1125     tcpip_state *sp = (struct tcpip_state *)h->cprivate;
1126     char *r, *buf = sp->buf;
1127     size_t len;
1128     struct hostent *host;
1129
1130     len = sizeof(addr);
1131     if (getpeername(h->iofile, (struct sockaddr*) &addr, &len) < 0)
1132     {
1133         h->cerrno = CSYSERR;
1134         return 0;
1135     }
1136     if ((host = gethostbyaddr((char*)&addr.sin_addr, sizeof(addr.sin_addr),
1137                               AF_INET)))
1138         r = (char*) host->h_name;
1139     else
1140         r = inet_ntoa(addr.sin_addr);
1141     sprintf(buf, "tcp:%s", r);
1142 #if HAVE_OPENSSL_SSL_H
1143     if (sp->ctx)
1144         sprintf(buf, "ssl:%s", r);
1145 #endif
1146     return buf;
1147 }
1148
1149 int static tcpip_set_blocking(COMSTACK p, int blocking)
1150 {
1151     unsigned long flag;
1152
1153     if (p->blocking == blocking)
1154         return 1;
1155     if(blocking)
1156 #ifdef WIN32
1157         flag = blocking;
1158     if (ioctlsocket(p->iofile, FIONBIO, &flag) < 0)
1159         return 0;
1160 #else
1161     flag = fcntl(p->iofile, F_GETFL, 0);
1162     if(!blocking)
1163         flag = flag & ~O_NONBLOCK;
1164     else
1165         flag = flag | O_NONBLOCK;
1166     if (fcntl(p->iofile, F_SETFL, flag) < 0)
1167         return 0;
1168 #endif
1169     p->blocking = blocking;
1170     return 1;
1171 }