71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
tlscan2pcileech_coe.py
|
||
将 TLScan 的 .tlscan 文件转换成 PCILeech 所需的 COE 格式
|
||
用法：
|
||
python tlscan2pcileech_coe.py input.tlscan
|
||
结果：
|
||
同目录下生成 input_rw1.coe 和 input_rw1c.coe
|
||
"""
|
||
import os
|
||
import sys
|
||
import datetime
|
||
import xml.etree.ElementTree as ET
|
||
|
||
# ---------- 参数 ----------
|
||
if len(sys.argv) < 2:
|
||
print("用法: python tlscan2pcileech_coe.py <xxx.tlscan>")
|
||
sys.exit(1)
|
||
|
||
src = sys.argv[1]
|
||
base, _ = os.path.splitext(src)
|
||
dst_rw1 = base + "_rw1.coe"
|
||
dst_rw1c = base + "_rw1c.coe"
|
||
|
||
# ---------- 读取 TLScan ----------
|
||
try:
|
||
bs_hex = ET.parse(src).find('.//bytes').text
|
||
except Exception as e:
|
||
sys.exit(f"解析 TLScan 文件失败: {e}")
|
||
|
||
bs_hex = ''.join(bs_hex.split())
|
||
if len(bs_hex) != 8192: # 4096 字节 => 8192 字符
|
||
sys.exit(f"期望 8192 个十六进制字符，实际 {len(bs_hex)}")
|
||
|
||
# ---------- 生成掩码 ----------
|
||
# 4096 字节 => 1024 个 dword
|
||
words_rw1 = [0] * 1024
|
||
words_rw1c = [0] * 1024
|
||
|
||
for i in range(4096):
|
||
val = int(bs_hex[i*2:i*2+2], 16)
|
||
# 简易规则：0x01 视为 RW1C/RW1CS
|
||
if val != 0x00:
|
||
words_rw1[i//4] |= 1 << ((i % 4) * 8)
|
||
if val == 0x01:
|
||
words_rw1c[i//4] |= 1 << ((i % 4) * 8)
|
||
|
||
# ---------- 写 COE ----------
|
||
def write_coe(path, words):
|
||
with open(path, 'w') as f:
|
||
f.write(f'; Converted to COE from "{src}" on {datetime.datetime.now()}\n')
|
||
f.write('memory_initialization_radix=16;\nmemory_initialization_vector=\n')
|
||
|
||
for blk in range(16):
|
||
offset = blk * 256
|
||
f.write(f'; {offset:04x}\n')
|
||
for line in range(16):
|
||
idx = blk * 64 + line * 4
|
||
line_words = [words[idx + j] for j in range(4)]
|
||
f.write(','.join(f'{w:08x}' for w in line_words) + ',\n')
|
||
|
||
# 去掉最后一个逗号
|
||
f.seek(f.tell() - 2, os.SEEK_SET) # 回退 2 字节（,\n）
|
||
f.truncate()
|
||
f.write('\n;\n')
|
||
|
||
write_coe(dst_rw1, words_rw1)
|
||
write_coe(dst_rw1c, words_rw1c)
|
||
|
||
print(f'已生成:\n {dst_rw1}\n {dst_rw1c}') |