Deal with generated files.
[openssl.git] / crypto / bio / b_sock.c
1 /* crypto/bio/b_sock.c */
2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
3  * All rights reserved.
4  *
5  * This package is an SSL implementation written
6  * by Eric Young (eay@cryptsoft.com).
7  * The implementation was written so as to conform with Netscapes SSL.
8  * 
9  * This library is free for commercial and non-commercial use as long as
10  * the following conditions are aheared to.  The following conditions
11  * apply to all code found in this distribution, be it the RC4, RSA,
12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
13  * included with this distribution is covered by the same copyright terms
14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
15  * 
16  * Copyright remains Eric Young's, and as such any Copyright notices in
17  * the code are not to be removed.
18  * If this package is used in a product, Eric Young should be given attribution
19  * as the author of the parts of the library used.
20  * This can be in the form of a textual message at program startup or
21  * in documentation (online or textual) provided with the package.
22  * 
23  * Redistribution and use in source and binary forms, with or without
24  * modification, are permitted provided that the following conditions
25  * are met:
26  * 1. Redistributions of source code must retain the copyright
27  *    notice, this list of conditions and the following disclaimer.
28  * 2. Redistributions in binary form must reproduce the above copyright
29  *    notice, this list of conditions and the following disclaimer in the
30  *    documentation and/or other materials provided with the distribution.
31  * 3. All advertising materials mentioning features or use of this software
32  *    must display the following acknowledgement:
33  *    "This product includes cryptographic software written by
34  *     Eric Young (eay@cryptsoft.com)"
35  *    The word 'cryptographic' can be left out if the rouines from the library
36  *    being used are not cryptographic related :-).
37  * 4. If you include any Windows specific code (or a derivative thereof) from 
38  *    the apps directory (application code) you must include an acknowledgement:
39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
40  * 
41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
51  * SUCH DAMAGE.
52  * 
53  * The licence and distribution terms for any publically available version or
54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
55  * copied and put under another distribution licence
56  * [including the GNU Public Licence.]
57  */
58
59 #ifndef NO_SOCK
60
61 #include <stdio.h>
62 #include <stdlib.h>
63 #include <errno.h>
64 #define USE_SOCKETS
65 #include "cryptlib.h"
66 #include "bio.h"
67
68 /*      BIOerr(BIO_F_WSASTARTUP,BIO_R_WSASTARTUP ); */
69
70 #ifdef WIN16
71 #define SOCKET_PROTOCOL 0 /* more microsoft stupidity */
72 #else
73 #define SOCKET_PROTOCOL IPPROTO_TCP
74 #endif
75
76 #ifdef SO_MAXCONN
77 #define MAX_LISTEN  SOMAXCONN
78 #elif defined(SO_MAXCONN)
79 #define MAX_LISTEN  SO_MAXCONN
80 #else
81 #define MAX_LISTEN  32
82 #endif
83
84 #ifdef WINDOWS
85 static int wsa_init_done=0;
86 #endif
87
88 static unsigned long BIO_ghbn_hits=0L;
89 static unsigned long BIO_ghbn_miss=0L;
90
91 #define GHBN_NUM        4
92 static struct ghbn_cache_st
93         {
94         char name[129];
95         struct hostent *ent;
96         unsigned long order;
97         } ghbn_cache[GHBN_NUM];
98
99 #ifndef NOPROTO
100 static int get_ip(char *str,unsigned char *ip);
101 static void ghbn_free(struct hostent *a);
102 static struct hostent *ghbn_dup(struct hostent *a);
103 #else
104 static int get_ip();
105 static void ghbn_free();
106 static struct hostent *ghbn_dup();
107 #endif
108
109 int BIO_get_host_ip(str,ip)
110 char *str;
111 unsigned char *ip;
112         {
113         int i;
114         struct hostent *he;
115
116         i=get_ip(str,ip);
117         if (i > 0) return(1);
118         if (i < 0)
119                 {
120                 BIOerr(BIO_F_BIO_GET_HOST_IP,BIO_R_INVALID_IP_ADDRESS);
121                 ERR_add_error_data(2,"host=",str);
122                 return(0);
123                 }
124         else
125                 { /* do a gethostbyname */
126                 if (!BIO_sock_init()) return(0);
127
128                 he=BIO_gethostbyname(str);
129                 if (he == NULL)
130                         {
131                         BIOerr(BIO_F_BIO_GET_HOST_IP,BIO_R_BAD_HOSTNAME_LOOKUP);
132                         ERR_add_error_data(2,"host=",str);
133                         return(0);
134                         }
135
136                 /* cast to short because of win16 winsock definition */
137                 if ((short)he->h_addrtype != AF_INET)
138                         {
139                         BIOerr(BIO_F_BIO_GET_HOST_IP,BIO_R_GETHOSTBYNAME_ADDR_IS_NOT_AF_INET);
140                         ERR_add_error_data(2,"host=",str);
141                         return(0);
142                         }
143                 for (i=0; i<4; i++)
144                         ip[i]=he->h_addr_list[0][i];
145                 }
146         return(1);
147         }
148
149 int BIO_get_port(str,port_ptr)
150 char *str;
151 unsigned short *port_ptr;
152         {
153         int i;
154         struct servent *s;
155
156         if (str == NULL)
157                 {
158                 BIOerr(BIO_F_BIO_GET_PORT,BIO_R_NO_PORT_DEFINED);
159                 return(0);
160                 }
161         i=atoi(str);
162         if (i != 0)
163                 *port_ptr=(unsigned short)i;
164         else
165                 {
166                 s=getservbyname(str,"tcp");
167                 if (s == NULL)
168                         {
169                         if (strcmp(str,"http") == 0)
170                                 *port_ptr=80;
171                         else if (strcmp(str,"telnet") == 0)
172                                 *port_ptr=23;
173                         else if (strcmp(str,"socks") == 0)
174                                 *port_ptr=1080;
175                         else if (strcmp(str,"https") == 0)
176                                 *port_ptr=443;
177                         else if (strcmp(str,"ssl") == 0)
178                                 *port_ptr=443;
179                         else if (strcmp(str,"ftp") == 0)
180                                 *port_ptr=21;
181                         else if (strcmp(str,"gopher") == 0)
182                                 *port_ptr=70;
183 #if 0
184                         else if (strcmp(str,"wais") == 0)
185                                 *port_ptr=21;
186 #endif
187                         else
188                                 {
189                                 SYSerr(SYS_F_GETSERVBYNAME,get_last_socket_error());
190                                 ERR_add_error_data(3,"service='",str,"'");
191                                 return(0);
192                                 }
193                         return(1);
194                         }
195                 *port_ptr=htons((unsigned short)s->s_port);
196                 }
197         return(1);
198         }
199
200 int BIO_sock_error(sock)
201 int sock;
202         {
203         int j,i,size;
204                  
205         size=sizeof(int);
206
207         i=getsockopt(sock,SOL_SOCKET,SO_ERROR,(char *)&j,&size);
208         if (i < 0)
209                 return(1);
210         else
211                 return(j);
212         }
213
214 long BIO_ghbn_ctrl(cmd,iarg,parg)
215 int cmd;
216 int iarg;
217 char *parg;
218         {
219         int i;
220         char **p;
221
222         switch (cmd)
223                 {
224         case BIO_GHBN_CTRL_HITS:
225                 return(BIO_ghbn_hits);
226                 /* break; */
227         case BIO_GHBN_CTRL_MISSES:
228                 return(BIO_ghbn_miss);
229                 /* break; */
230         case BIO_GHBN_CTRL_CACHE_SIZE:
231                 return(GHBN_NUM);
232                 /* break; */
233         case BIO_GHBN_CTRL_GET_ENTRY:
234                 if ((iarg >= 0) && (iarg <GHBN_NUM) &&
235                         (ghbn_cache[iarg].order > 0))
236                         {
237                         p=(char **)parg;
238                         if (p == NULL) return(0);
239                         *p=ghbn_cache[iarg].name;
240                         ghbn_cache[iarg].name[128]='\0';
241                         return(1);
242                         }
243                 return(0);
244                 /* break; */
245         case BIO_GHBN_CTRL_FLUSH:
246                 for (i=0; i<GHBN_NUM; i++)
247                         ghbn_cache[i].order=0;
248                 break;
249         default:
250                 return(0);
251                 }
252         return(1);
253         }
254
255 static struct hostent *ghbn_dup(a)
256 struct hostent *a;
257         {
258         struct hostent *ret;
259         int i,j;
260
261         MemCheck_off();
262         ret=(struct hostent *)Malloc(sizeof(struct hostent));
263         if (ret == NULL) return(NULL);
264         memset(ret,0,sizeof(struct hostent));
265
266         for (i=0; a->h_aliases[i] != NULL; i++)
267                 ;
268         i++;
269         ret->h_aliases=(char **)Malloc(sizeof(char *)*i);
270         memset(ret->h_aliases,0,sizeof(char *)*i);
271         if (ret == NULL) goto err;
272
273         for (i=0; a->h_addr_list[i] != NULL; i++)
274                 ;
275         i++;
276         ret->h_addr_list=(char **)Malloc(sizeof(char *)*i);
277         memset(ret->h_addr_list,0,sizeof(char *)*i);
278         if (ret->h_addr_list == NULL) goto err;
279
280         j=strlen(a->h_name)+1;
281         if ((ret->h_name=Malloc(j)) == NULL) goto err;
282         memcpy((char *)ret->h_name,a->h_name,j+1);
283         for (i=0; a->h_aliases[i] != NULL; i++)
284                 {
285                 j=strlen(a->h_aliases[i])+1;
286                 if ((ret->h_aliases[i]=Malloc(j)) == NULL) goto err;
287                 memcpy(ret->h_aliases[i],a->h_aliases[i],j+1);
288                 }
289         ret->h_length=a->h_length;
290         ret->h_addrtype=a->h_addrtype;
291         for (i=0; a->h_addr_list[i] != NULL; i++)
292                 {
293                 if ((ret->h_addr_list[i]=Malloc(a->h_length)) == NULL)
294                         goto err;
295                 memcpy(ret->h_addr_list[i],a->h_addr_list[i],a->h_length);
296                 }
297         if (0)
298                 {
299 err:    
300                 if (ret != NULL)
301                         ghbn_free(ret);
302                 ret=NULL;
303                 }
304         MemCheck_on();
305         return(ret);
306         }
307
308 static void ghbn_free(a)
309 struct hostent *a;
310         {
311         int i;
312
313         if (a->h_aliases != NULL)
314                 {
315                 for (i=0; a->h_aliases[i] != NULL; i++)
316                         Free(a->h_aliases[i]);
317                 Free(a->h_aliases);
318                 }
319         if (a->h_addr_list != NULL)
320                 {
321                 for (i=0; a->h_addr_list[i] != NULL; i++)
322                         Free(a->h_addr_list[i]);
323                 Free(a->h_addr_list);
324                 }
325         if (a->h_name != NULL) Free((char *)a->h_name);
326         Free(a);
327         }
328
329 struct hostent *BIO_gethostbyname(name)
330 char *name;
331         {
332         struct hostent *ret;
333         int i,lowi=0,j;
334         unsigned long low= (unsigned long)-1;
335
336 /*      return(gethostbyname(name)); */
337
338         CRYPTO_w_lock(CRYPTO_LOCK_BIO_GETHOSTBYNAME);
339         j=strlen(name);
340         if (j < 128)
341                 {
342                 for (i=0; i<GHBN_NUM; i++)
343                         {
344                         if (low > ghbn_cache[i].order)
345                                 {
346                                 low=ghbn_cache[i].order;
347                                 lowi=i;
348                                 }
349                         if (ghbn_cache[i].order > 0)
350                                 {
351                                 if (strncmp(name,ghbn_cache[i].name,128) == 0)
352                                         break;
353                                 }
354                         }
355                 }
356         else
357                 i=GHBN_NUM;
358
359         if (i == GHBN_NUM) /* no hit*/
360                 {
361                 BIO_ghbn_miss++;
362                 ret=gethostbyname(name);
363
364                 if (ret == NULL) return(NULL);
365                 if (j > 128) return(ret); /* too big to cache */
366
367                 /* else add to cache */
368                 if (ghbn_cache[lowi].ent != NULL)
369                         ghbn_free(ghbn_cache[lowi].ent);
370
371                 strncpy(ghbn_cache[lowi].name,name,128);
372                 ghbn_cache[lowi].ent=ghbn_dup(ret);
373                 ghbn_cache[lowi].order=BIO_ghbn_miss+BIO_ghbn_hits;
374                 }
375         else
376                 {
377                 BIO_ghbn_hits++;
378                 ret= ghbn_cache[i].ent;
379                 ghbn_cache[i].order=BIO_ghbn_miss+BIO_ghbn_hits;
380                 }
381         CRYPTO_w_unlock(CRYPTO_LOCK_BIO_GETHOSTBYNAME);
382         return(ret);
383         }
384
385 int BIO_sock_init()
386         {
387 #ifdef WINDOWS
388         static struct WSAData wsa_state;
389
390         if (!wsa_init_done)
391                 {
392                 int err;
393           
394 #ifdef SIGINT
395                 signal(SIGINT,(void (*)(int))BIO_sock_cleanup);
396 #endif
397                 wsa_init_done=1;
398                 memset(&wsa_state,0,sizeof(wsa_state));
399                 if (WSAStartup(0x0101,&wsa_state)!=0)
400                         {
401                         err=WSAGetLastError();
402                         SYSerr(SYS_F_WSASTARTUP,err);
403                         BIOerr(BIO_F_BIO_SOCK_INIT,BIO_R_WSASTARTUP);
404                         return(-1);
405                         }
406                 }
407 #endif /* WINDOWS */
408         return(1);
409         }
410
411 void BIO_sock_cleanup()
412         {
413 #ifdef WINDOWS
414         if (wsa_init_done)
415                 {
416                 wsa_init_done=0;
417                 WSACancelBlockingCall();
418                 WSACleanup();
419                 }
420 #endif
421         }
422
423 int BIO_socket_ioctl(fd,type,arg)
424 int fd;
425 long type;
426 unsigned long *arg;
427         {
428         int i;
429
430         i=ioctlsocket(fd,type,arg);
431         if (i < 0)
432                 SYSerr(SYS_F_IOCTLSOCKET,get_last_socket_error());
433         return(i);
434         }
435
436 /* The reason I have implemented this instead of using sscanf is because
437  * Visual C 1.52c gives an unresolved external when linking a DLL :-( */
438 static int get_ip(str,ip)
439 char *str;
440 unsigned char ip[4];
441         {
442         unsigned int tmp[4];
443         int num=0,c,ok=0;
444
445         tmp[0]=tmp[1]=tmp[2]=tmp[3]=0;
446
447         for (;;)
448                 {
449                 c= *(str++);
450                 if ((c >= '0') && (c <= '9'))
451                         {
452                         ok=1;
453                         tmp[num]=tmp[num]*10+c-'0';
454                         if (tmp[num] > 255) return(-1);
455                         }
456                 else if (c == '.')
457                         {
458                         if (!ok) return(-1);
459                         if (num == 3) break;
460                         num++;
461                         ok=0;
462                         }
463                 else if ((num == 3) && ok)
464                         break;
465                 else
466                         return(0);
467                 }
468         ip[0]=tmp[0];
469         ip[1]=tmp[1];
470         ip[2]=tmp[2];
471         ip[3]=tmp[3];
472         return(1);
473         }
474
475 int BIO_get_accept_socket(host,bind_mode)
476 char *host;
477 int bind_mode;
478         {
479         int ret=0;
480         struct sockaddr_in server,client;
481         int s= -1,cs;
482         unsigned char ip[4];
483         short port;
484         char *str,*h,*p,*e;
485         unsigned long l;
486         int err_num;
487
488         if (!BIO_sock_init()) return(INVALID_SOCKET);
489
490         if ((str=BUF_strdup(host)) == NULL) return(INVALID_SOCKET);
491
492         h=p=NULL;
493         h=str;
494         for (e=str; *e; e++)
495                 {
496                 if (*e == ':')
497                         {
498                         p= &(e[1]);
499                         *e='\0';
500                         }
501                 else if (*e == '/')
502                         {
503                         *e='\0';
504                         break;
505                         }
506                 }
507
508         if (p == NULL)
509                 {
510                 p=h;
511                 h="*";
512                 }
513
514         if (!BIO_get_port(p,&port)) return(INVALID_SOCKET);
515
516         memset((char *)&server,0,sizeof(server));
517         server.sin_family=AF_INET;
518         server.sin_port=htons((unsigned short)port);
519
520         if (strcmp(h,"*") == 0)
521                 server.sin_addr.s_addr=INADDR_ANY;
522         else
523                 {
524                 if (!BIO_get_host_ip(h,&(ip[0]))) return(INVALID_SOCKET);
525                 l=(unsigned long)
526                         ((unsigned long)ip[0]<<24L)|
527                         ((unsigned long)ip[1]<<16L)|
528                         ((unsigned long)ip[2]<< 8L)|
529                         ((unsigned long)ip[3]);
530                 server.sin_addr.s_addr=htonl(l);
531                 }
532
533 again:
534         s=socket(AF_INET,SOCK_STREAM,SOCKET_PROTOCOL);
535         if (s == INVALID_SOCKET)
536                 {
537                 SYSerr(SYS_F_SOCKET,get_last_socket_error());
538                 ERR_add_error_data(3,"port='",host,"'");
539                 BIOerr(BIO_F_BIO_GET_ACCEPT_SOCKET,BIO_R_UNABLE_TO_CREATE_SOCKET);
540                 goto err;
541                 }
542
543 #ifdef SO_REUSEADDR
544         if (bind_mode == BIO_BIND_REUSEADDR)
545                 {
546                 int i=1;
547
548                 ret=setsockopt(s,SOL_SOCKET,SO_REUSEADDR,(char *)&i,sizeof(i));
549                 bind_mode=BIO_BIND_NORMAL;
550                 }
551 #endif
552         if (bind(s,(struct sockaddr *)&server,sizeof(server)) == -1)
553                 {
554 #ifdef SO_REUSEADDR
555                 err_num=get_last_socket_error();
556                 if ((bind_mode == BIO_BIND_REUSEADDR_IF_UNUSED) &&
557                         (err_num == EADDRINUSE))
558                         {
559                         memcpy((char *)&client,(char *)&server,sizeof(server));
560                         if (strcmp(h,"*") == 0)
561                                 client.sin_addr.s_addr=htonl(0x7F000001);
562                         cs=socket(AF_INET,SOCK_STREAM,SOCKET_PROTOCOL);
563                         if (cs != INVALID_SOCKET)
564                                 {
565                                 int ii;
566                                 ii=connect(cs,(struct sockaddr *)&client,
567                                         sizeof(client));
568                                 closesocket(cs);
569                                 if (ii == INVALID_SOCKET)
570                                         {
571                                         bind_mode=BIO_BIND_REUSEADDR;
572                                         closesocket(s);
573                                         goto again;
574                                         }
575                                 /* else error */
576                                 }
577                         /* else error */
578                         }
579 #endif
580                 SYSerr(SYS_F_BIND,err_num);
581                 ERR_add_error_data(3,"port='",host,"'");
582                 BIOerr(BIO_F_BIO_GET_ACCEPT_SOCKET,BIO_R_UNABLE_TO_BIND_SOCKET);
583                 goto err;
584                 }
585         if (listen(s,MAX_LISTEN) == -1)
586                 {
587                 SYSerr(SYS_F_BIND,get_last_socket_error());
588                 ERR_add_error_data(3,"port='",host,"'");
589                 BIOerr(BIO_F_BIO_GET_ACCEPT_SOCKET,BIO_R_UNABLE_TO_LISTEN_SOCKET);
590                 goto err;
591                 }
592         ret=1;
593 err:
594         if (str != NULL) Free(str);
595         if ((ret == 0) && (s != INVALID_SOCKET))
596                 {
597                 closesocket(s);
598                 s= INVALID_SOCKET;
599                 }
600         return(s);
601         }
602
603 int BIO_accept(sock,addr)
604 int sock;
605 char **addr;
606         {
607         int ret=INVALID_SOCKET;
608         static struct sockaddr_in from;
609         unsigned long l;
610         short port;
611         int len;
612         char *p;
613
614         memset((char *)&from,0,sizeof(from));
615         len=sizeof(from);
616         ret=accept(sock,(struct sockaddr *)&from,&len);
617         if (ret == INVALID_SOCKET)
618                 {
619                 SYSerr(SYS_F_ACCEPT,get_last_socket_error());
620                 BIOerr(BIO_F_BIO_ACCEPT,BIO_R_ACCEPT_ERROR);
621                 goto end;
622                 }
623
624         if (addr == NULL) goto end;
625
626         l=ntohl(from.sin_addr.s_addr);
627         port=ntohs(from.sin_port);
628         if (*addr == NULL)
629                 {
630                 if ((p=Malloc(24)) == NULL)
631                         {
632                         BIOerr(BIO_F_BIO_ACCEPT,ERR_R_MALLOC_FAILURE);
633                         goto end;
634                         }
635                 *addr=p;
636                 }
637         sprintf(*addr,"%d.%d.%d.%d:%d",
638                 (unsigned char)(l>>24L)&0xff,
639                 (unsigned char)(l>>16L)&0xff,
640                 (unsigned char)(l>> 8L)&0xff,
641                 (unsigned char)(l     )&0xff,
642                 port);
643 end:
644         return(ret);
645         }
646
647 int BIO_set_tcp_ndelay(s,on)
648 int s;
649 int on;
650         {
651         int ret=0;
652 #if defined(TCP_NODELAY) && (defined(IPPROTO_TCP) || defined(SOL_TCP))
653         int opt;
654
655 #ifdef SOL_TCP
656         opt=SOL_TCP;
657 #else
658 #ifdef IPPROTO_TCP
659         opt=IPPROTO_TCP;
660 #endif
661 #endif
662         
663         ret=setsockopt(s,opt,TCP_NODELAY,(char *)&on,sizeof(on));
664 #endif
665         return(ret == 0);
666         }
667 #endif
668
669 int BIO_socket_nbio(s,mode)
670 int s;
671 int mode;
672         {
673         int ret= -1;
674         unsigned long l;
675
676         l=mode;
677 #ifdef FIONBIO
678         ret=BIO_socket_ioctl(s,FIONBIO,&l);
679 #endif
680         return(ret == 0);
681         }