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 package org.apache.harmony.xnet.provider.jsse; 19 20 import java.security.KeyStore; 21 import java.security.MessageDigest; 22 import java.security.PrivateKey; 23 import java.security.cert.Certificate; 24 25 import junit.framework.TestCase; 26 27 /** 28 * Tests for <code>DigitalSignature</code> constructor and methods 29 */ 30 public class DigitalSignatureTest extends TestCase { 31 32 private PrivateKey key; 33 private Certificate cert; 34 35 @Override 36 public void setUp() throws Exception { 37 38 char[] pwd = JSSETestData.KS_PASSWORD; 39 KeyStore ks = JSSETestData.getKeyStore(); 40 KeyStore.PrivateKeyEntry entry = (KeyStore.PrivateKeyEntry) ks 41 .getEntry("ssl_test_store", 42 new KeyStore.PasswordProtection(pwd)); 43 key = entry.getPrivateKey(); 44 cert = entry.getCertificate(); 45 } 46 47 public void testDigitalSignature_1() throws Exception { 48 49 MessageDigest md5 = null; 50 MessageDigest sha = null; 51 52 md5 = MessageDigest.getInstance("MD5"); 53 sha = MessageDigest.getInstance("SHA-1"); 54 55 DigitalSignature ds_sign = new DigitalSignature( 56 CipherSuite.KeyExchange_RSA_EXPORT); 57 DigitalSignature ds_verify = new DigitalSignature( 58 CipherSuite.KeyExchange_RSA_EXPORT); 59 ds_sign.init(key); 60 // use pivateKeyEncoding as signed data 61 byte[] pivateKeyEncoding = key.getEncoded(); 62 ds_sign.update(pivateKeyEncoding); 63 byte[] hash = ds_sign.sign(); 64 65 // verify 66 byte[] md5_hash = new byte[16]; 67 byte[] sha_hash = new byte[20]; 68 sha.update(pivateKeyEncoding); 69 md5.update(pivateKeyEncoding); 70 71 sha.digest(sha_hash, 0, sha_hash.length); 72 md5.digest(md5_hash, 0, md5_hash.length); 73 74 ds_verify.init(cert); 75 ds_verify.setMD5(md5_hash); 76 ds_verify.setSHA(sha_hash); 77 78 assertTrue(ds_verify.verifySignature(hash)); 79 } 80 81 public void testDigitalSignature_2() throws Exception { 82 83 DigitalSignature ds_sign = new DigitalSignature( 84 CipherSuite.KeyExchange_RSA_EXPORT); 85 DigitalSignature ds_verify = new DigitalSignature( 86 CipherSuite.KeyExchange_RSA_EXPORT); 87 ds_sign.init(key); 88 89 byte[] md5_hash = new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 90 14, 15, 16 }; 91 byte[] sha_hash = new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 92 14, 15, 16, 17, 18, 19, 20 }; 93 ds_sign.setMD5(md5_hash); 94 ds_sign.setSHA(sha_hash); 95 byte[] hash = ds_sign.sign(); 96 97 // verify 98 ds_verify.init(cert); 99 ds_verify.setMD5(md5_hash); 100 ds_verify.setSHA(sha_hash); 101 assertTrue(ds_verify.verifySignature(hash)); 102 } 103 104 }