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