1 files changed, 15 insertions, 5 deletions

diff --git a/src/main/java/org/gnunet/voting/RegisterCommand.java b/src/main/java/org/gnunet/voting/RegisterCommand.java
index f9d4c4d..1878c48 100644
--- a/src/main/java/org/gnunet/voting/RegisterCommand.java
+++ b/src/main/java/org/gnunet/voting/RegisterCommand.java
@@ -49,16 +49,18 @@ import org.gnunet.mesh.TunnelEndHandler;
 import org.gnunet.testbed.CompressedConfig;
 import org.gnunet.util.Configuration;
 import org.gnunet.util.PeerIdentity;
+import org.gnunet.voting.messages.BallotRegisterFailureMessage;
 import org.gnunet.voting.messages.BallotRegisterRequestMessage;
-import org.gnunet.voting.messages.BallotRegisterRespondMessage;
-import org.gnunet.voting.messages.SubmitMessage;
-import org.gnunet.voting.messages.SubmitResponseMessage;
+import org.gnunet.voting.messages.BallotRegisterSuccessMessage;
 
 import java.io.File;
 import java.io.IOException;
 import java.util.List;
 import java.util.Random;
 
+/**
+ * Command for registering a ballot with an authority.
+ */
 public class RegisterCommand extends MeshRunabout implements TunnelEndHandler {
     private final String ballotFilename;
     private Ballot ballot;
@@ -74,9 +76,9 @@ public class RegisterCommand extends MeshRunabout implements TunnelEndHandler {
             throw new AssertionError();
     }
 
-    public void visit(BallotRegisterRespondMessage m) {
+    public void visit(BallotRegisterSuccessMessage m) {
         submitted = true;
-        System.out.println("vote successfully registered");
+        System.out.println("ballot successfully registered");
         ballot.addRegistrationSignature(currentAuthority, m.registrationSignature);
         try {
             Files.write(ballot.serialize(), new File(ballotFilename), Charsets.UTF_8);
@@ -88,6 +90,14 @@ public class RegisterCommand extends MeshRunabout implements TunnelEndHandler {
         mesh.destroy();
     }
 
+
+    public void visit(BallotRegisterFailureMessage m) {
+        submitted = true;
+        System.out.println("registering failed: " + m.reason);
+        tunnel.destroy();
+        mesh.destroy();
+    }
+
     public RegisterCommand(Configuration cfg, String ballotFilename) {
         this.cfg = cfg;
         this.ballotFilename = ballotFilename;