X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=src%2Fserver%2Fmain.c;h=d6fefbc61438f1f0a98f09517d162e3c5c9d9ccc;hb=1b1b4f396d34f08e2e2932e7a69229dcc51a1da2;hp=0692d6a62eecc2cbbcf24d9e962da5eb7aa17849;hpb=10fc79a2f4f72165cb1ce1ba78c84faa8d56f5db;p=tpg%2Fopendispense2.git diff --git a/src/server/main.c b/src/server/main.c index 0692d6a..d6fefbc 100644 --- a/src/server/main.c +++ b/src/server/main.c @@ -18,20 +18,31 @@ #include #include #include +#include #include "../cokebank.h" // === IMPORTS === extern void Init_Handlers(void); extern void Load_Itemlist(void); extern void Server_Start(void); +extern int gbServer_RunInBackground; extern int giServer_Port; extern char *gsItemListFile; extern char *gsCoke_SerialPort; extern char *gsSnack_SerialPort; +extern char *gsDoor_Password; + +// === PROTOTYPES === +void *Periodic_Thread(void *Unused); // === GLOBALS === int giDebugLevel = 0; char *gsCokebankPath = "cokebank.db"; +// - Functions called every 20s (or so) +#define ciMaxPeriodics 10 +struct sPeriodicCall { + void (*Function)(void); +} gaPeriodicCalls[ciMaxPeriodics]; // === CODE === void sigint_handler() @@ -39,9 +50,25 @@ void sigint_handler() exit(0); } +void PrintUsage(const char *progname) +{ + fprintf(stderr, "Usage: %s\n", progname); + fprintf(stderr, " -p Set server port (default 11020)\n"); + fprintf(stderr, " -d Set debug level (0 - 2, default 0)\n"); + fprintf(stderr, " --itemsfile\n"); + fprintf(stderr, " Set debug level (0 - 2, default 0)\n"); + fprintf(stderr, " --cokeport\n"); + fprintf(stderr, " Coke machine serial port (Default \"/dev/ttyS0\")\n"); + fprintf(stderr, " --doorpass\n"); + fprintf(stderr, " Door LAT password file (Default empty password)\n"); + fprintf(stderr, " --cokebank\n"); + fprintf(stderr, " Coke bank database file (Default \"cokebank.db\")\n"); +} + int main(int argc, char *argv[]) { int i; + pthread_t timer_thread; // Parse Arguments for( i = 1; i < argc; i++ ) @@ -52,36 +79,74 @@ int main(int argc, char *argv[]) switch(arg[1]) { case 'p': + if( i + 1 >= argc ) return -1; giServer_Port = atoi(argv[++i]); break; case 'd': + if( i + 1 >= argc ) return -1; giDebugLevel = atoi(argv[++i]); break; + case 'D': + gbServer_RunInBackground = 1; + return -1; default: // Usage Error? - break; + PrintUsage(argv[0]); + return -1; } } else if( arg[0] == '-' && arg[1] == '-' ) { if( strcmp(arg, "--itemsfile") == 0 ) { + if( i + 1 >= argc ) return -1; gsItemListFile = argv[++i]; } else if( strcmp(arg, "--cokeport") == 0 ) { + if( i + 1 >= argc ) return -1; gsCoke_SerialPort = argv[++i]; } else if( strcmp(arg, "--snackport") == 0 ) { + if( i + 1 >= argc ) return -1; gsSnack_SerialPort = argv[++i]; } + else if( strcmp(arg, "--doorpass") == 0 ) { + FILE *fp; + char buf[30]; + if( i + 1 >= argc ) return -1; + fp = fopen(argv[++i], "r"); + if( !fp ) { + fprintf(stderr, "ERROR: Unable to read password file\n"); + perror("reading LAT password"); + return -1; + } + fgets(buf, sizeof buf, fp); + fclose(fp); + gsDoor_Password = strdup(buf); + } + else if( strcmp(arg, "--cokebank") == 0 ) { + if( i + 1 >= argc ) return -1; + gsCokebankPath = argv[++i]; + } + else if( strcmp(arg, "--daemonise") == 0 ) { + gbServer_RunInBackground = 1; + } + else if( strcmp(arg, "--dont-daemonise") == 0 ) { + gbServer_RunInBackground = 1; + } else { // Usage error? + PrintUsage(argv[0]); + return -1; } } else { // Usage Error? + PrintUsage(argv[0]); + return -1; } } signal(SIGINT, sigint_handler); + signal(SIGTERM, sigint_handler); openlog("odispense2", 0, LOG_LOCAL4); @@ -92,12 +157,46 @@ int main(int argc, char *argv[]) Load_Itemlist(); + pthread_create( &timer_thread, NULL, Periodic_Thread, NULL ); + Server_Start(); + pthread_kill(timer_thread, SIGKILL); return 0; } +void *Periodic_Thread(void *Unused) +{ + int i; + Unused = NULL; // quiet, gcc + + for( ;; ) + { + sleep(10); // Sleep for a while +// printf("Periodic firing\n"); + for( i = 0; i < ciMaxPeriodics; i ++ ) + { + if( gaPeriodicCalls[i].Function ) + gaPeriodicCalls[i].Function(); + } + } + return NULL; +} + +void AddPeriodicFunction(void (*Fcn)(void)) +{ + int i; + for( i = 0; i < ciMaxPeriodics; i ++ ) + { + if( gaPeriodicCalls[i].Function ) continue; + gaPeriodicCalls[i].Function = Fcn; + return ; + } + + fprintf(stderr, "Error: No space for %p in periodic list\n", Fcn); +} + int RunRegex(regex_t *regex, const char *string, int nMatches, regmatch_t *matches, const char *errorMessage) { int ret;