crypto.c revision 285612 
1#include "config.h"
2#include "unity.h"
3#include "ntp_types.h"
4
5#include "sntptest.h"
6#include "crypto.h"
7
8#define MD5_LENGTH 16
9#define SHA1_LENGTH 20
10
11void test_MakeMd5Mac(void) {
12
13	const char* PKT_DATA = "abcdefgh0123";
14	const int PKT_LEN = strlen(PKT_DATA);
15	const char* EXPECTED_DIGEST =
16		"\x52\x6c\xb8\x38\xaf\x06\x5a\xfb\x6c\x98\xbb\xc0\x9b\x0a\x7a\x1b";
17	char actual[MD5_LENGTH];
18
19	struct key md5;
20	md5.next = NULL;
21	md5.key_id = 10;
22	md5.key_len = 6;
23	memcpy(&md5.key_seq, "md5seq", md5.key_len);
24	memcpy(&md5.type, "MD5", 4);
25
26	TEST_ASSERT_EQUAL(MD5_LENGTH,
27			  make_mac((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5, actual));
28
29	TEST_ASSERT_TRUE(memcmp(EXPECTED_DIGEST, actual, MD5_LENGTH) == 0);
30}
31
32
33void test_MakeSHA1Mac(void) {
34#ifdef OPENSSL
35	const char* PKT_DATA = "abcdefgh0123";
36	const int PKT_LEN = strlen(PKT_DATA);
37	const char* EXPECTED_DIGEST =
38		"\x17\xaa\x82\x97\xc7\x17\x13\x6a\x9b\xa9"
39		"\x63\x85\xb4\xce\xbe\x94\xa0\x97\x16\x1d";
40	char actual[SHA1_LENGTH];
41
42	struct key sha1;
43	sha1.next = NULL;
44	sha1.key_id = 20;
45	sha1.key_len = 7;
46	memcpy(&sha1.key_seq, "sha1seq", sha1.key_len);
47	memcpy(&sha1.type, "SHA1", 5);
48
49	TEST_ASSERT_EQUAL(SHA1_LENGTH,
50			  make_mac((char*)PKT_DATA, PKT_LEN, SHA1_LENGTH, &sha1, actual));
51
52	TEST_ASSERT_TRUE(memcmp(EXPECTED_DIGEST, actual, SHA1_LENGTH) == 0);
53#else
54	TEST_IGNORE_MESSAGE("OpenSSL not found, skipping...");
55#endif	/* OPENSSL */
56}
57
58
59void test_VerifyCorrectMD5(void) {
60	const char* PKT_DATA =
61		"sometestdata"		// Data
62		"\0\0\0\0"			// Key-ID (unused)
63		"\xc7\x58\x99\xdd\x99\x32\x0f\x71" // MAC
64		"\x2b\x7b\xfe\x4f\xa2\x32\xcf\xac";
65	const int PKT_LEN = 12;
66
67	struct key md5;
68	md5.next = NULL;
69	md5.key_id = 0;
70	md5.key_len = 6;
71	memcpy(&md5.key_seq, "md5key", md5.key_len);
72	memcpy(&md5.type, "MD5", 4);
73
74	TEST_ASSERT_TRUE(auth_md5((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5));
75}
76
77
78void test_VerifySHA1(void) {
79#ifdef OPENSSL
80	const char* PKT_DATA =
81		"sometestdata"		// Data
82		"\0\0\0\0"			// Key-ID (unused)
83		"\xad\x07\xde\x36\x39\xa6\x77\xfa\x5b\xce" // MAC
84		"\x2d\x8a\x7d\x06\x96\xe6\x0c\xbc\xed\xe1";
85	const int PKT_LEN = 12;
86
87	struct key sha1;
88	sha1.next = NULL;
89	sha1.key_id = 0;
90	sha1.key_len = 7;
91	memcpy(&sha1.key_seq, "sha1key", sha1.key_len);
92	memcpy(&sha1.type, "SHA1", 5);
93
94	TEST_ASSERT_TRUE(auth_md5((char*)PKT_DATA, PKT_LEN, SHA1_LENGTH, &sha1));
95#else
96	TEST_IGNORE_MESSAGE("OpenSSL not found, skipping...");
97#endif	/* OPENSSL */
98}
99
100void test_VerifyFailure(void) {
101	/* We use a copy of the MD5 verification code, but modify
102	 * the last bit to make sure verification fails. */
103	const char* PKT_DATA =
104		"sometestdata"		// Data
105		"\0\0\0\0"			// Key-ID (unused)
106		"\xc7\x58\x99\xdd\x99\x32\x0f\x71"	// MAC
107		"\x2b\x7b\xfe\x4f\xa2\x32\xcf\x00"; // Last byte is wrong!
108	const int PKT_LEN = 12;
109
110	struct key md5;
111	md5.next = NULL;
112	md5.key_id = 0;
113	md5.key_len = 6;
114	memcpy(&md5.key_seq, "md5key", md5.key_len);
115	memcpy(&md5.type, "MD5", 4);
116
117	TEST_ASSERT_FALSE(auth_md5((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5));
118}
119
120void test_PacketSizeNotMultipleOfFourBytes(void) {
121	const char* PKT_DATA = "123456";
122	const int PKT_LEN = 6;
123	char actual[MD5_LENGTH];
124
125	struct key md5;
126	md5.next = NULL;
127	md5.key_id = 10;
128	md5.key_len = 6;
129	memcpy(&md5.key_seq, "md5seq", md5.key_len);
130	memcpy(&md5.type, "MD5", 4);
131
132	TEST_ASSERT_EQUAL(0, make_mac((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5, actual));
133}
134
135