1 files changed, 71 insertions, 0 deletions
diff --git a/src/cz/crcs/ectester/standalone/test/base/KeyGeneratorTestable.java b/src/cz/crcs/ectester/standalone/test/base/KeyGeneratorTestable.java
new file mode 100644
index 0000000..b561b8b
--- /dev/null
+++ b/src/cz/crcs/ectester/standalone/test/base/KeyGeneratorTestable.java
@@ -0,0 +1,71 @@
+package cz.crcs.ectester.standalone.test.base;
+
+import java.security.InvalidAlgorithmParameterException;
+import java.security.KeyPair;
+import java.security.KeyPairGenerator;
+import java.security.spec.ECParameterSpec;
+
+/**
+ * @author Jan Jancar johny@neuromancer.sk
+ */
+public class KeyGeneratorTestable extends StandaloneTestable<KeyGeneratorTestable.KeyGeneratorStage> {
+    private KeyPair kp;
+    private KeyPairGenerator kpg;
+    private int keysize = 0;
+    private ECParameterSpec spec = null;
+
+    public KeyGeneratorTestable(KeyPairGenerator kpg) {
+        this.kpg = kpg;
+    }
+
+    public KeyGeneratorTestable(KeyPairGenerator kpg, int keysize) {
+        this.kpg = kpg;
+        this.keysize = keysize;
+    }
+
+    public KeyGeneratorTestable(KeyPairGenerator kpg, ECParameterSpec spec) {
+        this.kpg = kpg;
+        this.spec = spec;
+    }
+
+    public KeyPairGenerator getKpg() {
+        return kpg;
+    }
+
+    public KeyPair getKeyPair() {
+        return kp;
+    }
+
+    @Override
+    public void run() {
+        try {
+            stage = KeyGeneratorStage.Init;
+            try {
+                if (spec != null) {
+                    kpg.initialize(spec);
+                } else if (keysize != 0) {
+                    kpg.initialize(keysize);
+                }
+            } catch (InvalidAlgorithmParameterException e) {
+                ok = false;
+                hasRun = true;
+                return;
+            }
+
+            stage = KeyGeneratorStage.GenKeyPair;
+            kp = kpg.genKeyPair();
+
+            ok = true;
+        } catch (Exception ex) {
+            ok = false;
+            error = true;
+            errorCause = ex;
+        }
+        hasRun = true;
+    }
+
+    public enum KeyGeneratorStage {
+        Init,
+        GenKeyPair
+    }
+}