]> code.citadel.org Git - citadel.git/blobdiff - webcit/tcp_sockets.c
* Started moving all of the global variables into a struct, to facilitate
[citadel.git] / webcit / tcp_sockets.c
index a26023555bcd26b51ca5f11bbd9dab0fcab8871c..d7ec10712e545c5abecb64f565c4911529e45ff4 100644 (file)
@@ -3,17 +3,19 @@
  * 
  * TCP socket module for WebCit
  *
- * version 1.3
- *
+ * $Id$
  */
 
 #include <stdlib.h>
+#ifdef HAVE_UNISTD_H
 #include <unistd.h>
+#endif
 #include <stdio.h>
 #include <signal.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
+#include <arpa/inet.h>
 #include <netdb.h>
 #include <string.h>
 #include <stdarg.h>
@@ -27,84 +29,66 @@ char server_is_local = 0;
 
 extern int errno;
 
-int serv_sock;
-
-#ifndef DONT_PROTO_HTONS
-u_short htons();
-#endif
-u_long inet_addr();
-
-
-void timeout() {
-       printf("\rConnection timed out.\n");
+RETSIGTYPE timeout(int signum)
+{
+       fprintf(stderr, "Connection timed out.\n");
        exit(3);
-       }
+}
 
-int connectsock(host,service,protocol)
-char *host;
-char *service;
-char *protocol; {
+int connectsock(char *host, char *service, char *protocol)
+{
        struct hostent *phe;
        struct servent *pse;
        struct protoent *ppe;
        struct sockaddr_in sin;
-       int s,type;
+       int s, type;
 
-       bzero((char *)&sin,sizeof(sin));
+       bzero((char *) &sin, sizeof(sin));
        sin.sin_family = AF_INET;
 
-       pse=getservbyname(service,protocol);
+       pse = getservbyname(service, protocol);
        if (pse) {
                sin.sin_port = pse->s_port;
-               }
-       else if ((sin.sin_port = htons((u_short)atoi(service))) == 0) {
-               fprintf(stderr,"can't get %s service entry: %s\n",
-                       service,strerror(errno));
-               return(-1);
-               }
-       
-       phe=gethostbyname(host);
+       } else if ((sin.sin_port = htons((u_short) atoi(service))) == 0) {
+               fprintf(stderr, "Can't get %s service entry\n", service);
+               return (-1);
+       }
+       phe = gethostbyname(host);
        if (phe) {
-               bcopy(phe->h_addr,(char *)&sin.sin_addr,phe->h_length);
-               }
-       else if ((sin.sin_addr.s_addr = inet_addr(host))==INADDR_NONE) {
-               fprintf(stderr,"can't get %s host entry: %s\n",
-                       host,strerror(errno));
-               return(-1);
-               }
-
-       if ((ppe=getprotobyname(protocol))==0) {
-               fprintf(stderr,"can't get %s protocol entry: %s\n",
-                       protocol,strerror(errno));
-               return(-1);
-               }
-
-       if (!strcmp(protocol,"udp"))
+               bcopy(phe->h_addr, (char *) &sin.sin_addr, phe->h_length);
+       } else if ((sin.sin_addr.s_addr = inet_addr(host)) == INADDR_NONE) {
+               fprintf(stderr, "Can't get %s host entry: %s\n",
+                       host, strerror(errno));
+               return (-1);
+       }
+       if ((ppe = getprotobyname(protocol)) == 0) {
+               fprintf(stderr, "Can't get %s protocol entry: %s\n",
+                       protocol, strerror(errno));
+               return (-1);
+       }
+       if (!strcmp(protocol, "udp"))
                type = SOCK_DGRAM;
        else
                type = SOCK_STREAM;
 
-       s = socket(PF_INET,type,ppe->p_proto);
-       if (s<0) {
-               fprintf(stderr,"can't create socket: %s\n",strerror(errno));
-               return(-1);
-               }
-
-
-       signal(SIGALRM,timeout);
+       s = socket(PF_INET, type, ppe->p_proto);
+       if (s < 0) {
+               fprintf(stderr, "Can't create socket: %s\n", strerror(errno));
+               return (-1);
+       }
+       signal(SIGALRM, timeout);
        alarm(30);
 
-       if (connect(s,(struct sockaddr *)&sin,sizeof(sin))<0) {
-               fprintf(stderr,"can't connect to %s.%s: %s\n",
-                       host,service,strerror(errno));
-               return(-1);
-               }
-
+       if (connect(s, (struct sockaddr *) &sin, sizeof(sin)) < 0) {
+               fprintf(stderr, "can't connect to %s.%s: %s\n",
+                       host, service, strerror(errno));
+               return (-1);
+       }
        alarm(0);
