# -*- coding: utf-8 -*-
import pandas as pd
from binascii import unhexlify, hexlify

# ----------------------------
# 配置：已知 keystream 前缀
# ----------------------------
ks_name_hex = "2663077431161e64ab"  # 9 bytes
ks_phone_hex = "2663077431161e64ab9755"  # 11 bytes
ks_id_hex = "2663077431161e64ab975542fdee50499abd"  # 18 bytes

ks_name = unhexlify(ks_name_hex)
ks_phone = unhexlify(ks_phone_hex)
ks_id = unhexlify(ks_id_hex)


# ----------------------------
# 辅助函数
# ----------------------------
def xor(a, b):
    """字节异或"""
    return bytes(x ^ y for x, y in zip(a, b))


# ----------------------------
# 读取 Excel 文件
# ----------------------------
input_file = "1.xlsx"  # 输入文件
output_file = "1_decrypted.xlsx"  # 输出文件

df_input = pd.read_excel(input_file)

decrypted_rows = []
for idx, row in df_input.iterrows():
    nctb = unhexlify(row['姓名'])
    pctb = unhexlify(row['手机号'])
    ictb = unhexlify(row['身份证号'])

    # 解密前缀
    name_bytes = xor(nctb[:len(ks_name)], ks_name)
    phone_bytes = xor(pctb[:len(ks_phone)], ks_phone)
    id_bytes = xor(ictb[:len(ks_id)], ks_id)

    # 尝试解码
    try:
        name_plain = name_bytes.decode('utf-8')
    except:
        name_plain = name_bytes.decode('utf-8', errors='replace')
    phone_plain = phone_bytes.decode('utf-8', errors='replace')
    id_plain = id_bytes.decode('utf-8', errors='replace')

    decrypted_rows.append({
        '序号': row['序号'],
        '解出_姓名(前9字节)': name_plain,
        '解出_手机号(前11字节)': phone_plain,
        '解出_身份证(前18字节)': id_plain,
        'name_hex': hexlify(name_bytes).decode(),
        'phone_hex': hexlify(phone_bytes).decode(),
        'id_hex': hexlify(id_bytes).decode()
    })

# ----------------------------
# 保存结果
# ----------------------------
df_decrypted = pd.DataFrame(decrypted_rows)
df_decrypted.to_excel(output_file, index=False)
print(f"解密完成，结果已保存到 {output_file}")