From: Lokesh Walase Date: Fri, 19 Jun 2015 14:54:45 +0000 (+0530) Subject: Test - crypto.c from sntp X-Git-Tag: NTP_4_3_43~2^2~7^2~1^2 X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=9265f809103214abbf4c82225dde806e6a4f959f;p=thirdparty%2Fntp.git Test - crypto.c from sntp bk: 55842d35bDhHEzd5GOhQWGjpsXkVXQ --- diff --git a/sntp/tests/Makefile.am b/sntp/tests/Makefile.am index cc8b08fac..1cba3719e 100644 --- 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 index 000000000..15ba803a0 --- /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 index 000000000..a6f89f5aa --- /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)); +} +