Inside Bitcoin

1 /*********************************************************************
2 * Copyright (c) 2016 Pieter Wuille                                   *
3 * Distributed under the MIT software license, see the accompanying   *
4 * file COPYING or http://www.opensource.org/licenses/mit-license.php.*
5 **********************************************************************/
6
7#include "ctaes.h"
8
9#include <stdio.h>
10#include <string.h>
11#include <assert.h>
12
13typedef struct {
14    int keysize;
15    const char* key;
16    const char* plain;
17    const char* cipher;
18} ctaes_test;
19
20static const ctaes_test ctaes_tests[] = {
21    /* AES test vectors from FIPS 197. */
22    {128, "000102030405060708090a0b0c0d0e0f", "00112233445566778899aabbccddeeff", "69c4e0d86a7b0430d8cdb78070b4c55a"},
23    {192, "000102030405060708090a0b0c0d0e0f1011121314151617", "00112233445566778899aabbccddeeff", "dda97ca4864cdfe06eaf70a0ec0d7191"},
24    {256, "000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f", "00112233445566778899aabbccddeeff", "8ea2b7ca516745bfeafc49904b496089"},
25
26    /* AES-ECB test vectors from NIST sp800-38a. */
27    {128, "2b7e151628aed2a6abf7158809cf4f3c", "6bc1bee22e409f96e93d7e117393172a", "3ad77bb40d7a3660a89ecaf32466ef97"},
28    {128, "2b7e151628aed2a6abf7158809cf4f3c", "ae2d8a571e03ac9c9eb76fac45af8e51", "f5d3d58503b9699de785895a96fdbaaf"},
29    {128, "2b7e151628aed2a6abf7158809cf4f3c", "30c81c46a35ce411e5fbc1191a0a52ef", "43b1cd7f598ece23881b00e3ed030688"},
30    {128, "2b7e151628aed2a6abf7158809cf4f3c", "f69f2445df4f9b17ad2b417be66c3710", "7b0c785e27e8ad3f8223207104725dd4"},
31    {192, "8e73b0f7da0e6452c810f32b809079e562f8ead2522c6b7b", "6bc1bee22e409f96e93d7e117393172a", "bd334f1d6e45f25ff712a214571fa5cc"},
32    {192, "8e73b0f7da0e6452c810f32b809079e562f8ead2522c6b7b", "ae2d8a571e03ac9c9eb76fac45af8e51", "974104846d0ad3ad7734ecb3ecee4eef"},
33    {192, "8e73b0f7da0e6452c810f32b809079e562f8ead2522c6b7b", "30c81c46a35ce411e5fbc1191a0a52ef", "ef7afd2270e2e60adce0ba2face6444e"},
34    {192, "8e73b0f7da0e6452c810f32b809079e562f8ead2522c6b7b", "f69f2445df4f9b17ad2b417be66c3710", "9a4b41ba738d6c72fb16691603c18e0e"},
35    {256, "603deb1015ca71be2b73aef0857d77811f352c073b6108d72d9810a30914dff4", "6bc1bee22e409f96e93d7e117393172a", "f3eed1bdb5d2a03c064b5a7e3db181f8"},
36    {256, "603deb1015ca71be2b73aef0857d77811f352c073b6108d72d9810a30914dff4", "ae2d8a571e03ac9c9eb76fac45af8e51", "591ccb10d410ed26dc5ba74a31362870"},
37    {256, "603deb1015ca71be2b73aef0857d77811f352c073b6108d72d9810a30914dff4", "30c81c46a35ce411e5fbc1191a0a52ef", "b6ed21b99ca6f4f9f153e7b1beafed1d"},
38    {256, "603deb1015ca71be2b73aef0857d77811f352c073b6108d72d9810a30914dff4", "f69f2445df4f9b17ad2b417be66c3710", "23304b7a39f9f3ff067d8d8f9e24ecc7"}
39};
40
41static void from_hex(unsigned char* data, int len, const char* hex) {
42    int p;
43    for (p = 0; p < len; p++) {
44        int v = 0;
45        int n;
46        for (n = 0; n < 2; n++) {
47            assert((*hex >= '0' && *hex <= '9') || (*hex >= 'a' && *hex <= 'f'));
48            if (*hex >= '0' && *hex <= '9') {
49                v |= (*hex - '0') << (4 * (1 - n));
50            } else {
51                v |= (*hex - 'a' + 10) << (4 * (1 - n));
52            }
53            hex++;
54        }
55        *(data++) = v;
56    }
57    assert(*hex == 0);
58}
59
60int main(void) {
61    int i;
62    int fail = 0;
63    for (i = 0; i < sizeof(ctaes_tests) / sizeof(ctaes_tests[0]); i++) {
64        unsigned char key[32], plain[16], cipher[16], ciphered[16], deciphered[16];
65        const ctaes_test* test = &ctaes_tests[i];
66        assert(test->keysize == 128 || test->keysize == 192 || test->keysize == 256);
67        from_hex(plain, 16, test->plain);
68        from_hex(cipher, 16, test->cipher);
69        switch (test->keysize) {
70            case 128: {
71                AES128_ctx ctx;
72                from_hex(key, 16, test->key);
73                AES128_init(&ctx, key);
74                AES128_encrypt(&ctx, 1, ciphered, plain);
75                AES128_decrypt(&ctx, 1, deciphered, cipher);
76                break;
77            }
78            case 192: {
79                AES192_ctx ctx;
80                from_hex(key, 24, test->key);
81                AES192_init(&ctx, key);
82                AES192_encrypt(&ctx, 1, ciphered, plain);
83                AES192_decrypt(&ctx, 1, deciphered, cipher);
84                break;
85            }
86            case 256: {
87                AES256_ctx ctx;
88                from_hex(key, 32, test->key);
89                AES256_init(&ctx, key);
90                AES256_encrypt(&ctx, 1, ciphered, plain);
91                AES256_decrypt(&ctx, 1, deciphered, cipher);
92                break;
93            }
94        }
95        if (memcmp(cipher, ciphered, 16)) {
96            fprintf(stderr, "E(key=\"%s\", plain=\"%s\") != \"%s\"\n", test->key, test->plain, test->cipher);
97            fail++;
98        }
99        if (memcmp(plain, deciphered, 16)) {
100            fprintf(stderr, "D(key=\"%s\", cipher=\"%s\") != \"%s\"\n", test->key, test->cipher, test->plain);
101            fail++;
102        }
103    }
104    if (fail == 0) {
105        fprintf(stderr, "All tests successful\n");
106    } else {
107        fprintf(stderr, "%i tests failed\n", fail);
108    }
109    return (fail != 0);
110}
111