From: Bernard Blackham Date: Fri, 25 Jun 2004 14:23:08 +0000 (+0000) Subject: Initial stuff X-Git-Tag: IMPORT-FROM-UCCCVS~157 X-Git-Url: https://git.ucc.asn.au/?a=commitdiff_plain;h=34659fae1f24885353883f43ee9f7acb64fa0629;p=uccvend-vendserver.git Initial stuff --- diff --git a/sql-edition/servers/LATClient.py b/sql-edition/servers/LATClient.py new file mode 100644 index 0000000..4b2a1d9 --- /dev/null +++ b/sql-edition/servers/LATClient.py @@ -0,0 +1,54 @@ +from socket import * + +LATCP_SOCKET = '/var/run/latlogin' + +LAT_VERSION = '1.21' +LATCP_CMD_VERSION = 8 +LATCP_CMD_TERMINALSESSION = 26 +LATCP_CMD_ERRORMSG = 99 + +class LATClient: + def __init__(self, service = None, node = None, port = None, \ + localport = None, password = None, is_queued = False): + self.sock = socket(AF_UNIX, SOCK_STREAM, 0); + self.sock.connect(LATCP_SOCKET) + self.send_msg(LATCP_CMD_VERSION, LAT_VERSION+'\000') + (cmd, msg) = self.read_reply() + if service == None: service = '' + if node == None: node = '' + if port == None: port = '' + if localport == None: localport = '' + if password == None: password = '' + if is_queued == True: + is_queued = 1 + else: + is_queued = 0 + self.send_msg(LATCP_CMD_TERMINALSESSION, '%c%c%s%c%s%c%s%c%s%c%s' % \ + (is_queued, + len(service), service, + len(node), node, + len(port), port, + len(localport), localport, + len(password), password + )) + (cmd, msg) = self.read_reply() + if cmd == LATCP_CMD_ERRORMSG: + raise Exception + + def send_msg(self, cmd, msg): + self.sock.send('%c%c%c%s'%(cmd, len(msg)/256, len(msg)%256, msg)) + + def read_reply(self): + head = self.sock.recv(3) + if len(head) != 3: + sys.stderr.write('Error: Short LAT packet\n') + return None + cmd = head[0] + length = ord(head[1])*256 + ord(head[2]) + msg = self.sock.recv(length) + if cmd == LATCP_CMD_ERRORMSG: + sys.stderr.write('Error: Received LAT error: %s\n'%msg) + return (cmd, msg) + + def get_fh(self): + return (self.sock.makefile('r'), self.sock.makefile('w')) diff --git a/sql-edition/servers/VendServer.py b/sql-edition/servers/VendServer.py new file mode 100755 index 0000000..57a2682 --- /dev/null +++ b/sql-edition/servers/VendServer.py @@ -0,0 +1,44 @@ +#!/usr/bin/python + +import sys, os, string, socket, time +from popen2 import popen2 +from pyPgSQL import PgSQL +from LATClient import LATClient + +class VendingMachine: + def __init__(self, rfh, wfh): + self.rfh = rfh + self.wfh = wfh + self.wfh.write('\n') + self.await_prompt() + + def await_prompt(self): + self.wfh.flush() + state = 0 + s = '' + while state != 3: + s = self.rfh.read(1) + if s == '': raise Exception + if s == '\n' and state == 0: state = 1 + if s == '#' and state == 1: state = 2 + if s == ' ' and state == 2: state = 3 + + def get_response(self): + self.wfh.flush() + s = '' + while s == '': + s = self.rfh.readline() + if s == '': return None + s = s.strip('\r\n') + return s + + def ping(self): + self.wfh.write('PING\n') + return self.get_response() + +if __name__ == '__main__': + # Open vending machine via LAT + latclient = LATClient(service = 'VEND', password = 'dmscptd') + (rfh, wfh) = latclient.get_fh() + v = VendingMachine(rfh, wfh) + print v.ping()