Updated the encription fuction

This commit is contained in:
ashutosh.nehete 2025-12-06 16:08:43 +05:30
parent 4282a6e1f7
commit 9fbaf03ce4

View File

@ -3,6 +3,7 @@ using Microsoft.AspNetCore.Mvc.Filters;
using Newtonsoft.Json;
using Newtonsoft.Json.Serialization;
using System.Security.Cryptography;
using System.Text;
public class EncryptResponseAttribute : TypeFilterAttribute
{
@ -71,29 +72,62 @@ public class EncryptResponseAttribute : TypeFilterAttribute
aes.Mode = CipherMode.CBC;
aes.Padding = PaddingMode.PKCS7;
// We do NOT use 'using' on the MemoryStream here yet,
// because we need to read from it after the CryptoStream finishes.
// 1. Convert string to bytes directly (Avoids StreamWriter encoding issues)
var plainBytes = Encoding.UTF8.GetBytes(plainText);
using var ms = new MemoryStream();
// Write IV first (16 bytes)
// 2. Write IV (16 bytes)
ms.Write(aes.IV, 0, aes.IV.Length);
using (var encryptor = aes.CreateEncryptor(aes.Key, aes.IV))
using (var cs = new CryptoStream(ms, encryptor, CryptoStreamMode.Write))
using (var sw = new StreamWriter(cs))
{
// CRITICAL FIX: Use Async Write
await sw.WriteAsync(plainText);
// 3. Write Data
await cs.WriteAsync(plainBytes, 0, plainBytes.Length);
// Flush the writer, but do not close the underlying streams yet via 'using' exit
await sw.FlushAsync();
// 4. CRITICAL: Flush the final block (Padding) to the MemoryStream
// Without this, Dart receives incomplete data and throws "Invalid Padding"
cs.FlushFinalBlock();
}
// At this point, CryptoStream is closed (disposed by using block),
// causing the final block to be flushed to MemoryStream.
// MemoryStream is technically closed, but .NET allows ToArray() on closed MemoryStreams.
// 5. Convert full stream to Base64
return Convert.ToBase64String(ms.ToArray());
}
//private async Task<string> EncryptAsync(string plainText)
//{
// if (string.IsNullOrEmpty(plainText)) return plainText;
// using var aes = Aes.Create();
// aes.Key = Convert.FromBase64String(_keyBase64);
// aes.GenerateIV();
// aes.Mode = CipherMode.CBC;
// aes.Padding = PaddingMode.PKCS7;
// // We do NOT use 'using' on the MemoryStream here yet,
// // because we need to read from it after the CryptoStream finishes.
// using var ms = new MemoryStream();
// // Write IV first (16 bytes)
// ms.Write(aes.IV, 0, aes.IV.Length);
// using (var encryptor = aes.CreateEncryptor(aes.Key, aes.IV))
// using (var cs = new CryptoStream(ms, encryptor, CryptoStreamMode.Write))
// using (var sw = new StreamWriter(cs))
// {
// // CRITICAL FIX: Use Async Write
// await sw.WriteAsync(plainText);
// // Flush the writer, but do not close the underlying streams yet via 'using' exit
// await sw.FlushAsync();
// }
// // At this point, CryptoStream is closed (disposed by using block),
// // causing the final block to be flushed to MemoryStream.
// // MemoryStream is technically closed, but .NET allows ToArray() on closed MemoryStreams.
// return Convert.ToBase64String(ms.ToArray());
//}
}
}