path: root/vchat-client.c

/*
 * vchat-client - alpha version
 * vchat-client.c - main() and utility functions
 *
 * Copyright (C) 2001 Andreas Kotes <count@flatline.de>
 *
 * This program is free software. It can be redistributed and/or modified,
 * provided that this copyright notice is kept intact. This program is
 * distributed in the hope that it will be useful, but without any warranty;
 * without even the implied warranty of merchantability or fitness for a
 * particular purpose. In no event shall the copyright holder be liable for
 * any direct, indirect, incidental or special damages arising in any way out
 * of the use of this software. 
 *
 */

/* general includes */
#include <sys/types.h>
#include <sys/time.h>
#include <string.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <fcntl.h>
#include <signal.h>
#include <readline/readline.h>
#include <openssl/ssl.h>
#include "vchat.h"

/* version of this module */
unsigned char *vchat_cl_version = "$Id$";

/* externally used variables */
/*   we're logged in */
unsigned int loggedin = 0;
/*   we run as long as this is true */
unsigned int status = 1;
/*   error string to show after exit */
unsigned char errstr[ERRSTRSIZE] = "\0";

/* locally global variables */
/* our list of filedescriptors */
static fd_set masterfds;

/* declaration of configuration array */
#include "vchat-config.h"

/* servers filedescriptor from vchat-protocol.c */
extern int serverfd;

void setnoption (unsigned char *, unsigned char *);

static void parsecfg(unsigned char *line) {
  int bytes;
  unsigned char *param=line;
  unsigned char *value=NULL;
  
  /* handle quotes value is empty, so wecan use it */
  value = strchr(line,'#');
  if (value) { /* the line contains a cute little quote */
  	value[0]='\0'; /* ignore the rest of the line */
  }
  
  /* now split the line into two parts */
  value = strchr(line,'=');
  if (!value) return; /* exit if strchr fails */
  value[0]='\0';
  value++;

  /* "trim" values */
  while ((value[0] == ' ')||(value[0] == '\t'))
    value++;
  bytes = strlen(value);
  while ((value[bytes-1] == ' ')||(value[bytes-1] == '\t')) {
    value[bytes-1] = '\0';
    bytes=strlen(value);
  } 
  /* bytes should be strlen(value) */
  if ( value[bytes-1] == '"' ) value[bytes-1] = '\0';
  if ( value[0] == '"' ) value++;
  
  /* "trim" param */
  while ((param[0] == ' ')||(param[0] == '\t'))
    param++;
  bytes = strlen(param);
  while ((param[bytes-1] == ' ')||(param[bytes-1] == '\t')) {
    param[bytes-1] = '\0';
    bytes=strlen(param);
  }
  /* bytes should be strlen(param) */
  if ( param[bytes-1] == '\"' ) param[bytes-1] = '\0';
  if ( param[0] == '\"' ) param++;
  
  if ((!param)||(!value)) return; /* failsave */
  
  //fprintf(stderr,"\"%s\" -> \"%s\"\n",param,value);
  setnoption(param,value);
}

static void parseformats(unsigned char *line) {
  int i;
  unsigned char *tmp = NULL;  

  /* read a format line from file, syntax is
     FS_XXX = "formatstring"
  */

  while( *line == ' ') line++;

  if( strlen( line ) > TMPSTRSIZE ) return;

  if( *line != '#') /* allow to comment out the line */
      for (i = 0; formatstrings[i].formatstr; i++)
          if (!strncasecmp(formatstrings[i].idstring, line, strlen( formatstrings[i].idstring) ))
          {
              unsigned char *tail = line + strlen( formatstrings[i].idstring);
              while( *tail==' ' || *tail=='\t') tail++; /* and skip whitespaces */

              if( *tail++ == '=' )
              {
                  while( *tail==' ' || *tail=='\t') tail++;
                  if( *(tail++)=='\"' )
                  {
                      int j, k = 0, stringends = 0, backslash=0;
                      for ( j = 0; tail[j] && !stringends; j++)
                      {
                          switch( tail[j] ) {
                          case '^':
                              if ( tail[j+1] != '^' )
                                  tmpstr[k++] = 1;
                              break;
                          case '\\':
                              backslash=1-backslash;
                              tmpstr[k++] = '\\';
                              break;
                          case '\"':
                              if (backslash) k--; else stringends = 1;
                          default:
                              tmpstr[k++] = tail[j];
                              backslash = 0;
                          }
                      }

                      if ( stringends && ( (tmp = (unsigned char *)malloc( 1 + j )) != NULL ) )
                      {
                          memcpy( tmp, tmpstr, k);
                          tmp[k-1]=0;
                          formatstrings[i].formatstr = tmp;
                      }
                  }
              }
          }

}

