1 files changed, 44 insertions, 0 deletions
diff --git a/source/tests/test_crypt.c b/source/tests/test_crypt.c
new file mode 100644
index 0000000..a2ebd49
--- /dev/null
+++ b/source/tests/test_crypt.c
@@ -0,0 +1,44 @@
+#include "tests.h"
+
+#include "utils.h"
+#include "crypt.h"
+
+
+#define MIN_BUFSIZ 8192
+#define MAX_BUFSIZ 65536
+
+
+BOOL test_crypt(void)
+{
+    uint32_t key[8], iv[8];
+    size_t ivkeysize = 0, maxsiz = 0;
+
+    maxsiz = MIN_BUFSIZ + (__rdtsc() % (MAX_BUFSIZ-MIN_BUFSIZ+1));
+    ivkeysize = 1 + (__rdtsc() % (sizeof(key)/sizeof(key[0])));
+
+    char* randstr = test_randstring(maxsiz);
+    ERRETCP( randstr != NULL );
+    size_t randlen = strlen(randstr);
+    ERRETCP( maxsiz == randlen );
+    ERRETCP( randlen >= MIN_BUFSIZ && MAX_BUFSIZ >= randlen );
+
+    for (size_t i = 0; i < ivkeysize; ++i) {
+        while(key[i] == 0) key[i] = xor32_randomkey();
+        while(iv[i] == 0) iv[i] = xor32_randomkey();
+    }
+
+    size_t encsiz = maxsiz + (ivkeysize*sizeof(key[0]));
+    char* encBuf = calloc(encsiz, sizeof(char));
+    for (size_t i = 0; i < encsiz; ++i)
+        ERRETCPDW_NOLOG( *(encBuf + i) == 0x0, *(encBuf + i) );
+    memcpy(encBuf, randstr, randlen);
+    size_t newsiz = xor32n_pcbc_crypt_buf((uint32_t*)encBuf, maxsiz, &iv[0], &key[0], ivkeysize);
+    ERRETCP( memcmp(encBuf, randstr, maxsiz) != 0 );
+    size_t oldsiz = xor32n_pcbc_crypt_buf((uint32_t*)encBuf, newsiz, &iv[0], &key[0], ivkeysize);
+    ERRETCP( oldsiz == newsiz );
+    ERRETCP( memcmp(encBuf, randstr, maxsiz) == 0 );
+    free(encBuf);
+
+    COMPAT(free)(randstr);
+    return TRUE;
+}