98 lines
3.1 KiB
Python
Executable File
98 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import sys
|
|
from typing import Dict, List
|
|
|
|
from common.basedir import BASEDIR
|
|
|
|
# messages reserved for CAN based ignition (see can_ignition_hook function in panda/board/drivers/can)
|
|
# (addr, len)
|
|
CAN_IGNITION_MSGS = {
|
|
'gm': [(0x1F1, 8), (0x160, 5)],
|
|
#'tesla' : [(0x348, 8)],
|
|
}
|
|
|
|
def _get_fingerprints():
|
|
# read all the folders in selfdrive/car and return a dict where:
|
|
# - keys are all the car names that which we have a fingerprint dict for
|
|
# - values are dicts of fingeprints for each trim
|
|
fingerprints = {}
|
|
for car_folder in [x[0] for x in os.walk(BASEDIR + '/selfdrive/car')]:
|
|
car_name = car_folder.split('/')[-1]
|
|
try:
|
|
fingerprints[car_name] = __import__(f'selfdrive.car.{car_name}.values', fromlist=['FINGERPRINTS']).FINGERPRINTS
|
|
except (ImportError, OSError, AttributeError):
|
|
pass
|
|
|
|
return fingerprints
|
|
|
|
|
|
def check_fingerprint_consistency(f1, f2):
|
|
# return false if it finds a fingerprint fully included in another
|
|
# max message worth checking is 1800, as above that they usually come too infrequently and not
|
|
# usable for fingerprinting
|
|
|
|
max_msg = 1800
|
|
|
|
is_f1_in_f2 = True
|
|
for k in f1:
|
|
if (k not in f2 or f1[k] != f2[k]) and k < max_msg:
|
|
is_f1_in_f2 = False
|
|
|
|
is_f2_in_f1 = True
|
|
for k in f2:
|
|
if (k not in f1 or f2[k] != f1[k]) and k < max_msg:
|
|
is_f2_in_f1 = False
|
|
|
|
return not is_f1_in_f2 and not is_f2_in_f1
|
|
|
|
|
|
def check_can_ignition_conflicts(fingerprints, brands):
|
|
# loops through all the fingerprints and exits if CAN ignition dedicated messages
|
|
# are found in unexpected fingerprints
|
|
|
|
for brand_can, msgs_can in CAN_IGNITION_MSGS.items():
|
|
for i, f in enumerate(fingerprints):
|
|
for msg_can in msgs_can:
|
|
if brand_can != brands[i] and msg_can[0] in f and msg_can[1] == f[msg_can[0]]:
|
|
print("CAN ignition dedicated msg %d with len %d found in %s fingerprints!" % (msg_can[0], msg_can[1], brands[i]))
|
|
print("TEST FAILED")
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
fingerprints = _get_fingerprints()
|
|
|
|
fingerprints_flat: List[Dict] = []
|
|
car_names = []
|
|
brand_names = []
|
|
for brand in fingerprints:
|
|
for car in fingerprints[brand]:
|
|
fingerprints_flat += fingerprints[brand][car]
|
|
for i in range(len(fingerprints[brand][car])):
|
|
car_names.append(car)
|
|
brand_names.append(brand)
|
|
|
|
# first check if CAN ignition specific messages are unexpectedly included in other fingerprints
|
|
check_can_ignition_conflicts(fingerprints_flat, brand_names)
|
|
|
|
valid = True
|
|
for idx1, f1 in enumerate(fingerprints_flat):
|
|
for idx2, f2 in enumerate(fingerprints_flat):
|
|
if idx1 < idx2 and not check_fingerprint_consistency(f1, f2):
|
|
valid = False
|
|
print(f"Those two fingerprints are inconsistent {car_names[idx1]} {car_names[idx2]}")
|
|
print("")
|
|
print(', '.join("%d: %d" % v for v in sorted(f1.items())))
|
|
print("")
|
|
print(', '.join("%d: %d" % v for v in sorted(f2.items())))
|
|
print("")
|
|
|
|
print(f"Found {len(fingerprints_flat)} individual fingerprints")
|
|
if not valid or len(fingerprints_flat) == 0:
|
|
print("TEST FAILED")
|
|
sys.exit(1)
|
|
else:
|
|
print("TEST SUCESSFUL")
|