X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=src%2Fserver%2Fhandler_coke.c;h=79d0a4697b2bdb4218a43cf119bb7d2a6e84c26e;hb=74b6a69bdc5a9f96d5781939167ae5af0e038375;hp=a8a1fee142f16498fd295513ba3fe8dd43ba964f;hpb=ac89c817538f549be3c044202443f90b0684b879;p=tpg%2Fopendispense2.git diff --git a/src/server/handler_coke.c b/src/server/handler_coke.c index a8a1fee..79d0a46 100644 --- a/src/server/handler_coke.c +++ b/src/server/handler_coke.c @@ -17,14 +17,24 @@ #include #include #include +#include +#include #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, ...); @@ -41,143 +51,142 @@ 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; +time_t gtCoke_LastDispenseTime; // == CODE === int Coke_InitHandler() { - printf("connecting to coke machine...\n"); + CompileRegex(&gCoke_StatusRegex, "^slot\\s+([0-9]+)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED); + + printf("Connecting to coke machine on '%s'\n", gsCoke_SerialPort); 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"); - 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"); + 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(); + } + else + fprintf(stderr, "Coke machine timed out.\n"); } - CompileRegex(&gCoke_StatusRegex, "^slot\\s+([0-9]+)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED); + AddPeriodicFunction(Coke_int_UpdateSlotStatuses); + 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 - - // Can't dispense if the machine is not connected - if( giCoke_SerialFD == -1 ) - return -2; - - #if TRACE_COKE - printf("Coke_CanDispense: Flushing\n"); - #endif - + char *status; - // Wait for a prompt - ret = 0; - while( WaitForColon() && ret < 3 ) - { - // Flush the input buffer - char tmpbuf[512]; - read(giCoke_SerialFD, tmpbuf, sizeof(tmpbuf)); - #if TRACE_COKE - printf("Coke_CanDispense: sending 'd7'\n"); - #endif - Writef("d7\r\n"); - ret ++; - } - // Check for a timeout error - if( !(ret < 3) ) { - fprintf(stderr, "Coke machine timed out\n"); - return -2; // -EMYBAD + // 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 ]; - // You need to do a 'd7' before reading the status - // - Otherwise it sometimes reports a full slot as empty - // [TPG] (2011-02-19) - if( ret == 0 ) - { - Writef("d7\r\n"); - WaitForColon(); + TRACE("Machine responded slot %i status '%s'\n", Slot, status); + + 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'\n", Item); - #endif + if( giCoke_SerialFD == -1 ) return ; - // Ask the coke machine - Writef("s%i\r\n", Item); - - #if TRACE_COKE - printf("Coke_CanDispense: reading response\n"); - #endif - // Read from the machine (ignoring empty lines) - while( (ret = ReadLine(sizeof(tmp)-1, tmp)) == 0 ); - #if TRACE_COKE - printf("ret = %i, tmp = '%s'\n", ret, tmp); - #endif - // Read back-echoed lines - while( tmp[0] == ':' || tmp[1] != 'l' ) + 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; + } + } while(tmp[0] != ':' && tmp[1] != 's'); + + for( i = 0; i <= 6; i ++ ) { - 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"); + // Read until non-blank line + while( (len = ReadLine(sizeof tmp, tmp)) == 0 ) ; + if( len == -1 ) { + TRACE("Read failed on slot %i\n", i); + goto ret; // I give up :( } - return -1; + TRACE("tmp = '%s'\n", tmp); + Coke_int_GetSlotStatus(tmp, i); } - - #if TRACE_COKE - printf("Coke_CanDispense: wait for the prompt again\n"); - #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; + // 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 ]; - - #if TRACE_COKE - printf("Coke_CanDispense: Machine responded slot status '%s'\n", status); - #endif + TRACE("Updated\n"); - 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]; } /** @@ -186,7 +195,7 @@ int Coke_CanDispense(int UNUSED(User), int Item) int Coke_DoDispense(int UNUSED(User), int Item) { char tmp[32]; - int ret; + int ret, len; // Sanity please if( Item < 0 || Item > 6 ) return -1; @@ -194,11 +203,21 @@ int Coke_DoDispense(int UNUSED(User), int Item) // Can't dispense if the machine is not connected if( giCoke_SerialFD == -1 ) return -2; + + // Make sure there are not two dispenses within n seconds + if( time(NULL) - gtCoke_LastDispenseTime < 10 ) + { + printf("Wait %li seconds?\n", 10 - (time(NULL) - gtCoke_LastDispenseTime)); + sleep( 10 - (time(NULL) - gtCoke_LastDispenseTime) ); + printf("wait done\n"); + } + + // LOCK + pthread_mutex_lock(&gCoke_Mutex); - #if TRACE_COKE - printf("Coke_DoDispense: flushing input\n"); - #endif + TRACE("flushing input\n"); + // Wait for prompt ret = 0; while( WaitForColon() && ret < 3 ) @@ -206,44 +225,68 @@ int Coke_DoDispense(int UNUSED(User), int Item) // Flush the input buffer char tmpbuf[512]; read(giCoke_SerialFD, tmpbuf, sizeof(tmpbuf)); - #if TRACE_COKE - printf("Coke_DoDispense: sending 'd7'\n"); - #endif + TRACE("sending 'd7'\n"); Writef("d7\r\n"); + ret ++; + } + if( ret == 3 ) + { + TRACE("timed out\n"); + pthread_mutex_unlock(&gCoke_Mutex); + return -1; } - #if TRACE_COKE - printf("Coke_DoDispense: sending 'd%i'\n", Item); - #endif + TRACE("sending 'd%i'\n", Item); // Dispense Writef("d%i\r\n", Item); // Read empty lines and echo-backs do { ret = ReadLine(sizeof(tmp)-1, tmp); - if( ret == -1 ) return -1; - #if TRACE_COKE - printf("Coke_DoDispense: read %i '%s'\n", ret, tmp); - #endif + 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\n"); - #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; + } + 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 } - printf("Machine returned unknown value '%s'\n", tmp); - - return -1; + gtCoke_LastDispenseTime = time(NULL); + + // Release and return + pthread_mutex_unlock(&gCoke_Mutex); + + //return ret; + // HACK!!! + return 0; } char ReadChar() @@ -262,13 +305,13 @@ 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; } @@ -335,8 +378,6 @@ int ReadLine(int len, char *output) } } - //printf("ReadLine: output=%s\n", output); - if( !ch ) return -1; return i; }