1 files changed, 34 insertions, 2 deletions

diff --git a/src/main/java/org/gnunet/voting/SubmitCommand.java b/src/main/java/org/gnunet/voting/SubmitCommand.java
index d00ea69..43e10c9 100644
--- a/src/main/java/org/gnunet/voting/SubmitCommand.java
+++ b/src/main/java/org/gnunet/voting/SubmitCommand.java
@@ -21,18 +21,27 @@
 package org.gnunet.voting;
 
 
+import com.google.common.base.Charsets;
+import com.google.common.io.Files;
 import org.gnunet.mesh.Mesh;
 import org.gnunet.mesh.MeshRunabout;
 import org.gnunet.mesh.TunnelEndHandler;
+import org.gnunet.util.AbsoluteTime;
 import org.gnunet.util.Configuration;
+import org.gnunet.util.CryptoECC;
 import org.gnunet.util.PeerIdentity;
+import org.gnunet.voting.messages.SubmitFailureMessage;
 import org.gnunet.voting.messages.SubmitMessage;
-import org.gnunet.voting.messages.SubmitResponseMessage;
+import org.gnunet.voting.messages.SubmitSuccessMessage;
 
 import java.io.File;
+import java.io.IOException;
 import java.util.List;
 import java.util.Random;
 
+/**
+ * Command that submits a ballot with a vote to a remaining authority.
+ */
 public class SubmitCommand extends MeshRunabout implements TunnelEndHandler {
     private final String ballotFilename;
     private Ballot ballot;
@@ -40,6 +49,7 @@ public class SubmitCommand extends MeshRunabout implements TunnelEndHandler {
     private Mesh mesh;
     private Mesh.Tunnel<Void> tunnel;
     private boolean submitted = false;
+    private PeerIdentity currentAuthority;
 
     @Override
     public void onTunnelEnd(Mesh.Tunnel tunnel) {
@@ -47,9 +57,23 @@ public class SubmitCommand extends MeshRunabout implements TunnelEndHandler {
             throw new AssertionError();
     }
 
-    public void visit(SubmitResponseMessage m) {
+    public void visit(SubmitSuccessMessage m) {
         submitted = true;
         System.out.println("vote successfully submitted");
+        ballot.addVoucher(currentAuthority, m.voucherSig);
+        try {
+            Files.write(ballot.serialize(), new File(ballotFilename), Charsets.UTF_8);
+        } catch (IOException e) {
+            System.out.println("could not write ballot file");
+            return;
+        }
+        tunnel.destroy();
+        mesh.destroy();
+    }
+
+    public void visit(SubmitFailureMessage m) {
+        submitted = true;
+        System.out.println("vote not submitted: " + m.reason);
         tunnel.destroy();
         mesh.destroy();
     }
@@ -74,9 +98,17 @@ public class SubmitCommand extends MeshRunabout implements TunnelEndHandler {
         Random r = new Random();
         PeerIdentity authority = remainingAuthorities.get(r.nextInt(remainingAuthorities.size()));
         System.out.println("submitting to authority" + authority.toString());
+        currentAuthority = authority;
         mesh = new Mesh(cfg, null, this);
         tunnel = mesh.createTunnel(authority, TallyAuthorityDaemon.MESH_PORT, true, true, null);
         SubmitMessage m = new SubmitMessage();
+        if (ballot.voterPub == null) {
+            throw new InvalidBallotException("no voter in ballot");
+        }
+        m.voterPub = ballot.voterPub;
+        // FIXME: implement certs
+        m.groupCertExpiration = AbsoluteTime.FOREVER.asMessage();
+        m.groupCert = CryptoECC.Signature.randomSignature();
         m.ballotGuid = ballot.getBallotGuid();
         m.choiceId = ballot.choiceId;
         tunnel.send(m);