+++ /dev/null
-# vim:ts=4
-import re
-from CRC import do_crc
-from select import select
-import socket, logging
-from time import time, sleep
-from MIFAREClient import MIFAREClient
-
-asynchronous_responses = [ '400', '401', # door open/closed
- '610', # switches changed
- '200', '201', '202', '203', '204', '205', '206',
- '207', '208', '209', '211', # key presses
- ]
-DOOR = 1
-SWITCH = 2
-KEY = 3
-TICK = 4
-MIFARE = 5
-
-class VendingException(Exception): pass
-
-class VendingMachine:
- def __init__(self, rfh, wfh, use_mifare):
- self.events = []
- # Secret
- self.secret = 'SN4CKZ0RZZZZZZZZ'
- self.rfh = rfh
- self.wfh = wfh
- self.challenge = None
- # Initialise ourselves into a known state
- self.wfh.write('\n')
- self.await_prompt()
- self.wfh.write('echo off\n')
- self.await_prompt()
- self.wfh.write('PING\n')
- code = ''
- while code != '000':
- code = self.get_response()[0]
- self.get_switches()
- if use_mifare:
- self.mifare = MIFAREClient()
- self.mifare_timeout = 0
- else:
- self.mifare = None
-
- def await_prompt(self):
- self.wfh.flush()
- state = 1
- timeout = 0.5
- prefix = ''
- s = ''
- # mtearle - vending machine was dying wait for a response from
- # the hardware, suspect it was missing characters
- #
- # fixed by migration to pyserial - but future good place to start
- while True:
- try:
- s = self.rfh.read(1)
- except socket.error:
- raise VendingException('failed to read input from vending machine')
- if s == '': raise VendingException('nothing read!')
- if (s != '#' and s != '%') and state == 1: prefix += s
- if s == '\n' or s == '\r':
- state = 1
- prefix = ''
- if (s == '#' or s == '%') and state == 1: state = 2
- if s == ' ' and state == 2:
- if prefix == '':
- self.challenge = None
- return
- if re.search('^[0-9a-fA-F]{4}$', prefix):
- self.challenge = int(prefix, 16)
- return
-
- def get_response(self, async = False):
- self.wfh.flush()
- while True:
- s = ''
- while s == '':
- s = self.rfh.readline()
- if s == '':
- raise VendingException('Input socket has closed!')
- s = s.strip('\r\n')
- code = s[0:3]
- text = s[4:]
- if code in asynchronous_responses:
- self.handle_event(code, text)
- if async: return None
- else:
- self.await_prompt()
- return (code, text)
-
- def get_switches(self):
- self.wfh.write('S\n')
- (code, text) = self.get_response()
- if code != '600':
- return (False, code, text)
- self.interpret_switches(text)
- return (True, code, text)
-
- def interpret_switches(self, text):
- self.switches = (int(text[0:2], 16) << 8) | int(text[3:5], 16)
-
- def handle_event(self, code, text):
- if code == '400':
- self.events.append((DOOR, 1))
- elif code == '401':
- self.events.append((DOOR, 0))
- elif code == '610':
- # NOP this. Nothing handles this yet.
- #self.events.append((SWITCH, None))
- self.interpret_switches(text)
- elif code[0] == '2':
- self.events.append((KEY, int(code[1:3])))
- else:
- logging.warning('Unhandled event! (%s %s)\n'%(code,text))
-
- def authed_message(self, message):
- print 'self.challenge = %04x' % self.challenge
- if self.challenge == None:
- return message
- crc = do_crc('%c%c'%(self.challenge >> 8, self.challenge & 0xff))
- crc = do_crc(self.secret, crc)
- crc = do_crc(message, crc)
- print 'output = "%s|%04x"' % (message, crc)
- return message+'|'+('%04x'%crc)
-
- def ping(self):
- self.wfh.write('PING\n')
- (code, string) = self.get_response()
- return (code == '000', code, string)
-
- def vend(self, item):
- if not re.search('^[0-9][0-9]$', item):
- return (False, 'Invalid item requested (%s)'%item)
- self.wfh.write(self.authed_message(('V%s'%item))+'\n')
- (code, string) = self.get_response()
- return (code == '100', code, string)
-
- def beep(self, duration = None, synchronous = True):
- msg = 'B'
- if synchronous: msg += 'S'
- if duration != None:
- if duration > 255: duration = 255
- if duration < 1: duration = 1
- msg += '%02x'%duration
- self.wfh.write(msg+'\n')
- (code, string) = self.get_response()
- return (code == '500', code, string)
-
- def silence(self, duration = None, synchronous = True):
- msg = 'C'
- if synchronous: msg += 'S'
- if duration != None:
- if duration > 255: duration = 255
- if duration < 1: duration = 1
- msg += '%02x'%duration
- self.wfh.write(msg+'\n')
- (code, string) = self.get_response()
- return (code == '501', code, string)
-
- def display(self, string):
- if len(string) > 10:
- string = string[0:10]
- string = re.sub('(.)\.', lambda match: '.'+match.group(1), string)
- self.wfh.write('D'+string+'\n')
- (code, string) = self.get_response()
- return (code == '300', code, string)
-
- def next_event(self, timeout = None):
- # we don't want to buffer in the serial port, so we get all the events
- # we can ASAP.
-
- # Never have no timeout...
- if timeout == None: timeout = 60*60*24*365
-
- # Make sure we go through the loop at least once.
- if timeout < 0: timeout = 0
-
- while timeout >= 0:
- this_timeout = min(timeout, 0.2)
- timeout -= this_timeout
-
- (r, _, _) = select([self.rfh], [], [], this_timeout)
- if r:
- self.get_response(async = True)
- timeout = 0
-
- if self.mifare:
- now = time()
- if now > self.mifare_timeout:
- self.mifare_timeout = now + 0.5
- mifare_uid = self.mifare.get_card_id()
- if mifare_uid != None:
- logging.info('Got MIFARE card id %s'%(repr(mifare_uid)))
- self.events.append((MIFARE, mifare_uid))
- timeout = 0
- if timeout == 0:
- break
-
- if len(self.events) == 0: return (TICK, time())
- ret = self.events[0]
- del self.events[0]
- return ret