X-Git-Url: https://git.ucc.asn.au/?p=uccvend-vendserver.git;a=blobdiff_plain;f=sql-edition%2Fservers%2FLATClient.py;h=dedc1ff3d8ee816b6ce7936d08d5a98e92617839;hp=f1a9ca0cbcc18a59644d72b95e75400068c4dc2a;hb=34e039b5c1fa1650ff68d410bc95515c5394d565;hpb=f72d11b6eee9031792be32ecd92764e2bbb8db83

diff --git a/sql-edition/servers/LATClient.py b/sql-edition/servers/LATClient.py
index f1a9ca0..dedc1ff 100644
--- a/sql-edition/servers/LATClient.py
+++ b/sql-edition/servers/LATClient.py
@@ -28,9 +28,14 @@ def write_expecting_echo(rfh, wfh, message):
 	while len(message) > 0:
 		wfh.write(message[0])
 		wfh.flush()
-		echo = rfh.read(1)
-		if echo != message[0]:
-			print "Gah, expected echo when writing", message[0]
+		if message[0] == '\n':
+			echo = rfh.read(2)
+			expected_echo = '\r\n'
+		else:
+			echo = rfh.read(1)
+			expected_echo = message[0]
+		if echo != expected_echo:
+			print "Gah, expected echo when writing", repr(message[0]), "but got", repr(echo)
 		message = message[1:]
 	print "  --> Sent and echoed."
 
@@ -93,6 +98,7 @@ class LATClient:
 		self.sock.shutdown(2)
 		self.sock.close()
 		mopw, mopr = popen4('/usr/sbin/moprc '+self.server_name)
+		read_for_a_bit(mopr)
 		write_expecting_echo(mopr, mopw, '\n')
 		read_for_a_bit(mopr)
 		write_expecting_echo(mopr, mopw, self.connect_password+'\n')