Skip to content

Commit

Permalink
allow user to perform detached file hash
Browse files Browse the repository at this point in the history
  • Loading branch information
@elantiguamsft
elantiguamsft committed Oct 24, 2023
1 parent 16a81fa commit fa874eb
Show file tree
Hide file tree
Showing 2 changed files with 222 additions and 52 deletions.
111 changes: 111 additions & 0 deletions CoseSign1.Tests/DetachedSignatureFactoryTests.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

namespace CoseSign1.Tests;

using System.Runtime.Intrinsics.Arm;

/// <summary>
/// Class for Testing Methods of <see cref="DetachedSignatureFactory"/>
/// </summary>
Expand Down Expand Up @@ -74,6 +76,49 @@ public async Task TestCreateDetachedSignatureAsync()
memStream.Seek(0, SeekOrigin.Begin);
}

[Test]
public async Task TestCreateDetachedSignatureHashProvidedAsync()
{
ICoseSigningKeyProvider coseSigningKeyProvider = SetupMockSigningKeyProvider(nameof(TestCreateDetachedSignatureHashProvidedAsync));
using DetachedSignatureFactory factory = new();
byte[] randomBytes = new byte[50];
new Random().NextBytes(randomBytes);
using HashAlgorithm hasher = CoseSign1MessageDetachedSignatureExtensions.CreateHashAlgorithmFromName(factory.HashAlgorithmName)
?? throw new Exception($"Failed to get hash algorithm from {nameof(CoseSign1MessageDetachedSignatureExtensions.CreateHashAlgorithmFromName)}");
byte[] hash = hasher!.ComputeHash(randomBytes);
using MemoryStream hashStream = new(hash);

// test the sync method
Assert.Throws<ArgumentNullException>(() => factory.CreateDetachedSignature(hash, coseSigningKeyProvider, string.Empty));
CoseSign1Message detachedSignature = factory.CreateDetachedSignature(hash, coseSigningKeyProvider, "application/test.payload", payloadHashed: true);
detachedSignature.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-sha256");
detachedSignature.SignatureMatches(randomBytes).Should().BeTrue();

Assert.Throws<ArgumentNullException>(() => factory.CreateDetachedSignature(hashStream, coseSigningKeyProvider, string.Empty));
hashStream.Seek(0, SeekOrigin.Begin);
CoseSign1Message detachedSignature2 = factory.CreateDetachedSignature(hashStream, coseSigningKeyProvider, "application/test.payload", payloadHashed: true);
detachedSignature2.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature2.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-sha256");
detachedSignature2.SignatureMatches(randomBytes).Should().BeTrue();
hashStream.Seek(0, SeekOrigin.Begin);

// test the async methods
Assert.ThrowsAsync<ArgumentNullException>(() => factory.CreateDetachedSignatureAsync(hash, coseSigningKeyProvider, string.Empty));
CoseSign1Message detachedSignature3 = await factory.CreateDetachedSignatureAsync(hash, coseSigningKeyProvider, "application/test.payload", payloadHashed: true);
detachedSignature3.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature3.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-sha256");
detachedSignature3.SignatureMatches(randomBytes).Should().BeTrue();

Assert.ThrowsAsync<ArgumentNullException>(() => factory.CreateDetachedSignatureAsync(hashStream, coseSigningKeyProvider, string.Empty));
hashStream.Seek(0, SeekOrigin.Begin);
CoseSign1Message detachedSignature4 = await factory.CreateDetachedSignatureAsync(hashStream, coseSigningKeyProvider, "application/test.payload", payloadHashed: true);
detachedSignature4.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature4.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-sha256");
detachedSignature4.SignatureMatches(randomBytes).Should().BeTrue();
hashStream.Seek(0, SeekOrigin.Begin);
}

[Test]
public async Task TestCreateDetachedSignatureBytesAsync()
{
Expand Down Expand Up @@ -114,6 +159,49 @@ public async Task TestCreateDetachedSignatureBytesAsync()
detachedSignature4.SignatureMatches(memStream).Should().BeTrue();
}

