examples: add bencryption_bench

examples/CMakeLists.txt

@@ -23,3 +23,6 @@ target_link_libraries(bavl_test security)
 
 add_executable(hashtable_bench hashtable_bench.c)
 target_link_libraries(hashtable_bench system security)
+
+add_executable(bencryption_bench bencryption_bench.c)
+target_link_libraries(bencryption_bench system security)

examples/bencryption_bench.c

@@ -0,0 +1,134 @@
+/**
+ * @file bencryption_bench.c
+ * @author Ambroz Bizjak <ambrop7@gmail.com>
+ * 
+ * @section LICENSE
+ * 
+ * This file is part of BadVPN.
+ * 
+ * BadVPN is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2
+ * as published by the Free Software Foundation.
+ * 
+ * BadVPN is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdint.h>
+#include <limits.h>
+
+#include <security/BRandom.h>
+#include <security/BEncryption.h>
+#include <system/DebugObject.h>
+
+static void usage (char *name)
+{
+    printf(
+        "Usage: %s <enc/dec> <ciper> <num_blocks> <num_ops>\n"
+        "    <cipher> is one of (blowfish, aes).\n",
+        name
+    );
+    
+    exit(1);
+}
+
+int main (int argc, char **argv)
+{
+    if (argc <= 0) {
+        return 1;
+    }
+    
+    if (argc != 5) {
+        usage(argv[0]);
+    }
+    
+    char *mode_str = argv[1];
+    char *cipher_str = argv[2];
+    
+    int mode;
+    int cipher;
+    int num_blocks = atoi(argv[3]);
+    int num_ops = atoi(argv[4]);
+    
+    if (!strcmp(mode_str, "enc")) {
+        mode = BENCRYPTION_MODE_ENCRYPT;
+    }
+    else if (!strcmp(mode_str, "dec")) {
+        mode = BENCRYPTION_MODE_DECRYPT;
+    }
+    else {
+        usage(argv[0]);
+    }
+    
+    if (!strcmp(cipher_str, "blowfish")) {
+        cipher = BENCRYPTION_CIPHER_BLOWFISH;
+    }
+    else if (!strcmp(cipher_str, "aes")) {
+        cipher = BENCRYPTION_CIPHER_AES;
+    }
+    else {
+        usage(argv[0]);
+    }
+    
+    if (num_blocks < 0 || num_ops < 0) {
+        usage(argv[0]);
+    }
+    
+    int key_size = BEncryption_cipher_key_size(cipher);
+    int block_size = BEncryption_cipher_block_size(cipher);
+    
+    uint8_t key[key_size];
+    BRandom_randomize(key, sizeof(key));
+    
+    uint8_t iv[block_size];
+    BRandom_randomize(iv, sizeof(iv));
+    
+    int unit_size = num_blocks * block_size;
+    
+    printf("unit size %d\n", unit_size);
+    
+    uint8_t *buf1 = malloc(unit_size);
+    if (!buf1) {
+        printf("malloc failed");
+        goto fail0;
+    }
+    
+    uint8_t *buf2 = malloc(unit_size);
+    if (!buf2) {
+        printf("malloc failed");
+        goto fail1;
+    }
+    
+    BEncryption enc;
+    BEncryption_Init(&enc, mode, cipher, key);
+    
+    uint8_t *in = buf1;
+    uint8_t *out = buf2;
+    BRandom_randomize(in, unit_size);
+    
+    for (int i = 0; i < num_ops; i++) {
+        BEncryption_Encrypt(&enc, in, out, unit_size, iv);
+        
+        uint8_t *t = in;
+        in = out;
+        out = t;
+    }
+    
+    BEncryption_Free(&enc);
+    free(buf2);
+fail1:
+    free(buf1);
+fail0:
+    DebugObjectGlobal_Finish();
+    
+    return 0;
+}