197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes/* 297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * Copyright (C) 2012 Google Inc. 397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * 497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * Licensed under the Apache License, Version 2.0 (the "License"); 597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * you may not use this file except in compliance with the License. 697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * You may obtain a copy of the License at 797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * 897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * http://www.apache.org/licenses/LICENSE-2.0 997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * 1097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * Unless required by applicable law or agreed to in writing, software 1197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * distributed under the License is distributed on an "AS IS" BASIS, 1297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * See the License for the specific language governing permissions and 1497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * limitations under the License. 1597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes */ 1697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 1797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughespackage benchmarks.regression; 1897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 19ea13f8291a92b6f47f50011da1d5e8c107984bc3Paul Duffinimport com.google.caliper.BeforeExperiment; 2097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughesimport com.google.caliper.Param; 2197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughesimport java.security.spec.AlgorithmParameterSpec; 2297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughesimport java.util.HashMap; 2397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughesimport java.util.Map; 2497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughesimport javax.crypto.Cipher; 25ea13f8291a92b6f47f50011da1d5e8c107984bc3Paul Duffinimport javax.crypto.KeyGenerator; 26ea13f8291a92b6f47f50011da1d5e8c107984bc3Paul Duffinimport javax.crypto.SecretKey; 2797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughesimport javax.crypto.spec.IvParameterSpec; 2897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 2997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes/** 3097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * Cipher benchmarks. Only runs on AES currently because of the combinatorial 3197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes * explosion of the test as it stands. 3297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes */ 33ea13f8291a92b6f47f50011da1d5e8c107984bc3Paul Duffinpublic class CipherBenchmark { 3497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 3597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private static final int DATA_SIZE = 8192; 3697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private static final byte[] DATA = new byte[DATA_SIZE]; 3797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 3897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private static final int IV_SIZE = 16; 3997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 4097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private static final byte[] IV = new byte[IV_SIZE]; 4197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 4297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes static { 4397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes for (int i = 0; i < DATA_SIZE; i++) { 4497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes DATA[i] = (byte) i; 4597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 4697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes for (int i = 0; i < IV_SIZE; i++) { 4797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes IV[i] = (byte) i; 4897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 4997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 5097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 5197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes @Param private Algorithm algorithm; 5297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 5397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes public enum Algorithm { 5497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes AES, 5597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes }; 5697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 5797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes @Param private Mode mode; 5897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 5997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes public enum Mode { 6097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes CBC, 6197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes CFB, 6297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes CTR, 6397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes ECB, 6497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes OFB, 6597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes }; 6697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 6797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes @Param private Padding padding; 6897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 6997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes public enum Padding { 7097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes NOPADDING, 7197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes PKCS1PADDING, 7297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes }; 7397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 7497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes @Param({"128", "192", "256"}) private int keySize; 7597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 7697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes @Param({"16", "32", "64", "128", "1024", "8192"}) private int inputSize; 7797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 7897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes @Param private Implementation implementation; 7997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 8097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes public enum Implementation { OpenSSL, BouncyCastle }; 8197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 8297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private String providerName; 8397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 8497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes // Key generation isn't part of the benchmark so cache the results 8597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private static Map<Integer, SecretKey> KEY_SIZES = new HashMap<Integer, SecretKey>(); 8697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 8797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private String cipherAlgorithm; 8897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private SecretKey key; 8997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 9097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private byte[] output = new byte[DATA.length]; 9197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 9297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private Cipher cipherEncrypt; 9397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 9497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private Cipher cipherDecrypt; 9597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 9697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes private AlgorithmParameterSpec spec; 9797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 98ea13f8291a92b6f47f50011da1d5e8c107984bc3Paul Duffin @BeforeExperiment 99ea13f8291a92b6f47f50011da1d5e8c107984bc3Paul Duffin protected void setUp() throws Exception { 10097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes cipherAlgorithm = algorithm.toString() + "/" + mode.toString() + "/" 10197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes + padding.toString(); 10297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 10397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes String keyAlgorithm = algorithm.toString(); 10497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes key = KEY_SIZES.get(keySize); 10597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes if (key == null) { 10697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes KeyGenerator generator = KeyGenerator.getInstance(keyAlgorithm); 10797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes generator.init(keySize); 10897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes key = generator.generateKey(); 10997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes KEY_SIZES.put(keySize, key); 11097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 11197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 11297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes switch (implementation) { 11397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes case OpenSSL: 11497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes providerName = "AndroidOpenSSL"; 11597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes break; 11697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes case BouncyCastle: 11797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes providerName = "BC"; 11897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes break; 11997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes default: 12097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes throw new RuntimeException(implementation.toString()); 12197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 12297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 12397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes if (mode != Mode.ECB) { 12497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes spec = new IvParameterSpec(IV); 12597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 12697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 12797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes cipherEncrypt = Cipher.getInstance(cipherAlgorithm, providerName); 12897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes cipherEncrypt.init(Cipher.ENCRYPT_MODE, key, spec); 12997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 13097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes cipherDecrypt = Cipher.getInstance(cipherAlgorithm, providerName); 13197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes cipherDecrypt.init(Cipher.DECRYPT_MODE, key, spec); 13297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 13397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 13497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes public void timeEncrypt(int reps) throws Exception { 13597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes for (int i = 0; i < reps; ++i) { 13697aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes cipherEncrypt.doFinal(DATA, 0, inputSize, output); 13797aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 13897aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 13997aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes 14097aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes public void timeDecrypt(int reps) throws Exception { 14197aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes for (int i = 0; i < reps; ++i) { 14297aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes cipherDecrypt.doFinal(DATA, 0, inputSize, output); 14397aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 14497aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes } 14597aba27f961a5a6f37dcaf7f455df371e250ede3Elliott Hughes} 146