X-Git-Url: https://git.ucc.asn.au/?p=tpg%2Fopendispense2.git;a=blobdiff_plain;f=src%2Fserver%2Fhandler_coke.c;h=b9a9cf3271bbbd6e0a30e8441d7eee1ef165855b;hp=978e3f5fe538d00c3dba7bb5ed973decdf82eded;hb=6d657891a5410b7d93cc90f376a3ef27b72b20f6;hpb=142d8dc0b172eac2e1a735df4bd0fe3461e41252 diff --git a/src/server/handler_coke.c b/src/server/handler_coke.c index 978e3f5..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 #include +#include +#include #include -#include -#include -#include +#include +#include + +#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,6 +34,12 @@ int Coke_InitHandler(); int Coke_CanDispense(int User, int Item); int Coke_DoDispense(int User, int Item); + 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 = { @@ -29,74 +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 = open(gsCoke_SerialPort, O_RDWR); - 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; + } + + // Open modbus + if( !gbCoke_DummyMode ) + { + Coke_int_ConnectToPLC(); } - regcomp(&gCoke_StatusRegex, "^slot\\s+(\\d)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED); + 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]; - - // Sanity please - if( Item < 0 || Item > 6 ) return -1; + int slot; - // Ask the coke machine - sprintf(tmp, "s%i\n", Item); - write(giCoke_SerialFD, tmp, 2); + // Check for 'dummy' mode + if( gbCoke_DummyMode ) + return 0; - // Read the response - read(giCoke_SerialFD, tmp, sizeof(tmp)-1); - regexec(&gCoke_StatusRegex, tmp, sizeof(matches)/sizeof(matches[0]), matches, 0); + // Get slot + slot = Coke_int_GetSlotFromItem(Item, 0); + if(slot < 0) return -1; - tmp[ matches[3].rm_eo ] = '\0'; - status = &tmp[ matches[3].rm_so ]; + return Coke_int_IsSlotEmpty(slot); +} - printf("Machine responded slot status '%s'\n", status); +/** + * \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; - if( strcmp(status, "full") == 0 ) + // 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); + + return Coke_int_DropSlot(slot); +} + +// --- 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( modbus_connect(gCoke_Modbus) ) + { + gtCoke_LastReconnectTime = time(NULL); + perror("coke - modbus_connect"); + modbus_free(gCoke_Modbus); + gCoke_Modbus = NULL; return 1; + } return 0; } -/** - * \brief Actually do a dispense from the coke machine - */ -int Coke_DoDispense(int User, int Item) +int Coke_int_GetSlotFromItem(int Item, int bDispensing) { - char tmp[32], *status; - regmatch_t matches[4]; - - // Sanity please if( Item < 0 || Item > 6 ) return -1; - // Dispense - sprintf(tmp, "d%i\n", Item); - write(giCoke_SerialFD, tmp, 2); - - // Get status - read(giCoke_SerialFD, tmp, sizeof(tmp)-1); - regexec(&gCoke_StatusRegex, tmp, sizeof(matches)/sizeof(matches[0]), matches, 0); + // Non-coke slots + if( Item < 6 ) + return Item; - tmp[ matches[3].rm_eo ] = '\0'; - status = &tmp[ matches[3].rm_so ]; + // 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 + } + } + + // 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 status == 0; +} + +int Coke_int_DropSlot(int Slot) +{ + uint8_t res; + + if(Slot < 0 || Slot > 9) return -1; - printf("Machine responded slot status '%s'\n", status); + // 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 _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; +}