Embed Adblock Plus

mobile/android/base/distribution/Distribution.java

 /* -*- Mode: Java; c-basic-offset: 4; tab-width: 20; indent-tabs-mode: nil; -*-
  * This Source Code Form is subject to the terms of the Mozilla Public
  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
  * You can obtain one at http://mozilla.org/MPL/2.0/. */
 
 package org.mozilla.gecko.distribution;
 
 import java.io.BufferedInputStream;
 import java.io.File;
 import java.io.FileNotFoundException;
@@ skipping 322 matching lines @@
      * set and populated.
      *
      * This method is *only* protected for use from testDistribution.
      *
      * @return true if we've set a distribution.
      */
     @RobocopTarget
     protected boolean doInit() {
         ThreadUtils.assertNotOnUiThread();
 
+        // Firefox for Android normally ignores application scope extensions,
+        // and thus doesn't put the extensions directory in place. In Adblock
+        // Browser however, we need this mechanism.
+        //
+        // This logic doesn't really belong here. However, here the change is
+        // minimally invasive, and we're more likely to notice when the logic
+        // changes for the distributions directory.
+        try {
+            copyExtensionsFiles();
+        } catch (IOException e) {
+            Log.e(LOGTAG, "Error copying extensions files from APK.", e);
+        }
+
         // Bail if we've already tried to initialize the distribution, and
         // there wasn't one.
         final SharedPreferences settings;
         if (prefsBranch == null) {
             settings = GeckoSharedPrefs.forApp(context);
         } else {
             settings = context.getSharedPreferences(prefsBranch, Activity.MODE_PRIVATE);
         }
 
         String keyName = context.getPackageName() + ".distribution_state";
@@ skipping 290 matching lines @@
                     fileStream.close();
                 }
             }
         } finally {
             zip.close();
         }
 
         return distributionSet;
     }
 
+    /**
+     * Copies the /extensions folder out of the APK and into the app's data directory.
+     * Adapted from copyFiles().
+     */
+    private void copyExtensionsFiles() throws IOException {
+        final File applicationPackage = new File(packagePath);
+        final ZipFile zip = new ZipFile(applicationPackage);
+
+        try {
+            final byte[] buffer = new byte[1024];
+
+            final Enumeration<? extends ZipEntry> zipEntries = zip.entries();
+            while (zipEntries.hasMoreElements()) {
+                final ZipEntry fileEntry = zipEntries.nextElement();
+                final String name = fileEntry.getName();
+
+                if (fileEntry.isDirectory()) {
+                    // We'll let getDataFile deal with creating the directory hierarchy.
+                    continue;
+                }
+
+                if (!name.startsWith("extensions/")) {
+                    continue;
+                }
+
+                final File outFile = getDataFile(name);
+                if (outFile == null) {
+                    continue;
+                }
+
+                final InputStream fileStream = zip.getInputStream(fileEntry);
+                try {
+                    writeStream(fileStream, outFile, fileEntry.getTime(), buffer);
+                } finally {
+                    fileStream.close();
+                }
+            }
+        } finally {
+            zip.close();
+        }
+    }
+
     private void writeStream(InputStream fileStream, File outFile, final long modifiedTime, byte[] buffer)
             throws FileNotFoundException, IOException {
         final OutputStream outStream = new FileOutputStream(outFile);
         try {
             int count;
             while ((count = fileStream.read(buffer)) > 0) {
                 outStream.write(buffer, 0, count);
             }
 
             outFile.setLastModified(modifiedTime);
@@ skipping 114 matching lines @@
     }
 
     /**
      * A safe way for callers to determine if this Distribution instance
      * represents a real live distribution.
      */
     public boolean exists() {
         return state == STATE_SET;
     }
 }