1 files changed, 59 insertions, 0 deletions
diff --git a/test_crypto.c b/test_crypto.c
new file mode 100644
index 0000000..67894eb
--- /dev/null
+++ b/test_crypto.c
@@ -0,0 +1,59 @@
+
+#include "brandt.h"
+#include "crypto.h"
+#include "test.h"
+
+int
+test_brandt_ec_keypair_create ()
+{
+	gcry_mpi_t skey;
+	gcry_mpi_t x1 = gcry_mpi_new(0);
+	gcry_mpi_t x2 = gcry_mpi_new(0);
+	gcry_mpi_t y1 = gcry_mpi_new(0);
+	gcry_mpi_t y2 = gcry_mpi_new(0);
+	gcry_mpi_point_t pkey1;
+	gcry_mpi_point_t pkey2 = gcry_mpi_point_new(0);
+
+	brandt_ec_keypair_create (&pkey1, &skey);
+	check(skey, "no sec key created");
+	check(pkey1, "no pub key created");
+	check(pkey2, "could not init pkey2");
+	check(x1, "could not init x1");
+	check(x2, "could not init x2");
+	check(y1, "could not init y1");
+	check(y2, "could not init y2");
+
+	gcry_mpi_ec_mul(pkey2, skey, ec_gen, ec_ctx);
+	check(!gcry_mpi_ec_get_affine(x1, y1, pkey1, ec_ctx), "could not get affine coords for pkey1");
+	check(!gcry_mpi_ec_get_affine(x2, y2, pkey2, ec_ctx), "could not get affine coords for pkey1");
+	check(!gcry_mpi_cmp(x1, x2), "x component does not match");
+	check(!gcry_mpi_cmp(y1, y2), "y component does not match");
+
+	gcry_mpi_release(skey);
+	gcry_mpi_release(x1);
+	gcry_mpi_release(x2);
+	gcry_mpi_release(y1);
+	gcry_mpi_release(y2);
+	gcry_mpi_point_release(pkey1);
+	gcry_mpi_point_release(pkey2);
+}
+
+int
+main (int argc, char *argv[])
+{
+	int repeat = 100;
+	int i;
+
+	BRANDT_init();
+
+	for (i = 0; i < repeat; i++)
+	{
+		run(test_brandt_ec_keypair_create);
+	}
+
+	if (!ret)
+	{
+		fputs("All tests passed", stderr);
+	}
+	return ret;
+}