kodDatabase.c revision 1.1.1.3.6.3
1 1.1.1.3.6.3 snj /* $NetBSD: kodDatabase.c,v 1.1.1.3.6.3 2016/05/08 21:51:05 snj Exp $ */ 2 1.1.1.3.6.2 snj 3 1.1.1.3.6.2 snj #include "config.h" 4 1.1.1.3.6.2 snj 5 1.1.1.3.6.3 snj #include "ntp_workimpl.h" 6 1.1.1.3.6.2 snj #include "ntp_types.h" 7 1.1.1.3.6.2 snj #include "sntptest.h" 8 1.1.1.3.6.2 snj #include "ntp_stdlib.h" 9 1.1.1.3.6.2 snj #include "sntp-opts.h" 10 1.1.1.3.6.2 snj #include "kod_management.h" 11 1.1.1.3.6.2 snj #include "ntp_io.h" 12 1.1.1.3.6.2 snj 13 1.1.1.3.6.2 snj #include "unity.h" 14 1.1.1.3.6.2 snj 15 1.1.1.3.6.2 snj void setUp(void); 16 1.1.1.3.6.2 snj void test_SingleEntryHandling(void); 17 1.1.1.3.6.2 snj void test_MultipleEntryHandling(void); 18 1.1.1.3.6.2 snj void test_NoMatchInSearch(void); 19 1.1.1.3.6.2 snj void test_AddDuplicate(void); 20 1.1.1.3.6.2 snj void test_DeleteEntry(void); 21 1.1.1.3.6.2 snj 22 1.1.1.3.6.2 snj 23 1.1.1.3.6.2 snj void 24 1.1.1.3.6.2 snj setUp(void) { 25 1.1.1.3.6.2 snj kod_init_kod_db("/dev/null", TRUE); 26 1.1.1.3.6.3 snj init_lib(); 27 1.1.1.3.6.2 snj } 28 1.1.1.3.6.2 snj 29 1.1.1.3.6.2 snj 30 1.1.1.3.6.2 snj void 31 1.1.1.3.6.2 snj test_SingleEntryHandling(void) { 32 1.1.1.3.6.2 snj const char HOST[] = "192.0.2.5"; 33 1.1.1.3.6.2 snj const char REASON[] = "DENY"; 34 1.1.1.3.6.2 snj 35 1.1.1.3.6.2 snj add_entry(HOST, REASON); 36 1.1.1.3.6.2 snj 37 1.1.1.3.6.2 snj struct kod_entry* result; 38 1.1.1.3.6.2 snj 39 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(1, search_entry(HOST, &result)); 40 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL_STRING(HOST, result->hostname); 41 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL_STRING(REASON, result->type); 42 1.1.1.3.6.2 snj } 43 1.1.1.3.6.2 snj 44 1.1.1.3.6.2 snj 45 1.1.1.3.6.2 snj void 46 1.1.1.3.6.2 snj test_MultipleEntryHandling(void) { 47 1.1.1.3.6.2 snj const char HOST1[] = "192.0.2.3"; 48 1.1.1.3.6.2 snj const char REASON1[] = "DENY"; 49 1.1.1.3.6.2 snj 50 1.1.1.3.6.2 snj const char HOST2[] = "192.0.5.5"; 51 1.1.1.3.6.2 snj const char REASON2[] = "RATE"; 52 1.1.1.3.6.2 snj 53 1.1.1.3.6.2 snj const char HOST3[] = "192.0.10.1"; 54 1.1.1.3.6.2 snj const char REASON3[] = "DENY"; 55 1.1.1.3.6.2 snj 56 1.1.1.3.6.2 snj add_entry(HOST1, REASON1); 57 1.1.1.3.6.2 snj add_entry(HOST2, REASON2); 58 1.1.1.3.6.2 snj add_entry(HOST3, REASON3); 59 1.1.1.3.6.2 snj 60 1.1.1.3.6.2 snj struct kod_entry* result; 61 1.1.1.3.6.2 snj 62 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(1, search_entry(HOST1, &result)); 63 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL_STRING(HOST1, result->hostname); 64 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL_STRING(REASON1, result->type); 65 1.1.1.3.6.2 snj 66 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(1, search_entry(HOST2, &result)); 67 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL_STRING(HOST2, result->hostname); 68 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL_STRING(REASON2, result->type); 69 1.1.1.3.6.2 snj 70 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(1, search_entry(HOST3, &result)); 71 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL_STRING(HOST3, result->hostname); 72 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL_STRING(REASON3, result->type); 73 1.1.1.3.6.2 snj 74 1.1.1.3.6.2 snj free(result); 75 1.1.1.3.6.2 snj } 76 1.1.1.3.6.2 snj 77 1.1.1.3.6.2 snj 78 1.1.1.3.6.2 snj void 79 1.1.1.3.6.2 snj test_NoMatchInSearch(void) { 80 1.1.1.3.6.2 snj const char HOST_ADD[] = "192.0.2.6"; 81 1.1.1.3.6.2 snj const char HOST_NOTADD[] = "192.0.6.1"; 82 1.1.1.3.6.2 snj const char REASON[] = "DENY"; 83 1.1.1.3.6.2 snj 84 1.1.1.3.6.2 snj add_entry(HOST_ADD, REASON); 85 1.1.1.3.6.2 snj 86 1.1.1.3.6.2 snj struct kod_entry* result; 87 1.1.1.3.6.2 snj 88 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(0, search_entry(HOST_NOTADD, &result)); 89 1.1.1.3.6.2 snj TEST_ASSERT_TRUE(result == NULL); 90 1.1.1.3.6.2 snj } 91 1.1.1.3.6.2 snj 92 1.1.1.3.6.2 snj 93 1.1.1.3.6.2 snj void 94 1.1.1.3.6.2 snj test_AddDuplicate(void) { 95 1.1.1.3.6.2 snj const char HOST[] = "192.0.2.3"; 96 1.1.1.3.6.2 snj const char REASON1[] = "RATE"; 97 1.1.1.3.6.2 snj const char REASON2[] = "DENY"; 98 1.1.1.3.6.2 snj 99 1.1.1.3.6.2 snj add_entry(HOST, REASON1); 100 1.1.1.3.6.2 snj struct kod_entry* result1; 101 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(1, search_entry(HOST, &result1)); 102 1.1.1.3.6.2 snj 103 1.1.1.3.6.2 snj /* 104 1.1.1.3.6.2 snj * Sleeps for two seconds since we want to ensure that 105 1.1.1.3.6.2 snj * the timestamp is updated to a new value. 106 1.1.1.3.6.2 snj */ 107 1.1.1.3.6.2 snj sleep(2); 108 1.1.1.3.6.2 snj 109 1.1.1.3.6.2 snj add_entry(HOST, REASON2); 110 1.1.1.3.6.2 snj struct kod_entry* result2; 111 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(1, search_entry(HOST, &result2)); 112 1.1.1.3.6.2 snj 113 1.1.1.3.6.2 snj TEST_ASSERT_FALSE(result1->timestamp == result2->timestamp); 114 1.1.1.3.6.2 snj 115 1.1.1.3.6.2 snj free(result1); 116 1.1.1.3.6.2 snj free(result2); 117 1.1.1.3.6.2 snj } 118 1.1.1.3.6.2 snj 119 1.1.1.3.6.2 snj 120 1.1.1.3.6.2 snj void 121 1.1.1.3.6.2 snj test_DeleteEntry(void) { 122 1.1.1.3.6.2 snj const char HOST1[] = "192.0.2.1"; 123 1.1.1.3.6.2 snj const char HOST2[] = "192.0.2.2"; 124 1.1.1.3.6.2 snj const char HOST3[] = "192.0.2.3"; 125 1.1.1.3.6.2 snj const char REASON[] = "DENY"; 126 1.1.1.3.6.2 snj 127 1.1.1.3.6.2 snj add_entry(HOST1, REASON); 128 1.1.1.3.6.2 snj add_entry(HOST2, REASON); 129 1.1.1.3.6.2 snj add_entry(HOST3, REASON); 130 1.1.1.3.6.2 snj 131 1.1.1.3.6.2 snj struct kod_entry* result; 132 1.1.1.3.6.2 snj 133 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(1, search_entry(HOST2, &result)); 134 1.1.1.3.6.2 snj free(result); 135 1.1.1.3.6.2 snj 136 1.1.1.3.6.2 snj delete_entry(HOST2, REASON); 137 1.1.1.3.6.2 snj 138 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(0, search_entry(HOST2, &result)); 139 1.1.1.3.6.2 snj 140 1.1.1.3.6.2 snj // Ensure that the other entry is still there. 141 1.1.1.3.6.2 snj TEST_ASSERT_EQUAL(1, search_entry(HOST1, &result)); 142 1.1.1.3.6.2 snj free(result); 143 1.1.1.3.6.2 snj } 144
Indexes created Tue Mar 24 00:24:13 UTC 2026