Some POW improvements

domain/src/main/java/ch/dissem/bitmessage/BitmessageContext.java

@@ -31,6 +31,8 @@ import org.slf4j.LoggerFactory;
 
 import java.net.InetAddress;
 import java.util.Arrays;
+import java.util.Timer;
+import java.util.TimerTask;
 import java.util.concurrent.*;
 
 import static ch.dissem.bitmessage.entity.Plaintext.Status.*;
@@ -72,6 +74,13 @@ public class BitmessageContext {
         // As this thread is used for parts that do POW, which itself uses parallel threads, only
         // one should be executed at any time.
         pool = Executors.newFixedThreadPool(1);
+
+        new Timer().schedule(new TimerTask() {
+            @Override
+            public void run() {
+                ctx.getProofOfWorkService().doMissingProofOfWork();
+            }
+        }, 30_000); // After 30 seconds
     }
 
     public AddressRepository addresses() {
@@ -206,6 +215,19 @@ public class BitmessageContext {
         }
     }
 
+    /**
+     * Send a custom message to a specific node (that should implement handling for this message type) and returns
+     * the response, which in turn is expected to be a {@link CustomMessage}.
+     *
+     * @param server  the node's address
+     * @param port    the node's port
+     * @param request the request
+     * @return the response
+     */
+    public CustomMessage send(InetAddress server, int port, CustomMessage request) {
+        return ctx.getNetworkHandler().send(server, port, request);
+    }
+
     public void cleanup() {
         ctx.getInventory().cleanup();
     }
@@ -276,6 +298,14 @@ public class BitmessageContext {
         );
     }
 
+    /**
+     * Returns the {@link InternalContext} - normally you wouldn't need it,
+     * unless you are doing something crazy with the protocol.
+     */
+    public InternalContext internals() {
+        return ctx;
+    }
+
     public interface Listener {
         void receive(Plaintext plaintext);
     }

domain/src/main/java/ch/dissem/bitmessage/InternalContext.java

@@ -140,6 +140,10 @@ public class InternalContext {
         return proofOfWorkEngine;
     }
 
+    public ProofOfWorkService getProofOfWorkService() {
+        return proofOfWorkService;
+    }
+
     public long[] getStreams() {
         long[] result = new long[streams.size()];
         int i = 0;

domain/src/main/java/ch/dissem/bitmessage/ProofOfWorkService.java

@@ -20,6 +20,13 @@ public class ProofOfWorkService implements ProofOfWorkEngine.Callback, InternalC
     private ProofOfWorkRepository powRepo;
     private MessageRepository messageRepo;
 
+    public void doMissingProofOfWork() {
+        for (byte[] initialHash : powRepo.getItems()) {
+            ProofOfWorkRepository.Item item = powRepo.getItem(initialHash);
+            security.doProofOfWork(item.object, item.nonceTrialsPerByte, item.extraBytes, this);
+        }
+    }
+
     public void doProofOfWork(ObjectMessage object) {
         doProofOfWork(null, object);
     }
@@ -29,7 +36,7 @@ public class ProofOfWorkService implements ProofOfWorkEngine.Callback, InternalC
         long extraBytes = recipient == null ? 0 : recipient.getPubkey().getExtraBytes();
 
         powRepo.putObject(object, nonceTrialsPerByte, extraBytes);
-        if (object.getPayload() instanceof PlaintextHolder){
+        if (object.getPayload() instanceof PlaintextHolder) {
             Plaintext plaintext = ((PlaintextHolder) object.getPayload()).getPlaintext();
             plaintext.setInitialHash(security.getInitialHash(object));
             messageRepo.save(plaintext);
@@ -39,7 +46,7 @@ public class ProofOfWorkService implements ProofOfWorkEngine.Callback, InternalC
 
     @Override
     public void onNonceCalculated(byte[] initialHash, byte[] nonce) {
-        ObjectMessage object = powRepo.getObject(initialHash);
+        ObjectMessage object = powRepo.getItem(initialHash).object;
         object.setNonce(nonce);
 //        messageCallback.proofOfWorkCompleted(payload);
         Plaintext plaintext = messageRepo.getMessage(initialHash);
@@ -48,6 +55,7 @@ public class ProofOfWorkService implements ProofOfWorkEngine.Callback, InternalC
             messageRepo.save(plaintext);
         }
         ctx.getInventory().storeObject(object);
+        ctx.getProofOfWorkRepository().removeObject(initialHash);
         ctx.getNetworkHandler().offer(object.getInventoryVector());
 //        messageCallback.messageOffered(payload, object.getInventoryVector());
     }

domain/src/main/java/ch/dissem/bitmessage/entity/CustomMessage.java

@@ -53,13 +53,21 @@ public class CustomMessage implements MessagePayload {
         return Command.CUSTOM;
     }
 
-    public byte[] getData() throws IOException {
+    public String getCustomCommand() {
+        return command;
+    }
+
+    public byte[] getData() {
         if (data != null) {
             return data;
         } else {
-            ByteArrayOutputStream out = new ByteArrayOutputStream();
-            write(out);
-            return out.toByteArray();
+            try {
+                ByteArrayOutputStream out = new ByteArrayOutputStream();
+                write(out);
+                return out.toByteArray();
+            } catch (IOException e) {
+                throw new RuntimeException(e);
+            }
         }
     }
 

domain/src/main/java/ch/dissem/bitmessage/ports/ProofOfWorkRepository.java

@@ -2,15 +2,31 @@ package ch.dissem.bitmessage.ports;
 
 import ch.dissem.bitmessage.entity.ObjectMessage;
 
+import java.util.List;
+
 /**
  * Objects that proof of work is currently being done for.
  *
  * @author Christian Basler
  */
 public interface ProofOfWorkRepository {
-    ObjectMessage getObject(byte[] initialHash);
+    Item getItem(byte[] initialHash);
+
+    List<byte[]> getItems();
 
     void putObject(ObjectMessage object, long nonceTrialsPerByte, long extraBytes);
 
-    void removeObject(ObjectMessage object);
+    void removeObject(byte[] initialHash);
+
+    class Item {
+        public final ObjectMessage object;
+        public final long nonceTrialsPerByte;
+        public final long extraBytes;
+
+        public Item(ObjectMessage object, long nonceTrialsPerByte, long extraBytes) {
+            this.object = object;
+            this.nonceTrialsPerByte = nonceTrialsPerByte;
+            this.extraBytes = extraBytes;
+        }
+    }
 }