56 lines
2.4 KiB
Python
56 lines
2.4 KiB
Python
"""激活 keygen 测试 — 向量按 docs/firmware-analysis.md §4.2 手工推算。"""
|
||
import pytest
|
||
|
||
from ppclock import protocol as P
|
||
|
||
|
||
class TestKeygen:
|
||
def test_keygen_vector(self):
|
||
"""MAC 00:11:22:33:44:55 → [49,14,32,98,15,81]（ror2(((m<<1)&FF)^EF)%101 手算）"""
|
||
code = P.keygen_code(bytes([0x00, 0x11, 0x22, 0x33, 0x44, 0x55]))
|
||
assert list(code) == [49, 14, 32, 98, 15, 81]
|
||
|
||
def test_keygen_all_zero_mac(self):
|
||
# t = 0^0xEF = 0xEF; ror2(0xEF)=0xFB=251; 251%101=49
|
||
assert list(P.keygen_code(bytes(6))) == [49] * 6
|
||
|
||
def test_keygen_all_ff_mac(self):
|
||
# t = (0x1FE & 0xFF)^0xEF = 0xFE^0xEF = 0x11; rol2(0x11)=0x44=68; 68%101=68
|
||
assert list(P.keygen_code(b"\xFF" * 6)) == [68] * 6
|
||
|
||
def test_keygen_requires_6_bytes(self):
|
||
with pytest.raises(ValueError):
|
||
P.keygen_code(b"\x01" * 5)
|
||
|
||
|
||
class TestRecoverMac:
|
||
def test_recover_real_device(self):
|
||
"""实机向量：ID '81233F3C267112' ↔ 广播 18:BC:5A:5D:BF:28（NVDS 序=显示序反转）"""
|
||
mac = P.recover_mac_from_device_id("81233F3C267112")
|
||
assert mac == bytes([0x28, 0xBF, 0x5D, 0x5A, 0xBC, 0x18])
|
||
assert mac[::-1] == bytes.fromhex("18BC5A5DBF28")
|
||
|
||
def test_recover_vector(self):
|
||
"""MAC [9A,01,99,50,10,20] → ID '79894900980101'（|MAC5-99|,|MAC4-99|,|MAC3-99|,
|
||
|MAC2-99|,|MAC1-99|,|MAC0-99|,flags；flags=各字节≥0x9A 位掩码=0x01）"""
|
||
mac = P.recover_mac_from_device_id("79894900980101")
|
||
assert list(mac) == [0x9A, 0x01, 0x99, 0x50, 0x10, 0x20]
|
||
|
||
def test_roundtrip_all_high(self):
|
||
# 全 ≥0x9A：flags=0x3F；v=|m-0x99|
|
||
mac = bytes([0x9A, 0xB0, 0xC1, 0xFF, 0xA0, 0x9B])
|
||
vals = [abs(mac[5] - 0x99), abs(mac[4] - 0x99), abs(mac[3] - 0x99),
|
||
abs(mac[2] - 0x99), abs(mac[1] - 0x99), abs(mac[0] - 0x99)]
|
||
dev_id = "".join(f"{v:02X}" for v in vals) + "3F"
|
||
assert P.recover_mac_from_device_id(dev_id) == mac
|
||
|
||
def test_id_must_be_14_hex(self):
|
||
with pytest.raises(ValueError):
|
||
P.recover_mac_from_device_id("123")
|
||
|
||
def test_keygen_after_recover_matches(self):
|
||
"""recovered MAC → keygen：与直接对原 MAC keygen 一致"""
|
||
mac = bytes([0x28, 0xBF, 0x5D, 0x5A, 0xBC, 0x18])
|
||
recovered = P.recover_mac_from_device_id("81233F3C267112")
|
||
assert P.keygen_code(recovered) == P.keygen_code(mac)
|