1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#include "config.h"
#include "fileHandlingTest.h"
#include "ntp_stdlib.h"
#include "ntp_types.h"
#include "crypto.h"
#include "unity.h"
//typedef int bool;
bool CompareKeys(struct key expected, struct key actual) {
if (expected.key_id != actual.key_id){
printf("Expected key_id: %d", expected.key_id);
printf(" but was: %d\n", actual.key_id);
return FALSE;
}
if (expected.key_len != actual.key_len){
printf("Expected key_len: %d", expected.key_len);
printf(" but was: %d\n", actual.key_len);
return FALSE;
}
if (strcmp(expected.type, actual.type) != 0){
printf("Expected key_type: %s", expected.type);
printf(" but was: %s\n", actual.type);
return FALSE;
}
if (memcmp(expected.key_seq, actual.key_seq, expected.key_len) != 0){
printf("Key mismatch!\n");
return FALSE;
}
return TRUE;
}
bool CompareKeysAlternative(int key_id,
int key_len,
const char* type,
const char* key_seq,
struct key actual) {
struct key temp;
temp.key_id = key_id;
temp.key_len = key_len;
strlcpy(temp.type, type, sizeof(temp.type));
memcpy(temp.key_seq, key_seq, key_len);
return CompareKeys(temp, actual);
}
void test_ReadEmptyKeyFile() {
struct key* keys = NULL;
TEST_ASSERT_EQUAL(0, auth_init(CreatePath("key-test-empty", INPUT_DIR), &keys));
TEST_ASSERT_TRUE(keys == NULL);
}
void test_ReadASCIIKeys() {
struct key* keys = NULL;
TEST_ASSERT_EQUAL(2, auth_init(CreatePath("key-test-ascii", INPUT_DIR), &keys));
TEST_ASSERT_TRUE(keys != NULL);
struct key* result = NULL;
get_key(40, &result);
TEST_ASSERT_TRUE(result != NULL);
TEST_ASSERT_TRUE(CompareKeysAlternative(40, 11, "MD5", "asciikeyTwo", *result));
result = NULL;
get_key(50, &result);
TEST_ASSERT_TRUE(result != NULL);
TEST_ASSERT_TRUE(CompareKeysAlternative(50, 11, "MD5", "asciikeyOne", *result));
}
void test_ReadHexKeys() {
struct key* keys = NULL;
TEST_ASSERT_EQUAL(3, auth_init(CreatePath("key-test-hex", INPUT_DIR), &keys));
TEST_ASSERT_TRUE(keys != NULL);
struct key* result = NULL;
get_key(10, &result);
TEST_ASSERT_TRUE(result != NULL);
TEST_ASSERT_TRUE(CompareKeysAlternative(10, 13, "MD5",
"\x01\x23\x45\x67\x89\xab\xcd\xef\x01\x23\x45\x67\x89", *result));
result = NULL;
get_key(20, &result);
TEST_ASSERT_TRUE(result != NULL);
char data1[15]; memset(data1, 0x11, 15);
TEST_ASSERT_TRUE(CompareKeysAlternative(20, 15, "MD5", data1, *result));
result = NULL;
get_key(30, &result);
TEST_ASSERT_TRUE(result != NULL);
char data2[13]; memset(data2, 0x01, 13);
TEST_ASSERT_TRUE(CompareKeysAlternative(30, 13, "MD5", data2, *result));
}
void test_ReadKeyFileWithComments() {
struct key* keys = NULL;
TEST_ASSERT_EQUAL(2, auth_init(CreatePath("key-test-comments", INPUT_DIR), &keys));
TEST_ASSERT_TRUE(keys != NULL);
struct key* result = NULL;
get_key(10, &result);
TEST_ASSERT_TRUE(result != NULL);
char data[15]; memset(data, 0x01, 15);
TEST_ASSERT_TRUE(CompareKeysAlternative(10, 15, "MD5", data, *result));
result = NULL;
get_key(34, &result);
TEST_ASSERT_TRUE(result != NULL);
TEST_ASSERT_TRUE(CompareKeysAlternative(34, 3, "MD5", "xyz", *result));
}
void test_ReadKeyFileWithInvalidHex() {
struct key* keys = NULL;
TEST_ASSERT_EQUAL(1, auth_init(CreatePath("key-test-invalid-hex", INPUT_DIR), &keys));
TEST_ASSERT_TRUE(keys != NULL);
struct key* result = NULL;
get_key(10, &result);
TEST_ASSERT_TRUE(result != NULL);
char data[15]; memset(data, 0x01, 15);
TEST_ASSERT_TRUE(CompareKeysAlternative(10, 15, "MD5", data, *result));
result = NULL;
get_key(30, &result); // Should not exist, and result should remain NULL.
TEST_ASSERT_TRUE(result == NULL);
}
|
