Learning/mRemoteNG
1
0
Fork 0
mirror of https://github.com/mRemoteNG/mRemoteNG.git synced 2026-02-17 14:07:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added tests for the key generator

Browse Source
This commit is contained in:
David Sparer
2016-10-13 18:01:36 -06:00
parent 966f7b2d61
commit 1f9218bd5a
3 changed files with 98 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

91
mRemoteNGTests/Security/KeyDerivation/Pkcs5S2KeyGeneratorTests.cs Normal file
View File

@@ -0,0 +1,91 @@
using System;
using mRemoteNG.Security.KeyDerivation;
using NUnit.Framework;
namespace mRemoteNGTests.Security.KeyDerivation
{
public class Pkcs5S2KeyGeneratorTests
{
[Test]
public void DefaultConstructorThrowsNoException()
{
// ReSharper disable once ObjectCreationAsStatement
Assert.DoesNotThrow(() => new Pkcs5S2KeyGenerator());
}
[Test]
public void CreatingGeneratorWithLowIterationCountThrowsError()
{
// ReSharper disable once ObjectCreationAsStatement
Assert.Throws<ArgumentOutOfRangeException>(() => new Pkcs5S2KeyGenerator(256, 999));
}
[Test]
public void CreatingGeneratorWithNegativeKeyBitSizeThrowsError()
{
// ReSharper disable once ObjectCreationAsStatement
Assert.Throws<ArgumentOutOfRangeException>(() => new Pkcs5S2KeyGenerator(-1));
}
[Test]
public void IdenticalParametersProduceIdenticalKeys()
{
var keyDerivationFunction = new Pkcs5S2KeyGenerator();
var key1 = keyDerivationFunction.DeriveKey("", new byte[0]);
var key2 = keyDerivationFunction.DeriveKey("", new byte[0]);
Assert.That(key1, Is.EquivalentTo(key2));
}
[Test]
public void DifferingIterationsProduceDifferingKeys()
{
var keyDerivationFunction1 = new Pkcs5S2KeyGenerator(256, 1001);
var keyDerivationFunction2 = new Pkcs5S2KeyGenerator(256, 1002);
var key1 = keyDerivationFunction1.DeriveKey("", new byte[0]);
var key2 = keyDerivationFunction2.DeriveKey("", new byte[0]);
Assert.That(key1, Is.Not.EquivalentTo(key2));
}
[Test]
public void DifferingKeysizeProduceDifferingKeys()
{
var keyDerivationFunction1 = new Pkcs5S2KeyGenerator();
var keyDerivationFunction2 = new Pkcs5S2KeyGenerator(512);
var key1 = keyDerivationFunction1.DeriveKey("", new byte[0]);
var key2 = keyDerivationFunction2.DeriveKey("", new byte[0]);
Assert.That(key1, Is.Not.EquivalentTo(key2));
}
[Test]
public void DifferingPasswordsProduceDifferingKeys()
{
var keyDerivationFunction = new Pkcs5S2KeyGenerator();
var key1 = keyDerivationFunction.DeriveKey("a", new byte[0]);
var key2 = keyDerivationFunction.DeriveKey("b", new byte[0]);
Assert.That(key1, Is.Not.EquivalentTo(key2));
}
[Test]
public void DifferingSaltsProduceDifferingKeys()
{
var keyDerivationFunction = new Pkcs5S2KeyGenerator();
var key1 = keyDerivationFunction.DeriveKey("", new byte[0]);
var key2 = keyDerivationFunction.DeriveKey("", new byte[] {1});
Assert.That(key1, Is.Not.EquivalentTo(key2));
}
[TestCase(0)]
[TestCase(1)]
[TestCase(8)]
[TestCase(9)]
[TestCase(256)]
[TestCase(333)]
public void KeyLengthIsKeyBitSizeDividedBy8(int keyBitSize)
{
var keyDerivationFunction = new Pkcs5S2KeyGenerator(keyBitSize);
var key = keyDerivationFunction.DeriveKey("", new byte[0]);
Assert.That(key.Length, Is.EqualTo(keyBitSize / 8));
}
}
}

1
mRemoteNGTests/mRemoteNGTests.csproj
View File

@@ -117,6 +117,7 @@
<Compile Include="Connection\ConnectionInfoComparerTests.cs" />
<Compile Include="Connection\Protocol\ProtocolListTests.cs" />
<Compile Include="Security\Authentication\PasswordAuthenticatorTests.cs" />
<Compile Include="Security\KeyDerivation\Pkcs5S2KeyGeneratorTests.cs" />
<Compile Include="Tools\ExternalToolsArgumentParserTests.cs" />
<Compile Include="Tree\ConnectionTreeDragAndDropHandlerTests.cs" />
<Compile Include="Tree\ConnectionTreeModelTests.cs" />

7
mRemoteV1/Security/KeyDerivation/Pkcs5S2KeyGenerator.cs
View File

@@ -1,4 +1,5 @@
using Org.BouncyCastle.Crypto;
using System;
using Org.BouncyCastle.Crypto;
using Org.BouncyCastle.Crypto.Generators;
using Org.BouncyCastle.Crypto.Parameters;
@@ -12,6 +13,10 @@ namespace mRemoteNG.Security.KeyDerivation
public Pkcs5S2KeyGenerator(int keyBitSize = 256, int iterations = 1000)
{
if (iterations < 1000)
throw new ArgumentOutOfRangeException($"Minimum value of {nameof(iterations)} is 1000");
if (keyBitSize < 0)
throw new ArgumentOutOfRangeException($"{nameof(keyBitSize)} must be positive");
_keyBitSize = keyBitSize;
_iterations = iterations;
}