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

AzureAD · Jul 26, 2023 · 057ceab · 057ceab
1 parent d1fd8fa
commit 057ceab
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
         }
     }
 }