Issue 1771 - Update RootTools

src/org/adblockplus/android/configurators/IptablesProxyConfigurator.java

 /*
- * This file is part of Adblock Plus <http://adblockplus.org/>,
- * Copyright (C) 2006-2014 Eyeo GmbH
+ * This file is part of Adblock Plus <https://adblockplus.org/>,
+ * Copyright (C) 2006-2015 Eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
@@ skipping 34 matching lines @@
 
   private final Context context;
   private String iptables;
   private boolean isRegistered = false;
 
   public IptablesProxyConfigurator(final Context context)
   {
     this.context = context;
   }
 
-  private static List<String> sendShell(final String command, final int timeout) throws IOException, TimeoutException,
+  private static List<String> runRootCommand(final String command, final int timeout) throws IOException, TimeoutException,
       RootDeniedException
   {
-    final CommandOutput cmd = new CommandOutput(0, DEFAULT_TIMEOUT, command);
+    final CapturingOutputCommand cmd = new CapturingOutputCommand(0, DEFAULT_TIMEOUT, command);
 
     Shell.runRootCommand(cmd);
 
     cmd.waitForCompletion();
 
     return cmd.output;
   }
 
   @Override
   public boolean initialize()
   {
     try
     {
       // If we don't set `handlerEnabled` to `false`, RootTools uses Handlers
       // which get executed on the UI thread which in fact renders it useless
       // for our purpose (as it either finished too late or blocks forever).
       RootTools.handlerEnabled = false;
 
       if (!RootTools.isAccessGiven())
       {
         throw new IllegalStateException("No root access");
       }
 
-      final File ipt = this.context.getFileStreamPath("iptables");
-
-      if (!ipt.exists())
-      {
-        throw new FileNotFoundException("No iptables executable");
-      }
-
-      final String path = ipt.getAbsolutePath();
-
-      sendShell("chmod 700 " + path, DEFAULT_TIMEOUT);
+      final String path = getIptablesExecutablePath();
+
+      runRootCommand("chmod 700 " + path, DEFAULT_TIMEOUT);
 
       boolean compatible = false;
       boolean version = false;
 
       final String command = path + " --version\n" + path + " -L -t nat -n\n";
 
-      final List<String> result = sendShell(command, DEFAULT_TIMEOUT);
+      final List<String> result = runRootCommand(command, DEFAULT_TIMEOUT);
 
       for (final String line : result)
       {
         if (line.contains("OUTPUT"))
         {
           compatible = true;
         }
         if (line.contains("v1.4."))
         {
           version = true;
         }
       }
 
       if (!(compatible && version))
       {
         throw new IllegalStateException("Incompatible iptables excutable");
       }
 
       this.iptables = path;
 
       return true;
     }
     catch (final Exception e)
     {
-      Log.e(TAG, "iptables setup failed", e);
       return false;
     }
+  }
+
+  private String getIptablesExecutablePath() throws FileNotFoundException
+  {
+    File iptablesExecutable = new File("/system/bin/iptables");
+    if (!iptablesExecutable.exists())
+    {
+      Log.i(TAG, "iptables not found on the system, using embedded binary");
+      iptablesExecutable = context.getFileStreamPath("iptables");
+    }
+    if (!iptablesExecutable.exists())
+      throw new FileNotFoundException("No iptables executable");
+    return iptablesExecutable.getAbsolutePath();
   }
 
   @Override
   public boolean registerProxy(final InetAddress address, final int port)
   {
     try
     {
       final int uid = this.context.getPackageManager().getPackageInfo(this.context.getPackageName(), 0).applicationInfo.uid;
 
       final StringBuilder cmd = new StringBuilder();
       cmd.append(this.iptables);
       cmd.append(IPTABLES_RETURN.replace("{{UID}}", String.valueOf(uid)));
       cmd.append('\n');
       cmd.append(this.iptables);
       cmd.append(IPTABLES_ADD_HTTP.replace("{{PORT}}", String.valueOf(port)));
 
-      sendShell(cmd.toString(), DEFAULT_TIMEOUT);
+      runRootCommand(cmd.toString(), DEFAULT_TIMEOUT);
 
       this.isRegistered = true;
 
       return true;
     }
     catch (final Exception e)
     {
       // I leave this logging message for now, passing 'init' and failing 'register' definitely is a failure
       Log.e(TAG, "Couldn't register proxy using iptables.", e);
       return false;
     }
   }
 
   @Override
   public void unregisterProxy()
   {
     try
     {
-      sendShell(this.iptables + " -t nat -F OUTPUT", DEFAULT_TIMEOUT);
+      runRootCommand(this.iptables + " -t nat -F OUTPUT", DEFAULT_TIMEOUT);
     }
     catch (final Exception e)
     {
       Log.w(TAG, "Failed to unregister proxy using iptables.", e);
     }
     finally
     {
       this.isRegistered = false;
     }
   }
@@ skipping 15 matching lines @@
 
       final File ipt = context.getFileStreamPath("iptables");
 
       if (!ipt.exists())
       {
         throw new FileNotFoundException("No iptables executable");
       }
 
       final String path = ipt.getAbsolutePath();
 
-      sendShell("chmod 700 " + path, DEFAULT_TIMEOUT);
+      runRootCommand("chmod 700 " + path, DEFAULT_TIMEOUT);
 
       boolean compatible = false;
       boolean version = false;
 
       String command = path + " --version\n" + path + " -L -t nat -n\n";
 
-      final List<String> result = sendShell(command, DEFAULT_TIMEOUT);
+      final List<String> result = runRootCommand(command, DEFAULT_TIMEOUT);
 
       for (final String line : result)
       {
         if (line.contains("OUTPUT"))
         {
           compatible = true;
         }
         if (line.contains("v1.4."))
         {
           version = true;
         }
       }
 
       if (!(compatible && version))
       {
         throw new IllegalStateException("Incompatible iptables excutable");
       }
 
       command = path + " -L -t nat -n\n";
 
-      return sendShell(command, DEFAULT_TIMEOUT);
+      return runRootCommand(command, DEFAULT_TIMEOUT);
     }
     catch (final Throwable t)
     {
       return null;
     }
   }
 
   @Override
   public ProxyRegistrationType getType()
   {
@@ skipping 11 matching lines @@
   {
     return false;
   }
 
   @Override
   public String toString()
   {
     return "[ProxyConfigurator: " + this.getType() + "]";
   }
 
-  private final static class CommandOutput extends Command
-  {
-    private final Semaphore completionNotify = new Semaphore(1);
+  private final static class CapturingOutputCommand extends Command
+  {
+    private final Semaphore running = new Semaphore(1);
 
     public List<String> output = new ArrayList<String>();
 
-    public CommandOutput(final int id, final int timeout, final String command)
+    public CapturingOutputCommand(final int id, final int timeout, final String command)
     {
       super(id, timeout, command);
 
-      this.completionNotify.acquireUninterruptibly();
+      this.running.acquireUninterruptibly();
     }
 
     @Override
     public void commandOutput(int id, String line)
     {
       this.output.add(line);
     }
 
     @Override
     public void commandCompleted(int id, int exitCode)
     {
-      this.completionNotify.release();
+      this.running.release();
     }
 
     @Override
     public void commandTerminated(int id, String reason)
     {
-      this.completionNotify.release();
+      this.running.release();
     }
 
     public void waitForCompletion()
     {
-      this.completionNotify.acquireUninterruptibly();
+      this.running.acquireUninterruptibly();
     }
   }
 }