major bugfixes, still need clean up + improvements, not to mention speex and video...
[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("-i <iface> \t specify network interface");
38   puts("");
39
40   return SUCCESS;
41 }
42
43 int main(int argc,char **argv) {
44
45   /* TESTING BY NOW */
46
47   t_ivac ivac;
48   int i;
49
50   /* default values */
51   strcpy(ivac.username,"ivac");
52   ivac.net.l_port=IVAC_LISTEN_PORT;
53   ivac.net.l_udp_port=IVAC_UDP_PORT;
54   strcpy(ivac.audio.dsp_dev,SOUND_DEVICE);
55   strcpy(ivac.net.nic,"eth0");
56  
57   /* parse argv and change default values */
58   for(i=1;i<argc;i++) {
59     if(argv[i][0]=='-') {
60       switch(argv[i][1]) {
61         case 'h':
62           usage();
63           return SUCCESS;
64         case 'n':
65           strncpy(ivac.username,argv[++i],CHAR_USERNAME);
66           break;
67         case 'p':
68           ivac.net.l_port=atoi(argv[++i]);
69           break;
70         case 'u':
71           ivac.net.l_udp_port=atoi(argv[++i]);
72           break;
73         case 'd':
74           strncpy(ivac.audio.dsp_dev,argv[++i],MAX_CHAR_DEVICE);
75           break;
76         case 'i':
77           strncpy(ivac.net.nic,argv[++i],MAX_NIC_DEVICE);
78         default:
79           usage();
80           return ERROR;
81       }
82     }
83     else usage();
84   }
85
86   /* clear challenger struct */
87   for(i=0;i<MAX_CONNECTIONS;i++)
88     memset(&(ivac.challenger[i]),0,sizeof(t_challenger));
89   /* zero console buffer */
90   for(i=0;i<IVAC_CONSOLE_LEN;i++)
91     memset(ivac.console[i],0,IVAC_CONSOLE_STRING_LEN);
92
93   /* set capabilities (futur: set by check routines) */
94   ivac.g_cap=NETWORK;
95   ivac.av_cap=AUDIO|VIDEO|DUPLEX;
96
97   /* set event timeout */
98   ivac.event.timeout.tv_sec=IVAC_S_SEC;
99   ivac.event.timeout.tv_usec=IVAC_S_USEC;
100   /* event init */
101   if(event_init(&(ivac.event))==E_ERROR) {
102     ivac_shutdown(&ivac);
103     return ERROR;
104   }
105
106   /* input init */
107   ivac.input.mode=CONTENT_BUFFER;
108   ivac.input.c_count=0;
109   if(input_init(&(ivac.input))==I_ERROR) {
110     ivac_shutdown(&ivac);
111     return ERROR;
112   }
113
114   /* network init */
115   if(network_init(&(ivac.net))==N_ERROR) {
116     printf("[ivac] use 'fuser -n tcp %d' to determine the process to kill!\n",
117            ivac.net.l_port);
118     ivac_shutdown(&ivac);
119     return ERROR;
120   }
121
122   /* network udp init */
123   network_udp_listen_init(&(ivac.net));
124
125   /* add listening port(s) + stdin to (read) event system */
126   event_math(ivac.net.l_fd,&(ivac.event),READ,ADD);
127   event_math(ivac.net.l_udp_fd,&(ivac.event),READ,ADD);
128   event_math(0,&(ivac.event),READ,ADD);
129
130   /* use hardcoded audio settings by now */
131   ivac.audio.fmt=BIT_8;
132   ivac.audio.channels=MONO;
133   ivac.audio.speed=8000;
134   /* audio init */
135   audio_init(&(ivac.audio));
136   /* audio setup */
137   audio_setup(&(ivac.audio));
138
139   /* display init */
140   display_init(&(ivac.display));
141
142   /* display */
143   ivac_display(&ivac);
144   ivac_display_content(&ivac);
145   ivac_display_draw(&ivac);
146
147   /* start event system - callbacks used: ivac_event_cb + ivac_regular_cb */
148   event_start(&(ivac.event),(void *)&ivac,ivac_event_cb,ivac_regular_cb);
149
150   return SUCCESS;
151 }
152
153 int ivac_shutdown(t_ivac *ivac) {
154
155   display_shutdown(&(ivac->display));
156   network_shutdown(&(ivac->net));
157   network_udp_shutdown(&(ivac->net));
158   input_shutdown(&(ivac->input));
159   event_stop(&(ivac->event));
160   audio_shutdown(&(ivac->audio));
161
162   return SUCCESS;
163 }
164
165 int ivac_send_info(int channel,t_ivac *ivac) {
166
167   char data[SEND_N_MAX];
168   int size;
169
170   size=strlen(ivac->username);
171
172   data[0]=IVAC_SEND_NAME;
173   data[1]=size;
174   strncpy(data+2,ivac->username,size);
175   size+=2;
176
177   data[size]=IVAC_SEND_G_CAP;
178   data[size+1]=1;
179   data[size+2]=ivac->g_cap;
180   size+=3;
181
182   data[size]=IVAC_SEND_AV_CAP;
183   data[size+1]=2;
184   data[size+2]=(ivac->av_cap)>>8;
185   data[size+3]=(ivac->av_cap)&0xff;
186   size+=4;
187
188   if(network_send(ivac->net.connection[channel].fd,data,size)==N_ERROR) {
189     puts("[ivac] ivac_send_info failed");
190     return ERROR;
191   }
192
193   return SUCCESS;
194 }
195
196 int ivac_send_quit(int channel,t_ivac *ivac) {
197
198   char data[7]; /* one more for \0 */
199
200   data[0]=IVAC_SEND_QUIT;
201   data[1]=4;
202   strcpy(data+2,"quit");
203
204   if(network_send(ivac->net.connection[channel].fd,data,6)==N_ERROR) {
205     puts("[ivac] ivac_send_quit failed");
206     return ERROR;
207   }
208
209   return SUCCESS;
210 }
211
212 int ivac_receive_info(int channel,t_ivac *ivac) {
213
214   char data[SEND_N_MAX];
215   char c_str[IVAC_CONSOLE_STRING_LEN];
216   int count,length;
217
218   count=0;
219
220   if((length=network_receive(ivac->net.connection[channel].fd,
221                              data,SEND_N_MAX))==N_ERROR) {
222     puts("[ivac] ivac_receive_info failed");
223     return ERROR;
224   }
225
226   while(length-count) {
227     switch(data[count]) {
228       case IVAC_SEND_NAME:
229         strncpy(ivac->challenger[channel].name,data+count+2,data[count+1]);
230         ivac->challenger[channel].name[(int)data[count+1]]='\0';
231         count+=(data[count+1]+2);
232         break;
233       case IVAC_SEND_G_CAP:
234         ivac->challenger[channel].g_cap=data[count+2];
235         count+=3;
236         break;
237       case IVAC_SEND_AV_CAP:
238         ivac->challenger[channel].av_cap=data[count+2]<<8;
239         ivac->challenger[channel].av_cap|=data[count+3];
240         count+=4;
241         break;
242       case IVAC_SEND_QUIT:
243         if(!(strncmp(data+count+2,"quit",data[1])))
244           sprintf(c_str,"channel %02d: connection closed by remote host",
245                   channel);
246           event_math(ivac->net.connection[channel].fd,&(ivac->event),
247                      READ,REMOVE);
248           network_close(&(ivac->net),channel);
249           memset(&(ivac->challenger[channel]),0,sizeof(t_challenger));
250           ivac_add_to_monitor(ivac,c_str);
251           count+=6;
252           break;
253       default:
254         sprintf(c_str,"ivac_receive_info, unknown character: 0x%02x\n",
255                data[count]);
256         ivac_add_to_monitor(ivac,c_str);
257         return ERROR;
258         break;
259     }
260   }
261
262   return length;
263 }
264
265 int ivac_event_cb(t_event *event,void *ptr) {
266  
267   t_ivac *ivac;
268   int channel;
269   char c_str[IVAC_CONSOLE_STRING_LEN];
270
271   ivac=(t_ivac *)ptr;
272
273   /* incoming connection -- first contact => send info */
274   if(FD_ISSET(ivac->net.l_fd,&(event->rfds))) {
275     channel=network_manage_incoming(&(ivac->net));
276     if(channel==N_E_ACCEPT)
277       sprintf(c_str,"accept failed");
278     else if(channel==N_E_MAXC)
279       sprintf(c_str,"maximum connections reached");
280     else {
281       sprintf(c_str,"connection from %s port %d on channel %d",
282               ivac->net.connection[channel].ip,
283               ivac->net.connection[channel].port,channel);
284       ivac_add_to_monitor(ivac,c_str);
285       event_math(ivac->net.connection[channel].fd,event,READ,ADD);
286       ivac_send_info(channel,ivac);
287     }
288     ivac_display_content(ivac);
289     ivac_display_draw(ivac);
290   }
291
292   /* wait for user info */
293   for(channel=0;channel<MAX_CONNECTIONS;channel++) {
294     if(ivac->net.connection[channel].status&C_ESTABL) {
295       /* remote is sending info */
296       if(FD_ISSET(ivac->net.connection[channel].fd,&(event->rfds))) {
297         if(ivac_receive_info(channel,ivac)==0) {
298           event_math(ivac->net.connection[channel].fd,event,READ,REMOVE);
299           network_close(&(ivac->net),channel);
300           sprintf(c_str,"channel %02d: broken pipe - disconnected",channel);
301           ivac_add_to_monitor(ivac,c_str);
302         }
303         ivac_display_content(ivac);
304         ivac_display_draw(ivac);
305       }
306     }
307   } 
308
309   /* user interaction */
310   if(FD_ISSET(0,&(event->rfds)))
311     input_get_event(&(ivac->input),ivac_parse_command,ivac);
312
313   return SUCCESS;
314 }
315
316 int ivac_regular_cb(t_event *event,void *ptr) {
317
318   t_ivac *ivac;
319
320   ivac=(t_ivac *)ptr;
321
322   /* usual jobs like audio & video transmit ... */
323
324   return SUCCESS;
325 }
326
327 int ivac_parse_command(t_input *input,void *ptr) {
328
329   t_ivac *ivac;
330   int i,j,k;
331   int len;
332   int channel;
333   char *data,valid;
334   char c_str[IVAC_CONSOLE_STRING_LEN];
335   char arg[IVAC_ARG_COUNT][IVAC_ARG_LEN];
336 #ifdef DEBUG
337   char debug_string[IVAC_CONSOLE_STRING_LEN];
338 #endif
339
340   ivac=(t_ivac *)ptr;
341   data=input->content;
342   valid=0;
343
344   /* refresh prompt content only */
345 #ifdef DEBUG
346   sprintf(debug_string,"-> %02x c_count = %d",input->content[input->c_count-1],
347                        input->c_count);
348   ivac_add_to_monitor(ivac,debug_string);
349 #endif
350   ivac_display_prompt_content(ivac);
351   ivac_display_draw(ivac);
352
353   /* parse command routines */
354   if(data[input->c_count-1]=='\n'||data[input->c_count-1]=='\r') {
355
356 #ifdef DEBUG
357     sprintf(debug_string,"got %02x",data[input->c_count-1]);
358     ivac_add_to_monitor(ivac,debug_string);
359 #endif
360
361     /* delete console string + args */
362     memset(c_str,0,IVAC_CONSOLE_STRING_LEN);
363     for(j=0;j<IVAC_ARG_COUNT;j++) memset(arg[j],0,IVAC_ARG_LEN);
364
365     /* get args */
366     len=0;
367     while(data[len]!='\n'&&data[len]!='\r') len++;
368 #ifdef DEBUG
369     sprintf(debug_string,"len = %d\n",len);
370     ivac_add_to_monitor(ivac,debug_string);
371 #endif
372     i=0; j=0;
373     while((i<len) && (j<IVAC_ARG_COUNT)) {
374       k=0;
375       while(data[i+k]==' ') i++;
376       while(data[i+k]!=' '&&data[i+k]!='\n'&&data[i+k]!='\r') {
377         arg[j][k]=data[i+k];
378         k++;
379       }
380       arg[j][k]='\0';
381       j++;
382       i+=k;
383     }
384
385 #ifdef DEBUG
386     sprintf(debug_string,"got args: %s ...",arg[0]);
387     ivac_add_to_monitor(ivac,debug_string);
388 #endif
389
390     /* parse command  aka arg[0] */
391     if(!(strncmp(arg[0],"quit",4))) {
392       valid=1;
393       sprintf(c_str,"ivac shutdown ...");
394       for(i=0;i<MAX_CONNECTIONS;i++) {
395         if(ivac->net.connection[i].status&C_ESTABL) {
396           ivac_send_quit(i,ivac);
397           network_close(&(ivac->net),i);
398         }
399       }
400       ivac_shutdown(ivac);
401     }
402     if(!(strncmp(arg[0],"set",3))) {
403       valid=1;
404       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
405       else channel=-1;
406       if(!(strncmp(arg[1],"name",4))) {
407         strncpy(ivac->username,arg[2],CHAR_USERNAME);
408         sprintf(c_str,"changed username to %s",ivac->username);
409       }
410       else if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
411         i=network_set_connection_info(&(ivac->net),channel,arg[2],atoi(arg[3]));
412         if(i==N_E_IN_USE)
413           sprintf(c_str,"channel %02d: connection in use",channel);
414         if(i==N_SUCCESS)
415           sprintf(c_str,"channel %02d: set connection info",channel);
416       } else snprintf(c_str,IVAC_CONSOLE_STRING_LEN,"unknown argument: '%s'",
417                       arg[1]);
418     }
419     if(!(strncmp(arg[0],"connect",7))) {
420       valid=1;
421       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
422       else channel=-1;
423       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
424         i=network_connect(&(ivac->net),channel);
425         if(i==N_E_IN_USE)
426           sprintf(c_str,"channel %02d: connection in use",channel);
427         else if(i==N_E_NO_INFO)
428           sprintf(c_str,"channel %02d: channel not configured",channel);
429         else if(i==N_SUCCESS) {
430           sprintf(c_str,"channel %02d: connected to %s:%d",channel,
431                   ivac->net.connection[channel].ip,
432                   ivac->net.connection[channel].port);
433           event_math(ivac->net.connection[channel].fd,&(ivac->event),READ,ADD);
434           ivac_send_info(channel,ivac);
435         }
436         else if(i==N_E_CONNECT)
437           sprintf(c_str,"channel %02d: connection error - %s",
438                         channel,sys_errlist[errno]);
439       }
440       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
441     } 
442     if(!(strncmp(arg[0],"close",5))) {
443       valid=1;
444       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
445       else channel=-1;
446       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
447         if(ivac->net.connection[channel].status&C_ESTABL) {
448           ivac_send_quit(channel,ivac);
449           event_math(ivac->net.connection[channel].fd,&(ivac->event),
450                      READ,REMOVE);
451           network_close(&(ivac->net),channel);
452           sprintf(c_str,"channel %02d: connection closed",channel);
453           memset(&(ivac->challenger[channel]),0,sizeof(t_challenger));
454         }
455         else
456           sprintf(c_str,"channel %02d: no active connection",channel);
457       }
458       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
459     }
460     if(!(strncmp(arg[0],"select",6))) {
461       valid=1;
462       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
463       else channel=-1;
464       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
465         network_select(&(ivac->net),channel);
466         sprintf(c_str,"selected channel %d",channel);
467       }
468       else if(arg[1][0]=='*') {
469         network_select(&(ivac->net),MAX_CONNECTIONS);
470         strcpy(c_str,"selected all channels");
471       }
472       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
473     }
474     if(!(strncmp(arg[0],"deselect",8))) {
475       valid=1;
476       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
477       else channel=-1;
478       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
479         network_deselect(&(ivac->net),channel);
480         sprintf(c_str,"deselected channel %d",channel);
481       }
482       else if(arg[1][0]=='*') {
483         network_deselect(&(ivac->net),MAX_CONNECTIONS);
484         strcpy(c_str,"deselected all channels");
485       }
486       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
487     }
488
489     if(!valid)
490       snprintf(c_str,IVAC_CONSOLE_STRING_LEN,"unknown command: '%s'",arg[0]);
491
492     /* call network functions */
493     network_manage_connection(&(ivac->net));
494
495     /* add console string to console buffer */
496     ivac_add_to_monitor(ivac,c_str);
497
498     /* delete input content buffer + reset counter */
499     memset(input->content,0,input->c_count-1);
500     input->c_count=0;
501
502     /* refresh whole display content */
503     ivac_display_content(ivac);
504     ivac_display_draw(ivac);
505   }
506
507   return SUCCESS;
508 }
509
510 int ivac_display_head(t_ivac *ivac) {
511
512   display_line(&(ivac->display),0,0,ivac->display.max_x,0,'#');
513   display_string(&(ivac->display),(ivac->display.max_x-strlen(PROG_NAME))/2,1,
514                  PROG_NAME,strlen(PROG_NAME));
515   display_line(&(ivac->display),0,2,ivac->display.max_x,2,'#');
516
517   return SUCCESS;
518 }
519
520 int ivac_display_box(t_ivac *ivac) {
521
522   display_line(&(ivac->display),0,3,ivac->display.max_x,3,'-');
523   display_line(&(ivac->display),0,5,ivac->display.max_x,5,'-');
524   display_string(&(ivac->display),0,6,"connections:",12);
525   display_line(&(ivac->display),
526                0,ivac->display.max_y-IVAC_PROMPT_LEN-IVAC_CONSOLE_LEN-1,
527                ivac->display.max_x,
528                ivac->display.max_y-IVAC_PROMPT_LEN-IVAC_CONSOLE_LEN-1,'-');
529  
530   return SUCCESS;
531 }
532
533 int ivac_display_box_content(t_ivac *ivac) {
534
535   int channel;
536   char string[MAX_BOX_CHARS];
537
538   /* prepare challenger names */
539   for(channel=0;channel<MAX_CONNECTIONS;channel++)
540     if(ivac->challenger[channel].name[0]==0)
541       strcpy(ivac->challenger[channel].name,"<empty>");
542
543   sprintf(string,"username: %s, capabilities: %02x|%04x",
544           ivac->username,ivac->g_cap,ivac->av_cap);
545   display_string(&(ivac->display),0,4,string,strlen(string));
546
547   for(channel=0;channel<MAX_CONNECTIONS;channel++) {
548     if(ivac->net.connection[channel].status&C_INFO_A) {
549       snprintf(string,MAX_BOX_CHARS,
550                "channel %02d: %s:%d, status: %02x, name: %s\n",channel,
551                ivac->net.connection[channel].ip,
552                ivac->net.connection[channel].port,
553                ivac->net.connection[channel].status,
554                ivac->challenger[channel].name);
555       display_string(&(ivac->display),0,IVAC_HEAD_LEN+4+channel,
556                      string,strlen(string));
557     }
558   }
559
560   return SUCCESS;
561 }
562
563 int ivac_display_console(t_ivac *ivac) {
564
565   display_string(&(ivac->display),
566                  0,ivac->display.max_y-IVAC_PROMPT_LEN-IVAC_CONSOLE_LEN,
567                  "console messages:",17);
568
569   return SUCCESS;
570 }
571
572 int ivac_display_console_content(t_ivac *ivac) {
573
574   int i,size;
575   char string[IVAC_CONSOLE_STRING_LEN];
576
577   for(i=0;i<IVAC_CONSOLE_LEN;i++) {
578     size=strlen(ivac->console[i]);
579     size=(size<IVAC_CONSOLE_STRING_LEN)?size:IVAC_CONSOLE_STRING_LEN;
580     memcpy(string,ivac->console[i],size);
581     memset(string+size,' ',IVAC_CONSOLE_STRING_LEN-size);
582     display_string(&(ivac->display),
583                    0,ivac->display.max_y-IVAC_PROMPT_LEN-IVAC_CONSOLE_LEN+1+i,
584                    string,IVAC_CONSOLE_STRING_LEN);
585   }
586
587   return SUCCESS;
588 }
589
590 int ivac_display_prompt(t_ivac *ivac) {
591
592   display_string(&(ivac->display),0,ivac->display.max_y-1,"prompt: ",8);
593
594   return SUCCESS;
595 }
596
597 int ivac_display_prompt_content(t_ivac *ivac) {
598
599   char string[IVAC_CONSOLE_STRING_LEN];
600   int count;
601
602   for(count=0;count<ivac->input.c_count;count++)
603     string[count]=ivac->input.content[count];
604   for(count=ivac->input.c_count;count<IVAC_CONSOLE_STRING_LEN-8;count++)
605     string[count]=' ';
606   display_string(&(ivac->display),8,ivac->display.max_y-1,
607                  string,IVAC_CONSOLE_STRING_LEN-8);
608
609   return SUCCESS;
610 }
611
612 int ivac_display_content(t_ivac *ivac) {
613
614   /* display box content */
615   ivac_display_box_content(ivac);
616
617   /* display console content */
618   ivac_display_console_content(ivac);
619
620   /* display prompt content */
621   ivac_display_prompt_content(ivac);
622
623   return SUCCESS;
624 }
625
626 int ivac_display(t_ivac *ivac) {
627
628   /* display static head */
629   ivac_display_head(ivac);
630
631   /* display static box */
632   ivac_display_box(ivac);
633
634   /* display static console */
635   ivac_display_console(ivac);
636
637   /* display command prompt */
638   ivac_display_prompt(ivac);
639
640   return SUCCESS;
641 }
642
643 int ivac_display_draw(t_ivac *ivac) {
644   
645   display_draw(&(ivac->display));
646   display_set_cursor(&(ivac->display),ivac->input.c_count+8,
647                      ivac->display.max_y-1);
648
649   return SUCCESS;
650 }
651
652 int ivac_add_to_monitor(t_ivac *ivac,char *msg) {
653
654   int i;
655
656   for(i=0;i<IVAC_CONSOLE_LEN-1;i++)
657     memcpy(ivac->console[i],ivac->console[i+1],IVAC_CONSOLE_STRING_LEN);
658   memcpy(ivac->console[IVAC_CONSOLE_LEN-1],msg,IVAC_CONSOLE_STRING_LEN);
659
660   ivac_display_console_content(ivac);
661   ivac_display_draw(ivac);
662
663   return SUCCESS;
664 }