X-Git-Url: https://git.ucc.asn.au/?p=tpg%2Fopendispense2.git;a=blobdiff_plain;f=src%2Fserver%2Fhandler_coke.c;h=b9a9cf3271bbbd6e0a30e8441d7eee1ef165855b;hp=62ea17f0564f4e673ea90c5a14223a38729d0351;hb=6d657891a5410b7d93cc90f376a3ef27b72b20f6;hpb=ffc52312097ac25aaca6d20a132242f5f0670c5b

diff --git a/src/server/handler_coke.c b/src/server/handler_coke.c
index 62ea17f..b9a9cf3 100644
--- a/src/server/handler_coke.c
+++ b/src/server/handler_coke.c
@@ -11,15 +11,22 @@
  * - Remember, the coke machine echoes your text back to you!
  */
 #include "common.h"
+#include "../common/config.h"
 #include <stdio.h>
 #include <string.h>
+#include <stdarg.h>
+#include <pthread.h>
 #include <unistd.h>
-#include <sys/stat.h>
-#include <fcntl.h>
-#include <regex.h>
+#include <modbus/modbus.h>
+#include <errno.h>
 
-#define READ_TIMEOUT	2	// 2 seconds for ReadChar
-#define TRACE_COKE	1
+#define MIN_DISPENSE_PERIOD	2
+#define COKE_RECONNECT_RATELIMIT	2
+
+// === CONSTANTS ===
+const int	ciCoke_MinPeriod = 5;
+const int	ciCoke_DropBitBase = 1024;
+const int	ciCoke_StatusBitBase = 16;
 
 // === IMPORTS ===
 
@@ -27,8 +34,12 @@
  int	Coke_InitHandler();
  int	Coke_CanDispense(int User, int Item);
  int	Coke_DoDispense(int User, int Item);
- 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);
+static int	_ReadBit(int BitNum, uint8_t *Value);
+static int	_WriteBit(int BitNum, uint8_t Value);
 
 // === GLOBALS ===
 tHandler	gCoke_Handler = {
@@ -37,260 +48,223 @@ tHandler	gCoke_Handler = {
 	Coke_CanDispense,
 	Coke_DoDispense
 };
-char	*gsCoke_SerialPort = "/dev/ttyS0";
- int	giCoke_SerialFD;
-regex_t	gCoke_StatusRegex;
+const char	*gsCoke_ModbusAddress = "130.95.13.73";
+modbus_t	*gCoke_Modbus;
+time_t	gtCoke_LastDispenseTime;
+time_t	gtCoke_LastReconnectTime;
+ 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;
 	}
-	
-	CompileRegex(&gCoke_StatusRegex, "^slot\\s+([0-9]+)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED);
+
+	// Open modbus
+	if( !gbCoke_DummyMode )
+	{
+		Coke_int_ConnectToPLC();
+	}
+
 	return 0;
 }
 
 int Coke_CanDispense(int UNUSED(User), int Item)
 {
-	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;
+	 int	slot;
 	
-	#if TRACE_COKE
-	printf("Coke_CanDispense: Flushing");
-	#endif
+	// Check for 'dummy' mode
+	if( gbCoke_DummyMode )
+		return 0;
+
+	// Get slot
+	slot = Coke_int_GetSlotFromItem(Item, 0);
+	if(slot < 0)	return -1;
+
+	return Coke_int_IsSlotEmpty(slot);
+}
+
+/**
+ * \brief Actually do a dispense from the coke machine
+ */
+int Coke_DoDispense(int UNUSED(User), int Item)
+{
+	 int	slot;
+	// Check for 'dummy' mode
+	if( gbCoke_DummyMode )
+		return 0;
+
+	// Get slot
+	slot = Coke_int_GetSlotFromItem(Item, 1);
+	if(slot < 0)	return -1;
 	
-	// Flush the input buffer
+	// Make sure there are not two dispenses within n seconds
+	if( time(NULL) - gtCoke_LastDispenseTime < ciCoke_MinPeriod )
 	{
-		char	tmpbuf[512];
-		read(giCoke_SerialFD, tmpbuf, sizeof(tmpbuf));
-	}
-	
-	// 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 );
-
-	if( ret == 3 ) {
-		fprintf(stderr, "Coke machine timed out\n");
-		return -2;	// -EMYBAD
+		 int	delay = ciCoke_MinPeriod - (time(NULL) - gtCoke_LastDispenseTime);
+		Debug_Debug("Waiting for %i seconds (rate limit)", delay);
+		sleep( delay );
+		Debug_Debug("wait done");
 	}
+	gtCoke_LastDispenseTime = time(NULL);
 
-	// TODO: Handle "not ok" response to D7
-	
-	#if TRACE_COKE
-	printf("Coke_CanDispense: sending 's%i'", Item);
-	#endif
-	
-	// 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);
-	}
+	return Coke_int_DropSlot(slot);
+}
 
