Extend calls for #18

ELynx · Sep 8, 2016 · 4b31fc2 · 4b31fc2
1 parent 8e8cb90
commit 4b31fc2
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/app/src/main/java/com/elynx/pogoxmitm/module/ModuleManager.java b/app/src/main/java/com/elynx/pogoxmitm/module/ModuleManager.java
@@ -70,7 +70,7 @@ public interface ModuleManager {
      * @param data Data of request
      * @return Modified data if changes are made, null otherwise
      */
-    ByteString processRequest(Requests.RequestType type, ByteString data);
+    ByteString processRequest(Requests.RequestType type, ByteString data, int exchangeId, boolean connectionOk);
 
     /**
      * Process response from server to client
@@ -79,5 +79,5 @@ public interface ModuleManager {
      * @param data Data of response
      * @return Modified data if changes are made, null otherwise
      */
-    ByteString processResponse(Requests.RequestType type, ByteString data);
+    ByteString processResponse(Requests.RequestType type, ByteString data, int exchangeId, boolean connectionOk);
 }
diff --git a/app/src/main/java/com/elynx/pogoxmitm/test/AlwaysFails.java b/app/src/main/java/com/elynx/pogoxmitm/test/AlwaysFails.java
@@ -44,11 +44,11 @@ public Set<Requests.RequestType> responseTypes() {
         return new HashSet<>();
     }
 
-    public ByteString processRequest(Requests.RequestType type, ByteString data) {
+    public ByteString processRequest(Requests.RequestType type, ByteString data, int exchangeId, boolean connectionOk) {
         return null;
     }
 
-    public ByteString processResponse(Requests.RequestType type, ByteString data) {
+    public ByteString processResponse(Requests.RequestType type, ByteString data, int exchangeId, boolean connectionOk) {
         return null;
     }
 }
diff --git a/app/src/main/java/com/elynx/pogoxmitm/test/ResponseOnlyTest.java b/app/src/main/java/com/elynx/pogoxmitm/test/ResponseOnlyTest.java
@@ -15,7 +15,7 @@ public String userspaceName() {
     }
 
     public String userspaceBrief() {
-        return "Module that does nothing";
+        return "Module that doubles even responses";
     }
 
     public String userspaceInfo() {
@@ -33,8 +33,8 @@ public Set<Requests.RequestType> responseTypes() {
         return types;
     }
 
-    public ByteString processResponse(Requests.RequestType type, ByteString data) {
-        if (data.size() < 5) {
+    public ByteString processResponse(Requests.RequestType type, ByteString data, int exchangeId, boolean connectionOk) {
+        if (connectionOk && ((exchangeId & 1) == 0)) {
             //TODO valid?
             return data.concat(data);
         }