X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=src%2Fserver%2Fhandler_coke.c;h=a899ad588aa93c020db782c628116b5cdfefa946;hb=907e45a8c930a4ff4055b4e593744a89190732c3;hp=da9fa18d2501326c252c670ac7617cf0fcfd7ae0;hpb=39d4f8129c6413d3e41916041d239ea6d3041228;p=tpg%2Fopendispense2.git

diff --git a/src/server/handler_coke.c b/src/server/handler_coke.c
index da9fa18..a899ad5 100644
--- a/src/server/handler_coke.c
+++ b/src/server/handler_coke.c
@@ -13,27 +13,29 @@
 #include "common.h"
 #include <stdio.h>
 #include <string.h>
-#include <unistd.h>
-#include <sys/stat.h>
-#include <fcntl.h>
-#include <regex.h>
 #include <stdarg.h>
 #include <pthread.h>
+#include <unistd.h>
+#include <modbus/modbus.h>
+#include <errno.h>
 
-#define READ_TIMEOUT	2	// 2 seconds for ReadChar
-#define TRACE_COKE	1
+#define MIN_DISPENSE_PERIOD	5
+
+// === CONSTANTS ===
+const int	ciCoke_MinPeriod = 5;
+const int	ciCoke_DropBitBase = 1024;
+const int	ciCoke_StatusBitBase = 16;
 
 // === 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);
