Change #include filenames from <foo.h> to <openssl.h>.
[openssl.git] / crypto / bio / b_sock.c
index a45909527c9f27d11da148441a314112cd76f1a9..328394efe42614d41e20af26e19a0e405268a155 100644 (file)
@@ -63,7 +63,7 @@
 #include <errno.h>
 #define USE_SOCKETS
 #include "cryptlib.h"
-#include "bio.h"
+#include <openssl/bio.h>
 
 /*     BIOerr(BIO_F_WSASTARTUP,BIO_R_WSASTARTUP ); */
 
@@ -97,7 +97,7 @@ static struct ghbn_cache_st
        } ghbn_cache[GHBN_NUM];
 
 #ifndef NOPROTO
-static int get_ip(char *str,unsigned char *ip);
+static int get_ip(const char *str,unsigned char *ip);
 static void ghbn_free(struct hostent *a);
 static struct hostent *ghbn_dup(struct hostent *a);
 #else
@@ -106,9 +106,7 @@ static void ghbn_free();
 static struct hostent *ghbn_dup();
 #endif
 
-int BIO_get_host_ip(str,ip)
-char *str;
-unsigned char *ip;
+int BIO_get_host_ip(const char *str, unsigned char *ip)
        {
        int i;
        struct hostent *he;
@@ -146,9 +144,7 @@ unsigned char *ip;
        return(1);
        }
 
-int BIO_get_port(str,port_ptr)
-char *str;
-short *port_ptr;
+int BIO_get_port(const char *str, unsigned short *port_ptr)
        {
        int i;
        struct servent *s;
@@ -197,24 +193,21 @@ short *port_ptr;
        return(1);
        }
 
-int BIO_sock_error(sock)
-int sock;
+int BIO_sock_error(int sock)
        {
-       int j,i,size;
+       int j,i;
+       int size;
                 
        size=sizeof(int);
 
-       i=getsockopt(sock,SOL_SOCKET,SO_ERROR,(char *)&j,&size);
+       i=getsockopt(sock,SOL_SOCKET,SO_ERROR,&j,&size);
        if (i < 0)
                return(1);
        else
                return(j);
        }
 
-long BIO_ghbn_ctrl(cmd,iarg,parg)
-int cmd;
-int iarg;
-char *parg;
+long BIO_ghbn_ctrl(int cmd, int iarg, char *parg)
        {
        int i;
        char **p;
@@ -223,13 +216,13 @@ char *parg;
                {
        case BIO_GHBN_CTRL_HITS:
                return(BIO_ghbn_hits);
-               break;
+               /* break; */
        case BIO_GHBN_CTRL_MISSES:
                return(BIO_ghbn_miss);
-               break;
+               /* break; */
        case BIO_GHBN_CTRL_CACHE_SIZE:
                return(GHBN_NUM);
-               break;
+               /* break; */
        case BIO_GHBN_CTRL_GET_ENTRY:
                if ((iarg >= 0) && (iarg <GHBN_NUM) &&
                        (ghbn_cache[iarg].order > 0))
@@ -241,7 +234,7 @@ char *parg;
                        return(1);
                        }
                return(0);
-               break;
+               /* break; */
        case BIO_GHBN_CTRL_FLUSH:
                for (i=0; i<GHBN_NUM; i++)
                        ghbn_cache[i].order=0;
@@ -252,77 +245,82 @@ char *parg;
        return(1);
        }
 
