Test - crypto.c from sntp

bk: 55842d35bDhHEzd5GOhQWGjpsXkVXQ

diff --git a/sntp/tests/Makefile.am b/sntp/tests/Makefile.am
index cc8b08fac668d89e817c2d04d81f8c7b528cd863..1cba3719eef1721bca3114a6f0a74d42466f6877 100644 (file)
--- a/sntp/tests/Makefile.am
+++ b/sntp/tests/Makefile.am
@@ -8,6 +8,7 @@ run_unity =     cd $(srcdir) && ruby ../../sntp/unity/auto/generate_test_runner.rb
 EXTRA_PROGRAMS +=      test-keyFile
 
 check_PROGRAMS =               \
+       test-crypto         \
        test-kodDatabase        \
        test-networking         \
        $(NULL)
@@ -28,7 +29,7 @@ base_SOURCES =                        \
 
 tests_SOURCES =                        \
        $(base_SOURCES)         \
-       crypto.cpp              \
+       g_crypto.cpp            \
        keyFile.cpp             \
        g_kodDatabase.cpp       \
        kodFile.cpp             \
@@ -104,11 +105,20 @@ AM_LDFLAGS = $(LDFLAGS_NTP)
 
 
 BUILT_SOURCES +=                       \
+       $(srcdir)/run-crypto.c  \
        $(srcdir)/run-networking.c      \
        $(srcdir)/run-kodDatabase.c     \
        $(srcdir)/run-keyFile.c         \
        $(NULL)
 
+test_crypto_CFLAGS =           \
+       -I$(top_srcdir)/unity           \
+       $(NULL)
+
+test_crypto_LDADD =                    \
+       $(unity_tests_LDADD)            \
+       $(NULL)
+
 test_networking_CFLAGS =               \
        -I$(top_srcdir)/unity           \
        $(NULL)
@@ -134,6 +144,12 @@ test_keyFile_LDADD =                       \
        $(NULL)
 
 
+test_crypto_SOURCES =          \
+       crypto.c                        \
+       run-crypto.c            \
+       ../version.c                    \
+       $(NULL)
+
 test_networking_SOURCES =              \
        networking.c                    \
        run-networking.c                \
@@ -151,6 +167,9 @@ test_keyFile_SOURCES =                      \
        run-keyFile.c                   \
        $(NULL)
 
+$(srcdir)/run-crypto.c: $(srcdir)/crypto.c $(std_unity_list)
+       $(run_unity) crypto.c run-crypto.c
+
 $(srcdir)/run-networking.c: $(srcdir)/networking.c $(std_unity_list)
        $(run_unity) networking.c run-networking.c
 

