fix error handling in Shapeshifter class

1 files changed, 45 insertions, 16 deletions

diff --git a/app/src/main/java/se/leap/bitmaskclient/pluggableTransports/Shapeshifter.java b/app/src/main/java/se/leap/bitmaskclient/pluggableTransports/Shapeshifter.java
index f7b18083..6d3b1db2 100644
--- a/app/src/main/java/se/leap/bitmaskclient/pluggableTransports/Shapeshifter.java
+++ b/app/src/main/java/se/leap/bitmaskclient/pluggableTransports/Shapeshifter.java
@@ -18,12 +18,16 @@
 package se.leap.bitmaskclient.pluggableTransports;
 
 import android.os.AsyncTask;
+import android.os.Handler;
+import android.os.Looper;
 import android.util.Log;
 
+import java.lang.ref.WeakReference;
+
 import de.blinkt.openvpn.core.VpnStatus;
 import shapeshifter.ShapeShifter;
 
-public class Shapeshifter {
+public class Shapeshifter  {
 
     public static final String DISPATCHER_PORT = "4430";
     public static final String DISPATCHER_IP = "127.0.0.1";
@@ -32,31 +36,48 @@ public class Shapeshifter {
     ShapeShifter shapeShifter;
     ShapeshifterErrorListner shapeshifterErrorListner;
 
+    public interface ShapeshifterErrorListenerCallback {
+        void onStarted();
+    }
+
     public  Shapeshifter(Obfs4Options options) {
         shapeShifter = new ShapeShifter();
+        setup(options);
+        Log.d(TAG, "shapeshifter initialized with: \n" + shapeShifter.toString());
+    }
+
+    private void setup(Obfs4Options options) {
         shapeShifter.setSocksAddr(DISPATCHER_IP+":"+DISPATCHER_PORT);
         shapeShifter.setTarget(options.remoteIP+":"+options.remotePort);
         shapeShifter.setCert(options.cert);
-        Log.d(TAG, "shapeshifter initialized with: \n" + shapeShifter.toString());
     }
 
-    public boolean start() {
+    public void start() {
         try {
-            shapeshifterErrorListner = new ShapeshifterErrorListner();
+            ShapeshifterErrorListenerCallback errorListenerCallback = () -> {
+                ;
+                Handler handler = new Handler(Looper.getMainLooper());
+                handler.postDelayed(() -> {
+                    try {
+                        Log.d(TAG, "shapeshifter open");
+                        shapeShifter.open();
+                    } catch (Exception e) {
+                        e.printStackTrace();
+                    }
+                }, 200);
+            };
+            shapeshifterErrorListner = new ShapeshifterErrorListner(errorListenerCallback);
             shapeshifterErrorListner.execute(shapeShifter);
-            shapeShifter.open();
-            Log.d(TAG, "shapeshifter opened");
-            return true;
+
         } catch (Exception e) {
-            VpnStatus.logError("SHAPESHIFTER ERROR " + e.getLocalizedMessage());
             e.printStackTrace();
         }
-        return false;
     }
 
     public boolean stop() {
         try {
             shapeShifter.close();
+            shapeshifterErrorListner.cancel(true);
             Log.d(TAG, "shapeshifter closed");
             return true;
         } catch (Exception e) {
@@ -67,28 +88,36 @@ public class Shapeshifter {
     }
 
     static class ShapeshifterErrorListner extends AsyncTask<ShapeShifter, Void, Void> {
+
+        WeakReference<ShapeshifterErrorListenerCallback> callbackWeakReference;
+
+        public ShapeshifterErrorListner(ShapeshifterErrorListenerCallback callback) {
+            callbackWeakReference = new WeakReference<>(callback);
+        }
+
         @Override
         protected Void doInBackground(ShapeShifter... shapeShifters) {
             ShapeShifter shapeshifter = shapeShifters[0];
+            Log.d(TAG, "Shapeshifter error listener started");
+            ShapeshifterErrorListenerCallback callback = callbackWeakReference.get();
+            if (callback != null) {
+                callback.onStarted();
+            }
             try {
                 shapeshifter.getLastError();
             } catch (Exception e) {
                 e.printStackTrace();
-                VpnStatus.logError("SHAPESHIFTER ERROR " + e.getLocalizedMessage());
+                Log.e(TAG, "SHAPESHIFTER ERROR: " + e.getLocalizedMessage());
+                VpnStatus.logError(VpnStatus.ErrorType.SHAPESHIFTER);
+                VpnStatus.logError(e.getLocalizedMessage());
                 try {
                     shapeshifter.close();
                 } catch (Exception ex) {
                     ex.printStackTrace();
-                    VpnStatus.logError("SHAPESHIFTER ERROR " + e.getLocalizedMessage());
                 }
             }
             return null;
         }
-
-        @Override
-        protected void onPostExecute(Void aVoid) {
-            super.onPostExecute(aVoid);
-        }
     }
 
 }