-	// Catch an error	
-	if( ret <= 0 ) {
-		fprintf(stderr, "Coke machine is not being chatty (read = %i)\n", ret);
-		if( ret == -1 ) {
-			perror("Coke Machine");
-		}
+// --- INTERNAL FUNCTIONS ---
+int Coke_int_ConnectToPLC(void)
+{
+	// Ratelimit
+	if( time(NULL) - gtCoke_LastReconnectTime < COKE_RECONNECT_RATELIMIT )
 		return -1;
+
+	if( !gCoke_Modbus )
+	{
+		gCoke_Modbus = modbus_new_tcp(gsCoke_ModbusAddress, 502);
+		if( !gCoke_Modbus )
+		{
+			perror("coke - modbus_new_tcp");
+			gtCoke_LastReconnectTime = time(NULL);
+			return 1;
+		}
+	}
+	else {
+		// Preven resource leaks
+		modbus_close(gCoke_Modbus);
 	}
+	Debug_Notice("Connecting to coke PLC machine on '%s'\n", gsCoke_ModbusAddress);
 	
-	#if TRACE_COKE
-	printf("Coke_CanDispense: wait for the prompt again");
-	#endif
+	if( modbus_connect(gCoke_Modbus) )
+	{
+		gtCoke_LastReconnectTime = time(NULL);
+		perror("coke - modbus_connect");
+		modbus_free(gCoke_Modbus);
+		gCoke_Modbus = NULL;
+		return 1;
+	}
 
-	// Eat rest of response
-	WaitForColon();
+	return 0;
+}
 
-	// Parse status response
-	ret = RunRegex(&gCoke_StatusRegex, tmp, sizeof(matches)/sizeof(matches[0]), matches, "Bad Response");
-	if( ret ) {
-		return -1;
+int Coke_int_GetSlotFromItem(int Item, int bDispensing)
+{
+	if( Item < 0 || Item > 6 )	return -1;
+
+	// Non-coke slots
+	if( Item < 6 )
+		return Item;
+	
+	// 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 ++ )
+	{
+		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
+		}
 	}
 
-	// Get slot status
-	tmp[ matches[3].rm_eo ] = '\0';
-	status = &tmp[ matches[3].rm_so ];
+	// Coke is empty!
+	// - Return 6, even if it's empty, the checks elsewhere will avoid problems
+	return 6;
+}
 
-	printf("Machine responded slot status '%s'\n", status);
-	
-	#if TRACE_COKE
-	printf("Coke_CanDispense: done");
-	#endif
+int Coke_int_IsSlotEmpty(int Slot)
+{
+	uint8_t status;
 
-	if( strcmp(status, "full") == 0 )
-		return 0;
+	if( Slot < 0 || Slot > 9 )	return -1;
+
+	errno = 0;
+	if( _ReadBit(ciCoke_StatusBitBase + Slot, &status) )
+	{
+		perror("Coke_int_IsSlotEmpty - modbus_read_bits");
+		return -2;
+	}
 
-	return 1;
+	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	i, ret;
+	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 )
-		return -2;
-	
-	#if TRACE_COKE
-	printf("Coke_DoDispense: flushing input");
-	#endif
-	// Flush the input buffer
+	// Check if a dispense is in progress
+	if( _ReadBit(ciCoke_DropBitBase + Slot, &res) )
 	{
-		char	tmpbuf[512];
-		read(giCoke_SerialFD, tmpbuf, sizeof(tmpbuf));
+		perror("Coke_int_DropSlot - modbus_read_bits#1");
+		return -2;
 	}
-	
-	// 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
-	// 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] == ':' ) {
-		ret = ReadLine(sizeof(tmp)-1, tmp);
-		if( ret == -1 )	return -1;
+	if( res != 0 )
+	{
+		// Manual dispense in progress
+		return -1;
 	}
-	// Get status
-	ret = ReadLine(sizeof(tmp)-1, tmp);
-	if( ret == -1 )	return -1;
 
-	WaitForColon();	// Eat up rest of response
-	
-	#if TRACE_COKE
-	printf("Coke_DoDispense: done");
-	#endif
-
-	// TODO: Regex
-	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;
+	// Dispense
+	if( _WriteBit(ciCoke_DropBitBase + Slot, 1) )
+	{
+		perror("Coke_int_DropSlot - modbus_write_bit");
+		return -2;
 	}
 
-	printf("Machine returned unknown value '%s'\n", tmp);	
-
-	return -1;
-}
-
-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;
+	// Check that it started
+	usleep(1000);	// 1ms
+	if( _ReadBit(ciCoke_DropBitBase + Slot, &res) )
+	{
+		perror("Coke_int_DropSlot - modbus_read_bits#2");
+		return -2;
 	}
-	
-	ret = read(giCoke_SerialFD, &ch, 1);
-	if( ret != 1 ) {
-		printf("ret = %i\n", ret);
-		return 0;
+	if( res == 0 )
+	{
+		// Oops!, no drink
+		Log_Error("Drink dispense failed, bit lowered too quickly");
+		Debug_Notice("Drink dispense failed, bit lowered too quickly");
+		return 1;
 	}
 	
-	return ch;
+	return 0;
 }
 
-int WaitForColon()
+int _ReadBit(int BitNum, uint8_t *Value)
 {
-	fd_set	readfs;
-	char	ch = 0;
-	
-	FD_SET(giCoke_SerialFD, &readfs);
-	
-	while( (ch = ReadChar()) != ':' && ch != 0);
-	
-	if( ch == 0 )	return -1;	// Timeout
-	
-	return 0;
+	errno = 0;
+	if( !gCoke_Modbus && Coke_int_ConnectToPLC() )
+		return -1;
+	if( modbus_read_bits( gCoke_Modbus, BitNum, 1, Value) >= 0 )
+		return 0;
+	if( Coke_int_ConnectToPLC() )
+		return -1;
+	if( modbus_read_bits( gCoke_Modbus, BitNum, 1, Value) >= 0 )
+		return 0;
+	return -1;
 }
 
-int ReadLine(int len, char *output)
+int _WriteBit(int BitNum, uint8_t Value)
 {
-	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;
+	errno = 0;
+	if( !gCoke_Modbus && Coke_int_ConnectToPLC() )
+		return -1;
+	if( modbus_write_bit( gCoke_Modbus, BitNum, Value != 0 ) >= 0 )
+		return 0;
+	// Error case
+	if( Coke_int_ConnectToPLC() )
+		return -1;
+	if( modbus_write_bit( gCoke_Modbus, BitNum, Value != 0 ) >= 0 )
+		return 0;
+	return -1;
 }
 
-