Remove include of sys/select.h from tcpip.c . It is not needed.
[yaz-moved-to-github.git] / src / tcpip.c
1 /*
2  * Copyright (C) 1995-2007, Index Data ApS
3  * See the file LICENSE for details.
4  *
5  * $Id: tcpip.c,v 1.37 2007-11-09 21:52:38 adam Exp $
6  */
7 /**
8  * \file tcpip.c
9  * \brief Implements TCP/IP + SSL COMSTACK.
10  */
11
12 #include <stdio.h>
13 #include <string.h>
14 #include <stdlib.h>
15 #include <errno.h>
16 #include <fcntl.h>
17 #include <signal.h>
18 #if HAVE_SYS_TYPES_H
19 #include <sys/types.h>
20 #endif
21 #if HAVE_SYS_TIME_H
22 #include <sys/time.h>
23 #endif
24 #if HAVE_UNISTD_H
25 #include <unistd.h>
26 #endif
27
28 #ifdef WIN32
29
30 /* VS 2003 or later has getaddrinfo; older versions do not */
31 #include <winsock2.h>
32 #if _MSC_VER >= 1300
33 #include <ws2tcpip.h>
34 #define HAVE_GETADDRINFO 1
35 #else
36 #define HAVE_GETADDRINFO 0
37 #endif
38
39 #else
40 #include <netinet/in.h>
41 #include <netdb.h>
42 #include <arpa/inet.h>
43 #include <netinet/tcp.h>
44 #endif
45
46 #if HAVE_SYS_SOCKET_H
47 #include <sys/socket.h>
48 #endif
49 #if HAVE_SYS_WAIT_H
50 #include <sys/wait.h>
51 #endif
52
53 #if HAVE_OPENSSL_SSL_H
54 #include <openssl/ssl.h>
55 #include <openssl/err.h>
56 #endif
57
58 #include <yaz/comstack.h>
59 #include <yaz/tcpip.h>
60 #include <yaz/nmem.h>
61
62 static int tcpip_close(COMSTACK h);
63 static int tcpip_put(COMSTACK h, char *buf, int size);
64 static int tcpip_get(COMSTACK h, char **buf, int *bufsize);
65 static int tcpip_put_connect(COMSTACK h, char *buf, int size);
66 static int tcpip_get_connect(COMSTACK h, char **buf, int *bufsize);
67 static int tcpip_connect(COMSTACK h, void *address);
68 static int tcpip_more(COMSTACK h);
69 static int tcpip_rcvconnect(COMSTACK h);
70 static int tcpip_bind(COMSTACK h, void *address, int mode);
71 static int tcpip_listen(COMSTACK h, char *raddr, int *addrlen,
72                  int (*check_ip)(void *cd, const char *a, int len, int type),
73                  void *cd);
74 static int tcpip_set_blocking(COMSTACK p, int blocking);
75
76 #if HAVE_OPENSSL_SSL_H
77 static int ssl_get(COMSTACK h, char **buf, int *bufsize);
78 static int ssl_put(COMSTACK h, char *buf, int size);
79 #endif
80
81 static COMSTACK tcpip_accept(COMSTACK h);
82 static char *tcpip_addrstr(COMSTACK h);
83 static void *tcpip_straddr(COMSTACK h, const char *str);
84
85 #if 0
86 #define TRC(x) x
87 #else
88 #define TRC(X)
89 #endif
90
91 #ifndef YAZ_SOCKLEN_T
92 #define YAZ_SOCKLEN_T int
93 #endif
94
95 /* this state is used for both SSL and straight TCP/IP */
96 typedef struct tcpip_state
97 {
98     char *altbuf; /* alternate buffer for surplus data */
99     int altsize;  /* size as xmalloced */
100     int altlen;   /* length of data or 0 if none */
101
102     int written;  /* -1 if we aren't writing */
103     int towrite;  /* to verify against user input */
104     int (*complete)(const char *buf, int len); /* length/complete. */
105 #if HAVE_GETADDRINFO
106     struct addrinfo *ai;
107 #else
108     struct sockaddr_in addr;  /* returned by cs_straddr */
109 #endif
110     char buf[128]; /* returned by cs_addrstr */
111 #if HAVE_OPENSSL_SSL_H
112     SSL_CTX *ctx;       /* current CTX. */
113     SSL_CTX *ctx_alloc; /* If =ctx it is owned by CS. If 0 it is not owned */
114     SSL *ssl;
115     char cert_fname[256];
116 #endif
117     char *connect_request_buf;
118     int connect_request_len;
119     char *connect_response_buf;
120     int connect_response_len;
121 } tcpip_state;
122
123 #ifdef WIN32
124 static int tcpip_init (void)
125 {
126     static int initialized = 0;
127     if (!initialized)
128     {
129         WORD requested;
130         WSADATA wd;
131
132         requested = MAKEWORD(1, 1);
133         if (WSAStartup(requested, &wd))
134             return 0;
135         initialized = 1;
136     }
137     return 1;
138 }
139 #else
140 static int tcpip_init (void)
141 {
142     return 1;
143 }
144 #endif
145
146 /*
147  * This function is always called through the cs_create() macro.
148  * s >= 0: socket has already been established for us.
149  */
150 COMSTACK tcpip_type(int s, int flags, int protocol, void *vp)
151 {
152     COMSTACK p;
153     tcpip_state *sp;
154
155     if (!tcpip_init ())
156         return 0;
157     if (!(p = (struct comstack *)xmalloc(sizeof(struct comstack))))
158         return 0;
159     if (!(sp = (struct tcpip_state *)(p->cprivate =
160                                          xmalloc(sizeof(tcpip_state)))))
161         return 0;
162
163     p->flags = flags;
164
165     p->io_pending = 0;
166     p->iofile = s;
167     p->type = tcpip_type;
168     p->protocol = (enum oid_proto) protocol;
169
170     p->f_connect = tcpip_connect;
171     p->f_rcvconnect = tcpip_rcvconnect;
172     p->f_get = tcpip_get;
173     p->f_put = tcpip_put;
174     p->f_close = tcpip_close;
175     p->f_more = tcpip_more;
176     p->f_bind = tcpip_bind;
177     p->f_listen = tcpip_listen;
178     p->f_accept = tcpip_accept;
179     p->f_addrstr = tcpip_addrstr;
180     p->f_straddr = tcpip_straddr;
181     p->f_set_blocking = tcpip_set_blocking;
182     p->max_recv_bytes = 5000000;
183
184     p->state = s < 0 ? CS_ST_UNBND : CS_ST_IDLE; /* state of line */
185     p->event = CS_NONE;
186     p->cerrno = 0;
187     p->stackerr = 0;
188     p->user = 0;
189
190 #if HAVE_OPENSSL_SSL_H
191     sp->ctx = sp->ctx_alloc = 0;
192     sp->ssl = 0;
193     strcpy(sp->cert_fname, "yaz.pem");
194 #endif
195
196 #if HAVE_GETADDRINFO
197     sp->ai = 0;
198 #endif
199     sp->altbuf = 0;
200     sp->altsize = sp->altlen = 0;
201     sp->towrite = sp->written = -1;
202     if (protocol == PROTO_WAIS)
203         sp->complete = completeWAIS;
204     else
205         sp->complete = cs_complete_auto;
206
207     sp->connect_request_buf = 0;
208     sp->connect_request_len = 0;
209     sp->connect_response_buf = 0;
210     sp->connect_response_len = 0;
211
212     p->timeout = COMSTACK_DEFAULT_TIMEOUT;
213     TRC(fprintf(stderr, "Created new TCPIP comstack\n"));
214
215     return p;
216 }
217
218 COMSTACK yaz_tcpip_create(int s, int flags, int protocol,
219                           const char *connect_host)
220 {
221     COMSTACK p = tcpip_type(s, flags, protocol, 0);
222     if (!p)
223         return 0;
224     if (connect_host)
225     {
226         tcpip_state *sp = (tcpip_state *) p->cprivate;
227         sp->connect_request_buf = xmalloc(strlen(connect_host) + 30);
228         sprintf(sp->connect_request_buf, "CONNECT %s HTTP/1.0\r\n\r\n",
229                 connect_host);
230         sp->connect_request_len = strlen(sp->connect_request_buf);
231         p->f_put = tcpip_put_connect;
232         p->f_get = tcpip_get_connect;
233         sp->complete = cs_complete_auto_head; /* only want HTTP header */
234     }
235     return p;
236 }
237
238
239 #if HAVE_OPENSSL_SSL_H
240
241 COMSTACK ssl_type(int s, int flags, int protocol, void *vp)
242 {
243     tcpip_state *sp;
244     COMSTACK p;
245
246     p = tcpip_type (s, flags, protocol, 0);
247     if (!p)
248         return 0;
249     p->f_get = ssl_get;
250     p->f_put = ssl_put;
251     p->type = ssl_type;
252     sp = (tcpip_state *) p->cprivate;
253
254     sp->ctx = (SSL_CTX *) vp;  /* may be NULL */
255
256     /* note: we don't handle already opened socket in SSL mode - yet */
257     return p;
258 }
259 #endif
260
261 #if HAVE_GETADDRINFO
262 /* resolve using getaddrinfo */
263 struct addrinfo *tcpip_getaddrinfo(const char *str, const char *port)
264 {
265     struct addrinfo hints, *res;
266     int error;
267     char host[512], *p;
268
269     hints.ai_flags = 0;
270     hints.ai_family = AF_UNSPEC;
271     hints.ai_socktype = SOCK_STREAM;
272     hints.ai_protocol = 0;
273     hints.ai_addrlen        = 0;
274     hints.ai_addr           = NULL;
275     hints.ai_canonname      = NULL;
276     hints.ai_next           = NULL;
277
278     strncpy(host, str, sizeof(host)-1);
279     host[sizeof(host)-1] = 0;
280     if ((p = strchr(host, '/')))
281         *p = 0;
282     if ((p = strrchr(host, ':')))
283     {
284         *p = '\0';
285         port = p+1;
286     }
287
288     if (!strcmp("@", host))
289     {
290         hints.ai_flags = AI_PASSIVE;
291         error = getaddrinfo(0, port, &hints, &res);
292     }
293     else
294     {
295         error = getaddrinfo(host, port, &hints, &res);
296     }
297     if (error)
298         return 0;
299     return res;
300 }
301
302 #endif
303 /* gethostbyname .. old systems */
304 int tcpip_strtoaddr_ex(const char *str, struct sockaddr_in *add,
305                        int default_port)
306 {
307     struct hostent *hp;
308     char *p, buf[512];
309     short int port = default_port;
310 #ifdef WIN32
311     unsigned long tmpadd;
312 #else
313     in_addr_t tmpadd;
314 #endif
315     TRC(fprintf(stderr, "tcpip_strtoaddress: %s\n", str ? str : "NULL"));
316     add->sin_family = AF_INET;
317     strncpy(buf, str, sizeof(buf)-1);
318     buf[sizeof(buf)-1] = 0;
319     if ((p = strchr(buf, '/')))
320         *p = 0;
321     if ((p = strrchr(buf, ':')))
322     {
323         *p = 0;
324         port = atoi(p + 1);
325     }
326     add->sin_port = htons(port);
327     if (!strcmp("@", buf))
328     {
329         add->sin_addr.s_addr = INADDR_ANY;
330     }
331     else if ((tmpadd = inet_addr(buf)) != -1)
332     {
333         memcpy(&add->sin_addr.s_addr, &tmpadd, sizeof(struct in_addr));
334     }
335     else if ((hp = gethostbyname(buf)))
336     {
337         memcpy(&add->sin_addr.s_addr, *hp->h_addr_list,
338                sizeof(struct in_addr));
339     }
340     else
341         return 0;
342     return 1;
343 }
344
345
346 #if HAVE_GETADDRINFO
347 void *tcpip_straddr(COMSTACK h, const char *str)
348 {
349     tcpip_state *sp = (tcpip_state *)h->cprivate;
350     const char *port = "210";
351     if (h->protocol == PROTO_HTTP)
352         port = "80";
353     if (!tcpip_init ())
354         return 0;
355
356     if (sp->ai)
357         freeaddrinfo(sp->ai);
358     sp->ai = tcpip_getaddrinfo(str, port);
359     if (sp->ai && h->state == CS_ST_UNBND)
360     {
361         int s = -1;
362         struct addrinfo *ai = sp->ai;
363         for (; ai; ai = ai->ai_next)
364         {
365             s = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
366             if (s != -1)
367                 break;
368         }
369         if (s == -1)
370             return 0;
371         h->iofile = s;
372         
373         if (!tcpip_set_blocking(h, h->flags))
374             return 0;
375     }
376     return sp->ai;
377 }
378 #else
379 void *tcpip_straddr(COMSTACK h, const char *str)
380 {
381     tcpip_state *sp = (tcpip_state *)h->cprivate;
382     int port = 210;
383     if (h->protocol == PROTO_HTTP)
384         port = 80;
385
386     if (!tcpip_init ())
387         return 0;
388     if (!tcpip_strtoaddr_ex (str, &sp->addr, port))
389         return 0;
390     if (h->state == CS_ST_UNBND)
391     {
392         int s;
393         s = socket(AF_INET, SOCK_STREAM, 0);
394         if (s < 0)
395             return 0;
396         h->iofile = s;
397
398         if (!tcpip_set_blocking(h, h->flags))
399             return 0;
400     }
401     return &sp->addr;
402 }
403 #endif
404
405 int tcpip_more(COMSTACK h)
406 {
407     tcpip_state *sp = (tcpip_state *)h->cprivate;
408     
409     return sp->altlen && (*sp->complete)(sp->altbuf, sp->altlen);
410 }
411
412 /*
413  * connect(2) will block (sometimes) - nothing we can do short of doing
414  * weird things like spawning subprocesses or threading or some weird junk
415  * like that.
416  */
417 int tcpip_connect(COMSTACK h, void *address)
418 {
419 #if HAVE_GETADDRINFO
420     tcpip_state *sp = (tcpip_state *)h->cprivate;
421 #else
422     struct sockaddr_in *add = (struct sockaddr_in *) address;
423 #endif
424     int r;
425 #ifdef __sun__
426     int recbuflen;
427     YAZ_SOCKLEN_T rbufsize = sizeof(recbuflen);
428 #endif
429     TRC(fprintf(stderr, "tcpip_connect\n"));
430     h->io_pending = 0;
431     if (h->state != CS_ST_UNBND)
432     {
433         h->cerrno = CSOUTSTATE;
434         return -1;
435     }
436 #if HAVE_GETADDRINFO
437     if (sp->ai != (struct addrinfo *) address)
438     {
439         h->cerrno = CSOUTSTATE;
440         return -1;
441     }
442 #endif
443 #ifdef __sun__
444     /* On Suns, you must set a bigger Receive Buffer BEFORE a call to connect
445      * This gives the connect a chance to negotiate with the other side
446      * (see 'man tcp') 
447      */
448     if ( getsockopt(h->iofile, SOL_SOCKET, SO_RCVBUF, (void *)&recbuflen, &rbufsize ) < 0 )
449     {
450         h->cerrno = CSYSERR;
451         return -1;
452     }
453     TRC(fprintf( stderr, "Current Size of TCP Receive Buffer= %d\n",
454                  recbuflen ));
455     recbuflen *= 10; /* lets be optimistic */
456     if ( setsockopt(h->iofile, SOL_SOCKET, SO_RCVBUF, (void *)&recbuflen, rbufsize ) < 0 )
457     {
458         h->cerrno = CSYSERR;
459         return -1;
460     }
461     if ( getsockopt(h->iofile, SOL_SOCKET, SO_RCVBUF, (void *)&recbuflen, &rbufsize ) )
462     {
463         h->cerrno = CSYSERR;
464         return -1;
465     }
466     TRC(fprintf( stderr, "New Size of TCP Receive Buffer = %d\n",
467                  recbuflen ));
468 #endif
469
470 #if HAVE_GETADDRINFO
471     r = connect(h->iofile, sp->ai->ai_addr, sp->ai->ai_addrlen);
472     freeaddrinfo(sp->ai);
473     sp->ai = 0;
474 #else
475     r = connect(h->iofile, (struct sockaddr *) add, sizeof(*add));
476 #endif
477     if (r < 0)
478     {
479 #ifdef WIN32
480         if (WSAGetLastError() == WSAEWOULDBLOCK)
481         {
482             h->event = CS_CONNECT;
483             h->state = CS_ST_CONNECTING;
484             h->io_pending = CS_WANT_WRITE;
485             return 1;
486         }
487 #else
488         if (yaz_errno() == EINPROGRESS)
489         {
490             h->event = CS_CONNECT;
491             h->state = CS_ST_CONNECTING;
492             h->io_pending = CS_WANT_WRITE|CS_WANT_READ;
493             return 1;
494         }
495 #endif
496         h->cerrno = CSYSERR;
497         return -1;
498     }
499     h->event = CS_CONNECT;
500     h->state = CS_ST_CONNECTING;
501
502     return tcpip_rcvconnect (h);
503 }
504
505 /*
506  * nop
507  */
508 int tcpip_rcvconnect(COMSTACK h)
509 {
510 #if HAVE_OPENSSL_SSL_H
511     tcpip_state *sp = (tcpip_state *)h->cprivate;
512 #endif
513     TRC(fprintf(stderr, "tcpip_rcvconnect\n"));
514
515     if (h->state == CS_ST_DATAXFER)
516         return 0;
517     if (h->state != CS_ST_CONNECTING)
518     {
519         h->cerrno = CSOUTSTATE;
520         return -1;
521     }
522 #if HAVE_OPENSSL_SSL_H
523     if (h->type == ssl_type && !sp->ctx)
524     {
525         SSL_library_init();
526         SSL_load_error_strings();
527
528         sp->ctx = sp->ctx_alloc = SSL_CTX_new (SSLv23_method());
529         if (!sp->ctx)
530         {
531             h->cerrno = CSERRORSSL;
532             return -1;
533         }
534     }
535     if (sp->ctx)
536     {
537         int res;
538
539         if (!sp->ssl)
540         {
541             sp->ssl = SSL_new (sp->ctx);
542             SSL_set_fd (sp->ssl, h->iofile);
543         }
544         res = SSL_connect (sp->ssl);
545         if (res <= 0)
546         {
547             int err = SSL_get_error(sp->ssl, res);
548             if (err == SSL_ERROR_WANT_READ)
549             {
550                 h->io_pending = CS_WANT_READ;
551                 return 1;
552             }
553             if (err == SSL_ERROR_WANT_WRITE)
554             {
555                 h->io_pending = CS_WANT_WRITE;
556                 return 1;
557             }
558             h->cerrno = CSERRORSSL;
559             return -1;
560         }
561     }
562 #endif
563     h->event = CS_DATA;
564     h->state = CS_ST_DATAXFER;
565     return 0;
566 }
567
568 #define CERTF "ztest.pem"
569 #define KEYF "ztest.pem"
570
571 static void tcpip_setsockopt (int fd)
572 {
573 #if 0
574     int len = 4096;
575     int set = 1;
576     
577     if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char*)&set, sizeof(int)))
578     {
579         yaz_log(LOG_WARN|LOG_ERRNO, "setsockopt TCP_NODELAY");
580     }
581     if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, (char*)&len, sizeof(int)))
582     {
583         yaz_log(LOG_WARN|LOG_ERRNO, "setsockopt SNDBUF");
584     }
585     if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, (char*)&len, sizeof(int)))
586     {
587         yaz_log(LOG_WARN|LOG_ERRNO, "setsockopt RCVBUF");
588     }
589 #endif
590 }
591
592 static int tcpip_bind(COMSTACK h, void *address, int mode)
593 {
594     int r;
595     tcpip_state *sp = (tcpip_state *)h->cprivate;
596 #if HAVE_GETADDRINFO
597 #else
598     struct sockaddr *addr = (struct sockaddr *)address;
599 #endif
600 #ifdef WIN32
601     BOOL one = 1;
602 #else
603     int one = 1;
604 #endif
605
606 #if HAVE_GETADDRINFO
607     if (sp->ai != (struct addrinfo *) address)
608     {
609         h->cerrno = CSOUTSTATE;
610         return -1;
611     }
612 #endif
613
614 #if HAVE_OPENSSL_SSL_H
615     if (h->type == ssl_type && !sp->ctx)
616     {
617         SSL_library_init();
618         SSL_load_error_strings();
619
620         sp->ctx = sp->ctx_alloc = SSL_CTX_new (SSLv23_method());
621         if (!sp->ctx)
622         {
623             h->cerrno = CSERRORSSL;
624             return -1;
625         }
626     }
627     if (sp->ctx)
628     {
629         if (sp->ctx_alloc)
630         {
631             int res;
632             res = SSL_CTX_use_certificate_chain_file(sp->ctx, sp->cert_fname);
633             if (res <= 0)
634             {
635                 ERR_print_errors_fp(stderr);
636                 exit (2);
637             }
638             res = SSL_CTX_use_PrivateKey_file (sp->ctx, sp->cert_fname,
639                                                SSL_FILETYPE_PEM);
640             if (res <= 0)
641             {
642                 ERR_print_errors_fp(stderr);
643                 exit (3);
644             }
645             res = SSL_CTX_check_private_key (sp->ctx);
646             if (res <= 0)
647             {
648                 ERR_print_errors_fp(stderr);
649                 exit(5);
650             }
651         }
652         TRC (fprintf (stderr, "ssl_bind\n"));
653     }
654     else
655     {
656         TRC (fprintf (stderr, "tcpip_bind\n"));
657     }
658 #else
659     TRC (fprintf (stderr, "tcpip_bind\n"));
660 #endif
661 #ifndef WIN32
662     if (setsockopt(h->iofile, SOL_SOCKET, SO_REUSEADDR, (char*) 
663         &one, sizeof(one)) < 0)
664     {
665         h->cerrno = CSYSERR;
666         return -1;
667     }
668 #endif
669     tcpip_setsockopt(h->iofile);
670 #if HAVE_GETADDRINFO
671     r = bind(h->iofile, sp->ai->ai_addr, sp->ai->ai_addrlen);
672     freeaddrinfo(sp->ai);
673     sp->ai = 0;
674 #else
675     r = bind(h->iofile, addr, sizeof(struct sockaddr_in));
676 #endif
677     if (r)
678     {
679         h->cerrno = CSYSERR;
680         return -1;
681     }
682     /* Allow a maximum-sized backlog of waiting-to-connect clients */
683     if (mode == CS_SERVER && listen(h->iofile, SOMAXCONN) < 0)
684     {
685         h->cerrno = CSYSERR;
686         return -1;
687     }
688     h->state = CS_ST_IDLE;
689     h->event = CS_LISTEN;
690     return 0;
691 }
692
693 int tcpip_listen(COMSTACK h, char *raddr, int *addrlen,
694                  int (*check_ip)(void *cd, const char *a, int len, int t),
695                  void *cd)
696 {
697     struct sockaddr_in addr;
698     YAZ_SOCKLEN_T len = sizeof(addr);
699
700     TRC(fprintf(stderr, "tcpip_listen pid=%d\n", getpid()));
701     if (h->state != CS_ST_IDLE)
702     {
703         h->cerrno = CSOUTSTATE;
704         return -1;
705     }
706     h->newfd = accept(h->iofile, (struct sockaddr*)&addr, &len);
707     if (h->newfd < 0)
708     {
709         if (
710 #ifdef WIN32
711             WSAGetLastError() == WSAEWOULDBLOCK
712 #else
713             yaz_errno() == EWOULDBLOCK 
714 #ifdef EAGAIN
715 #if EAGAIN != EWOULDBLOCK
716             || yaz_errno() == EAGAIN
717 #endif
718 #endif
719 #endif
720             )
721             h->cerrno = CSNODATA;
722         else
723             h->cerrno = CSYSERR;
724         return -1;
725     }
726     if (addrlen && (size_t) (*addrlen) >= sizeof(struct sockaddr_in))
727         memcpy(raddr, &addr, *addrlen = sizeof(struct sockaddr_in));
728     else if (addrlen)
729         *addrlen = 0;
730     if (check_ip && (*check_ip)(cd, (const char *) &addr,
731         sizeof(addr), AF_INET))
732     {
733         h->cerrno = CSDENY;
734 #ifdef WIN32
735         closesocket(h->newfd);
736 #else
737         close(h->newfd);
738 #endif
739         h->newfd = -1;
740         return -1;
741     }
742     h->state = CS_ST_INCON;
743     tcpip_setsockopt (h->newfd);
744     return 0;
745 }
746
747 COMSTACK tcpip_accept(COMSTACK h)
748 {
749     COMSTACK cnew;
750     tcpip_state *state, *st = (tcpip_state *)h->cprivate;
751 #ifdef WIN32
752     unsigned long tru = 1;
753 #endif
754
755     TRC(fprintf(stderr, "tcpip_accept\n"));
756     if (h->state == CS_ST_INCON)
757     {
758         if (!(cnew = (COMSTACK)xmalloc(sizeof(*cnew))))
759         {
760             h->cerrno = CSYSERR;
761 #ifdef WIN32
762             closesocket(h->newfd);
763 #else
764             close(h->newfd);
765 #endif
766             h->newfd = -1;
767             return 0;
768         }
769         memcpy(cnew, h, sizeof(*h));
770         cnew->iofile = h->newfd;
771         cnew->io_pending = 0;
772         if (!(state = (tcpip_state *)
773               (cnew->cprivate = xmalloc(sizeof(tcpip_state)))))
774         {
775             h->cerrno = CSYSERR;
776             if (h->newfd != -1)
777             {
778 #ifdef WIN32
779                 closesocket(h->newfd);
780 #else
781                 close(h->newfd);
782 #endif
783                 h->newfd = -1;
784             }
785             return 0;
786         }
787         if (!tcpip_set_blocking(cnew, cnew->flags))
788         {
789             h->cerrno = CSYSERR;
790             if (h->newfd != -1)
791             {
792 #ifdef WIN32
793                 closesocket(h->newfd);
794 #else
795                 close(h->newfd);
796 #endif
797                 h->newfd = -1;
798             }
799             xfree (cnew);
800             xfree (state);
801             return 0;
802         }
803         h->newfd = -1;
804         state->altbuf = 0;
805         state->altsize = state->altlen = 0;
806         state->towrite = state->written = -1;
807         state->complete = st->complete;
808 #if HAVE_GETADDRINFO
809         state->ai = 0;
810 #endif
811         cnew->state = CS_ST_ACCEPT;
812         h->state = CS_ST_IDLE;
813         
814 #if HAVE_OPENSSL_SSL_H
815         state->ctx = st->ctx;
816         state->ctx_alloc = 0;
817         state->ssl = st->ssl;
818         if (state->ctx)
819         {
820             state->ssl = SSL_new (state->ctx);
821             SSL_set_fd (state->ssl, cnew->iofile);
822         }
823         state->connect_request_buf = 0;
824         state->connect_response_buf = 0;
825 #endif
826         h = cnew;
827     }
828     if (h->state == CS_ST_ACCEPT)
829     {
830 #if HAVE_OPENSSL_SSL_H
831         tcpip_state *state = (tcpip_state *)h->cprivate;
832         if (state->ctx)
833         {
834             int res = SSL_accept (state->ssl);
835             TRC(fprintf(stderr, "SSL_accept\n"));
836             if (res <= 0)
837             {
838                 int err = SSL_get_error(state->ssl, res);
839                 if (err == SSL_ERROR_WANT_READ)
840                 {
841                     h->io_pending = CS_WANT_READ;
842                     return h;
843                 }
844                 if (err == SSL_ERROR_WANT_WRITE)
845                 {
846                     h->io_pending = CS_WANT_WRITE;
847                     return h;
848                 }
849                 cs_close (h);
850                 return 0;
851             }
852         }
853 #endif
854     }
855     else
856     {
857         h->cerrno = CSOUTSTATE;
858         return 0;
859     }
860     h->io_pending = 0;
861     h->state = CS_ST_DATAXFER;
862     h->event = CS_DATA;
863     return h;
864 }
865
866 #define CS_TCPIP_BUFCHUNK 4096
867
868 /*
869  * Return: -1 error, >1 good, len of buffer, ==1 incomplete buffer,
870  * 0=connection closed.
871  */
872 int tcpip_get(COMSTACK h, char **buf, int *bufsize)
873 {
874     tcpip_state *sp = (tcpip_state *)h->cprivate;
875     char *tmpc;
876     int tmpi, berlen, rest, req, tomove;
877     int hasread = 0, res;
878
879     TRC(fprintf(stderr, "tcpip_get: bufsize=%d\n", *bufsize));
880     if (sp->altlen) /* switch buffers */
881     {
882         TRC(fprintf(stderr, "  %d bytes in altbuf (0x%x)\n", sp->altlen,
883             (unsigned) sp->altbuf));
884         tmpc = *buf;
885         tmpi = *bufsize;
886         *buf = sp->altbuf;
887         *bufsize = sp->altsize;
888         hasread = sp->altlen;
889         sp->altlen = 0;
890         sp->altbuf = tmpc;
891         sp->altsize = tmpi;
892     }
893     h->io_pending = 0;
894     while (!(berlen = (*sp->complete)(*buf, hasread)))
895     {
896         if (!*bufsize)
897         {
898             if (!(*buf = (char *)xmalloc(*bufsize = CS_TCPIP_BUFCHUNK)))
899             {
900                 h->cerrno = CSYSERR;
901                 return -1;
902             }
903         }
904         else if (*bufsize - hasread < CS_TCPIP_BUFCHUNK)
905             if (!(*buf =(char *)xrealloc(*buf, *bufsize *= 2)))
906             {
907                 h->cerrno = CSYSERR;
908                 return -1;
909             }
910 #ifdef __sun__
911         yaz_set_errno( 0 );
912         /* unfortunatly, sun sometimes forgets to set errno in recv
913            when EWOULDBLOCK etc. would be required (res = -1) */
914 #endif
915         res = recv(h->iofile, *buf + hasread, CS_TCPIP_BUFCHUNK, 0);
916         TRC(fprintf(stderr, "  recv res=%d, hasread=%d\n", res, hasread));
917         if (res < 0)
918         {
919             TRC(fprintf(stderr, "  recv errno=%d, (%s)\n", yaz_errno(), 
920                       strerror(yaz_errno())));
921 #ifdef WIN32
922             if (WSAGetLastError() == WSAEWOULDBLOCK)
923             {
924                 h->io_pending = CS_WANT_READ;
925                 break;
926             }
927             else
928             {
929                 h->cerrno = CSYSERR;
930                 return -1;
931             }
932 #else
933             if (yaz_errno() == EWOULDBLOCK 
934 #ifdef EAGAIN   
935 #if EAGAIN != EWOULDBLOCK
936                 || yaz_errno() == EAGAIN
937 #endif
938 #endif
939                 || yaz_errno() == EINPROGRESS
940 #ifdef __sun__
941                 || yaz_errno() == ENOENT /* Sun's sometimes set errno to this */
942 #endif
943                 )
944             {
945                 h->io_pending = CS_WANT_READ;
946                 break;
947             }
948             else if (yaz_errno() == 0)
949                 continue;
950             else
951             {
952                 h->cerrno = CSYSERR;
953                 return -1;
954             }
955 #endif
956         }
957         else if (!res)
958             return hasread;
959         hasread += res;
960         if (hasread > h->max_recv_bytes)
961         {
962             h->cerrno = CSBUFSIZE;
963             return -1;
964         }
965     }
966     TRC (fprintf (stderr, "  Out of read loop with hasread=%d, berlen=%d\n",
967                   hasread, berlen));
968     /* move surplus buffer (or everything if we didn't get a BER rec.) */
969     if (hasread > berlen)
970     {
971         tomove = req = hasread - berlen;
972         rest = tomove % CS_TCPIP_BUFCHUNK;
973         if (rest)
974             req += CS_TCPIP_BUFCHUNK - rest;
975         if (!sp->altbuf)
976         {
977             if (!(sp->altbuf = (char *)xmalloc(sp->altsize = req)))
978             {
979                 h->cerrno = CSYSERR;
980                 return -1;
981             }
982         } else if (sp->altsize < req)
983             if (!(sp->altbuf =(char *)xrealloc(sp->altbuf, sp->altsize = req)))
984             {
985                 h->cerrno = CSYSERR;
986                 return -1;
987             }
988         TRC(fprintf(stderr, "  Moving %d bytes to altbuf(0x%x)\n", tomove,
989             (unsigned) sp->altbuf));
990         memcpy(sp->altbuf, *buf + berlen, sp->altlen = tomove);
991     }
992     if (berlen < CS_TCPIP_BUFCHUNK - 1)
993         *(*buf + berlen) = '\0';
994     return berlen ? berlen : 1;
995 }
996
997
998 #if HAVE_OPENSSL_SSL_H
999 /*
1000  * Return: -1 error, >1 good, len of buffer, ==1 incomplete buffer,
1001  * 0=connection closed.
1002  */
1003 int ssl_get(COMSTACK h, char **buf, int *bufsize)
1004 {
1005     tcpip_state *sp = (tcpip_state *)h->cprivate;
1006     char *tmpc;
1007     int tmpi, berlen, rest, req, tomove;
1008     int hasread = 0, res;
1009
1010     TRC(fprintf(stderr, "ssl_get: bufsize=%d\n", *bufsize));
1011     if (sp->altlen) /* switch buffers */
1012     {
1013         TRC(fprintf(stderr, "  %d bytes in altbuf (0x%x)\n", sp->altlen,
1014             (unsigned) sp->altbuf));
1015         tmpc = *buf;
1016         tmpi = *bufsize;
1017         *buf = sp->altbuf;
1018         *bufsize = sp->altsize;
1019         hasread = sp->altlen;
1020         sp->altlen = 0;
1021         sp->altbuf = tmpc;
1022         sp->altsize = tmpi;
1023     }
1024     h->io_pending = 0;
1025     while (!(berlen = (*sp->complete)(*buf, hasread)))
1026     {
1027         if (!*bufsize)
1028         {
1029             if (!(*buf = (char *)xmalloc(*bufsize = CS_TCPIP_BUFCHUNK)))
1030                 return -1;
1031         }
1032         else if (*bufsize - hasread < CS_TCPIP_BUFCHUNK)
1033             if (!(*buf =(char *)xrealloc(*buf, *bufsize *= 2)))
1034                 return -1;
1035         res = SSL_read (sp->ssl, *buf + hasread, CS_TCPIP_BUFCHUNK);
1036         TRC(fprintf(stderr, "  SSL_read res=%d, hasread=%d\n", res, hasread));
1037         if (res <= 0)
1038         {
1039             int ssl_err = SSL_get_error(sp->ssl, res);
1040             if (ssl_err == SSL_ERROR_WANT_READ)
1041             {
1042                 h->io_pending = CS_WANT_READ;
1043                 break;
1044             }
1045             if (ssl_err == SSL_ERROR_WANT_WRITE)
1046             {
1047                 h->io_pending = CS_WANT_WRITE;
1048                 break;
1049             }
1050             if (res == 0)
1051                 return 0;
1052             h->cerrno = CSERRORSSL;
1053             return -1;
1054         }
1055         hasread += res;
1056     }
1057     TRC (fprintf (stderr, "  Out of read loop with hasread=%d, berlen=%d\n",
1058         hasread, berlen));
1059     /* move surplus buffer (or everything if we didn't get a BER rec.) */
1060     if (hasread > berlen)
1061     {
1062         tomove = req = hasread - berlen;
1063         rest = tomove % CS_TCPIP_BUFCHUNK;
1064         if (rest)
1065             req += CS_TCPIP_BUFCHUNK - rest;
1066         if (!sp->altbuf)
1067         {
1068             if (!(sp->altbuf = (char *)xmalloc(sp->altsize = req)))
1069                 return -1;
1070         } else if (sp->altsize < req)
1071             if (!(sp->altbuf =(char *)xrealloc(sp->altbuf, sp->altsize = req)))
1072                 return -1;
1073         TRC(fprintf(stderr, "  Moving %d bytes to altbuf(0x%x)\n", tomove,
1074             (unsigned) sp->altbuf));
1075         memcpy(sp->altbuf, *buf + berlen, sp->altlen = tomove);
1076     }
1077     if (berlen < CS_TCPIP_BUFCHUNK - 1)
1078         *(*buf + berlen) = '\0';
1079     return berlen ? berlen : 1;
1080 }
1081 #endif
1082
1083 /*
1084  * Returns 1, 0 or -1
1085  * In nonblocking mode, you must call again with same buffer while
1086  * return value is 1.
1087  */
1088 int tcpip_put(COMSTACK h, char *buf, int size)
1089 {
1090     int res;
1091     struct tcpip_state *state = (struct tcpip_state *)h->cprivate;
1092
1093     TRC(fprintf(stderr, "tcpip_put: size=%d\n", size));
1094     h->io_pending = 0;
1095     h->event = CS_DATA;
1096     if (state->towrite < 0)
1097     {
1098         state->towrite = size;
1099         state->written = 0;
1100     }
1101     else if (state->towrite != size)
1102     {
1103         h->cerrno = CSWRONGBUF;
1104         return -1;
1105     }
1106     while (state->towrite > state->written)
1107     {
1108         if ((res =
1109              send(h->iofile, buf + state->written, size -
1110                   state->written, 
1111 #ifdef MSG_NOSIGNAL
1112                   MSG_NOSIGNAL
1113 #else
1114                   0
1115 #endif
1116                  )) < 0)
1117         {
1118             if (
1119 #ifdef WIN32
1120                 WSAGetLastError() == WSAEWOULDBLOCK
1121 #else
1122                 yaz_errno() == EWOULDBLOCK 
1123 #ifdef EAGAIN
1124 #if EAGAIN != EWOULDBLOCK
1125              || yaz_errno() == EAGAIN
1126 #endif
1127 #endif
1128 #ifdef __sun__
1129                 || yaz_errno() == ENOENT /* Sun's sometimes set errno to this value! */
1130 #endif
1131                 || yaz_errno() == EINPROGRESS
1132 #endif
1133                 )
1134             {
1135                 TRC(fprintf(stderr, "  Flow control stop\n"));
1136                 h->io_pending = CS_WANT_WRITE;
1137                 return 1;
1138             }
1139             h->cerrno = CSYSERR;
1140             return -1;
1141         }
1142         state->written += res;
1143         TRC(fprintf(stderr, "  Wrote %d, written=%d, nbytes=%d\n",
1144                     res, state->written, size));
1145     }
1146     state->towrite = state->written = -1;
1147     TRC(fprintf(stderr, "  Ok\n"));
1148     return 0;
1149 }
1150
1151
1152 #if HAVE_OPENSSL_SSL_H
1153 /*
1154  * Returns 1, 0 or -1
1155  * In nonblocking mode, you must call again with same buffer while
1156  * return value is 1.
1157  */
1158 int ssl_put(COMSTACK h, char *buf, int size)
1159 {
1160     int res;
1161     struct tcpip_state *state = (struct tcpip_state *)h->cprivate;
1162
1163     TRC(fprintf(stderr, "ssl_put: size=%d\n", size));
1164     h->io_pending = 0;
1165     h->event = CS_DATA;
1166     if (state->towrite < 0)
1167     {
1168         state->towrite = size;
1169         state->written = 0;
1170     }
1171     else if (state->towrite != size)
1172     {
1173         h->cerrno = CSWRONGBUF;
1174         return -1;
1175     }
1176     while (state->towrite > state->written)
1177     {
1178         res = SSL_write (state->ssl, buf + state->written,
1179                          size - state->written);
1180         if (res <= 0)
1181         {
1182             int ssl_err = SSL_get_error(state->ssl, res);
1183             if (ssl_err == SSL_ERROR_WANT_READ)
1184             {
1185                 h->io_pending = CS_WANT_READ;
1186                 return 1;
1187             }
1188             if (ssl_err == SSL_ERROR_WANT_WRITE)
1189             {
1190                 h->io_pending = CS_WANT_WRITE;
1191                 return 1;
1192             }
1193             h->cerrno = CSERRORSSL;
1194             return -1;
1195         }
1196         state->written += res;
1197         TRC(fprintf(stderr, "  Wrote %d, written=%d, nbytes=%d\n",
1198                     res, state->written, size));
1199     }
1200     state->towrite = state->written = -1;
1201     TRC(fprintf(stderr, "  Ok\n"));
1202     return 0;
1203 }
1204 #endif
1205
1206 int tcpip_close(COMSTACK h)
1207 {
1208     tcpip_state *sp = (struct tcpip_state *)h->cprivate;
1209
1210     TRC(fprintf(stderr, "tcpip_close\n"));
1211     if (h->iofile != -1)
1212     {
1213 #if HAVE_OPENSSL_SSL_H
1214         if (sp->ssl)
1215         {
1216             SSL_shutdown (sp->ssl);
1217         }
1218 #endif
1219 #ifdef WIN32
1220         closesocket(h->iofile);
1221 #else
1222         close(h->iofile);
1223 #endif
1224     }
1225     if (sp->altbuf)
1226         xfree(sp->altbuf);
1227 #if HAVE_OPENSSL_SSL_H
1228     if (sp->ssl)
1229     {
1230         TRC (fprintf(stderr, "SSL_free\n"));
1231         SSL_free (sp->ssl);
1232     }
1233     sp->ssl = 0;
1234     if (sp->ctx_alloc)
1235         SSL_CTX_free (sp->ctx_alloc);
1236 #endif
1237 #if HAVE_GETADDRINFO
1238     if (sp->ai)
1239         freeaddrinfo(sp->ai);
1240 #endif
1241     xfree(sp->connect_request_buf);
1242     xfree(sp->connect_response_buf);
1243     xfree(sp);
1244     xfree(h);
1245     return 0;
1246 }
1247
1248 char *tcpip_addrstr(COMSTACK h)
1249 {
1250     tcpip_state *sp = (struct tcpip_state *)h->cprivate;
1251     char *r = 0, *buf = sp->buf;
1252
1253 #if HAVE_GETADDRINFO
1254     char host[120];
1255     struct sockaddr_storage addr;
1256     YAZ_SOCKLEN_T len = sizeof(addr);
1257     
1258     if (getpeername(h->iofile, (struct sockaddr *)&addr, &len) < 0)
1259     {
1260         h->cerrno = CSYSERR;
1261         return 0;
1262     }
1263     if (getnameinfo((struct sockaddr *) &addr, len, host, sizeof(host)-1, 
1264                     0, 0, 
1265                     (h->flags & CS_FLAGS_NUMERICHOST) ? NI_NUMERICHOST : 0))
1266     {
1267         r = "unknown";
1268     }
1269     else
1270         r = host;
1271     
1272 #else
1273
1274     struct sockaddr_in addr;
1275     YAZ_SOCKLEN_T len = sizeof(addr);
1276     struct hostent *host;
1277     
1278     if (getpeername(h->iofile, (struct sockaddr*) &addr, &len) < 0)
1279     {
1280         h->cerrno = CSYSERR;
1281         return 0;
1282     }
1283     if (!(h->flags & CS_FLAGS_NUMERICHOST))
1284     {
1285         if ((host = gethostbyaddr((char*)&addr.sin_addr,
1286                                   sizeof(addr.sin_addr),
1287                                   AF_INET)))
1288             r = (char*) host->h_name;
1289     }
1290     if (!r)
1291         r = inet_ntoa(addr.sin_addr);        
1292 #endif
1293
1294     if (h->protocol == PROTO_HTTP)
1295         sprintf(buf, "http:%s", r);
1296     else
1297         sprintf(buf, "tcp:%s", r);
1298 #if HAVE_OPENSSL_SSL_H
1299     if (sp->ctx)
1300     {
1301         if (h->protocol == PROTO_HTTP)
1302             sprintf(buf, "https:%s", r);
1303         else
1304             sprintf(buf, "ssl:%s", r);
1305     }
1306 #endif
1307     return buf;
1308 }
1309
1310 int static tcpip_set_blocking(COMSTACK p, int flags)
1311 {
1312     unsigned long flag;
1313     
1314 #ifdef WIN32
1315     flag = (flags & CS_FLAGS_BLOCKING) ? 0 : 1;
1316     if (ioctlsocket(p->iofile, FIONBIO, &flag) < 0)
1317         return 0;
1318 #else
1319     flag = fcntl(p->iofile, F_GETFL, 0);
1320     if (flags & CS_FLAGS_BLOCKING)
1321         flag = flag & ~O_NONBLOCK;  /* blocking */
1322     else
1323     {
1324         flag = flag | O_NONBLOCK;   /* non-blocking */
1325         signal(SIGPIPE, SIG_IGN);
1326     }
1327     if (fcntl(p->iofile, F_SETFL, flag) < 0)
1328         return 0;
1329 #endif
1330     p->flags = flags;
1331     return 1;
1332 }
1333
1334 #if HAVE_OPENSSL_SSL_H
1335 int cs_set_ssl_ctx(COMSTACK cs, void *ctx)
1336 {
1337     struct tcpip_state *sp;
1338     if (!cs || cs->type != ssl_type)
1339         return 0;
1340     sp = (struct tcpip_state *) cs->cprivate;
1341     if (sp->ctx_alloc)
1342         return 0;
1343     sp->ctx = (SSL_CTX *) ctx;
1344     return 1;
1345 }
1346
1347 void *cs_get_ssl(COMSTACK cs)
1348 {
1349     struct tcpip_state *sp;
1350     if (!cs || cs->type != ssl_type)
1351         return 0;
1352     sp = (struct tcpip_state *) cs->cprivate;
1353     return sp->ssl;  
1354 }
1355
1356 int cs_set_ssl_certificate_file(COMSTACK cs, const char *fname)
1357 {
1358     struct tcpip_state *sp;
1359     if (!cs || cs->type != ssl_type)
1360         return 0;
1361     sp = (struct tcpip_state *) cs->cprivate;
1362     strncpy(sp->cert_fname, fname, sizeof(sp->cert_fname)-1);
1363     sp->cert_fname[sizeof(sp->cert_fname)-1] = '\0';
1364     return 1;
1365 }
1366
1367 int cs_get_peer_certificate_x509(COMSTACK cs, char **buf, int *len)
1368 {
1369     SSL *ssl = (SSL *) cs_get_ssl(cs);
1370     if (ssl)
1371     {
1372         X509 *server_cert = SSL_get_peer_certificate (ssl);
1373         if (server_cert)
1374         {
1375             BIO *bio = BIO_new(BIO_s_mem());
1376             char *pem_buf;
1377             /* get PEM buffer in memory */
1378             PEM_write_bio_X509(bio, server_cert);
1379             *len = BIO_get_mem_data(bio, &pem_buf);
1380             *buf = (char *) xmalloc(*len);
1381             memcpy(*buf, pem_buf, *len);
1382             BIO_free(bio);
1383             return 1;
1384         }
1385     }
1386     return 0;
1387 }
1388 #else
1389 int cs_set_ssl_ctx(COMSTACK cs, void *ctx)
1390 {
1391     return 0;
1392 }
1393
1394 void *cs_get_ssl(COMSTACK cs)
1395 {
1396     return 0;
1397 }
1398
1399 int cs_get_peer_certificate_x509(COMSTACK cs, char **buf, int *len)
1400 {
1401     return 0;
1402 }
1403
1404 int cs_set_ssl_certificate_file(COMSTACK cs, const char *fname)
1405 {
1406     return 0;
1407 }
1408 #endif
1409
1410
1411 static int tcpip_put_connect(COMSTACK h, char *buf, int size)
1412 {
1413     struct tcpip_state *state = (struct tcpip_state *)h->cprivate;
1414
1415     int r = tcpip_put(h, state->connect_request_buf,
1416                       state->connect_request_len);
1417     if (r == 0)
1418     {
1419         /* it's sent */
1420         h->f_put = tcpip_put; /* switch to normal tcpip put */
1421         r = tcpip_put(h, buf, size);
1422     }
1423     return r;
1424 }
1425
1426 static int tcpip_get_connect(COMSTACK h, char **buf, int *bufsize)
1427 {
1428     struct tcpip_state *state = (struct tcpip_state *)h->cprivate;
1429     int r;
1430
1431     r = tcpip_get(h, &state->connect_response_buf, 
1432                   &state->connect_response_len);
1433     if (r < 1)
1434         return r;
1435     /* got the connect response completely */
1436     state->complete = cs_complete_auto; /* switch to normal tcpip get */
1437     h->f_get = tcpip_get;
1438     return tcpip_get(h, buf, bufsize);
1439 }
1440
1441
1442 /*
1443  * Local variables:
1444  * c-basic-offset: 4
1445  * indent-tabs-mode: nil
1446  * End:
1447  * vim: shiftwidth=4 tabstop=8 expandtab
1448  */
1449