Imported Upstream version 2.0.0.11353upstream/2.0.0.11353

1 files changed, 44 insertions, 0 deletions

diff --git a/src/mscorlib/shared/Interop/Windows/BCrypt/Interop.BCryptGenRandom.cs b/src/mscorlib/shared/Interop/Windows/BCrypt/Interop.BCryptGenRandom.cs
new file mode 100644
index 0000000000..bc357125b5
--- /dev/null
+++ b/src/mscorlib/shared/Interop/Windows/BCrypt/Interop.BCryptGenRandom.cs
@@ -0,0 +1,44 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+using System;
+using System.Diagnostics;
+using System.Runtime.InteropServices;
+
+internal partial class Interop
+{
+    internal partial class BCrypt
+    {
+        internal static unsafe int BCryptGenRandom(byte* pbBuffer, int count)
+        {
+            Debug.Assert(pbBuffer != null);
+            Debug.Assert(count >= 0);
+
+            return BCryptGenRandom(IntPtr.Zero, pbBuffer, count, BCRYPT_USE_SYSTEM_PREFERRED_RNG);
+        }
+
+        private const int BCRYPT_USE_SYSTEM_PREFERRED_RNG = 0x00000002;
+        internal const int STATUS_SUCCESS = 0x0;
+        internal const int STATUS_NO_MEMORY = unchecked((int)0xC0000017);
+
+        [DllImport(Libraries.BCrypt, CharSet = CharSet.Unicode)]
+        private static unsafe extern int BCryptGenRandom(IntPtr hAlgorithm, byte* pbBuffer, int cbBuffer, int dwFlags);
+    }
+
+    internal static unsafe void GetRandomBytes(byte* buffer, int length)
+    {
+        int status = BCrypt.BCryptGenRandom(buffer, length);
+        if (status != BCrypt.STATUS_SUCCESS)
+        {
+            if (status == BCrypt.STATUS_NO_MEMORY)
+            {
+                throw new OutOfMemoryException();
+            }
+            else
+            {
+                throw new InvalidOperationException();
+            }
+        }
+    }
+}