added unit tests for checking tags

relay_station/tests/Makefile

@@ -0,0 +1,13 @@
+CC=gcc
+CFLAGS=-g -ggdb -Wall -std=gnu99
+
+TARGETS=check_tagged
+
+all: $(TARGETS)
+
+check_tagged: check_tagged.c ../flow.c ../crypto.o ../relay.o ../cryptothread.o ../ptwist168.o ../packet.o ../util.o ../flow.h ../crypto.h ../relay.h ../cryptothread.h ../ptwist.h ../packet.h ../util.h
+	gcc -g -ggdb -o $@ $^ -L/usr/local/lib -I/usr/local/include -lssl -lcrypto -ldl -lpthread -lpcap -lcheck_pic -lrt -lm -lsubunit
+
+clean:
+	-rm $(TARGETS)
+

relay_station/tests/check_tagged.c

@@ -0,0 +1,119 @@
+/** check_tagged.c
+ *
+ * Test the tag-check functionality of Slitheen
+ */
+
+#include <check.h>
+#include <stdlib.h>
+
+#include "../flow.h"
+#include "../crypto.h"
+#include "../cryptothread.h"
+#include "../packet.h"
+#include "../util.h"
+
+START_TEST(test_recognize_notag){
+    struct packet_info *info;
+    uint8_t *data;
+    FILE *fp;
+    uint64_t fsize;
+
+    //populate packet_info with a tagged ClientHello message
+    fp = fopen("data/packet_untagged.dat", "rb");
+    if (fp == NULL) {
+        perror("fopen");
+        ck_abort();
+    }
+
+    fseek(fp, 0, SEEK_END);
+    fsize = ftell(fp);
+    fseek(fp, 0, SEEK_SET);
+    data = smalloc(fsize);
+
+    int32_t result = fread(data, fsize, 1, fp);
+
+    fclose(fp);
+
+    info = smalloc(sizeof(struct packet_info));
+
+    extract_packet_headers(data, info);
+
+    ck_assert_int_eq(check_handshake(info), 0);
+}
+END_TEST
+
+START_TEST(test_recognize_tag){
+    struct packet_info *info;
+    uint8_t *data;
+    FILE *fp;
+    uint64_t fsize;
+
+    //populate packet_info with a tagged ClientHello message
+    fp = fopen("data/packet_tagged.dat", "rb");
+    if (fp == NULL) {
+        perror("fopen");
+        ck_abort();
+    }
+
+    fseek(fp, 0, SEEK_END);
+    fsize = ftell(fp);
+    fseek(fp, 0, SEEK_SET);
+    data = smalloc(fsize);
+
+    int32_t result = fread(data, fsize, 1, fp);
+
+    fclose(fp);
+    
+    info = smalloc(sizeof(struct packet_info));
+
+    extract_packet_headers(data, info);
+
+    ck_assert_int_eq(check_handshake(info), 1);
+
+}
+END_TEST
+
+Suite *tag_suite(void) {
+    Suite *s;
+    TCase *tc_core;
+
+    s = suite_create("Tag");
+
+    tc_core = tcase_create("Core");
+    tcase_add_test(tc_core, test_recognize_tag);
+    tcase_add_test(tc_core, test_recognize_notag);
+    suite_add_tcase(s, tc_core);
+
+    return s;
+}
+
+
+int main(void){
+
+    int number_failed;
+    Suite *s;
+    SRunner *sr;
+
+    //initialize Slitheen structures
+    if(init_tables()){
+        exit(1);
+    }
+    if(init_session_cache()){
+        exit(1);
+    }
+    init_crypto_locks();
+
+
+    s = tag_suite();
+    sr = srunner_create(s);
+
+    srunner_set_fork_status(sr, CK_NOFORK);
+
+    srunner_run_all(sr, CK_NORMAL);
+    number_failed = srunner_ntests_failed(sr);
+    srunner_free(sr);
+
+    crypto_locks_cleanup();
+
+    return (number_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
+}