qmc_test.go

package main

import (
	"bytes"
	"os"
	"testing"
)

// TestKuwoEkeyDecrypt 验证酷我 ekey 的 DES-ECB 解密 + raw key 提取 + QMC 密钥派生，
// 与 nonebot-plugin-kuwo (src/qmc.rs) 的 Python 实现逐字节对比。
func TestKuwoEkeyDecrypt(t *testing.T) {
	ekeyBytes, err := os.ReadFile("testdata/test_ekey.txt")
	if err != nil {
		t.Skipf("跳过：未找到 testdata/test_ekey.txt: %v", err)
	}
	ekey := string(bytes.TrimSpace(ekeyBytes))

	// 验证 kuwoBase64Decrypt（DES-ECB 解密 + 去尾部零）
	decrypted, err := kuwoBase64Decrypt(ekey)
	if err != nil {
		t.Fatalf("kuwoBase64Decrypt 失败: %v", err)
	}
	if expected, eerr := os.ReadFile("testdata/expected_kuwo_b64dec.bin"); eerr == nil {
		if !bytes.Equal(decrypted, expected) {
			t.Errorf("kuwoBase64Decrypt 输出不匹配: got %d 字节, want %d 字节", len(decrypted), len(expected))
		}
	}

	// 验证 extractQmcRawKey（从尾部提取 704/364 字节 base64 子串）
	rawKey, err := extractQmcRawKey(ekey)
	if err != nil {
		t.Fatalf("extractQmcRawKey 失败: %v", err)
	}
	if expected, eerr := os.ReadFile("testdata/expected_raw_key.bin"); eerr == nil {
		if !bytes.Equal(rawKey, expected) {
			t.Errorf("rawKey 不匹配: got %d 字节, want %d 字节", len(rawKey), len(expected))
		}
	}

	// 验证 deriveKey（QMC 密钥派生：base64 解码 → V2 前缀检查 → V1 派生）
	qmcKey, err := deriveKey(rawKey)
	if err != nil {
		t.Fatalf("deriveKey 失败: %v", err)
	}
	if expected, eerr := os.ReadFile("testdata/expected_qmc_key.bin"); eerr == nil {
		if !bytes.Equal(qmcKey, expected) {
			t.Errorf("qmcKey 不匹配: got %d 字节, want %d 字节", len(qmcKey), len(expected))
		}
	}

	t.Logf("解密链验证通过: ekey(%d) → rawKey(%d) → qmcKey(%d)", len(ekey), len(rawKey), len(qmcKey))
}