contrib/ntp/tests/libntp/a_md5encrypt.c

   1 #include "config.h"
   2 #include "unity.h"
   3
   4 #ifdef OPENSSL
   5 # include "openssl/err.h"
   6 # include "openssl/rand.h"
   7 # include "openssl/evp.h"
   8 #endif
   9 #include "ntp.h"
  10 #include "ntp_stdlib.h"
  11
  12 u_long current_time = 4;
  13
  14
  15 /*
  16  * Example packet with MD5 hash calculated manually.
  17  */
  18 const int keytype = KEY_TYPE_MD5;
  19 const char *key = "abcdefgh";
  20 const u_short keyLength = 8;
  21 const char *packet = "ijklmnopqrstuvwx";
  22 #define packetLength 16
  23 #define keyIdLength  4
  24 #define digestLength 16
  25 const int totalLength = packetLength + keyIdLength + digestLength;
  26 const char *expectedPacket = "ijklmnopqrstuvwx\0\0\0\0\x0c\x0e\x84\xcf\x0b\xb7\xa8\x68\x8e\x52\x38\xdb\xbc\x1c\x39\x53";
  27
  28
  29 void test_Encrypt(void);
  30 void test_DecryptValid(void);
  31 void test_DecryptInvalid(void);
  32 void test_IPv4AddressToRefId(void);
  33 void test_IPv6AddressToRefId(void);
  34
  35
  36 void
  37 test_Encrypt(void) {
  38         char *packetPtr;
  39         int length;
  40
  41         packetPtr = emalloc(totalLength * sizeof(*packetPtr));
  42
  43         memset(packetPtr + packetLength, 0, keyIdLength);
  44         memcpy(packetPtr, packet, packetLength);
  45
  46         cache_secretsize = keyLength;
  47
  48         length = MD5authencrypt(keytype, (u_char*)key, (u_int32*)packetPtr, packetLength);
  49
  50         TEST_ASSERT_TRUE(MD5authdecrypt(keytype, (u_char*)key, (u_int32*)packetPtr, packetLength, length));
  51
  52         TEST_ASSERT_EQUAL(20, length);
  53         TEST_ASSERT_EQUAL_MEMORY(expectedPacket, packetPtr, totalLength);
  54
  55         free(packetPtr);
  56 }
  57
  58 void
  59 test_DecryptValid(void) {
  60         cache_secretsize = keyLength;
  61
  62         TEST_ASSERT_TRUE(MD5authdecrypt(keytype, (u_char*)key, (u_int32*)expectedPacket, packetLength, 20));
  63 }
  64
  65 void
  66 test_DecryptInvalid(void) {
  67         cache_secretsize = keyLength;
  68
  69         const char *invalidPacket = "ijklmnopqrstuvwx\0\0\0\0\x0c\x0e\x84\xcf\x0b\xb7\xa8\x68\x8e\x52\x38\xdb\xbc\x1c\x39\x54";
  70
  71         TEST_ASSERT_FALSE(MD5authdecrypt(keytype, (u_char*)key, (u_int32*)invalidPacket, packetLength, 20));
  72 }
  73
  74 void
  75 test_IPv4AddressToRefId(void) {
  76         sockaddr_u addr;
  77         addr.sa4.sin_family = AF_INET;
  78         u_int32 address;
  79
  80         addr.sa4.sin_port = htons(80);
  81
  82         address = inet_addr("192.0.2.1");
  83         addr.sa4.sin_addr.s_addr = address;
  84
  85         TEST_ASSERT_EQUAL(address, addr2refid(&addr));
  86 }
  87
  88 void
  89 test_IPv6AddressToRefId(void) {
  90         const struct in6_addr address = {
  91                 0x20, 0x01, 0x0d, 0xb8,
  92                 0x85, 0xa3, 0x08, 0xd3,
  93                 0x13, 0x19, 0x8a, 0x2e,
  94                 0x03, 0x70, 0x73, 0x34
  95         };
  96         sockaddr_u addr;
  97
  98         addr.sa6.sin6_family = AF_INET6;
  99
 100         addr.sa6.sin6_addr = address;
 101
 102         const int expected = 0x75cffd52;
 103
 104 #if 0
 105         TEST_ASSERT_EQUAL(expected, addr2refid(&addr));
 106 #else
 107         TEST_IGNORE_MESSAGE("Skipping because of big endian problem?");
 108 #endif
 109 }