/* UNUSED uncomment if needed 
static void parseknownhosts(unsigned char *line) {
}
*/

/* load config file */
static void
loadcfg (unsigned char *file,void (*lineparser) (unsigned char *))
{
  int fd;
  int bytes,bufoff=0;
  unsigned char *tmp = NULL;
#define BUFSIZE 4096
  unsigned char buf[BUFSIZE];     /* data buffer */
  unsigned char *ltmp = buf;
  unsigned char *tildex = NULL;
  buf[BUFSIZE-1] = '\0'; /* sanity stop */
  
  if (!file) return;
  if (!file[0]) return;
  if (file[0] == '~')
    tildex = tilde_expand(file);
   else
    tildex = file;
  fd = open(tildex,O_RDONLY);
  if (fd == -1) {
    snprintf (errstr, TMPSTRSIZE, "Can't open config-file \"%s\": %s.", tildex, sys_errlist[errno]);
  } else {
    while ((bytes = read(fd,&buf[bufoff],BUFSIZE-bufoff-1))) {
      if (bytes < 0) {
        close(fd);
        return;
      } else {
        /* terminate string */
        buf[bytes + bufoff] = '\0';
        /* as long as there are lines .. */
        while ((tmp = strchr (ltmp, '\n')) != NULL) {
          /* did the server send CR+LF instead of LF with the last line? */
          if (tmp[-1] == '\r')
            tmp[-1] = '\0';
          
          /* remove newline from previous message, advance pointer of next
           * message */
          tmp[0] = '\0';
          tmp++;

          /* we have a last message? give away to line handler! */
          if (ltmp[0])
            {
              lineparser(ltmp);
            }

          /* move line along .. */
          ltmp = tmp;
        }
        /* buffer exhausted, move partial line to start of buffer and go
         * on .. */
        bufoff = (bytes+bufoff) - (ltmp-buf);
        if (bufoff > 0)
          memmove (buf, ltmp, bufoff);
        else
          bufoff = 0;
      }
    }
    close(fd);
  }
}

void
loadconfig (unsigned char *file)
{
  loadcfg(file,parsecfg);
}

void
loadformats (unsigned char *file)
{
  loadcfg(file,parseformats);
}

/* get-format-string */
unsigned char *
getformatstr (formtstr id)
{
  int i;
  for (i = 0; formatstrings[i].formatstr; i++)
      if (formatstrings[i].id == id) return formatstrings[i].formatstr;
  return NULL;
}

/* get-string-option, fetches *char-value of variable named by option */
unsigned char *
getstroption (confopt option)
{
  int i;
#ifdef DEBUG
  fprintf(stderr,"getstroption: %d\n",option);
#endif
  for (i = 0; configoptions[i].type != CO_NIL; i++)
    if ((configoptions[i].id == option) && (configoptions[i].type == CO_STR)) {
      if (!configoptions[i].value)
         return configoptions[i].defaultvalue;
       else
         return configoptions[i].value;
    }
  return NULL;
}

/* set-string-option, puts *char-value to variable named by option */
void
setstroption (confopt option, unsigned char *string)
{
  int i;
#ifdef DEBUG
  fprintf(stderr,"setstroption: %d to %s\n",option,string);
#endif
  for (i = 0; configoptions[i].type != CO_NIL; i++)
    if ((configoptions[i].id == option) && (configoptions[i].type == CO_STR)) {
      if (configoptions[i].value)
         free(configoptions[i].value);
      if (string)
         configoptions[i].value = strdup(string);
       else
         configoptions[i].value = NULL;
      if (configoptions[i].localvar)
         *configoptions[i].localvar = configoptions[i].value;
    }
}

