Add C3ID implementation

src/Cask/CrossCompanyCorrelatingId.cs

@@ -0,0 +1,96 @@
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+
+using System.Diagnostics;
+using System.Security.Cryptography;
+using System.Text;
+
+using static CommonAnnotatedSecurityKeys.Limits;
+
+namespace CommonAnnotatedSecurityKeys;
+
+internal static class CrossCompanyCorrelatingId
+{
+    /// <summary>
+    /// The size of a Cross-Company Correlating ID (aka C3ID) in bytes.
+    /// </summary>
+    public const int SizeInBytes = 15;
+
+    private static ReadOnlySpan<byte> CompanyPrefix => "Cross"u8;
+    private static ReadOnlySpan<byte> CompanySuffix => "CorrelatingId:"u8;
+    private static ReadOnlySpan<byte> Hex => "0123456789ABCDEF"u8;
+    private const int HexCharsPerByte = 2;
+
+    /// <summary>
+    /// Computes the Cross-Company Correlating Id (aka C3ID) bytes for the given
+    /// company and text and writes them to the destination span.
+    /// </summary>
+    public static void Compute(string company, string text, Span<byte> destination)
+    {
+        Debug.Assert(destination.Length >= SizeInBytes);
+
+        // Input: $"Cross{company}CorrelatingId:{SHA256Hex(text))}" encoded in UTF-8 
+        int companyByteCount = Encoding.UTF8.GetByteCount(company);
+        int inputByteCount =
+            CompanyPrefix.Length +
+            companyByteCount +
+            CompanySuffix.Length +
+            (SHA256.HashSizeInBytes * HexCharsPerByte);
+
+        Span<byte> input = inputByteCount <= MaxStackAlloc ? stackalloc byte[inputByteCount] : new byte[inputByteCount];
+        Span<byte> inputDestination = input;
+
+        // 'Cross'
+        CompanyPrefix.CopyTo(inputDestination);
+        inputDestination = inputDestination[CompanyPrefix.Length..];
+
+        // {company}
+        Encoding.UTF8.GetBytes(company.AsSpan(), inputDestination);
+        inputDestination = inputDestination[companyByteCount..];
+
+        // 'CorrelatingId:'
+        CompanySuffix.CopyTo(inputDestination);
+        inputDestination = inputDestination[CompanySuffix.Length..];
+
+        // SHA256 hash of UTF-8 encoded text, converted to uppercase UTF-8 encoded hex
+        Sha256Hex(text, inputDestination);
+
+        // Compute second SHA256 of above input, truncate, and copy to destination
+        Span<byte> sha = stackalloc byte[SHA256.HashSizeInBytes];
+        SHA256.HashData(input, sha);
+        sha[..SizeInBytes].CopyTo(destination);
+    }
+
+    /// <summary>
+    /// Computes the SHA256 of the text encoded as UTF-8 and writes the result
+    /// to the destination as UTF-8 encoded uppercase hex.
+    /// </summary>
+    private static void Sha256Hex(string text, Span<byte> destination)
+    {
+        Debug.Assert(destination.Length >= SHA256.HashSizeInBytes * HexCharsPerByte);
+
+        int byteCount = Encoding.UTF8.GetByteCount(text);
+        Span<byte> bytes = byteCount <= MaxStackAlloc ? stackalloc byte[byteCount] : new byte[byteCount];
+        Encoding.UTF8.GetBytes(text.AsSpan(), bytes);
+
+        Span<byte> sha = stackalloc byte[SHA256.HashSizeInBytes];
+        SHA256.HashData(bytes, sha);
+        ConvertToHex(sha, destination);
+    }
+
+    /// <summary>
+    /// Converts bytes to UTF-8 encoded uppercase hex. Directly, without
+    /// allocation or UTF-16 to UTF-8 conversion.
+    /// </summary>
+    private static void ConvertToHex(ReadOnlySpan<byte> bytes, Span<byte> destination)
+    {
+        Debug.Assert(destination.Length >= bytes.Length * HexCharsPerByte);
+
+        for (int src = 0, dst = 0; src < bytes.Length; src++, dst += HexCharsPerByte)
+        {
+            byte b = bytes[src];
+            destination[dst] = Hex[b >> 4];
+            destination[dst + 1] = Hex[b & 0xF];
+        }
+    }
+}

