src/network.c

   1 /* network.c -- network management stuff
   2  *
   3  * author: hackbard@hackdaworld.dyndns.org
   4  *
   5  */
   6
   7 #include "network.h"
   8
   9 int network_init(t_net *net) {
  10
  11   struct sockaddr_in addr;
  12   int true;
  13
  14   puts("[network] initializing network ...");
  15
  16   memset(net->connection,0,MAX_CONNECTIONS*sizeof(t_connection));
  17   net->c_count=0;
  18   net->sendmask=0;
  19
  20   if((net->l_fd=socket(AF_INET,SOCK_STREAM,0))==-1) {
  21     perror("[network] socket call");
  22     return N_ERROR;
  23   }
  24
  25   memset(&addr,0,sizeof(struct sockaddr));
  26   addr.sin_family=AF_INET;
  27   addr.sin_port=htons(net->l_port);
  28   addr.sin_addr.s_addr=INADDR_ANY;
  29
  30   /* prevent addres in use error message */
  31   true=1;
  32   if(setsockopt(net->l_fd,SOL_SOCKET,SO_REUSEADDR,&true,sizeof(true))==-1) {
  33     perror("[network] setsockopt call");
  34     return N_ERROR;
  35   }
  36
  37   if(bind(net->l_fd,(struct sockaddr *)&addr,
  38           sizeof(struct sockaddr))==-1) {
  39     perror("[network] bind call");
  40     return N_ERROR;
  41   }
  42
  43   if(listen(net->l_fd,MAX_LISTEN_QUEUE)==-1) {
  44     perror("[network] listen call");
  45     return N_ERROR;
  46   }
  47
  48   printf("[network] listen on %s port %d\n",inet_ntoa(addr.sin_addr),
  49                                           net->l_port);
  50
  51   return N_SUCCESS;
  52 }
  53
  54 int network_shutdown(t_net *net) {
  55
  56   if(close(net->l_fd)==-1) {
  57     perror("[network] close call");
  58     return N_ERROR;
  59   }
  60
  61   puts("[network] shutdown");
  62
  63   return N_SUCCESS;
  64 }
  65
  66 int network_set_listen_port(t_net *net,in_port_t port) {
  67
  68   net->l_port=port;
  69
  70   return N_SUCCESS;
  71 }
  72
  73 int network_manage_connection(t_net *net) {
  74
  75   int i;
  76   struct sockaddr_in addr;
  77
  78   for(i=0;i<MAX_CONNECTIONS;i++) {
  79
  80     if(net->connection[i].status&C_IN_USE) {
  81
  82       if(net->connection[i].status&C_HANGUP) {
  83         if(close(net->connection[i].fd)==-1) {
  84           perror("[network] close call");
  85           return N_ERROR;
  86         }
  87         printf("[network] connection %d closed\n",i);
  88         net->connection[i].status=0;
  89       }
  90
  91       if(net->connection[i].status&C_INFO_A) {
  92
  93         if(!net->connection[i].status&C_SOCKET) {
  94           if((net->connection[i].fd=socket(AF_INET,SOCK_STREAM,0))==-1) {
  95             perror("[network] socket call");
  96             return N_ERROR;
  97           }
  98         }
  99
 100         if(!net->connection[i].status&C_ESTABL) {
 101
 102           memset(&addr,0,sizeof(struct sockaddr));
 103           addr.sin_family=AF_INET;
 104           addr.sin_port=htons(net->connection[i].port);
 105           if(!inet_aton(net->connection[i].ip,&(addr.sin_addr))) {
 106             perror("[network] inet_aton call");
 107             return N_ERROR;
 108           }
 109
 110           if(connect(net->connection[i].fd,(struct sockaddr *)&addr,
 111                      sizeof(struct sockaddr))==-1) {
 112             perror("[network] connect call");
 113             return N_ERROR;
 114           }
 115
 116           printf("[network] established connection to %s port %d on channel %d\n",
 117                  net->connection[i].ip,net->connection[i].port,i);
 118
 119         }
 120
 121       }
 122
 123     }
 124
 125   }
 126
 127   return N_SUCCESS;
 128 }
 129
 130 int network_manage_incoming(t_net *net) {
 131
 132   int channel;
 133   struct sockaddr_in addr;
 134   int len;
 135
 136   for(channel=0;channel<MAX_CONNECTIONS;channel++) {
 137     if(!net->connection[channel].status&C_IN_USE) {
 138       if((net->connection[channel].fd=accept(net->l_fd,
 139                                        (struct sockaddr *)&addr,
 140                                        &len))==-1) {
 141         perror("[network] accept call");
 142         return N_ERROR;
 143       }
 144       strncpy(net->connection[channel].ip,inet_ntoa(addr.sin_addr),IP_DIGITS);
 145       net->connection[channel].port=ntohs(addr.sin_port);
 146       net->connection[channel].status=C_IN_USE|C_INFO_A|C_SOCKET|C_ESTABL;
 147       printf("[network] established connection from %s port %d on channel %d\n",
 148              net->connection[channel].ip,net->connection[channel].port,channel);
 149       return channel;
 150     }
 151   }
 152
 153   puts("[network] maximum connections reached");
 154   return N_ERROR;
 155 }
 156
 157 int network_send(int fd,unsigned char *data,int datasize) {
 158
 159   int count,left;
 160
 161   count=0;
 162   left=datasize;
 163
 164   while(left) {
 165     if((count=write(fd,data+datasize-left,left))==-1) {
 166       perror("[network] write call");
 167       return N_ERROR;
 168     }
 169     left-=count;
 170   }
 171
 172   return N_SUCCESS;
 173 }
 174
 175 int network_receive(int fd,unsigned char *data,int datasize) {
 176
 177   int count;
 178
 179   if((count=read(fd,data,datasize))==-1) {
 180     perror("[network] read call");
 181     return N_ERROR;
 182   }
 183
 184   return count;
 185 }
 186
 187 int send_info(int channel,t_net *net,char *name) {
 188
 189   char data[SEND_N_MAX];
 190   int size;
 191
 192   size=strlen(name);
 193
 194   data[0]=SEND_N_NAME;
 195   data[1]=size;
 196   strncpy(data+2,name,size);
 197   size+=2;
 198
 199   data[size]=SEND_N_G_CAP;
 200   data[size+1]=sizeof(unsigned char);
 201   data[size+1+sizeof(unsigned char)]=net->cap;
 202   size+=(sizeof(unsigned char)+2);
 203
 204   data[size]=SEND_N_AV_CAP;
 205   data[size+1]=sizeof(unsigned short);
 206   data[size+1+sizeof(unsigned short)]=net->avcap;
 207   size+=(sizeof(unsigned short)+2);
 208
 209   if(network_send(net->connection[channel].fd,data,size)==N_ERROR) {
 210     puts("[network] send_info failed");
 211     return N_ERROR;
 212   }
 213
 214   return N_SUCCESS;
 215 }
 216
 217 int receive_info(int channel,t_net *net) {
 218
 219   char data[CHAR_N_UNAME+2];
 220   int count,length;
 221
 222   count=0;
 223
 224   if((length=network_receive(net->connection[channel].fd,
 225                              data,SEND_N_MAX))==N_ERROR) {
 226     puts("[network] receive_info failed");
 227     return N_ERROR;
 228   }
 229
 230   while(length-count) {
 231     switch(data[count]) {
 232       case SEND_N_NAME:
 233         strncpy(net->connection[channel].name,&data[count+2],data[count+1]);
 234         net->connection[channel].name[(int)data[count+2]]='\0';
 235         count+=(data[count+2]+2);
 236         break;
 237       case SEND_N_G_CAP:
 238         net->connection[channel].cap=data[count+4];
 239         count+=(sizeof(unsigned char)+2);
 240         break;
 241       case SEND_N_AV_CAP:
 242         net->connection[channel].avcap=data[count+3]<<8;
 243         net->connection[channel].avcap|=data[count+4];
 244         count+=(sizeof(unsigned short)+2);
 245         break;
 246       default:
 247         puts("[network] receive_info, unknown character");
 248         return N_ERROR;
 249     }
 250   }
 251
 252   return N_SUCCESS;
 253 }