test: add unit test for hash verify

twsc-james · twsc-james · commit 8ef55e84e680 · 2024-07-24T15:06:06.000+08:00
diff --git a/EccSDK.sln b/EccSDK.sln
@@ -2,6 +2,8 @@
 Microsoft Visual Studio Solution File, Format Version 12.00
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "EccSDK", "EccSDK\EccSDK.csproj", "{5EAC669B-2882-4A01-BC96-F39E04092834}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "EccSdkUnitTest", "EccSdkUnitTest\EccSdkUnitTest.csproj", "{09C5648F-11F0-446C-B0EC-835BD965CB6C}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -12,5 +14,9 @@ Global
 		{5EAC669B-2882-4A01-BC96-F39E04092834}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{5EAC669B-2882-4A01-BC96-F39E04092834}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{5EAC669B-2882-4A01-BC96-F39E04092834}.Release|Any CPU.Build.0 = Release|Any CPU
+		{09C5648F-11F0-446C-B0EC-835BD965CB6C}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{09C5648F-11F0-446C-B0EC-835BD965CB6C}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{09C5648F-11F0-446C-B0EC-835BD965CB6C}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{09C5648F-11F0-446C-B0EC-835BD965CB6C}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 EndGlobal
diff --git a/EccSDK/ChameleonHashHelper.cs b/EccSDK/ChameleonHashHelper.cs
@@ -4,7 +4,7 @@
 
 namespace EccGrpcSDK;
 
-public class ChameleonHashHelper
+public static class ChameleonHashHelper
 {
     // Kn = public key
     // kn = private key
@@ -23,12 +23,13 @@ public static bool Verify(ChameleonHashRequest request, ECPoint rightChameleonHa
         return GetChameleonHash(request).Equals(rightChameleonHash);
     }
 
-    private static ECPoint GetChameleonHash(ChameleonHashRequest request)
+    public static ECPoint GetChameleonHash(ChameleonHashRequest request)
     {
         // chameleonHash = [Kn x H(m)] + [P x sessionKey]
         var msgHash = HashHelper.Sha256(request.Message);
         var rP = request.KeyPair.BasePoint.Multiply(request.Signature);
         var chameleonHash = request.KeyPair.PublicKey.Multiply(msgHash).Add(rP).Normalize();
         return chameleonHash;
     }
+
 }
diff --git a/EccSDK/EccGenerator.cs b/EccSDK/EccGenerator.cs
@@ -6,9 +6,9 @@
 
 namespace EccGrpcSDK;
 
-public class EccGenerator : IEccGenerator
+public static class EccGenerator
 {
-    public KeyPair GenerateKeyPair(int keySize)
+    public static KeyPair GenerateKeyPair(int keySize)
     {
         var gen = new ECKeyPairGenerator("ECDSA");
         gen.Init(new KeyGenerationParameters(new SecureRandom(), keySize));
@@ -22,9 +22,11 @@ public KeyPair GenerateKeyPair(int keySize)
         {
             PublicKey = publicKey.Q, 
             PrivateKey = privateKey.D,
-            BasePoint = publicKey.Parameters.G
+            BasePoint = publicKey.Parameters.G,
+            KeySize = keySize 
         };
         
         return generateKeyPair;
     }
+    
 }
diff --git a/EccSDK/IEccGenerator.cs b/EccSDK/IEccGenerator.cs
@@ -1,5 +1,6 @@
 using EccGrpcSDK.models;
 using Org.BouncyCastle.Crypto;
+using Org.BouncyCastle.Math;
 
 namespace EccGrpcSDK;
 
diff --git a/EccSDK/models/KeyPair.cs b/EccSDK/models/KeyPair.cs
@@ -8,4 +8,5 @@ public class KeyPair
     public BigInteger PrivateKey { get; set; }
     public ECPoint PublicKey { get; set; }
     public ECPoint BasePoint { get; set; }
+    public int KeySize { get; set; }
 }
