forked from M-Labs/artiq
1
0
Fork 0

devices/corecom_serial: RPC server

This commit is contained in:
Sebastien Bourdeauducq 2014-07-07 19:11:43 +02:00
parent ffd2bdc4eb
commit 09e78a1f5e
1 changed files with 60 additions and 5 deletions

View File

@ -1,6 +1,23 @@
import os
import struct
import termios
import os, termios, struct
from enum import Enum
class _MsgType(Enum):
KERNEL_FINISHED = 0x01
RPC_REQUEST = 0x02
def _write_exactly(f, data):
remaining = len(data)
pos = 0
while remaining:
written = f.write(data[pos:])
remaining -= written
pos += written
def _read_exactly(f, n):
r = bytes()
while(len(r) < n):
r += f.read(n - len(r))
return r
class CoreCom:
def __init__(self, dev="/dev/ttyUSB1", baud=115200):
@ -31,5 +48,43 @@ class CoreCom:
self.close()
def run(self, kcode):
self.port.write(struct.pack(">LL", 0x5a5a5a5a, len(kcode)))
self.port.write(kcode)
_write_exactly(self.port, struct.pack(">ll", 0x5a5a5a5a, len(kcode)))
_write_exactly(self.port, kcode)
# FIXME: when loading immediately after a board reset, we erroneously get some zeros back.
# Ignore them with a warning for now.
spurious_zero_count = 0
while True:
(reply, ) = struct.unpack("b", _read_exactly(self.port, 1))
if reply == 0:
spurious_zero_count += 1
elif reply == 0x4f:
break
else:
raise IOError("Incorrect reply from device: "+hex(reply))
if spurious_zero_count:
print("Warning: received {} spurious zeros".format(spurious_zero_count))
def _wait_sync(self):
recognized = 0
while recognized < 4:
(c, ) = struct.unpack("b", _read_exactly(self.port, 1))
if c == 0x5a:
recognized += 1
else:
recognized = 0
def serve(self, rpc_map):
while True:
self._wait_sync()
msg = _MsgType(*struct.unpack("b", _read_exactly(self.port, 1)))
if msg == _MsgType.KERNEL_FINISHED:
return
elif msg == _MsgType.RPC_REQUEST:
rpc_num, n_args = struct.unpack(">hb", _read_exactly(self.port, 3))
args = []
for i in range(n_args):
args.append(*struct.unpack(">l", _read_exactly(self.port, 4)))
r = rpc_map[rpc_num](*args)
if r is None:
r = 0
_write_exactly(self.port, struct.pack(">l", r))