webcit/tcp_sockets.c

   1 /*
   2  * tcp_sockets.c
   3  *
   4  * TCP socket module for WebCit
   5  *
   6  * $Id$
   7  */
   8
   9 #include <stdlib.h>
  10 #include <unistd.h>
  11 #include <stdio.h>
  12 #include <signal.h>
  13 #include <sys/types.h>
  14 #include <sys/socket.h>
  15 #include <netinet/in.h>
  16 #include <arpa/inet.h>
  17 #include <netdb.h>
  18 #include <string.h>
  19 #include <stdarg.h>
  20 #include "webcit.h"
  21
  22 char server_is_local = 0;
  23
  24 #ifndef INADDR_NONE
  25 #define INADDR_NONE 0xffffffff
  26 #endif
  27
  28 extern int errno;
  29
  30 int serv_sock;
  31
  32
  33 void timeout(int signum) {
  34         printf("\rConnection timed out.\n");
  35         exit(3);
  36         }
  37
  38 int connectsock(char *host, char *service, char *protocol)
  39 {
  40         struct hostent *phe;
  41         struct servent *pse;
  42         struct protoent *ppe;
  43         struct sockaddr_in sin;
  44         int s,type;
  45
  46         bzero((char *)&sin,sizeof(sin));
  47         sin.sin_family = AF_INET;
  48
  49         pse=getservbyname(service,protocol);
  50         if (pse) {
  51                 sin.sin_port = pse->s_port;
  52                 }
  53         else if ((sin.sin_port = htons((u_short)atoi(service))) == 0) {
  54                 fprintf(stderr,"can't get %s service entry\n", service);
  55                 return(-1);
  56                 }
  57
  58         phe=gethostbyname(host);
  59         if (phe) {
  60                 bcopy(phe->h_addr,(char *)&sin.sin_addr,phe->h_length);
  61                 }
  62         else if ((sin.sin_addr.s_addr = inet_addr(host))==INADDR_NONE) {
  63                 fprintf(stderr,"can't get %s host entry: %s\n",
  64                         host,strerror(errno));
  65                 return(-1);
  66                 }
  67
  68         if ((ppe=getprotobyname(protocol))==0) {
  69                 fprintf(stderr,"can't get %s protocol entry: %s\n",
  70                         protocol,strerror(errno));
  71                 return(-1);
  72                 }
  73
  74         if (!strcmp(protocol,"udp"))
  75                 type = SOCK_DGRAM;
  76         else
  77                 type = SOCK_STREAM;
  78
  79         s = socket(PF_INET,type,ppe->p_proto);
  80         if (s<0) {
  81                 fprintf(stderr,"can't create socket: %s\n",strerror(errno));
  82                 return(-1);
  83                 }
  84
  85
  86         signal(SIGALRM,timeout);
  87         alarm(30);
  88
  89         if (connect(s,(struct sockaddr *)&sin,sizeof(sin))<0) {
  90                 fprintf(stderr,"can't connect to %s.%s: %s\n",
  91                         host,service,strerror(errno));
  92                 return(-1);
  93                 }
  94
  95         alarm(0);
  96         signal(SIGALRM,SIG_IGN);
  97
  98         return(s);
  99         }
 100
 101
 102
 103
 104 /*
 105  * Input binary data from socket
 106  */
 107 void serv_read(char *buf, int bytes)
 108 {
 109         int len,rlen;
 110
 111         len = 0;
 112         while(len<bytes) {
 113                 rlen = read(serv_sock,&buf[len],bytes-len);
 114                 if (rlen<1) {
 115                         fprintf(stderr, "Server connection broken: %s\n",
 116                                 strerror(errno));
 117                         connected = 0;
 118                         logged_in = 0;
 119                         return;
 120                         }
 121                 len = len + rlen;
 122                 }
 123         }
 124
 125
 126 /*
 127  * input string from pipe
 128  */
 129 void serv_gets(char *strbuf)
 130 {
 131         int ch,len;
 132         char buf[2];
 133
 134         len = 0;
 135         strcpy(strbuf,"");
 136         do {
 137                 serv_read(&buf[0], 1);
 138                 ch = buf[0];
 139                 strbuf[len++] = ch;
 140                 } while((ch!=10)&&(ch!=13)&&(ch!=0)&&(len<255));
 141         strbuf[len-1] = 0;
 142         /* fprintf(stderr, ">%s\n", strbuf); */
 143         }
 144
 145
 146
 147 /*
 148  * Attach to a Citadel server
 149  */
 150 void attach_to_server(int argc, char **argv)
 151 {
 152         if (argc==1) {
 153                 server_is_local = 1;
 154                 serv_sock = connectsock("localhost","citadel","tcp");
 155                 }
 156         if (argc==2) {
 157                 serv_sock = connectsock(argv[1],"citadel","tcp");
 158                 if ( (!strcmp(argv[1],"localhost"))
 159                 || (!strcmp(argv[1],"127.0.0.1")) ) server_is_local = 1;
 160                 }
 161         if (argc>=3) serv_sock = connectsock(argv[1],argv[2],"tcp");
 162
 163         if (serv_sock < 0) exit(errno);
 164         }
 165
 166
 167 /*
 168  * send binary to server
 169  */
 170 void serv_write(char *buf, int nbytes)
 171 {
 172         int bytes_written = 0;
 173         int retval;
 174         while (bytes_written < nbytes) {
 175                 retval = write(serv_sock, &buf[bytes_written],
 176                         nbytes - bytes_written);
 177                 if (retval < 1) {
 178                         fprintf(stderr, "Server connection broken: %s\n",
 179                                 strerror(errno));
 180                         connected = 0;
 181                         logged_in = 0;
 182                         return;
 183                         }
 184                 bytes_written = bytes_written + retval;
 185                 }
 186         }
 187
 188
 189 /*
 190  * send line to server
 191  */
 192 void serv_puts(char *string)
 193 {
 194         char buf[256];
 195
 196         sprintf(buf,"%s\n",string);
 197         serv_write(buf, strlen(buf));
 198         }
 199
 200
 201 /*
 202  * convenience function to send stuff to the server
 203  */
 204 void serv_printf(const char *format, ...) {
 205         va_list arg_ptr;
 206         char buf[256];
 207
 208         va_start(arg_ptr, format);
 209         vsprintf(buf, format, arg_ptr);
 210         va_end(arg_ptr);
 211
 212         strcat(buf, "\n");
 213         serv_write(buf, strlen(buf));
 214         /* fprintf(stderr, "<%s", buf); */
 215         }
 216
 217