package com.cscn;

import javacard.framework.APDU;
import javacard.framework.ISO7816;
import javacard.framework.ISOException;
import javacard.framework.JCSystem;
import javacard.framework.Util;

/**
 * 仅做一次“自检调用”：
 * 1) ctx.init(KEY, IV); Enc(Input) == EncExpected ?
 * 2) ctx.init(KEY, IV); Enc(Enc(Input)) == Input ?
 * Response: 2字节 [encMatch, dblEncRestored]，1=真，0=假
 */
public final class Method {

	// ======= 请用你的向量替换下面占位内容 =======
	// Key: 32字节
	private static final byte[] KEY32 = new byte[] {
			// TODO: 替换成你的32字节Key
			(byte)0x00,(byte)0x01,(byte)0x02,(byte)0x03,(byte)0x04,(byte)0x05,(byte)0x06,(byte)0x07,
			(byte)0x08,(byte)0x09,(byte)0x0A,(byte)0x0B,(byte)0x0C,(byte)0x0D,(byte)0x0E,(byte)0x0F,
			(byte)0x10,(byte)0x11,(byte)0x12,(byte)0x13,(byte)0x14,(byte)0x15,(byte)0x16,(byte)0x17,
			(byte)0x18,(byte)0x19,(byte)0x1A,(byte)0x1B,(byte)0x1C,(byte)0x1D,(byte)0x1E,(byte)0x1F
	};

	// IV: 25字节（ZUC-256）
	private static final byte[] IV25 = new byte[] {
			// TODO: 替换成你的25字节IV
			(byte)0xA0,(byte)0xA1,(byte)0xA2,(byte)0xA3,(byte)0xA4,(byte)0xA5,(byte)0xA6,(byte)0xA7,
			(byte)0xA8,(byte)0xA9,(byte)0xAA,(byte)0xAB,(byte)0xAC,(byte)0xAD,(byte)0xAE,(byte)0xAF,
			(byte)0xB0,(byte)0xB1,(byte)0xB2,(byte)0xB3,(byte)0xB4,(byte)0xB5,(byte)0xB6,(byte)0xB7,
			(byte)0xB8
	};

	// Input: 建议长度为 4 的倍数，便于不走 finish 的残字节路径
	private static final byte[] INPUT = new byte[] {
			// TODO: 替换成你的明文（示例32字节）
			(byte)0x11,(byte)0x22,(byte)0x33,(byte)0x44,(byte)0x55,(byte)0x66,(byte)0x77,(byte)0x88,
			(byte)0x99,(byte)0xAA,(byte)0xBB,(byte)0xCC,(byte)0xDD,(byte)0xEE,(byte)0xFF,(byte)0x00,
			(byte)0x10,(byte)0x20,(byte)0x30,(byte)0x40,(byte)0x50,(byte)0x60,(byte)0x70,(byte)0x80,
			(byte)0x90,(byte)0xA0,(byte)0xB0,(byte)0xC0,(byte)0xD0,(byte)0xE0,(byte)0xF0,(byte)0x00
	};

	// EncResult: 期望密文（与上面的 Input 配套）
	private static final byte[] ENC_EXPECTED = new byte[] {
			// TODO: 替换成你“已知正确”的密文（示例随便填；不改会导致第一项校验=0）
			(byte)0x21,(byte)0x32,(byte)0x43,(byte)0x54,(byte)0x65,(byte)0x76,(byte)0x87,(byte)0x98,
			(byte)0xA9,(byte)0xBA,(byte)0xCB,(byte)0xDC,(byte)0xED,(byte)0xFE,(byte)0x0F,(byte)0x1E,
			(byte)0x2D,(byte)0x3C,(byte)0x4B,(byte)0x5A,(byte)0x69,(byte)0x78,(byte)0x87,(byte)0x96,
			(byte)0xA5,(byte)0xB4,(byte)0xC3,(byte)0xD2,(byte)0xE1,(byte)0xF0,(byte)0x0F,(byte)0x00
	};
	// ======================================

	// 运行时缓冲：放RAM，避免写EEPROM
	private final byte[] buf1; // Enc(Input)
	private final byte[] buf2; // Enc(Enc(Input)) -> 应为 Input
	private final Zuc256EncryptCtx ctx;

	public Method() {
		short L = (short) INPUT.length;
		buf1 = JCSystem.makeTransientByteArray(L, JCSystem.CLEAR_ON_DESELECT);
		buf2 = JCSystem.makeTransientByteArray(L, JCSystem.CLEAR_ON_DESELECT);
		ctx  = new Zuc256EncryptCtx(); // 仅创建一次
	}

	public void processData(APDU apdu) {
		byte[] apduBuf = apdu.getBuffer();

		// 第一次：Enc(Input)
		ctx.init(KEY32, IV25);
		ctx.update(INPUT, (short) INPUT.length, buf1);
		ctx.finish(buf1); // 若 Input 长度为 4 的倍数则通常无副作用，留着更稳妥

		boolean encMatch = (Util.arrayCompare(buf1, (short)0, ENC_EXPECTED, (short)0, (short)INPUT.length) == 0);

		// 第二次：Enc(Enc(Input)) 应还原 Input
		ctx.init(KEY32, IV25);
		ctx.update(buf1, (short) INPUT.length, buf2);
		ctx.finish(buf2);
		boolean dblOk = (Util.arrayCompare(buf2, (short)0, INPUT, (short)0, (short)INPUT.length) == 0);

		// 返回 2 字节结果：[encMatch, dblOk]，1=真, 0=假
		apduBuf[0] = (byte)(encMatch ? 1 : 0);
		apduBuf[1] = (byte)(dblOk    ? 1 : 0);
		short outLen = 2;

		apdu.setOutgoing();
		apdu.setOutgoingLength(outLen);
		apdu.sendBytes((short)0, outLen);
	}

	// 本任务不需要更新Key，这里留空
	public void updateKey(APDU apdu) {
		ISOException.throwIt(ISO7816.SW_FUNC_NOT_SUPPORTED);
	}
}