-static struct hostent *ghbn_dup(a)
-struct hostent *a;
+static struct hostent *ghbn_dup(struct hostent *a)
        {
        struct hostent *ret;
        int i,j;
 
-       ret=(struct hostent *)malloc(sizeof(struct hostent));
+       MemCheck_off();
+       ret=(struct hostent *)Malloc(sizeof(struct hostent));
        if (ret == NULL) return(NULL);
        memset(ret,0,sizeof(struct hostent));
 
        for (i=0; a->h_aliases[i] != NULL; i++)
                ;
        i++;
-       ret->h_aliases=(char **)malloc(sizeof(char *)*i);
+       ret->h_aliases=(char **)Malloc(sizeof(char *)*i);
        memset(ret->h_aliases,0,sizeof(char *)*i);
        if (ret == NULL) goto err;
 
        for (i=0; a->h_addr_list[i] != NULL; i++)
                ;
        i++;
-       ret->h_addr_list=(char **)malloc(sizeof(char *)*i);
+       ret->h_addr_list=(char **)Malloc(sizeof(char *)*i);
        memset(ret->h_addr_list,0,sizeof(char *)*i);
        if (ret->h_addr_list == NULL) goto err;
 
        j=strlen(a->h_name)+1;
-       if ((ret->h_name=malloc(j)) == NULL) goto err;
-       memcpy((char *)ret->h_name,a->h_name,j);
+       if ((ret->h_name=Malloc(j)) == NULL) goto err;
+       memcpy((char *)ret->h_name,a->h_name,j+1);
        for (i=0; a->h_aliases[i] != NULL; i++)
                {
                j=strlen(a->h_aliases[i])+1;
-               if ((ret->h_aliases[i]=malloc(j)) == NULL) goto err;
-               memcpy(ret->h_aliases[i],a->h_aliases[i],j);
+               if ((ret->h_aliases[i]=Malloc(j)) == NULL) goto err;
+               memcpy(ret->h_aliases[i],a->h_aliases[i],j+1);
                }
        ret->h_length=a->h_length;
        ret->h_addrtype=a->h_addrtype;
        for (i=0; a->h_addr_list[i] != NULL; i++)
                {
-               if ((ret->h_addr_list[i]=malloc(a->h_length)) == NULL)
+               if ((ret->h_addr_list[i]=Malloc(a->h_length)) == NULL)
                        goto err;
                memcpy(ret->h_addr_list[i],a->h_addr_list[i],a->h_length);
                }
-       return(ret);
+       if (0)
+               {
 err:   
-       if (ret != NULL)
-               ghbn_free(ret);
-       return(NULL);
+               if (ret != NULL)
+                       ghbn_free(ret);
+               ret=NULL;
+               }
+       MemCheck_on();
+       return(ret);
        }
 
-static void ghbn_free(a)
-struct hostent *a;
+static void ghbn_free(struct hostent *a)
        {
        int i;
 
+       if(a == NULL)
+           return;
+
        if (a->h_aliases != NULL)
                {
                for (i=0; a->h_aliases[i] != NULL; i++)
-                       free(a->h_aliases[i]);
-               free(a->h_aliases);
+                       Free(a->h_aliases[i]);
+               Free(a->h_aliases);
                }
        if (a->h_addr_list != NULL)
                {
                for (i=0; a->h_addr_list[i] != NULL; i++)
-                       free(a->h_addr_list[i]);
-               free(a->h_addr_list);
+                       Free(a->h_addr_list[i]);
+               Free(a->h_addr_list);
                }
-       if (a->h_name != NULL) free((char *)a->h_name);
-       free(a);
+       if (a->h_name != NULL) Free((char *)a->h_name);
+       Free(a);
        }
 
-struct hostent *BIO_gethostbyname(name)
-char *name;
+struct hostent *BIO_gethostbyname(const char *name)
        {
        struct hostent *ret;
        int i,lowi=0,j;
@@ -377,7 +375,7 @@ char *name;
        return(ret);
        }
 
-int BIO_sock_init()
+int BIO_sock_init(void)
        {
 #ifdef WINDOWS
        static struct WSAData wsa_state;
@@ -403,7 +401,7 @@ int BIO_sock_init()
        return(1);
        }
 
-void BIO_sock_cleanup()
+void BIO_sock_cleanup(void)
        {
 #ifdef WINDOWS
        if (wsa_init_done)
@@ -415,10 +413,7 @@ void BIO_sock_cleanup()
 #endif
        }
 
