Fixes fixes fixes, init.d script
[tpg/opendispense2.git] / src / server / main.c
index 80183ba..eedafc3 100644 (file)
 #include <string.h>
 #include <signal.h>
 #include "common.h"
+#include <termios.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <stdarg.h>
+#include <syslog.h>
+#include <pthread.h>
+#include "../cokebank.h"
 
 // === IMPORTS ===
-extern void    Init_Cokebank(const char *Argument);    // cokebank.c
 extern void    Init_Handlers(void);
 extern void    Load_Itemlist(void);
 extern void    Server_Start(void);
 extern int     giServer_Port;
-extern char*   gsItemListFile;
-extern char*   gsCoke_SerialPort;
-extern char*   gsSnack_SerialPort;
+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()
@@ -33,9 +49,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++ )
@@ -46,46 +78,195 @@ 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;
                        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 {
                                // Usage error?
+                               PrintUsage(argv[0]);
+                               return -1;
                        }
                }
                else {
                        // Usage Error?
+                       PrintUsage(argv[0]);
+                       return -1;
                }
        }
        
        signal(SIGINT, sigint_handler);
        
-       Init_Cokebank(gsCokebankPath);
+       openlog("odispense2", 0, LOG_LOCAL4);
+       
+       if( Bank_Initialise(gsCokebankPath) )
+               return -1;
 
        Init_Handlers();
 
        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;
+       
+       ret = regexec(regex, string, nMatches, matches, 0);
+       if( ret == REG_NOMATCH ) {
+               return -1;
+       }
+       if( ret ) {
+               size_t  len = regerror(ret, regex, NULL, 0);
+               char    errorStr[len];
+               regerror(ret, regex, errorStr, len);
+               printf("string = '%s'\n", string);
+               fprintf(stderr, "%s\n%s", errorMessage, errorStr);
+               exit(-1);
+       }
+       
+       return ret;
+}
+
+void CompileRegex(regex_t *regex, const char *pattern, int flags)
+{
+        int    ret = regcomp(regex, pattern, flags);
+       if( ret ) {
+               size_t  len = regerror(ret, regex, NULL, 0);
+               char    errorStr[len];
+               regerror(ret, regex, errorStr, len);
+               fprintf(stderr, "Regex compilation failed - %s\n", errorStr);
+               exit(-1);
+       }
+}
+
+// Serial helper
+int InitSerial(const char *File, int BaudRate)
+{
+       struct termios  info;
+        int    baud;
+        int    fd;
+       
+       fd = open(File, O_RDWR | O_NOCTTY | O_NONBLOCK);
+       if( fd == -1 )  return -1;
+       
+       switch(BaudRate)
+       {
+       case 9600:      baud = B9600;   break;
+       default:        close(fd);      return -1;
+       }
+       
+       info.c_lflag = 0;       // Non-Canoical, No Echo
+       info.c_cflag = baud | CS8 | CLOCAL | CREAD;     // baud, 8N1
+       cfsetspeed(&info, baud);
+       info.c_cc[VTIME] = 0;   // No time limit
+       info.c_cc[VMIN] = 1;    // Block until 1 char
+       
+       tcflush(fd, TCIFLUSH);
+       tcsetattr(fd, TCSANOW, &info);
+       
+       return fd;
+}
+
+
+/**
+ * \brief Create a formatted heap string
+ */
+char *mkstr(const char *Format, ...)
+{
+       va_list args;
+        int    len;
+       char    *ret;
+
+       va_start(args, Format);
+       len = vsnprintf(NULL, 0, Format, args);
+       va_end(args);
+
+       ret = malloc( len + 1 );
+       if(!ret)        return NULL;
+
+       va_start(args, Format);
+       vsprintf(ret, Format, args);
+       va_end(args);
+       
+       return ret;
+}
+

UCC git Repository :: git.ucc.asn.au