114 lines
3.8 KiB
Python
114 lines
3.8 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import argparse
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
class Main:
|
||
|
|
def __init__(self):
|
||
|
|
# command args
|
||
|
|
self.parser = argparse.ArgumentParser()
|
||
|
|
self.parser.add_argument("-d", "--debug", action="store_true", help="Debug")
|
||
|
|
self.subparsers = self.parser.add_subparsers(help="sub-command help")
|
||
|
|
self.parser_check = self.subparsers.add_parser(
|
||
|
|
"check", help="Check Option Bytes"
|
||
|
|
)
|
||
|
|
self.parser_check.set_defaults(func=self.check)
|
||
|
|
self.parser_set = self.subparsers.add_parser("set", help="Set Option Bytes")
|
||
|
|
self.parser_set.set_defaults(func=self.set)
|
||
|
|
# logging
|
||
|
|
self.logger = logging.getLogger()
|
||
|
|
# OB
|
||
|
|
self.ob = {}
|
||
|
|
|
||
|
|
def __call__(self):
|
||
|
|
self.args = self.parser.parse_args()
|
||
|
|
if "func" not in self.args:
|
||
|
|
self.parser.error("Choose something to do")
|
||
|
|
# configure log output
|
||
|
|
self.log_level = logging.DEBUG if self.args.debug else logging.INFO
|
||
|
|
self.logger.setLevel(self.log_level)
|
||
|
|
self.handler = logging.StreamHandler(sys.stdout)
|
||
|
|
self.handler.setLevel(self.log_level)
|
||
|
|
self.formatter = logging.Formatter("%(asctime)s [%(levelname)s] %(message)s")
|
||
|
|
self.handler.setFormatter(self.formatter)
|
||
|
|
self.logger.addHandler(self.handler)
|
||
|
|
# execute requested function
|
||
|
|
self.loadOB()
|
||
|
|
self.args.func()
|
||
|
|
|
||
|
|
def loadOB(self):
|
||
|
|
self.logger.info(f"Loading Option Bytes data")
|
||
|
|
file_path = os.path.join(os.path.dirname(sys.argv[0]), "ob.data")
|
||
|
|
file = open(file_path, "r")
|
||
|
|
for line in file.readlines():
|
||
|
|
k, v, o = line.split(":")
|
||
|
|
self.ob[k.strip()] = v.strip(), o.strip()
|
||
|
|
|
||
|
|
def check(self):
|
||
|
|
self.logger.info(f"Checking Option Bytes")
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(
|
||
|
|
["STM32_Programmer_CLI", "-q", "-c port=swd", "-ob displ"]
|
||
|
|
)
|
||
|
|
assert output
|
||
|
|
except Exception as e:
|
||
|
|
self.logger.error(f"Failed to call STM32_Programmer_CLI")
|
||
|
|
self.logger.exception(e)
|
||
|
|
return
|
||
|
|
ob_correct = True
|
||
|
|
for line in output.decode().split("\n"):
|
||
|
|
line = line.strip()
|
||
|
|
if not ":" in line:
|
||
|
|
self.logger.debug(f"Skipping line: {line}")
|
||
|
|
continue
|
||
|
|
key, data = line.split(":", 1)
|
||
|
|
key = key.strip()
|
||
|
|
data = data.strip()
|
||
|
|
if not key in self.ob.keys():
|
||
|
|
self.logger.debug(f"Skipping key: {key}")
|
||
|
|
continue
|
||
|
|
self.logger.debug(f"Processing key: {key} {data}")
|
||
|
|
value, comment = data.split(" ", 1)
|
||
|
|
value = value.strip()
|
||
|
|
comment = comment.strip()
|
||
|
|
if self.ob[key][0] != value:
|
||
|
|
self.logger.error(
|
||
|
|
f"Invalid OB: {key} {value}, expected: {self.ob[key][0]}"
|
||
|
|
)
|
||
|
|
ob_correct = False
|
||
|
|
if ob_correct:
|
||
|
|
self.logger.info(f"OB Check OK")
|
||
|
|
else:
|
||
|
|
self.logger.error(f"OB Check FAIL")
|
||
|
|
exit(255)
|
||
|
|
|
||
|
|
def set(self):
|
||
|
|
self.logger.info(f"Setting Option Bytes")
|
||
|
|
options = []
|
||
|
|
for key, (value, attr) in self.ob.items():
|
||
|
|
if "w" in attr:
|
||
|
|
options.append(f"{key}={value}")
|
||
|
|
try:
|
||
|
|
output = subprocess.check_output(
|
||
|
|
[
|
||
|
|
"STM32_Programmer_CLI",
|
||
|
|
"-q",
|
||
|
|
"-c port=swd",
|
||
|
|
f"-ob {' '.join(options)}",
|
||
|
|
]
|
||
|
|
)
|
||
|
|
assert output
|
||
|
|
self.logger.info(f"Success")
|
||
|
|
except Exception as e:
|
||
|
|
self.logger.error(f"Failed to call STM32_Programmer_CLI")
|
||
|
|
self.logger.exception(e)
|
||
|
|
return
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
Main()()
|