#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 0
+
+#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);
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\r\n");
+ if( !WaitForColon() )
+ {
+ Writef("n1 Slot1\r\n");
+ WaitForColon();
+ Writef("n2 Slot2\r\n");
+ WaitForColon();
+ Writef("n3 Slot3\r\n");
+ WaitForColon();
+ Writef("n4 Slot4\r\n");
+ WaitForColon();
+ Writef("n5 Slot5\r\n");
+ WaitForColon();
+ Writef("n6 Coke\r\n");
+ WaitForColon();
+
+ Coke_int_UpdateSlotStatuses();
+ }
+ else
+ fprintf(stderr, "Coke machine timed out.\n");
+ }
+
+ 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;
-
- // Wait for a prompt
- ret = 0;
- do {
- write(giCoke_SerialFD, "d7\r\n", 4);
- } while( WaitForColon() && ret++ < 3 );
-
- 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;
}
- // TODO: Handle "not ok" response to D7
+ // Get slot status
+ Buffer[ matches[3].rm_eo ] = '\0';
+ status = &Buffer[ matches[3].rm_so ];
- // Ask the coke machine
- sprintf(tmp, "s%i\r\n", Item);
- write(giCoke_SerialFD, tmp, 4);
+ TRACE("Machine responded slot %i status '%s'\n", Slot, status);
- // 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);
+ if( strcmp(status, "full") == 0 ) {
+ gaCoke_CachedStatus[Slot] = 0; // 0: Avaliiable
+ return 0;
+ }
+ else {
+ gaCoke_CachedStatus[Slot] = 1; // 1: Empty
+ return 1;
}
+}
- // Catch an error
- if( ret <= 0 ) {
- fprintf(stderr, "Coke machine is not being chatty (read = %i)\n", ret);
- if( ret == -1 ) {
- perror("Coke Machine");
+/**
+ * \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( giCoke_SerialFD == -1 ) return ;
+
+ 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');
+
+ for( i = 0; i <= 6; i ++ )
+ {
+ // 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 :(
+ }
+ TRACE("tmp = '%s'\n", tmp);
+ Coke_int_GetSlotStatus(tmp, i);
}
-
- // 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 ];
-
- printf("Machine responded slot status '%s'\n", status);
+ 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];
}
/**
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;
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);
+
+ TRACE("flushing input\n");
+
+
// Wait for prompt
- i = 0;
- do {
- write(Item, "d7\r\n", 4);
- } while( WaitForColon() && i++ < 3 );
+ 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;
+ }
+ 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
+
+ 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()
int ret;
struct timeval timeout;
- timeout.tv_sec = 5; // 5 second timeout
+ timeout.tv_sec = READ_TIMEOUT;
timeout.tv_usec = 0;
FD_ZERO(&readfs);
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;
}
}
- //printf("ReadLine: output=%s\n", output);
-
if( !ch ) return -1;
return i;
}