1/*
2 *  Licensed to the Apache Software Foundation (ASF) under one or more
3 *  contributor license agreements.  See the NOTICE file distributed with
4 *  this work for additional information regarding copyright ownership.
5 *  The ASF licenses this file to You under the Apache License, Version 2.0
6 *  (the "License"); you may not use this file except in compliance with
7 *  the License.  You may obtain a copy of the License at
8 *
9 *     http://www.apache.org/licenses/LICENSE-2.0
10 *
11 *  Unless required by applicable law or agreed to in writing, software
12 *  distributed under the License is distributed on an "AS IS" BASIS,
13 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 *  See the License for the specific language governing permissions and
15 *  limitations under the License.
16 */
17
18/**
19* @author Boris V. Kuznetsov
20* @version $Revision$
21*/
22
23package tests.java.security;
24
25import java.security.NoSuchAlgorithmException;
26import java.security.NoSuchProviderException;
27import java.security.Provider;
28import java.security.SecureRandom;
29import java.security.Security;
30import junit.framework.TestCase;
31import org.apache.harmony.security.tests.support.RandomImpl;
32
33/**
34 * Tests for <code>SecureRandom</code> constructor and methods
35 *
36 */
37public class SecureRandomTest extends TestCase {
38
39    /**
40     * SRProvider
41     */
42    Provider p;
43
44    /*
45     * @see TestCase#setUp()
46     */
47    protected void setUp() throws Exception {
48        super.setUp();
49        p = new SRProvider();
50        Security.insertProviderAt(p, 1);
51    }
52
53    /*
54     * @see TestCase#tearDown()
55     */
56    protected void tearDown() throws Exception {
57        super.tearDown();
58        Security.removeProvider(p.getName());
59    }
60
61    public final void testNext() {
62        MySecureRandom sr = new MySecureRandom();
63        if (sr.nextElement(1) != 1 || sr.nextElement(2) != 3 || sr.nextElement(3) != 7) {
64            fail("next failed");
65        }
66    }
67
68    /*
69     * Class under test for void setSeed(long)
70     */
71    public final void testSetSeedlong() {
72        SecureRandom sr = new SecureRandom();
73        sr.setSeed(12345);
74        if (!RandomImpl.runEngineSetSeed) {
75            fail("setSeed failed");
76        }
77    }
78
79    public final void testNextBytes() {
80        byte[] b = new byte[5];
81        SecureRandom sr = new SecureRandom();
82        sr.nextBytes(b);
83        for (int i = 0; i < b.length; i++) {
84            if (b[i] != (byte)(i + 0xF1)) {
85                fail("nextBytes failed");
86            }
87        }
88
89        try {
90            sr.nextBytes(null);
91            fail("expected exception");
92        } catch (Exception e) {
93            // ok
94        }
95    }
96
97    /*
98     * Class under test for void SecureRandom()
99     */
100    public final void testSecureRandom() {
101        SecureRandom sr = new SecureRandom();
102        if (!sr.getAlgorithm().equals("someRandom")  ||
103                sr.getProvider()!= p) {
104            fail("incorrect SecureRandom implementation" + p.getName());
105        }
106    }
107
108    /*
109     * Class under test for void SecureRandom(byte[])
110     */
111    public final void testSecureRandombyteArray() {
112        byte[] b = {1,2,3};
113        new SecureRandom(b);
114
115        if (!RandomImpl.runEngineSetSeed) {
116            fail("No setSeed");
117        }
118
119
120    }
121
122    /*
123     * Class under test for SecureRandom getInstance(String)
124     */
125    public final void testGetInstanceString() {
126        SecureRandom sr = null;
127        try {
128            sr = SecureRandom.getInstance("someRandom");
129        } catch (NoSuchAlgorithmException e) {
130            fail(e.toString());
131        }
132        if (sr.getProvider() != p || !"someRandom".equals(sr.getAlgorithm())) {
133            fail("getInstance failed");
134        }
135    }
136
137    /*
138     * Class under test for SecureRandom getInstance(String, String)
139     */
140    public final void testGetInstanceStringString() throws Exception {
141        SecureRandom sr = SecureRandom.getInstance("someRandom", "SRProvider");
142        if (sr.getProvider() != p || !"someRandom".equals(sr.getAlgorithm())) {
143            fail("getInstance failed");
144        }
145
146        try {
147            SecureRandom r = SecureRandom.getInstance("anotherRandom", "SRProvider");
148            fail("expected NoSuchAlgorithmException");
149        } catch (NoSuchAlgorithmException e) {
150            // ok
151        } catch (NoSuchProviderException e) {
152            fail("unexpected: " + e);
153        } catch (IllegalArgumentException e) {
154            fail("unexpected: " + e);
155        } catch (NullPointerException e) {
156            fail("unexpected: " + e);
157        }
158
159        try {
160            SecureRandom r = SecureRandom.getInstance("someRandom", "UnknownProvider");
161            fail("expected NoSuchProviderException");
162        } catch (NoSuchProviderException e) {
163            // ok
164        } catch (NoSuchAlgorithmException e) {
165            fail("unexpected: " + e);
166        } catch (IllegalArgumentException e) {
167            fail("unexpected: " + e);
168        } catch (NullPointerException e) {
169            fail("unexpected: " + e);
170        }
171
172        try {
173            SecureRandom r = SecureRandom.getInstance("someRandom", (String)null);
174            fail("expected IllegalArgumentException");
175        } catch (NoSuchProviderException e) {
176            fail("unexpected: " + e);
177        } catch (NoSuchAlgorithmException e) {
178            fail("unexpected: " + e);
179        } catch (IllegalArgumentException e) {
180            // ok
181        } catch (NullPointerException e) {
182            fail("unexpected: " + e);
183        }
184
185        try {
186            SecureRandom r = SecureRandom.getInstance(null, "SRProvider");
187            fail("expected NullPointerException");
188        } catch (NoSuchProviderException e) {
189            fail("unexpected: " + e);
190        } catch (NoSuchAlgorithmException e) {
191            fail("unexpected: " + e);
192        } catch (IllegalArgumentException e) {
193            fail("unexpected: " + e);
194        } catch (NullPointerException e) {
195            // ok
196        }
197    }
198
199    /*
200     * Class under test for SecureRandom getInstance(String, Provider)
201     */
202    public final void testGetInstanceStringProvider() throws Exception {
203        Provider p = new SRProvider();
204        SecureRandom sr = SecureRandom.getInstance("someRandom", p);
205        if (sr.getProvider() != p || !"someRandom".equals(sr.getAlgorithm())) {
206            fail("getInstance failed");
207        }
208
209        try {
210            SecureRandom r = SecureRandom.getInstance("unknownRandom", p);
211            fail("expected NoSuchAlgorithmException");
212        } catch (NoSuchAlgorithmException e) {
213            // ok
214        } catch (IllegalArgumentException e) {
215            fail("unexpected: " + e);
216        } catch (NullPointerException e) {
217            fail("unexpected: " + e);
218        }
219
220
221        try {
222            SecureRandom r = SecureRandom.getInstance(null, p);
223            fail("expected NullPointerException");
224        } catch (NoSuchAlgorithmException e) {
225            fail("unexpected: " + e);
226        } catch (IllegalArgumentException e) {
227            fail("unexpected: " + e);
228        } catch (NullPointerException e) {
229            // ok
230        }
231
232        try {
233            SecureRandom r = SecureRandom.getInstance("anyRandom", (Provider)null);
234            fail("expected IllegalArgumentException");
235        } catch (NoSuchAlgorithmException e) {
236            fail("unexpected: " + e);
237        } catch (IllegalArgumentException e) {
238            // ok
239        } catch (NullPointerException e) {
240            fail("unexpected: " + e);
241        }
242
243
244    }
245
246    /*
247     * Class under test for void setSeed(byte[])
248     */
249    public final void testSetSeedbyteArray() {
250        byte[] b = {1,2,3};
251        SecureRandom sr = new SecureRandom();
252        sr.setSeed(b);
253        if (!RandomImpl.runEngineSetSeed) {
254            fail("setSeed failed");
255        }
256
257    }
258
259    public final void testGetSeed() {
260        byte[] b = SecureRandom.getSeed(4);
261        if( b.length != 4) {
262            fail("getSeed failed");
263        }
264    }
265
266    public final void testGenerateSeed() {
267        SecureRandom sr = new SecureRandom();
268        byte[] b = sr.generateSeed(4);
269        for (int i = 0; i < b.length; i++) {
270            if (b[i] != (byte)i) {
271                fail("generateSeed failed");
272            }
273        }
274    }
275
276
277
278    public class SRProvider extends Provider {
279        public SRProvider() {
280            super("SRProvider", 1.0, "SRProvider for testing");
281            put("SecureRandom.someRandom",
282                    "org.apache.harmony.security.tests.support.RandomImpl");
283        }
284    }
285
286    class MySecureRandom extends SecureRandom {
287        public MySecureRandom(){
288            super();
289        }
290
291        public int nextElement(int numBits) {
292            return super.next(numBits);
293        }
294    }
295}
296