*/
-/* ! \file */
-
#include <config.h>
-#include <atf-c.h>
+#if HAVE_CMOCKA
+
+#include <stdarg.h>
+#include <stddef.h>
+#include <setjmp.h>
+#include <stdlib.h>
#include <inttypes.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/mman.h>
+#define UNIT_TESTING
+#include <cmocka.h>
+
#include <isc/mem.h>
#include <isc/print.h>
#include <isc/random.h>
#define MAP_FILE 0
#endif
+static int
+_setup(void **state) {
+ isc_result_t result;
+
+ UNUSED(state);
+
+ result = dns_test_begin(NULL, false);
+ assert_int_equal(result, ISC_R_SUCCESS);
+
+ return (0);
+}
+
+static int
+_teardown(void **state) {
+ UNUSED(state);
+
+ dns_test_end();
+
+ return (0);
+}
+
typedef struct data_holder {
int len;
const char *data;
(data->len != 0 && data->data != NULL));
result = isc_stdio_tell(file, &where);
- if (result != ISC_R_SUCCESS)
+ if (result != ISC_R_SUCCESS) {
return (result);
+ }
temp = *data;
temp.data = (data->len == 0
isc_crc64_update(crc, (void *)&temp, sizeof(temp));
ret = fwrite(&temp, sizeof(data_holder_t), 1, file);
- if (ret != 1)
+ if (ret != 1) {
return (ISC_R_FAILURE);
+ }
if (data->len > 0) {
isc_crc64_update(crc, (const void *)data->data, data->len);
ret = fwrite(data->data, data->len, 1, file);
- if (ret != 1)
+ if (ret != 1) {
return (ISC_R_FAILURE);
+ }
}
return (ISC_R_SUCCESS);
}
static isc_result_t
-fix_data(dns_rbtnode_t *p, void *base, size_t max, void *arg,
- uint64_t *crc)
-{
+fix_data(dns_rbtnode_t *p, void *base, size_t max, void *arg, uint64_t *crc) {
data_holder_t *data = p->data;
size_t size;
REQUIRE(p != NULL);
- if (data == NULL)
- printf("fixing data: data NULL\n");
- else
- printf("fixing data: len %d, data %p\n", data->len, data->data);
-
if (data == NULL ||
(data->len == 0 && data->data != NULL) ||
(data->len != 0 && data->data == NULL))
+ {
return (ISC_R_INVALIDFILE);
+ }
size = max - ((char *)p - (char *)base);
if (data->len > (int) size || data->data > (const char *) max) {
- printf("data invalid\n");
return (ISC_R_INVALIDFILE);
}
isc_crc64_update(crc, (void *)data, sizeof(*data));
- data->data = (data->len == 0)
- ? NULL
- : (char *)data + sizeof(data_holder_t);
+ data->data = NULL;
+ if (data->len != 0) {
+ data->data = (char *)data + sizeof(data_holder_t);
+ }
- if (data->len > 0)
+ if (data->len > 0) {
isc_crc64_update(crc, (const void *)data->data, data->len);
+ }
return (ISC_R_SUCCESS);
}
if (name != NULL) {
result = dns_rbt_addname(rbt, name, &testdatap->data);
- ATF_CHECK_STREQ(dns_result_totext(result), "success");
+ assert_int_equal(result, ISC_R_SUCCESS);
}
testdatap++;
}
data = NULL;
result = dns_rbt_findname(rbt, name, 0, foundname,
(void *) &data);
- ATF_CHECK_STREQ(dns_result_totext(result), "success");
+ assert_int_equal(result, ISC_R_SUCCESS);
testdatap++;
}
}
static void
-data_printer(FILE *out, void *datap)
-{
+data_printer(FILE *out, void *datap) {
data_holder_t *data = (data_holder_t *)datap;
fprintf(out, "%d bytes, %s", data->len, data->data);
}
-ATF_TC(serialize);
-ATF_TC_HEAD(serialize, tc) {
- atf_tc_set_md_var(tc, "descr", "Test writing an rbt to file");
-}
-ATF_TC_BODY(serialize, tc) {
+/* Test writing an rbt to file */
+static void
+serialize_test(void **state) {
dns_rbt_t *rbt = NULL;
isc_result_t result;
FILE *rbtfile = NULL;
off_t filesize = 0;
char *base;
- UNUSED(tc);
+ UNUSED(state);
isc_mem_debugging = ISC_MEM_DEBUGRECORD;
- result = dns_test_begin(NULL, true);
- ATF_CHECK_STREQ(dns_result_totext(result), "success");
result = dns_rbt_create(mctx, delete_data, NULL, &rbt);
- ATF_CHECK_STREQ(dns_result_totext(result), "success");
+ assert_int_equal(result, ISC_R_SUCCESS);
add_test_data(mctx, rbt);
/*
* Serialize the tree.
*/
- printf("serialization begins.\n");
rbtfile = fopen("./zone.bin", "w+b");
- ATF_REQUIRE(rbtfile != NULL);
+ assert_non_null(rbtfile);
result = dns_rbt_serialize_tree(rbtfile, rbt, write_data, NULL,
&offset);
- ATF_REQUIRE(result == ISC_R_SUCCESS);
+ assert_true(result == ISC_R_SUCCESS);
dns_rbt_destroy(&rbt);
/*
- * Deserialize the tree
- */
- printf("deserialization begins.\n");
-
- /*
+ * Deserialize the tree.
* Map in the whole file in one go
*/
fd = open("zone.bin", O_RDWR);
isc_file_getsizefd(fd, &filesize);
- base = mmap(NULL, filesize,
- PROT_READ|PROT_WRITE,
+ base = mmap(NULL, filesize, PROT_READ|PROT_WRITE,
MAP_FILE|MAP_PRIVATE, fd, 0);
- ATF_REQUIRE(base != NULL && base != MAP_FAILED);
+ assert_true(base != NULL && base != MAP_FAILED);
close(fd);
result = dns_rbt_deserialize_tree(base, filesize, 0, mctx,
NULL, &rbt_deserialized);
/* Test to make sure we have a valid tree */
- ATF_REQUIRE(result == ISC_R_SUCCESS);
- if (rbt_deserialized == NULL)
- atf_tc_fail("deserialized rbt is null!"); /* Abort execution. */
+ assert_true(result == ISC_R_SUCCESS);
+ if (rbt_deserialized == NULL) {
+ fail_msg("deserialized rbt is null!"); /* Abort execution. */
+ }
check_test_data(rbt_deserialized);
dns_rbt_destroy(&rbt_deserialized);
munmap(base, filesize);
unlink("zone.bin");
- dns_test_end();
}
-ATF_TC(deserialize_corrupt);
-ATF_TC_HEAD(deserialize_corrupt, tc) {
- atf_tc_set_md_var(tc, "descr", "Test reading a corrupt map file");
-}
-ATF_TC_BODY(deserialize_corrupt, tc) {
+/* Test reading a corrupt map file */
+static void
+deserialize_corrupt_test(void **state) {
dns_rbt_t *rbt = NULL;
isc_result_t result;
FILE *rbtfile = NULL;
char *base, *p, *q;
int i;
- UNUSED(tc);
+ UNUSED(state);
isc_mem_debugging = ISC_MEM_DEBUGRECORD;
- result = dns_test_begin(NULL, true);
- ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
-
/* Set up map file */
result = dns_rbt_create(mctx, delete_data, NULL, &rbt);
- ATF_CHECK_EQ(result, ISC_R_SUCCESS);
+ assert_int_equal(result, ISC_R_SUCCESS);
add_test_data(mctx, rbt);
rbtfile = fopen("./zone.bin", "w+b");
- ATF_REQUIRE(rbtfile != NULL);
+ assert_non_null(rbtfile);
result = dns_rbt_serialize_tree(rbtfile, rbt, write_data, NULL,
&offset);
- ATF_REQUIRE(result == ISC_R_SUCCESS);
+ assert_true(result == ISC_R_SUCCESS);
dns_rbt_destroy(&rbt);
/* Read back with random fuzzing */
fd = open("zone.bin", O_RDWR);
isc_file_getsizefd(fd, &filesize);
- base = mmap(NULL, filesize,
- PROT_READ|PROT_WRITE,
+ base = mmap(NULL, filesize, PROT_READ|PROT_WRITE,
MAP_FILE|MAP_PRIVATE, fd, 0);
- ATF_REQUIRE(base != NULL && base != MAP_FAILED);
+ assert_true(base != NULL && base != MAP_FAILED);
close(fd);
/* Randomly fuzz a portion of the memory */
delete_data, NULL,
fix_data, NULL,
NULL, &rbt_deserialized);
- printf("%d: %s\n", i, isc_result_totext(result));
/* Test to make sure we have a valid tree */
- ATF_REQUIRE(result == ISC_R_SUCCESS ||
+ assert_true(result == ISC_R_SUCCESS ||
result == ISC_R_INVALIDFILE);
- if (result != ISC_R_SUCCESS)
- ATF_REQUIRE(rbt_deserialized == NULL);
+ if (result != ISC_R_SUCCESS) {
+ assert_null(rbt_deserialized);
+ }
- if (rbt_deserialized != NULL)
+ if (rbt_deserialized != NULL) {
dns_rbt_destroy(&rbt_deserialized);
+ }
munmap(base, filesize);
}
unlink("zone.bin");
- dns_test_end();
}
-
-ATF_TC(serialize_align);
-ATF_TC_HEAD(serialize_align, tc) {
- atf_tc_set_md_var(tc, "descr",
- "Test the dns_rbt_serialize_align() function.");
+/* Test the dns_rbt_serialize_align() function */
+static void
+serialize_align_test(void **state) {
+ UNUSED(state);
+
+ assert_true(dns_rbt_serialize_align(0) == 0);
+ assert_true(dns_rbt_serialize_align(1) == 8);
+ assert_true(dns_rbt_serialize_align(2) == 8);
+ assert_true(dns_rbt_serialize_align(3) == 8);
+ assert_true(dns_rbt_serialize_align(4) == 8);
+ assert_true(dns_rbt_serialize_align(5) == 8);
+ assert_true(dns_rbt_serialize_align(6) == 8);
+ assert_true(dns_rbt_serialize_align(7) == 8);
+ assert_true(dns_rbt_serialize_align(8) == 8);
+ assert_true(dns_rbt_serialize_align(9) == 16);
+ assert_true(dns_rbt_serialize_align(0xff) == 0x100);
+ assert_true(dns_rbt_serialize_align(0x301) == 0x308);
}
-ATF_TC_BODY(serialize_align, tc) {
- UNUSED(tc);
-
- ATF_CHECK(dns_rbt_serialize_align(0) == 0);
- ATF_CHECK(dns_rbt_serialize_align(1) == 8);
- ATF_CHECK(dns_rbt_serialize_align(2) == 8);
- ATF_CHECK(dns_rbt_serialize_align(3) == 8);
- ATF_CHECK(dns_rbt_serialize_align(4) == 8);
- ATF_CHECK(dns_rbt_serialize_align(5) == 8);
- ATF_CHECK(dns_rbt_serialize_align(6) == 8);
- ATF_CHECK(dns_rbt_serialize_align(7) == 8);
- ATF_CHECK(dns_rbt_serialize_align(8) == 8);
- ATF_CHECK(dns_rbt_serialize_align(9) == 16);
- ATF_CHECK(dns_rbt_serialize_align(0xff) == 0x100);
- ATF_CHECK(dns_rbt_serialize_align(0x301) == 0x308);
+
+int
+main(void) {
+ const struct CMUnitTest tests[] = {
+ cmocka_unit_test_setup_teardown(serialize_test,
+ _setup, _teardown),
+ cmocka_unit_test_setup_teardown(deserialize_corrupt_test,
+ _setup, _teardown),
+ cmocka_unit_test(serialize_align_test),
+ };
+
+ return (cmocka_run_group_tests(tests, NULL, NULL));
}
-/*
- * Main
- */
-ATF_TP_ADD_TCS(tp) {
- ATF_TP_ADD_TC(tp, serialize);
- ATF_TP_ADD_TC(tp, deserialize_corrupt);
- ATF_TP_ADD_TC(tp, serialize_align);
+#else /* HAVE_CMOCKA */
+
+#include <stdio.h>
- return (atf_no_error());
+int
+main(void) {
+ printf("1..0 # Skipped: cmocka not available\n");
+ return (0);
}
+
+#endif
projects / thirdparty / bind9.git / commitdiff
