mirror of
https://git.sr.ht/~thestr4ng3r/chiaki
synced 2025-03-12 05:25:23 -07:00
81 lines
2.4 KiB
C
81 lines
2.4 KiB
C
// SPDX-License-Identifier: LicenseRef-AGPL-3.0-only-OpenSSL
|
|
|
|
#include <munit.h>
|
|
|
|
#include <chiaki/fec.h>
|
|
#include <chiaki/base64.h>
|
|
|
|
typedef struct fec_test_case_t
|
|
{
|
|
unsigned int k;
|
|
unsigned int m;
|
|
const int erasures[0x10];
|
|
const char *frame_buffer_b64;
|
|
const size_t unit_size;
|
|
} FECTestCase;
|
|
|
|
#include "fec_test_cases.inl"
|
|
|
|
static MunitResult test_fec_case(FECTestCase *test_case)
|
|
{
|
|
size_t b64len = strlen(test_case->frame_buffer_b64);
|
|
uint8_t *frame_buffer_ref = malloc(b64len);
|
|
munit_assert_not_null(frame_buffer_ref);
|
|
|
|
size_t stride = ((test_case->unit_size + 0xf) / 0x10) * 0x10;
|
|
size_t frame_buffer_size = stride * (test_case->k + test_case->m);
|
|
uint8_t *frame_buffer = malloc(frame_buffer_size);
|
|
munit_assert_not_null(frame_buffer);
|
|
|
|
ChiakiErrorCode err = chiaki_base64_decode(test_case->frame_buffer_b64, b64len, frame_buffer_ref, &b64len);
|
|
munit_assert_int(err, ==, CHIAKI_ERR_SUCCESS);
|
|
munit_assert_size(b64len, ==, test_case->unit_size * (test_case->k + test_case->m));
|
|
|
|
for(size_t i=0; i<test_case->k + test_case->m; i++)
|
|
memcpy(frame_buffer + i * stride, frame_buffer_ref + i * test_case->unit_size, test_case->unit_size);
|
|
|
|
size_t erasures_count = 0;
|
|
for(const int *e = test_case->erasures; *e >= 0; e++, erasures_count++);
|
|
|
|
// write garbage over erasures
|
|
for(size_t i=0; i<erasures_count; i++)
|
|
{
|
|
unsigned int e = test_case->erasures[i];
|
|
munit_assert_uint(e, <, test_case->k + test_case->m);
|
|
memset(frame_buffer + stride * e, 0x42, test_case->unit_size);
|
|
}
|
|
|
|
err = chiaki_fec_decode(frame_buffer, test_case->unit_size, stride, test_case->k, test_case->m, (const unsigned int *)test_case->erasures, erasures_count);
|
|
munit_assert_int(err, ==, CHIAKI_ERR_SUCCESS);
|
|
|
|
for(size_t i=0; i<test_case->k; i++)
|
|
munit_assert_memory_equal(test_case->unit_size, frame_buffer + i * stride, frame_buffer_ref + i * test_case->unit_size);
|
|
|
|
free(frame_buffer);
|
|
free(frame_buffer_ref);
|
|
return MUNIT_OK;
|
|
}
|
|
|
|
static MunitParameterEnum fec_params[] = {
|
|
{ "test_case", fec_test_case_ids },
|
|
{ NULL, NULL },
|
|
};
|
|
|
|
static MunitResult test_fec(const MunitParameter params[], void *test_user)
|
|
{
|
|
unsigned long test_case_id = strtoul(params[0].value, NULL, 0);
|
|
return test_fec_case(&fec_test_cases[test_case_id]);
|
|
}
|
|
|
|
MunitTest tests_fec[] = {
|
|
{
|
|
"/fec",
|
|
test_fec,
|
|
NULL,
|
|
NULL,
|
|
MUNIT_TEST_OPTION_NONE,
|
|
fec_params
|
|
},
|
|
{ NULL, NULL, NULL, NULL, MUNIT_TEST_OPTION_NONE, NULL }
|
|
};
|