Home | History | Annotate | Download | only in regression
      1 /*
      2  * Copyright (C) 2010 Google Inc.
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  * http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package benchmarks.regression;
     18 
     19 import com.google.caliper.Param;
     20 import com.google.caliper.SimpleBenchmark;
     21 import com.android.org.bouncycastle.crypto.Digest;
     22 
     23 public class DigestBenchmark extends SimpleBenchmark {
     24 
     25     private static final int DATA_SIZE = 8192;
     26     private static final byte[] DATA = new byte[DATA_SIZE];
     27     static {
     28         for (int i = 0; i < DATA_SIZE; i++) {
     29             DATA[i] = (byte)i;
     30         }
     31     }
     32 
     33     @Param private Algorithm algorithm;
     34 
     35     public enum Algorithm { MD5, SHA1, SHA256,  SHA384, SHA512 };
     36 
     37     @Param private Implementation implementation;
     38 
     39     public enum Implementation { OPENSSL, BOUNCYCASTLE };
     40 
     41     private Class<? extends Digest> digestClass;
     42 
     43     @Override protected void setUp() throws Exception {
     44         String className = "com.android.org.bouncycastle.crypto.digests.";
     45         switch (implementation) {
     46             case OPENSSL:
     47                 className += ("OpenSSLDigest$" + algorithm);
     48                 break;
     49             case BOUNCYCASTLE:
     50                 className += (algorithm + "Digest");
     51                 break;
     52             default:
     53                 throw new RuntimeException(implementation.toString());
     54         }
     55         this.digestClass = (Class<? extends Digest>)Class.forName(className);
     56     }
     57 
     58     public void time(int reps) throws Exception {
     59         for (int i = 0; i < reps; ++i) {
     60             Digest digest = digestClass.newInstance();
     61             digest.update(DATA, 0, DATA_SIZE);
     62             digest.doFinal(new byte[digest.getDigestSize()], 0);
     63         }
     64     }
     65 }
     66