From 14f3dfb2cd39d552c3c2b767f1f045d58759572f Mon Sep 17 00:00:00 2001
From: J08nY
Date: Wed, 4 Oct 2017 22:07:42 +0200
Subject: Fix ANSI X9.62 method not respecting -p or --cofactor options.

---
 src/exhaustive/exhaustive.c | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/exhaustive/exhaustive.c')

diff --git a/src/exhaustive/exhaustive.c b/src/exhaustive/exhaustive.c
index b55bcca..41e3ef1 100644
--- a/src/exhaustive/exhaustive.c
+++ b/src/exhaustive/exhaustive.c
@@ -76,7 +76,13 @@ static void exhaustive_ginit(gen_f *generators, const config_t *cfg) {
 			default:
 				break;
 		}
-		generators[OFFSET_ORDER] = &order_gen_any;
+		if (cfg->prime) {
+			generators[OFFSET_ORDER] = &order_gen_prime;
+		} else if (cfg->cofactor) {
+			generators[OFFSET_ORDER] = &order_gen_smallfact;
+		} else {
+			generators[OFFSET_ORDER] = &order_gen_any;
+		}
 	} else {
 		// setup normal generators
 		generators[OFFSET_SEED] = &gen_skip;
-- 
cgit v1.2.3-70-g09d2