/* set-named-option, puts string to variable named by name */
void
setnoption (unsigned char *name, unsigned char *string)
{
  int i;
#ifdef DEBUG
  fprintf(stderr,"setstrnoption: %s to %s\n",name,string);
#endif
  for (i = 0; configoptions[i].type != CO_NIL; i++)
    if (!strcmp(configoptions[i].varname,name)) {
      if (configoptions[i].type == CO_STR) {
         if (configoptions[i].value)
           free(configoptions[i].value);
         if (string)
           configoptions[i].value = strdup(string);
         else
           configoptions[i].value = NULL;
       } else if (configoptions[i].type == CO_INT) {
         configoptions[i].value = (char *) atoi(string);
       }
      if (configoptions[i].localvar)
         *configoptions[i].localvar = configoptions[i].value;
    }
}

/* get-integer-option, fetches int-value of variable named by option */
int
getintoption (confopt option)
{
  int i;
#ifdef DEBUG
  fprintf(stderr,"getintoption: %d\n",option);
#endif
  for (i = 0; configoptions[i].type != CO_NIL; i++)
    if ((configoptions[i].id == option) && (configoptions[i].type == CO_INT)) {
      if ((int)configoptions[i].value == -1)
         return (int) configoptions[i].defaultvalue;   
       else
         return (int) configoptions[i].value;
    }
  return 0;
}

/* set-integer-option, puts int-value to variable named by option */
void
setintoption (confopt option, int value)
{
  int i;
#ifdef DEBUG
  fprintf(stderr,"setintoption: %d to %d\n",option,value);
#endif
  for (i = 0; configoptions[i].type != CO_NIL; i++)
    if ((configoptions[i].id == option) && (configoptions[i].type == CO_INT)) {
       configoptions[i].value = (char *) value;
       if (configoptions[i].localvar)
         *configoptions[i].localvar = configoptions[i].value;
    }
}

int quitrequest = 0;

/* cleanup-hook, for SIGINT */
void
cleanup (int signal)
{
  if( signal == SIGINT ) {
      switch( quitrequest >> 2 ) {
      case 0:
          flushout( );
          writeout( "  Press Ctrl+C twice now to confirm ");
          showout( );
          quitrequest+=4;
          return;
          break;
      case 1:
          flushout( );
          writeout( "  Press Ctrl+C twice now to confirm ");
          writeout( "  Press Ctrl+C once  now to confirm ");
          showout( );
          quitrequest+=4;
          return;
          break;
      default:
          break;
      }
  }
  /* restore terminal state */
  exitui ();
  /* clear userlist */
  ul_clear ();
  /* close server connection */
  if (serverfd > 0)
    close (serverfd);
  /* inform user if we where killed by signal */
  if (signal > 1)
    {
      fprintf (stderr, "vchat-client: terminated with signal %d.\n", signal);
    } else if (errstr[0])
      fputs   (errstr, stderr);
  /* end of story */
  exit (0);
}

void calleverysecond( void ) {
  /* timetriggered execution, don't rely on being called every 1000us */
  /* rather see it as a chance for being called 9 times in 10 seconds */
  /* so check time() */

  if(quitrequest)
      quitrequest--;
  if(outputcountdown && !--outputcountdown)
      hideout( );
}

/* this function is called in the master loop */
void
eventloop (void)
{
  /* get fresh copy of filedescriptor list */
  fd_set readfds = masterfds;
  struct timeval tv = { 1, 0};

  switch (select (serverfd + 1, &readfds, NULL, NULL, &tv))
  {
  case -1:
      /* EINTR is most likely a SIGWINCH - ignore for now */
      if (errno != EINTR)
	{
	  snprintf (tmpstr, TMPSTRSIZE, "Select fails, %s.", sys_errlist[errno]);
          strncpy(errstr,tmpstr,TMPSTRSIZE-2);
          errstr[TMPSTRSIZE-2] = '\0';
          strcat(errstr,"\n");
	  writecf (FS_ERR,tmpstr);
          /* see this as an error condition and bail out */
	  status = 0;
	}
        break;
  case 0:
      /* time out reached */
      calleverysecond();
      break;
  default:
      /* something to read from user & we're logged in or have a cert? */
      if (FD_ISSET (0, &readfds) && loggedin)
	userinput ();

      /* something to read from server? */
      if (FD_ISSET (serverfd, &readfds))
	networkinput ();
      break;
  }
}

