From cd4d5581c1044f2d383f406438c24a5a257d1d2f Mon Sep 17 00:00:00 2001
From: Sun Yimin <emmansun@users.noreply.github.com>
Date: Mon, 11 Sep 2023 17:20:02 +0800
Subject: [PATCH] sm3: add more test vectors

---
 sm3/sm3_test.go | 252 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 252 insertions(+)

diff --git a/sm3/sm3_test.go b/sm3/sm3_test.go
index 7e2512d..0603e1e 100644
--- a/sm3/sm3_test.go
+++ b/sm3/sm3_test.go
@@ -5,6 +5,7 @@ import (
 	"crypto/sha256"
 	"encoding"
 	"encoding/base64"
+	"encoding/hex"
 	"fmt"
 	"hash"
 	"io"
@@ -98,6 +99,257 @@ func TestGoldenMarshal(t *testing.T) {
 	}
 }
 
+var sm3TestVector = []struct {
+	out       string
+	in        string
+}{
+	// Test vectors from Crypto++
+	{
+		"1ab21d8355cfa17f8e61194831e81a8f22bec8c728fefb747ed035eb5082aa2b",
+		"",
+	},
+	{
+		"82ec580fe6d36ae4f81cae3c73f4a5b3b5a09c943172dc9053c69fd8e18dca1e",
+		"61626364",
+	},
+	{
+		"b58b85b795b34879c354428f7c78cd1486c4ef25ea4c5d68e611ff41c15731ef",
+		"6162636461626364",
+	},
+	{
+		"fd959b2560dadd0c0839144be6090cb665915156179c1fa6dc00292da7a2b9c2",
+		"616263646162636461626364",
+	},
+	{
+		"639c6f6b30d93ecebd559a953ba2eb72705db7d2be82bbf32979380e02124971",
+		"61626364616263646162636461626364",
+	},
+	{
+		"3f0371287a1d1fd198e12bba07e94ae5815dc7e06ba45856b6e53e56f1594f23",
+		"6162636461626364616263646162636461626364",
+	},
+	{
+		"8d15c0b9e7540b5f41b359774127ee51d126a3c780357336c7d39d6ffc01f130",
+		"616263646162636461626364616263646162636461626364",
+	},
+	{
+		"0a5a2fafba54c2a6593b18c5877c50c26bee5369bc7f07b0c66a641e49295419",
+		"61626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"73edef5c9d3710f14dbaf892f50ce9dfab48e462d837d93ec0f9422c5f2a4007",
+		"6162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"28a6a907842a5b4a360cead2ff6f0b96f1b28c12e5c9ed0be58169c26863b0d8",
+		"616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"035be9acd343d3711a61972ea6a80d4deb38e40c901f1cd20786cf57c82ce8ed",
+		"61626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"7a04f699def87c0ef8a9dd44d46a71a39e6b594bc467298d04454e52aa922dca",
+		"6162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"6fe77b627c1a12bc367d13c8f07b32ffbde1a537b1b9cb061bf7d75a692e02bb",
+		"616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"4251f280685451d9f73d75e6d59f4e3f140f3a0b22f9cb0416e4dd15c1410d11",
+		"61626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"9a032f0cf27e4b408f252452d451cac51a422d43ae73ab6cd7ec2483241358e9",
+		"6162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"7b0685c88114bee154296c262d619a3d43c4fbf325d5dcb6f2bbdbf96d4275db",
+		"616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"debe9ff92275b8a138604889c18e5a4d6fdb70e5387e5765293dcba39c0c5732",
+		"61626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"860f7ad118996a6f631c5e4ac693157aefda97a18a873d3323f64c28a8a44fc5",
+		"6162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"0c5e778ef656184f8c9ee54f0fe0c9ce5059e02c771325184619be82c92a8c5a",
+		"616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"bb78eaeb6b00d13d43ab682b8d65512aa9e91fbc8c6c2841c6b96345f44f9652",
+		"61626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"716f5396cc9312609c28c645344ec695fdabdb9fbf11de36c1b33ac5291cca6c",
+		"6162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"24ce4fbaab353814f890561cbde3d10308c33ac6831ff74e236e8e0525b2cd4e",
+		"616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"5852844cbc2d742d4b129392646b3a1029c54f9813eec409012bf2c4bf0acdeb",
+		"61626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"3efb9d84992a87607dc43d91fb818cdbebe3f6cfac66456495f6cb922f9c2fb3",
+		"6162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	{
+		"994bf36eb6c0099b21d6ead2cf71490ea57aa845f0feed97d7f8ce2788e22342",
+		"616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364616263646162636461626364",
+	},
+	// Test vectors from GmSSL
+	{
+		"2b6173f01b9002cac00ec91c8b481867c2e35bf408bde154d62844da1b5b6e18",
+		"8f",
+	},
+	{
+		"b49515e07f9e777070d725fd49d41579d263cffb4ec3fa65b93c87b473d91cb9",
+		"bf23",
+	},
+	{
+		"171e937ace4b4e18fef1efbae2feaaa7f1022f722ffc82155a24e75ae4ac4b29",
+		"7a9719",
+	},
+	{
+		"041326870d40091c58ce7a252458e86cf19664ca6aa7e11db8952329b7688a1c",
+		"fb39e8cc",
+	},
+	{
+		"b2249a453544ca8699e07b369382001e8d741c76cad673851a4a25be5c1765cd",
+		"95aaf89fe9",
+	},
+	{
+		"5ac22e272e9c6e7cef2c17178302817d865cb178c6c04b331b0e5ee61c2014f9",
+		"7688740e973d",
+	},
+	{
+		"667509d27e794d11621817e993b8da4accacb3f83a719e635e0ea455905040de",
+		"d7b976b5f139c5",
+	},
+	{
+		"911b051b02636a5efe12ec8fafa3b18d55b1aa6c4bc9e07789b2d44470b94167",
+		"391b4861669a3de0",
+	},
+	{
+		"429debaa692b6aeb85aae6666ed2f8c206395d6a87560699dca23d6035918b28",
+		"e71ec750cc768837a8",
+	},
+	{
+		"140f2130d810310c5d0a988e9b380678031c4a9050aabe1a7b5603a98f452684",
+		"898fb851bb90bceefb6d",
+	},
+	{
+		"3b4c28a7bd3ab41b65525f580edcbe1a308c3fa09d821c192858a349581ca55b",
+		"7d33665e8ef4f522b55df7",
+	},
+	{
+		"ecc8181aaffc4dbce05c8ad2b3544bf6838791d6c9966de2996318b33ee37b8b",
+		"815e16c63b984946c07ad55a",
+	},
+	{
+		"5b617c321f8769b5227bc02471c117f44dfdf816ebabb60575d07edef1f880c0",
+		"87eb36f4116f4d2aa05c9d2b56",
+	},
+	{
+		"911048c7853c404970a9230fd997a95b5b4a00ff74a8aafbbce72b4f9444de8f",
+		"f670d1ed03316874263b039cb53a",
+	},
+	{
+		"ddc82e25eb0caa33ee7229a12ac69a7b9b72345236e835857e81bd2173edc532",
+		"102cb525fa7e0de28efe6d2f2ad1b1",
+	},
+	{
+		"ad2b435cda7615f0ac367aeca843a69189b665552233f4804439090f86f3009f",
+		"74c2e33d49e8c2a5144e82edc04218f9",
+	},
+	{
+		"78b026963e21183970483449ee9272a877a2254c89c63f0f6a0b4b86dea1db33",
+		"706c9721e103771b688e97636ab26b412c",
+	},
+	{
+		"eb46d32b15ebf5c5fd2af316c9717065cc21376053945f04cb6f92d31e1666e9",
+		"ac24ded9e54844b3e0ca037ec1d0f8103fec",
+	},
+	{
+		"c1761751af06a5be5233bd7a8d6fc497fc1ea3f36cafb988e457f281615bf867",
+		"1b39c5a77a7cd6d538c3d2d1079f3f5338c620",
+	},
+	{
+		"7239694c9b61cdf9e62d5d48bafdc283f2041a9c0845e0dd975bb3a8e5a01198",
+		"65a647b8dd26dcaece299788b13ac05a0e71c577",
+	},
+	{
+		"6b70357406f59521ba34f961aa4c7d78a088db325595d6267af5d499f1d95b1f",
+		"98f13468ff3c69382495d34ba12e1e4bd895bec29a",
+	},
+	{
+		"d98dea9c0a614fb689750c75fd4c6d5e91447104f8f563dbea1f50602da018f5",
+		"03de8240c26bbaef801d579d2f2be849ffeab779e056",
+	},
+	{
+		"acfaaad49a8b94d1df81ea951a30ee2b899fba79f60c452f63af3c79e1283944",
+		"a733f4adfbe6b110c1ef6fd3599b680a6fa5644fbcd073",
+	},
+	{
+		"1ec1f230391a74818cf79b24c9b4df63f0a291e317eed70990c65efad9b5ad33",
+		"1cb083e53c6242d1dd62510ccfb3d5496d692b8017023bae",
+	},
+	{
+		"9aa7e33e6d4edc5e487a4a15b9bfa013d167d256afd86df7770468107df19f87",
+		"6a8000baf113ddee93437ce36c9b28f69f0df4a0500633eeab",
+	},
+	{
+		"6f5321a4bd11eb49dcfc9214d1391fcb733bedca2148f05a18afcc3d141211e6",
+		"d2d424446f9b2b29efcf93f2da96208a90a39fb0dc7e46ad8f53",
+	},
+	{
+		"6194eeda953c297c20ec9876e4f0d51c44c1ccc4037cc90e6ba184572c84ef2f",
+		"db1bcaf89851cbcc847347eb4b1d0c40ea6f3e0de1853d713704bc",
+	},
+	{
+		"e715a625d3d1271f8a3aadcf215f91161c1bc161bd645627b78c707f0c8b3b3f",
+		"bab74a186887702204977bceefb1bba1d52c9c58667e7fe0e3f4bc19",
+	},
+	{
+		"4c77eb46a03cc08c37ee567127c0bce8f952ba5fdd729ff6fc44e3ffe307a507",
+		"36cdc511b3082afd547f4331298f879aa742c634c10e0263f80efdeece",
+	},
+	{
+		"6a1ad547946c39bb22dda3dc8b861010790d8cebf4cf8dc3c9493ca8d4c92f70",
+		"072bedbcbe5f9295ec6a2bf22490d984bc7388fe99035916c1262e8b592c",
+	},
+	{
+		"8abe85411e3ce2c2a2e582eda291d45fcd3a4a5cd7e1851a7e67a49fa2f3ba9a",
+		"6009dced63eef9985da162f1beda1b56e0c17d51119b1aaeee7b28050bd1dd",
+	},
+	{
+		"355422a0c526088b16c5207d1adad939cf9642150ebe110e213f23884f8db789",
+		"45dc649b6c57a9e21e9b069d06e12729b80a62facd901c6e9ed10c38ba5c2e65",
+	},
+	{
+		"d730e1a1482889c9f4f2b661941c2e1ea6a12f1c78e70908df164b33c6226dee",
+		"8d8ac08987feb09407263bc7cd471911372f1daebe2c879b258c0efbef9c544aba",
+	},
+}
+
+
+func TestSM3(t *testing.T) {
+	for i, tt := range sm3TestVector {
+		input, _ := hex.DecodeString(tt.in)
+		res := Sum(input)
+		if hex.EncodeToString(res[:]) != tt.out {
+			t.Errorf("case %v failed, in: %v ", i, tt.in)
+		}
+	}
+}
+
 func TestSize(t *testing.T) {
 	c := New()
 	if got := c.Size(); got != Size {