X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=src%2Fserver%2Fhandler_coke.c;h=ff051c8400b283f9a4239b85f824784bd03220e4;hb=008596161453e64354973268125058b49aee60d3;hp=61b26491237066b6ccd1b6755f8918d4a1a87657;hpb=5ea4314d1ee0769f8b521df2c81fe65f3a654959;p=tpg%2Fopendispense2.git

diff --git a/src/server/handler_coke.c b/src/server/handler_coke.c
index 61b2649..ff051c8 100644
--- a/src/server/handler_coke.c
+++ b/src/server/handler_coke.c
@@ -19,7 +19,8 @@
 #include <modbus/modbus.h>
 #include <errno.h>
 
-#define MIN_DISPENSE_PERIOD	5
+#define MIN_DISPENSE_PERIOD	2
+#define COKE_RECONNECT_RATELIMIT	2
 
 // === CONSTANTS ===
 const int	ciCoke_MinPeriod = 5;
@@ -36,6 +37,8 @@ const int	ciCoke_StatusBitBase = 16;
  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 = {
@@ -47,6 +50,7 @@ tHandler	gCoke_Handler = {
 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;
 
@@ -103,10 +107,11 @@ int Coke_DoDispense(int UNUSED(User), int Item)
 	if( time(NULL) - gtCoke_LastDispenseTime < ciCoke_MinPeriod )
 	{
 		 int	delay = ciCoke_MinPeriod - (time(NULL) - gtCoke_LastDispenseTime);
-		printf("Wait %i seconds?\n", delay);
+		Debug_Debug("Waiting for %i seconds (rate limit)", delay);
 		sleep( delay );
-		printf("wait done\n");
+		Debug_Debug("wait done");
 	}
+	gtCoke_LastDispenseTime = time(NULL);
 
 	return Coke_int_DropSlot(slot);
 }
@@ -114,19 +119,29 @@ int Coke_DoDispense(int UNUSED(User), int Item)
 // --- 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;
 		}
 	}
-	printf("Connecting to coke PLC machine on '%s'\n", gsCoke_ModbusAddress);
+	else {
+		// Preven resource leaks
+		modbus_close(gCoke_Modbus);
+	}
+	Debug_Notice("Connecting to coke PLC machine on '%s'\n", gsCoke_ModbusAddress);
 	
 	if( modbus_connect(gCoke_Modbus) )
 	{
+		gtCoke_LastReconnectTime = time(NULL);
 		perror("coke - modbus_connect");
 		modbus_free(gCoke_Modbus);
 		gCoke_Modbus = NULL;
@@ -170,13 +185,9 @@ int Coke_int_IsSlotEmpty(int Slot)
 
 	if( Slot < 0 || Slot > 9 )	return -1;
 
-	if(!gCoke_Modbus)
-		return -2;
-
 	errno = 0;
-	if( modbus_read_bits(gCoke_Modbus, ciCoke_StatusBitBase + Slot, 1, &status) <= 0 )
+	if( _ReadBit(ciCoke_StatusBitBase + Slot, &status) )
 	{
-		// TODO: Retry connection, or fail
 		perror("Coke_int_IsSlotEmpty - modbus_read_bits");
 		return -2;
 	}
@@ -190,12 +201,8 @@ int Coke_int_DropSlot(int Slot)
 
 	if(Slot < 0 || Slot > 9)	return -1;
 
-	// Can't dispense if the machine is not connected
-	if( !gCoke_Modbus )
-		return -2;
-
 	// Check if a dispense is in progress
-	if( modbus_read_bits(gCoke_Modbus, ciCoke_DropBitBase + Slot, 1, &res) <= 0 )
+	if( _ReadBit(ciCoke_DropBitBase + Slot, &res) )
 	{
 		perror("Coke_int_DropSlot - modbus_read_bits#1");
 		return -2;
@@ -207,7 +214,7 @@ int Coke_int_DropSlot(int Slot)
 	}
 
 	// Dispense
-	if( modbus_write_bit(gCoke_Modbus, ciCoke_DropBitBase + Slot, 1) < 0 )
+	if( _WriteBit(ciCoke_DropBitBase + Slot, 1) )
 	{
 		perror("Coke_int_DropSlot - modbus_write_bit");
 		return -2;
@@ -215,7 +222,7 @@ int Coke_int_DropSlot(int Slot)
 
 	// Check that it started
 	usleep(1000);	// 1ms
-	if( modbus_read_bits(gCoke_Modbus, ciCoke_DropBitBase + Slot, 1, &res) <= 0 )
+	if( _ReadBit(ciCoke_DropBitBase + Slot, &res) )
 	{
 		perror("Coke_int_DropSlot - modbus_read_bits#2");
 		return -2;
@@ -223,10 +230,40 @@ int Coke_int_DropSlot(int Slot)
 	if( res == 0 )
 	{
 		// Oops!, no drink
-		printf("Drink dispense failed, bit lowered too quickly\n");
+		Log_Error("Drink dispense failed, bit lowered too quickly");
+		Debug_Notice("Drink dispense failed, bit lowered too quickly");
 		return 1;
 	}
 	
 	return 0;
 }
 
+int _ReadBit(int BitNum, uint8_t *Value)
+{
+	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;
+}
+