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 }