mirror of
https://github.com/dekuNukem/USB4VC.git
synced 2025-10-31 11:26:46 -07:00
108 lines
2.9 KiB
Python
108 lines
2.9 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
import spidev
|
|
import RPi.GPIO as GPIO
|
|
|
|
PBOARD_RESET_PIN = 25
|
|
PBOARD_BOOT0_PIN = 12
|
|
|
|
SLAVE_REQ_PIN = 16
|
|
GPIO.setmode(GPIO.BCM)
|
|
GPIO.setup(PBOARD_RESET_PIN, GPIO.IN)
|
|
GPIO.setup(PBOARD_BOOT0_PIN, GPIO.IN)
|
|
GPIO.setup(SLAVE_REQ_PIN, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
|
|
|
|
is_dfu = False
|
|
|
|
def enter_dfu():
|
|
# RESET LOW: Enter reset
|
|
GPIO.setup(PBOARD_RESET_PIN, GPIO.OUT)
|
|
GPIO.output(PBOARD_RESET_PIN, GPIO.LOW)
|
|
time.sleep(0.05)
|
|
# BOOT0 HIGH: Boot into DFU mode
|
|
GPIO.setup(PBOARD_BOOT0_PIN, GPIO.OUT)
|
|
GPIO.output(PBOARD_BOOT0_PIN, GPIO.HIGH)
|
|
time.sleep(0.05)
|
|
# Release RESET, BOOT0 still HIGH, STM32 now in DFU mode
|
|
GPIO.setup(PBOARD_RESET_PIN, GPIO.IN)
|
|
time.sleep(1)
|
|
|
|
def exit_dfu():
|
|
# Release BOOT0
|
|
GPIO.setup(PBOARD_BOOT0_PIN, GPIO.IN)
|
|
# Activate RESET
|
|
GPIO.setup(PBOARD_RESET_PIN, GPIO.OUT)
|
|
GPIO.output(PBOARD_RESET_PIN, GPIO.LOW)
|
|
time.sleep(0.05)
|
|
# Release RESET, BOOT0 is LOW, STM32 boots in normal mode
|
|
GPIO.setup(PBOARD_RESET_PIN, GPIO.IN)
|
|
time.sleep(0.2)
|
|
|
|
def flash_firmware(fw_path):
|
|
for x in range(5):
|
|
print(f"----------------- {fw_path.split('/')[-1]} -----------------")
|
|
enter_dfu()
|
|
if is_dfu:
|
|
exit_code = os.system(f'sudo dfu-util --device ,0483:df11 -a 0 -D {fw_path}') >> 8
|
|
else:
|
|
exit_code = os.system(f'sudo stm32flash -w {fw_path} -a 0x3b /dev/i2c-1') >> 8
|
|
exit_dfu()
|
|
if exit_code != 0:
|
|
for x in range(5):
|
|
print("!!!!!!!!!!!!!!!!! TEST FLASH FAILED !!!!!!!!!!!!!!!!!")
|
|
exit()
|
|
|
|
if(len(sys.argv) < 3):
|
|
print (__file__ + ' payload_fw test_fw')
|
|
exit()
|
|
|
|
os.system("clear")
|
|
|
|
pcard_spi = spidev.SpiDev(0, 0)
|
|
pcard_spi.max_speed_hz = 2000000
|
|
|
|
payload_fw_path = sys.argv[1]
|
|
test_fw_path = sys.argv[2]
|
|
|
|
if '.dfu' in payload_fw_path.lower() or '.dfu' in test_fw_path.lower():
|
|
is_dfu = True
|
|
|
|
flash_firmware(test_fw_path)
|
|
req_result = []
|
|
for x in range(10):
|
|
req_result.append(GPIO.input(SLAVE_REQ_PIN))
|
|
time.sleep(0.1)
|
|
|
|
print(req_result)
|
|
|
|
if 0 not in req_result or 1 not in req_result or req_result.count(0) <= 3 or req_result.count(1) <= 3:
|
|
for x in range(5):
|
|
print("!!!!!!!!!!!!!!!!! SLAVE REQ ERROR !!!!!!!!!!!!!!!!!")
|
|
exit()
|
|
|
|
while 1:
|
|
if len(input("Press enter to continue\n")) == 0:
|
|
break;
|
|
|
|
flash_firmware(payload_fw_path)
|
|
|
|
SPI_MOSI_MAGIC = 0xde
|
|
SPI_MOSI_MSG_TYPE_INFO_REQUEST = 1
|
|
|
|
nop_spi_msg_template = [SPI_MOSI_MAGIC] + [0]*31
|
|
info_request_spi_msg_template = [SPI_MOSI_MAGIC, 0, SPI_MOSI_MSG_TYPE_INFO_REQUEST] + [0]*29
|
|
|
|
this_msg = list(info_request_spi_msg_template)
|
|
pcard_spi.xfer(this_msg)
|
|
time.sleep(0.1)
|
|
response = pcard_spi.xfer(list(nop_spi_msg_template))
|
|
time.sleep(0.1)
|
|
print(response)
|
|
|
|
if response[0] != 205:
|
|
for x in range(5):
|
|
print("!!!!!!!!!!!!!!!!! WRONG RESPONSE !!!!!!!!!!!!!!!!!")
|
|
else:
|
|
print("----------------- OK OK OK OK OK OK -----------------")
|
|
print("----------------- OK OK OK OK OK OK -----------------") |