-       signal(SIGALRM,SIG_IGN);
+       signal(SIGALRM, SIG_IGN);
 
-       return(s);
-       }
+       return (s);
+}
 
 
 
@@ -112,111 +96,93 @@ char *protocol; {
 /*
  * Input binary data from socket
  */
-void serv_read(buf,bytes)
-char buf[];
-int bytes; {
-        int len,rlen;
-
-        len = 0;
-        while(len<bytes) {
-                rlen = read(serv_sock,&buf[len],bytes-len);
-                if (rlen<1) {
-                        printf("Network error - connection terminated.\n");
-                        printf("%s\n", strerror(errno));
-                        exit(3);
-                        }
-                len = len + rlen;
-                }
-        }
+void serv_read(char *buf, int bytes)
+{
+       int len, rlen;
+
+       len = 0;
+       while (len < bytes) {
+               rlen = read(WC->serv_sock, &buf[len], bytes - len);
+               if (rlen < 1) {
+                       fprintf(stderr, "Server connection broken: %s\n",
+                               strerror(errno));
+                       WC->connected = 0;
+                       WC->logged_in = 0;
+                       return;
+               }
+               len = len + rlen;
+       }
+}
 
 
 /*
  * input string from pipe
  */
-void serv_gets(strbuf)
-char strbuf[]; {
-       int ch,len,rlen;
+void serv_gets(char *strbuf)
+{
+       int ch, len;
        char buf[2];
 
        len = 0;
-       strcpy(strbuf,"");
+       strcpy(strbuf, "");
        do {
                serv_read(&buf[0], 1);
                ch = buf[0];
                strbuf[len++] = ch;
-               } while((ch!=10)&&(ch!=13)&&(ch!=0)&&(len<255));
-       strbuf[len-1] = 0;
-       }
+       } while ((ch != 10) && (ch != 13) && (ch != 0) && (len < 255));
+       strbuf[len - 1] = 0;
+       /* fprintf(stderr, ">%s\n", strbuf); */
+}
 
 
 
 /*
- * Attach to a Citadel server
+ * send binary to server
  */
-void attach_to_server(argc,argv)
-int argc;
-char *argv[]; {
-       if (argc==1) {
-               server_is_local = 1;
-               serv_sock = connectsock("localhost","citadel","tcp");
-               }
-       if (argc==2) {
-               serv_sock = connectsock(argv[1],"citadel","tcp");
-               if ( (!strcmp(argv[1],"localhost"))
-               || (!strcmp(argv[1],"127.0.0.1")) ) server_is_local = 1;
+void serv_write(char *buf, int nbytes)
+{
+       int bytes_written = 0;
+       int retval;
+       while (bytes_written < nbytes) {
+               retval = write(WC->serv_sock, &buf[bytes_written],
+                              nbytes - bytes_written);
+               if (retval < 1) {
+                       fprintf(stderr, "Server connection broken: %s\n",
+                               strerror(errno));
+                       WC->connected = 0;
+                       WC->logged_in = 0;
+                       return;
                }
-       if (argc>=3) serv_sock = connectsock(argv[1],argv[2],"tcp");
-
-       if (serv_sock < 0) exit(errno);
+               bytes_written = bytes_written + retval;
        }
-
-
-/*
- * send binary to server
- */
-void serv_write(buf, nbytes)
-char buf[];
-int nbytes; {
-        int bytes_written = 0;
-        int retval;
-        while (bytes_written < nbytes) {
-                retval = write(serv_sock, &buf[bytes_written],
-                        nbytes - bytes_written);
-                if (retval < 1) {
-                        printf("Network error - connection terminated.\n");
-                        printf("%s\n", strerror(errno));
-                        exit(3);
-                        }
-                bytes_written = bytes_written + retval;
-                }
-        }
+}
 
 
 /*
  * send line to server
  */
-void serv_puts(string)
-char *string; {
+void serv_puts(char *string)
+{
        char buf[256];
 
-       sprintf(buf,"%s\n",string);
+       sprintf(buf, "%s\n", string);
        serv_write(buf, strlen(buf));
-       }
+}
 
 
 /*
  * convenience function to send stuff to the server
  */
-void serv_printf(const char *format, ...) {   
-        va_list arg_ptr;   
+void serv_printf(const char *format,...)
+{
+       va_list arg_ptr;
        char buf[256];
 
-               va_start(arg_ptr, format);   
-               vsprintf(buf, format, arg_ptr);   
-               va_end(arg_ptr);   
+       va_start(arg_ptr, format);
+       vsprintf(buf, format, arg_ptr);
+       va_end(arg_ptr);
 
        strcat(buf, "\n");
        serv_write(buf, strlen(buf));
-       }
-
-
+       /* fprintf(stderr, "<%s", buf); */
+}