Break out Package read/write

1 files changed, 104 insertions, 0 deletions

diff --git a/test/test-package.cc b/test/test-package.cc
new file mode 100644
index 0000000..ce15c88
--- /dev/null
+++ b/test/test-package.cc
@@ -0,0 +1,104 @@
+// -*- mode: c++; c-basic-offset: 2; -*-
+
+#include "common.hh"
+#include "test.hh"
+
+#include "package.hh"
+
+namespace {
+
+void setup(Package* pkg) {
+  pkg->id = 42;
+  pkg->timestamp.tv_sec = 123;
+  pkg->timestamp.tv_nsec = 999999999;
+  pkg->source_port = 0;
+  pkg->source_host = "source";
+  pkg->target_port = 65535;
+  pkg->target_host = "target";
+}
+
+bool pkg_eq(Package const& p1, Package const& p2) {
+  return p1.id == p2.id
+    && p1.timestamp.tv_sec == p2.timestamp.tv_sec
+    && p1.timestamp.tv_nsec == p2.timestamp.tv_nsec
+    && p1.source_port == p2.source_port
+    && p1.source_host.compare(p2.source_host) == 0
+    && p1.target_port == p2.target_port
+    && p1.target_host.compare(p2.target_host) == 0;
+}
+
+bool test_sanity() {
+  Package pkg1;
+  Package pkg2;
+
+  setup(&pkg1);
+
+  uint8_t tmp[4096];
+  auto size = write_package(pkg1, tmp, sizeof(tmp));
+  ASSERT_TRUE(size < sizeof(tmp));
+
+  ASSERT_EQ(size, read_package(&pkg2, tmp, sizeof(tmp)));
+
+  ASSERT_TRUE(pkg_eq(pkg1, pkg2));
+
+  return true;
+}
+
+bool test_overflow() {
+  Package pkg1;
+  setup(&pkg1);
+
+  uint8_t tmp[4096];
+  auto size = write_package(pkg1, tmp, sizeof(tmp));
+  for (size_t i = 0; i < size; ++i) {
+    memset(tmp + size, 0, sizeof(tmp) - size);
+    auto s2 = write_package(pkg1, tmp + size, i);
+    ASSERT_EQ(size, s2);
+    ASSERT_EQ(0, tmp[size + i]);
+  }
+  auto s2 = write_package(pkg1, tmp + size, size);
+  ASSERT_EQ(size, s2);
+  ASSERT_EQ(0, memcmp(tmp, tmp + size, size));
+  return true;
+}
+
+bool test_null() {
+  Package pkg1;
+  setup(&pkg1);
+
+  uint8_t tmp[4096];
+  auto size = write_package(pkg1, tmp, sizeof(tmp));
+  auto s2 = write_package(pkg1, NULL, 0);
+  ASSERT_EQ(size, s2);
+  s2 = write_package(pkg1, NULL, sizeof(tmp));
+  ASSERT_EQ(size, s2);
+
+  return true;
+}
+
+bool test_empty() {
+  Package pkg1;
+  Package pkg2;
+  setup(&pkg1);
+  pkg1.source_host.clear();
+  pkg1.target_host.clear();
+
+  uint8_t tmp[4096];
+  auto size = write_package(pkg1, tmp, sizeof(tmp));
+  ASSERT_EQ(size, read_package(&pkg2, tmp, sizeof(tmp)));
+
+  ASSERT_TRUE(pkg_eq(pkg1, pkg2));
+
+  return true;
+}
+
+}  // namespace
+
+int main(void) {
+  BEFORE;
+  RUN(test_sanity());
+  RUN(test_overflow());
+  RUN(test_null());
+  RUN(test_empty());
+  AFTER;
+}