src/Cask/Polyfill.cs

@@ -69,6 +69,11 @@ internal static class Extensions
     {
         public static unsafe string GetString(this Encoding encoding, ReadOnlySpan<byte> bytes)
         {
+            if (bytes.Length == 0)
+            {
+                return string.Empty;
+            }
+
             fixed (byte* ptr = bytes)
             {
                 return encoding.GetString(ptr, bytes.Length);
@@ -77,6 +82,11 @@ public static unsafe string GetString(this Encoding encoding, ReadOnlySpan<byte>
 
         public static unsafe int GetByteCount(this Encoding encoding, ReadOnlySpan<char> chars)
         {
+            if (chars.Length == 0)
+            {
+                return 0;
+            }
+
             fixed (char* ptr = chars)
             {
                 return encoding.GetByteCount(ptr, chars.Length);
@@ -85,6 +95,11 @@ public static unsafe int GetByteCount(this Encoding encoding, ReadOnlySpan<char>
 
         public static unsafe int GetBytes(this Encoding encoding, ReadOnlySpan<char> chars, Span<byte> bytes)
         {
+            if (chars.Length == 0)
+            {
+                return 0;
+            }
+
             fixed (char* charPtr = chars)
             fixed (byte* bytePtr = bytes)
             {

src/Tests/Cask.Tests/CrossCompanyCorrelatingIdTests.cs

@@ -0,0 +1,40 @@
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+
+using Xunit;
+
+namespace CommonAnnotatedSecurityKeys.Tests;
+
+public class CrossCompanyCorrelatingIdTests
+{
+    [Theory]
+    [InlineData("", "EZ3GxRsKq+Dp21GvyCpQ")]
+    [InlineData("Hello world", "R8ogeP7QfTFvL5qAATry")]
+    [InlineData("😁", "f/BTV0j6A8km4KDw7aJz")]
+    public void Test_Basic(string text, string expected)
+    {
+        string actual = ComputeC3IDBase64(company: "Microsoft", text);
+        Assert.Equal(expected, actual);
+    }
+
+    [Fact]
+    public void Test_LargeText()
+    {
+        string actual = ComputeC3IDBase64(company: "Microsoft", text: new string('x', 300));
+        Assert.Equal("QjHXB4Bu8voB3eJcJagI", actual);
+    }
+
+    [Fact]
+    public void Test_LargeCompany()
+    {
+        string actual = ComputeC3IDBase64(company: new string('x', 300), text: "test");
+        Assert.Equal("rG1CONo8M3lcBqzxyIpf", actual);
+    }
+
+    private static string ComputeC3IDBase64(string company, string text)
+    {
+        byte[] bytes = new byte[CrossCompanyCorrelatingId.SizeInBytes];
+        CrossCompanyCorrelatingId.Compute(company, text, bytes);
+        return Convert.ToBase64String(bytes);
+    }
+}

src/Tests/Cask.Tests/PolyfillTests.cs

@@ -271,6 +271,31 @@ public void Random_NotDeterministic()
         Assert.False(random1.SequenceEqual(random2), "RandomNumberGenerator produced two identical 32-byte sequences.");
     }
 
+    [Fact]
+    public void Encoding_GetString_Empty()
+    {
+        ReadOnlySpan<byte> data = [];
+        string text = Encoding.UTF8.GetString(data);
+        Assert.Equal("", text);
+    }
+
+    [Fact]
+    public void Encoding_GetByteCount_Empty()
+    {
+        ReadOnlySpan<char> text = "".AsSpan();
+        int byteCount = Encoding.UTF8.GetByteCount(text);
+        Assert.Equal(0, byteCount);
+    }
+
+    [Fact]
+    public void Encoding_GetBytes_Empty()
+    {
+        ReadOnlySpan<char> text = "".AsSpan();
+        Span<byte> bytes = [];
+        int bytesWritten = Encoding.UTF8.GetBytes(text, bytes);
+        Assert.Equal(0, bytesWritten);
+    }
+
 #if NETFRAMEWORK // We don't need to stress test the modern BCL :)
     [Fact]
     public async Task Polyfill_ThreadingStress()