-int BIO_socket_ioctl(fd,type,arg)
-int fd;
-long type;
-unsigned long *arg;
+int BIO_socket_ioctl(int fd, long type, unsigned long *arg)
        {
        int i;
 
@@ -430,9 +425,7 @@ unsigned long *arg;
 
 /* The reason I have implemented this instead of using sscanf is because
  * Visual C 1.52c gives an unresolved external when linking a DLL :-( */
-static int get_ip(str,ip)
-char *str;
-unsigned char ip[4];
+static int get_ip(const char *str, unsigned char ip[4])
        {
        unsigned int tmp[4];
        int num=0,c,ok=0;
@@ -467,16 +460,17 @@ unsigned char ip[4];
        return(1);
        }
 
-int BIO_get_accept_socket(host)
-char *host;
+int BIO_get_accept_socket(char *host, int bind_mode)
        {
        int ret=0;
-       struct sockaddr_in server;
-       int s= -1;
+       struct sockaddr_in server,client;
+       int s= -1,cs;
        unsigned char ip[4];
-       short port;
-       char *str,*h,*p,*e;
+       unsigned short port;
+       char *str,*e;
+       const char *h,*p;
        unsigned long l;
+       int err_num;
 
        if (!BIO_sock_init()) return(INVALID_SOCKET);
 
@@ -508,7 +502,7 @@ char *host;
 
        memset((char *)&server,0,sizeof(server));
        server.sin_family=AF_INET;
-       server.sin_port=htons((unsigned short)port);
+       server.sin_port=htons(port);
 
        if (strcmp(h,"*") == 0)
                server.sin_addr.s_addr=INADDR_ANY;
@@ -517,12 +511,13 @@ char *host;
                if (!BIO_get_host_ip(h,&(ip[0]))) return(INVALID_SOCKET);
                l=(unsigned long)
                        ((unsigned long)ip[0]<<24L)|
-                       ((unsigned long)ip[0]<<16L)|
-                       ((unsigned long)ip[0]<< 8L)|
-                       ((unsigned long)ip[0]);
+                       ((unsigned long)ip[1]<<16L)|
+                       ((unsigned long)ip[2]<< 8L)|
+                       ((unsigned long)ip[3]);
                server.sin_addr.s_addr=htonl(l);
                }
 
+again:
        s=socket(AF_INET,SOCK_STREAM,SOCKET_PROTOCOL);
        if (s == INVALID_SOCKET)
                {
@@ -531,9 +526,45 @@ char *host;
                BIOerr(BIO_F_BIO_GET_ACCEPT_SOCKET,BIO_R_UNABLE_TO_CREATE_SOCKET);
                goto err;
                }
+
+#ifdef SO_REUSEADDR
+       if (bind_mode == BIO_BIND_REUSEADDR)
+               {
+               int i=1;
+
+               ret=setsockopt(s,SOL_SOCKET,SO_REUSEADDR,(char *)&i,sizeof(i));
+               bind_mode=BIO_BIND_NORMAL;
+               }
+#endif
        if (bind(s,(struct sockaddr *)&server,sizeof(server)) == -1)
                {
-               SYSerr(SYS_F_BIND,get_last_socket_error());
+#ifdef SO_REUSEADDR
+               err_num=get_last_socket_error();
+               if ((bind_mode == BIO_BIND_REUSEADDR_IF_UNUSED) &&
+                       (err_num == EADDRINUSE))
+                       {
+                       memcpy((char *)&client,(char *)&server,sizeof(server));
+                       if (strcmp(h,"*") == 0)
+                               client.sin_addr.s_addr=htonl(0x7F000001);
+                       cs=socket(AF_INET,SOCK_STREAM,SOCKET_PROTOCOL);
+                       if (cs != INVALID_SOCKET)
+                               {
+                               int ii;
+                               ii=connect(cs,(struct sockaddr *)&client,
+                                       sizeof(client));
+                               closesocket(cs);
+                               if (ii == INVALID_SOCKET)
+                                       {
+                                       bind_mode=BIO_BIND_REUSEADDR;
+                                       closesocket(s);
+                                       goto again;
+                                       }
+                               /* else error */
+                               }
+                       /* else error */
+                       }
+#endif
+               SYSerr(SYS_F_BIND,err_num);
                ERR_add_error_data(3,"port='",host,"'");
                BIOerr(BIO_F_BIO_GET_ACCEPT_SOCKET,BIO_R_UNABLE_TO_BIND_SOCKET);
                goto err;
@@ -550,24 +581,18 @@ err:
        if (str != NULL) Free(str);
        if ((ret == 0) && (s != INVALID_SOCKET))
                {
-#ifdef WINDOWS
                closesocket(s);
-#else
-               close(s);
-#endif
                s= INVALID_SOCKET;
                }
        return(s);
        }
 
-int BIO_accept(sock,addr)
-int sock;
-char **addr;
+int BIO_accept(int sock, char **addr)
        {
        int ret=INVALID_SOCKET;
        static struct sockaddr_in from;
        unsigned long l;
-       short port;
+       unsigned short port;
        int len;
        char *p;
 
@@ -604,9 +629,7 @@ end:
        return(ret);
        }
 
-int BIO_set_tcp_ndelay(s,on)
-int s;
-int on;
+int BIO_set_tcp_ndelay(int s, int on)
        {
        int ret=0;
 #if defined(TCP_NODELAY) && (defined(IPPROTO_TCP) || defined(SOL_TCP))
@@ -626,3 +649,14 @@ int on;
        }
 #endif
 
+int BIO_socket_nbio(int s, int mode)
+       {
+       int ret= -1;
+       unsigned long l;
+
+       l=mode;
+#ifdef FIONBIO
+       ret=BIO_socket_ioctl(s,FIONBIO,&l);
+#endif
+       return(ret == 0);
+       }