init repo
This commit is contained in:
551
rfid_tester.py
Normal file
551
rfid_tester.py
Normal file
@@ -0,0 +1,551 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
MT6 RFID 读卡器测试程序
|
||||
基于 USB HID 私有协议
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
import time
|
||||
|
||||
# 添加 PySimpleGUI 目录到路径
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), 'PySimpleGUI'))
|
||||
|
||||
import PySimpleGUI as sg
|
||||
|
||||
try:
|
||||
import hid
|
||||
except ImportError:
|
||||
sg.popup_error('请安装 hidapi 库:\npip install hidapi')
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
# ==================== 协议常量 ====================
|
||||
REPORT_ID_SET_REQ = 0x01
|
||||
REPORT_ID_GET_RES = 0x03
|
||||
REPORT_SIZE = 256
|
||||
|
||||
# 设备信息
|
||||
VENDOR_ID = 0xFFFF # 默认VID
|
||||
PRODUCT_ID = 0x0035 # 默认PID
|
||||
|
||||
# ==================== 协议函数 ====================
|
||||
|
||||
def calc_outer_checksum(data_len_bytes: bytes, payload: bytes) -> int:
|
||||
"""计算外层帧校验（XOR）"""
|
||||
result = 0
|
||||
for b in data_len_bytes + payload:
|
||||
result ^= b
|
||||
return result
|
||||
|
||||
|
||||
def build_setreq_frame(data: bytes) -> bytes:
|
||||
"""
|
||||
构建 SetReq 帧
|
||||
:param data: Data 段数据
|
||||
:return: 完整的 256 字节帧
|
||||
"""
|
||||
# Frame Length: 从 0x07 到 Checksum（包含校验和，不含结束符）
|
||||
# = 2 (constant) + 2 (data length) + len(data) + 1 (checksum)
|
||||
frame_length = 5 + len(data)
|
||||
frame_length_bytes = frame_length.to_bytes(2, 'little')
|
||||
|
||||
# Data Length
|
||||
data_length_bytes = len(data).to_bytes(2, 'little')
|
||||
|
||||
# Checksum
|
||||
checksum = calc_outer_checksum(data_length_bytes, data)
|
||||
|
||||
# 构建帧
|
||||
frame = bytearray(REPORT_SIZE)
|
||||
frame[0] = REPORT_ID_SET_REQ
|
||||
frame[1:5] = b'\x00\x00\x00\x00' # Fixed
|
||||
frame[5:7] = frame_length_bytes # Frame Length
|
||||
frame[7:9] = b'\x00\x02' # Constant
|
||||
frame[9:11] = data_length_bytes # Data Length
|
||||
frame[11:11+len(data)] = data # Data
|
||||
frame[11+len(data)] = checksum # Checksum
|
||||
frame[12+len(data)] = 0x03 # End Marker
|
||||
|
||||
return bytes(frame)
|
||||
|
||||
|
||||
def parse_getres_frame(data: bytes) -> tuple:
|
||||
"""
|
||||
解析 GetRes 帧
|
||||
:param data: 接收到的数据
|
||||
:return: (status, response_data) 或 None
|
||||
"""
|
||||
if len(data) < 12:
|
||||
return None
|
||||
|
||||
# 检查 Report ID
|
||||
if data[0] != REPORT_ID_GET_RES:
|
||||
return None
|
||||
|
||||
# 解析帧结构
|
||||
frame_length = int.from_bytes(data[5:7], 'little')
|
||||
data_length = int.from_bytes(data[9:11], 'little')
|
||||
|
||||
# 提取 Data 段
|
||||
response_data = data[11:11+data_length]
|
||||
|
||||
if len(response_data) < 1:
|
||||
return None
|
||||
|
||||
status = response_data[0]
|
||||
return (status, response_data[1:])
|
||||
|
||||
|
||||
# ==================== 命令函数 ====================
|
||||
|
||||
def cmd_get_version() -> bytes:
|
||||
"""读版本号命令"""
|
||||
return bytes([0xc0])
|
||||
|
||||
|
||||
def cmd_buzzer(on: bool) -> bytes:
|
||||
"""蜂鸣器控制命令"""
|
||||
return bytes([0xcd, 0x01 if on else 0x00])
|
||||
|
||||
|
||||
def cmd_set_power(power: int) -> bytes:
|
||||
"""设置功率命令 (0-9)"""
|
||||
power = max(0, min(9, power))
|
||||
return bytes([0xcc, power])
|
||||
|
||||
|
||||
def cmd_set_mode(mode: int) -> bytes:
|
||||
"""设置工作模式命令
|
||||
1: 单标签巡查
|
||||
2: 被动模式
|
||||
3: 多标签巡查
|
||||
"""
|
||||
return bytes([0x0f, mode])
|
||||
|
||||
|
||||
def cmd_read_epc() -> bytes:
|
||||
"""读取 EPC 命令"""
|
||||
return bytes([0xce, 0xbb, 0x00, 0x22, 0x00, 0x00, 0x22, 0x7e])
|
||||
|
||||
|
||||
def cmd_select_card(epc_data: bytes) -> bytes:
|
||||
"""选中卡命令"""
|
||||
# 固定前缀 + EPC 数据
|
||||
payload = bytes([0x01, 0x00, 0x00, 0x00, 0x20, 0x10, 0x00]) + epc_data
|
||||
|
||||
# 计算 checksum
|
||||
card_op_cmd = bytes([0x00, 0x0c])
|
||||
checksum = 0
|
||||
for b in card_op_cmd + payload:
|
||||
checksum = (checksum + b) & 0xFF
|
||||
|
||||
internal_len = len(payload).to_bytes(2, 'little')
|
||||
return bytes([0xce, 0xbb]) + card_op_cmd + internal_len + payload + bytes([checksum, 0x7e])
|
||||
|
||||
|
||||
def cmd_write_epc(old_epc_crc: bytes, new_epc: bytes) -> bytes:
|
||||
"""写入 EPC 命令"""
|
||||
epc_len_indicator = (len(new_epc) * 4) // 4
|
||||
if epc_len_indicator == 0:
|
||||
epc_len_indicator = 8
|
||||
|
||||
word_count = 2 + len(new_epc) // 2
|
||||
|
||||
payload = bytes([0x00, 0x00, 0x00, 0x00, 0x01, 0x00]) # Reserved
|
||||
payload += bytes([0x00, 0x00]) # Reserved
|
||||
payload += bytes([word_count]) # Word Count
|
||||
payload += old_epc_crc # Old EPC CRC
|
||||
payload += bytes([epc_len_indicator, 0x00]) # EPC Len Indicator, Status
|
||||
payload += new_epc # New EPC Data
|
||||
|
||||
# 计算 checksum
|
||||
card_op_cmd = bytes([0x00, 0x49])
|
||||
checksum = 0
|
||||
for b in card_op_cmd + len(payload).to_bytes(2, 'little') + payload:
|
||||
checksum = (checksum + b) & 0xFF
|
||||
|
||||
internal_len = len(payload).to_bytes(2, 'little')
|
||||
return bytes([0xce, 0xbb]) + card_op_cmd + internal_len + payload + bytes([checksum, 0x7e])
|
||||
|
||||
|
||||
# ==================== 设备管理类 ====================
|
||||
|
||||
class RFIDDevice:
|
||||
def __init__(self):
|
||||
self.device = None
|
||||
self.vendor_id = VENDOR_ID
|
||||
self.product_id = PRODUCT_ID
|
||||
|
||||
def list_devices(self):
|
||||
"""列出所有 HID 设备"""
|
||||
devices = []
|
||||
try:
|
||||
for d in hid.enumerate():
|
||||
devices.append({
|
||||
'vendor_id': d['vendor_id'],
|
||||
'product_id': d['product_id'],
|
||||
'manufacturer_string': d.get('manufacturer_string', ''),
|
||||
'product_string': d.get('product_string', ''),
|
||||
'serial_number': d.get('serial_number', ''),
|
||||
'interface_number': d.get('interface_number', -1),
|
||||
'path': d['path']
|
||||
})
|
||||
except Exception as e:
|
||||
print(f"枚举设备失败: {e}")
|
||||
return devices
|
||||
|
||||
def connect(self, vendor_id=None, product_id=None):
|
||||
"""连接设备"""
|
||||
if vendor_id:
|
||||
self.vendor_id = vendor_id
|
||||
if product_id:
|
||||
self.product_id = product_id
|
||||
|
||||
try:
|
||||
self.device = hid.device()
|
||||
self.device.open(self.vendor_id, self.product_id)
|
||||
return True, f"已连接到设备 (VID: 0x{self.vendor_id:04x}, PID: 0x{self.product_id:04x})"
|
||||
except Exception as e:
|
||||
return False, f"连接失败: {e}"
|
||||
|
||||
def disconnect(self):
|
||||
"""断开连接"""
|
||||
if self.device:
|
||||
try:
|
||||
self.device.close()
|
||||
except:
|
||||
pass
|
||||
self.device = None
|
||||
return "已断开连接"
|
||||
|
||||
def is_connected(self):
|
||||
"""检查是否已连接"""
|
||||
return self.device is not None
|
||||
|
||||
def send_command(self, cmd_data: bytes):
|
||||
"""发送命令并接收响应"""
|
||||
if not self.device:
|
||||
return None, "设备未连接"
|
||||
|
||||
try:
|
||||
# 构建并发送 SetReq
|
||||
frame = build_setreq_frame(cmd_data)
|
||||
self.device.send_feature_report(frame)
|
||||
|
||||
# 接收 GetRes
|
||||
time.sleep(0.1)
|
||||
response = self.device.get_feature_report(REPORT_ID_GET_RES, REPORT_SIZE)
|
||||
|
||||
# 解析响应
|
||||
result = parse_getres_frame(bytes(response))
|
||||
if result is None:
|
||||
return None, "响应格式错误"
|
||||
|
||||
status, data = result
|
||||
return data, None
|
||||
except Exception as e:
|
||||
return None, f"通信错误: {e}"
|
||||
|
||||
|
||||
# ==================== GUI 界面 ====================
|
||||
|
||||
def create_window():
|
||||
"""创建主窗口"""
|
||||
sg.theme('LightBlue2')
|
||||
|
||||
# 设备连接区域
|
||||
device_frame = [
|
||||
[sg.Text('VID (十六进制):'), sg.Input('FFFF', size=(8, 1), key='-VID-'),
|
||||
sg.Text('PID (十六进制):'), sg.Input('0035', size=(8, 1), key='-PID-'),
|
||||
sg.Button('枚举设备', key='-ENUM-'),
|
||||
sg.Button('连接', key='-CONNECT-'),
|
||||
sg.Button('断开', key='-DISCONNECT-')],
|
||||
[sg.Text('状态: ', size=(12, 1)), sg.Text('未连接', key='-STATUS-', text_color='red')]
|
||||
]
|
||||
|
||||
# 基本命令区域
|
||||
basic_frame = [
|
||||
[sg.Button('读取版本号', key='-VERSION-', size=(15, 1)),
|
||||
sg.Button('打开蜂鸣器', key='-BUZZER_ON-', size=(15, 1)),
|
||||
sg.Button('关闭蜂鸣器', key='-BUZZER_OFF-', size=(15, 1))],
|
||||
[sg.Text('功率 (0-9):'), sg.Slider(range=(0, 9), default_value=8, orientation='h', size=(20, 15), key='-POWER-'),
|
||||
sg.Button('设置功率', key='-SET_POWER-')],
|
||||
[sg.Text('工作模式:'),
|
||||
sg.Radio('单标签巡查', 'mode', key='-MODE1-', default=True),
|
||||
sg.Radio('被动模式', 'mode', key='-MODE2-'),
|
||||
sg.Radio('多标签巡查', 'mode', key='-MODE3-'),
|
||||
sg.Button('设置模式', key='-SET_MODE-')]
|
||||
]
|
||||
|
||||
# EPC 操作区域
|
||||
epc_frame = [
|
||||
[sg.Button('读取 EPC', key='-READ_EPC-', size=(15, 1)),
|
||||
sg.Button('选中卡', key='-SELECT_CARD-', size=(15, 1))],
|
||||
[sg.Text('EPC 数据 (十六进制):'), sg.Input('', size=(40, 1), key='-EPC_DATA-')],
|
||||
[sg.Text('新 EPC (十六进制):'), sg.Input('', size=(40, 1), key='-NEW_EPC-')],
|
||||
[sg.Button('写入 EPC', key='-WRITE_EPC-', size=(15, 1))]
|
||||
]
|
||||
|
||||
# 日志输出区域
|
||||
log_frame = [
|
||||
[sg.Multiline('', size=(80, 15), key='-LOG-', autoscroll=True, disabled=True)]
|
||||
]
|
||||
|
||||
layout = [
|
||||
[sg.Frame('设备连接', device_frame)],
|
||||
[sg.Frame('基本命令', basic_frame)],
|
||||
[sg.Frame('EPC 操作', epc_frame)],
|
||||
[sg.Frame('日志', log_frame)],
|
||||
[sg.Button('清空日志', key='-CLEAR_LOG-'), sg.Button('退出', key='-EXIT-')]
|
||||
]
|
||||
|
||||
return sg.Window('MT6 RFID 读卡器测试程序', layout, finalize=True)
|
||||
|
||||
|
||||
def log_message(window, message, is_hex=False):
|
||||
"""输出日志消息"""
|
||||
timestamp = time.strftime('%H:%M:%S')
|
||||
if is_hex:
|
||||
if isinstance(message, bytes):
|
||||
hex_str = ' '.join(f'{b:02x}' for b in message)
|
||||
window['-LOG-'].print(f'[{timestamp}] {hex_str}')
|
||||
else:
|
||||
window['-LOG-'].print(f'[{timestamp}] {message}')
|
||||
else:
|
||||
window['-LOG-'].print(f'[{timestamp}] {message}')
|
||||
|
||||
|
||||
def hex_to_bytes(hex_str: str) -> bytes:
|
||||
"""十六进制字符串转字节"""
|
||||
hex_str = hex_str.replace(' ', '').replace(',', '')
|
||||
return bytes.fromhex(hex_str)
|
||||
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
window = create_window()
|
||||
rfid = RFIDDevice()
|
||||
device_list = []
|
||||
|
||||
while True:
|
||||
event, values = window.read()
|
||||
|
||||
if event in (sg.WIN_CLOSED, '-EXIT-'):
|
||||
rfid.disconnect()
|
||||
break
|
||||
|
||||
# 枚举设备
|
||||
if event == '-ENUM-':
|
||||
device_list = rfid.list_devices()
|
||||
if device_list:
|
||||
log_message(window, f"找到 {len(device_list)} 个 HID 设备")
|
||||
device_strs = []
|
||||
for i, d in enumerate(device_list):
|
||||
s = f"[{i}] VID:0x{d['vendor_id']:04x} PID:0x{d['product_id']:04x} " \
|
||||
f"IF:{d['interface_number']} {d['manufacturer_string']} {d['product_string']}"
|
||||
device_strs.append(s)
|
||||
log_message(window, s)
|
||||
|
||||
# 创建设备选择窗口
|
||||
select_layout = [
|
||||
[sg.Listbox(values=device_strs, size=(80, min(10, len(device_strs))), key='-SELECTED-')],
|
||||
[sg.Button('使用选中设备'), sg.Button('取消')]
|
||||
]
|
||||
select_window = sg.Window('选择设备', select_layout, modal=True)
|
||||
while True:
|
||||
sel_event, sel_values = select_window.read()
|
||||
if sel_event in (sg.WIN_CLOSED, '取消'):
|
||||
select_window.close()
|
||||
break
|
||||
if sel_event == '使用选中设备':
|
||||
selected_idx = sel_values.get('-SELECTED-', [])
|
||||
if selected_idx:
|
||||
idx = device_strs.index(selected_idx[0])
|
||||
d = device_list[idx]
|
||||
window['-VID-'].update(f"{d['vendor_id']:04x}")
|
||||
window['-PID-'].update(f"{d['product_id']:04x}")
|
||||
log_message(window, f"已选择设备: VID=0x{d['vendor_id']:04x}, PID=0x{d['product_id']:04x}, IF={d['interface_number']}")
|
||||
select_window.close()
|
||||
break
|
||||
else:
|
||||
log_message(window, "未找到 HID 设备")
|
||||
|
||||
# 连接设备
|
||||
if event == '-CONNECT-':
|
||||
try:
|
||||
vid = int(values['-VID-'], 16)
|
||||
pid = int(values['-PID-'], 16)
|
||||
success, msg = rfid.connect(vid, pid)
|
||||
log_message(window, msg)
|
||||
if success:
|
||||
window['-STATUS-'].update('已连接', text_color='green')
|
||||
else:
|
||||
window['-STATUS-'].update('连接失败', text_color='red')
|
||||
except ValueError:
|
||||
log_message(window, "请输入有效的十六进制 VID/PID")
|
||||
|
||||
# 断开连接
|
||||
if event == '-DISCONNECT-':
|
||||
msg = rfid.disconnect()
|
||||
log_message(window, msg)
|
||||
window['-STATUS-'].update('未连接', text_color='red')
|
||||
|
||||
# 读取版本号
|
||||
if event == '-VERSION-':
|
||||
if not rfid.is_connected():
|
||||
log_message(window, "错误: 设备未连接")
|
||||
else:
|
||||
log_message(window, "发送命令: 读版本号")
|
||||
data, err = rfid.send_command(cmd_get_version())
|
||||
if err:
|
||||
log_message(window, f"错误: {err}")
|
||||
else:
|
||||
log_message(window, f"响应 (hex): ", is_hex=True)
|
||||
log_message(window, data, is_hex=True)
|
||||
try:
|
||||
version = data.decode('ascii', errors='ignore')
|
||||
log_message(window, f"版本: {version}")
|
||||
except:
|
||||
pass
|
||||
|
||||
# 蜂鸣器控制
|
||||
if event == '-BUZZER_ON-':
|
||||
if not rfid.is_connected():
|
||||
log_message(window, "错误: 设备未连接")
|
||||
else:
|
||||
log_message(window, "发送命令: 打开蜂鸣器")
|
||||
data, err = rfid.send_command(cmd_buzzer(True))
|
||||
if err:
|
||||
log_message(window, f"错误: {err}")
|
||||
else:
|
||||
log_message(window, f"响应: {data.hex() if data else '无数据'}")
|
||||
|
||||
if event == '-BUZZER_OFF-':
|
||||
if not rfid.is_connected():
|
||||
log_message(window, "错误: 设备未连接")
|
||||
else:
|
||||
log_message(window, "发送命令: 关闭蜂鸣器")
|
||||
data, err = rfid.send_command(cmd_buzzer(False))
|
||||
if err:
|
||||
log_message(window, f"错误: {err}")
|
||||
else:
|
||||
log_message(window, f"响应: {data.hex() if data else '无数据'}")
|
||||
|
||||
# 设置功率
|
||||
if event == '-SET_POWER-':
|
||||
if not rfid.is_connected():
|
||||
log_message(window, "错误: 设备未连接")
|
||||
else:
|
||||
power = int(values['-POWER-'])
|
||||
log_message(window, f"发送命令: 设置功率 = {power}")
|
||||
data, err = rfid.send_command(cmd_set_power(power))
|
||||
if err:
|
||||
log_message(window, f"错误: {err}")
|
||||
else:
|
||||
log_message(window, f"响应: {data.hex() if data else '无数据'}")
|
||||
|
||||
# 设置工作模式
|
||||
if event == '-SET_MODE-':
|
||||
if not rfid.is_connected():
|
||||
log_message(window, "错误: 设备未连接")
|
||||
else:
|
||||
mode = 1
|
||||
if values['-MODE2-']:
|
||||
mode = 2
|
||||
elif values['-MODE3-']:
|
||||
mode = 3
|
||||
mode_names = {1: '单标签巡查', 2: '被动模式', 3: '多标签巡查'}
|
||||
log_message(window, f"发送命令: 设置模式 = {mode_names.get(mode, mode)}")
|
||||
data, err = rfid.send_command(cmd_set_mode(mode))
|
||||
if err:
|
||||
log_message(window, f"错误: {err}")
|
||||
else:
|
||||
log_message(window, f"响应: {data.hex() if data else '无数据'}")
|
||||
|
||||
# 读取 EPC
|
||||
if event == '-READ_EPC-':
|
||||
if not rfid.is_connected():
|
||||
log_message(window, "错误: 设备未连接")
|
||||
else:
|
||||
log_message(window, "发送命令: 读取 EPC")
|
||||
data, err = rfid.send_command(cmd_read_epc())
|
||||
if err:
|
||||
log_message(window, f"错误: {err}")
|
||||
else:
|
||||
log_message(window, f"响应 (hex): ", is_hex=True)
|
||||
log_message(window, data, is_hex=True)
|
||||
# 解析 EPC
|
||||
if data and len(data) > 10:
|
||||
# 检查 Card Op Resp
|
||||
if data[2:4] == b'\x02\x22':
|
||||
# 有卡
|
||||
rssi = data[6] if len(data) > 6 else 0
|
||||
epc_len_indicator = data[7] if len(data) > 7 else 0
|
||||
epc_len = epc_len_indicator // 4 if epc_len_indicator > 0 else 0
|
||||
epc_data = data[9:9+epc_len] if len(data) > 9 else b''
|
||||
log_message(window, f"RSSI: 0x{rssi:02x}")
|
||||
log_message(window, f"EPC: {epc_data.hex()}")
|
||||
window['-EPC_DATA-'].update(epc_data.hex())
|
||||
elif data[2:4] == b'\x01\xff':
|
||||
log_message(window, "无卡或读取失败")
|
||||
else:
|
||||
log_message(window, f"未知响应: {data[2:4].hex()}")
|
||||
|
||||
# 选中卡
|
||||
if event == '-SELECT_CARD-':
|
||||
if not rfid.is_connected():
|
||||
log_message(window, "错误: 设备未连接")
|
||||
else:
|
||||
epc_hex = values['-EPC_DATA-'].strip()
|
||||
if not epc_hex:
|
||||
log_message(window, "错误: 请输入 EPC 数据")
|
||||
else:
|
||||
try:
|
||||
epc_data = hex_to_bytes(epc_hex)
|
||||
log_message(window, f"发送命令: 选中卡 EPC={epc_data.hex()}")
|
||||
data, err = rfid.send_command(cmd_select_card(epc_data))
|
||||
if err:
|
||||
log_message(window, f"错误: {err}")
|
||||
else:
|
||||
log_message(window, f"响应 (hex): {data.hex() if data else '无数据'}")
|
||||
if data and len(data) > 6:
|
||||
if data[6] == 0x00:
|
||||
log_message(window, "选中成功")
|
||||
else:
|
||||
log_message(window, f"选中失败: 0x{data[6]:02x}")
|
||||
except ValueError:
|
||||
log_message(window, "错误: EPC 数据格式错误，请使用十六进制")
|
||||
|
||||
# 写入 EPC
|
||||
if event == '-WRITE_EPC-':
|
||||
if not rfid.is_connected():
|
||||
log_message(window, "错误: 设备未连接")
|
||||
else:
|
||||
new_epc_hex = values['-NEW_EPC-'].strip()
|
||||
if not new_epc_hex:
|
||||
log_message(window, "错误: 请输入新的 EPC 数据")
|
||||
else:
|
||||
try:
|
||||
new_epc = hex_to_bytes(new_epc_hex)
|
||||
# 这里需要一个旧的 EPC CRC，暂时使用示例值
|
||||
old_epc_crc = bytes([0xca, 0x9e]) # 需要根据实际 EPC 计算
|
||||
log_message(window, f"发送命令: 写入 EPC={new_epc.hex()}")
|
||||
data, err = rfid.send_command(cmd_write_epc(old_epc_crc, new_epc))
|
||||
if err:
|
||||
log_message(window, f"错误: {err}")
|
||||
else:
|
||||
log_message(window, f"响应 (hex): {data.hex() if data else '无数据'}")
|
||||
except ValueError:
|
||||
log_message(window, "错误: EPC 数据格式错误，请使用十六进制")
|
||||
|
||||
# 清空日志
|
||||
if event == '-CLEAR_LOG-':
|
||||
window['-LOG-'].update('')
|
||||
|
||||
window.close()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user