def printControlRequest(submit, data_str, data_size, pipe_str):
if submit.m_ctrl.bRequestType & URB_TRANSFER_IN:
indented("buff = controlRead(0x%02X, 0x%02X, 0x%04X, 0x%04X, %u)" %
(submit.m_ctrl.bRequestType, submit.m_ctrl.bRequest,
submit.m_ctrl.wValue, submit.m_ctrl.wIndex, data_size))
else:
indented("controlWrite(0x%02X, 0x%02X, 0x%04X, 0x%04X, %s)" %
(submit.m_ctrl.bRequestType, submit.m_ctrl.bRequest,
submit.m_ctrl.wValue, submit.m_ctrl.wIndex, data_str))
def header(self):
comment("Generated by usbrply")
# comment("Date: %s" % (UVDCurDateTime()))
comment("cmd: %s" % (' '.join(sys.argv), ))
indented("")
if self.wrapper:
print('''
import binascii
import time
import usb1
def validate_read(expected, actual, msg):
if expected != actual:
print('Failed %s' % msg)
print(' Expected; %s' % binascii.hexlify(expected,))
print(' Actual: %s' % binascii.hexlify(actual,))
#raise Exception('failed validate: %s' % msg)
''')
print('def replay(dev):')
indent_inc()
print('''\
def bulkRead(endpoint, length, timeout=None):
return dev.bulkRead(endpoint, length, timeout=(1000 if timeout is None else timeout))
def bulkWrite(endpoint, data, timeout=None):
dev.bulkWrite(endpoint, data, timeout=(1000 if timeout is None else timeout))
def controlRead(request_type, request, value, index, length,
timeout=None):
return dev.controlRead(request_type, request, value, index, length,
timeout=(1000 if timeout is None else timeout))
def controlWrite(request_type, request, value, index, data,
timeout=None):
dev.controlWrite(request_type, request, value, index, data,
timeout=(1000 if timeout is None else timeout))
''')
def comment(s):
indented('# %s' % (s, ))
def run(self, j):
self.header()
# Last wire command (ie non-comment)
# Used to optionally generate timing
prevd = None
for d in j["data"]:
# print(d)
"""
FIXME: think need to add timing info to JSON?
if self.sleep and prevd:
# mind order of operations here...was having round off issues
ds = self.submit.m_urb.sec - prev.sec
dt = ds + self.submit.m_urb.usec//1.e6 - prev.usec//1.e6
if dt < -1.e-6:
# stupid reversed packets
if 0:
print('prev sec: %s' % prev.sec)
print('prev usec: %s' % prev.usec)
print('this sec: %s' % self.submit.m_urb.sec)
print('this usec: %s' % self.submit.m_urb.usec)
raise Exception("bad calc: %s" % dt)
elif dt >= 0.001:
indented('time.sleep(%.3f)' % (dt,))
"""
packet_numbering = ''
if d["type"] == "comment":
comment(d["v"])
continue
if self.packet_numbers:
if "packm" in d:
packet_numbering = "packet %s/%s" % (d["packn"],
d["packm"])
else:
packet_numbering = "packet %s/?" % (d["packn"], )
else:
# TODO: consider counting instead of by captured index
packet_numbering = "packet"
if d["type"] == "controlRead":
# Is it legal to have a 0 length control in?
indented(
"buff = controlRead(0x%02X, 0x%02X, 0x%04X, 0x%04X, %u)" %
(d["bRequestType"], d["bRequest"], d["wValue"],
d["wIndex"], d["wLength"]))
indented("validate_read(%s, buff, \"%s\")" %
(bytes2AnonArray(d["data"]), packet_numbering))
elif d["type"] == "controlWrite":
data_str = bytes2AnonArray(d["data"])
indented("controlWrite(0x%02X, 0x%02X, 0x%04X, 0x%04X, %s)" %
(d["bRequestType"], d["bRequest"], d["wValue"],
d["wIndex"], data_str))
elif d["type"] == "bulkRead":
data_str = "\"\""
indented("buff = bulkRead(0x%02X, 0x%04X)" %
(d["endp"], d["len"]))
indented("validate_read(%s, buff, \"%s\")" %
(bytes2AnonArray(d["data"]), packet_numbering))
elif d["type"] == "bulkWrite":
# Note that its the submit from earlier, not the ack that we care about
data_str = bytes2AnonArray(d["data"])
# def bulkWrite(self, endpoint, data, timeout=0):
indented("bulkWrite(0x%02X, %s)" % (d["endp"], data_str))
# these aren't event added to JSON right now
# print('%s# WARNING: omitting interrupt' % (indent,))
prevd = d
self.footer()