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