network/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   fprintf(stderr,"[network] initializing network ...\n");
  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   if(bind(net->l_fd,(struct sockaddr *)&addr,
  31           sizeof(struct sockaddr))==-1) {
  32     /* try harder ... */
  33     true=1;
  34     if(setsockopt(net->l_fd,SOL_SOCKET,SO_REUSEADDR,&true,sizeof(true))==-1) {
  35       perror("[network] setsockopt call");
  36       return N_ERROR;
  37     }
  38     fprintf(stderr,"[network] reused address\n");
  39   }
  40
  41   if(listen(net->l_fd,MAX_LISTEN_QUEUE)==-1) {
  42     perror("[network] listen call");
  43     return N_ERROR;
  44   }
  45
  46   fprintf(stderr,"[network] listen on %s port %d\n",inet_ntoa(addr.sin_addr),
  47                                                     net->l_port);
  48
  49   return N_SUCCESS;
  50 }
  51
  52 int network_shutdown(t_net *net) {
  53
  54   int channel;
  55
  56   for(channel=0;channel<MAX_CONNECTIONS;channel++)
  57     if(net->connection[channel].status&C_SOCKET)
  58       close(net->connection[channel].fd);
  59
  60   if(close(net->l_fd)==-1) {
  61     perror("[network] close call");
  62     return N_ERROR;
  63   }
  64
  65   fprintf(stderr,"[network] shutdown\n");
  66
  67   return N_SUCCESS;
  68 }
  69
  70 int network_set_listen_port(t_net *net,in_port_t port) {
  71
  72   net->l_port=port;
  73
  74   return N_SUCCESS;
  75 }
  76
  77 int network_manage_connection(t_net *net) {
  78
  79   int i;
  80   struct sockaddr_in addr;
  81
  82   for(i=0;i<MAX_CONNECTIONS;i++) {
  83
  84     if(net->connection[i].status&C_IN_USE) {
  85
  86       if(net->connection[i].status&C_HANGUP) {
  87         if(close(net->connection[i].fd)==-1) {
  88           perror("[network] close call");
  89           return N_E_CLOSE;
  90         }
  91         fprintf(stderr,"[network] connection %d closed\n",i);
  92         net->connection[i].status=0;
  93       }
  94
  95       if(net->connection[i].status&C_INFO_A) {
  96
  97         if(!(net->connection[i].status&C_SOCKET)) {
  98           if((net->connection[i].fd=socket(AF_INET,SOCK_STREAM,0))==-1) {
  99             perror("[network] socket call");
 100             return N_ERROR;
 101           }
 102           net->connection[i].status|=C_SOCKET;
 103         }
 104
 105         if((!(net->connection[i].status&C_ESTABL))&&
 106            (net->connection[i].status&C_SOCKET)) {
 107
 108           memset(&addr,0,sizeof(struct sockaddr));
 109           addr.sin_family=AF_INET;
 110           addr.sin_port=htons(net->connection[i].port);
 111           if(!inet_aton(net->connection[i].ip,&(addr.sin_addr))) {
 112             perror("[network] inet_aton call");
 113             return N_ERROR;
 114           }
 115
 116           if(connect(net->connection[i].fd,(struct sockaddr *)&addr,
 117                      sizeof(struct sockaddr))==-1) {
 118             perror("[network] connect call");
 119             return N_E_CONNECT;
 120           }
 121
 122           fprintf(stderr,"[network] established connection to ");
 123           fprintf(stderr,"%s port %d on channel %d\n",net->connection[i].ip,
 124                                                      net->connection[i].port,i);
 125           net->connection[i].status|=C_ESTABL;
 126
 127         }
 128
 129       }
 130
 131     }
 132
 133   }
 134
 135   return N_SUCCESS;
 136 }
 137
 138 int network_connect(t_net *net,int channel) {
 139
 140   if(net->connection[channel].status&C_IN_USE) {
 141     fprintf(stderr,"[network] connect failed, channel %02d in use\n",channel);
 142     return N_E_IN_USE;
 143   }
 144   if(!(net->connection[channel].status&C_INFO_A)) {
 145     fprintf(stderr,"[network] connect failed, missing configuration for channel %02d\n",channel);
 146     return N_E_NO_INFO;
 147   }
 148
 149   /* try connect & return result */
 150   net->connection[channel].status|=C_IN_USE;
 151   return(network_manage_connection(net)); /* could be other channel too */
 152 }
 153
 154 int network_close(t_net *net,int channel) {
 155
 156   if(!(net->connection[channel].status&C_ESTABL)) {
 157     fprintf(stderr,"[network] close failed, channel %02d not active\n",channel);
 158     return N_E_NC;
 159   }
 160
 161   net->connection[channel].status|=C_HANGUP;
 162   return(network_manage_connection(net)); /* could be other channel too */
 163 }
 164
 165 int network_close_all(t_net *net) {
 166
 167   int channel;
 168
 169   for(channel=0;channel<MAX_CONNECTIONS;channel++)
 170     if(net->connection[channel].status&C_ESTABL) network_close(net,channel);
 171
 172   return N_SUCCESS;
 173 }
 174
 175 int network_set_connection_info(t_net *net,int channel,char *ip,int port) {
 176
 177   if(net->connection[channel].status&C_IN_USE) {
 178     fprintf(stderr,"[network] set connection failed, channel %02d in use\n",
 179                    channel);
 180     return N_E_IN_USE;
 181   }
 182
 183   strncpy(net->connection[channel].ip,ip,IP_DIGITS);
 184   net->connection[channel].port=port;
 185   net->connection[channel].status|=C_INFO_A;
 186
 187   return N_SUCCESS;
 188 }
 189
 190 int network_select(t_net *net,int channel) {
 191
 192   int mask;
 193
 194   if(channel==MAX_CONNECTIONS) mask=0xffffffff;
 195   else mask=(1<<channel);
 196   net->sendmask|=mask;
 197
 198   return N_SUCCESS;
 199 }
 200
 201 int network_deselect(t_net *net,int channel) {
 202
 203   int mask;
 204
 205   if(channel==MAX_CONNECTIONS) mask=0;
 206   else mask=~(1<<channel);
 207   net->sendmask&=mask;
 208
 209   return N_SUCCESS;
 210 }
 211
 212 int network_manage_incoming(t_net *net) {
 213
 214   int channel;
 215   struct sockaddr_in addr;
 216   int len;
 217
 218   for(channel=0;channel<MAX_CONNECTIONS;channel++) {
 219     if(!(net->connection[channel].status&C_IN_USE)) {
 220       if((net->connection[channel].fd=accept(net->l_fd,
 221                                        (struct sockaddr *)&addr,
 222                                        &len))==-1) {
 223         perror("[network] accept call");
 224         return N_E_ACCEPT;
 225       }
 226       strncpy(net->connection[channel].ip,inet_ntoa(addr.sin_addr),IP_DIGITS);
 227       net->connection[channel].port=ntohs(addr.sin_port);
 228       net->connection[channel].status=C_IN_USE|C_INFO_A|C_SOCKET|C_ESTABL;
 229       fprintf(stderr,"[network] established connection from %s port %d on channel %d\n",net->connection[channel].ip,net->connection[channel].port,channel);
 230       return channel;
 231     }
 232   }
 233
 234   fprintf(stderr,"[network] maximum connections reached\n");
 235   return N_E_MAXC;
 236 }
 237
 238 int network_send(int fd,unsigned char *data,int datasize) {
 239
 240   int count,left;
 241
 242   count=0;
 243   left=datasize;
 244
 245   while(left) {
 246     if((count=write(fd,data+datasize-left,left))==-1) {
 247       perror("[network] write call");
 248       return N_ERROR;
 249     }
 250     left-=count;
 251   }
 252
 253   return N_SUCCESS;
 254 }
 255
 256 int network_receive(int fd,unsigned char *data,int datasize) {
 257
 258   int count;
 259
 260   if((count=read(fd,data,datasize))==-1) {
 261     perror("[network] read call");
 262     return N_ERROR;
 263   }
 264
 265   return count;
 266 }
 267
 268 int network_udp_listen_init(t_net *net) {
 269
 270   struct sockaddr_in addr;
 271   int true;
 272
 273   if((net->l_udp_fd=socket(AF_INET,SOCK_DGRAM,0))==-1) {
 274     perror("[network] socket call (udp-receive)");
 275     return N_ERROR;
 276   }
 277
 278   memset(&addr,0,sizeof(struct sockaddr));
 279   addr.sin_family=AF_INET;
 280   addr.sin_port=htons(net->l_udp_port);
 281   addr.sin_addr.s_addr=INADDR_ANY;
 282   if(bind(net->l_udp_fd,(struct sockaddr *)&addr,sizeof(struct sockaddr))==-1) {
 283     /* try harder ... */
 284     true=1;
 285     if(setsockopt(net->l_udp_fd,SOL_SOCKET,SO_REUSEADDR,
 286                   &true,sizeof(true))==-1) {
 287       perror("[network] setsockopt call (udp)");
 288       return N_ERROR;
 289     }
 290     fprintf(stderr,"[network] reused address (udp)\n");
 291   }
 292
 293   fprintf(stderr,"[network] listening on port %d (udp)\n",net->l_udp_port);
 294
 295   if((net->s_udp_fd=socket(AF_INET,SOCK_DGRAM,0))==-1) {
 296     perror("[network] socket call (udp-send)");
 297     return N_ERROR;
 298   }
 299
 300   return N_SUCCESS;
 301 }
 302
 303 int network_udp_receive(t_net *net,int channel, unsigned char *data,int count) {
 304
 305   struct sockaddr_in addr;
 306   socklen_t len;
 307
 308   if((count=recvfrom(net->l_udp_fd,data,count,0,
 309                      (struct sockaddr *)&addr,&len))==-1) {
 310     perror("[network] recvfrom call");
 311     return N_ERROR;
 312   }
 313
 314   if(strncmp(net->connection[channel].ip,inet_ntoa(addr.sin_addr),IP_DIGITS)) {
 315     fprintf(stderr,"[network] packet from unknown: %s\n",
 316             inet_ntoa(addr.sin_addr));
 317     return N_UDP_WRONG_SENDER;
 318   }
 319
 320   return N_SUCCESS;
 321 }
 322
 323 int network_udp_send(t_net *net,int channel, unsigned char *data,int size) {
 324
 325   int count,left;
 326   struct sockaddr_in addr;
 327
 328   count=0;
 329   left=count;
 330
 331   memset(&addr,0,sizeof(struct sockaddr));
 332   addr.sin_family=AF_INET;
 333   addr.sin_port=htons(net->l_udp_port);
 334   inet_aton(net->connection[channel].ip,&(addr.sin_addr));
 335
 336   while(left) {
 337     if((count=sendto(net->s_udp_fd,data+size-left,left,0,
 338                      (struct sockaddr *)&addr,sizeof(struct sockaddr)))==-1) {
 339       perror("[network] sendto call");
 340       return N_ERROR;
 341     }
 342     left-=count;
 343   }
 344
 345   return N_SUCCESS;
 346 }
 347
 348 int network_udp_shutdown(t_net *net) {
 349
 350   if(close(net->l_udp_fd)==-1) {
 351     perror("[network] close call (udp-receive)");
 352     return N_ERROR;
 353   }
 354
 355   if(close(net->s_udp_fd)==-1) {
 356     perror("[network] close call (udp-send)");
 357     return N_ERROR;
 358   }
 359
 360   return N_SUCCESS;
 361 }