]> hackdaworld.org Git - my-code/ivac.git/blob - network.c
b9c68aa09316e0ba8613daa28998d05edf199c1a
[my-code/ivac.git] / 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,int port) {
270
271   struct sockaddr_in addr;
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     perror("[network] bind call (udp)");
284     return N_ERROR;
285   }
286
287   printf("[network] listening on port %d (udp)\n",net->l_udp_port);
288
289   if((net->s_udp_fd=socket(AF_INET,SOCK_DGRAM,0))==-1) {
290     perror("[network] socket call (udp-send)");
291     return N_ERROR;
292   }
293
294   return N_SUCCESS;
295 }
296
297 int network_set_udp_ports(t_net *net,int port) {
298
299   net->l_udp_port=port;
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 }