from PyQt5 import QtWidgets, QtCore, QtGui
|
from can.interfaces.canalystii import CANalystIIBus, TIMING_DICT
|
from can.interfaces.vector import VectorBus
|
from can.bus import BusABC
|
import asyncio
|
import time
|
import aioisotp
|
from uds.client import Client
|
from udsoncan import MemoryLocation
|
import configparser
|
|
|
class FlashBootloaderWidget(QtWidgets.QWidget):
|
|
def __init__(self, parent=None, flags=QtCore.Qt.WindowFlags()):
|
super().__init__(parent=parent, flags=flags)
|
self.initUI()
|
self.in_programming = False
|
self.bus: BusABC = None
|
self.drv_data: bytes = None
|
self.app_data: bytes = None
|
|
try:
|
config = configparser.ConfigParser()
|
config.read('config.ini')
|
self.tx_id = int(config['CAN_ID']['tx_id'], 16)
|
self.rx_id = int(config['CAN_ID']['rx_id'], 16)
|
self.append_text_to_log_display('read config.ini success')
|
except Exception as e:
|
self.append_text_to_log_display(
|
'read config.ini error, use 0x692 and 0x693 for default tx_id and rx_id', '#ff0000')
|
self.tx_id = 0x692
|
self.rx_id = 0x693
|
|
def initUI(self):
|
|
self.setWindowTitle('Flash Bootloader')
|
self.setFont(QtGui.QFont('Segoe UI'))
|
desktop = QtWidgets.QApplication.desktop() # 获得屏幕尺寸
|
self.resize(desktop.width()/2, desktop.height()/3)
|
le_drv = QtWidgets.QLineEdit(self)
|
le_drv.setReadOnly(True)
|
le_drv.setObjectName('le_drv')
|
btn_drv = QtWidgets.QPushButton(self)
|
btn_drv.setText('driver')
|
btn_drv.setObjectName('btn_drv')
|
|
le_app = QtWidgets.QLineEdit(self)
|
le_app.setReadOnly(True)
|
le_app.setObjectName('le_app')
|
btn_app = QtWidgets.QPushButton(self)
|
btn_app.setText('app')
|
btn_app.setObjectName('btn_app')
|
|
gl0 = QtWidgets.QGridLayout()
|
gl0.addWidget(le_drv, 0, 0, 1, 1)
|
gl0.addWidget(btn_drv, 0, 1, 1, 1)
|
gl0.addWidget(le_app, 1, 0, 1, 1)
|
gl0.addWidget(btn_app, 1, 1, 1, 1)
|
|
cb_interface_type = QtWidgets.QComboBox(self)
|
cb_interface_type.addItems(['vector', 'canalystii'])
|
cb_interface_type.setObjectName('cb_interface_type')
|
|
cb_baudrate = QtWidgets.QComboBox(self)
|
cb_baudrate.addItems([('%dK' % (i/1000)) for i in TIMING_DICT.keys()])
|
cb_baudrate.setCurrentIndex(14)
|
cb_baudrate.setObjectName('cb_baudrate')
|
|
cb_can_channel = QtWidgets.QComboBox(self)
|
cb_can_channel.addItems(['CH1/3', 'CH2/4'])
|
cb_can_channel.setObjectName('cb_can_channel')
|
|
btn_open = QtWidgets.QPushButton(self)
|
btn_open.setText('open')
|
btn_open.setObjectName('btn_open')
|
|
hb0 = QtWidgets.QHBoxLayout()
|
hb0.addWidget(cb_interface_type)
|
hb0.addWidget(cb_baudrate)
|
hb0.addWidget(cb_can_channel)
|
hb0.addWidget(btn_open)
|
|
cb_only_main = QtWidgets.QCheckBox(self)
|
cb_only_main.setText('only programming step')
|
|
btn_start = QtWidgets.QPushButton(self)
|
btn_start.setText('start programming')
|
btn_start.setObjectName('btn_start')
|
btn_start.setDisabled(True)
|
|
hb1 = QtWidgets.QHBoxLayout()
|
hb1.addWidget(cb_only_main)
|
hb1.addWidget(btn_start)
|
|
progress_bar = QtWidgets.QProgressBar()
|
progress_bar.setObjectName('progress_bar')
|
|
hb2 = QtWidgets.QHBoxLayout()
|
hb2.addWidget(progress_bar)
|
|
# checkbox1 = QtWidgets.QCheckBox(self)
|
# checkbox2 = QtWidgets.QCheckBox(self)
|
# checkbox3 = QtWidgets.QCheckBox(self)
|
# checkbox4 = QtWidgets.QCheckBox(self)
|
# checkbox5 = QtWidgets.QCheckBox(self)
|
# checkbox6 = QtWidgets.QCheckBox(self)
|
# checkbox7 = QtWidgets.QCheckBox(self)
|
# checkbox8 = QtWidgets.QCheckBox(self)
|
# checkbox9 = QtWidgets.QCheckBox(self)
|
# checkbox10 = QtWidgets.QCheckBox(self)
|
# checkbox11 = QtWidgets.QCheckBox(self)
|
|
# # https://blog.csdn.net/jianfengxia/article/details/86623321
|
|
# checkbox1.setText('Enter extended session')
|
# checkbox2.setText('Stop setting of DTCs')
|
# checkbox3.setText('Disable non-diagnostic communication')
|
# vb_pre = QtWidgets.QVBoxLayout()
|
# vb_pre.setAlignment(QtCore.Qt.AlignTop)
|
# vb_pre.addWidget(checkbox1)
|
# vb_pre.addWidget(checkbox2)
|
# vb_pre.addWidget(checkbox3)
|
# gb_pre = QtWidgets.QGroupBox(self)
|
# gb_pre.setTitle('pre-programming')
|
# gb_pre.setLayout(vb_pre)
|
|
# checkbox4.setText('Enter programming session')
|
# checkbox5.setText('Request seed')
|
# checkbox6.setText('Send key')
|
# checkbox7.setText('Write Programming Date')
|
# checkbox8.setText('Erase Application Software Memory')
|
# checkbox9.setText('Download Application Software')
|
# checkbox10.setText(
|
# 'Check Programming Application Software Dependencie')
|
|
# vb_main = QtWidgets.QVBoxLayout()
|
# vb_main.setAlignment(QtCore.Qt.AlignTop)
|
# vb_main.addWidget(checkbox4)
|
# vb_main.addWidget(checkbox5)
|
# vb_main.addWidget(checkbox6)
|
# vb_main.addWidget(checkbox7)
|
# vb_main.addWidget(checkbox8)
|
# vb_main.addWidget(checkbox9)
|
# vb_main.addWidget(checkbox10)
|
# gb_main = QtWidgets.QGroupBox(self)
|
# gb_main.setTitle('programming')
|
# gb_main.setLayout(vb_main)
|
|
# checkbox11.setText('ECUReset')
|
# vb_post = QtWidgets.QVBoxLayout()
|
# vb_post.setAlignment(QtCore.Qt.AlignTop)
|
# vb_post.addWidget(checkbox11)
|
# gb_post = QtWidgets.QGroupBox(self)
|
# gb_post.setTitle('post-programming')
|
# gb_post.setLayout(vb_post)
|
|
# hb3 = QtWidgets.QHBoxLayout()
|
# hb3.addWidget(gb_pre)
|
# hb3.addWidget(gb_main)
|
# hb3.addWidget(gb_post)
|
|
tb_log_display = QtWidgets.QTextBrowser(self)
|
tb_log_display.setObjectName('tb_log_display')
|
|
hb4 = QtWidgets.QHBoxLayout()
|
hb4.addWidget(tb_log_display)
|
|
vb = QtWidgets.QVBoxLayout()
|
vb.addLayout(gl0)
|
vb.addLayout(hb0)
|
vb.addLayout(hb1)
|
vb.addLayout(hb2)
|
# vb.addLayout(hb3)
|
vb.addLayout(hb4)
|
|
self.setLayout(vb)
|
|
QtCore.QMetaObject.connectSlotsByName(self)
|
|
async def pre_programming(self, client: Client):
|
|
self.append_text_to_log_display('# 预编程步骤')
|
|
# 进入extended session
|
self.append_text_to_log_display('>>> 进入extended session')
|
response = await client.change_session(3)
|
# print(response)
|
|
# 检查编程条件
|
self.append_text_to_log_display('>>> 检查编程条件')
|
response = await client.start_routine(0xff02)
|
# print(response)
|
|
# 关闭DTC的存储
|
self.append_text_to_log_display('>>> 关闭DTC的存储')
|
response = await client.control_dtc_setting(2)
|
# print(response)
|
|
# 关闭与诊断无关的报文
|
self.append_text_to_log_display('>>> 关闭与诊断无关的报文')
|
response = await client.communication_control(0x03, 0x01)
|
# print(response)
|
|
async def programming(self, client: Client):
|
|
self.append_text_to_log_display('# 主编程步骤')
|
|
# 进入programming session
|
self.append_text_to_log_display('>>> 进入programming session')
|
response = await client.change_session(2)
|
# print(response)
|
|
# 请求种子
|
self.append_text_to_log_display('>>> 请求种子')
|
response = await client.request_seed(1)
|
# print(response)
|
|
# 发送密钥
|
self.append_text_to_log_display('>>> 发送密钥')
|
response = await client.send_key(1, bytes([0xb3, 0x42]))
|
# print(response)
|
|
# 发送driver文件
|
self.append_text_to_log_display('>>> 发送driver文件')
|
drv_file = self.drv_data
|
address = drv_file[16] << 24 | drv_file[17] << 16 | drv_file[18] << 8 | drv_file[19] << 0
|
memorysize = drv_file[20] << 24 | drv_file[21] << 16 | drv_file[22] << 8 | drv_file[23] << 0
|
memory_location = MemoryLocation(address, memorysize, 32, 32)
|
response = await client.request_download(memory_location)
|
# print(response)
|
|
max_length = response.service_data.max_length
|
|
# 有效数据长度, 去除sid和sequence两个字节
|
payload_length = max_length - 2
|
|
count = (len(drv_file) + payload_length - 1) // payload_length
|
|
base = self.get_progressbar_pos()
|
for i in range(count):
|
start = i * payload_length
|
end = start + payload_length
|
response = await client.transfer_data((i + 1) % 256, drv_file[start:end])
|
self.set_progressbar_pos(base + end)
|
# print(response)
|
|
response = await client.request_transfer_exit()
|
# print(response)
|
|
# driver文件完整性检验
|
self.append_text_to_log_display('>>> driver文件完整性检验')
|
response = await client.start_routine(0xf001, drv_file[0:4])
|
# print(response)
|
|
app_file = self.app_data
|
address = app_file[16] << 24 | app_file[17] << 16 | app_file[18] << 8 | app_file[19] << 0
|
memorysize = app_file[20] << 24 | app_file[21] << 16 | app_file[22] << 8 | app_file[23] << 0
|
memory_location = MemoryLocation(address, memorysize, 32, 32)
|
|
# 删除app存储空间
|
self.append_text_to_log_display('>>> 删除app存储空间')
|
data = b''
|
data += memory_location.alfid.get_byte()
|
data += memory_location.get_address_bytes()
|
data += memory_location.get_memorysize_bytes()
|
response = await client.start_routine(0xff00, data)
|
# print(response)
|
|
# 发送app文件
|
self.append_text_to_log_display('>>> 发送app文件')
|
response = await client.request_download(memory_location)
|
# print(response)
|
|
max_length = response.service_data.max_length
|
|
# 有效数据长度, 去除sid和sequence两个字节
|
payload_length = max_length - 2
|
|
count = (len(app_file) + payload_length - 1) // payload_length
|
|
base = self.get_progressbar_pos()
|
for i in range(count):
|
start = i * payload_length
|
end = start + payload_length
|
response = await client.transfer_data((i + 1) % 256, app_file[start:end])
|
self.set_progressbar_pos(base + end)
|
# print(response)
|
|
response = await client.request_transfer_exit()
|
# print(response)
|
|
# app文件完整性检验
|
self.append_text_to_log_display('>>> app文件完整性检验')
|
response = await client.start_routine(0xf001, app_file[0:4])
|
# print(response)
|
|
async def post_programming(self, client: Client):
|
|
self.append_text_to_log_display('# 后编程步骤')
|
|
# 回到default session
|
self.append_text_to_log_display('>>> 回到default session')
|
response = await client.change_session(1, True)
|
# print(response)
|
|
async def start_programming(self):
|
|
network = aioisotp.ISOTPNetwork(bus=self.bus)
|
|
with network.open():
|
|
reader, writer = await network.open_connection(self.rx_id, self.tx_id)
|
|
client = Client(reader, writer)
|
|
try:
|
await client.tester_present(timeout=0.3)
|
except asyncio.TimeoutError:
|
# canalystii 第一条请求无效
|
# timeout >= 0.3
|
if isinstance(self.bus, CANalystIIBus):
|
pass
|
elif isinstance(self.bus, VectorBus):
|
self.append_text_to_log_display('查看是否选错CAN通道', '#ff0000')
|
|
except Exception as e:
|
self.append_text_to_log_display('%s' % e, '#ff0000')
|
|
self.set_progressbar_pos(0)
|
|
t1 = time.time()
|
|
try:
|
|
await self.pre_programming(client)
|
await self.programming(client)
|
await self.post_programming(client)
|
|
except asyncio.TimeoutError:
|
|
self.append_text_to_log_display('request timeout', '#ff0000')
|
|
except Exception as e:
|
|
self.append_text_to_log_display('%s' % e, '#ff0000')
|
|
t2 = time.time()
|
|
self.append_text_to_log_display('finished in %.2f sec' % (t2 - t1))
|
self.in_programming = False
|
self.findChild(QtWidgets.QPushButton,
|
'btn_start').setDisabled(True)
|
|
def set_progressbar_pos(self, pos):
|
|
drv_data_len = len(
|
self.drv_data) if self.drv_data is not None else 0
|
app_data_len = len(
|
self.app_data) if self.app_data is not None else 0
|
total_len = drv_data_len + app_data_len
|
if pos > total_len:
|
pos = total_len
|
elif pos < 0:
|
pos = 0
|
|
self.findChild(QtWidgets.QProgressBar, 'progress_bar').setValue(pos)
|
|
def get_progressbar_pos(self):
|
|
return self.findChild(QtWidgets.QProgressBar, 'progress_bar').value()
|
|
def append_text_to_log_display(self, text, color='#000000'):
|
|
text = '<font color="%s">%s</font>' % (color, text)
|
self.findChild(QtWidgets.QTextBrowser, 'tb_log_display').append(text)
|
|
def clear_log_display(self):
|
|
self.findChild(QtWidgets.QTextBrowser, 'tb_log_display').clear()
|
|
@QtCore.pyqtSlot()
|
def on_btn_drv_clicked(self):
|
|
if self.in_programming:
|
self.append_text_to_log_display('## 正在编程中')
|
return
|
|
fileName, _ = QtWidgets.QFileDialog.getOpenFileName(
|
self, 'load driver file', '.', 'All Files (*);;Bin Files (*.bin)')
|
|
if fileName != '':
|
self.findChild(QtWidgets.QLineEdit, 'le_drv').setText(fileName)
|
with open(fileName, 'rb') as fd:
|
self.drv_data = fd.read()
|
drv_data_len = len(
|
self.drv_data) if self.drv_data is not None else 0
|
app_data_len = len(
|
self.app_data) if self.app_data is not None else 0
|
total_len = drv_data_len + app_data_len
|
self.findChild(QtWidgets.QProgressBar,
|
'progress_bar').setRange(0, total_len)
|
|
@QtCore.pyqtSlot()
|
def on_btn_app_clicked(self):
|
|
if self.in_programming:
|
self.append_text_to_log_display('## 正在编程中')
|
return
|
|
fileName, _ = QtWidgets.QFileDialog.getOpenFileName(
|
self, 'load app file', '.', 'All Files (*);;Bin Files (*.bin)')
|
|
if fileName != '':
|
self.findChild(QtWidgets.QLineEdit, 'le_app').setText(fileName)
|
with open(fileName, 'rb') as fd:
|
self.app_data = fd.read()
|
drv_data_len = len(
|
self.drv_data) if self.drv_data is not None else 0
|
app_data_len = len(
|
self.app_data) if self.app_data is not None else 0
|
total_len = drv_data_len + app_data_len
|
self.findChild(QtWidgets.QProgressBar,
|
'progress_bar').setRange(0, total_len)
|
|
@QtCore.pyqtSlot()
|
def on_btn_start_clicked(self):
|
|
if self.drv_data is None or self.app_data is None:
|
self.append_text_to_log_display('请加载driver文件和app文件')
|
return
|
|
if self.bus is None:
|
self.append_text_to_log_display('请打开设备')
|
return
|
|
if self.in_programming:
|
self.append_text_to_log_display('## 正在编程中')
|
return
|
|
self.in_programming = True
|
loop = asyncio.get_event_loop()
|
loop.create_task(self.start_programming())
|
|
@QtCore.pyqtSlot()
|
def on_btn_open_clicked(self):
|
|
if self.in_programming:
|
self.append_text_to_log_display('## 正在编程中')
|
return
|
|
interface_type_index = self.findChild(
|
QtWidgets.QComboBox, 'cb_interface_type').currentIndex()
|
|
baudrate_index = self.findChild(
|
QtWidgets.QComboBox, 'cb_baudrate').currentIndex()
|
|
can_channel_index = self.findChild(
|
QtWidgets.QComboBox, 'cb_can_channel').currentIndex()
|
|
# bus = VectorBus(channel=1, bitrate=500000, can_filters=can_filters)
|
# bus = CANalystIIBus(channel=0, baud=500000, can_filters=can_filters)
|
|
self.clear_log_display()
|
|
can_filters = [
|
{'can_id': self.rx_id, 'can_mask': 0xFFFFFFFF}
|
]
|
|
bitrate = list(TIMING_DICT.keys())[baudrate_index]
|
|
try:
|
|
if interface_type_index == 0:
|
self.bus = VectorBus(channel=can_channel_index,
|
bitrate=bitrate, can_filters=can_filters)
|
|
elif interface_type_index == 1:
|
self.bus = CANalystIIBus(channel=can_channel_index,
|
baud=bitrate, can_filters=can_filters)
|
|
self.append_text_to_log_display('open device success')
|
self.findChild(QtWidgets.QPushButton,
|
'btn_start').setDisabled(False)
|
|
except Exception as e:
|
self.append_text_to_log_display('%s' % e, '#ff0000')
|
|
@QtCore.pyqtSlot(str)
|
def on_cb_interface_type_currentIndexChanged(self, text):
|
|
if text == 'vector':
|
self.findChild(QtWidgets.QComboBox,
|
'cb_can_channel').setItemText(0, 'CH1/3')
|
self.findChild(QtWidgets.QComboBox,
|
'cb_can_channel').setItemText(1, 'CH2/4')
|
elif text == 'canalystii':
|
self.findChild(QtWidgets.QComboBox,
|
'cb_can_channel').setItemText(0, 'CAN1')
|
self.findChild(QtWidgets.QComboBox,
|
'cb_can_channel').setItemText(1, 'CAN2')
|
|
# @PyQt5.QtCore.pyqtSlot(参数)
|
# def on_发送者对象名称_发射信号名称(self, 参数):
|
# pass
|
