mockable

Author: DeagleGross

src/DataProtection/DataProtection/test/Microsoft.AspNetCore.DataProtection.Tests/Cng/CngAuthenticatedEncryptorBaseTests.cs

@@ -83,46 +83,63 @@ public void Decrypt_HandlesEmptyCiphertextPointerFixup()
         Assert.Equal(new byte[] { 0x20, 0x21, 0x22 }, retVal);
     }
 
-    internal abstract class MockableEncryptor : IOptimizedAuthenticatedEncryptor, ISpanAuthenticatedEncryptor, IDisposable
+    internal abstract unsafe class MockableEncryptor : IOptimizedAuthenticatedEncryptor, ISpanAuthenticatedEncryptor, IDisposable
     {
         public abstract byte[] DecryptHook(IntPtr pbCiphertext, uint cbCiphertext, IntPtr pbAdditionalAuthenticatedData, uint cbAdditionalAuthenticatedData);
         public abstract byte[] EncryptHook(IntPtr pbPlaintext, uint cbPlaintext, IntPtr pbAdditionalAuthenticatedData, uint cbAdditionalAuthenticatedData, uint cbPreBuffer, uint cbPostBuffer);
 
-        public int GetEncryptedSize(int plainTextLength)
-        {
-            throw new NotImplementedException();
-        }
+        public int GetEncryptedSize(int plainTextLength) => 1000;
+        public int GetDecryptedSize(int cipherTextLength) => 1000;
 
-        public int GetDecryptedSize(int cipherTextLength)
+        public byte[] Decrypt(ArraySegment<byte> ciphertext, ArraySegment<byte> additionalAuthenticatedData)
         {
-            throw new NotImplementedException();
-        }
+            fixed (byte* pbCiphertext = ciphertext.Array)
+            fixed (byte* pbAAD = additionalAuthenticatedData.Array)
+            {
+                IntPtr ptrCiphertext = (IntPtr)(pbCiphertext + ciphertext.Offset);
+                IntPtr ptrAAD = (IntPtr)(pbAAD + additionalAuthenticatedData.Offset);
 
-        public bool TryEncrypt(ReadOnlySpan<byte> plaintext, ReadOnlySpan<byte> additionalAuthenticatedData, Span<byte> destination, out int bytesWritten)
-        {
-            throw new NotImplementedException();
+                return DecryptHook(ptrCiphertext, (uint)ciphertext.Count, ptrAAD, (uint)additionalAuthenticatedData.Count);
+            }
         }
 
-        public bool TryDecrypt(ReadOnlySpan<byte> cipherText, ReadOnlySpan<byte> additionalAuthenticatedData, Span<byte> destination, out int bytesWritten)
+        public byte[] Encrypt(ArraySegment<byte> plaintext, ArraySegment<byte> additionalAuthenticatedData, uint preBufferSize, uint postBufferSize)
         {
-            throw new NotImplementedException();
-        }
+            fixed (byte* pbPlaintext = plaintext.Array)
+            fixed (byte* pbAAD = additionalAuthenticatedData.Array)
+            {
+                IntPtr ptrPlaintext = (IntPtr)(pbPlaintext + plaintext.Offset);
+                IntPtr ptrAAD = (IntPtr)(pbAAD + additionalAuthenticatedData.Offset);
 
-        public byte[] Decrypt(ArraySegment<byte> ciphertext, ArraySegment<byte> additionalAuthenticatedData)
-        {
-            throw new NotImplementedException();
+                return EncryptHook(ptrPlaintext, (uint)plaintext.Count, ptrAAD, (uint)additionalAuthenticatedData.Count, preBufferSize, postBufferSize);
+            }
         }
 
         public byte[] Encrypt(ArraySegment<byte> plaintext, ArraySegment<byte> additionalAuthenticatedData)
+            => Encrypt(plaintext, additionalAuthenticatedData, 0, 0);
+
+        public bool TryEncrypt(ReadOnlySpan<byte> plaintext, ReadOnlySpan<byte> additionalAuthenticatedData, Span<byte> destination, out int bytesWritten)
         {
-            throw new NotImplementedException();
+            var encrypted = Encrypt(ToArraySegment(plaintext), ToArraySegment(additionalAuthenticatedData));
+            encrypted.CopyTo(destination);
+            bytesWritten = encrypted.Length;
+            return true;
         }
 
-        public byte[] Encrypt(ArraySegment<byte> plaintext, ArraySegment<byte> additionalAuthenticatedData, uint preBufferSize, uint postBufferSize)
+        public bool TryDecrypt(ReadOnlySpan<byte> cipherText, ReadOnlySpan<byte> additionalAuthenticatedData, Span<byte> destination, out int bytesWritten)
         {
-            throw new NotImplementedException();
+            var encrypted = Decrypt(ToArraySegment(cipherText), ToArraySegment(additionalAuthenticatedData));
+            encrypted.CopyTo(destination);
+            bytesWritten = encrypted.Length;
+            return true;
         }
 
         public void Dispose() { }
+
+        ArraySegment<byte> ToArraySegment(ReadOnlySpan<byte> span)
+        {
+            var array = span.ToArray();
+            return new ArraySegment<byte>(array);
+        }
     }
 }

src/DataProtection/samples/KeyManagementSimulator/Program.cs

@@ -282,8 +282,16 @@ sealed class MockAuthenticatedEncryptor : ISpanAuthenticatedEncryptor
     public byte[] Decrypt(ArraySegment<byte> ciphertext, ArraySegment<byte> _additionalAuthenticatedData) => ciphertext.ToArray();
     public byte[] Encrypt(ArraySegment<byte> plaintext, ArraySegment<byte> _additionalAuthenticatedData) => plaintext.ToArray();
 
+    public int GetDecryptedSize(int cipherTextLength) => cipherTextLength;
     public int GetEncryptedSize(int plainTextLength) => plainTextLength;
 
+    public bool TryDecrypt(ReadOnlySpan<byte> cipherText, ReadOnlySpan<byte> additionalAuthenticatedData, Span<byte> destination, out int bytesWritten)
+    {
+        var result = cipherText.TryCopyTo(destination);
+        bytesWritten = destination.Length;
+        return result;
+    }
+
     public bool TryEncrypt(ReadOnlySpan<byte> plainText, ReadOnlySpan<byte> additionalAuthenticatedData, Span<byte> destination, out int bytesWritten)
     {
         var result = plainText.TryCopyTo(destination);