public void TestRandomBytes(Symmetric_AES.KeySize keySize, int?numBytes)
{
var key = Symmetric_AES.GenerateRandomKey(keySize);
byte[] input = null;
if (numBytes.HasValue)
{
input = GetRandomBytes(numBytes.Value, numBytes.Value);
}
byte[] encrypted, decrypted;
using (var algorithm = new Symmetric_AES_ECB())
{
encrypted = algorithm.EncryptBytes(input, key);
}
using (var algorithm = new Symmetric_AES_ECB())
{
decrypted = algorithm.DecryptBytes(encrypted, key);
}
if (input == null)
{
Assert.IsNull(encrypted, "encrypted");
Assert.IsNull(decrypted, "decrypted");
}
else
{
Assert.IsTrue(encrypted.Length >= numBytes, "encrypted.Length");
Assert.IsTrue(decrypted.SequenceEqual(input), "Decrypted does not match original.");
}
}
public void TestIncorrectKeyFails(Symmetric_AES.KeySize keySize)
{
var key = Symmetric_AES.GenerateRandomKey(keySize);
byte[] input = GetRandomBytes(1000, 5000);
byte[] encrypted;
using (var algorithm = new Symmetric_AES_ECB())
{
encrypted = algorithm.EncryptBytes(input, key);
}
key = Symmetric_AES.GenerateRandomKey(keySize);
using (var algorithm = new Symmetric_AES_ECB())
{
algorithm.DecryptBytes(encrypted, key);
}
}
public void TestSameInputProducesSameOutput(Symmetric_AES.KeySize keySize)
{
var key = Symmetric_AES.GenerateRandomKey(keySize);
for (int counter = 0; counter < 10; ++counter)
{
var encryptedAsBase64 = new HashSet <string>();
using (var algorithm = new Symmetric_AES_ECB())
{
var inputBytes = GetRandomBytes(1024, 4096);
for (int i = 0; i < 100; ++i)
{
encryptedAsBase64.Add(Convert.ToBase64String(algorithm.EncryptBytes(inputBytes, key)));
}
}
Assert.AreEqual(1, encryptedAsBase64.Count, "Should be only 1 distinct value");
}
}
public void TestUnicodeString(Symmetric_AES.KeySize keySize)
{
var key = Symmetric_AES.GenerateRandomKey(keySize);
string input = Convert.ToBase64String(SecureRandomizer.GetRandomBytes(100)) + "\u01e2\u01f0\u020e\u0229";
byte[] encrypted;
string decrypted;
using (var algorithm = new Symmetric_AES_ECB())
{
encrypted = algorithm.EncryptString(input, key);
}
using (var algorithm = new Symmetric_AES_ECB())
{
Assert.IsTrue(encrypted.Length > 100, "encrypted.Length");
decrypted = algorithm.DecryptBytes(encrypted, key).AsString();
}
Assert.AreEqual(input, decrypted, string.Format("{0} | {1}", input, decrypted));
}
public void TestRandomStream(Symmetric_AES.KeySize keySize, int?numBytes)
{
var key = Symmetric_AES.GenerateRandomKey(keySize);
byte[] input = null;
if (numBytes.HasValue)
{
input = GetRandomBytes(numBytes.Value, numBytes.Value);
}
byte[] encrypted = null;
using (var inputStream = numBytes.HasValue ? new MemoryStream(input) : null)
{
using (var outputStream = numBytes.HasValue ? new MemoryStream() : null)
{
using (var algorithm = new Symmetric_AES_ECB())
{
try
{
algorithm.EncryptStream(inputStream, outputStream, key);
if (numBytes.HasValue)
{
encrypted = outputStream.ToArray();
Assert.IsTrue(encrypted.Length >= numBytes, "encrypted.Length");
}
else
{
Assert.Fail("Should have thrown exception");
}
}
catch (Exception ex)
{
if (numBytes.HasValue)
{
Assert.Fail("Should not have thrown exception");
}
else
{
Assert.AreEqual(typeof(ArgumentNullException), ex.GetType(), "Exception type");
}
}
}
}
}
using (var inputStream = numBytes.HasValue ? new MemoryStream(encrypted) : null)
{
using (var outputStream = numBytes.HasValue ? new MemoryStream() : null)
{
using (var algorithm = new Symmetric_AES_ECB())
{
algorithm.DecryptStream(inputStream, outputStream, key);
}
if (numBytes.HasValue)
{
byte[] decrypted = outputStream.ToArray();
Assert.IsTrue(decrypted.SequenceEqual(input), "Decrypted does not match original.");
}
else
{
Assert.IsNull(inputStream, "inputStream");
Assert.IsNull(outputStream, "outputStream");
}
}
}
}