diff --git a/sntp/tests/crypto.c b/sntp/tests/crypto.c
new file mode 100644 (file)
index 0000000..15ba803
--- /dev/null
+++ b/sntp/tests/crypto.c
@@ -0,0 +1,127 @@
+#include "config.h"
+#include "unity.h"
+#include "ntp_types.h"
+
+#include "c_sntptest.h"
+#include "crypto.h"
+
+#define MD5_LENGTH 16
+#define SHA1_LENGTH 20
+
+void test_MakeMd5Mac(void) {
+
+       const char* PKT_DATA = "abcdefgh0123";
+       const int PKT_LEN = strlen(PKT_DATA);
+       const char* EXPECTED_DIGEST =
+               "\x52\x6c\xb8\x38\xaf\x06\x5a\xfb\x6c\x98\xbb\xc0\x9b\x0a\x7a\x1b";
+       char actual[MD5_LENGTH];
+
+       struct key md5;
+       md5.next = NULL;
+       md5.key_id = 10;
+       md5.key_len = 6;
+       memcpy(&md5.key_seq, "md5seq", md5.key_len);
+       memcpy(&md5.type, "MD5", 4);
+
+       TEST_ASSERT_EQUAL(MD5_LENGTH,
+                         make_mac((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5, actual));
+
+       TEST_ASSERT_TRUE(memcmp(EXPECTED_DIGEST, actual, MD5_LENGTH) == 0);
+}
+
+#ifdef OPENSSL
+void test_MakeSHA1Mac(void) {
+       const char* PKT_DATA = "abcdefgh0123";
+       const int PKT_LEN = strlen(PKT_DATA);
+       const char* EXPECTED_DIGEST =
+               "\x17\xaa\x82\x97\xc7\x17\x13\x6a\x9b\xa9"
+               "\x63\x85\xb4\xce\xbe\x94\xa0\x97\x16\x1d";
+       char actual[SHA1_LENGTH];
+
+       struct key sha1;
+       sha1.next = NULL;
+       sha1.key_id = 20;
+       sha1.key_len = 7;
+       memcpy(&sha1.key_seq, "sha1seq", sha1.key_len);
+       memcpy(&sha1.type, "SHA1", 5);
+
+       TEST_ASSERT_EQUAL(SHA1_LENGTH,
+                         make_mac((char*)PKT_DATA, PKT_LEN, SHA1_LENGTH, &sha1, actual));
+
+       TEST_ASSERT_TRUE(memcmp(EXPECTED_DIGEST, actual, SHA1_LENGTH) == 0);
+}
+#endif /* OPENSSL */
+
+void test_VerifyCorrectMD5(void) {
+       const char* PKT_DATA =
+               "sometestdata"          // Data
+               "\0\0\0\0"                      // Key-ID (unused)
+               "\xc7\x58\x99\xdd\x99\x32\x0f\x71" // MAC
+               "\x2b\x7b\xfe\x4f\xa2\x32\xcf\xac";
+       const int PKT_LEN = 12;
+
+       struct key md5;
+       md5.next = NULL;
+       md5.key_id = 0;
+       md5.key_len = 6;
+       memcpy(&md5.key_seq, "md5key", md5.key_len);
+       memcpy(&md5.type, "MD5", 4);
+
+       TEST_ASSERT_TRUE(auth_md5((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5));
+}
+
+#ifdef OPENSSL
+void test_VerifySHA1(void) {
+       const char* PKT_DATA =
+               "sometestdata"          // Data
+               "\0\0\0\0"                      // Key-ID (unused)
+               "\xad\x07\xde\x36\x39\xa6\x77\xfa\x5b\xce" // MAC
+               "\x2d\x8a\x7d\x06\x96\xe6\x0c\xbc\xed\xe1";
+       const int PKT_LEN = 12;
+
+       struct key sha1;
+       sha1.next = NULL;
+       sha1.key_id = 0;
+       sha1.key_len = 7;
+       memcpy(&sha1.key_seq, "sha1key", sha1.key_len);
+       memcpy(&sha1.type, "SHA1", 5);
+
+       TEST_ASSERT_TRUE(auth_md5((char*)PKT_DATA, PKT_LEN, SHA1_LENGTH, &sha1));
+}
+#endif /* OPENSSL */
+
+void test_VerifyFailure(void) {
+       /* We use a copy of the MD5 verification code, but modify
+        * the last bit to make sure verification fails. */
+       const char* PKT_DATA =
+               "sometestdata"          // Data
+               "\0\0\0\0"                      // Key-ID (unused)
+               "\xc7\x58\x99\xdd\x99\x32\x0f\x71"      // MAC
+               "\x2b\x7b\xfe\x4f\xa2\x32\xcf\x00"; // Last byte is wrong!
+       const int PKT_LEN = 12;
+
+       struct key md5;
+       md5.next = NULL;
+       md5.key_id = 0;
+       md5.key_len = 6;
+       memcpy(&md5.key_seq, "md5key", md5.key_len);
+       memcpy(&md5.type, "MD5", 4);
+
+       TEST_ASSERT_FALSE(auth_md5((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5));
+}
+
+void test_PacketSizeNotMultipleOfFourBytes(void) {
+       const char* PKT_DATA = "123456";
+       const int PKT_LEN = 6;
+       char actual[MD5_LENGTH];
+
+       struct key md5;
+       md5.next = NULL;
+       md5.key_id = 10;
+       md5.key_len = 6;
+       memcpy(&md5.key_seq, "md5seq", md5.key_len);
+       memcpy(&md5.type, "MD5", 4);
+
+       TEST_ASSERT_EQUAL(0, make_mac((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5, actual));
+}
+

diff --git a/sntp/tests/g_crypto.cpp b/sntp/tests/g_crypto.cpp
new file mode 100644 (file)
index 0000000..a6f89f5
--- /dev/null
+++ b/sntp/tests/g_crypto.cpp
@@ -0,0 +1,129 @@
+#include "sntptest.h"
+
+extern "C" {
+#include "crypto.h"
+};
+
+class cryptoTest : public sntptest {
+};
+
+#define MD5_LENGTH 16
+#define SHA1_LENGTH 20
+
+TEST_F(cryptoTest, MakeMd5Mac) {
+
+       const char* PKT_DATA = "abcdefgh0123";
+       const int PKT_LEN = strlen(PKT_DATA);
+       const char* EXPECTED_DIGEST =
+               "\x52\x6c\xb8\x38\xaf\x06\x5a\xfb\x6c\x98\xbb\xc0\x9b\x0a\x7a\x1b";
+       char actual[MD5_LENGTH];
+
+       key md5;
+       md5.next = NULL;
+       md5.key_id = 10;
+       md5.key_len = 6;
+       memcpy(&md5.key_seq, "md5seq", md5.key_len);
+       memcpy(&md5.type, "MD5", 4);
+
+       EXPECT_EQ(MD5_LENGTH,
+                         make_mac((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5, actual));
+
+       EXPECT_TRUE(memcmp(EXPECTED_DIGEST, actual, MD5_LENGTH) == 0);
+}
+
+#ifdef OPENSSL
+TEST_F(cryptoTest, MakeSHA1Mac) {
+       const char* PKT_DATA = "abcdefgh0123";
+       const int PKT_LEN = strlen(PKT_DATA);
+       const char* EXPECTED_DIGEST =
+               "\x17\xaa\x82\x97\xc7\x17\x13\x6a\x9b\xa9"
+               "\x63\x85\xb4\xce\xbe\x94\xa0\x97\x16\x1d";
+       char actual[SHA1_LENGTH];
+
+       key sha1;
+       sha1.next = NULL;
+       sha1.key_id = 20;
+       sha1.key_len = 7;
+       memcpy(&sha1.key_seq, "sha1seq", sha1.key_len);
+       memcpy(&sha1.type, "SHA1", 5);
+
+       EXPECT_EQ(SHA1_LENGTH,
+                         make_mac((char*)PKT_DATA, PKT_LEN, SHA1_LENGTH, &sha1, actual));
+
+       EXPECT_TRUE(memcmp(EXPECTED_DIGEST, actual, SHA1_LENGTH) == 0);
+}
+#endif /* OPENSSL */
+
+TEST_F(cryptoTest, VerifyCorrectMD5) {
+       const char* PKT_DATA =
+               "sometestdata"          // Data
+               "\0\0\0\0"                      // Key-ID (unused)
+               "\xc7\x58\x99\xdd\x99\x32\x0f\x71" // MAC
+               "\x2b\x7b\xfe\x4f\xa2\x32\xcf\xac";
+       const int PKT_LEN = 12;
+
+       key md5;
+       md5.next = NULL;
+       md5.key_id = 0;
+       md5.key_len = 6;
+       memcpy(&md5.key_seq, "md5key", md5.key_len);
+       memcpy(&md5.type, "MD5", 4);
+
+       EXPECT_TRUE(auth_md5((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5));
+}
+
+#ifdef OPENSSL
+TEST_F(cryptoTest, VerifySHA1) {
+       const char* PKT_DATA =
+               "sometestdata"          // Data
+               "\0\0\0\0"                      // Key-ID (unused)
+               "\xad\x07\xde\x36\x39\xa6\x77\xfa\x5b\xce" // MAC
+               "\x2d\x8a\x7d\x06\x96\xe6\x0c\xbc\xed\xe1";
+       const int PKT_LEN = 12;
+
+       key sha1;
+       sha1.next = NULL;
+       sha1.key_id = 0;
+       sha1.key_len = 7;
+       memcpy(&sha1.key_seq, "sha1key", sha1.key_len);
+       memcpy(&sha1.type, "SHA1", 5);
+
+       EXPECT_TRUE(auth_md5((char*)PKT_DATA, PKT_LEN, SHA1_LENGTH, &sha1));
+}
+#endif /* OPENSSL */
+
+TEST_F(cryptoTest, VerifyFailure) {
+       /* We use a copy of the MD5 verification code, but modify
+        * the last bit to make sure verification fails. */
+       const char* PKT_DATA =
+               "sometestdata"          // Data
+               "\0\0\0\0"                      // Key-ID (unused)
+               "\xc7\x58\x99\xdd\x99\x32\x0f\x71"      // MAC
+               "\x2b\x7b\xfe\x4f\xa2\x32\xcf\x00"; // Last byte is wrong!
+       const int PKT_LEN = 12;
+
+       key md5;
+       md5.next = NULL;
+       md5.key_id = 0;
+       md5.key_len = 6;
+       memcpy(&md5.key_seq, "md5key", md5.key_len);
+       memcpy(&md5.type, "MD5", 4);
+
+       EXPECT_FALSE(auth_md5((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5));
+}
+
+TEST_F(cryptoTest, PacketSizeNotMultipleOfFourBytes) {
+       const char* PKT_DATA = "123456";
+       const int PKT_LEN = 6;
+       char actual[MD5_LENGTH];
+
+       key md5;
+       md5.next = NULL;
+       md5.key_id = 10;
+       md5.key_len = 6;
+       memcpy(&md5.key_seq, "md5seq", md5.key_len);
+       memcpy(&md5.type, "MD5", 4);
+
+       EXPECT_EQ(0, make_mac((char*)PKT_DATA, PKT_LEN, MD5_LENGTH, &md5, actual));
+}
+