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

diff --git a/src/server/handler_coke.c b/src/server/handler_coke.c
index cd02fa7..b9a9cf3 100644
--- a/src/server/handler_coke.c
+++ b/src/server/handler_coke.c
@@ -6,14 +6,27 @@
  *
  * This file is licenced under the 3-clause BSD Licence. See the file
  * COPYING for full details.
+ *
+ * NOTES:
+ * - 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 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 ===
 
@@ -21,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 = {
@@ -31,172 +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+(\\d)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED);
+
+	// Open modbus
+	if( !gbCoke_DummyMode )
+	{
+		Coke_int_ConnectToPLC();
+	}
+
 	return 0;
 }
 
-int Coke_CanDispense(int User, int Item)
+int Coke_CanDispense(int UNUSED(User), int Item)
 {
-	char	tmp[32], *status;
-	regmatch_t	matches[4];
-	 int	ret;
-
-	// Sanity please
-	if( Item < 0 || Item > 6 )	return -1;	// -EYOURBAD
-	
-	write(giCoke_SerialFD, "d7\r\n", 4);
-	write(giCoke_SerialFD, "d7\r\n", 4);
-	write(giCoke_SerialFD, "d7\r\n", 4);
+	 int	slot;
 	
-	if( WaitForColon() ) {
-		fprintf(stderr, "Coke machine timed out\n");
-		return -2;	// -EMYBAD
-	}
-	
-	// Ask the coke machine
-	sprintf(tmp, "s%i\r\n", Item);
-	write(giCoke_SerialFD, tmp, 4);
-	
-	if( WaitForColon() ) {
-		fprintf(stderr, "Coke machine timed out (after initial)\n");
-		return -2;	// -EMYBAD
-	}
-
-	ret = ReadLine(sizeof(tmp)-1, tmp);
-	printf("ret = %i, tmp = '%s'\n", ret, tmp);
-	
-	if( ret <= 0 ) {
-		fprintf(stderr, "Coke machine is not being chatty (read = %i)\n", ret);
-		if( ret == -1 ) {
-			perror("Coke Machine");
-		}
-		return -1;
-	}
-	ret = RunRegex(&gCoke_StatusRegex, tmp, sizeof(matches)/sizeof(matches[0]), matches, "Bad Response");
-	if( ret ) {
-		return -1;
-	}
-
-	tmp[ matches[3].rm_eo ] = '\0';
-	status = &tmp[ matches[3].rm_so ];
-
-	printf("Machine responded slot status '%s'\n", status);
-
-	if( strcmp(status, "full") == 0 )
+	// Check for 'dummy' mode
+	if( gbCoke_DummyMode )
 		return 0;
 
-	return 1;
+	// 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 User, int Item)
+int Coke_DoDispense(int UNUSED(User), int Item)
 {
-	char	tmp[32], *status;
-	regmatch_t	matches[4];
+	 int	slot;
+	// Check for 'dummy' mode
+	if( gbCoke_DummyMode )
+		return 0;
 
-	// Sanity please
-	if( Item < 0 || Item > 6 )	return -1;
+	// Get slot
+	slot = Coke_int_GetSlotFromItem(Item, 1);
+	if(slot < 0)	return -1;
+	
+	// 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);
+		Debug_Debug("Waiting for %i seconds (rate limit)", delay);
+		sleep( delay );
+		Debug_Debug("wait done");
+	}
+	gtCoke_LastDispenseTime = time(NULL);
 
-	WaitForColon();
+	return Coke_int_DropSlot(slot);
+}
 
-	// Dispense
-	sprintf(tmp, "d%i\r\n", Item);
-	write(giCoke_SerialFD, tmp, 4);
-	
-	WaitForColon();
+// --- INTERNAL FUNCTIONS ---
+int Coke_int_ConnectToPLC(void)
+{
+	// Ratelimit
+	if( time(NULL) - gtCoke_LastReconnectTime < COKE_RECONNECT_RATELIMIT )
+		return -1;
 
-	// Get status
-	ReadLine(sizeof(tmp)-1, tmp);
+	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);
 	
-	tmp[ matches[3].rm_eo ] = '\0';
-	status = &tmp[ matches[3].rm_so ];
-
-	printf("Machine responded slot status '%s'\n", status);
+	if( modbus_connect(gCoke_Modbus) )
+	{
+		gtCoke_LastReconnectTime = time(NULL);
+		perror("coke - modbus_connect");
+		modbus_free(gCoke_Modbus);
+		gCoke_Modbus = NULL;
+		return 1;
+	}
 
 	return 0;
 }
 
-char ReadChar()
+int Coke_int_GetSlotFromItem(int Item, int bDispensing)
 {
-	fd_set	readfs;
-	char	ch = 0;
-	 int	ret;
-	struct timeval	timeout;
-	
-	timeout.tv_sec = 5;	// 5 second timeout
-	timeout.tv_usec = 0;
-	
-	FD_ZERO(&readfs);
-	FD_SET(giCoke_SerialFD, &readfs);
+	if( Item < 0 || Item > 6 )	return -1;
+
+	// Non-coke slots
+	if( Item < 6 )
+		return Item;
 	
-	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;
+	// 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
+		}
 	}
-	
-	ret = read(giCoke_SerialFD, &ch, 1);
-	if( ret != 1 ) {
-		printf("ret = %i\n", ret);
-		return 0;
+
+	// Coke is empty!
+	// - Return 6, even if it's empty, the checks elsewhere will avoid problems
+	return 6;
+}
+
+int Coke_int_IsSlotEmpty(int Slot)
+{
+	uint8_t status;
+
+	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 ch;
+
+	return status == 0;
 }
 
-int WaitForColon()
+int Coke_int_DropSlot(int Slot)
 {
-	fd_set	readfs;
-	char	ch = 0;
-	
-	FD_SET(giCoke_SerialFD, &readfs);
-	
-	while( (ch = ReadChar()) != ':' && ch != 0);
-	
-	if( ch == 0 )	return -1;	// Timeout
+	uint8_t res;
+
+	if(Slot < 0 || Slot > 9)	return -1;
+
+	// Check if a dispense is in progress
+	if( _ReadBit(ciCoke_DropBitBase + Slot, &res) )
+	{
+		perror("Coke_int_DropSlot - modbus_read_bits#1");
+		return -2;
+	}
+	if( res != 0 )
+	{
+		// Manual dispense in progress
+		return -1;
+	}
+
+	// Dispense
+	if( _WriteBit(ciCoke_DropBitBase + Slot, 1) )
+	{
+		perror("Coke_int_DropSlot - modbus_write_bit");
+		return -2;
+	}
+
+	// Check that it started
+	usleep(1000);	// 1ms
+	if( _ReadBit(ciCoke_DropBitBase + Slot, &res) )
+	{
+		perror("Coke_int_DropSlot - modbus_read_bits#2");
+		return -2;
+	}
+	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 0;
 }
 
-int ReadLine(int len, char *output)
+int _ReadBit(int BitNum, uint8_t *Value)
 {
-	char	ch;
-	 int	i = 0;
-	
-	for(;;)
-	{
-		ch = ReadChar();
-		
-		
-		if( i < len )
-			output[i++] = ch;
-		
-		if( ch == '\0' ) {
-			return -1;
-		}
-		if( ch == '\n' || ch == '\r' ) {
-			if( i < len )
-				output[--i] = '\0';
-			return i;
-		}
-	}
+	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 _WriteBit(int BitNum, uint8_t Value)
+{
+	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;
+}