| /* |
| * 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.commons.crypto; |
| |
| import java.nio.ByteBuffer; |
| import java.util.Properties; |
| |
| import javax.crypto.Cipher; |
| import javax.crypto.spec.IvParameterSpec; |
| import javax.crypto.spec.SecretKeySpec; |
| |
| import org.apache.commons.crypto.cipher.CryptoCipher; |
| import org.apache.commons.crypto.cipher.CryptoCipherFactory; |
| import org.apache.commons.crypto.random.CryptoRandom; |
| import org.apache.commons.crypto.random.CryptoRandomFactory; |
| |
| import static org.junit.jupiter.api.Assertions.assertEquals; |
| |
| public abstract class AbstractBenchmark { |
| |
| private static final byte[] KEY = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, |
| 0x09, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16 }; |
| private static final byte[] IV = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, |
| 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 }; |
| private static final SecretKeySpec keySpec = new SecretKeySpec(KEY, "AES"); |
| private static final IvParameterSpec ivSpec = new IvParameterSpec(IV); |
| private static final byte[] BUFFER = new byte[1000]; |
| |
| public AbstractBenchmark() { |
| super(); |
| } |
| |
| protected void random(final String cipherClass) throws Exception { |
| final CryptoRandom random = getRandom(cipherClass); |
| random.nextBytes(new byte[1000]); |
| random.nextBytes(new byte[1000]); |
| random.close(); |
| } |
| |
| protected void encipher(final String cipherClass) throws Exception { |
| final CryptoCipher enCipher = getCipher(cipherClass); |
| enCipher.init(Cipher.ENCRYPT_MODE, keySpec, ivSpec); |
| final int bufferSize = 1024; |
| final ByteBuffer inBuffer = ByteBuffer.allocateDirect(bufferSize); |
| final ByteBuffer outBuffer = ByteBuffer.allocateDirect(bufferSize); |
| inBuffer.put(BUFFER); |
| inBuffer.flip(); |
| enCipher.doFinal(inBuffer, outBuffer); |
| enCipher.close(); |
| } |
| |
| protected CryptoRandom getRandom(final String className) throws Exception { |
| final Properties props = new Properties(); |
| props.setProperty(CryptoRandomFactory.CLASSES_KEY, className); |
| final CryptoRandom cryptoRandom = CryptoRandomFactory.getCryptoRandom(props); |
| assertEquals(className, cryptoRandom.getClass().getCanonicalName()); |
| return cryptoRandom; |
| } |
| |
| protected CryptoCipher getCipher(final String className) throws Exception { |
| final Properties properties = new Properties(); |
| properties.setProperty(CryptoCipherFactory.CLASSES_KEY, className); |
| final CryptoCipher cipher = CryptoCipherFactory.getCryptoCipher("AES/CBC/PKCS5Padding", properties); |
| assertEquals(className, cipher.getClass().getCanonicalName()); |
| return cipher; |
| } |
| |
| } |