1/*
2 * Copyright (c) 2012, 2017, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/*
25 * @test
26 * @bug 6405536
27 * @summary basic test of ECDSA signatures for P-256 and P-384 from the
28 * example data in "Suite B Implementer's Guide to FIPS 186-3".
29 * @library ..
30 * @library ../../../../java/security/testlibrary
31 * @modules java.base/sun.security.util
32 *          jdk.crypto.cryptoki
33 * @compile -XDignore.symbol.file TestECDH2.java
34 * @run main/othervm TestECDH2
35 * @run main/othervm TestECDH2 sm
36 */
37
38import java.math.BigInteger;
39import java.security.AlgorithmParameters;
40import java.security.KeyFactory;
41import java.security.KeyPair;
42import java.security.KeyPairGenerator;
43import java.security.PrivateKey;
44import java.security.Provider;
45import java.security.PublicKey;
46import java.security.spec.ECGenParameterSpec;
47import java.security.spec.ECParameterSpec;
48import java.security.spec.ECPoint;
49import java.security.spec.ECPrivateKeySpec;
50import java.security.spec.ECPublicKeySpec;
51import java.util.Arrays;
52import javax.crypto.KeyAgreement;
53
54public class TestECDH2 extends PKCS11Test {
55
56    // values of the keys we use for the tests
57
58    // keypair using NIST P-256
59    private final static String privD256 = "70a12c2db16845ed56ff68cfc21a472b3f04d7d6851bf6349f2d7d5b3452b38a";
60    private final static String pubX256 = "8101ece47464a6ead70cf69a6e2bd3d88691a3262d22cba4f7635eaff26680a8";
61    private final static String pubY256 = "d8a12ba61d599235f67d9cb4d58f1783d3ca43e78f0a5abaa624079936c0c3a9";
62
63    // keypair using NIST P-384
64    private final static String privD384 = "c838b85253ef8dc7394fa5808a5183981c7deef5a69ba8f4f2117ffea39cfcd90e95f6cbc854abacab701d50c1f3cf24";
65    private final static String pubX384 = "1fbac8eebd0cbf35640b39efe0808dd774debff20a2a329e91713baf7d7f3c3e81546d883730bee7e48678f857b02ca0";
66    private final static String pubY384 = "eb213103bd68ce343365a8a4c3d4555fa385f5330203bdd76ffad1f3affb95751c132007e1b240353cb0a4cf1693bdf9";
67
68    private KeyFactory kf = null;
69    private KeyPairGenerator kpg = null;
70
71    private static void testKeyAgreement(KeyPair kpA, KeyPair kpB, Provider p)
72        throws Exception {
73        KeyAgreement ka1 = KeyAgreement.getInstance("ECDH", p);
74        ka1.init(kpA.getPrivate());
75        ka1.doPhase(kpB.getPublic(), true);
76        byte[] s1 = ka1.generateSecret();
77
78        KeyAgreement ka2 = KeyAgreement.getInstance("ECDH", p);
79        ka2.init(kpB.getPrivate());
80        ka2.doPhase(kpA.getPublic(), true);
81        byte[] s2 = ka2.generateSecret();
82        if (Arrays.equals(s1, s2) == false) {
83            System.out.println("expected: " + toString(s1));
84            System.out.println("actual:   " + toString(s2));
85            throw new Exception("Generated secrets do not match");
86        }
87    }
88
89    private KeyPair genECKeyPair(String curvName, String privD, String pubX,
90                                 String pubY, Provider p) throws Exception {
91        AlgorithmParameters params = AlgorithmParameters.getInstance("EC", p);
92        params.init(new ECGenParameterSpec(curvName));
93        ECParameterSpec ecParams = params.getParameterSpec(ECParameterSpec.class);
94        ECPrivateKeySpec privKeySpec =
95            new ECPrivateKeySpec(new BigInteger(privD, 16), ecParams);
96        ECPublicKeySpec pubKeySpec =
97            new ECPublicKeySpec(new ECPoint(new BigInteger(pubX, 16),
98                                            new BigInteger(pubY, 16)),
99                                ecParams);
100        PrivateKey privKey = kf.generatePrivate(privKeySpec);
101        PublicKey pubKey = kf.generatePublic(pubKeySpec);
102        return new KeyPair(pubKey, privKey);
103    }
104    private KeyPair genECKeyPair(String curvName) throws Exception {
105        ECGenParameterSpec genParams = new ECGenParameterSpec(curvName);
106        kpg.initialize(genParams, null);
107        return kpg.generateKeyPair();
108    }
109    public static void main(String[] args) throws Exception {
110        main(new TestECDH2(), args);
111    }
112
113    @Override
114    public void main(Provider provider) throws Exception {
115        if (provider.getService("KeyAgreement", "ECDH") == null) {
116            System.out.println("ECDH not supported, skipping");
117            return;
118        }
119
120        if (isBadNSSVersion(provider)) {
121            return;
122        }
123
124        kf = KeyFactory.getInstance("EC", provider);
125        kpg = KeyPairGenerator.getInstance("EC", provider);
126
127        System.out.println("Testing against NIST P-256");
128
129        long start = System.currentTimeMillis();
130        KeyPair kp256A =
131            genECKeyPair("secp256r1", privD256, pubX256, pubY256, provider);
132        KeyPair kp256B = genECKeyPair("secp256r1");
133        testKeyAgreement(kp256A, kp256B, provider);
134
135        System.out.println("Testing against NIST P-384");
136        KeyPair kp384A =
137            genECKeyPair("secp384r1", privD384, pubX384, pubY384, provider);
138        KeyPair kp384B = genECKeyPair("secp384r1");
139        testKeyAgreement(kp384A, kp384B, provider);
140
141        long stop = System.currentTimeMillis();
142        System.out.println("All tests passed (" + (stop - start) + " ms).");
143    }
144}
145