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