test/jdk/sun/security/provider/SecureRandom/AbstractDrbg/SpecTest.java - platform/libcore - Git at Google

 /*
  * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * This code is free software; you can redistribute it and/or modify it
  * under the terms of the GNU General Public License version 2 only, as
  * published by the Free Software Foundation.
  *
  * This code is distributed in the hope that it will be useful, but WITHOUT
  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  * version 2 for more details (a copy is included in the LICENSE file that
  * accompanied this code).
  *
  * You should have received a copy of the GNU General Public License version
  * 2 along with this work; if not, write to the Free Software Foundation,
  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  *
  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  * or visit www.oracle.com if you need additional information or have any
  * questions.
  */

 /* @test
  * @bug 8051408 8157308 8130181
  * @modules java.base/sun.security.provider
  * @build java.base/sun.security.provider.S
  * @run main SpecTest
  * @summary check the AbstractDrbg API etc
  */

 import java.security.*;
 import sun.security.provider.S;
 import static java.security.DrbgParameters.Capability.*;

 /**
  * This test makes sure the AbstractDrbg API works as specified. It also
  * checks the SecureRandom API.
  *
  * The implementations must be patched into java.base/sun.security.provider
  * because AbstractDrbg is not a public interface.
  */
 public class SpecTest {

     public static void main(String args[]) throws Exception {

         // getInstance from a provider.

         Provider p = new All("A", "0", "");
         byte[] bytes = new byte[100];

         // A non-DRBG
         iae(() -> SecureRandom.getInstance("S1", null, p));
         nsae(() -> SecureRandom.getInstance("S1",
                 new SecureRandomParameters() {}, p));

         SecureRandom s1 = SecureRandom.getInstance("S1", p);
         if (s1.getParameters() != null) {
             throw new Exception();
         }

         iae(() -> s1.nextBytes(bytes, null));
         uoe(() -> s1.nextBytes(bytes, new SecureRandomParameters() {}));
         uoe(() -> s1.reseed());
         iae(() -> s1.reseed(null));
         uoe(() -> s1.reseed(new SecureRandomParameters() {}));

         // A weak DRBG
         iae(() -> SecureRandom.getInstance("S2", null, p));
         nsae(() -> SecureRandom.getInstance("S2",
                 new SecureRandomParameters() {}, p));
         nsae(() -> SecureRandom.getInstance("S2",
                 DrbgParameters.instantiation(256, NONE, null), p));
         nsae(() -> SecureRandom.getInstance("S2",
                 DrbgParameters.instantiation(-1, PR_AND_RESEED, null), p));
         nsae(() -> SecureRandom.getInstance("S2",
                 DrbgParameters.instantiation(-1, RESEED_ONLY, null), p));

         SecureRandom s2 = SecureRandom.getInstance("S2",
                 DrbgParameters.instantiation(-1, NONE, null), p);
         equals(s2, "S2,SQUEEZE,128,none");
         equals(s2.getParameters(), "128,none,null");

         npe(() -> s2.nextBytes(null));
         iae(() -> s2.nextBytes(bytes, null));
         iae(() -> s2.nextBytes(bytes, new SecureRandomParameters() {}));
         uoe(() -> s2.reseed());
         iae(() -> s2.reseed(null));

         iae(() -> s2.nextBytes(bytes,
                 DrbgParameters.nextBytes(-1, false, new byte[101])));
         s2.nextBytes(new byte[101],
                 DrbgParameters.nextBytes(-1, false, new byte[100]));
         s2.nextBytes(bytes,
                 DrbgParameters.nextBytes(-1, false, new byte[100]));

         // A strong DRBG
         iae(() -> SecureRandom.getInstance("S3", null, p));
         nsae(() -> SecureRandom.getInstance("S3",
                 new SecureRandomParameters() {}, p));
         SecureRandom.getInstance("S3",
                 DrbgParameters.instantiation(192, PR_AND_RESEED, null), p);

         SecureRandom s3 = SecureRandom.getInstance("S3", p);
         equals(s3, "S3,SQUEEZE,128,reseed_only");
         equals(s3.getParameters(), "128,reseed_only,null");

         iae(() -> s3.nextBytes(bytes,
                 DrbgParameters.nextBytes(192, false, null)));
         iae(() -> s3.nextBytes(bytes,
                 DrbgParameters.nextBytes(112, true, null)));
         iae(() -> s3.reseed(new SecureRandomParameters() {}));

         SecureRandom s32 = SecureRandom.getInstance(
                 "S3", DrbgParameters.instantiation(192, PR_AND_RESEED, null), p);
         equals(s32, "S3,SQUEEZE,192,pr_and_reseed");
         equals(s32.getParameters(), "192,pr_and_reseed,null");

         s32.nextBytes(bytes, DrbgParameters.nextBytes(192, false, null));
         s32.nextBytes(bytes, DrbgParameters.nextBytes(112, true, null));
         s32.reseed();
         s32.reseed(DrbgParameters.reseed(true, new byte[100]));

         // getInstance from competitive providers.

         Provider l = new Legacy("L", "0", "");
         Provider w = new Weak("W", "0", "");
         Provider s = new Strong("S", "0", "");

         Security.addProvider(l);
         Security.addProvider(w);
         Security.addProvider(s);

         SecureRandom s4;

         try {
             s4 = SecureRandom.getInstance("S");
             if (s4.getProvider() != l) {
                 throw new Exception();
             }

             nsae(() -> SecureRandom.getInstance(
                     "S", DrbgParameters.instantiation(256, NONE, null)));

             s4 = SecureRandom.getInstance(
                     "S", DrbgParameters.instantiation(192, NONE, null));
             if (s4.getProvider() != s) {
                 throw new Exception();
             }

             s4 = SecureRandom.getInstance(
                     "S", DrbgParameters.instantiation(128, PR_AND_RESEED, null));
             if (s4.getProvider() != s) {
                 throw new Exception();
             }

             s4 = SecureRandom.getInstance(
                     "S", DrbgParameters.instantiation(128, RESEED_ONLY, null));
             if (s4.getProvider() != s) {
                 throw new Exception();
             }

             s4 = SecureRandom.getInstance(
                     "S", DrbgParameters.instantiation(128, NONE, null));
             if (s4.getProvider() != w) {
                 throw new Exception();
             }
         } finally {
             Security.removeProvider("L");
             Security.removeProvider("W");
             Security.removeProvider("S");
         }
     }

     public static class All extends Provider {
         protected All(String name, String version, String info) {
             super(name, version, info);
             put("SecureRandom.S1", S.S1.class.getName());
             put("SecureRandom.S2", S.S2.class.getName());
             put("SecureRandom.S3", S.S3.class.getName());
         }
     }

     // Providing S with no params support
     public static class Legacy extends Provider {
         protected Legacy(String name, String version, String info) {
             super(name, version, info);
             put("SecureRandom.S", S.S1.class.getName());
         }
     }

     public static class Weak extends Provider {
         protected Weak(String name, String version, String info) {
             super(name, version, info);
             put("SecureRandom.S", S.S2.class.getName());
         }
     }

     public static class Strong extends Provider {
         protected Strong(String name, String version, String info) {
             super(name, version, info);
             put("SecureRandom.S", S.S3.class.getName());
         }
     }

     static void nsae(RunnableWithException r) throws Exception {
         checkException(r, NoSuchAlgorithmException.class);
     }

     static void iae(RunnableWithException r) throws Exception {
         checkException(r, IllegalArgumentException.class);
     }

     static void uoe(RunnableWithException r) throws Exception {
         checkException(r, UnsupportedOperationException.class);
     }

     static void npe(RunnableWithException r) throws Exception {
         checkException(r, NullPointerException.class);
     }

     interface RunnableWithException {
         void run() throws Exception;
     }

     static void checkException(RunnableWithException r, Class ex)
             throws Exception {
         try {
             r.run();
         } catch (Exception e) {
             if (ex.isAssignableFrom(e.getClass())) {
                 return;
             }
             throw e;
         }
         throw new Exception("No exception thrown");
     }

     static void equals(Object o, String s) throws Exception {
         if (!o.toString().equals(s)) {
             throw new Exception(o.toString() + " is not " + s);
         }
     }
 }
	/*
	* Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
	*
	* This code is free software; you can redistribute it and/or modify it
	* under the terms of the GNU General Public License version 2 only, as
	* published by the Free Software Foundation.
	*
	* This code is distributed in the hope that it will be useful, but WITHOUT
	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
	* version 2 for more details (a copy is included in the LICENSE file that
	* accompanied this code).
	*
	* You should have received a copy of the GNU General Public License version
	* 2 along with this work; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
	*
	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
	* or visit www.oracle.com if you need additional information or have any
	* questions.
	*/

	/* @test
	* @bug 8051408 8157308 8130181
	* @modules java.base/sun.security.provider
	* @build java.base/sun.security.provider.S
	* @run main SpecTest
	* @summary check the AbstractDrbg API etc
	*/

	import java.security.*;
	import sun.security.provider.S;
	import static java.security.DrbgParameters.Capability.*;

	/**
	* This test makes sure the AbstractDrbg API works as specified. It also
	* checks the SecureRandom API.
	*
	* The implementations must be patched into java.base/sun.security.provider
	* because AbstractDrbg is not a public interface.
	*/
	public class SpecTest {

	public static void main(String args[]) throws Exception {

	// getInstance from a provider.

	Provider p = new All("A", "0", "");
	byte[] bytes = new byte[100];

	// A non-DRBG
	iae(() -> SecureRandom.getInstance("S1", null, p));
	nsae(() -> SecureRandom.getInstance("S1",
	new SecureRandomParameters() {}, p));

	SecureRandom s1 = SecureRandom.getInstance("S1", p);
	if (s1.getParameters() != null) {
	throw new Exception();
	}

	iae(() -> s1.nextBytes(bytes, null));
	uoe(() -> s1.nextBytes(bytes, new SecureRandomParameters() {}));
	uoe(() -> s1.reseed());
	iae(() -> s1.reseed(null));
	uoe(() -> s1.reseed(new SecureRandomParameters() {}));

	// A weak DRBG
	iae(() -> SecureRandom.getInstance("S2", null, p));
	nsae(() -> SecureRandom.getInstance("S2",
	new SecureRandomParameters() {}, p));
	nsae(() -> SecureRandom.getInstance("S2",
	DrbgParameters.instantiation(256, NONE, null), p));
	nsae(() -> SecureRandom.getInstance("S2",
	DrbgParameters.instantiation(-1, PR_AND_RESEED, null), p));
	nsae(() -> SecureRandom.getInstance("S2",
	DrbgParameters.instantiation(-1, RESEED_ONLY, null), p));

	SecureRandom s2 = SecureRandom.getInstance("S2",
	DrbgParameters.instantiation(-1, NONE, null), p);
	equals(s2, "S2,SQUEEZE,128,none");
	equals(s2.getParameters(), "128,none,null");

	npe(() -> s2.nextBytes(null));
	iae(() -> s2.nextBytes(bytes, null));
	iae(() -> s2.nextBytes(bytes, new SecureRandomParameters() {}));
	uoe(() -> s2.reseed());
	iae(() -> s2.reseed(null));

	iae(() -> s2.nextBytes(bytes,
	DrbgParameters.nextBytes(-1, false, new byte[101])));
	s2.nextBytes(new byte[101],
	DrbgParameters.nextBytes(-1, false, new byte[100]));
	s2.nextBytes(bytes,
	DrbgParameters.nextBytes(-1, false, new byte[100]));

	// A strong DRBG
	iae(() -> SecureRandom.getInstance("S3", null, p));
	nsae(() -> SecureRandom.getInstance("S3",
	new SecureRandomParameters() {}, p));
	SecureRandom.getInstance("S3",
	DrbgParameters.instantiation(192, PR_AND_RESEED, null), p);

	SecureRandom s3 = SecureRandom.getInstance("S3", p);
	equals(s3, "S3,SQUEEZE,128,reseed_only");
	equals(s3.getParameters(), "128,reseed_only,null");

	iae(() -> s3.nextBytes(bytes,
	DrbgParameters.nextBytes(192, false, null)));
	iae(() -> s3.nextBytes(bytes,
	DrbgParameters.nextBytes(112, true, null)));
	iae(() -> s3.reseed(new SecureRandomParameters() {}));

	SecureRandom s32 = SecureRandom.getInstance(
	"S3", DrbgParameters.instantiation(192, PR_AND_RESEED, null), p);
	equals(s32, "S3,SQUEEZE,192,pr_and_reseed");
	equals(s32.getParameters(), "192,pr_and_reseed,null");

	s32.nextBytes(bytes, DrbgParameters.nextBytes(192, false, null));
	s32.nextBytes(bytes, DrbgParameters.nextBytes(112, true, null));
	s32.reseed();
	s32.reseed(DrbgParameters.reseed(true, new byte[100]));

	// getInstance from competitive providers.

	Provider l = new Legacy("L", "0", "");
	Provider w = new Weak("W", "0", "");
	Provider s = new Strong("S", "0", "");

	Security.addProvider(l);
	Security.addProvider(w);
	Security.addProvider(s);

	SecureRandom s4;

	try {
	s4 = SecureRandom.getInstance("S");
	if (s4.getProvider() != l) {
	throw new Exception();
	}

	nsae(() -> SecureRandom.getInstance(
	"S", DrbgParameters.instantiation(256, NONE, null)));

	s4 = SecureRandom.getInstance(
	"S", DrbgParameters.instantiation(192, NONE, null));
	if (s4.getProvider() != s) {
	throw new Exception();
	}

	s4 = SecureRandom.getInstance(
	"S", DrbgParameters.instantiation(128, PR_AND_RESEED, null));
	if (s4.getProvider() != s) {
	throw new Exception();
	}

	s4 = SecureRandom.getInstance(
	"S", DrbgParameters.instantiation(128, RESEED_ONLY, null));
	if (s4.getProvider() != s) {
	throw new Exception();
	}

	s4 = SecureRandom.getInstance(
	"S", DrbgParameters.instantiation(128, NONE, null));
	if (s4.getProvider() != w) {
	throw new Exception();
	}
	} finally {
	Security.removeProvider("L");
	Security.removeProvider("W");
	Security.removeProvider("S");
	}
	}

	public static class All extends Provider {
	protected All(String name, String version, String info) {
	super(name, version, info);
	put("SecureRandom.S1", S.S1.class.getName());
	put("SecureRandom.S2", S.S2.class.getName());
	put("SecureRandom.S3", S.S3.class.getName());
	}
	}

	// Providing S with no params support
	public static class Legacy extends Provider {
	protected Legacy(String name, String version, String info) {
	super(name, version, info);
	put("SecureRandom.S", S.S1.class.getName());
	}
	}

	public static class Weak extends Provider {
	protected Weak(String name, String version, String info) {
	super(name, version, info);
	put("SecureRandom.S", S.S2.class.getName());
	}
	}

	public static class Strong extends Provider {
	protected Strong(String name, String version, String info) {
	super(name, version, info);
	put("SecureRandom.S", S.S3.class.getName());
	}
	}

	static void nsae(RunnableWithException r) throws Exception {
	checkException(r, NoSuchAlgorithmException.class);
	}

	static void iae(RunnableWithException r) throws Exception {
	checkException(r, IllegalArgumentException.class);
	}

	static void uoe(RunnableWithException r) throws Exception {
	checkException(r, UnsupportedOperationException.class);
	}

	static void npe(RunnableWithException r) throws Exception {
	checkException(r, NullPointerException.class);
	}

	interface RunnableWithException {
	void run() throws Exception;
	}

	static void checkException(RunnableWithException r, Class ex)
	throws Exception {
	try {
	r.run();
	} catch (Exception e) {
	if (ex.isAssignableFrom(e.getClass())) {
	return;
	}
	throw e;
	}
	throw new Exception("No exception thrown");
	}

	static void equals(Object o, String s) throws Exception {
	if (!o.toString().equals(s)) {
	throw new Exception(o.toString() + " is not " + s);
	}
	}
	}