rizin/test/unit/test_base64.c
2021-02-03 15:40:29 -01:00

62 lines
1.8 KiB
C

#include <rz_util.h>
#include "minunit.h"
bool test_rz_base64_decode_dyn(void) {
char *hello = (char *)rz_base64_decode_dyn("aGVsbG8=", -1);
mu_assert_streq(hello, "hello", "base64_decode_dyn");
free(hello);
mu_end;
}
bool test_rz_base64_decode(void) {
ut8 *hello = malloc(50);
int status = rz_base64_decode(hello, "aGVsbG8=", -1);
mu_assert_eq(status, (int)strlen("hello"), "valid base64 decoding");
mu_assert_streq((char *)hello, "hello", "base64 decoding");
free(hello);
mu_end;
}
bool test_rz_base64_decode_invalid(void) {
ut8 *hello = malloc(50);
int status = rz_base64_decode(hello, "\x01\x02\x03\x04\x00", -1);
// Returns the length of the decoded string, 0 == invalid input.
mu_assert_eq(status, -1, "invalid base64 decoding");
free(hello);
mu_end;
}
int test_rz_base64_encode_dyn(void) {
char *hello = rz_base64_encode_dyn((const ut8 *)"hello", 6);
mu_assert_streq(hello, "aGVsbG8A", "base64_encode_dyn");
free(hello);
hello = rz_base64_encode_dyn((const ut8 *)"hello1", 7);
mu_assert_streq(hello, "aGVsbG8xAA==", "base64_encode_dyn");
free(hello);
hello = rz_base64_encode_dyn((const ut8 *)"hello12", 8);
mu_assert_streq(hello, "aGVsbG8xMgA=", "base64_encode_dyn");
free(hello);
hello = rz_base64_encode_dyn((const ut8 *)"hello123", 9);
mu_assert_streq(hello, "aGVsbG8xMjMA", "base64_encode_dyn");
free(hello);
mu_end;
}
int test_rz_base64_encode(void) {
char *hello = malloc(50);
rz_base64_encode(hello, (const ut8 *)"hello", 5);
mu_assert_streq(hello, "aGVsbG8=", "base64_encode");
free(hello);
mu_end;
}
int all_tests() {
mu_run_test(test_rz_base64_decode_dyn);
mu_run_test(test_rz_base64_decode);
mu_run_test(test_rz_base64_decode_invalid);
mu_run_test(test_rz_base64_encode_dyn);
mu_run_test(test_rz_base64_encode);
return tests_passed != tests_run;
}
mu_main(all_tests)