Serialize.java revision 14557:3bcf2d4fdb54 
1/*
2 * Copyright (c) 1998, 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/* @test
25 * @bug 4102896
26 * @summary Make sure that a SecureRandom object can be serialized
27 * @run main/othervm Serialize
28 */
29
30import java.security.*;
31import java.io.*;
32
33public class Serialize {
34
35    public static void main(String args[]) throws Exception {
36        System.setProperty("java.security.egd", "file:/dev/urandom");
37
38        for (String alg: new String[]{
39                "SHA1PRNG", "DRBG", "Hash_DRBG", "HMAC_DRBG", "CTR_DRBG",
40                "Hash_DRBG,SHA-512,192,pr_and_reseed"}) {
41
42            System.out.println("Testing " + alg);
43            SecureRandom s1;
44
45            // A SecureRandom can be s11ned and des11ned at any time.
46
47            // Brand new.
48            s1 = getInstance(alg);
49            revive(s1).nextInt();
50            if (alg.contains("DRBG")) {
51                revive(s1).reseed();
52            }
53
54            // Used
55            s1 = getInstance(alg);
56            s1.nextInt();    // state set
57            revive(s1).nextInt();
58            if (alg.contains("DRBG")) {
59                revive(s1).reseed();
60            }
61
62            // Automatically reseeded
63            s1 = getInstance(alg);
64            if (alg.contains("DRBG")) {
65                s1.reseed();
66            }
67            revive(s1).nextInt();
68            if (alg.contains("DRBG")) {
69                revive(s1).reseed();
70            }
71
72            // Manually seeded
73            s1 = getInstance(alg);
74            s1.setSeed(1L);
75            revive(s1).nextInt();
76            if (alg.contains("DRBG")) {
77                revive(s1).reseed();
78            }
79        }
80    }
81
82    private static SecureRandom getInstance(String alg) throws Exception {
83        if (alg.equals("SHA1PRNG") || alg.equals("DRBG")) {
84            return SecureRandom.getInstance(alg);
85        } else {
86            String old = Security.getProperty("securerandom.drbg.config");
87            try {
88                Security.setProperty("securerandom.drbg.config", alg);
89                return SecureRandom.getInstance("DRBG");
90            } finally {
91                Security.setProperty("securerandom.drbg.config", old);
92            }
93        }
94    }
95
96    private static SecureRandom revive(SecureRandom oldOne) throws Exception {
97        ByteArrayOutputStream bout = new ByteArrayOutputStream();
98        new ObjectOutputStream(bout).writeObject(oldOne);
99        SecureRandom newOne = (SecureRandom) new ObjectInputStream(
100                new ByteArrayInputStream(bout.toByteArray())).readObject();
101        if (!oldOne.toString().equals(newOne.toString())) {
102            throw new Exception(newOne + " is not " + oldOne);
103        }
104        return newOne;
105    }
106}
107