From ff6be88e469608a67945a274ec2180aee3f3ccd2 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Mon, 28 May 2018 18:37:01 +0200
Subject: Handle exceptions in standalone testables better.

---
 .../standalone/test/SignatureTestable.java         | 101 ++++++++++++---------
 1 file changed, 56 insertions(+), 45 deletions(-)

(limited to 'src/cz/crcs/ectester/standalone/test/SignatureTestable.java')

diff --git a/src/cz/crcs/ectester/standalone/test/SignatureTestable.java b/src/cz/crcs/ectester/standalone/test/SignatureTestable.java
index c77e9c5..6bc9b30 100644
--- a/src/cz/crcs/ectester/standalone/test/SignatureTestable.java
+++ b/src/cz/crcs/ectester/standalone/test/SignatureTestable.java
@@ -57,54 +57,65 @@ public class SignatureTestable extends BaseTestable {
 
     @Override
     public void run() {
-        if (kgt != null) {
-            signKey = (ECPrivateKey) kgt.getKeyPair().getPrivate();
-            verifyKey = (ECPublicKey) kgt.getKeyPair().getPublic();
-        }
-
-        try {
-            sig.initSign(signKey);
-        } catch (InvalidKeyException e) {
-            throw new TestException(e);
-        }
-
-        try {
-            sig.update(data);
-        } catch (SignatureException e) {
-            ok = false;
-            hasRun = true;
-            return;
-        }
-
-        try {
-            signature = sig.sign();
-        } catch (SignatureException e) {
-            ok = false;
-            hasRun = true;
-            return;
-        }
-
-        try {
-            sig.initVerify(verifyKey);
-        } catch (InvalidKeyException e) {
-            throw new TestException(e);
-        }
-
-        try {
-            sig.update(data);
-        } catch (SignatureException e) {
-            ok = false;
-            hasRun = true;
-            return;
-        }
-
         try {
-            verified = sig.verify(signature);
-        } catch (SignatureException e) {
+            if (kgt != null) {
+                signKey = (ECPrivateKey) kgt.getKeyPair().getPrivate();
+                verifyKey = (ECPublicKey) kgt.getKeyPair().getPublic();
+            }
+
+            try {
+                sig.initSign(signKey);
+            } catch (InvalidKeyException e) {
+                ok = false;
+                hasRun = true;
+                return;
+            }
+
+            try {
+                sig.update(data);
+            } catch (SignatureException e) {
+                ok = false;
+                hasRun = true;
+                return;
+            }
+
+            try {
+                signature = sig.sign();
+            } catch (SignatureException e) {
+                ok = false;
+                hasRun = true;
+                return;
+            }
+
+            try {
+                sig.initVerify(verifyKey);
+            } catch (InvalidKeyException e) {
+                ok = false;
+                hasRun = true;
+                return;
+            }
+
+            try {
+                sig.update(data);
+            } catch (SignatureException e) {
+                ok = false;
+                hasRun = true;
+                return;
+            }
+
+            try {
+                verified = sig.verify(signature);
+            } catch (SignatureException e) {
+                ok = false;
+                hasRun = true;
+            }
+
+            ok = true;
+        } catch (Exception ex) {
             ok = false;
-            hasRun = true;
+            error = true;
+            errorCause = ex;
         }
-        ok = true;
         hasRun = true;
     }
 }
-- 
cgit v1.2.3-70-g09d2