my_crypt unittest

2 files changed, 105 insertions, 1 deletions

diff --git a/unittest/mysys/CMakeLists.txt b/unittest/mysys/CMakeLists.txt
index 8c09a466451..7b7a861781b 100644
--- a/unittest/mysys/CMakeLists.txt
+++ b/unittest/mysys/CMakeLists.txt
@@ -13,7 +13,7 @@
 # along with this program; if not, write to the Free Software
 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 
-MY_ADD_TESTS(bitmap base64 my_vsnprintf my_atomic my_rdtsc lf my_malloc
+MY_ADD_TESTS(bitmap base64 my_vsnprintf my_atomic my_rdtsc lf my_malloc aes
              LINK_LIBRARIES mysys)
 
 MY_ADD_TESTS(ma_dyncol
diff --git a/unittest/mysys/aes-t.c b/unittest/mysys/aes-t.c
new file mode 100644
index 00000000000..83cd39ff325
--- /dev/null
+++ b/unittest/mysys/aes-t.c
@@ -0,0 +1,104 @@
+/* Copyright (c) 2003, 2006, 2007 MySQL AB, 2009 Sun Microsystems, Inc.
+   Use is subject to license terms.
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License as
+   published by the Free Software Foundation; version 2 of the License.
+
+   This program 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 St, Fifth Floor, Boston, MA 02110-1301  USA */
+
+#include <my_global.h>
+#include <my_sys.h>
+#include <my_crypt.h>
+#include <tap.h>
+#include <string.h>
+#include <ctype.h>
+
+#define DO_TEST_P(mode, slen, fill, dlen, hash)                         \
+  do {                                                                  \
+    memset(src, fill, src_len= slen);                                   \
+    ok(my_aes_encrypt_ ## mode(src, src_len, dst, &dst_len,             \
+                     key, sizeof(key), iv, sizeof(iv), 0) == MY_AES_OK, \
+      "encrypt " #mode " %u", src_len);                                 \
+    ok(dst_len <= my_aes_get_size(src_len),                             \
+       "my_aes_get_size(%u) >= %u", src_len, dst_len);                  \
+    my_md5(md5, (char*)dst, dst_len);                                   \
+    ok(dst_len == dlen && memcmp(md5, hash, sizeof(md5)) == 0, "md5");  \
+    ok(my_aes_decrypt_ ## mode(dst, dst_len, ddst, &ddst_len,           \
+                     key, sizeof(key), iv, sizeof(iv), 0) == MY_AES_OK, \
+       "decrypt " #mode " %u", dst_len);                                \
+    ok(ddst_len == src_len && memcmp(src, ddst, src_len) == 0, "memcmp"); \
+  } while(0)
+
+#define DO_TEST_N(mode, slen, fill, dlen, hash)                         \
+  do {                                                                  \
+    memset(src, fill, src_len= slen);                                   \
+    ok(my_aes_encrypt_ ## mode(src, src_len, dst, &dst_len,             \
+                     key, sizeof(key), iv, sizeof(iv), 1) == MY_AES_OK, \
+      "encrypt " #mode " %u nopad", src_len);                           \
+    my_md5(md5, (char*)dst, dst_len);                                   \
+    ok(dst_len == dlen && memcmp(md5, hash, sizeof(md5)) == 0, "md5");  \
+    ok(my_aes_decrypt_ ## mode(dst, dst_len, ddst, &ddst_len,           \
+                     key, sizeof(key), iv, sizeof(iv), 1) == MY_AES_OK, \
+       "decrypt " #mode " %u", dst_len);                                \
+    ok(ddst_len == src_len && memcmp(src, ddst, src_len) == 0, "memcmp"); \
+  } while(0)
+
+#define DO_TEST_X(mode, slen, fill, dlen, hash)                         \
+  do {                                                                  \
+    memset(src, fill, src_len= slen);                                   \
+    ok(my_aes_encrypt_ ## mode(src, src_len, dst, &dst_len,             \
+                     key, sizeof(key), iv, sizeof(iv)) == MY_AES_OK,    \
+      "encrypt " #mode " %u", src_len);                                 \
+    my_md5(md5, (char*)dst, dst_len);                                   \
+    ok(dst_len == dlen && memcmp(md5, hash, sizeof(md5)) == 0, "md5");  \
+    ok(my_aes_decrypt_ ## mode(dst, dst_len, ddst, &ddst_len,           \
+                     key, sizeof(key), iv, sizeof(iv)) == MY_AES_OK,    \
+       "decrypt " #mode " %u", dst_len);                                \
+    ok(ddst_len == src_len && memcmp(src, ddst, src_len) == 0, "memcmp"); \
+  } while(0)
+
+/* useful macro for debugging */
+#define PRINT_MD5()                                     \
+  do {                                                  \
+    uint i;                                             \
+    printf("\"");                                       \
+    for (i=0; i < sizeof(md5); i++)                     \
+      printf("\\x%02x", md5[i]);                        \
+    printf("\"\n");                                     \
+  } while(0);
+
+
+int
+main(int argc __attribute__((unused)),char *argv[])
+{
+  uchar key[16]= {1,2,3,4,5,6,7,8,9,0,1,2,3,4,5,6};
+  uchar iv[16]=  {2,3,4,5,6,7,8,9,0,1,2,3,4,5,6,7};
+  uchar src[1000], dst[1100], ddst[1000];
+  uchar md5[MY_MD5_HASH_SIZE];
+  uint src_len, dst_len, ddst_len;
+
+  MY_INIT(argv[0]);
+
+  plan(44);
+  DO_TEST_P(ecb, 200, '.', 208, "\xd8\x73\x8e\x3a\xbc\x66\x99\x13\x7f\x90\x23\x52\xee\x97\x6f\x9a");
+  DO_TEST_P(ecb, 128, '?', 144, "\x19\x58\x33\x85\x4c\xaa\x7f\x06\xd1\xb2\xec\xd7\xb7\x6a\xa9\x5b");
+  DO_TEST_P(cbc, 159, '%', 160, "\x4b\x03\x18\x3d\xf1\xa7\xcd\xa1\x46\xb3\xc6\x8a\x92\xc0\x0f\xc9");
+  DO_TEST_P(cbc, 192, '@', 208, "\x54\xc4\x75\x1d\xff\xe0\xf6\x80\xf0\x85\xbb\x8b\xda\x07\x21\x17");
+  DO_TEST_N(ecb, 200, '.', 200, "\xbf\xec\x43\xd1\x66\x8d\x01\xad\x3a\x25\xee\xa6\x3d\xc6\xc4\x68");
+  DO_TEST_N(ecb, 128, '?', 128, "\x5b\x44\x20\xf3\xd9\xb4\x9d\x74\x5e\xb7\x5a\x0a\xe7\x32\x35\xc3");
+  DO_TEST_N(cbc, 159, '%', 159, "\xf3\x6e\x40\x00\x3c\x08\xa0\xb1\x2d\x1f\xcf\xce\x54\xc9\x73\x83");
+  DO_TEST_N(cbc, 192, '@', 192, "\x30\xe5\x28\x8c\x4a\x3b\x02\xd7\x56\x40\x59\x25\xac\x58\x09\x22");
+  DO_TEST_X(ctr, 200, '.', 200, "\x5a\x77\x19\xea\x67\x50\xe3\xab\x7f\x39\x6f\xc4\xa8\x09\xc5\x88");
+  DO_TEST_X(gcm, 128, '?', 144, "\x54\x6a\x7c\xa2\x04\xdc\x6e\x80\x1c\xcd\x5f\x7a\x7b\x08\x9e\x9d");
+
+  my_end(0);
+  return exit_status();
+}