1 /*
   2  * Copyright (c) 2013, 2016, 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 7200306 8029158
  27  * @summary verify that P11Signature impl will error out when initialized
  28  * with unsupported key sizes
  29  * @library ..
  30  * @key randomness
  31  * @modules jdk.crypto.token
  32  * @run main/othervm TestDSAKeyLength
  33  * @run main/othervm TestDSAKeyLength sm
  34  */
  35 
  36 import java.security.InvalidKeyException;
  37 import java.security.KeyPair;
  38 import java.security.KeyPairGenerator;
  39 import java.security.Provider;
  40 import java.security.SecureRandom;
  41 import java.security.Signature;
  42 
  43 public class TestDSAKeyLength extends PKCS11Test {
  44 
  45     public static void main(String[] args) throws Exception {
  46         main(new TestDSAKeyLength(), args);
  47     }
  48 
  49     @Override
  50     public void main(Provider provider) throws Exception {
  51         if (isNSS(provider) && getNSSVersion() >= 3.14) {
  52             System.out.println("Skip testing NSS " + getNSSVersion());
  53             return;
  54         }
  55 
  56         /*
  57          * Use Solaris SPARC 11.2 or later to avoid an intermittent failure
  58          * when running SunPKCS11-Solaris (8044554)
  59          */
  60         if (provider.getName().equals("SunPKCS11-Solaris") &&
  61             props.getProperty("os.name").equals("SunOS") &&
  62             props.getProperty("os.arch").equals("sparcv9") &&
  63             props.getProperty("os.version").compareTo("5.11") <= 0 &&
  64             getDistro().compareTo("11.2") < 0) {
  65 
  66             System.out.println("SunPKCS11-Solaris provider requires " +
  67                 "Solaris SPARC 11.2 or later, skipping");
  68             return;
  69         }
  70 
  71         KeyPairGenerator kpg = KeyPairGenerator.getInstance("DSA", "SUN");
  72         kpg.initialize(2048, new SecureRandom());
  73         KeyPair pair = kpg.generateKeyPair();
  74 
  75         boolean status = true;
  76         Signature sig = Signature.getInstance("SHA1withDSA", provider);
  77         try {
  78             sig.initSign(pair.getPrivate());
  79             status = false;
  80         } catch (InvalidKeyException ike) {
  81             System.out.println("Expected IKE thrown for initSign()");
  82         }
  83         try {
  84             sig.initVerify(pair.getPublic());
  85             status = false;
  86         } catch (InvalidKeyException ike) {
  87             System.out.println("Expected IKE thrown for initVerify()");
  88         }
  89         if (status) {
  90             System.out.println("Test Passed");
  91         } else {
  92             throw new Exception("Test Failed - expected IKE not thrown");
  93         }
  94     }
  95 }