void usage(unsigned char *name) {
    printf   ("usage: %s [-C config-file] [-l] [-z] [-s host] [-p port] [-c channel] [-n nickname]\n",name);
    puts     ("   -C   load a second config-file, overriding the first one\n");
    puts     ("   -l   local connect (no SSL + connects localhost:2323)\n");
    puts     ("   -z   don't use certificate files\n");
    printf   ("   -s   set server (default \"%s\")\n",getstroption(CF_SERVERHOST));
    printf   ("   -p   set port (default %d)\n",getintoption(CF_SERVERPORT));
    printf   ("   -c   set channel (default %d)\n",getintoption(CF_CHANNEL));
    if (nick)
       printf("   -n   set nickname (default \"%s\")\n",nick);
    else
       puts  ("   -n   set nickname\n");
    printf   ("   -f   set from (default \"%s\")\n",getstroption(CF_FROM));
    puts     ("   -h   gives this help\n");
}

/* main - d'oh */
int
main (int argc, char **argv)
{
  int pchar;
  int cmdsunparsed = 1;

  loadconfig (GLOBAL_CONFIG_FILE);
  loadconfig (getstroption (CF_CONFIGFILE));
  loadformats(GLOBAL_FORMAT_FILE);
  loadformats(getstroption (CF_FORMFILE));

  /* parse commandline */
  while (cmdsunparsed) {
      pchar = getopt(argc,argv,"C:lzs:p:c:n:f:h");
#ifdef DEBUG
      fprintf(stderr,"parse commandline: %d ('%c'): %s\n",pchar,pchar,optarg);
#endif

      switch (pchar) {
	  case -1 : cmdsunparsed = 0; break;
          case 'C': loadconfig(optarg); break;
          case 'l': setintoption(CF_USESSL,0); break;
          case 'z': setintoption(CF_USECERT,0); break;
          case 's': setstroption(CF_SERVERHOST,optarg); break;
          case 'p': setintoption(CF_SERVERPORT,strtol(optarg,NULL,10)); break;
          case 'c': setintoption(CF_CHANNEL,strtol(optarg,NULL,10)); break;
          case 'n': setstroption(CF_NICK,optarg); break;
          case 'f': setstroption(CF_FROM,optarg); break;
          case 'h': usage(argv[0]); exit(0); break;
          default : usage(argv[0]); exit(1);
      }
  }

  if (optind < argc) { usage(argv[0]); exit(1); }

  if (!getintoption(CF_USESSL)) {
      setstroption(CF_SERVERHOST,"localhost");
      setintoption(CF_SERVERPORT,2323);
  } else {
      SSL_library_init ();
      SSL_load_error_strings ();
    }

  
  /* install signal handler */
  signal (SIGINT, cleanup);
  signal (SIGHUP, cleanup);
  signal (SIGTERM, cleanup);
  signal (SIGQUIT, cleanup);

  /* initialize userinterface */
  initui ();

  /* attempt connection */
  if (!vcconnect (getstroption(CF_SERVERHOST), getintoption(CF_SERVERPORT)))
    {
      snprintf (tmpstr, TMPSTRSIZE, "Could not connect to server, %s.",
		sys_errlist[errno]);
      strncpy(errstr,tmpstr,TMPSTRSIZE-2);
      errstr[TMPSTRSIZE-2] = '\0';
      strcat(errstr,"\n");
      writecf (FS_ERR,tmpstr);
      /* exit condition */
      status = 0;
    }
  else
    {
      /* add stdin & server to masterdfs */
      FD_ZERO (&masterfds);
      FD_SET (0, &masterfds);
      FD_SET (serverfd, &masterfds);
    }

  while (status)
    eventloop ();

  /* sanely close connection to server */
  vcdisconnect ();

  /* call cleanup-hook without signal */
  cleanup (0);
  return 0;
}