trezor-firmware/trezorlib/transport_serial.py

'''SerialTransport implements wire transport over serial port.'''

# Local serial port loopback: socat PTY,link=COM8 PTY,link=COM9

import serial
from select import select
from transport import Transport

class SerialTransport(Transport):
    def __init__(self, device, *args, **kwargs):
        self.serial = None
        super(SerialTransport, self).__init__(device, *args, **kwargs)
        
    def _open(self):
        self.serial = serial.Serial(self.device, 115200, timeout=10, writeTimeout=10)
    
    def _close(self):
        self.serial.close()
        self.serial = None
    
    def ready_to_read(self):
        rlist, _, _ = select([self.serial], [], [], 0)
        return len(rlist) > 0
    
    def _write(self, msg, protobuf_msg):
        try:
            self.serial.write(msg)
            self.serial.flush()
        except serial.SerialException:
            print "Error while writing to socket"
            raise
        
    def _read(self):
        try:
            (msg_type, datalen) = self._read_headers(self.serial)
            return (msg_type, self.serial.read(datalen))
        except serial.SerialException:
            print "Failed to read from device"
            raise