tests/test-aes128.c

   1 /*
   2  * Copyright (c) 2009, 2010, 2013, 2014 Nicira, 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 #include <config.h>
  18 #include <ctype.h>
  19 #include "aes128.h"
  20 #include "util.h"
  21 #include "ovstest.h"
  22
  23 static void
  24 hex_to_uint8(const char *input, uint8_t *output, size_t n)
  25 {
  26     size_t i;
  27
  28     if (strlen(input) != n * 2) {
  29         goto error;
  30     }
  31     for (i = 0; i < n; i++) {
  32         bool ok;
  33
  34         output[i] = hexits_value(&input[i * 2], 2, &ok);
  35         if (!ok) {
  36             goto error;
  37         }
  38     }
  39     return;
  40
  41 error:
  42     ovs_fatal(0, "\"%s\" is not exactly %"PRIuSIZE" hex digits", input, n * 2);
  43 }
  44
  45 static void
  46 test_aes128_main(int argc, char *argv[])
  47 {
  48     struct aes128 aes;
  49     uint8_t plaintext[16];
  50     uint8_t ciphertext[16];
  51     uint8_t key[16];
  52     size_t i;
  53
  54     if (argc != 3) {
  55         ovs_fatal(0, "usage: %s KEY PLAINTEXT, where KEY and PLAINTEXT each "
  56                   "consist of 32 hex digits", argv[0]);
  57     }
  58
  59     hex_to_uint8(argv[1], key, 16);
  60     hex_to_uint8(argv[2], plaintext, 16);
  61
  62     aes128_schedule(&aes, key);
  63     aes128_encrypt(&aes, plaintext, ciphertext);
  64     for (i = 0; i < 16; i++) {
  65         printf("%02x", ciphertext[i]);
  66     }
  67     putchar('\n');
  68 }
  69
  70 OVSTEST_REGISTER("test-aes128", test_aes128_main);