75 lines
2.1 KiB
C
75 lines
2.1 KiB
C
|
#include "crypto.h"
|
||
|
|
#include <string.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
|
||
|
|
#define mu_assert(message, test) do { if (!(test)) return message; } while (0)
|
||
|
|
#define mu_run_test(test) do { char *message = test(); tests_run++; \
|
||
|
|
if (message) return message; } while (0)
|
||
|
|
int tests_run;
|
||
|
|
|
||
|
|
static char* testRoundtrip() {
|
||
|
|
char* text = strdup("DIESXISTXEINXTEST");
|
||
|
|
char* result = (char*) malloc(strlen(text) + 1);
|
||
|
|
char* result2 = (char*) malloc(strlen(text) + 1);
|
||
|
|
|
||
|
|
KEY key = { 1, "KEY" };
|
||
|
|
encrypt(key, text, result);
|
||
|
|
decrypt(key, result, result2);
|
||
|
|
|
||
|
|
mu_assert("Roundtrip failed", strcmp(result2, text) == 0);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static char* testIllegalKey() {
|
||
|
|
char* text = strdup("DIESXISTXEINXTEST");
|
||
|
|
char* result = (char*) malloc(strlen(text) + 1);
|
||
|
|
int error;
|
||
|
|
|
||
|
|
KEY key1 = { 1, "" };
|
||
|
|
error = encrypt(key1, text, result);
|
||
|
|
mu_assert("Zero key not detected", error == E_KEY_TOO_SHORT);
|
||
|
|
|
||
|
|
KEY key2 = { 1, "abcd"};
|
||
|
|
error = encrypt(key2, text, result);
|
||
|
|
mu_assert("Illegal char in key not detected: abcd", error == E_KEY_ILLEGAL_CHAR);
|
||
|
|
|
||
|
|
KEY key3 = { 1, "@[\\]^_"};
|
||
|
|
error = encrypt(key3, text, result);
|
||
|
|
mu_assert("Illegal char in key not detected: @[\\]^_", error == E_KEY_ILLEGAL_CHAR);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static char* textIllegalMessage() {
|
||
|
|
KEY key1 = { 1, "KEY" };
|
||
|
|
int error = encrypt(key1, "ABCD AB", NULL);
|
||
|
|
mu_assert("Illegal char in message not detected: ABCD AB", error == E_MESSAGE_ILLEGAL_CHAR);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static char* textIllegalCypher() {
|
||
|
|
KEY key1 = { 1, "KEY" };
|
||
|
|
int error = decrypt(key1, "ABCD AB", NULL);
|
||
|
|
mu_assert("Illegal char in cypher not detected: ABCD AB", error == E_CYPHER_ILLEGAL_CHAR);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static char* allTests() {
|
||
|
|
mu_run_test(testRoundtrip);
|
||
|
|
mu_run_test(testIllegalKey);
|
||
|
|
mu_run_test(textIllegalMessage);
|
||
|
|
mu_run_test(textIllegalCypher);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int main() {
|
||
|
|
char *result = allTests();
|
||
|
|
|
||
|
|
if (result != 0) printf("%s\n", result);
|
||
|
|
else printf("ALL TESTS PASSED\n");
|
||
|
|
|
||
|
|
printf("Tests run: %d\n", tests_run);
|
||
|
|
|
||
|
|
return result != 0;
|
||
|
|
}
|