Fix broken build
[openssl.git] / apps / s_socket.c
index a9228f42a6b2b433ffdb263cd0b7a6fe59422e9c..e83baf4e709279dee9a452c990fa276eadb073a7 100644 (file)
@@ -91,16 +91,21 @@ typedef unsigned int u_int;
 #include "netdb.h"
 #endif
 
-static struct hostent *GetHostByName(char *name);
+static struct hostent *GetHostByName(const char *name);
 #if defined(OPENSSL_SYS_WINDOWS) || (defined(OPENSSL_SYS_NETWARE) && !defined(NETWARE_BSDSOCK))
 static void ssl_sock_cleanup(void);
 #endif
 static int ssl_sock_init(void);
-static int init_client_ip(int *sock,unsigned char ip[4], int port, int type);
+static int init_client_ip(int *sock, const unsigned char ip[4], int port,
+                         int type);
 static int init_server(int *sock, int port, int type);
 static int init_server_long(int *sock, int port,char *ip, int type);
 static int do_accept(int acc_sock, int *sock, char **host);
-static int host_ip(char *str, unsigned char ip[4]);
+static int host_ip(const char *str, unsigned char ip[4]);
+#ifndef NO_SYS_UN_H
+static int init_server_unix(int *sock, const char *path);
+static int do_accept_unix(int acc_sock, int *sock);
+#endif
 
 #ifdef OPENSSL_SYS_WIN16
 #define SOCKET_PROTOCOL        0 /* more microsoft stupidity */
@@ -228,18 +233,18 @@ static int ssl_sock_init(void)
        return(1);
        }
 
-int init_client(int *sock, char *host, int port, int type)
+int init_client(int *sock, const char *host, int port, int type)
        {
        unsigned char ip[4];
 
+       ip[0] = ip[1] = ip[2] = ip[3] = 0;
        if (!host_ip(host,&(ip[0])))
-               {
-               return(0);
-               }
-       return(init_client_ip(sock,ip,port,type));
+               return 0;
+       return init_client_ip(sock,ip,port,type);
        }
 
-static int init_client_ip(int *sock, unsigned char ip[4], int port, int type)
+static int init_client_ip(int *sock, const unsigned char ip[4], int port,
+                         int type)
        {
        unsigned long addr;
        struct sockaddr_in them;
@@ -269,7 +274,7 @@ static int init_client_ip(int *sock, unsigned char ip[4], int port, int type)
                {
                i=0;
                i=setsockopt(s,SOL_SOCKET,SO_KEEPALIVE,(char *)&i,sizeof(i));
-               if (i < 0) { perror("keepalive"); return(0); }
+               if (i < 0) { closesocket(s); perror("keepalive"); return(0); }
                }
 #endif
 
@@ -279,7 +284,32 @@ static int init_client_ip(int *sock, unsigned char ip[4], int port, int type)
        return(1);
        }
 
