implemented Zach's suggestion

UCSD-PL · Nov 4, 2012 · 763182a · 763182a
1 parent 48febdb
commit 763182a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/compiler/kernel-template/client/msg.py-template b/compiler/kernel-template/client/msg.py-template
@@ -10,9 +10,9 @@ def init():
   KCHAN = socket.fromfd(fd, socket.AF_UNIX, socket.SOCK_STREAM)
 
 def recv_num():
-  h = struct.unpack('>B', KCHAN.recv(1))[0]
-  l = struct.unpack('>B', KCHAN.recv(1))[0]
-  return (h * 256 + l)
+  s = KCHAN.recv(2)
+  n = struct.unpack('>H', s)[0]
+  return n
 
 def recv_str():
   n = recv_num()