post-quantum/NTRUSign.java at master · disparate1/post-quantum · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.security.Provider;
import java.security.Security;
import java.util.Random;

import org.bouncycastle.crypto.AsymmetricCipherKeyPair;
import org.bouncycastle.crypto.digests.SHA256Digest;
import org.bouncycastle.jce.provider.BouncyCastleProvider;
import org.bouncycastle.pqc.crypto.ntru.NTRUSigner;
import org.bouncycastle.pqc.crypto.ntru.NTRUSigningKeyGenerationParameters;
import org.bouncycastle.pqc.crypto.ntru.NTRUSigningKeyPairGenerator;
import org.bouncycastle.pqc.crypto.ntru.NTRUSigningPrivateKeyParameters;
import org.bouncycastle.pqc.crypto.ntru.NTRUSigningPublicKeyParameters;

public class NTRUSign {

	public static void main(String[] args) throws IOException {
		Security.addProvider(new BouncyCastleProvider());
		//testSignVerify(new NTRUSigningKeyGenerationParameters(239, 1024, 7, 6, 4, 1, 1, 0.200, 350, 150, false, true, 0, new SHA256Digest()));
		//testSignVerify(NTRUSigningKeyGenerationParameters.APR2011_439_PROD);
		testSignVerify(NTRUSigningKeyGenerationParameters.APR2011_743_PROD);


	}


	private static void testSignVerify(NTRUSigningKeyGenerationParameters params)
	        throws IOException {
	    NTRUSigner ntru = new NTRUSigner(params.getSigningParameters());
	    NTRUSigningKeyPairGenerator kGen = new NTRUSigningKeyPairGenerator();

	    kGen.init(params);

	    AsymmetricCipherKeyPair kp = kGen.generateKeyPair();

	    ByteArrayOutputStream bos1 = new ByteArrayOutputStream();
		ByteArrayOutputStream bos2 = new ByteArrayOutputStream();

	    NTRUSigningPublicKeyParameters pk = (NTRUSigningPublicKeyParameters) kp.getPublic();
	    NTRUSigningPrivateKeyParameters sk = (NTRUSigningPrivateKeyParameters) kp.getPrivate();

	    pk.writeTo(bos1);
	    sk.writeTo(bos2);
        byte[] encodedPub = bos1.toByteArray();
        byte[] encodedPriv = bos2.toByteArray();

        System.out.println("pub key size: "+encodedPub.length);
		System.out.println("priv key size: "+encodedPriv.length);

	    byte[] msg = "hi".getBytes();


	    // sign and verify
	    ntru.init(true, sk);

	    ntru.update(msg, 0, msg.length);

	    byte[] s = ntru.generateSignature();

	    ntru.init(false, pk);

	    ntru.update(msg, 0, msg.length);

	    boolean valid = ntru.verifySignature(s);
	    System.out.println("message: "+new String(msg));
	    System.out.println("signature: "+new String(s));
	    System.out.println("signature size: "+s.length);
	    System.out.println("valid: "+valid);
	}

}