+ int	Coke_int_ConnectToPLC(void);
+ int	Coke_int_GetSlotFromItem(int Item, int bDispensing);
+ int	Coke_int_IsSlotEmpty(int Slot);
+ int	Coke_int_DropSlot(int Slot);
 
 // === GLOBALS ===
 tHandler	gCoke_Handler = {
@@ -42,324 +44,188 @@ tHandler	gCoke_Handler = {
 	Coke_CanDispense,
 	Coke_DoDispense
 };
-char	*gsCoke_SerialPort = "/dev/ttyS0";
- int	giCoke_SerialFD;
-regex_t	gCoke_StatusRegex;
- int	gaCoke_CachedStatus[7];
-pthread_mutex_t	gCoke_Mutex = PTHREAD_MUTEX_INITIALIZER;
+const char	*gsCoke_ModbusAddress = "130.95.13.73";
+modbus_t	*gCoke_Modbus;
+time_t	gtCoke_LastDispenseTime;
+ int	gbCoke_DummyMode = 1;
+ int	giCoke_NextCokeSlot = 0;
 
 // == CODE ===
 int Coke_InitHandler()
 {
-	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);
+	// Configuable dummy/blank mode (all dispenses succeed)
+	// TODO: Find a better way of handling missing/invalid options
+	if( Config_GetValueCount("coke_dummy_mode") > 0 )
+	{
+		gbCoke_DummyMode = Config_GetValue_Bool("coke_dummy_mode", 0);
+		if(gbCoke_DummyMode == -1)	gbCoke_DummyMode = 0;
 	}
-	else {
-		// 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();
-		}
+
+	// Open modbus
+	if( !gbCoke_DummyMode )
+	{
+		Coke_int_ConnectToPLC();
 	}
-	
-	AddPeriodicFunction(Coke_int_UpdateSlotStatuses);
-	
-	CompileRegex(&gCoke_StatusRegex, "^slot\\s+([0-9]+)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED);
+
 	return 0;
 }
 
-int Coke_int_GetSlotStatus(char *Buffer, int Slot)
+int Coke_CanDispense(int UNUSED(User), int Item)
 {
-	regmatch_t	matches[4];
-	 int	ret;
-	char	*status;	
+	 int	slot;
 	
-	// Parse status response
-	ret = RunRegex(&gCoke_StatusRegex, Buffer, sizeof(matches)/sizeof(matches[0]), matches, "Bad Response");
-	if( ret ) {
-		return -1;
-	}
+	// Check for 'dummy' mode
+	if( gbCoke_DummyMode )
+		return 0;
 
-	// Get slot status
-	Buffer[ matches[3].rm_eo ] = '\0';
-	status = &Buffer[ matches[3].rm_so ];
-	
-	#if TRACE_COKE
-	printf("Coke_CanDispense: Machine responded slot status '%s'\n", status);
-	#endif
+	// Get slot
+	slot = Coke_int_GetSlotFromItem(Item, 0);
+	if(slot < 0)	return -1;
 
-	if( strcmp(status, "full") == 0 ) {
-		gaCoke_CachedStatus[Slot] = 0;	// 0: Avaliiable
-		return 0;
-	}
-	else {
-		gaCoke_CachedStatus[Slot] = 1;	// 1: Empty
-		return 1;
-	}
+	return Coke_int_IsSlotEmpty(slot);
 }
 
 /**
- * \brief Update the status of all coke slots
- * \note Uses goto to reduce the chance of the lock being kept
+ * \brief Actually do a dispense from the coke machine
  */
-void Coke_int_UpdateSlotStatuses(void)
+int Coke_DoDispense(int UNUSED(User), int Item)
 {
-	 int	i, len;
-	char	tmp[40];
-	
-	if( giCoke_SerialFD == -1 )	return ;
+	 int	slot;
+	// Check for 'dummy' mode
+	if( gbCoke_DummyMode )
+		return 0;
+
+	// Get slot
+	slot = Coke_int_GetSlotFromItem(Item, 1);
+	if(slot < 0)	return -1;
 	
-	pthread_mutex_lock(&gCoke_Mutex);
+	// Make sure there are not two dispenses within n seconds
+	if( time(NULL) - gtCoke_LastDispenseTime < ciCoke_MinPeriod )
+	{
+		 int	delay = ciCoke_MinPeriod - (time(NULL) - gtCoke_LastDispenseTime);
+		printf("Wait %i seconds?\n", delay);
+		sleep( delay );
+		printf("wait done\n");
+	}
+
+	return Coke_int_DropSlot(slot);
+}
+
+// --- INTERNAL FUNCTIONS ---
+int Coke_int_ConnectToPLC(void)
+{
+	if( !gCoke_Modbus )
+	{
+		gCoke_Modbus = modbus_new_tcp(gsCoke_ModbusAddress, 502);
+		if( !gCoke_Modbus )
+		{
+			perror("coke - modbus_new_tcp");
+			return 1;
+		}
+	}
+	printf("Connecting to coke PLC machine on '%s'\n", gsCoke_ModbusAddress);
 	
-	WaitForColon();
-	#if TRACE_COKE
-	printf("Coke_int_UpdateSlotStatuses: send d7\n");
-	#endif
-	Writef("d7\r\n");	// Update slot statuses
-	if( WaitForColon() )	goto ret;
-	#if TRACE_COKE
-	printf("Coke_int_UpdateSlotStatuses: send s\n");
-	#endif
-	Writef("s\r\n");
-	ReadLine(sizeof tmp, tmp);	// Read back what we just said
+	if( modbus_connect(gCoke_Modbus) )
+	{
+		perror("coke - modbus_connect");
+		modbus_free(gCoke_Modbus);
+		gCoke_Modbus = NULL;
+		return 1;
+	}
+
+	return 0;
+}
+
+int Coke_int_GetSlotFromItem(int Item, int bDispensing)
+{
+	if( Item < 0 || Item > 6 )	return -1;
+
+	// Non-coke slots
+	if( Item < 6 )
+		return Item;
 	
-	for( i = 0; i <= 6; i ++ )
+	// Iterate though coke slots and find the first one with a drink avaliable
+	// `giCoke_NextCokeSlot` ensures that the slots rotate
+	for( int i = 0; i < 4; i ++ )
 	{
-		len = ReadLine(sizeof tmp, tmp);
-		if( len == -1 ) {
-			#if TRACE_COKE
-			printf("Coke_int_UpdateSlotStatuses: Read failed on slot %i\n", i);
-			#endif
-			goto ret;	// I give up :(
+		int slot = 6 + (i + giCoke_NextCokeSlot) % 4;
+		if( !Coke_int_IsSlotEmpty(slot) )
+		{
+			if(bDispensing) {
+				giCoke_NextCokeSlot ++;
+				if(giCoke_NextCokeSlot == 4)	giCoke_NextCokeSlot = 0;
+			}
+			return slot;	// Drink avaliable
 		}
-		#if TRACE_COKE
-		printf("Coke_int_UpdateSlotStatuses: tmp = '%s'\n", tmp);
-		#endif
-		Coke_int_GetSlotStatus(tmp, i);
 	}
 
-ret:
-	pthread_mutex_unlock(&gCoke_Mutex);
+	// Coke is empty!
+	// - Return 6, even if it's empty, the checks elsewhere will avoid problems
+	return 6;
 }
 
-int Coke_CanDispense(int UNUSED(User), int Item)
+int Coke_int_IsSlotEmpty(int Slot)
 {
-	// Sanity please
-	if( Item < 0 || Item > 6 )	return -1;	// -EYOURBAD
-	
-	// Can't dispense if the machine is not connected
-	if( giCoke_SerialFD == -1 )
+	uint8_t status;
+
+	if( Slot < 0 || Slot > 9 )	return -1;
+
+	if(!gCoke_Modbus)
 		return -2;
-	
-	return gaCoke_CachedStatus[Item];
+
+	errno = 0;
+	if( modbus_read_bits(gCoke_Modbus, ciCoke_StatusBitBase + Slot, 1, &status) <= 0 )
+	{
+		// TODO: Retry connection, or fail
+		perror("Coke_int_IsSlotEmpty - modbus_read_bits");
+		return -2;
+	}
+
+	return status == 0;
 }
 
-/**
- * \brief Actually do a dispense from the coke machine
- */
-int Coke_DoDispense(int UNUSED(User), int Item)
+int Coke_int_DropSlot(int Slot)
 {
-	char	tmp[32];
-	 int	ret, len;
+	uint8_t res;
 
-	// Sanity please
-	if( Item < 0 || Item > 6 )	return -1;
+	if(Slot < 0 || Slot > 9)	return -1;
 
 	// Can't dispense if the machine is not connected
-	if( giCoke_SerialFD == -1 )
+	if( !gCoke_Modbus )
 		return -2;
-	
-	// LOCK
-	pthread_mutex_lock(&gCoke_Mutex);
-	
-	#if TRACE_COKE
-	printf("Coke_DoDispense: flushing input\n");
-	#endif
-	
-	// Wait for prompt
-	ret = 0;
-	while( WaitForColon() && ret < 3 )
+
+	// Check if a dispense is in progress
+	if( modbus_read_bits(gCoke_Modbus, ciCoke_DropBitBase + Slot, 1, &res) <= 0 )
 	{
-		// Flush the input buffer
-		char	tmpbuf[512];
-		read(giCoke_SerialFD, tmpbuf, sizeof(tmpbuf));
-		#if TRACE_COKE
-		printf("Coke_DoDispense: sending 'd7'\n");
-		#endif
-		Writef("d7\r\n");
-		ret ++;
+		perror("Coke_int_DropSlot - modbus_read_bits#1");
+		return -2;
 	}
-	if( ret == 3 )
+	if( res != 0 )
 	{
-		#if TRACE_COKE
-		printf("Coke_DoDispense: timed out\n");
-		#endif
-		pthread_mutex_unlock(&gCoke_Mutex);
+		// Manual dispense in progress
 		return -1;
 	}
 
-	#if TRACE_COKE
-	printf("Coke_DoDispense: sending 'd%i'\n", Item);
-	#endif
 	// Dispense
-	Writef("d%i\r\n", Item);
-	
-	// Read empty lines and echo-backs
-	do {
-		ret = ReadLine(sizeof(tmp)-1, tmp);
-		if( ret == -1 ) {
-			pthread_mutex_unlock(&gCoke_Mutex);
-			return -1;
-		}
-		#if TRACE_COKE
-		printf("Coke_DoDispense: read %i '%s'\n", ret, tmp);
-		#endif
-	} while( ret == 0 || tmp[0] == ':' || tmp[0] == 'd' );
-
-	WaitForColon();	// Eat up rest of response
-	
-	#if TRACE_COKE
-	printf("Coke_DoDispense: done\n");
-	#endif
-
-	// 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)
-		ret = 0;
-	}
-	else {
-		printf("Coke_DoDispense: Machine returned unknown value '%s'\n", tmp);
-		ret = -1;
+	if( modbus_write_bit(gCoke_Modbus, ciCoke_DropBitBase + Slot, 1) < 0 )
+	{
+		perror("Coke_int_DropSlot - modbus_write_bit");
+		return -2;
 	}
-	
-	#if TRACE_COKE
-	printf("Coke_DoDispense: Updating slot status\n");
-	#endif
-	// Update status
-	Writef("s%i\r\n", Item);
-	len = ReadLine(sizeof tmp, tmp);
-	if(len == -1)	gaCoke_CachedStatus[Item] = -1;
-	Coke_int_GetSlotStatus(tmp, Item);
-	
-	// Release and return
-	pthread_mutex_unlock(&gCoke_Mutex);
-	
-	return ret;
-}
 
-char ReadChar()
-{
-	fd_set	readfs;
-	char	ch = 0;
-	 int	ret;
-	struct timeval	timeout;
-	
-	timeout.tv_sec = READ_TIMEOUT;
-	timeout.tv_usec = 0;
-	
-	FD_ZERO(&readfs);
-	FD_SET(giCoke_SerialFD, &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);
-		return 0;
-	}
-	
-	ret = read(giCoke_SerialFD, &ch, 1);
-	if( ret != 1 ) {
-		printf("ret = %i\n", ret);
-		return 0;
+	// Check that it started
+	usleep(1000);	// 1ms
+	if( modbus_read_bits(gCoke_Modbus, ciCoke_DropBitBase + Slot, 1, &res) <= 0 )
+	{
+		perror("Coke_int_DropSlot - modbus_read_bits#2");
+		return -2;
 	}
-	
-	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);
-	
+	if( res == 0 )
 	{
-		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);
+		// Oops!, no drink
+		return 1;
 	}
 	
-}
-
-int WaitForColon()
-{
-	fd_set	readfs;
-	char	ch = 0;
-	
-	FD_SET(giCoke_SerialFD, &readfs);
-	
-	while( (ch = ReadChar()) != ':' && ch != 0);
-	
-	if( ch == 0 )	return -1;	// Timeout
-	
 	return 0;
 }
 
-int ReadLine(int len, char *output)
-{
-	char	ch;
-	 int	i = 0;
-	
-	for(;;)
-	{
-		ch = ReadChar();
-			
-		if( i < len )
-			output[i++] = ch;
-		
-		if( ch == '\0' ) {
-			break;
-		}
-		if( ch == '\n' || ch == '\r' ) {
-			if( i < len )
-				output[--i] = '\0';
-			break;
-		}
-	}
-
-	//printf("ReadLine: output=%s\n", output);
-
-	if( !ch ) 	return -1;
-	return i;
-}
-
-