/* -*- mode: c; c-file-style: "openbsd" -*- */
/*
* Copyright (c) 2015 Vincent Bernat <bernat@luffy.cx>
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <stdlib.h>
#include <unistd.h>
#include <check.h>
#include <sys/queue.h>
#define MARSHAL_EXPORT
#include "check-compat.h"
#include "../src/marshal.h"
#include "../src/log.h"
/* This suite can be run in valgrind for memory leaks:
CK_FORK=no valgrind -v --leak-check=yes ./tests/check_marshal
*/
/* Use this callback to avoid some logs */
void donothing(int pri, const char *msg) {};
struct struct_simple {
int a1;
long a2;
char a3;
time_t a4;
char a5[7];
};
MARSHAL(struct_simple);
START_TEST(test_simple_structure) {
struct struct_simple source = {
.a1 = 78452,
.a2 = 48751424,
.a3 = 'h',
.a4 = 784254,
.a5 = { 'A', 'B', 'C', 'D', 'E', 'F', 'G'},
};
struct struct_simple *destination;
void *buffer;
size_t len, len2;
len = struct_simple_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct struct_simple));
len2 = struct_simple_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->a1, 78452);
ck_assert_int_eq(destination->a2, 48751424);
ck_assert_int_eq(destination->a3, 'h');
ck_assert_int_eq(destination->a4, 784254);
ck_assert_int_eq(destination->a5[0], 'A');
ck_assert_int_eq(destination->a5[1], 'B');
ck_assert_int_eq(destination->a5[2], 'C');
ck_assert_int_eq(destination->a5[3], 'D');
ck_assert_int_eq(destination->a5[4], 'E');
ck_assert_int_eq(destination->a5[5], 'F');
ck_assert_int_eq(destination->a5[6], 'G');
free(destination);
}
END_TEST
struct struct_sub {
int e1;
struct struct_simple e2;
char e3;
};
MARSHAL_BEGIN(struct_sub)
MARSHAL_SUBSTRUCT(struct_sub, struct_simple, e2)
MARSHAL_END(struct_sub);
START_TEST(test_substruct_structure) {
struct struct_sub source = {
.e1 = -5122,
.e2 = {
.a1 = 78452,
.a2 = 48751424,
.a3 = 'h',
.a4 = 784254,
.a5 = { 'A', 'B', 'C', 'D', 'E', 'F', 'G'},
},
.e3 = 'a',
};
struct struct_sub *destination;
void *buffer;
size_t len, len2;
len = struct_sub_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct struct_sub));
len2 = struct_sub_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->e1, -5122);
ck_assert_int_eq(destination->e2.a1, 78452);
ck_assert_int_eq(destination->e2.a2, 48751424);
ck_assert_int_eq(destination->e2.a3, 'h');
ck_assert_int_eq(destination->e2.a4, 784254);
ck_assert_int_eq(destination->e2.a5[0], 'A');
ck_assert_int_eq(destination->e2.a5[1], 'B');
ck_assert_int_eq(destination->e2.a5[2], 'C');
ck_assert_int_eq(destination->e2.a5[3], 'D');
ck_assert_int_eq(destination->e2.a5[4], 'E');
ck_assert_int_eq(destination->e2.a5[5], 'F');
ck_assert_int_eq(destination->e2.a5[6], 'G');
ck_assert_int_eq(destination->e3, 'a');
free(destination);
}
END_TEST
struct struct_onepointer {
int b1;
long b2;
char b3;
struct struct_simple *b4;
int b5;
};
MARSHAL_BEGIN(struct_onepointer)
MARSHAL_POINTER(struct_onepointer, struct_simple, b4)
MARSHAL_END(struct_onepointer);
START_TEST(test_pointer_structure) {
struct struct_simple source_simple = {
.a1 = 78452,
.a2 = 48751424,
.a3 = 'h',
.a4 = 784254,
.a5 = { 'A', 'B', 'C', 'D', 'E', 'F', 'G'},
};
struct struct_onepointer source = {
.b1 = 18,
.b2 = 15454,
.b3 = 'o',
.b4 = &source_simple,
.b5 = 333333,
};
struct struct_onepointer *destination;
void *buffer;
size_t len, len2;
len = struct_onepointer_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source_simple, 0, sizeof(struct struct_simple));
memset(&source, 0, sizeof(struct struct_onepointer));
len2 = struct_onepointer_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->b1, 18);
ck_assert_int_eq(destination->b2, 15454);
ck_assert_int_eq(destination->b3, 'o');
ck_assert_int_eq(destination->b4->a1, 78452);
ck_assert_int_eq(destination->b4->a2, 48751424);
ck_assert_int_eq(destination->b4->a3, 'h');
ck_assert_int_eq(destination->b4->a4, 784254);
ck_assert_int_eq(destination->b4->a5[0], 'A');
ck_assert_int_eq(destination->b4->a5[1], 'B');
ck_assert_int_eq(destination->b4->a5[2], 'C');
ck_assert_int_eq(destination->b4->a5[3], 'D');
ck_assert_int_eq(destination->b4->a5[4], 'E');
ck_assert_int_eq(destination->b4->a5[5], 'F');
ck_assert_int_eq(destination->b4->a5[6], 'G');
ck_assert_int_eq(destination->b5, 333333);
free(destination->b4); free(destination);
}
END_TEST
struct struct_nestedpointers {
int c1;
long c2;
struct struct_simple *c3;
struct struct_onepointer *c4;
int c5;
};
MARSHAL_BEGIN(struct_nestedpointers)
MARSHAL_POINTER(struct_nestedpointers, struct_simple, c3)
MARSHAL_POINTER(struct_nestedpointers, struct_onepointer, c4)
MARSHAL_END(struct_nestedpointers);
START_TEST(test_several_pointers_structure) {
struct struct_simple source_simple1 = {
.a1 = 78452,
.a2 = 48751424,
.a3 = 'h',
.a4 = 784254,
.a5 = { 'A', 'B', 'C', 'D', 'E', 'F', 'G'},
};
struct struct_simple source_simple2 = {
.a1 = 451,
.a2 = 451424,
.a3 = 'o',
.a4 = 74,
.a5 = { 'a', 'b', 'c', 'd', 'e', 'f', 'g'},
};
struct struct_onepointer source_onepointer = {
.b1 = 18,
.b2 = 15454,
.b3 = 'o',
.b4 = &source_simple1,
.b5 = 333333,
};
struct struct_nestedpointers source = {
.c1 = 4542,
.c2 = 5665454,
.c3 = &source_simple2,
.c4 = &source_onepointer,
.c5 = -545424,
};
struct struct_nestedpointers *destination;
void *buffer;
size_t len, len2;
len = struct_nestedpointers_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source_simple1, 0, sizeof(struct struct_simple));
memset(&source_simple2, 0, sizeof(struct struct_simple));
memset(&source_onepointer, 0, sizeof(struct struct_onepointer));
memset(&source, 0, sizeof(struct struct_nestedpointers));
len2 = struct_nestedpointers_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->c1, 4542);
ck_assert_int_eq(destination->c2, 5665454);
ck_assert_int_eq(destination->c3->a1, 451);
ck_assert_int_eq(destination->c3->a2, 451424);
ck_assert_int_eq(destination->c3->a3, 'o');
ck_assert_int_eq(destination->c3->a4, 74);
ck_assert_int_eq(destination->c3->a5[3], 'd');
ck_assert_int_eq(destination->c3->a5[4], 'e');
ck_assert_int_eq(destination->c3->a5[6], 'g');
ck_assert_int_eq(destination->c4->b1, 18);
ck_assert_int_eq(destination->c4->b2, 15454);
ck_assert_int_eq(destination->c4->b3, 'o');
ck_assert_int_eq(destination->c4->b4->a1, 78452);
ck_assert_int_eq(destination->c4->b4->a2, 48751424);
ck_assert_int_eq(destination->c4->b4->a3, 'h');
ck_assert_int_eq(destination->c4->b4->a4, 784254);
ck_assert_int_eq(destination->c4->b4->a5[0], 'A');
ck_assert_int_eq(destination->c4->b4->a5[1], 'B');
ck_assert_int_eq(destination->c4->b4->a5[2], 'C');
ck_assert_int_eq(destination->c4->b4->a5[3], 'D');
ck_assert_int_eq(destination->c4->b4->a5[4], 'E');
ck_assert_int_eq(destination->c4->b4->a5[5], 'F');
ck_assert_int_eq(destination->c4->b4->a5[6], 'G');
ck_assert_int_eq(destination->c4->b5, 333333);
free(destination->c3); free(destination->c4->b4);
free(destination->c4); free(destination);
}
END_TEST
START_TEST(test_null_pointers) {
struct struct_simple source_simple2 = {
.a1 = 451,
.a2 = 451424,
.a3 = 'o',
.a4 = 74,
.a5 = { 'a', 'b', 'c', 'd', 'e', 'f', 'g'},
};
struct struct_nestedpointers source = {
.c1 = 4542,
.c2 = 5665454,
.c3 = &source_simple2,
.c4 = NULL,
.c5 = -545424,
};
struct struct_nestedpointers *destination;
void *buffer;
size_t len, len2;
len = struct_nestedpointers_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source_simple2, 0, sizeof(struct struct_simple));
memset(&source, 0, sizeof(struct struct_nestedpointers));
len2 = struct_nestedpointers_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->c1, 4542);
ck_assert_int_eq(destination->c2, 5665454);
ck_assert_int_eq(destination->c3->a1, 451);
ck_assert_int_eq(destination->c3->a2, 451424);
ck_assert_int_eq(destination->c3->a3, 'o');
ck_assert_int_eq(destination->c3->a4, 74);
ck_assert_int_eq(destination->c3->a5[3], 'd');
ck_assert_int_eq(destination->c3->a5[4], 'e');
ck_assert_int_eq(destination->c3->a5[6], 'g');
ck_assert_ptr_eq(destination->c4, NULL);
free(destination->c3); free(destination);
}
END_TEST
struct struct_multipleref {
int f1;
struct struct_simple* f2;
struct struct_simple* f3;
struct struct_nestedpointers* f4;
};
MARSHAL_BEGIN(struct_multipleref)
MARSHAL_POINTER(struct_multipleref, struct_simple, f2)
MARSHAL_POINTER(struct_multipleref, struct_simple, f3)
MARSHAL_POINTER(struct_multipleref, struct_nestedpointers, f4)
MARSHAL_END(struct_multipleref);
START_TEST(test_multiple_references) {
struct struct_simple source_simple = {
.a1 = 451,
.a2 = 451424,
.a3 = 'o',
.a4 = 74,
.a5 = { 'a', 'b', 'c', 'd', 'e', 'f', 'g'},
};
struct struct_nestedpointers source_nested = {
.c3 = &source_simple,
.c4 = NULL,
};
struct struct_multipleref source = {
.f1 = 15,
.f2 = &source_simple,
.f3 = &source_simple,
.f4 = &source_nested,
};
struct struct_multipleref *destination;
void *buffer = NULL;
size_t len, len2;
len = struct_multipleref_serialize(&source, &buffer);
fail_unless(buffer != NULL, "Buffer is empty");
fail_unless(len > 0, "Unable to serialize");
memset(&source_simple, 0, sizeof(struct struct_simple));
memset(&source_nested, 0, sizeof(struct struct_nestedpointers));
memset(&source, 0, sizeof(struct struct_multipleref));
len2 = struct_multipleref_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->f1, 15);
ck_assert_ptr_eq(destination->f2, destination->f3);
ck_assert_ptr_eq(destination->f2, destination->f4->c3);
ck_assert_int_eq(destination->f2->a1, 451);
ck_assert_int_eq(destination->f2->a2, 451424);
ck_assert_int_eq(destination->f2->a3, 'o');
ck_assert_int_eq(destination->f2->a4, 74);
ck_assert_ptr_eq(destination->f4->c4, NULL);
free(destination->f2); free(destination->f4); free(destination);
}
END_TEST
struct struct_circularref {
int g1;
struct struct_circularref* g2;
};
MARSHAL_BEGIN(struct_circularref)
MARSHAL_POINTER(struct_circularref, struct_circularref, g2)
MARSHAL_END(struct_circularref);
START_TEST(test_circular_references) {
struct struct_circularref source = {
.g1 = 42,
.g2 = &source,
};
struct struct_circularref *destination;
void *buffer = NULL;
size_t len, len2;
len = struct_circularref_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct struct_circularref));
len2 = struct_circularref_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->g1, 42);
ck_assert_int_eq(destination->g2->g1, 42);
ck_assert_ptr_eq(destination->g2, destination->g2->g2);
free(destination);
}
END_TEST
START_TEST(test_too_small_unmarshal) {
struct struct_simple source_simple1;
struct struct_onepointer source_onepointer = {
.b4 = &source_simple1,
};
struct struct_nestedpointers source = {
.c3 = &source_simple1,
.c4 = &source_onepointer,
};
struct struct_nestedpointers *destination;
void *buffer;
size_t len, len2;
int i, j;
log_register(donothing);
len = struct_nestedpointers_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source_simple1, 0, sizeof(struct struct_simple));
memset(&source_onepointer, 0, sizeof(struct struct_onepointer));
memset(&source, 0, sizeof(struct struct_nestedpointers));
/* Loop 30 times to ease debugging leaks with valgrind */
for (j = 0; j < 30; j++) {
for (i = 0; i < len; i++) {
len2 = struct_nestedpointers_unserialize(buffer, 1, &destination);
fail_unless(len2 == 0,
"Should not be able to deserialize, too small (%d<%d)",
i, len);
}
}
len2 = struct_nestedpointers_unserialize(buffer, len + 5, &destination);
fail_unless(len2 == len, "Deserialized too much");
free(destination->c3);
free(destination->c4); free(destination); free(buffer);
log_register(NULL);
}
END_TEST
struct struct_simpleentry {
TAILQ_ENTRY(struct_simpleentry) s_entries;
int g1;
struct struct_simple *g2;
};
MARSHAL_BEGIN(struct_simpleentry)
MARSHAL_TQE(struct_simpleentry, s_entries)
MARSHAL_POINTER(struct_simpleentry, struct_simple, g2)
MARSHAL_END(struct_simpleentry);
TAILQ_HEAD(list_simple, struct_simpleentry);
MARSHAL_TQ(list_simple, struct_simpleentry);
START_TEST(test_simple_list) {
struct struct_simple source_simple = {
.a1 = 451,
.a2 = 451424,
.a3 = 'o',
.a4 = 74,
.a5 = { 'a', 'b', 'c', 'd', 'e', 'f', 'g'},
};
struct list_simple source;
struct struct_simpleentry entry1 = {
.g1 = 47,
.g2 = &source_simple,
};
struct struct_simpleentry entry2 = {
.g1 = 49,
.g2 = &source_simple,
};
struct struct_simpleentry entry3 = {
.g1 = 4700,
.g2 = NULL,
};
struct struct_simpleentry entry4 = {
.g1 = -47,
.g2 = &source_simple,
};
struct list_simple *destination;
void *buffer;
size_t len, len2;
struct struct_simpleentry *e1, *e2;
struct struct_simple *s;
TAILQ_INIT(&source);
TAILQ_INSERT_TAIL(&source, &entry1, s_entries);
TAILQ_INSERT_TAIL(&source, &entry2, s_entries);
TAILQ_INSERT_TAIL(&source, &entry3, s_entries);
TAILQ_INSERT_TAIL(&source, &entry4, s_entries);
len = list_simple_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct list_simple));
memset(&entry1, 0, sizeof(struct struct_simpleentry));
memset(&entry2, 0, sizeof(struct struct_simpleentry));
memset(&entry3, 0, sizeof(struct struct_simpleentry));
memset(&entry4, 0, sizeof(struct struct_simpleentry));
len2 = list_simple_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
e1 = TAILQ_FIRST(destination);
ck_assert_int_eq(e1->g1, 47);
s = e1->g2;
e2 = TAILQ_NEXT(e1, s_entries);
free(e1);
ck_assert_int_eq(e2->g1, 49);
ck_assert_ptr_eq(e2->g2, s);
e1 = TAILQ_NEXT(e2, s_entries);
free(e2);
ck_assert_int_eq(e1->g1, 4700);
ck_assert_ptr_eq(e1->g2, NULL);
e2 = TAILQ_NEXT(e1, s_entries);
free(e1);
ck_assert_int_eq(e2->g1, -47);
ck_assert_ptr_eq(e2->g2, s);
e1 = TAILQ_NEXT(e2, s_entries);
free(e2);
ck_assert_ptr_eq(e1, NULL);
free(s);
free(destination);
}
END_TEST
START_TEST(test_simple_repaired_list) {
struct struct_simple source_simple = {
.a1 = 451,
.a2 = 451424,
.a3 = 'o',
.a4 = 74,
.a5 = { 'a', 'b', 'c', 'd', 'e', 'f', 'g'},
};
struct list_simple source;
struct struct_simpleentry entry1 = {
.g1 = 47,
.g2 = &source_simple,
};
struct struct_simpleentry entry2 = {
.g1 = 49,
.g2 = &source_simple,
};
struct struct_simpleentry entry3 = {
.g1 = 4700,
.g2 = NULL,
};
struct struct_simpleentry entry4 = {
.g1 = -47,
.g2 = &source_simple,
};
struct struct_simpleentry entry5 = {
.g1 = -1000,
.g2 = NULL,
};
struct list_simple *destination;
void *buffer;
size_t len, len2;
struct struct_simpleentry *e1, *e2, *e3, *e4;
TAILQ_INIT(&source);
TAILQ_INSERT_TAIL(&source, &entry1, s_entries);
TAILQ_INSERT_TAIL(&source, &entry2, s_entries);
TAILQ_INSERT_TAIL(&source, &entry3, s_entries);
TAILQ_INSERT_TAIL(&source, &entry4, s_entries);
len = list_simple_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct list_simple));
memset(&entry1, 0, sizeof(struct struct_simpleentry));
memset(&entry2, 0, sizeof(struct struct_simpleentry));
memset(&entry3, 0, sizeof(struct struct_simpleentry));
memset(&entry4, 0, sizeof(struct struct_simpleentry));
len2 = list_simple_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
marshal_repair_tailq(struct_simpleentry, destination, s_entries);
e1 = TAILQ_FIRST(destination);
ck_assert_int_eq(e1->g1, 47);
e4 = TAILQ_LAST(destination, list_simple);
ck_assert_int_eq(e4->g1, -47);
e3 = TAILQ_PREV(e4, list_simple, s_entries);
ck_assert_int_eq(e3->g1, 4700);
e2 = TAILQ_PREV(e3, list_simple, s_entries);
ck_assert_int_eq(e2->g1, 49);
TAILQ_INSERT_TAIL(destination, &entry5, s_entries);
free(e1->g2);
free(e1);
free(e2);
free(e3);
free(e4);
free(destination);
}
END_TEST
START_TEST(test_empty_repaired_list) {
struct list_simple source;
size_t len, len2;
struct list_simple *destination;
void *buffer;
struct struct_simpleentry *e1;
struct struct_simpleentry entry5 = {
.g1 = -1000,
.g2 = NULL,
};
TAILQ_INIT(&source);
len = list_simple_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct list_simple));
len2 = list_simple_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
marshal_repair_tailq(struct_simpleentry, destination, s_entries);
e1 = TAILQ_FIRST(destination);
ck_assert_ptr_eq(e1, NULL);
e1 = TAILQ_LAST(destination, list_simple);
ck_assert_ptr_eq(e1, NULL);
TAILQ_INSERT_TAIL(destination, &entry5, s_entries);
free(destination);
}
END_TEST
struct struct_withlist {
int i1;
TAILQ_HEAD(, struct_simpleentry) i2;
int i3;
};
MARSHAL_BEGIN(struct_withlist)
MARSHAL_SUBTQ(struct_withlist, struct_simpleentry, i2)
MARSHAL_END(struct_withlist);
START_TEST(test_embedded_list) {
struct struct_withlist source = {
.i1 = 45424,
.i3 = 4542,
};
struct struct_simple source_simple = {
.a1 = 451,
.a2 = 451424,
.a3 = 'o',
.a4 = 74,
.a5 = { 'a', 'b', 'c', 'd', 'e', 'f', 'g'},
};
struct struct_simpleentry entry1 = {
.g1 = 47,
.g2 = &source_simple,
};
struct struct_simpleentry entry2 = {
.g1 = 49,
.g2 = &source_simple,
};
struct struct_simpleentry entry3 = {
.g1 = 4700,
.g2 = NULL,
};
struct struct_simpleentry entry4 = {
.g1 = -47,
.g2 = &source_simple,
};
struct struct_withlist *destination;
void *buffer;
size_t len, len2;
struct struct_simpleentry *e1, *e2;
struct struct_simple *s;
TAILQ_INIT(&source.i2);
TAILQ_INSERT_TAIL(&source.i2, &entry1, s_entries);
TAILQ_INSERT_TAIL(&source.i2, &entry2, s_entries);
TAILQ_INSERT_TAIL(&source.i2, &entry3, s_entries);
TAILQ_INSERT_TAIL(&source.i2, &entry4, s_entries);
len = struct_withlist_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct list_simple));
memset(&entry1, 0, sizeof(struct struct_simpleentry));
memset(&entry2, 0, sizeof(struct struct_simpleentry));
memset(&entry3, 0, sizeof(struct struct_simpleentry));
memset(&entry4, 0, sizeof(struct struct_simpleentry));
len2 = struct_withlist_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(destination->i1, 45424);
ck_assert_int_eq(destination->i3, 4542);
e1 = TAILQ_FIRST(&destination->i2);
ck_assert_int_eq(e1->g1, 47);
ck_assert_int_eq(e1->g2->a4, 74);
s = e1->g2;
e2 = TAILQ_NEXT(e1, s_entries);
free(e1);
ck_assert_int_eq(e2->g1, 49);
ck_assert_ptr_eq(e2->g2, s);
e1 = TAILQ_NEXT(e2, s_entries);
free(e2);
ck_assert_int_eq(e1->g1, 4700);
ck_assert_ptr_eq(e1->g2, NULL);
e2 = TAILQ_NEXT(e1, s_entries);
free(e1);
ck_assert_int_eq(e2->g1, -47);
ck_assert_ptr_eq(e2->g2, s);
e1 = TAILQ_NEXT(e2, s_entries);
free(e2);
ck_assert_ptr_eq(e1, NULL);
free(s);
free(destination);
}
END_TEST
struct struct_string {
int s1;
char *s2;
char *s3;
};
MARSHAL_BEGIN(struct_string)
MARSHAL_STR(struct_string, s2)
MARSHAL_STR(struct_string, s3)
MARSHAL_END(struct_string);
START_TEST(test_string) {
struct struct_string source = {
.s1 = 44444,
.s2 = "String 2",
.s3 = "String 3",
};
struct struct_string *destination;
void *buffer;
size_t len, len2;
len = struct_string_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct struct_string));
len2 = struct_string_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->s1, 44444);
ck_assert_str_eq(destination->s2, "String 2");
ck_assert_str_eq(destination->s3, "String 3");
free(destination->s2); free(destination->s3);
free(destination);
}
END_TEST
struct struct_fixedstring {
int s1;
char *s2;
int s2_len;
char *s3;
};
MARSHAL_BEGIN(struct_fixedstring)
MARSHAL_FSTR(struct_fixedstring, s2, s2_len)
MARSHAL_STR(struct_fixedstring, s3)
MARSHAL_END(struct_fixedstring);
START_TEST(test_fixed_string) {
struct struct_fixedstring source = {
.s1 = 44444,
.s2 = "String 2 Bla",
.s2_len = 8, /* Not 12! */
.s3 = "String 3",
};
struct struct_fixedstring *destination;
void *buffer;
size_t len, len2;
len = struct_fixedstring_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct struct_fixedstring));
len2 = struct_fixedstring_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->s1, 44444);
ck_assert_int_eq(destination->s2_len, 8);
ck_assert_int_eq(destination->s2[0], 'S');
ck_assert_int_eq(destination->s2[2], 'r');
ck_assert_int_eq(destination->s2[4], 'n');
ck_assert_int_eq(destination->s2[5], 'g');
ck_assert_int_eq(destination->s2[6], ' ');
ck_assert_int_eq(destination->s2[7], '2');
ck_assert_int_eq(destination->s2[8], '\0'); /* fixed string are null-terminated too */
ck_assert_str_eq(destination->s3, "String 3");
free(destination->s2); free(destination->s3);
free(destination);
}
END_TEST
struct struct_ignore {
int t1;
void *t2;
int t3;
};
MARSHAL_BEGIN(struct_ignore)
MARSHAL_IGNORE(struct_ignore, t2)
MARSHAL_END(struct_ignore);
START_TEST(test_ignore) {
struct struct_ignore source = {
.t1 = 4544,
.t2 = (void *)"String 2 Bla",
.t3 = 11111,
};
struct struct_ignore *destination;
void *buffer;
size_t len, len2;
len = struct_ignore_serialize(&source, &buffer);
fail_unless(len > 0, "Unable to serialize");
memset(&source, 0, sizeof(struct struct_ignore));
len2 = struct_ignore_unserialize(buffer, len, &destination);
fail_unless(len2 > 0, "Unable to deserialize");
free(buffer);
ck_assert_int_eq(len, len2);
ck_assert_int_eq(destination->t1, 4544);
ck_assert_ptr_eq(destination->t2, NULL);
ck_assert_int_eq(destination->t3, 11111);
free(destination);
}
END_TEST
START_TEST(test_equality) {
struct struct_simple source_simple1 = {
.a1 = 451,
.a2 = 451424,
.a3 = 'o',
.a4 = 74,
.a5 = { 'a', 'b', 'c', 'd', 'e', 'f', 'g'},
};
struct struct_simpleentry entry1 = {
.g1 = 47,
.g2 = &source_simple1,
};
struct struct_simple source_simple2;
struct struct_simpleentry entry2;
void *buffer1, *buffer2;
memcpy(&source_simple2, &source_simple1, sizeof(source_simple1));
memcpy(&entry2, &entry1, sizeof(entry1));
entry2.g2 = &source_simple2;
ssize_t len1 = struct_simpleentry_serialize(&entry1, &buffer1);
ssize_t len2 = struct_simpleentry_serialize(&entry2, &buffer2);
fail_unless(len1 > 0, "Unable to serialize");
fail_unless(len2 > 0, "Unable to serialize");
ck_assert_int_eq(len1, len2);
fail_unless(!memcmp(buffer1, buffer2, len1), "Same content should give the same serialization");
free(buffer1); free(buffer2);
}
END_TEST
Suite *
marshal_suite(void)
{
Suite *s = suite_create("Marshalling");
TCase *tc_marshal = tcase_create("Marshalling");
tcase_add_test(tc_marshal, test_simple_structure);
tcase_add_test(tc_marshal, test_substruct_structure);
tcase_add_test(tc_marshal, test_pointer_structure);
tcase_add_test(tc_marshal, test_several_pointers_structure);
tcase_add_test(tc_marshal, test_null_pointers);
tcase_add_test(tc_marshal, test_multiple_references);
tcase_add_test(tc_marshal, test_circular_references);
tcase_add_test(tc_marshal, test_too_small_unmarshal);
tcase_add_test(tc_marshal, test_simple_list);
tcase_add_test(tc_marshal, test_simple_repaired_list);
tcase_add_test(tc_marshal, test_empty_repaired_list);
tcase_add_test(tc_marshal, test_embedded_list);
tcase_add_test(tc_marshal, test_string);
tcase_add_test(tc_marshal, test_fixed_string);
tcase_add_test(tc_marshal, test_ignore);
tcase_add_test(tc_marshal, test_equality);
suite_add_tcase(s, tc_marshal);
return s;
}
int
main()
{
int number_failed;
Suite *s = marshal_suite();
SRunner *sr = srunner_create(s);
srunner_run_all(sr, CK_ENV);
number_failed = srunner_ntests_failed(sr);
srunner_free(sr);
return (number_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
}