Ramana Kumar d453296005
Add some verify proof tests in csharp
The numbers here are based on the Python tests
2022-10-05 23:35:43 +01:00

76 lines
2.8 KiB
C#

using System;
using System.Numerics;
using System.Text;
using System.Runtime.InteropServices;
class ckzg {
[DllImport("ckzg.dll", EntryPoint="bytes_to_bls_field_wrap")]
public static extern IntPtr bytes_to_bls_field(byte[] bytes);
[DllImport("ckzg.dll", EntryPoint="verify_kzg_proof_wrap")]
public static extern int verify_kzg_proof(byte[] c, byte[] x, byte[] y, byte[] p, IntPtr ts);
[DllImport("ckzg.dll", EntryPoint="load_trusted_setup_wrap")]
public static extern IntPtr load_trusted_setup(string filename);
[DllImport("ckzg.dll", EntryPoint="free_trusted_setup_wrap")]
public static extern void free_trusted_setup(IntPtr ts);
[DllImport("ckzg.dll", EntryPoint="free")]
private static extern void free(IntPtr p);
[DllImport("ckzg.dll", EntryPoint="uint64s_from_bls_field")]
private static extern IntPtr uint64s_from_bls_field(IntPtr fr);
public static BigInteger int_from_bls_field(IntPtr fr) {
IntPtr uptr = uint64s_from_bls_field(fr);
Int64[] int64s = new Int64[4];
Marshal.Copy(uptr, int64s, 0, 4);
free(uptr);
BigInteger result = new BigInteger(0);
BigInteger mult = new BigInteger(1);
for (int i = 0; i < 4; i++) {
result += Convert.ToUInt64(int64s[i]) * mult;
mult *= BigInteger.Pow(2, 64);
}
return result;
}
}
class tests {
// Convert.FromHexString replacement (since mono does not seem to have new enough C# libs)
public static byte[] HexadecimalStringToByteArray(String hexadecimalString)
{
int length = hexadecimalString.Length;
byte[] byteArray = new byte[length / 2];
for (int i = 0; i < length; i += 2){
byteArray[i / 2] = Convert.ToByte(hexadecimalString.Substring(i, 2), 16);
}
return byteArray;
}
private static void Main(string[] args)
{
IntPtr ts = ckzg.load_trusted_setup("../../src/trusted_setup.txt");
if (ts == IntPtr.Zero) {
Console.WriteLine("Failed to load trusted setup.");
return;
}
byte[] c = HexadecimalStringToByteArray("b91c022acf7bd3b63be69a4c19b781ea7a3d5df1cd66ceb7dd0f399610f0ee04695dace82e04bfb83af2b17d7319f87f");
byte[] x = HexadecimalStringToByteArray("0345f802a75a6c0d9cc5b8a1e71642b8fa80b0a78938edc6da1e591149578d1a");
byte[] y = HexadecimalStringToByteArray("3b17cab634c3795d311380f3bc93ce8e768efc0e2b9e79496cfc8f351594b472");
byte[] p = HexadecimalStringToByteArray("a5ddd6da04c47a9cd4628beb8d55ebd2e930a64dfa29f876ebf393cfd6574d48a3ce96ac5a2af4a4f9ec9caa47d304d3");
int result = ckzg.verify_kzg_proof(c, x, y, p, ts);
System.Diagnostics.Trace.Assert(result == 1, "Verification failed");
x[0] = 0x42;
result = ckzg.verify_kzg_proof(c, x, y, p, ts);
System.Diagnostics.Trace.Assert(result == 0, "Verification succeeded incorrectly");
ckzg.free_trusted_setup(ts);
Console.WriteLine("Tests passed");
}
}