mirror of https://github.com/ericonr/purr-c.git
105 lines
3.4 KiB
C
105 lines
3.4 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include <sys/random.h>
|
|
|
|
#include "purr.h"
|
|
#include "mmap_file.h"
|
|
|
|
static int compare_strings(const char *expected, const char *result, const char *function)
|
|
{
|
|
int rv = 0;
|
|
|
|
printf("%s(): ", function);
|
|
if (strcmp(expected, result)) {
|
|
rv = 1;
|
|
puts("failure");
|
|
printf("expected: %s\ngot: %s\n", expected, result);
|
|
} else {
|
|
puts("success");
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
|
|
static int compare_arrays(const uint8_t *expected, const uint8_t *result, size_t len, const char *function)
|
|
{
|
|
int rv = 0;
|
|
|
|
printf("%s(): ", function);
|
|
if (memcmp(expected, result, len)) {
|
|
rv = 1;
|
|
puts("failure");
|
|
printf("expected: %s\ngot: %s\n", print_hex(expected, len, false), print_hex(result, len, false));
|
|
} else {
|
|
puts("success");
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
|
|
int main()
|
|
{
|
|
int rv = 0;
|
|
|
|
{
|
|
/* formats.c */
|
|
uint8_t buf[] = {0x12, 0x02, 0x12, 0x4c, 0xa8};
|
|
const char *expected = "1202124ca8";
|
|
const char *result = print_hex(buf, sizeof buf, false);
|
|
rv = compare_strings(expected, result, "print_hex") ? 1 : rv;
|
|
}
|
|
|
|
{
|
|
/* urls.c */
|
|
const char *dirty = "https://hello.com/ash";
|
|
char clean[4096], path[4096], port[16];
|
|
int portn = clean_up_link(dirty, clean, path, port);
|
|
rv = compare_strings("hello.com", clean, "clean_up_link") ? 1 : rv;
|
|
rv = compare_strings("/ash", path, "clean_up_link") ? 1 : rv;
|
|
rv = compare_strings("443", port, "clean_up_link") ? 1 : rv;
|
|
assert(portn == HTTPS_PORT);
|
|
|
|
dirty = "http://hello.com";
|
|
portn = clean_up_link(dirty, clean, path, port);
|
|
rv = compare_strings("hello.com", clean, "clean_up_link") ? 1 : rv;
|
|
rv = compare_strings("/", path, "clean_up_link") ? 1 : rv;
|
|
rv = compare_strings("80", port, "clean_up_link") ? 1 : rv;
|
|
assert(portn == HTTP_PORT);
|
|
|
|
dirty = "https://bsd.ac/paste.html#sieqaqk_73fe_df51";
|
|
portn = clean_up_link(dirty, clean, path, port);
|
|
rv = compare_strings("bsd.ac", clean, "clean_up_link") ? 1 : rv;
|
|
rv = compare_strings("/paste.html#sieqaqk_73fe_df51", path, "clean_up_link") ? 1 : rv;
|
|
rv = compare_strings("443", port, "clean_up_link") ? 1 : rv;
|
|
assert(portn == HTTPS_PORT);
|
|
uint8_t key_exc[KEY_LEN] = {0x73, 0xfe};
|
|
uint8_t iv_exc[IV_LEN] = {0xdf, 0x51};
|
|
uint8_t *key, *iv;
|
|
int err = get_encryption_params(path, &key, &iv);
|
|
rv = compare_strings("/sieqaqk", path, "get_encryption_params") ? 1 : rv;
|
|
rv = compare_arrays(key_exc, key, KEY_LEN, "get_encryption_params") ? 1 : rv;
|
|
rv = compare_arrays(iv_exc, iv, IV_LEN, "get_encryption_params") ? 1 : rv;
|
|
assert(err == 0);
|
|
}
|
|
|
|
{
|
|
/* mmap_file.c */
|
|
int write_size = 1024 * 1024;
|
|
struct mmap_file f = {.size = 2 * write_size, .prot = PROT_MEM, .flags = MAP_MEM};
|
|
f.data = mmap(NULL, f.size, f.prot, f.flags, -1, 0);
|
|
assert(!ERROR_MMAP(f));
|
|
uint8_t *data = malloc(write_size);
|
|
getrandom(data, write_size, 0);
|
|
assert(data);
|
|
int written = write_into_mmap(&f, data, write_size);
|
|
assert(f.offset == written);
|
|
RESET_MMAP(f);
|
|
read_from_mmap(&f, write_size);
|
|
rv = compare_arrays(data, f.data, write_size, "{write_into,read_from}_mmap") ? 1 : rv;
|
|
}
|
|
|
|
return rv;
|
|
}
|