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   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     puts("[network] reused address");
  39   }
  40
  41   if(listen(net->l_fd,MAX_LISTEN_QUEUE)==-1) {
  42     perror("[network] listen call");
  43     return N_ERROR;
  44   }
  45
  46   printf("[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   puts("[network] shutdown");
  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         printf("[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           printf("[network] established connection to ");
 123           printf("%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     printf("[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     printf("[network] connect failed, missing configuration for channel %02d\n",
 146            channel);
 147     return N_E_NO_INFO;
 148   }
 149
 150   /* try connect & return result */
 151   net->connection[channel].status|=C_IN_USE;
 152   return(network_manage_connection(net)); /* could be other channel too */
 153 }
 154
 155 int network_close(t_net *net,int channel) {
 156
 157   if(!(net->connection[channel].status&C_ESTABL)) {
 158     printf("[network] close failed, channel %02d not active\n",channel);
 159     return N_E_NC;
 160   }
 161
 162   net->connection[channel].status|=C_HANGUP;
 163   return(network_manage_connection(net)); /* could be other channel too */
 164 }
 165
 166 int network_close_all(t_net *net) {
 167
 168   int channel;
 169
 170   for(channel=0;channel<MAX_CONNECTIONS;channel++)
 171     if(net->connection[channel].status&C_ESTABL) network_close(net,channel);
 172
 173   return N_SUCCESS;
 174 }
 175
 176 int network_set_connection_info(t_net *net,int channel,char *ip,int port) {
 177
 178   if(net->connection[channel].status&C_IN_USE) {
 179     printf("[network] set connection failed, channel %02d in use\n",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       printf("[network] established connection from %s port %d on channel %d\n",
 230              net->connection[channel].ip,net->connection[channel].port,channel);
 231       return channel;
 232     }
 233   }
 234
 235   puts("[network] maximum connections reached");
 236   return N_E_MAXC;
 237 }
 238
 239 int network_send(int fd,unsigned char *data,int datasize) {
 240
 241   int count,left;
 242
 243   count=0;
 244   left=datasize;
 245
 246   while(left) {
 247     if((count=write(fd,data+datasize-left,left))==-1) {
 248       perror("[network] write call");
 249       return N_ERROR;
 250     }
 251     left-=count;
 252   }
 253
 254   return N_SUCCESS;
 255 }
 256
 257 int network_receive(int fd,unsigned char *data,int datasize) {
 258
 259   int count;
 260
 261   if((count=read(fd,data,datasize))==-1) {
 262     perror("[network] read call");
 263     return N_ERROR;
 264   }
 265
 266   return count;
 267 }
 268
 269 int network_udp_listen_init(t_net *net) {
 270
 271   struct sockaddr_in addr;
 272   int true;
 273
 274   if((net->l_udp_fd=socket(AF_INET,SOCK_DGRAM,0))==-1) {
 275     perror("[network] socket call (udp-receive)");
 276     return N_ERROR;
 277   }
 278
 279   memset(&addr,0,sizeof(struct sockaddr));
 280   addr.sin_family=AF_INET;
 281   addr.sin_port=htons(net->l_udp_port);
 282   addr.sin_addr.s_addr=INADDR_ANY;
 283   if(bind(net->l_udp_fd,(struct sockaddr *)&addr,sizeof(struct sockaddr))==-1) {
 284     /* try harder ... */
 285     true=1;
 286     if(setsockopt(net->l_udp_fd,SOL_SOCKET,SO_REUSEADDR,
 287                   &true,sizeof(true))==-1) {
 288       perror("[network] setsockopt call (udp)");
 289       return N_ERROR;
 290     }
 291     puts("[network] reused address (udp)");
 292   }
 293
 294   printf("[network] listening on port %d (udp)\n",net->l_udp_port);
 295
 296   if((net->s_udp_fd=socket(AF_INET,SOCK_DGRAM,0))==-1) {
 297     perror("[network] socket call (udp-send)");
 298     return N_ERROR;
 299   }
 300
 301   return N_SUCCESS;
 302 }
 303
 304 int network_udp_receive(t_net *net,int channel, unsigned char *data,int count) {
 305
 306   struct sockaddr_in addr;
 307   socklen_t len;
 308
 309   if((count=recvfrom(net->l_udp_fd,data,count,0,
 310                      (struct sockaddr *)&addr,&len))==-1) {
 311     perror("[network] recvfrom call");
 312     return N_ERROR;
 313   }
 314
 315   if(strncmp(net->connection[channel].ip,inet_ntoa(addr.sin_addr),IP_DIGITS)) {
 316     printf("[network] packet from unknown: %s\n",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 }