[Test]
public async Task TestCreateDetachedSignatureBytesHashProvidedAsync()
{
ICoseSigningKeyProvider coseSigningKeyProvider = SetupMockSigningKeyProvider(nameof(TestCreateDetachedSignatureBytesHashProvidedAsync));
using DetachedSignatureFactory factory = new();
byte[] randomBytes = new byte[50];
new Random().NextBytes(randomBytes);
using HashAlgorithm hasher = CoseSign1MessageDetachedSignatureExtensions.CreateHashAlgorithmFromName(factory.HashAlgorithmName)
?? throw new Exception($"Failed to get hash algorithm from {nameof(CoseSign1MessageDetachedSignatureExtensions.CreateHashAlgorithmFromName)}");
byte[] hash = hasher!.ComputeHash(randomBytes);
using MemoryStream hashStream = new(hash);

// test the sync method
Assert.Throws<ArgumentNullException>(() => factory.CreateDetachedSignatureBytes(hash, coseSigningKeyProvider, string.Empty));
CoseSign1Message detachedSignature = CoseMessage.DecodeSign1(factory.CreateDetachedSignatureBytes(hash, coseSigningKeyProvider, "application/test.payload", payloadHashed: true).ToArray());
detachedSignature.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-sha256");
detachedSignature.SignatureMatches(randomBytes).Should().BeTrue();

Assert.Throws<ArgumentNullException>(() => factory.CreateDetachedSignatureBytes(hashStream, coseSigningKeyProvider, string.Empty));
hashStream.Seek(0, SeekOrigin.Begin);
CoseSign1Message detachedSignature2 = CoseMessage.DecodeSign1(factory.CreateDetachedSignatureBytes(hashStream, coseSigningKeyProvider, "application/test.payload", payloadHashed: true).ToArray());
detachedSignature2.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature2.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-sha256");
detachedSignature2.SignatureMatches(randomBytes).Should().BeTrue();
hashStream.Seek(0, SeekOrigin.Begin);

// test the async methods
Assert.ThrowsAsync<ArgumentNullException>(() => factory.CreateDetachedSignatureBytesAsync(hash, coseSigningKeyProvider, string.Empty));
CoseSign1Message detachedSignature3 = CoseMessage.DecodeSign1((await factory.CreateDetachedSignatureBytesAsync(hash, coseSigningKeyProvider, "application/test.payload", payloadHashed: true)).ToArray());
detachedSignature3.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature3.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-sha256");
detachedSignature3.SignatureMatches(randomBytes).Should().BeTrue();

Assert.ThrowsAsync<ArgumentNullException>(() => factory.CreateDetachedSignatureBytesAsync(hashStream, coseSigningKeyProvider, string.Empty));
hashStream.Seek(0, SeekOrigin.Begin);
CoseSign1Message detachedSignature4 = CoseMessage.DecodeSign1((await factory.CreateDetachedSignatureBytesAsync(hashStream, coseSigningKeyProvider, "application/test.payload", payloadHashed: true)).ToArray());
detachedSignature4.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature4.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-sha256");
hashStream.Seek(0, SeekOrigin.Begin);
detachedSignature4.SignatureMatches(randomBytes).Should().BeTrue();
}

[Test]
public void TestCreateDetachedSignatureMd5()
{
Expand All @@ -130,6 +218,29 @@ public void TestCreateDetachedSignatureMd5()
detachedSignature.SignatureMatches(randomBytes).Should().BeTrue();
}

[Test]
public void TestCreateDetachedSignatureMd5HashProvided()
{
ICoseSigningKeyProvider coseSigningKeyProvider = SetupMockSigningKeyProvider(nameof(TestCreateDetachedSignatureMd5));
using DetachedSignatureFactory factory = new();
byte[] randomBytes = new byte[50];
new Random().NextBytes(randomBytes);
using HashAlgorithm hasher = CoseSign1MessageDetachedSignatureExtensions.CreateHashAlgorithmFromName(HashAlgorithmName.MD5)
?? throw new Exception($"Failed to get hash algorithm from {nameof(CoseSign1MessageDetachedSignatureExtensions.CreateHashAlgorithmFromName)}");
byte[] hash = hasher!.ComputeHash(randomBytes);

// test the sync method
Assert.Throws<ArgumentNullException>(() => factory.CreateDetachedSignature(hash, coseSigningKeyProvider, string.Empty));
CoseSign1Message detachedSignature = CoseMessage.DecodeSign1(factory.CreateDetachedSignatureBytes(hash, coseSigningKeyProvider, "application/test.payload", payloadHashed: true).ToArray());
detachedSignature.ProtectedHeaders.ContainsKey(CoseHeaderLabel.ContentType).Should().BeTrue();
detachedSignature.ProtectedHeaders[CoseHeaderLabel.ContentType].GetValueAsString().Should().Be("application/test.payload+hash-md5");
detachedSignature.SignatureMatches(randomBytes).Should().BeTrue();

// test unknown hash length
// test the sync method
Assert.Throws<ArgumentException>(() => factory.CreateDetachedSignature(randomBytes, coseSigningKeyProvider, "application/test.payload", payloadHashed: true));
}

[Test]
public void TestCreateDetachedSignatureAlreadyProvided()
{
Expand Down
Loading

0 comments on commit fa874eb

Please sign in to comment.