47d1892ae77d0ca3baededd9f40310d4a5a7a9fc
[my-code/ivac.git] / src / ivac.c
1 /* ivac.c -- main ivac file
2  *
3  * author: hackbard@hackdaworld.dyndns.org
4  *         frank.zirkelbach@physik.uni-augsburg.de
5  *
6  * Copyright (C) 2004 Frank Zirkelbach
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21  *
22  */
23
24 #include "ivac.h"
25
26 int usage(void) {
27
28   puts("");
29   puts("usage: ivac <options>");
30   puts("");
31   puts("options:");
32   puts("-h \t\t show this help");
33   puts("-n <name> \t specify your name");
34   puts("-p <port> \t specify port to listen for incoming connections");
35   puts("-u <port> \t specify udp data port");
36   puts("-d <device> \t specify audio device");
37   puts("");
38
39   return SUCCESS;
40 }
41
42 int main(int argc,char **argv) {
43
44   /* TESTING BY NOW */
45
46   t_ivac ivac;
47   int i;
48
49   /* default values */
50   strcpy(ivac.username,"ivac");
51   ivac.net.l_port=IVAC_LISTEN_PORT;
52   ivac.net.l_udp_port=IVAC_UDP_PORT;
53   strcpy(ivac.audio.dsp_dev,SOUND_DEVICE);
54  
55   /* parse argv and change default values */
56   for(i=1;i<argc;i++) {
57     if(argv[i][0]=='-') {
58       switch(argv[i][1]) {
59         case 'h':
60           usage();
61           return SUCCESS;
62         case 'n':
63           strncpy(ivac.username,argv[++i],CHAR_USERNAME);
64           break;
65         case 'p':
66           ivac.net.l_port=atoi(argv[++i]);
67           break;
68         case 'u':
69           ivac.net.l_udp_port=atoi(argv[++i]);
70           break;
71         case 'd':
72           strncpy(ivac.audio.dsp_dev,argv[++i],MAX_CHAR_DEVICE);
73           break;
74         default:
75           usage();
76           return ERROR;
77       }
78     }
79     else usage();
80   }
81
82   /* clear challenger struct */
83   for(i=0;i<MAX_CONNECTIONS;i++)
84     memset(&(ivac.challenger[i]),0,sizeof(t_challenger));
85   /* zero console buffer */
86   for(i=0;i<IVAC_CONSOLE_LEN;i++)
87     memset(ivac.console[i],0,IVAC_CONSOLE_STRING_LEN);
88
89   /* set capabilities (futur: set by check routines) */
90   ivac.g_cap=NETWORK;
91   ivac.av_cap=AUDIO|VIDEO|DUPLEX;
92
93   /* set event timeout */
94   ivac.event.timeout.tv_sec=IVAC_S_SEC;
95   ivac.event.timeout.tv_usec=IVAC_S_USEC;
96   /* event init */
97   event_init(&(ivac.event));
98
99   /* input init */
100   ivac.input.mode=CONTENT_BUFFER;
101   input_init(&(ivac.input));
102
103   /* network init */
104   if(network_init(&(ivac.net))==N_ERROR) {
105     printf("[ivac] use 'fuser -n tcp %d' to determine the process to kill!\n",
106            ivac.net.l_port);
107     ivac_shutdown(&ivac);
108     return ERROR;
109   }
110   network_udp_listen_init(&(ivac.net));
111
112   /* add listening port(s) + stdin to (read) event system */
113   event_math(ivac.net.l_fd,&(ivac.event),READ,ADD);
114   event_math(ivac.net.l_udp_fd,&(ivac.event),READ,ADD);
115   event_math(0,&(ivac.event),READ,ADD);
116
117   /* display init */
118   display_init(&(ivac.display));
119
120   /* use hardcoded audio settings by now */
121   ivac.audio.fmt=BIT_8;
122   ivac.audio.channels=MONO;
123   ivac.audio.speed=8000;
124   /* audio init */
125   audio_init(&(ivac.audio));
126   audio_setup(&(ivac.audio));
127
128   /* display */
129   ivac_display(&(ivac.display));
130
131   /* start event system - callbacks used: ivac_event_cb + ivac_regular_cb */
132   event_start(&(ivac.event),(void *)&ivac,ivac_event_cb,ivac_regular_cb);
133
134   return SUCCESS;
135 }
136
137 int ivac_shutdown(t_ivac *ivac) {
138
139   network_shutdown(&(ivac->net));
140   network_udp_shutdown(&(ivac->net));
141   input_shutdown(&(ivac->input));
142   event_stop(&(ivac->event));
143   audio_shutdown(&(ivac->audio));
144   display_shutdown(&(ivac->display));
145
146   return SUCCESS;
147 }
148
149 int ivac_send_info(int channel,t_ivac *ivac) {
150
151   char data[SEND_N_MAX];
152   int size;
153
154   size=strlen(ivac->username);
155
156   data[0]=IVAC_SEND_NAME;
157   data[1]=size;
158   strncpy(data+2,ivac->username,size);
159   size+=2;
160
161   data[size]=IVAC_SEND_G_CAP;
162   data[size+1]=1;
163   data[size+2]=ivac->g_cap;
164   size+=3;
165
166   data[size]=IVAC_SEND_AV_CAP;
167   data[size+1]=2;
168   data[size+2]=(ivac->av_cap)>>8;
169   data[size+3]=(ivac->av_cap)&0xff;
170   size+=4;
171
172   if(network_send(ivac->net.connection[channel].fd,data,size)==N_ERROR) {
173     puts("[ivac] ivac_send_info failed");
174     return ERROR;
175   }
176
177   return SUCCESS;
178 }
179
180 int ivac_send_quit(int channel,t_ivac *ivac) {
181
182   char data[7]; /* one more for \0 */
183
184   data[0]=IVAC_SEND_QUIT;
185   data[1]=4;
186   strcpy(data+2,"quit");
187
188   if(network_send(ivac->net.connection[channel].fd,data,6)==N_ERROR) {
189     puts("[ivac] ivac_send_quit failed");
190     return ERROR;
191   }
192
193   return SUCCESS;
194 }
195
196 int ivac_receive_info(int channel,t_ivac *ivac) {
197
198   char data[SEND_N_MAX];
199   char c_str[IVAC_CONSOLE_STRING_LEN];
200   int count,length;
201
202   count=0;
203
204   if((length=network_receive(ivac->net.connection[channel].fd,
205                              data,SEND_N_MAX))==N_ERROR) {
206     puts("[ivac] ivac_receive_info failed");
207     return ERROR;
208   }
209
210   while(length-count) {
211     switch(data[count]) {
212       case IVAC_SEND_NAME:
213         strncpy(ivac->challenger[channel].name,data+count+2,data[count+1]);
214         ivac->challenger[channel].name[data[count+1]]='\0';
215         count+=(data[count+1]+2);
216         break;
217       case IVAC_SEND_G_CAP:
218         ivac->challenger[channel].g_cap=data[count+2];
219         count+=3;
220         break;
221       case IVAC_SEND_AV_CAP:
222         ivac->challenger[channel].av_cap=data[count+2]<<8;
223         ivac->challenger[channel].av_cap|=data[count+3];
224         count+=4;
225         break;
226       case IVAC_SEND_QUIT:
227         if(!(strncmp(data+count+2,"quit",data[1])))
228           sprintf(c_str,"channel %02d: connection closed by remote host",
229                   channel);
230           event_math(ivac->net.connection[channel].fd,&(ivac->event),
231                      READ,REMOVE);
232           network_close(&(ivac->net),channel);
233           memset(&(ivac->challenger[channel]),0,sizeof(t_challenger));
234           ivac_add_to_monitor(ivac,c_str);
235           count+=6;
236           break;
237       default:
238         sprintf(c_str,"ivac_receive_info, unknown character: 0x%02x\n",
239                data[count]);
240         ivac_add_to_monitor(ivac,c_str);
241         return ERROR;
242         break;
243     }
244   }
245
246   return length;
247 }
248
249 int ivac_event_cb(t_event *event,void *ptr) {
250  
251   t_ivac *ivac;
252   int channel;
253   char c_str[IVAC_CONSOLE_STRING_LEN];
254
255   ivac=(t_ivac *)ptr;
256
257   /* incoming connection -- first contact => send info */
258   if(FD_ISSET(ivac->net.l_fd,&(event->rfds))) {
259     channel=network_manage_incoming(&(ivac->net));
260     if(channel==N_E_ACCEPT)
261       sprintf(c_str,"accept failed");
262     else if(channel==N_E_MAXC)
263       sprintf(c_str,"maximum connections reached");
264     else {
265       sprintf(c_str,"connection from %s port %d on channel %d",
266               ivac->net.connection[channel].ip,
267               ivac->net.connection[channel].port,channel);
268       ivac_add_to_monitor(ivac,c_str);
269       event_math(ivac->net.connection[channel].fd,event,READ,ADD);
270       ivac_send_info(channel,ivac);
271     }
272     ivac_display_content(ivac);
273   }
274
275   /* wait for user info */
276   for(channel=0;channel<MAX_CONNECTIONS;channel++) {
277     if(ivac->net.connection[channel].status&C_ESTABL) {
278       /* remote is sending info */
279       if(FD_ISSET(ivac->net.connection[channel].fd,&(event->rfds))) {
280         if(ivac_receive_info(channel,ivac)==0) {
281           event_math(ivac->net.connection[channel].fd,event,READ,REMOVE);
282           network_close(&(ivac->net),channel);
283           sprintf(c_str,"channel %02d: broken pipe - disconnected",channel);
284           ivac_add_to_monitor(ivac,c_str);
285         }
286         ivac_display_content(ivac);
287       }
288     }
289   } 
290
291   /* user interaction */
292   if(FD_ISSET(0,&(event->rfds)))
293     input_get_event(&(ivac->input),ivac_parse_command,ivac);
294
295   return SUCCESS;
296 }
297
298 int ivac_regular_cb(t_event *event,void *ptr) {
299
300   /* usual jobs like audio & video transmit ... */
301
302   return SUCCESS;
303 }
304
305 int ivac_parse_command(t_input *input,void *ptr) {
306
307   t_ivac *ivac;
308   int i,j,k;
309   int len;
310   int channel;
311   char *data,valid;
312   char c_str[IVAC_CONSOLE_STRING_LEN];
313   char arg[IVAC_ARG_COUNT][IVAC_ARG_LEN];
314
315   ivac=(t_ivac *)ptr;
316   data=input->content;
317   valid=0;
318
319   /* refresh prompt content only! */
320   ivac_display_prompt_content(ivac);
321
322   /* parse command routines */
323   if(data[input->c_count-1]=='\n') {
324
325     /* delete console string + args */
326     memset(c_str,0,IVAC_CONSOLE_STRING_LEN);
327     for(j=0;j<IVAC_ARG_COUNT;j++) memset(arg[j],0,IVAC_ARG_LEN);
328
329     /* get args */
330     len=0;
331     while(data[len]!='\n') len++;
332     i=0; j=0;
333     while((i<len) && (j<IVAC_ARG_COUNT)) {
334       k=0;
335       while((data[i+k]!=' ')&&(data[i+k]!='\n')) {
336         arg[j][k]=data[i+k];
337         k++;
338       }
339       arg[j][k]='\0';
340       j++;
341       /* skip all ' ' */
342       while(data[i+k]==' ') k++;
343       i+=k;
344     }
345
346     /* parse command  aka arg[0] */
347     if(!(strncmp(arg[0],"quit",4))) {
348       valid=1;
349       sprintf(c_str,"ivac shutdown ...");
350       for(i=0;i<MAX_CONNECTIONS;i++) {
351         if(ivac->net.connection[i].status&C_ESTABL) {
352           ivac_send_quit(i,ivac);
353           network_close(&(ivac->net),i);
354         }
355       }
356       ivac_shutdown(ivac);
357     }
358     if(!(strncmp(arg[0],"set",3))) {
359       valid=1;
360       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
361       else channel=-1;
362       if(!(strncmp(arg[1],"name",4))) {
363         strncpy(ivac->username,arg[2],CHAR_USERNAME);
364         sprintf(c_str,"changed username to %s",ivac->username);
365       }
366       else if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
367         i=network_set_connection_info(&(ivac->net),channel,arg[2],atoi(arg[3]));
368         if(i==N_E_IN_USE)
369           sprintf(c_str,"channel %02d: connection in use",channel);
370         if(i==N_SUCCESS)
371           sprintf(c_str,"channel %02d: set connection info",channel);
372       } else snprintf(c_str,IVAC_CONSOLE_STRING_LEN,"unknown argument: '%s'",
373                       arg[1]);
374     }
375     if(!(strncmp(arg[0],"connect",7))) {
376       valid=1;
377       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
378       else channel=-1;
379       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
380         i=network_connect(&(ivac->net),channel);
381         if(i==N_E_IN_USE)
382           sprintf(c_str,"channel %02d: connection in use",channel);
383         else if(i==N_E_NO_INFO)
384           sprintf(c_str,"channel %02d: channel not configured",channel);
385         else {
386           sprintf(c_str,"channel %02d: connected to %s:%d",channel,
387                   ivac->net.connection[channel].ip,
388                   ivac->net.connection[channel].port);
389           event_math(ivac->net.connection[channel].fd,&(ivac->event),READ,ADD);
390           ivac_send_info(channel,ivac);
391         }
392       }
393       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
394     } 
395     if(!(strncmp(arg[0],"close",5))) {
396       valid=1;
397       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
398       else channel=-1;
399       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
400         if(ivac->net.connection[channel].status&C_ESTABL) {
401           ivac_send_quit(channel,ivac);
402           event_math(ivac->net.connection[channel].fd,&(ivac->event),
403                      READ,REMOVE);
404           network_close(&(ivac->net),channel);
405           sprintf(c_str,"channel %02d: connection closed",channel);
406           memset(&(ivac->challenger[channel]),0,sizeof(t_challenger));
407         }
408         else
409           sprintf(c_str,"channel %02d: no active connection",channel);
410       }
411       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
412     }
413     if(!(strncmp(arg[0],"select",6))) {
414       valid=1;
415       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
416       else channel=-1;
417       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
418         network_select(&(ivac->net),channel);
419         sprintf(c_str,"selected channel %d",channel);
420       }
421       else if(arg[1][0]='*') {
422         network_select(&(ivac->net),MAX_CONNECTIONS);
423         strcpy(c_str,"selected all channels");
424       }
425       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
426     }
427     if(!(strncmp(arg[0],"deselect",8))) {
428       valid=1;
429       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
430       else channel=-1;
431       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
432         network_deselect(&(ivac->net),channel);
433         sprintf(c_str,"deselected channel %d",channel);
434       }
435       else if(arg[1][0]='*') {
436         network_deselect(&(ivac->net),MAX_CONNECTIONS);
437         strcpy(c_str,"deselected all channels");
438       }
439       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
440     }
441
442     if(!valid)
443       snprintf(c_str,IVAC_CONSOLE_STRING_LEN,"unknown command: '%s'",arg[0]);
444
445     /* call network functions */
446     network_manage_connection(&(ivac->net));
447
448     /* add console string to console buffer */
449     ivac_add_to_monitor(ivac,c_str);
450
451     /* refresh whole display content */
452     ivac_display_content(ivac);
453
454     /* delete content buffer + reset counter */
455     memset(input->content,0,input->c_count-1);
456     input->c_count=0;
457
458   }
459
460   return SUCCESS;
461 }
462
463 int ivac_display_head(t_display *display) {
464
465   puts("#########################################################");
466   puts("##### ivac - -  Copyright (C) 2004 Frank Zirkelbach #####");
467   puts("#########################################################");
468
469   return SUCCESS;
470 }
471
472 int ivac_display_box(t_display *display) {
473  
474   return SUCCESS;
475 }
476
477 int ivac_display_box_content(t_ivac *ivac) {
478
479   int channel;
480
481   /* prepare challenger names */
482   for(channel=0;channel<MAX_CONNECTIONS;channel++)
483     if(ivac->challenger[channel].name[0]==0)
484       strcpy(ivac->challenger[channel].name,"<empty>");
485
486   return SUCCESS;
487 }
488
489 int ivac_display_console(t_display *display) {
490
491   return SUCCESS;
492 }
493
494 int ivac_display_console_content(t_ivac *ivac) {
495
496   int i;
497
498   for(i=0;i<IVAC_CONSOLE_LEN;i++)
499     printf("[ivac] console line %d: %s\n",i,ivac->console[i]);
500
501   return SUCCESS;
502 }
503
504 int ivac_display_prompt(t_display *display) {
505
506   return SUCCESS;
507 }
508
509 int ivac_display_prompt_content(t_ivac *ivac) {
510
511   printf("%c",ivac->input.content[ivac->input.c_count-1]);
512   fflush(NULL);
513
514   return SUCCESS;
515 }
516
517 int ivac_display(t_display *display) {
518
519   int x,y;
520
521   /* display head */
522   ivac_display_head(display);
523
524   /* display box */
525   ivac_display_box(display);
526
527   /* display console */
528   ivac_display_console(display);
529
530   /* display command prompt */
531   ivac_display_prompt(display);
532
533   return SUCCESS;
534 }
535
536 int ivac_display_content(t_ivac *ivac) {
537
538   /* display box content */
539   ivac_display_box_content(ivac);
540
541   /* display console content */
542   ivac_display_console_content(ivac);
543
544   /* display prompt content */
545   ivac_display_prompt_content(ivac);
546
547   return SUCCESS;
548 }
549
550 int ivac_add_to_monitor(t_ivac *ivac,char *msg) {
551
552   int i;
553
554   for(i=0;i<IVAC_CONSOLE_LEN-1;i++)
555     memcpy(ivac->console[i],ivac->console[i+1],IVAC_CONSOLE_STRING_LEN);
556   memcpy(ivac->console[IVAC_CONSOLE_LEN-1],msg,IVAC_CONSOLE_STRING_LEN);
557
558   return SUCCESS;
559 }