def get_key_pkt(seed, level=1):
# type: (Packet, int) -> Optional[Packet]
def key_function_int(s):
# type: (int) -> int
return 0xffffffff & ~s
def key_function_short(s):
# type: (int) -> int
return 0xffff & ~s
try:
s = seed.securitySeed
except AttributeError:
return None
fmt = None
key_function = None # Optional[Callable[[int], int]]
if len(s) == 2:
fmt = "H"
key_function = key_function_short
if len(s) == 4:
fmt = "I"
key_function = key_function_int
if key_function is not None and fmt is not None:
key = struct.pack(fmt, key_function(struct.unpack(fmt, s)[0]))
return cast(
Packet,
UDS() / UDS_SA(securityAccessType=level + 1, securityKey=key))
else:
return None
def get_seed_pkt(sock, level=1, record=b""):
# type: (_SocketUnion, int, bytes) -> Optional[Packet]
req = UDS() / UDS_SA(securityAccessType=level,
securityAccessDataRecord=record)
for _ in range(10):
seed = sock.sr1(req, timeout=5, verbose=False)
if seed is None:
return None
elif seed.service == 0x7f and \
UDS_Enumerator._get_negative_response_code(seed) != 0x37:
log_interactive.info("Security access no seed! NR: %s",
repr(seed))
return None
elif seed.service == 0x7f and seed.negativeResponseCode == 0x37:
log_interactive.info("Security access retry to get seed")
time.sleep(10)
continue
else:
return seed
return None
def _get_initial_requests(self, **kwargs):
# type: (Any) -> Iterable[Packet]
scan_range = kwargs.pop("scan_range", range(1, 256, 2))
return (UDS() / UDS_SA(securityAccessType=x) for x in scan_range)