mirror of
https://github.com/solokeys/solo2.git
synced 2026-03-11 17:15:15 -07:00
Nicolas Stalder
139 lines
3.4 KiB
Python
139 lines
3.4 KiB
Python
#
|
|
# Some tests for very low-level functionality,
|
|
# when even pyscard/pcscd don't connect.
|
|
#
|
|
|
|
import struct
|
|
|
|
import psutil
|
|
import pytest
|
|
import usb.core
|
|
|
|
# VID = 0x1209
|
|
# PID = 0xBEEE
|
|
VID = 0x072F
|
|
PID = 0x90CC
|
|
|
|
CCID_CLASS = 0xB
|
|
CCID_SUBCLASS = 0
|
|
# CCID_PROTOCOL_0 = 0
|
|
|
|
from usb.util import (
|
|
find_descriptor,
|
|
claim_interface,
|
|
ENDPOINT_TYPE_BULK,
|
|
ENDPOINT_OUT,
|
|
ENDPOINT_IN,
|
|
endpoint_type,
|
|
endpoint_direction,
|
|
)
|
|
|
|
|
|
def ccid_compose(msg_type, seq, slot=0, rsv=0, param=0, data=b""):
|
|
msg = struct.pack('<BiBBBH', msg_type, len(data), slot, seq, rsv, param) + data
|
|
assert len(msg) < 64 # otherwise need to split up
|
|
return msg
|
|
|
|
|
|
class CCID:
|
|
def __init__(self, dev):
|
|
self.dev = dev
|
|
self.cfg = cfg = dev.get_active_configuration()
|
|
|
|
self.intf = intf = find_descriptor(
|
|
cfg, bInterfaceClass=CCID_CLASS, bInterfaceSubClass=CCID_SUBCLASS
|
|
)
|
|
if intf is None:
|
|
raise ValueError("Not a CCID device")
|
|
|
|
claim_interface(dev, intf)
|
|
|
|
for ep in intf:
|
|
if (
|
|
endpoint_type(ep.bmAttributes) == ENDPOINT_TYPE_BULK
|
|
and endpoint_direction(ep.bEndpointAddress) == ENDPOINT_OUT
|
|
):
|
|
self.bulkout = ep.bEndpointAddress
|
|
if (
|
|
endpoint_type(ep.bmAttributes) == ENDPOINT_TYPE_BULK
|
|
and endpoint_direction(ep.bEndpointAddress) == ENDPOINT_IN
|
|
):
|
|
self.bulkin = ep.bEndpointAddress
|
|
|
|
self.seq = 0
|
|
self.timeout = 10000
|
|
|
|
def increment_seq(self):
|
|
self.seq = (self.seq + 1) & 0xFF
|
|
|
|
def ccid_get_result(self):
|
|
msg = self.dev.read(self.bulkin, 1024, self.timeout)
|
|
|
|
if len(msg) < 10:
|
|
print(msg)
|
|
raise ValueError("ccid_get_result")
|
|
|
|
# assert len(msg) == 14
|
|
|
|
msg_type = msg[0]
|
|
data_len = msg[1] + (msg[2] << 8) + (msg[3] << 16) + (msg[4] << 24)
|
|
slot = msg[5]
|
|
seq = msg[6]
|
|
status = msg[7]
|
|
error = msg[8]
|
|
chain = msg[9]
|
|
data = msg[10:]
|
|
# XXX: check msg_type, data_len, slot, seq, error
|
|
return None, None, msg
|
|
return (status, chain, data.tobytes())
|
|
|
|
def power_on(self):
|
|
msg = ccid_compose(0x62, self.seq, rsv=1) # Vcc=5V
|
|
self.dev.write(self.bulkout, msg, self.timeout)
|
|
self.increment_seq()
|
|
status, chain, data = self.ccid_get_result()
|
|
# XXX: check status, chain
|
|
self.atr = atr = data
|
|
|
|
return atr
|
|
|
|
def power_off(self):
|
|
msg = ccid_compose(0x63, self.seq, rsv=1) # Vcc=5V
|
|
self.dev.write(self.bulkout, msg, self.timeout)
|
|
self.increment_seq()
|
|
stuff = status, chain, data = self.ccid_get_result()
|
|
print(f"data = {data}")
|
|
# XXX: check status, chain
|
|
|
|
return stuff
|
|
|
|
|
|
@pytest.fixture
|
|
def ccid():
|
|
"""The CCID interface to Zissou."""
|
|
|
|
dev = usb.core.find(idVendor=VID, idProduct=PID)
|
|
assert dev is not None
|
|
|
|
yield CCID(dev)
|
|
|
|
# do teardown here if necessary
|
|
pass
|
|
|
|
|
|
def test_no_pcscd():
|
|
pcscds = [p for p in psutil.process_iter() if p.name() == "pcscd"]
|
|
assert not pcscds, "Running `pcscd` process interferes with tests."
|
|
|
|
|
|
def test_connect(ccid):
|
|
"""Check access (udev."""
|
|
|
|
power_off = ccid.power_off()[2]
|
|
assert 0x81 == power_off[0]
|
|
|
|
atr = ccid.power_on()
|
|
assert 0x80 == atr[0]
|
|
l = atr[1]
|
|
assert [0x3B, 0x8C, 0x80, 0x01] == list(atr[10 : 10 + l])
|