ASN1: Add tests for random data resillience

outfoxx · Jun 9, 2023 · d76fd29 · d76fd29
1 parent 8b3304b
commit d76fd29
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/Tests/ASN1Tests.swift b/Tests/ASN1Tests.swift
@@ -249,4 +249,33 @@ class ASN1Tests: XCTestCase {
     XCTAssertEqual(offset(5), -40920)
   }
 
+  func testRandomData() throws {
+
+    struct TestStruct: Codable, SchemaSpecified {
+      var id: OID
+      var data: Data
+
+      static var asn1Schema: Schema {
+        .sequence([
+          "id": .objectIdentifier(),
+          "data": .octetString(size: .is(16))
+        ])
+      }
+    }
+
+    let encoded = try ASN1.Encoder.encode(TestStruct(id: [1, 2, 3, 4, 5], data: Data(count: 16)))
+
+    for _ in 0 ..< 10000 {
+
+      var random = Data(count: encoded.count)
+      random.withUnsafeMutableBytes { ptr in
+        if SecRandomCopyBytes(nil, ptr.count, ptr.baseAddress!) != errSecSuccess {
+          fatalError("SecRandomCopyBytes failed")
+        }
+      }
+
+      XCTAssertThrowsError(try ASN1.Decoder.decode(TestStruct.self, from: random))
+    }
+  }
+
 }