import

darcs-hash:20060926215024-b59df-787a30444c3ae416c309322e1bdd26fd438fdc10.gz

2 files changed, 199 insertions, 0 deletions

diff --git a/tests/Makefile.am b/tests/Makefile.am
new file mode 100644
index 0000000..bd6fd5b
--- /dev/null
+++ b/tests/Makefile.am
@@ -0,0 +1,10 @@
+
+AM_CFLAGS = -std=c99 -Wall -Werror
+
+test_SOURCES = test.c $(top_srcdir)/stun/stun.h $(top_srcdir)/stun/stun.c
+test_CFLAGS = -I$(top_srcdir)/stun $(AM_CFLAGS) @CHECK_CFLAGS@ @GLIB_CFLAGS@
+test_LDADD = @CHECK_LIBS@ @GLIB_LIBS@
+
+TESTS = test
+noinst_PROGRAMS = test
+
diff --git a/tests/test.c b/tests/test.c
new file mode 100644
index 0000000..8b1278f
--- /dev/null
+++ b/tests/test.c
@@ -0,0 +1,189 @@
+
+#include <string.h>
+
+#include <check.h>
+
+#include <stun.h>
+
+START_TEST (test_attribute_pack)
+{
+  StunAttribute *attr = stun_attribute_mapped_address_new(0x02030405, 2345);
+  gchar *packed;
+  guint length;
+
+  length = stun_attribute_pack(attr, &packed);
+
+  fail_unless(12 == length);
+  fail_unless(NULL != packed);
+
+  fail_unless(0 == memcmp(packed,
+    "\x00\x01"          // type
+    "\x00\x08"          // length
+    "\x00\x01"          // padding, address family
+    "\x09\x29"          // port
+    "\x02\x03\x04\x05", // IP address
+    length));
+  g_free(packed);
+  stun_attribute_free(attr);
+}
+END_TEST
+
+START_TEST (test_attribute_dump)
+{
+  StunAttribute *attr = stun_attribute_mapped_address_new(0x02030405, 2345);
+  gchar *dump = stun_attribute_dump(attr);
+
+  fail_unless(NULL != dump);
+  fail_unless(0 == strcmp(dump, "MAPPED-ADDRESS 2.3.4.5:2345"));
+  g_free(dump);
+  stun_attribute_free(attr);
+}
+END_TEST
+
+START_TEST (test_attribute_unpack)
+{
+  StunAttribute *attr = stun_attribute_unpack(12,
+    "\x00\x01"         // type
+    "\x00\x08"         // length
+    "\x00\x01"         // padding, address family
+    "\x09\x29"         // port
+    "\x02\x03\x04\x05" // IP address
+    );
+
+  fail_unless(NULL != attr);
+  fail_unless(attr->type == STUN_ATTRIBUTE_MAPPED_ADDRESS);
+  fail_unless(attr->address.af == 1);
+  fail_unless(attr->address.port == 2345);
+  fail_unless(attr->address.ip == 0x02030405);
+  stun_attribute_free(attr);
+}
+END_TEST
+
+START_TEST (test_message_pack)
+{
+  StunMessage *msg = stun_message_binding_request_new();
+  gchar *packed;
+  guint length;
+
+  memcpy(msg->transaction_id,
+    "\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f", 16);
+  msg->attributes = g_malloc0(2 * sizeof(StunAttribute *));
+  msg->attributes[0] = stun_attribute_mapped_address_new(0x02030405, 2345);
+  length = stun_message_pack(msg, &packed);
+
+  fail_unless(packed != NULL);
+  fail_unless(length == 32);
+  fail_unless(0 == memcmp(packed + 0, "\x00\x01", 2));
+  fail_unless(0 == memcmp(packed + 2, "\x00\x20", 2));
+  fail_unless(0 == memcmp(packed + 4,
+    "\x00\x01\x02\x03"
+    "\x04\x05\x06\x07"
+    "\x08\x09\x0a\x0b"
+    "\x0c\x0d\x0e\x0f", 16));
+  fail_unless(0 == memcmp(packed + 20,
+    "\x00\x01"
+    "\x00\x08"
+    "\x00\x01"
+    "\x09\x29"
+    "\x02\x03\x04\x05", 12));
+
+  g_free(packed);
+  stun_message_free(msg);
+}
+END_TEST
+
+START_TEST (test_message_dump)
+{
+  StunMessage *msg = stun_message_binding_request_new();
+  gchar *dump;
+
+  msg->attributes = g_malloc0(2 * sizeof(StunAttribute *));
+  msg->attributes[0] = stun_attribute_mapped_address_new(0x02030405, 2345);
+
+  dump = stun_message_dump(msg);
+  fail_unless(NULL != dump);
+  fail_unless(0 == strcmp(dump,
+    "BINDING-REQUEST 00000000:00000000:00000000:00000000\n"
+    "  MAPPED-ADDRESS 2.3.4.5:2345"));
+  g_free(dump);
+  stun_message_free(msg);
+}
+END_TEST
+
+START_TEST (test_message_unpack)
+{
+  StunMessage *msg = stun_message_unpack(32,
+    "\x00\x01"         // type
+    "\x00\x0c"         // length
+    "\x00\x01\x02\x03" // transaction ID
+    "\x04\x05\x06\x07"
+    "\x08\x09\x0a\x0b"
+    "\x0c\x0d\x0e\x0f"
+    "\x00\x01"         // attr1 type
+    "\x00\x08"         // attr1 length
+    "\x00\x01"         // padding, address family
+    "\x09\x29"         // port
+    "\x02\x03\x04\x05" // IP address
+    );
+
+  fail_unless(msg->type == STUN_MESSAGE_BINDING_REQUEST);
+  fail_unless(msg->attributes[0] != NULL);
+  fail_unless(msg->attributes[0]->type == STUN_ATTRIBUTE_MAPPED_ADDRESS);
+  fail_unless(msg->attributes[0]->address.port == 2345);
+  fail_unless(msg->attributes[0]->address.ip == 0x02030405);
+  fail_unless(msg->attributes[1] == NULL);
+  stun_message_free(msg);
+}
+END_TEST
+
+Suite *
+stun_suite(void)
+{
+  Suite *suite;
+  TCase *tcase;
+
+  suite = suite_create("STUN");
+
+  tcase = tcase_create("attribute pack");
+  tcase_add_test(tcase, test_attribute_pack);
+  suite_add_tcase(suite, tcase);
+
+  tcase = tcase_create("attribute dump");
+  tcase_add_test(tcase, test_attribute_dump);
+  suite_add_tcase(suite, tcase);
+
+  tcase = tcase_create("attribute unpack");
+  tcase_add_test(tcase, test_attribute_unpack);
+  suite_add_tcase(suite, tcase);
+
+  tcase = tcase_create("message pack");
+  tcase_add_test(tcase, test_message_pack);
+  suite_add_tcase(suite, tcase);
+
+  tcase = tcase_create("message dump");
+  tcase_add_test(tcase, test_message_dump);
+  suite_add_tcase(suite, tcase);
+
+  tcase = tcase_create("message unpack");
+  tcase_add_test(tcase, test_message_unpack);
+  suite_add_tcase(suite, tcase);
+
+  return suite;
+}
+
+int
+main(void)
+{
+  Suite *suite;
+  SRunner *runner;
+  int failures;
+
+  suite = stun_suite();
+  runner = srunner_create(suite);
+  srunner_run_all(runner, CK_NORMAL | CK_NOFORK);
+  failures = srunner_ntests_failed(runner);
+  srunner_free(runner);
+
+  return (failures == 0) ? 0 : 1;
+}
+