X-Git-Url: https://git.ucc.asn.au/?p=uccvend-vendserver.git;a=blobdiff_plain;f=VendServer%2FOpenDispense.py;h=79cec66b0b180b2b04813ec60368ac255cfed83b;hp=de8ff840aab669452dcce1db447ecc06504768fd;hb=87633e7337e6bcf46a0be2fb057022ce8fb2eb7e;hpb=8cda0f578277ed1c4ce70c712d2e54311dd79e70

diff --git a/VendServer/OpenDispense.py b/VendServer/OpenDispense.py
index de8ff84..79cec66 100644
--- a/VendServer/OpenDispense.py
+++ b/VendServer/OpenDispense.py
@@ -12,6 +12,8 @@ import os
 import logging
 import re
 import pwd
+import base64
+import socket
 from subprocess import Popen, PIPE
 from LDAPConnector import get_uid,get_uname, set_card_id
 
@@ -28,6 +30,41 @@ class OpenDispense(DispenseInterface):
 		pass
 
 	def authUserIdPin(self, userId, pin):
+		return self.authUserIdPin_db(userId, pin)
+		#return self.authUserIdPin_file(userId, pin)
+	
+	def authUserIdPin_db(self, userId, pin):
+		userId = int(userId)
+
+		try:
+			# Get username (TODO: Store the user ID in the dispense database too)
+			info = pwd.getpwuid(userId)
+		except KeyError:
+			logging.info('getting pin for uid %d: user not in password file'%userId)
+			return False
+		
+		sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
+		sock.connect(DISPENSE_ENDPOINT)
+		logging.debug('connected to dispsrv')
+		sockf = sock.makefile()
+		sockf.write("AUTHIDENT\n"); sockf.flush()
+		rsp = sockf.readline()
+		assert "200" in rsp
+		logging.debug('authenticated')
+		sockf.write("PIN_CHECK %s %s\n" % (info.pw_name, pin)); sockf.flush()
+		rsp = sockf.readline()
+		if not "200" in rsp:
+			logging.info('checking pin for uid %d: Server said no - %r' % (userId, rsp))
+			return False
+		#Login Successful
+		logging.info('accepted pin for uid %d \'%s\'' % (userId, info.pw_name))
+		self._userid = userId
+		self._loggedIn = True
+		self._disabled = False
+		self._username = info.pw_name
+		return True
+
+	def authUserIdPin_file(self, userId, pin):
 		userId = int(userId)
 
 		try:
@@ -70,26 +107,33 @@ class OpenDispense(DispenseInterface):
 		return False
 
 	def authMifareCard(self, cardId):
+		self._loggedIn = False
+		self._username = None
 		if DISPSRV_MIFARE:
 			card_base64 = base64.b64encode(cardId)
 			
 			sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
 			sock.connect(DISPENSE_ENDPOINT)
-			sock.write("AUTHIDENT\n")
-			rsp = sock.readline()
+			logging.debug('connected to dispsrv')
+			sockf = sock.makefile()
+			sockf.write("AUTHIDENT\n"); sockf.flush()
+			rsp = sockf.readline()
 			assert "200" in rsp
-			sock.write("AUTHCARD %s\n" % (card_base64,))
-			rsp = sock.readline()
+			logging.debug('authenticated')
+			sockf.write("AUTHCARD %s\n" % (card_base64,)); sockf.flush()
+			rsp = sockf.readline()
 			if not "200" in rsp:
-				raise ValueError, "no UID found for card ID"
-			username = rsp.split('=')[1]
+				logging.info("Rejected card base64:%s" % (card_base64,))
+				return False
+			username = rsp.split('=')[1].strip()
+			logging.info("Accepted card base64:%s for %s" % (card_base64,username,))
 
 			# Check for thier username
 			try:
 				# Get info from the system (by username)
 				info = pwd.getpwnam(username)
 			except KeyError:
-				logging.info('getting pin for uid %d: user not in password file'%uid)
+				logging.info('getting info for user \'%s\': user not in password file' % (username,))
 				return False
 		else:
 			# Get the users ID
@@ -100,7 +144,7 @@ class OpenDispense(DispenseInterface):
 				# Get info from the system (by UID)
 				info = pwd.getpwuid(self._userid)
 			except KeyError:
-				logging.info('getting pin for uid %d: user not in password file'%uid)
+				logging.info('getting info for uid %d: user not in password file' % (self._userid,))
 				return False
 
 		# If we get this far all is good
@@ -109,6 +153,11 @@ class OpenDispense(DispenseInterface):
 		self._userid = info.pw_uid
 		self._username = info.pw_name
 		return True
+        def logOut(self):
+            self._loggedIn = False
+            self._disabled = False
+            self._userId = None
+            self._username = None
 
 	def addCard(self, cardId):
 		if not self.isLoggedIn():
@@ -118,14 +167,15 @@ class OpenDispense(DispenseInterface):
 			logging.info('Enrolling card %s to uid %s (%s)' % (cardId, self._userId, self._username))
 			sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
 			sock.connect(DISPENSE_ENDPOINT)
-			sock.write("AUTHIDENT\n")
-			rsp = sock.readline()
+			sockf = sock.makefile()
+			sockf.write("AUTHIDENT\n")
+			sockf.flush(); rsp = sockf.readline()
 			assert "200" in rsp
-			sock.write("SETEUSER %s\n", self._username)
-			rsp = sock.readline()
+			sockf.write("SETEUSER %s\n" % (self._username,))
+			sockf.flush(); rsp = sockf.readline()
 			assert "200" in rsp
-			sock.write("CARD_ADD %s\n", card_base64)
-			rsp = sock.readline()
+			sockf.write("CARD_ADD %s\n" % (card_base64,))
+			sockf.flush(); rsp = sockf.readline()
 			if "200" in rsp:
 				return True
 			else:
@@ -154,6 +204,7 @@ class OpenDispense(DispenseInterface):
 		return balance
 
 	def getItemInfo(self, itemId):
+		logging.debug("getItemInfo(%s)" % (itemId,))
 		itemId = OpenDispenseMapping.vendingMachineToOpenDispense(itemId)
 		args = ('dispense', 'iteminfo', itemId)
 		info, unused = Popen(args, close_fds=True, stdout=PIPE).communicate()
@@ -190,8 +241,20 @@ class OpenDispenseMapping():
 
 	@staticmethod
 	def vendingMachineToOpenDispense(itemId):
+		logging.debug("vendingMachineToOpenDispense(%s)" % (itemId,))
 		_mappingFile = "OpenDispenseMappings.conf"
-		fh = open(_mappingFile)
+		try:
+			fh = open(_mappingFile)
+		except IOError:
+			if itemId[1] == '8':
+				return 'coke:' + itemId[0]
+			elif itemId[1] == '5':
+				if itemId[0] == '5':
+					return 'door'
+				else:
+					return None
+			else:
+				return 'snack:' + itemId
 		map = ""
 		for line in fh:
 			#line = line.strip()