Fixes fixes fixes, init.d script
[tpg/opendispense2.git] / src / server / handler_coke.c
index 62ea17f..12e2431 100644 (file)
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <regex.h>
+#include <stdarg.h>
+#include <pthread.h>
 
 #define READ_TIMEOUT   2       // 2 seconds for ReadChar
 #define TRACE_COKE     1
 
+#if TRACE_COKE
+# define TRACE(v...) do{printf("%s: ",__func__);printf(v);}while(0)
+#else
+# define TRACE(...)
+#endif
+
 // === IMPORTS ===
 
 // === PROTOTYPES ===
  int   Coke_InitHandler();
+ int   Coke_int_GetSlotStatus(char *Buffer, int Slot);
+void   Coke_int_UpdateSlotStatuses(void);
  int   Coke_CanDispense(int User, int Item);
  int   Coke_DoDispense(int User, int Item);
+ int   Writef(const char *Format, ...);
  int   WaitForColon();
  int   ReadLine(int len, char *output);
 
@@ -40,115 +51,138 @@ tHandler  gCoke_Handler = {
 char   *gsCoke_SerialPort = "/dev/ttyS0";
  int   giCoke_SerialFD;
 regex_t        gCoke_StatusRegex;
+ int   gaCoke_CachedStatus[7];
+pthread_mutex_t        gCoke_Mutex = PTHREAD_MUTEX_INITIALIZER;
 
 // == CODE ===
 int Coke_InitHandler()
 {
+       CompileRegex(&gCoke_StatusRegex, "^slot\\s+([0-9]+)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED);
+       
        printf("connecting to coke machine...\n");
        
        giCoke_SerialFD = InitSerial(gsCoke_SerialPort, 9600);
        if( giCoke_SerialFD == -1 ) {
                fprintf(stderr, "ERROR: Unable to open coke serial port ('%s')\n", gsCoke_SerialPort);
        }
+       else {
+               int i;
+               for( i = 0; i < 7; i ++ )
+                       gaCoke_CachedStatus[i] = -1;
+               // Reset the slot names.
+               // - Dunno why this is needed, but the machine plays silly
+               //   sometimes.
+               Writef("n0 Slot0\n");
+               if( !WaitForColon() )
+               {
+                       Writef("n1 Slot1\n");
+                       WaitForColon();
+                       Writef("n2 Slot2\n");
+                       WaitForColon();
+                       Writef("n3 Slot3\n");
+                       WaitForColon();
+                       Writef("n4 Slot4\n");
+                       WaitForColon();
+                       Writef("n5 Slot5\n");
+                       WaitForColon();
+                       Writef("n6 Coke\n");
+                       
+                       Coke_int_UpdateSlotStatuses();
+               }
+       }
+       
+       AddPeriodicFunction(Coke_int_UpdateSlotStatuses);
        
-       CompileRegex(&gCoke_StatusRegex, "^slot\\s+([0-9]+)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED);
        return 0;
 }
 
-int Coke_CanDispense(int UNUSED(User), int Item)
+int Coke_int_GetSlotStatus(char *Buffer, int Slot)
 {
-       char    tmp[40], *status;
        regmatch_t      matches[4];
         int    ret;
-
-       // Sanity please
-       if( Item < 0 || Item > 6 )      return -1;      // -EYOURBAD
+       char    *status;        
        
-       // Can't dispense if the machine is not connected
-       if( giCoke_SerialFD == -1 )
-               return -2;
-       
-       #if TRACE_COKE
-       printf("Coke_CanDispense: Flushing");
-       #endif
-       
-       // Flush the input buffer
-       {
-               char    tmpbuf[512];
-               read(giCoke_SerialFD, tmpbuf, sizeof(tmpbuf));
+       // Parse status response
+       ret = RunRegex(&gCoke_StatusRegex, Buffer, sizeof(matches)/sizeof(matches[0]), matches, "Bad Response");
+       if( ret ) {
+               return -1;
        }
+
+       // Get slot status
+       Buffer[ matches[3].rm_eo ] = '\0';
+       status = &Buffer[ matches[3].rm_so ];
        
-       // Wait for a prompt
-       ret = 0;
-       do {
-               #if TRACE_COKE
-               printf("Coke_DoDispense: sending 'd7'");
-               #endif
-               write(giCoke_SerialFD, "d7\r\n", 4);
-       } while( WaitForColon() && ret++ < 3 );
+       TRACE("Machine responded slot %i status '%s'\n", Slot, status);
 
-       if( ret == 3 ) {
-               fprintf(stderr, "Coke machine timed out\n");
-               return -2;      // -EMYBAD
+       if( strcmp(status, "full") == 0 ) {
+               gaCoke_CachedStatus[Slot] = 0;  // 0: Avaliiable
+               return 0;
        }
+       else {
+               gaCoke_CachedStatus[Slot] = 1;  // 1: Empty
+               return 1;
+       }
+}
 
-       // TODO: Handle "not ok" response to D7
+/**
+ * \brief Update the status of all coke slots
+ * \note Uses goto to reduce the chance of the lock being kept
+ */
+void Coke_int_UpdateSlotStatuses(void)
+{
+        int    i, len;
+       char    tmp[40];
        
-       #if TRACE_COKE
-       printf("Coke_CanDispense: sending 's%i'", Item);
-       #endif
+       if( giCoke_SerialFD == -1 )     return ;
        
-       // Ask the coke machine
-       sprintf(tmp, "s%i\r\n", Item);
-       write(giCoke_SerialFD, tmp, 4);
-
-       #if TRACE_COKE
-       printf("Coke_CanDispense: reading response");
-       #endif
-       // Read from the machine (ignoring empty lines)
-       while( (ret = ReadLine(sizeof(tmp)-1, tmp)) == 0 );
-       printf("ret = %i, tmp = '%s'\n", ret, tmp);
-       if( tmp[0] == ':' ) {
-               ret = ReadLine(sizeof(tmp)-1, tmp);
-               printf("ret = %i, tmp = '%s'\n", ret, tmp);
-       }
-
-       // Catch an error       
-       if( ret <= 0 ) {
-               fprintf(stderr, "Coke machine is not being chatty (read = %i)\n", ret);
-               if( ret == -1 ) {
-                       perror("Coke Machine");
+       pthread_mutex_lock(&gCoke_Mutex);
+       
+       while( ReadLine(sizeof tmp, tmp) >= 0 ) ;
+       TRACE("send d7\n");
+       Writef("d7\r\n");       // Update slot statuses
+       if( WaitForColon() )    goto ret;
+       TRACE("send s\n");
+       Writef("s\r\n");
+       do {
+               i = ReadLine(sizeof tmp, tmp);  // Read back what we just said
+               if( i == -1 ) {
+                       TRACE("Eat read failed");
+                       goto ret;
                }
-               return -1;
-       }
+       } while(tmp[0] != ':' && tmp[1] != 's');
        
-       #if TRACE_COKE
-       printf("Coke_CanDispense: wait for the prompt again");
-       #endif
-
-       // Eat rest of response
-       WaitForColon();
-
-       // Parse status response
-       ret = RunRegex(&gCoke_StatusRegex, tmp, sizeof(matches)/sizeof(matches[0]), matches, "Bad Response");
-       if( ret ) {
-               return -1;
+       for( i = 0; i <= 6; i ++ )
+       {
+               len = ReadLine(sizeof tmp, tmp);
+               if( len == -1 ) {
+                       TRACE("Read failed on slot %i\n", i);
+                       goto ret;       // I give up :(
+               }
+               TRACE("tmp = '%s'\n", tmp);
+               Coke_int_GetSlotStatus(tmp, i);
+       }
+       // Eat blank line
+       len = ReadLine(sizeof tmp, tmp);
+       if( len == -1 ) {
+               TRACE("Read failed on blank line\n");
        }
 
-       // Get slot status
-       tmp[ matches[3].rm_eo ] = '\0';
-       status = &tmp[ matches[3].rm_so ];
+       TRACE("Updated\n");
 
-       printf("Machine responded slot status '%s'\n", status);
-       
-       #if TRACE_COKE
-       printf("Coke_CanDispense: done");
-       #endif
-
-       if( strcmp(status, "full") == 0 )
-               return 0;
+ret:
+       pthread_mutex_unlock(&gCoke_Mutex);
+}
 
-       return 1;
+int Coke_CanDispense(int UNUSED(User), int Item)
+{
+       // Sanity please
+       if( Item < 0 || Item > 6 )      return -1;      // -EYOURBAD
+       
+       // Can't dispense if the machine is not connected
+       if( giCoke_SerialFD == -1 )
+               return -2;
+       
+       return gaCoke_CachedStatus[Item];
 }
 
 /**
@@ -157,7 +191,7 @@ int Coke_CanDispense(int UNUSED(User), int Item)
 int Coke_DoDispense(int UNUSED(User), int Item)
 {
        char    tmp[32];
-        int    i, ret;
+        int    ret, len;
 
        // Sanity please
        if( Item < 0 || Item > 6 )      return -1;
@@ -166,60 +200,79 @@ int Coke_DoDispense(int UNUSED(User), int Item)
        if( giCoke_SerialFD == -1 )
                return -2;
        
-       #if TRACE_COKE
-       printf("Coke_DoDispense: flushing input");
-       #endif
-       // Flush the input buffer
+       // LOCK
+       pthread_mutex_lock(&gCoke_Mutex);
+       
+       TRACE("flushing input\n");
+       
+
+       // Wait for prompt
+       ret = 0;
+       while( WaitForColon() && ret < 3 )
        {
+               // Flush the input buffer
                char    tmpbuf[512];
                read(giCoke_SerialFD, tmpbuf, sizeof(tmpbuf));
+               TRACE("sending 'd7'\n");
+               Writef("d7\r\n");
+               ret ++;
+       }
+       if( ret == 3 )
+       {
+               TRACE("timed out\n");
+               pthread_mutex_unlock(&gCoke_Mutex);
+               return -1;
        }
-       
-       // Wait for prompt
-       i = 0;
-       do {
-               #if TRACE_COKE
-               printf("Coke_DoDispense: sending 'd7'");
-               #endif
-               write(Item, "d7\r\n", 4);
-       } while( WaitForColon() && i++ < 3 );
 
-       #if TRACE_COKE
-       printf("Coke_DoDispense: sending 'd%i'", Item);
-       #endif
+       TRACE("sending 'd%i'\n", Item);
        // Dispense
-       sprintf(tmp, "d%i\r\n", Item);
-       write(giCoke_SerialFD, tmp, 4);
-       
-       // Read empty lines
-       while( (ret = ReadLine(sizeof(tmp)-1, tmp)) == -1 );
-       if( ret == -1 ) return -1;
-       // Read d%i
-       while( tmp[0] == ':' ) {
+       Writef("d%i\r\n", Item);
+       
+       // Read empty lines and echo-backs
+       do {
                ret = ReadLine(sizeof(tmp)-1, tmp);
-               if( ret == -1 ) return -1;
-       }
-       // Get status
-       ret = ReadLine(sizeof(tmp)-1, tmp);
-       if( ret == -1 ) return -1;
+               if( ret == -1 ) {
+                       pthread_mutex_unlock(&gCoke_Mutex);
+                       return -1;
+               }
+               TRACE("read %i '%s'\n", ret, tmp);
+       } while( ret == 0 || tmp[0] == ':' || tmp[0] == 'd' );
 
        WaitForColon(); // Eat up rest of response
        
-       #if TRACE_COKE
-       printf("Coke_DoDispense: done");
-       #endif
+       TRACE("done\n");
 
-       // TODO: Regex
+       // TODO: Regex instead?
        if( strcmp(tmp, "ok") == 0 ) {
                // We think dispense worked
                // - The machine returns 'ok' if an empty slot is dispensed, even if
                //   it doesn't actually try to dispense (no sound)
-               return 0;
+               ret = 0;
        }
-
-       printf("Machine returned unknown value '%s'\n", tmp);   
-
-       return -1;
+       else {
+               printf("Coke_DoDispense: Machine returned unknown value '%s'\n", tmp);
+               ret = -1;
+       }
+       
+       TRACE("Updating slot status\n");
+       
+       // Update status
+       WaitForColon();
+       Writef("s%i\r\n", Item);
+       len = ReadLine(sizeof tmp, tmp);
+       if(len == -1)   gaCoke_CachedStatus[Item] = -1;
+       Coke_int_GetSlotStatus(tmp, Item);
+       {
+               char buf[512];
+               read(giCoke_SerialFD, buf, 512);        // Flush
+       }
+       
+       // Release and return
+       pthread_mutex_unlock(&gCoke_Mutex);
+       
+       //return ret;
+       // HACK!!!
+       return 0;
 }
 
 char ReadChar()
@@ -238,19 +291,43 @@ char ReadChar()
        ret = select(giCoke_SerialFD+1, &readfs, NULL, NULL, &timeout);
        if( ret == 0 )  return 0;       // Timeout
        if( ret != 1 ) {
-               printf("readchar return %i\n", ret);
+               printf("ReadChar: select return %i\n", ret);
                return 0;
        }
        
        ret = read(giCoke_SerialFD, &ch, 1);
        if( ret != 1 ) {
-               printf("ret = %i\n", ret);
+               printf("ReadChar: ret != 1 (%i)\n", ret);
                return 0;
        }
        
        return ch;
 }
 
+int Writef(const char *Format, ...)
+{
+       va_list args;
+        int    len;
+       
+       va_start(args, Format);
+       len = vsnprintf(NULL, 0, Format, args);
+       va_end(args);
+       
+       {
+               char    buf[len+1];
+               va_start(args, Format);
+               vsnprintf(buf, len+1, Format, args);
+               va_end(args);
+               
+               #if DEBUG
+               printf("Writef: %s", buf);
+               #endif
+               
+               return write(giCoke_SerialFD, buf, len);
+       }
+       
+}
+
 int WaitForColon()
 {
        fd_set  readfs;
@@ -287,8 +364,6 @@ int ReadLine(int len, char *output)
                }
        }
 
-       //printf("ReadLine: output=%s\n", output);
-
        if( !ch )       return -1;
        return i;
 }

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