identity(without persist & without test)

src/test/java/minsecurity/crypto/TestSM2.java

@@ -0,0 +1,135 @@
+package minsecurity.crypto;
+
+import minsecurity.crypto.sm2.SM2Base;
+import minsecurity.crypto.sm2.SM2PrivateKey;
+import minsecurity.crypto.sm2.SM2PublicKey;
+import org.bouncycastle.crypto.AsymmetricCipherKeyPair;
+import org.bouncycastle.crypto.CryptoException;
+import org.bouncycastle.crypto.InvalidCipherTextException;
+import org.bouncycastle.crypto.params.ECPrivateKeyParameters;
+import org.bouncycastle.crypto.params.ECPublicKeyParameters;
+import org.bouncycastle.pqc.math.linearalgebra.ByteUtils;
+import org.junit.Test;
+import static org.junit.Assert.*;
+import org.slf4j.LoggerFactory;
+
+import java.nio.ByteBuffer;
+import java.util.ArrayList;
+import java.util.List;
+
+/*
+ * @Author: hongyu guo
+ * @Description: 测试Sm2方法
+ * @Version: 1.0.0
+ * @Date: 20:15 2021/03/03
+ * @Copyright: MIN-Group；国家重大科技基础设施——未来网络北大实验室；深圳市信息论与未来网络重点实验室
+ */
+public class TestSM2 {
+    private static final org.slf4j.Logger logger = LoggerFactory.getLogger(TestSM2.class);
+    @Test
+    public void testSetGet() throws InvalidCipherTextException {
+        AsymmetricCipherKeyPair keyPair = SM2Base.generateKeyPairParameter();
+        ECPrivateKeyParameters priKey = (ECPrivateKeyParameters) keyPair.getPrivate();
+        ECPublicKeyParameters pubKey = (ECPublicKeyParameters) keyPair.getPublic();
+        byte[] d = priKey.getD().toByteArray();
+        byte[] x = pubKey.getQ().getAffineXCoord().getEncoded();
+        byte[] y = pubKey.getQ().getAffineYCoord().getEncoded();
+        logger.debug("d.len = {}, x.len = {}, y.len = {}",d.length, x.length, y.length);
+        // BigInteger bigInteger = priKey.getD();
+        SM2PrivateKey sm2PrivateKey = new SM2PrivateKey();
+        SM2PublicKey sm2PublicKey = new SM2PublicKey();
+
+        sm2PrivateKey.setBytes(d);
+
+        ByteBuffer buffer = ByteBuffer.allocate(65);
+        buffer.put((byte)4);
+        buffer.put(x);
+        buffer.put(y);
+        sm2PublicKey.setBytes(buffer.array());
+
+        assertArrayEquals(d, sm2PrivateKey.getBytes());
+        assertArrayEquals(buffer.array(), sm2PublicKey.getBytes());
+
+        String content = "hello world";
+        byte[] encBytes = sm2PublicKey.encrypt(content.getBytes());
+        byte[] decBytes = sm2PrivateKey.decrypt(encBytes);
+        // logger.debug(new String(decBytes, StandardCharsets.UTF_8));
+        assertArrayEquals(content.getBytes(), decBytes);
+
+
+    }
+
+
+    @Test
+    public void testEncAndDec() throws InvalidCipherTextException {
+        AsymmetricCipherKeyPair keyPair = SM2Base.generateKeyPairParameter();
+        ECPrivateKeyParameters priKey = (ECPrivateKeyParameters) keyPair.getPrivate();
+        ECPublicKeyParameters pubKey = (ECPublicKeyParameters) keyPair.getPublic();
+        byte[] d = priKey.getD().toByteArray();
+        // d = Arrays.copyOf(d,32);
+        byte[] x = pubKey.getQ().getAffineXCoord().getEncoded();
+        byte[] y = pubKey.getQ().getAffineYCoord().getEncoded();
+        logger.debug("d.len = {}, x.len = {}, y.len = {}",d.length, x.length, y.length);
+        // BigInteger bigInteger = priKey.getD();
+        SM2PrivateKey sm2PrivateKey = new SM2PrivateKey(d);
+        SM2PublicKey sm2PublicKey = new SM2PublicKey(x,y);
+
+        String content = "hello world";
+        byte[] encBytes = sm2PublicKey.encrypt(content.getBytes());
+        byte[] decBytes = sm2PrivateKey.decrypt(encBytes);
+        // logger.debug(new String(decBytes, StandardCharsets.UTF_8));
+        assertArrayEquals(content.getBytes(), decBytes);
+    }
+
+    @Test
+    public void testSignAndVerify() throws CryptoException {
+        AsymmetricCipherKeyPair keyPair = SM2Base.generateKeyPairParameter();
+        ECPrivateKeyParameters priKey = (ECPrivateKeyParameters) keyPair.getPrivate();
+        ECPublicKeyParameters pubKey = (ECPublicKeyParameters) keyPair.getPublic();
+        byte[] d = priKey.getD().toByteArray();
+        // d = Arrays.copyOf(d,32);
+        byte[] x = pubKey.getQ().getAffineXCoord().getEncoded();
+        byte[] y = pubKey.getQ().getAffineYCoord().getEncoded();
+        logger.debug("d.len = {}, x.len = {}, y.len = {}",d.length, x.length, y.length);
+
+        // BigInteger bigInteger = priKey.getD();
+        SM2PrivateKey sm2PrivateKey = new SM2PrivateKey(d);
+        SM2PublicKey sm2PublicKey = new SM2PublicKey(x,y);
+        String content = "7b227462734365727469666963617465223a7b2276657273696f6e223a312c2273657269616c4e756d626572223a312c227075626c69634b6579223a22424a304d7047714e5958706b695879575a4d6e5838776d6b694a51326f576d37336257615932525473706b695058342f37705443666332764e7272596e766c6a644d3448647850537531753272616363506c74485245593d222c227369676e6174757265416c676f726974686d223a302c227075626c69634b6579416c676f726974686d223a302c226973737565546f223a226973737565546f222c22697373756572223a22697373756572222c226e6f744265666f7265223a313631353032303130393339352c226e6f744166746572223a313631353032303130393539352c226b65795573616765223a322c2274696d657374616d70223a313631353032303130393339352c226361223a747275657d2c227369676e6174757265416c676f726974686d223a302c227369676e617475726556616c7565223a6e756c6c7d";
+        byte[] digest = sm2PrivateKey.sign(ByteUtils.fromHexString(content));
+        boolean flag = sm2PublicKey.verify(ByteUtils.fromHexString(content), digest);
+        assertTrue(flag);
+    }
+
+    // 按照艾宾浩斯记忆曲线
+    // 第x天刷完题, 需要在第x+1,x+2,x+4,x+7,x+15时重新记忆
+    public static void testMemory(int total, int numsInOneDay){
+        int days = total / numsInOneDay + 1;
+        List<Integer>[] res = new List[days + 15];
+        for(int i = 0; i < days + 15; i++){
+            res[i] = new ArrayList<Integer>();
+        }
+
+        for(int i = 0; i < days; i++){
+
+            for(int j = 1; j <= numsInOneDay; j++){
+                if(i * numsInOneDay + j <= total) {
+                    res[i].add((i * numsInOneDay + j) * -1);
+                    res[i + 1].add(i * numsInOneDay + j);
+                    res[i + 2].add(i * numsInOneDay + j);
+                    res[i + 4].add(i * numsInOneDay + j);
+                    res[i + 7].add(i * numsInOneDay + j);
+                    res[i + 15].add(i * numsInOneDay + j);
+                }
+            }
+
+        }
+        for(int i = 0; i < days + 15; i++){
+            System.out.println("第" + (i + 1) + "天: " + res[i]);
+        }
+    }
+
+    public static void main(String[] args) {
+        testMemory(75,2);
+    }
+}