/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.harmony.crypto.tests.javax.crypto;
import java.security.AlgorithmParameters;
import java.security.InvalidAlgorithmParameterException;
import java.security.InvalidKeyException;
import java.security.Key;
import java.security.NoSuchAlgorithmException;
import java.security.NoSuchProviderException;
import java.security.Provider;
import java.security.Security;
import java.security.spec.AlgorithmParameterSpec;
import javax.crypto.ExemptionMechanism;
import javax.crypto.ExemptionMechanismException;
import javax.crypto.ExemptionMechanismSpi;
import javax.crypto.ShortBufferException;
import javax.crypto.spec.SecretKeySpec;
import org.apache.harmony.crypto.tests.support.MyExemptionMechanismSpi;
import org.apache.harmony.security.tests.support.SpiEngUtils;
import junit.framework.TestCase;
/**
* Tests for <code>ExemptionMechanism</code> class constructors and methods
*
*/
public class ExemptionMechanism_ImplTest extends TestCase {
public static final String srvExemptionMechanism = "ExemptionMechanism";
private static final String defaultAlg = "EMech";
private static final String ExemptionMechanismProviderClass = "org.apache.harmony.crypto.tests.support.MyExemptionMechanismSpi";
private static final String[] invalidValues = SpiEngUtils.invalidValues;
private static final String[] validValues;
static {
validValues = new String[4];
validValues[0] = defaultAlg;
validValues[1] = defaultAlg.toLowerCase();
validValues[2] = "eMEch";
validValues[3] = "emecH";
}
Provider mProv;
protected void setUp() throws Exception {
super.setUp();
mProv = (new SpiEngUtils()).new MyProvider("MyExMechProvider",
"Provider for ExemptionMechanism testing",
srvExemptionMechanism.concat(".").concat(defaultAlg),
ExemptionMechanismProviderClass);
Security.insertProviderAt(mProv, 1);
}
/*
* @see TestCase#tearDown()
*/
protected void tearDown() throws Exception {
super.tearDown();
Security.removeProvider(mProv.getName());
}
private void checkResult(ExemptionMechanism exMech)
throws ExemptionMechanismException, ShortBufferException,
InvalidKeyException, InvalidAlgorithmParameterException {
Key key = new MyExemptionMechanismSpi().new tmpKey("Proba", new byte[0]);
byte[] emptyA = new byte[0];
int len = MyExemptionMechanismSpi.getLength();
byte[] byteA = new byte[len];
try {
exMech.genExemptionBlob();
fail("IllegalStateException must be thrown");
} catch (IllegalStateException e) {
}
try {
exMech.genExemptionBlob(byteA);
fail("IllegalStateException must be thrown");
} catch (IllegalStateException e) {
}
try {
exMech.genExemptionBlob(byteA, 1);
fail("IllegalStateException must be thrown");
} catch (IllegalStateException e) {
}
try {
exMech.getOutputSize(0);
fail("IllegalStateException must be thrown");
} catch (IllegalStateException e) {
}
exMech.init(key);
byte[] bbRes = exMech.genExemptionBlob();
assertEquals("Incorrect length", bbRes.length, len);
assertEquals("Incorrect result", exMech.genExemptionBlob(new byte[5]), 5);
assertEquals("Incorrect result", exMech.genExemptionBlob(bbRes), len);
AlgorithmParameters params = null;
exMech.init(key, params);
AlgorithmParameterSpec parSpec = null;
exMech.init(key, parSpec);
key = new SecretKeySpec(new byte[10], "DSA");
try {
exMech.init(key);
fail("ExemptionMechanismException must be thrown");
} catch (ExemptionMechanismException e) {
assertTrue("Empty message", (e.getMessage().length() > 0));
}
try {
exMech.init(key, params);
fail("ExemptionMechanismException must be thrown");
} catch (ExemptionMechanismException e) {
assertTrue("Empty message", (e.getMessage().length() > 0));
}
try {
exMech.init(key, parSpec);
fail("ExemptionMechanismException must be thrown");
} catch (ExemptionMechanismException e) {
assertTrue("Empty message", (e.getMessage().length() > 0));
}
}
/**
* Test for <code>getInstance(String algorithm)</code> method
* Assertions:
* throws NullPointerException when algorithm is null;
* throws NoSuchAlgorithmException when algorithm is incorrect;
* returns ExemptionMechanism object
*/
public void testGetInstance01() throws NoSuchAlgorithmException,
ExemptionMechanismException, InvalidAlgorithmParameterException,
ShortBufferException, InvalidKeyException {
try {
ExemptionMechanism.getInstance(null);
fail("NullPointerException or NoSuchAlgorithmException should be thrown if algorithm is null");
} catch (NullPointerException e) {
} catch (NoSuchAlgorithmException e) {
}
for (int i = 0; i < invalidValues.length; i++) {
try {
ExemptionMechanism.getInstance(invalidValues[i]);
fail("NoSuchAlgorithmException must be thrown (algorithm: "
.concat(invalidValues[i]).concat(")"));
} catch (NoSuchAlgorithmException e) {
}
}
ExemptionMechanism exMech;
for (int i = 0; i < validValues.length; i++) {
exMech = ExemptionMechanism.getInstance(validValues[i]);
assertEquals("Incorrect algorithm", exMech.getName(),
validValues[i]);
assertEquals("Incorrect provider", exMech.getProvider(), mProv);
checkResult(exMech);
}
}
/**
* Test for <code>getInstance(String algorithm, String provider)</code>
* method
* Assertions:
* throws NullPointerException when algorithm is null;
* throws NoSuchAlgorithmException when algorithm is null or incorrect;
* throws IllegalArgumentException when provider is null;
* throws NoSuchProviderException when provider is available;
* returns ExemptionMechanism object
*/
public void testGetInstance02() throws NoSuchAlgorithmException,
NoSuchProviderException, IllegalArgumentException,
ExemptionMechanismException, InvalidAlgorithmParameterException,
ShortBufferException, InvalidKeyException {
try {
ExemptionMechanism.getInstance(null, mProv.getName());
fail("NullPointerException or NoSuchAlgorithmException should be thrown if algorithm is null");
} catch (NullPointerException e) {
} catch (NoSuchAlgorithmException e) {
}
for (int i = 0; i < invalidValues.length; i++) {
try {
ExemptionMechanism.getInstance(invalidValues[i], mProv
.getName());
fail("NoSuchAlgorithmException must be thrown (algorithm: "
.concat(invalidValues[i]).concat(")"));
} catch (NoSuchAlgorithmException e) {
}
}
String prov = null;
for (int i = 0; i < validValues.length; i++) {
try {
ExemptionMechanism.getInstance(validValues[i], prov);
fail("IllegalArgumentException must be thrown when provider is null (algorithm: "
.concat(invalidValues[i]).concat(")"));
} catch (IllegalArgumentException e) {
}
}
for (int i = 0; i < validValues.length; i++) {
for (int j = 1; j < invalidValues.length; j++) {
try {
ExemptionMechanism.getInstance(validValues[i],
invalidValues[j]);
fail("NoSuchProviderException must be thrown (algorithm: "
.concat(invalidValues[i]).concat(" provider: ")
.concat(invalidValues[j]).concat(")"));
} catch (NoSuchProviderException e) {
}
}
}
ExemptionMechanism exMech;
for (int i = 0; i < validValues.length; i++) {
exMech = ExemptionMechanism.getInstance(validValues[i], mProv
.getName());
assertEquals("Incorrect algorithm", exMech.getName(),
validValues[i]);
assertEquals("Incorrect provider", exMech.getProvider().getName(),
mProv.getName());
checkResult(exMech);
}
}
/**
* Test for <code>getInstance(String algorithm, Provider provider)</code>
* method
* Assertions:
* throws NullPointerException when algorithm is null;
* throws NoSuchAlgorithmException when algorithm is null or incorrect;
* throws IllegalArgumentException when provider is null;
* returns ExemptionMechanism object
*/
public void testGetInstance03() throws NoSuchAlgorithmException,
IllegalArgumentException,
ExemptionMechanismException, InvalidAlgorithmParameterException,
ShortBufferException, InvalidKeyException {
try {
ExemptionMechanism.getInstance(null, mProv);
fail("NullPointerException or NoSuchAlgorithmException should be thrown if algorithm is null");
} catch (NullPointerException e) {
} catch (NoSuchAlgorithmException e) {
}
for (int i = 0; i < invalidValues.length; i++) {
try {
ExemptionMechanism.getInstance(invalidValues[i], mProv);
fail("NoSuchAlgorithmException must be thrown (algorithm: "
.concat(invalidValues[i]).concat(")"));
} catch (NoSuchAlgorithmException e) {
}
}
Provider prov = null;
for (int i = 0; i < validValues.length; i++) {
try {
ExemptionMechanism.getInstance(validValues[i], prov);
fail("IllegalArgumentException must be thrown when provider is null (algorithm: "
.concat(invalidValues[i]).concat(")"));
} catch (IllegalArgumentException e) {
}
}
ExemptionMechanism exMech;
for (int i = 0; i < validValues.length; i++) {
exMech = ExemptionMechanism.getInstance(validValues[i], mProv);
assertEquals("Incorrect algorithm", exMech.getName(),
validValues[i]);
assertEquals("Incorrect provider", exMech.getProvider(), mProv);
checkResult(exMech);
}
}
}
class myEM extends ExemptionMechanism {
public myEM(ExemptionMechanismSpi spi, Provider prov, String mechanism) {
super(spi, prov, mechanism);
}
public void finalize() {
try {
super.finalize();
} catch (Throwable e) {
throw new RuntimeException("finalize was broken", e);
}
}
}