EncodedKeySpec2Test.java revision 229e34b182b98e1dba15d3dc6341954986ae2b7a 
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
18package tests.security.spec;
19
20import junit.framework.TestCase;
21
22import java.security.Key;
23import java.security.KeyFactory;
24import java.security.KeyPair;
25import java.security.KeyPairGenerator;
26import java.security.interfaces.DSAPrivateKey;
27import java.security.interfaces.DSAPublicKey;
28import java.security.spec.EncodedKeySpec;
29import java.security.spec.PKCS8EncodedKeySpec;
30import java.security.spec.X509EncodedKeySpec;
31
32public class EncodedKeySpec2Test extends TestCase {
33
34    /**
35     * java.security.spec.EncodedKeySpec#getEncoded()
36     */
37    public void test_getEncoded() throws Exception {
38
39               KeyPairGenerator keyGen = KeyPairGenerator.getInstance("DSA");
40
41               keyGen.initialize(1024);
42               KeyPair keys = keyGen.generateKeyPair();
43
44
45               KeyFactory fact = KeyFactory.getInstance("DSA");
46
47
48               // check public key encoding
49               byte[] encoded = keys.getPublic().getEncoded();
50               Key key = fact.generatePublic(new X509EncodedKeySpec(encoded));
51
52               assertTrue("public key encodings were different",
53                           isEqual(key, keys.getPublic()));
54
55               // check private key encoding
56               encoded = keys.getPrivate().getEncoded();
57               key = fact.generatePrivate(new PKCS8EncodedKeySpec(encoded));
58
59               assertTrue("private key encodings were different",
60                           isEqual(key, keys.getPrivate()));
61    }
62
63    private boolean isEqual(Key key1, Key key2) {
64        if (key1 instanceof DSAPublicKey && key2 instanceof DSAPublicKey) {
65            DSAPublicKey dsa1 = ((DSAPublicKey) key1);
66            DSAPublicKey dsa2 = ((DSAPublicKey) key2);
67            return dsa1.getY().equals(dsa2.getY())
68                    && dsa1.getParams().getG().equals(dsa2.getParams().getG())
69                    && dsa1.getParams().getP().equals(dsa2.getParams().getP())
70                    && dsa1.getParams().getQ().equals(dsa2.getParams().getQ());
71
72        } else if (key1 instanceof DSAPrivateKey
73                && key2 instanceof DSAPrivateKey) {
74            DSAPrivateKey dsa1 = ((DSAPrivateKey) key1);
75            DSAPrivateKey dsa2 = ((DSAPrivateKey) key2);
76            return dsa1.getX().equals(dsa2.getX())
77                    && dsa1.getParams().getG().equals(dsa2.getParams().getG())
78                    && dsa1.getParams().getP().equals(dsa2.getParams().getP())
79                    && dsa1.getParams().getQ().equals(dsa2.getParams().getQ());
80        } else {
81            return false;
82        }
83    }
84}
85