diff --git a/EccSdkUnitTest/EccSdkTests.cs b/EccSdkUnitTest/EccSdkTests.cs
@@ -0,0 +1,48 @@
+using EccGrpcSDK;
+using EccGrpcSDK.models;
+using Org.BouncyCastle.Math;
+
+namespace EccSdkUnitTest;
+
+[TestFixture]
+public class EccSdkTests
+{
+    private KeyPair _keyPair;
+
+    [SetUp]
+    public void SetUp()
+    {
+        _keyPair = EccGenerator.GenerateKeyPair(256);
+    }
+
+    [Test]
+    public void should_pass_verification()
+    {
+        var sign = ChameleonHashHelper.Sign(new ChameleonHashRequest
+        {
+            KeyPair = _keyPair,
+            Message = "Hello World",
+            Order = _keyPair.PublicKey.Curve.Order,
+            SessionKey = new BigInteger("1234567890")
+        });
+
+        var chameleonHash = ChameleonHashHelper.GetChameleonHash(new ChameleonHashRequest
+        {
+            KeyPair = _keyPair,
+            Message = "Hello World",
+            Order = _keyPair.PublicKey.Curve.Order,
+            SessionKey = new BigInteger("1234567890"),
+            Signature = sign
+        });
+
+        var result = ChameleonHashHelper.Verify(new ChameleonHashRequest
+        {
+            KeyPair = _keyPair,
+            Message = "Hello World",
+            Order = _keyPair.PublicKey.Curve.Order,
+            Signature = sign
+        }, chameleonHash);
+
+        Assert.That(result, Is.True);
+    }
+}
diff --git a/EccSdkUnitTest/EccSdkUnitTest.csproj b/EccSdkUnitTest/EccSdkUnitTest.csproj
@@ -0,0 +1,28 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <TargetFramework>net8.0</TargetFramework>
+        <ImplicitUsings>enable</ImplicitUsings>
+        <Nullable>enable</Nullable>
+
+        <IsPackable>false</IsPackable>
+        <IsTestProject>true</IsTestProject>
+    </PropertyGroup>
+
+    <ItemGroup>
+        <PackageReference Include="coverlet.collector" Version="6.0.0"/>
+        <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.8.0"/>
+        <PackageReference Include="NUnit" Version="3.14.0"/>
+        <PackageReference Include="NUnit.Analyzers" Version="3.9.0"/>
+        <PackageReference Include="NUnit3TestAdapter" Version="4.5.0"/>
+    </ItemGroup>
+
+    <ItemGroup>
+        <Using Include="NUnit.Framework"/>
+    </ItemGroup>
+
+    <ItemGroup>
+      <ProjectReference Include="..\EccSDK\EccSDK.csproj" />
+    </ItemGroup>
+
+</Project>

Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@`
`4`	`4`
`5`	`5`	`namespace EccGrpcSDK;`
`6`	`6`
`7`		`-public class ChameleonHashHelper`
	`7`	`+public static class ChameleonHashHelper`
`8`	`8`	`{`
`9`	`9`	`// Kn = public key`
`10`	`10`	`// kn = private key`
`@@ -23,12 +23,13 @@ public static bool Verify(ChameleonHashRequest request, ECPoint rightChameleonHa`
`23`	`23`	`return GetChameleonHash(request).Equals(rightChameleonHash);`
`24`	`24`	`}`
`25`	`25`
`26`		`- private static ECPoint GetChameleonHash(ChameleonHashRequest request)`
	`26`	`+ public static ECPoint GetChameleonHash(ChameleonHashRequest request)`
`27`	`27`	`{`
`28`	`28`	`// chameleonHash = [Kn x H(m)] + [P x sessionKey]`
`29`	`29`	`var msgHash = HashHelper.Sha256(request.Message);`
`30`	`30`	`var rP = request.KeyPair.BasePoint.Multiply(request.Signature);`
`31`	`31`	`var chameleonHash = request.KeyPair.PublicKey.Multiply(msgHash).Add(rP).Normalize();`
`32`	`32`	`return chameleonHash;`
`33`	`33`	`}`
	`34`	`+`
`34`	`35`	`}`
Original file line number	Diff line number	Diff line change
`@@ -6,9 +6,9 @@`
`6`	`6`
`7`	`7`	`namespace EccGrpcSDK;`
`8`	`8`
`9`		`-public class EccGenerator : IEccGenerator`
	`9`	`+public static class EccGenerator`
`10`	`10`	`{`
`11`		`- public KeyPair GenerateKeyPair(int keySize)`
	`11`	`+ public static KeyPair GenerateKeyPair(int keySize)`
`12`	`12`	`{`
`13`	`13`	`var gen = new ECKeyPairGenerator("ECDSA");`
`14`	`14`	`gen.Init(new KeyGenerationParameters(new SecureRandom(), keySize));`
`@@ -22,9 +22,11 @@ public KeyPair GenerateKeyPair(int keySize)`
`22`	`22`	`{`
`23`	`23`	`PublicKey = publicKey.Q,`
`24`	`24`	`PrivateKey = privateKey.D,`
`25`		`- BasePoint = publicKey.Parameters.G`
	`25`	`+ BasePoint = publicKey.Parameters.G,`
	`26`	`+ KeySize = keySize`
`26`	`27`	`};`
`27`	`28`
`28`	`29`	`return generateKeyPair;`
`29`	`30`	`}`
	`31`	`+`
`30`	`32`	`}`
Original file line number	Diff line number	Diff line change
`@@ -8,4 +8,5 @@ public class KeyPair`
`8`	`8`	`public BigInteger PrivateKey { get; set; }`
`9`	`9`	`public ECPoint PublicKey { get; set; }`
`10`	`10`	`public ECPoint BasePoint { get; set; }`
	`11`	`+ public int KeySize { get; set; }`
`11`	`12`	`}`