Add small benchmark to the native cipher suite - see #755

native/src/test/java/net/md_5/bungee/NativeCipherTest.java

@@ -4,9 +4,12 @@ import io.netty.buffer.ByteBuf;
 import io.netty.buffer.Unpooled;
 import org.junit.Assert;
 import org.junit.Test;
+import org.junit.FixMethodOrder;
+import org.junit.runners.MethodSorters;
 import javax.crypto.SecretKey;
 import javax.crypto.spec.SecretKeySpec;
 
+@FixMethodOrder(MethodSorters.NAME_ASCENDING)
 public class NativeCipherTest
 {
 
@@ -16,6 +19,7 @@ public class NativeCipherTest
         50, -7, 89, 1, -11, -32, -118, -48, -2, -72, 105, 97, -70, -81
     };
     private final SecretKey secret = new SecretKeySpec( new byte[ 16 ], "AES" );
+    private static final int BENCHMARK_COUNT = 50000;
 
     @Test
     public void testOpenSSL() throws Exception
@@ -31,6 +35,21 @@ public class NativeCipherTest
         }
     }
 
+    @Test
+    public void testOpenSSLBenchmark() throws Exception
+    {
+        if ( NativeCipher.isSupported() )
+        {
+            boolean loaded = NativeCipher.load();
+            Assert.assertTrue( "Native cipher failed to load!", loaded );
+
+            NativeCipher cipher = new NativeCipher();
+
+            System.out.println( "Benchmarking OpenSSL cipher..." );
+            testBenchmark( cipher );
+        }
+    }
+
     @Test
     public void testJDK() throws Exception
     {
@@ -41,6 +60,16 @@ public class NativeCipherTest
         testACipher( cipher );
     }
 
+    @Test
+    public void testJDKBenchmark() throws Exception
+    {
+        // Create JDK cipher
+        BungeeCipher cipher = new FallbackCipher();
+
+        System.out.println( "Benchmarking Java cipher..." );
+        testBenchmark( cipher );
+    }
+
     /**
      * Hackish test which can test both native and fallback ciphers using direct
      * buffers.
@@ -71,4 +100,36 @@ public class NativeCipherTest
 
         System.out.println( "This cipher works correctly!" );
     }
+
+    public void testBenchmark(BungeeCipher cipher) throws Exception
+    {
+        // Create input buf
+        ByteBuf nativePlain = Unpooled.directBuffer( plainBytes.length );
+        nativePlain.writeBytes( plainBytes );
+        // Create expected buf
+        ByteBuf nativeCiphered = Unpooled.directBuffer( cipheredBytes.length );
+        nativeCiphered.writeBytes( cipheredBytes );
+        // Create output buf
+        ByteBuf out = Unpooled.directBuffer( plainBytes.length );
+
+        // Encrypt
+        cipher.init( true, secret );
+        long start = System.currentTimeMillis();
+        for ( int i = 0; i < BENCHMARK_COUNT; i++ )
+        {
+            cipher.cipher( nativePlain, out );
+            out.clear();
+        }
+        System.out.println( String.format( "Encryption Iteration: %d, Elapsed: %d ms", BENCHMARK_COUNT, System.currentTimeMillis() - start ) );
+
+        // Decrypt
+        cipher.init( false, secret );
+        start = System.currentTimeMillis();
+        for ( int i = 0; i < BENCHMARK_COUNT; i++ )
+        {
+            cipher.cipher( nativeCiphered, out );
+            out.clear();
+        }
+        System.out.println( String.format( "Decryption Iteration: %d, Elapsed: %d ms", BENCHMARK_COUNT, System.currentTimeMillis() - start ) );
+    }
 }