Cleanup crypto utilities (#2549)

* Consolidate crypto functions into single module

* Migrate rest of code to new crypto module

* Fix name conflict

1 files changed, 173 insertions, 0 deletions

diff --git a/tools/slang-unit-test/unit-test-crypto.cpp b/tools/slang-unit-test/unit-test-crypto.cpp
new file mode 100644
index 000000000..adb7b2218
--- /dev/null
+++ b/tools/slang-unit-test/unit-test-crypto.cpp
@@ -0,0 +1,173 @@
+// unit-test-sha1.cpp
+#include "tools/unit-test/slang-unit-test.h"
+
+#include "../../source/core/slang-crypto.h"
+
+using namespace Slang;
+
+SLANG_UNIT_TEST(crypto)
+{
+    // HashDigest
+    {
+        using Digest = HashDigest<8>;
+        Digest empty;
+        SLANG_CHECK(empty.data[0] == 0 && empty.data[1] == 0);
+        SLANG_CHECK(empty.toString() == "0000000000000000");
+
+        Digest all = Digest("ffffffffffffffff");
+        SLANG_CHECK(all.data[0] == 0xffffffff && all.data[1] == 0xffffffff);
+        SLANG_CHECK(all.toString() == "ffffffffffffffff");
+
+        SLANG_CHECK(empty == Digest("0000000000000000"));
+        SLANG_CHECK(all == Digest("ffffffffffffffff"));
+        SLANG_CHECK(empty != all);
+
+        SLANG_CHECK(Digest("invalid") == empty);
+        SLANG_CHECK(Digest("X000000000000000") == empty);
+        SLANG_CHECK(Digest(" 000000000000000") == empty);
+
+        SLANG_CHECK(Digest("0123456789abcdef").toString() == "0123456789abcdef");
+
+        Slang::ComPtr<ISlangBlob> blob = Digest("0123456789abcdef").toBlob();
+        const uint8_t check[] = { 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef };
+        SLANG_CHECK(blob->getBufferSize() == 8);
+        SLANG_CHECK(::memcmp(blob->getBufferPointer(), check, 8) == 0);
+
+        SLANG_CHECK(Digest(blob) == Digest("0123456789abcdef"));
+    }
+
+    // MD5
+
+    // Empty string
+    {
+        MD5 sha1;
+        auto digest = sha1.finalize();
+        SLANG_CHECK(digest.toString() == "d41d8cd98f00b204e9800998ecf8427e");
+    }
+
+    // One call to update()
+    {
+        MD5 sha1;
+        const String str("Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.");
+        sha1.update(str.getBuffer(), str.getLength());
+        auto digest = sha1.finalize();
+        SLANG_CHECK(digest.toString() == "818c6e601a24f72750da0f6c9b8ebe28");
+    }
+
+    // Two calls to update()
+    {
+        MD5 sha1;
+        const String str1("Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.");
+        const String str2("Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.");
+        sha1.update(str1.getBuffer(), str1.getLength());
+        sha1.update(str2.getBuffer(), str2.getLength());
+        auto digest = sha1.finalize();
+        SLANG_CHECK(digest.toString() == "87d3caecb0ab82faae84d60fde994aca");
+    }
+
+    // compute()
+    {
+        SLANG_CHECK(MD5::compute(nullptr, 0).toString() == "d41d8cd98f00b204e9800998ecf8427e");
+        const String str("Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.");
+        SLANG_CHECK(MD5::compute(str.getBuffer(), str.getLength()).toString() == "818c6e601a24f72750da0f6c9b8ebe28");
+    }    
+
+    // SHA1
+
+    // Empty string
+    {
+        SHA1 sha1;
+        auto digest = sha1.finalize();
+        SLANG_CHECK(digest.toString() == "da39a3ee5e6b4b0d3255bfef95601890afd80709");
+    }
+
+    // One call to update()
+    {
+        SHA1 sha1;
+        const String str("Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.");
+        sha1.update(str.getBuffer(), str.getLength());
+        auto digest = sha1.finalize();
+        SLANG_CHECK(digest.toString() == "cca0871ecbe200379f0a1e4b46de177e2d62e655");
+    }
+
+    // Two calls to update()
+    {
+        SHA1 sha1;
+        const String str1("Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.");
+        const String str2("Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.");
+        sha1.update(str1.getBuffer(), str1.getLength());
+        sha1.update(str2.getBuffer(), str2.getLength());
+        auto digest = sha1.finalize();
+        SLANG_CHECK(digest.toString() == "7a8213edf9976d2e693f27bbc7dc41546bcfcc97");
+    }
+
+    // compute()
+    {
+        SLANG_CHECK(SHA1::compute(nullptr, 0).toString() == "da39a3ee5e6b4b0d3255bfef95601890afd80709");
+        const String str("Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.");
+        SLANG_CHECK(SHA1::compute(str.getBuffer(), str.getLength()).toString() == "cca0871ecbe200379f0a1e4b46de177e2d62e655");
+    }
+
+    // DigestBuider
+
+    // Raw numerical values, etc.
+    {
+        DigestBuilder<MD5> builder;
+
+        int64_t valueA = -1;
+        uint64_t valueB = 1;
+        builder.append(valueA);
+        builder.append(valueB);
+
+        auto digest = builder.finalize();
+        SLANG_CHECK(digest.toString() == "5ba171e20898bdd205639013746f2679");
+    }
+
+    // List
+    {
+        DigestBuilder<MD5> builder;
+
+        List<int64_t> listA;
+        listA.add(1);
+        listA.add(2);
+        listA.add(3);
+        listA.add(4);
+        builder.append(listA);
+
+        auto digest = builder.finalize();
+        SLANG_CHECK(digest.toString() == "9f66c130786a1a05e4731f71a3c5f172");
+    }
+
+    // UnownedStringSlice
+    {
+        DigestBuilder<MD5> builder;
+
+        UnownedStringSlice stringSlice = UnownedStringSlice("String Slice Test");
+        builder.append(stringSlice);
+
+        auto digest = builder.finalize();
+        SLANG_CHECK(digest.toString() == "5d6cc58e1824a4dfd0cf57395b603316");
+    }
+
+    // String
+    {
+        DigestBuilder<MD5> builder;
+
+        String str = String("String Test");
+        builder.append(str);
+
+        auto digest = builder.finalize();
+        SLANG_CHECK(digest.toString() == "df5a79cc2170c7401cf0a506ceb0ce24");
+    }
+
+    // Digest
+    {
+        DigestBuilder<MD5> builder;
+
+        MD5::Digest hash;
+        builder.append(hash);
+
+        auto digest = builder.finalize();
+        SLANG_CHECK(digest.toString() == "4ae71336e44bf9bf79d2752e234818a5");
+    }    
+}