display api improvements - to be continued
[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   event_init(&(ivac.event));
102
103   /* input init */
104   ivac.input.mode=CONTENT_BUFFER;
105   input_init(&(ivac.input));
106
107   /* network init */
108   if(network_init(&(ivac.net))==N_ERROR) {
109     printf("[ivac] use 'fuser -n tcp %d' to determine the process to kill!\n",
110            ivac.net.l_port);
111     ivac_shutdown(&ivac);
112     return ERROR;
113   }
114   network_udp_listen_init(&(ivac.net));
115
116   /* add listening port(s) + stdin to (read) event system */
117   event_math(ivac.net.l_fd,&(ivac.event),READ,ADD);
118   event_math(ivac.net.l_udp_fd,&(ivac.event),READ,ADD);
119   event_math(0,&(ivac.event),READ,ADD);
120
121   /* display init */
122   display_init(&(ivac.display));
123
124   /* use hardcoded audio settings by now */
125   ivac.audio.fmt=BIT_8;
126   ivac.audio.channels=MONO;
127   ivac.audio.speed=8000;
128   /* audio init */
129   audio_init(&(ivac.audio));
130   audio_setup(&(ivac.audio));
131
132   /* display */
133 #ifdef XXX_GUI
134   /* "constant" graphics */
135   ivac_display_head(&ivac);
136   ivac_display_box(&ivac);
137   ivac_display_console(&ivac);
138 #endif
139   ivac_display(&ivac);
140
141   /* start event system - callbacks used: ivac_event_cb + ivac_regular_cb */
142   event_start(&(ivac.event),(void *)&ivac,ivac_event_cb,ivac_regular_cb);
143
144   return SUCCESS;
145 }
146
147 int ivac_shutdown(t_ivac *ivac) {
148
149   network_shutdown(&(ivac->net));
150   network_udp_shutdown(&(ivac->net));
151   input_shutdown(&(ivac->input));
152   event_stop(&(ivac->event));
153   audio_shutdown(&(ivac->audio));
154   display_shutdown(&(ivac->display));
155
156   return SUCCESS;
157 }
158
159 int ivac_send_info(int channel,t_ivac *ivac) {
160
161   char data[SEND_N_MAX];
162   int size;
163
164   size=strlen(ivac->username);
165
166   data[0]=IVAC_SEND_NAME;
167   data[1]=size;
168   strncpy(data+2,ivac->username,size);
169   size+=2;
170
171   data[size]=IVAC_SEND_G_CAP;
172   data[size+1]=1;
173   data[size+2]=ivac->g_cap;
174   size+=3;
175
176   data[size]=IVAC_SEND_AV_CAP;
177   data[size+1]=2;
178   data[size+2]=(ivac->av_cap)>>8;
179   data[size+3]=(ivac->av_cap)&0xff;
180   size+=4;
181
182   if(network_send(ivac->net.connection[channel].fd,data,size)==N_ERROR) {
183     puts("[ivac] ivac_send_info failed");
184     return ERROR;
185   }
186
187   return SUCCESS;
188 }
189
190 int ivac_send_quit(int channel,t_ivac *ivac) {
191
192   char data[7]; /* one more for \0 */
193
194   data[0]=IVAC_SEND_QUIT;
195   data[1]=4;
196   strcpy(data+2,"quit");
197
198   if(network_send(ivac->net.connection[channel].fd,data,6)==N_ERROR) {
199     puts("[ivac] ivac_send_quit failed");
200     return ERROR;
201   }
202
203   return SUCCESS;
204 }
205
206 int ivac_receive_info(int channel,t_ivac *ivac) {
207
208   char data[SEND_N_MAX];
209   char c_str[IVAC_CONSOLE_STRING_LEN];
210   int count,length;
211
212   count=0;
213
214   if((length=network_receive(ivac->net.connection[channel].fd,
215                              data,SEND_N_MAX))==N_ERROR) {
216     puts("[ivac] ivac_receive_info failed");
217     return ERROR;
218   }
219
220   while(length-count) {
221     switch(data[count]) {
222       case IVAC_SEND_NAME:
223         strncpy(ivac->challenger[channel].name,data+count+2,data[count+1]);
224         ivac->challenger[channel].name[data[count+1]]='\0';
225         count+=(data[count+1]+2);
226         break;
227       case IVAC_SEND_G_CAP:
228         ivac->challenger[channel].g_cap=data[count+2];
229         count+=3;
230         break;
231       case IVAC_SEND_AV_CAP:
232         ivac->challenger[channel].av_cap=data[count+2]<<8;
233         ivac->challenger[channel].av_cap|=data[count+3];
234         count+=4;
235         break;
236       case IVAC_SEND_QUIT:
237         if(!(strncmp(data+count+2,"quit",data[1])))
238           sprintf(c_str,"channel %02d: connection closed by remote host",
239                   channel);
240           event_math(ivac->net.connection[channel].fd,&(ivac->event),
241                      READ,REMOVE);
242           network_close(&(ivac->net),channel);
243           memset(&(ivac->challenger[channel]),0,sizeof(t_challenger));
244           ivac_add_to_monitor(ivac,c_str);
245           count+=6;
246           break;
247       default:
248         sprintf(c_str,"ivac_receive_info, unknown character: 0x%02x\n",
249                data[count]);
250         ivac_add_to_monitor(ivac,c_str);
251         return ERROR;
252         break;
253     }
254   }
255
256   return length;
257 }
258
259 int ivac_event_cb(t_event *event,void *ptr) {
260  
261   t_ivac *ivac;
262   int channel;
263   char c_str[IVAC_CONSOLE_STRING_LEN];
264
265   ivac=(t_ivac *)ptr;
266
267   /* incoming connection -- first contact => send info */
268   if(FD_ISSET(ivac->net.l_fd,&(event->rfds))) {
269     channel=network_manage_incoming(&(ivac->net));
270     if(channel==N_E_ACCEPT)
271       sprintf(c_str,"accept failed");
272     else if(channel==N_E_MAXC)
273       sprintf(c_str,"maximum connections reached");
274     else {
275       sprintf(c_str,"connection from %s port %d on channel %d",
276               ivac->net.connection[channel].ip,
277               ivac->net.connection[channel].port,channel);
278       ivac_add_to_monitor(ivac,c_str);
279       event_math(ivac->net.connection[channel].fd,event,READ,ADD);
280       ivac_send_info(channel,ivac);
281     }
282     ivac_display(ivac);
283   }
284
285   /* wait for user info */
286   for(channel=0;channel<MAX_CONNECTIONS;channel++) {
287     if(ivac->net.connection[channel].status&C_ESTABL) {
288       /* remote is sending info */
289       if(FD_ISSET(ivac->net.connection[channel].fd,&(event->rfds))) {
290         if(ivac_receive_info(channel,ivac)==0) {
291           event_math(ivac->net.connection[channel].fd,event,READ,REMOVE);
292           network_close(&(ivac->net),channel);
293           sprintf(c_str,"channel %02d: broken pipe - disconnected",channel);
294           ivac_add_to_monitor(ivac,c_str);
295         }
296         ivac_display(ivac);
297       }
298     }
299   } 
300
301   /* user interaction */
302   if(FD_ISSET(0,&(event->rfds)))
303     input_get_event(&(ivac->input),ivac_parse_command,ivac);
304
305   return SUCCESS;
306 }
307
308 int ivac_regular_cb(t_event *event,void *ptr) {
309
310   /* usual jobs like audio & video transmit ... */
311
312   return SUCCESS;
313 }
314
315 int ivac_parse_command(t_input *input,void *ptr) {
316
317   t_ivac *ivac;
318   int i,j,k;
319   int len;
320   int channel;
321   char *data,valid;
322   char c_str[IVAC_CONSOLE_STRING_LEN];
323   char arg[IVAC_ARG_COUNT][IVAC_ARG_LEN];
324
325   ivac=(t_ivac *)ptr;
326   data=input->content;
327   valid=0;
328
329   /* refresh prompt content only */
330   ivac_display_prompt_content(ivac);
331
332   /* parse command routines */
333   if(data[input->c_count-1]=='\n') {
334
335     /* delete console string + args */
336     memset(c_str,0,IVAC_CONSOLE_STRING_LEN);
337     for(j=0;j<IVAC_ARG_COUNT;j++) memset(arg[j],0,IVAC_ARG_LEN);
338
339     /* get args */
340     len=0;
341     while(data[len]!='\n') len++;
342     i=0; j=0;
343     while((i<len) && (j<IVAC_ARG_COUNT)) {
344       k=0;
345       while((data[i+k]!=' ')&&(data[i+k]!='\n')) {
346         arg[j][k]=data[i+k];
347         k++;
348       }
349       arg[j][k]='\0';
350       j++;
351       /* skip all ' ' */
352       while(data[i+k]==' ') k++;
353       i+=k;
354     }
355
356     /* parse command  aka arg[0] */
357     if(!(strncmp(arg[0],"quit",4))) {
358       valid=1;
359       sprintf(c_str,"ivac shutdown ...");
360       for(i=0;i<MAX_CONNECTIONS;i++) {
361         if(ivac->net.connection[i].status&C_ESTABL) {
362           ivac_send_quit(i,ivac);
363           network_close(&(ivac->net),i);
364         }
365       }
366       ivac_shutdown(ivac);
367     }
368     if(!(strncmp(arg[0],"set",3))) {
369       valid=1;
370       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
371       else channel=-1;
372       if(!(strncmp(arg[1],"name",4))) {
373         strncpy(ivac->username,arg[2],CHAR_USERNAME);
374         sprintf(c_str,"changed username to %s",ivac->username);
375       }
376       else if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
377         i=network_set_connection_info(&(ivac->net),channel,arg[2],atoi(arg[3]));
378         if(i==N_E_IN_USE)
379           sprintf(c_str,"channel %02d: connection in use",channel);
380         if(i==N_SUCCESS)
381           sprintf(c_str,"channel %02d: set connection info",channel);
382       } else snprintf(c_str,IVAC_CONSOLE_STRING_LEN,"unknown argument: '%s'",
383                       arg[1]);
384     }
385     if(!(strncmp(arg[0],"connect",7))) {
386       valid=1;
387       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
388       else channel=-1;
389       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
390         i=network_connect(&(ivac->net),channel);
391         if(i==N_E_IN_USE)
392           sprintf(c_str,"channel %02d: connection in use",channel);
393         else if(i==N_E_NO_INFO)
394           sprintf(c_str,"channel %02d: channel not configured",channel);
395         else {
396           sprintf(c_str,"channel %02d: connected to %s:%d",channel,
397                   ivac->net.connection[channel].ip,
398                   ivac->net.connection[channel].port);
399           event_math(ivac->net.connection[channel].fd,&(ivac->event),READ,ADD);
400           ivac_send_info(channel,ivac);
401         }
402       }
403       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
404     } 
405     if(!(strncmp(arg[0],"close",5))) {
406       valid=1;
407       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
408       else channel=-1;
409       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
410         if(ivac->net.connection[channel].status&C_ESTABL) {
411           ivac_send_quit(channel,ivac);
412           event_math(ivac->net.connection[channel].fd,&(ivac->event),
413                      READ,REMOVE);
414           network_close(&(ivac->net),channel);
415           sprintf(c_str,"channel %02d: connection closed",channel);
416           memset(&(ivac->challenger[channel]),0,sizeof(t_challenger));
417         }
418         else
419           sprintf(c_str,"channel %02d: no active connection",channel);
420       }
421       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
422     }
423     if(!(strncmp(arg[0],"select",6))) {
424       valid=1;
425       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
426       else channel=-1;
427       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
428         network_select(&(ivac->net),channel);
429         sprintf(c_str,"selected channel %d",channel);
430       }
431       else if(arg[1][0]='*') {
432         network_select(&(ivac->net),MAX_CONNECTIONS);
433         strcpy(c_str,"selected all channels");
434       }
435       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
436     }
437     if(!(strncmp(arg[0],"deselect",8))) {
438       valid=1;
439       if((arg[1][0]>='0')&&(arg[1][0]<='9')) channel=atoi(arg[1]);
440       else channel=-1;
441       if((channel>=0)&&(channel<MAX_CONNECTIONS)) {
442         network_deselect(&(ivac->net),channel);
443         sprintf(c_str,"deselected channel %d",channel);
444       }
445       else if(arg[1][0]='*') {
446         network_deselect(&(ivac->net),MAX_CONNECTIONS);
447         strcpy(c_str,"deselected all channels");
448       }
449       else sprintf(c_str,"invalid argument: '%s'",arg[1]);
450     }
451
452     if(!valid)
453       snprintf(c_str,IVAC_CONSOLE_STRING_LEN,"unknown command: '%s'",arg[0]);
454
455     /* call network functions */
456     network_manage_connection(&(ivac->net));
457
458     /* add console string to console buffer */
459     ivac_add_to_monitor(ivac,c_str);
460
461     /* refresh whole display content */
462     ivac_display(ivac);
463
464     /* delete content buffer + reset counter */
465     memset(input->content,0,input->c_count-1);
466     input->c_count=0;
467
468   }
469
470   return SUCCESS;
471 }
472
473 int ivac_display_head(t_ivac *ivac) {
474
475 #ifndef XXX_GUI
476   display_term_line(&(ivac->display),'#');
477   puts("##### ivac - -  Copyright (C) 2004 Frank Zirkelbach #####");
478   puts("#########################################################");
479 #endif
480
481   return SUCCESS;
482 }
483
484 int ivac_display_box(t_ivac *ivac) {
485
486 #ifndef XXX_GUI
487   puts("---------------------------------------------------------------------");
488   printf("username: %s, capabilities: %02x|%04x\n",ivac->username,ivac->g_cap,
489                                                    ivac->av_cap);
490   puts("---------------------------------------------------------------------");
491   puts("connections:");
492 #endif
493  
494   return SUCCESS;
495 }
496
497 int ivac_display_box_content(t_ivac *ivac) {
498
499   int channel;
500
501   /* prepare challenger names */
502   for(channel=0;channel<MAX_CONNECTIONS;channel++)
503     if(ivac->challenger[channel].name[0]==0)
504       strcpy(ivac->challenger[channel].name,"<empty>");
505
506 #ifndef XXX_GUI
507   for(channel=0;channel<MAX_CONNECTIONS;channel++) {
508     if(ivac->net.connection[channel].status&C_INFO_A)
509       printf("channel %02d: %s:%d, status: %02x, name: %s\n",channel,
510              ivac->net.connection[channel].ip,
511              ivac->net.connection[channel].port,
512              ivac->net.connection[channel].status,
513              ivac->challenger[channel].name);
514   }
515 #endif
516
517   return SUCCESS;
518 }
519
520 int ivac_display_console(t_ivac *ivac) {
521
522 #ifndef XXX_GUI
523   puts("---------------------------------------------------------------------");
524   puts("console messages:");
525 #endif
526
527   return SUCCESS;
528 }
529
530 int ivac_display_console_content(t_ivac *ivac) {
531
532   int i;
533
534 #ifndef XXX_GUI
535   for(i=0;i<IVAC_CONSOLE_LEN;i++)
536     printf("| %s\n",ivac->console[i]);
537 #endif
538
539   return SUCCESS;
540 }
541
542 int ivac_display_prompt(t_ivac *ivac) {
543
544 #ifndef XXX_GUI
545   puts("---------------------------------------------------------------------");
546   printf("prompt: ");
547 #endif
548
549   return SUCCESS;
550 }
551
552 int ivac_display_prompt_content(t_ivac *ivac) {
553
554 #ifndef XXX_GUI
555   printf("%c",ivac->input.content[ivac->input.c_count-1]);
556   fflush(NULL);
557 #endif
558
559   return SUCCESS;
560 }
561
562 int ivac_display(t_ivac *ivac) {
563
564   int x,y;
565
566 #ifndef XXX_GUI
567   /* display head */
568   ivac_display_head(ivac);
569
570   /* display box */
571   ivac_display_box(ivac);
572 #endif
573   ivac_display_box_content(ivac);
574
575 #ifndef XXX_GUI
576   /* display console */
577   ivac_display_console(ivac);
578 #endif
579   ivac_display_console_content(ivac);
580
581 #ifndef XXX_GUI
582   /* display command prompt */
583   ivac_display_prompt(ivac);
584 #endif
585   fflush(NULL);
586
587   return SUCCESS;
588 }
589
590 int ivac_add_to_monitor(t_ivac *ivac,char *msg) {
591
592   int i;
593
594   for(i=0;i<IVAC_CONSOLE_LEN-1;i++)
595     memcpy(ivac->console[i],ivac->console[i+1],IVAC_CONSOLE_STRING_LEN);
596   memcpy(ivac->console[IVAC_CONSOLE_LEN-1],msg,IVAC_CONSOLE_STRING_LEN);
597
598   return SUCCESS;
599 }