WhonixTorController: Implement TAKEOWNERSHIP command

alvasw · May 29, 2024 · 24928bc · 24928bc
1 parent 2807fcb
commit 24928bc
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/network/tor/tor/src/main/java/bisq/tor/controller/ControlCommandFailedException.java b/network/tor/tor/src/main/java/bisq/tor/controller/ControlCommandFailedException.java
@@ -18,6 +18,10 @@
 package bisq.tor.controller;
 
 public class ControlCommandFailedException extends RuntimeException {
+    public ControlCommandFailedException(String message) {
+        super(message);
+    }
+
     public ControlCommandFailedException(String message, Throwable cause) {
         super(message, cause);
     }

diff --git a/network/tor/tor/src/main/java/bisq/tor/controller/WhonixTorController.java b/network/tor/tor/src/main/java/bisq/tor/controller/WhonixTorController.java
@@ -52,6 +52,15 @@ public void addOnion(TorKeyPair torKeyPair, int onionPort, int localPort) throws
         String reply = receiveReply();
     }
 
+    public void takeOwnership() throws IOException {
+        String command = "TAKEOWNERSHIP\r\n";
+        sendCommand(command);
+        String reply = receiveReply();
+        if (!reply.equals("250 OK")) {
+            throw new ControlCommandFailedException("Couldn't take ownership");
+        }
+    }
+
     private void sendCommand(String command) throws IOException {
         byte[] commandBytes = command.getBytes(StandardCharsets.US_ASCII);
         outputStream.write(commandBytes);