1 /*
   2  * Copyright (c) 2018, 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 import java.security.*;
  24 import java.security.interfaces.RSAPrivateKey;
  25 import java.security.interfaces.RSAPublicKey;
  26 import java.security.spec.*;
  27 import java.util.Arrays;
  28 import java.util.stream.IntStream;
  29 import static javax.crypto.Cipher.PRIVATE_KEY;
  30 import static javax.crypto.Cipher.PUBLIC_KEY;
  31 
  32 /**
  33  * @test
  34  * @bug 8146293
  35  * @summary Test RSASSA-PSS AlgorithmParameters impl of SunRsaSign provider.
  36  * @run main PSSParametersTest
  37  */
  38 public class PSSParametersTest {
  39     /**
  40      * JDK default RSA Provider.
  41      */
  42     private static final String PROVIDER = "SunRsaSign";
  43 
  44     private static final String ALGO = "RSASSA-PSS";
  45 
  46     public static void main(String[] args) throws Exception {
  47         System.out.println("Testing against DEFAULT parameters");
  48         test(PSSParameterSpec.DEFAULT);
  49         System.out.println("Testing against custom parameters");
  50         test(new PSSParameterSpec("SHA-512/224", "MGF1", MGF1ParameterSpec.SHA384,
  51             100, 1));
  52         System.out.println("Test Passed");
  53     }
  54 
  55     // test against the given spec by initializing w/ it, generate the DER bytes,
  56     // then initialize another instance w/ the DER bytes, retrieve the spec.
  57     // compare both spec for equality and throw exception if the comparison failed.
  58     private static void test(PSSParameterSpec spec) throws Exception {
  59         AlgorithmParameters params = AlgorithmParameters.getInstance(ALGO, PROVIDER);
  60         params.init(spec);
  61         byte[] encoded = params.getEncoded();
  62         AlgorithmParameters params2 = AlgorithmParameters.getInstance(ALGO, PROVIDER);
  63         params2.init(encoded);
  64         PSSParameterSpec spec2 = params2.getParameterSpec(PSSParameterSpec.class);
  65         if (!isEqual(spec, spec2)) {
  66             throw new RuntimeException("Spec check Failed");
  67         }
  68     }
  69 
  70     private static boolean isEqual(PSSParameterSpec spec, PSSParameterSpec spec2)
  71         throws Exception {
  72         if (spec == spec2) return true;
  73         if (spec == null || spec2 == null) return false;
  74 
  75         if (!spec.getDigestAlgorithm().equals(spec2.getDigestAlgorithm())) {
  76             System.out.println("Different digest algorithms: " +
  77                 spec.getDigestAlgorithm() + " vs " + spec2.getDigestAlgorithm());
  78             return false;
  79         }
  80         if (!spec.getMGFAlgorithm().equals(spec2.getMGFAlgorithm())) {
  81             System.out.println("Different MGF algorithms: " +
  82                 spec.getMGFAlgorithm() + " vs " + spec2.getMGFAlgorithm());
  83             return false;
  84         }
  85         if (spec.getSaltLength() != spec2.getSaltLength()) {
  86             System.out.println("Different Salt Length: " +
  87                 spec.getSaltLength() + " vs " + spec2.getSaltLength());
  88             return false;
  89         }
  90         if (spec.getTrailerField() != spec2.getTrailerField()) {
  91             System.out.println("Different TrailerField: " +
  92                 spec.getTrailerField() + " vs " + spec2.getTrailerField());
  93             return false;
  94         }
  95         // continue checking MGF Parameters
  96         AlgorithmParameterSpec mgfParams = spec.getMGFParameters();
  97         AlgorithmParameterSpec mgfParams2 = spec2.getMGFParameters();
  98         if (mgfParams == mgfParams2) return true;
  99         if (mgfParams == null || mgfParams2 == null) {
 100             System.out.println("Different MGF Parameters: " +
 101                 mgfParams + " vs " + mgfParams2);
 102             return false;
 103         }
 104         if (mgfParams instanceof MGF1ParameterSpec) {
 105             if (mgfParams2 instanceof MGF1ParameterSpec) {
 106                 boolean result =
 107                     ((MGF1ParameterSpec)mgfParams).getDigestAlgorithm().equals
 108                          (((MGF1ParameterSpec)mgfParams2).getDigestAlgorithm());
 109                 if (!result) {
 110                     System.out.println("Different Digest algo in MGF Parameters: " +
 111                         ((MGF1ParameterSpec)mgfParams).getDigestAlgorithm() + " vs " +
 112                         ((MGF1ParameterSpec)mgfParams2).getDigestAlgorithm());
 113                 }
 114                 return result;
 115             } else {
 116                 System.out.println("Different MGF Parameters types: " +
 117                     mgfParams.getClass() + " vs " + mgfParams2.getClass());
 118                 return false;
 119             }
 120         }
 121         throw new RuntimeException("Unrecognized MGFParameters: " + mgfParams);
 122     }
 123 }