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.android.org.bouncycastle.crypto.Digest;
     20 import com.google.caliper.BeforeExperiment;
     21 import com.google.caliper.Param;
     22 
     23 public class DigestBenchmark {
     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     @BeforeExperiment
     44     protected void setUp() throws Exception {
     45         String className = "com.android.org.bouncycastle.crypto.digests.";
     46         switch (implementation) {
     47             case OPENSSL:
     48                 className += ("OpenSSLDigest$" + algorithm);
     49                 break;
     50             case BOUNCYCASTLE:
     51                 className += (algorithm + "Digest");
     52                 break;
     53             default:
     54                 throw new RuntimeException(implementation.toString());
     55         }
     56         this.digestClass = (Class<? extends Digest>)Class.forName(className);
     57     }
     58 
     59     public void time(int reps) throws Exception {
     60         for (int i = 0; i < reps; ++i) {
     61             Digest digest = digestClass.newInstance();
     62             digest.update(DATA, 0, DATA_SIZE);
     63             digest.doFinal(new byte[digest.getDigestSize()], 0);
     64         }
     65     }
     66 }
     67