Use one-shot hash APIs in a couple of places (#2159)

AzureAD · Sep 7, 2023 · 834e1f7 · 834e1f7
1 parent 3285bc2
commit 834e1f7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/src/Microsoft.IdentityModel.Protocols.SignedHttpRequest/SignedHttpRequestHandler.cs b/src/Microsoft.IdentityModel.Protocols.SignedHttpRequest/SignedHttpRequestHandler.cs
@@ -1213,11 +1213,18 @@ private static string CalculateBase64UrlEncodedHash(string data)
 
         private static string CalculateBase64UrlEncodedHash(byte[] bytes)
         {
+            byte[] hashedBytes;
+
+#if NET6_0_OR_GREATER
+            hashedBytes = SHA256.HashData(bytes);
+#else
             using (var hash = SHA256.Create())
             {
-                var hashedBytes = hash.ComputeHash(bytes);
-                return Base64UrlEncoder.Encode(hashedBytes);
+                hashedBytes = hash.ComputeHash(bytes);
             }
+#endif
+
+            return Base64UrlEncoder.Encode(hashedBytes);
         }
 
         /// <summary>

diff --git a/src/Microsoft.IdentityModel.Tokens/Utility.cs b/src/Microsoft.IdentityModel.Tokens/Utility.cs
@@ -251,10 +251,16 @@ internal static void Zero(byte[] byteArray)
 
         internal static byte[] GenerateSha256Hash(string input)
         {
+            byte[] bytes = Encoding.UTF8.GetBytes(input);
+
+#if NET6_0_OR_GREATER
+            return SHA256.HashData(bytes);
+#else
             using (var hash = SHA256.Create())
             {
-                return hash.ComputeHash(Encoding.UTF8.GetBytes(input));
+                return hash.ComputeHash(bytes);
             }
+#endif
         }
     }
 }