1 files changed, 11 insertions, 0 deletions

diff --git a/src/gen/curve.c b/src/gen/curve.c
index 4996c28..f89044a 100644
--- a/src/gen/curve.c
+++ b/src/gen/curve.c
@@ -3,6 +3,7 @@
  * Copyright (C) 2017-2018 J08nY
  */
 #include "curve.h"
+#include "math/twists.h"
 #include "point.h"
 #include "seed.h"
 #include "util/memory.h"
@@ -116,6 +117,16 @@ GENERATOR(curve_gen_any) {
 	}
 }
 
+GENERATOR(curve_gen_any_twist) {
+	int result = curve_gen_any(curve, args, state);
+	if (result != 1) {
+		return result;
+	} else {
+		twist_rand(curve);
+		return 1;
+	}
+}
+
 CHECK(curve_check_nonzero) {
 	if (gequal0(ell_get_disc(curve->curve))) {
 		return -3;