Move stuff related to generators to src/gen.

1 files changed, 162 insertions, 0 deletions

diff --git a/src/gen/curve.c b/src/gen/curve.c
new file mode 100644
index 0000000..bc382cd
--- /dev/null
+++ b/src/gen/curve.c
@@ -0,0 +1,162 @@
+/*
+ * ecgen, tool for generating Elliptic curve domain parameters
+ * Copyright (C) 2017 J08nY
+ */
+#include "curve.h"
+#include "point.h"
+#include "seed.h"
+#include "util/memory.h"
+
+curve_t *curve_new(void) { return try_calloc(sizeof(curve_t)); }
+
+curve_t *curve_copy(const curve_t *src, curve_t *dest) {
+	if (src->seed) dest->seed = seed_copy(src->seed, dest->seed);
+	if (src->field) dest->field = gcopy(src->field);
+	if (src->a) dest->a = gcopy(src->a);
+	if (src->b) dest->b = gcopy(src->b);
+	if (src->curve) dest->curve = gcopy(src->curve);
+	if (src->order) dest->order = gcopy(src->order);
+	if (src->generators) {
+		dest->generators = points_new_copy(src->generators, src->ngens);
+		dest->ngens = src->ngens;
+	}
+	if (src->points) {
+		dest->points = points_new_copy(src->points, src->npoints);
+		dest->npoints = src->npoints;
+	}
+	return dest;
+}
+
+curve_t *curve_new_copy(const curve_t *src) {
+	curve_t *result = curve_new();
+	return curve_copy(src, result);
+}
+
+curve_t *curve_clone(const curve_t *src, curve_t *dest) {
+	if (src->seed) dest->seed = seed_clone(src->seed, dest->seed);
+	if (src->field) dest->field = gclone(src->field);
+	if (src->a) dest->a = gclone(src->a);
+	if (src->b) dest->b = gclone(src->b);
+	if (src->curve) dest->curve = gclone(src->curve);
+	if (src->order) dest->order = gclone(src->order);
+	if (src->generators) {
+		dest->generators = points_new_clone(src->generators, src->ngens);
+		dest->ngens = src->ngens;
+	}
+	if (src->points) {
+		dest->points = points_new_clone(src->points, src->npoints);
+		dest->npoints = src->npoints;
+	}
+	return dest;
+}
+
+curve_t *curve_new_clone(const curve_t *src) {
+	curve_t *result = curve_new();
+	return curve_clone(src, result);
+}
+
+void curve_free(curve_t **curve) {
+	if (*curve) {
+		seed_free(&(*curve)->seed);
+		points_free_deep(&(*curve)->generators, (*curve)->ngens);
+		points_free_deep(&(*curve)->points, (*curve)->npoints);
+
+		if ((*curve)->curve) {
+			// TODO, this is possibly dangerous...
+			obj_free((*curve)->curve);
+			if (isclone((*curve)->curve)) {
+				gunclone((*curve)->curve);
+			}
+		}
+
+		if ((*curve)->field && isclone((*curve)->field)) {
+			gunclone((*curve)->field);
+		}
+		if ((*curve)->a && isclone((*curve)->a)) {
+			gunclone((*curve)->a);
+		}
+		if ((*curve)->b && isclone((*curve)->b)) {
+			gunclone((*curve)->b);
+		}
+		if ((*curve)->order && isclone((*curve)->order)) {
+			gunclone((*curve)->order);
+		}
+
+		pari_free(*curve);
+		*curve = NULL;
+	}
+}
+
+GENERATOR(curve_gen_any) {
+	pari_sp ltop = avma;
+	GEN v = gen_0;
+	switch (typ(curve->field)) {
+		case t_INT:
+			v = gtovec0(gen_0, 2);
+			gel(v, 1) = curve->a;
+			gel(v, 2) = curve->b;
+			break;
+		case t_FFELT:
+			v = gtovec0(gen_0, 5);
+			gel(v, 1) = gen_1;
+			gel(v, 2) = curve->a;
+			gel(v, 4) = curve->b;
+			break;
+		default:
+			pari_err_TYPE("curve_gen_any", curve->field);
+	}
+	GEN crv = ellinit(v, curve->field, -1);
+
+	if (glength(crv) == 0) {
+		avma = ltop;
+		return -3;
+	} else {
+		curve->curve = gerepilecopy(ltop, crv);
+		return 1;
+	}
+}
+
+GENERATOR(curve_gen_nonzero) {
+	pari_sp ltop = avma;
+	int any = curve_gen_any(curve, cfg, args);
+	if (any <= 0) {
+		return any;
+	}
+	if (gequal0(ell_get_disc(curve->curve))) {
+		avma = ltop;
+		return -3;
+	} else {
+		return 1;
+	}
+}
+
+static int curve_gen_seed_fp(curve_t *curve, const config_t *cfg, arg_t *args) {
+	// TODO implement
+	return INT_MIN;
+}
+
+static int curve_gen_seed_f2m(curve_t *curve, const config_t *cfg,
+                              arg_t *args) {
+	// TODO implement
+	return INT_MIN;
+}
+
+GENERATOR(curve_gen_seed) {
+	switch (typ(curve->field)) {
+		case t_INT:
+			return curve_gen_seed_fp(curve, cfg, args);
+		case t_FFELT:
+			return curve_gen_seed_f2m(curve, cfg, args);
+		default:
+			pari_err_TYPE("curve_gen_seed", curve->field);
+			return INT_MIN; /* NOT REACHABLE */
+	}
+}
+
+UNROLL(curve_unroll) {
+	if (curve->curve) {
+		obj_free(curve->curve);
+		curve->curve = NULL;
+	}
+	return -1;
+}