src/server/main.c

   1 /*
   2  * OpenDispense 2
   3  * UCC (University [of WA] Computer Club) Electronic Accounting System
   4  *
   5  * main.c - Initialisation Code
   6  *
   7  * This file is licenced under the 3-clause BSD Licence. See the file
   8  * COPYING for full details.
   9  */
  10 #include <stdlib.h>
  11 #include <stdio.h>
  12 #include <string.h>
  13 #include <signal.h>
  14 #include "common.h"
  15 #include <termios.h>
  16 #include <unistd.h>
  17 #include <sys/stat.h>
  18 #include <fcntl.h>
  19 #include <stdarg.h>
  20 #include <syslog.h>
  21
  22 // === IMPORTS ===
  23 extern void     Init_Cokebank(const char *Argument);    // cokebank.c
  24 extern void     Init_Handlers(void);
  25 extern void     Load_Itemlist(void);
  26 extern void     Server_Start(void);
  27 extern int      giServer_Port;
  28 extern char*    gsItemListFile;
  29 extern char*    gsCoke_SerialPort;
  30 extern char*    gsSnack_SerialPort;
  31
  32 // === GLOBALS ===
  33  int    giDebugLevel = 0;
  34 char    *gsCokebankPath = "cokebank.db";
  35
  36 // === CODE ===
  37 void sigint_handler()
  38 {
  39         exit(0);
  40 }
  41
  42 int main(int argc, char *argv[])
  43 {
  44          int    i;
  45
  46         // Parse Arguments
  47         for( i = 1; i < argc; i++ )
  48         {
  49                 char    *arg = argv[i];
  50                 if( arg[0] == '-' && arg[1] != '-')
  51                 {
  52                         switch(arg[1])
  53                         {
  54                         case 'p':
  55                                 giServer_Port = atoi(argv[++i]);
  56                                 break;
  57                         case 'd':
  58                                 giDebugLevel = atoi(argv[++i]);
  59                                 break;
  60                         default:
  61                                 // Usage Error?
  62                                 break;
  63                         }
  64                 }
  65                 else if( arg[0] == '-' && arg[1] == '-' ) {
  66                         if( strcmp(arg, "--itemsfile") == 0 ) {
  67                                 gsItemListFile = argv[++i];
  68                         }
  69                         else if( strcmp(arg, "--cokeport") == 0 ) {
  70                                 gsCoke_SerialPort = argv[++i];
  71                         }
  72                         else if( strcmp(arg, "--snackport") == 0 ) {
  73                                 gsSnack_SerialPort = argv[++i];
  74                         }
  75                         else {
  76                                 // Usage error?
  77                         }
  78                 }
  79                 else {
  80                         // Usage Error?
  81                 }
  82         }
  83
  84         signal(SIGINT, sigint_handler);
  85
  86         openlog("odispense2", 0, LOG_LOCAL4);
  87
  88         Init_Cokebank(gsCokebankPath);
  89
  90         Init_Handlers();
  91
  92         Load_Itemlist();
  93
  94         Server_Start();
  95
  96
  97         return 0;
  98 }
  99
 100 int RunRegex(regex_t *regex, const char *string, int nMatches, regmatch_t *matches, const char *errorMessage)
 101 {
 102          int    ret;
 103
 104         ret = regexec(regex, string, nMatches, matches, 0);
 105         if( ret ) {
 106                 size_t  len = regerror(ret, regex, NULL, 0);
 107                 char    errorStr[len];
 108                 regerror(ret, regex, errorStr, len);
 109                 printf("string = '%s'\n", string);
 110                 fprintf(stderr, "%s\n%s", errorMessage, errorStr);
 111                 exit(-1);
 112         }
 113
 114         return ret;
 115 }
 116
 117 void CompileRegex(regex_t *regex, const char *pattern, int flags)
 118 {
 119          int    ret = regcomp(regex, pattern, flags);
 120         if( ret ) {
 121                 size_t  len = regerror(ret, regex, NULL, 0);
 122                 char    errorStr[len];
 123                 regerror(ret, regex, errorStr, len);
 124                 fprintf(stderr, "Regex compilation failed - %s\n", errorStr);
 125                 exit(-1);
 126         }
 127 }
 128
 129 // Serial helper
 130 int InitSerial(const char *File, int BaudRate)
 131 {
 132         struct termios  info;
 133          int    baud;
 134          int    fd;
 135
 136         fd = open(File, O_RDWR | O_NOCTTY | O_NONBLOCK);
 137         if( fd == -1 )  return -1;
 138
 139         switch(BaudRate)
 140         {
 141         case 9600:      baud = B9600;   break;
 142         default:        close(fd);      return -1;
 143         }
 144
 145         info.c_lflag = 0;       // Non-Canoical, No Echo
 146         info.c_cflag = baud | CS8 | CLOCAL | CREAD;     // baud, 8N1
 147         cfsetspeed(&info, baud);
 148         info.c_cc[VTIME] = 0;   // No time limit
 149         info.c_cc[VMIN] = 1;    // Block until 1 char
 150
 151         tcflush(fd, TCIFLUSH);
 152         tcsetattr(fd, TCSANOW, &info);
 153
 154         return fd;
 155 }
 156
 157
 158 /**
 159  * \brief Create a formatted heap string
 160  */
 161 char *mkstr(const char *Format, ...)
 162 {
 163         va_list args;
 164          int    len;
 165         char    *ret;
 166
 167         va_start(args, Format);
 168         len = vsnprintf(NULL, 0, Format, args);
 169         va_end(args);
 170
 171         ret = malloc( len + 1 );
 172         if(!ret)        return NULL;
 173
 174         va_start(args, Format);
 175         vsprintf(ret, Format, args);
 176         va_end(args);
 177
 178         return ret;
 179 }
 180