5 # include "openssl/err.h"
6 # include "openssl/rand.h"
7 # include "openssl/evp.h"
10 #include "ntp_stdlib.h"
12 u_long current_time = 4;
16 * Example packet with MD5 hash calculated manually.
18 const int keytype = KEY_TYPE_MD5;
19 const u_char *key = (const u_char*)"abcdefgh";
20 const u_short keyLength = 8;
21 const u_char *packet = (const u_char*)"ijklmnopqrstuvwx";
22 #define packetLength 16
24 #define digestLength 16
25 #define totalLength (packetLength + keyIdLength + digestLength)
27 u_char u8 [totalLength];
30 "ijklmnopqrstuvwx\0\0\0\0\x0c\x0e\x84\xcf\x0b\xb7\xa8\x68\x8e\x52\x38\xdb\xbc\x1c\x39\x53"
33 u_char u8 [totalLength];
36 "ijklmnopqrstuvwx\0\0\0\0\x0c\x0e\x84\xcf\x0b\xb7\xa8\x68\x8e\x52\x38\xdb\xbc\x1c\x39\x54"
40 void test_Encrypt(void);
41 void test_DecryptValid(void);
42 void test_DecryptInvalid(void);
43 void test_IPv4AddressToRefId(void);
44 void test_IPv6AddressToRefId(void);
52 packetPtr = emalloc(totalLength * sizeof(*packetPtr));
54 memset(packetPtr + packetLength, 0, keyIdLength);
55 memcpy(packetPtr, packet, packetLength);
57 cache_secretsize = keyLength;
59 length = MD5authencrypt(keytype, key, packetPtr, packetLength);
61 TEST_ASSERT_TRUE(MD5authdecrypt(keytype, key, packetPtr, packetLength, length));
63 TEST_ASSERT_EQUAL(20, length);
64 TEST_ASSERT_EQUAL_MEMORY(expectedPacket.u8, packetPtr, totalLength);
70 test_DecryptValid(void) {
71 cache_secretsize = keyLength;
72 TEST_ASSERT_TRUE(MD5authdecrypt(keytype, key, expectedPacket.u32, packetLength, 20));
76 test_DecryptInvalid(void) {
77 cache_secretsize = keyLength;
78 TEST_ASSERT_FALSE(MD5authdecrypt(keytype, key, invalidPacket.u32, packetLength, 20));
82 test_IPv4AddressToRefId(void) {
84 addr.sa4.sin_family = AF_INET;
87 addr.sa4.sin_port = htons(80);
89 address = inet_addr("192.0.2.1");
90 addr.sa4.sin_addr.s_addr = address;
92 TEST_ASSERT_EQUAL(address, addr2refid(&addr));
96 test_IPv6AddressToRefId(void) {
97 const int expected = 0x75cffd52;
98 const struct in6_addr address = { { {
99 0x20, 0x01, 0x0d, 0xb8,
100 0x85, 0xa3, 0x08, 0xd3,
101 0x13, 0x19, 0x8a, 0x2e,
102 0x03, 0x70, 0x73, 0x34
106 addr.sa6.sin6_family = AF_INET6;
108 addr.sa6.sin6_addr = address;
112 TEST_ASSERT_EQUAL(expected, addr2refid(&addr));
115 TEST_IGNORE_MESSAGE("Skipping because of big endian problem?");