-int do_server(int port, int type, int *ret, int (*cb)(char *hostname, int s, unsigned char *context), unsigned char *context)
+#ifndef NO_SYS_UN_H
+int init_client_unix(int *sock, const char *server)
+       {
+       struct sockaddr_un them;
+       int s;
+
+       if (strlen(server) > (UNIX_PATH_MAX + 1)) return(0);
+       if (!ssl_sock_init()) return(0);
+
+       s=socket(AF_UNIX, SOCK_STREAM, 0);
+       if (s == INVALID_SOCKET) { perror("socket"); return(0); }
+
+       memset((char *)&them,0,sizeof(them));
+       them.sun_family=AF_UNIX;
+       strcpy(them.sun_path, server);
+
+       if (connect(s, (struct sockaddr *)&them, sizeof(them)) == -1)
+               { closesocket(s); perror("connect"); return(0); }
+       *sock=s;
+       return(1);
+       }
+#endif
+
+int do_server(int port, int type, int *ret,
+             int (*cb)(char *hostname, int s, int stype, unsigned char *context),
+             unsigned char *context, int naccept)
        {
        int sock;
        char *name = NULL;
@@ -297,7 +327,11 @@ int do_server(int port, int type, int *ret, int (*cb)(char *hostname, int s, uns
                {
                if (type==SOCK_STREAM)
                        {
+#ifdef OPENSSL_SSL_DEBUG_BROKEN_PROTOCOL
+                       if (do_accept(accept_socket,&sock,NULL) == 0)
+#else
                        if (do_accept(accept_socket,&sock,&name) == 0)
+#endif
                                {
                                SHUTDOWN(accept_socket);
                                return(0);
@@ -305,11 +339,13 @@ int do_server(int port, int type, int *ret, int (*cb)(char *hostname, int s, uns
                        }
                else
                        sock = accept_socket;
-               i=(*cb)(name,sock, context);
+               i=(*cb)(name,sock, type, context);
                if (name != NULL) OPENSSL_free(name);
                if (type==SOCK_STREAM)
                        SHUTDOWN2(sock);
-               if (i < 0)
+               if (naccept != -1)
+                       naccept--;
+               if (i < 0 || naccept == 0)
                        {
                        SHUTDOWN2(accept_socket);
                        return(i);
@@ -317,6 +353,43 @@ int do_server(int port, int type, int *ret, int (*cb)(char *hostname, int s, uns
                }
        }
 
+#ifndef NO_SYS_UN_H
+int do_server_unix(const char *path, int *ret,
+                  int (*cb)(char *hostname, int s, int stype, unsigned char *context),
+                  unsigned char *context, int naccept)
+       {
+       int sock;
+       int accept_socket = 0;
+       int i;
+
+       if (!init_server_unix(&accept_socket, path)) return(0);
+
+       if (ret != NULL)
+               *ret=accept_socket;
+       for (;;)
+               {
+               if (do_accept_unix(accept_socket, &sock) == 0)
+                       {
+                       SHUTDOWN(accept_socket);
+                       i = 0;
+                       goto out;
+                       }
+               i=(*cb)(NULL, sock, 0, context);
+               SHUTDOWN2(sock);
+               if (naccept != -1)
+                       naccept--;
+               if (i < 0 || naccept == 0)
+                       {
+                       SHUTDOWN2(accept_socket);
+                       goto out;
+                       }
+               }
+out:
+       unlink(path);
+       return(i);
+       }
+#endif
+
 static int init_server_long(int *sock, int port, char *ip, int type)
        {
        int ret=0;
@@ -375,6 +448,50 @@ static int init_server(int *sock, int port, int type)
        return(init_server_long(sock, port, NULL, type));
        }
 
+#ifndef NO_SYS_UN_H
+static int init_server_unix(int *sock, const char *path)
+       {
+       int ret = 0;
+       struct sockaddr_un server;
+       int s = -1;
+
+       if (strlen(path) > (UNIX_PATH_MAX + 1)) return(0);
+       if (!ssl_sock_init()) return(0);
+
+       s=socket(AF_UNIX, SOCK_STREAM, 0);
+       if (s == INVALID_SOCKET) goto err;
+
+       memset((char *)&server,0,sizeof(server));
+       server.sun_family=AF_UNIX;
+       strcpy(server.sun_path, path);
+
+       if (bind(s, (struct sockaddr *)&server, sizeof(server)) == -1)
+               {
+#ifndef OPENSSL_SYS_WINDOWS
+               perror("bind");
+#endif
+               goto err;
+               }
+       /* Make it 128 for linux */
+       if (listen(s,128) == -1)
+               {
+#ifndef OPENSSL_SYS_WINDOWS
+               perror("listen");
+#endif
+               unlink(path);
+               goto err;
+               }
+       *sock=s;
+       ret=1;
+err:
+       if ((ret == 0) && (s != -1))
+               {
+               SHUTDOWN(s);
+               }
+       return(ret);
+       }
+#endif
+
 static int do_accept(int acc_sock, int *sock, char **host)
        {
        int ret;
@@ -400,6 +517,7 @@ redoit:
        if (ret == INVALID_SOCKET)
                {
 #if defined(OPENSSL_SYS_WINDOWS) || (defined(OPENSSL_SYS_NETWARE) && !defined(NETWARE_BSDSOCK))
+               int i;
                i=WSAGetLastError();
                BIO_printf(bio_err,"accept error %d\n",i);
 #else
@@ -444,6 +562,7 @@ redoit:
                if ((*host=(char *)OPENSSL_malloc(strlen(h1->h_name)+1)) == NULL)
                        {
                        perror("OPENSSL_malloc");
+                       closesocket(ret);
                        return(0);
                        }
                BUF_strlcpy(*host,h1->h_name,strlen(h1->h_name)+1);
@@ -452,11 +571,13 @@ redoit:
                if (h2 == NULL)
                        {
                        BIO_printf(bio_err,"gethostbyname failure\n");
+                       closesocket(ret);
                        return(0);
                        }
                if (h2->h_addrtype != AF_INET)
                        {
                        BIO_printf(bio_err,"gethostbyname addr is not AF_INET\n");
+                       closesocket(ret);
                        return(0);
                        }
                }
@@ -465,6 +586,32 @@ end:
        return(1);
        }
 
+#ifndef NO_SYS_UN_H
+static int do_accept_unix(int acc_sock, int *sock)
+       {
+       int ret;
+
+       if (!ssl_sock_init()) return(0);
+
+redoit:
+       ret=accept(acc_sock, NULL, NULL);
+       if (ret == INVALID_SOCKET)
+               {
+               if (errno == EINTR)
+                       {
+                       /*check_timeout(); */
+                       goto redoit;
+                       }
+               fprintf(stderr,"errno=%d ",errno);
+               perror("accept");
+               return(0);
+               }
+
+       *sock=ret;
+       return(1);
+       }
+#endif
+
 int extract_host_port(char *str, char **host_ptr, unsigned char *ip,
             short *port_ptr)
        {
@@ -490,7 +637,7 @@ err:
        return(0);
        }
 
-static int host_ip(char *str, unsigned char ip[4])
+static int host_ip(const char *str, unsigned char ip[4])
        {
        unsigned int in[4]; 
        int i;
@@ -536,7 +683,7 @@ err:
        return(0);
        }
 
-int extract_port(char *str, short *port_ptr)
+int extract_port(const char *str, short *port_ptr)
        {
        int i;
        struct servent *s;
@@ -568,7 +715,7 @@ static struct ghbn_cache_st
 static unsigned long ghbn_hits=0L;
 static unsigned long ghbn_miss=0L;
 
-static struct hostent *GetHostByName(char *name)
+static struct hostent *GetHostByName(const char *name)
        {
        struct hostent *ret;
        int i,lowi=0;