Add MD5

awslabs · Feb 6, 2024 · 586667c · 586667c
1 parent 73c5392
commit 586667c
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/Source/AwsCommonRuntimeKit/crt/Hash.swift b/Source/AwsCommonRuntimeKit/crt/Hash.swift
@@ -8,13 +8,16 @@ import AwsCCal
 public enum HashAlgorithm {
     case SHA1
     case SHA256
+    case MD5
 
     var size: Int {
         switch self {
         case .SHA1:
             return Int(AWS_SHA1_LEN)
         case .SHA256:
             return Int(AWS_SHA256_LEN)
+        case .MD5:
+            return Int(AWS_MD5_LEN)
         }
     }
 }
@@ -29,6 +32,8 @@ public class Hash {
             rawValue = aws_sha1_new(allocator.rawValue)
         case .SHA256:
             rawValue = aws_sha256_new(allocator.rawValue)
+        case .MD5:
+            rawValue = aws_md5_new(allocator.rawValue)
         }
     }
 

diff --git a/Test/AwsCommonRuntimeKitTests/crt/HashTests.swift b/Test/AwsCommonRuntimeKitTests/crt/HashTests.swift
@@ -20,6 +20,24 @@ class HashTests: XCBaseTestCase {
 
         XCTAssertEqual(md5, "iB0/3YSo7maijL0IGOgA9g==")
     }
+
+    func testMd5PayloadWithUpdate() throws {
+        let hello = "Hello".data(using: .utf8)!
+        let world = "World".data(using: .utf8)!
+
+        let helloWorld = "HelloWorld".data(using: .utf8)!
+        let md5 = try! helloWorld.computeMD5().encodeToHexString()
+
+        let hash = Hash(algorithm: .MD5)
+        try hash.update(data: hello)
+        try hash.update(data: world)
+        let finalizedHash = try hash.finalize().encodeToHexString()
+
+
+        XCTAssertEqual(md5, finalizedHash)
+        XCTAssertEqual(finalizedHash, "68e109f0f40ca72a15e05cc22786f8e6")
+    }
+
 
     func testSha256() throws {
         let hello = "Hello".data(using: .utf8)!