weblayer: fixes strict mode issues for weblayer

This fixs a number of cases and enables strict mode in tests. Hopefully
that helps prevent any other access from being added.

The trickier cases are jacoco injects code that accesses the disk, and
LayoutInflater may trigger loading classes, which triggers disk
access.

BUG=1059770
TEST=covered by tests now

Change-Id: I50f81836f838b023fe09fd5f760cb7484edd8d26
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2100589
Commit-Queue: Scott Violet <sky@chromium.org>
Reviewed-by: Bo <boliu@chromium.org>
Cr-Commit-Position: refs/heads/master@{#751028}

components/browser_ui/modaldialog/android/java/src/org/chromium/components/browser_ui/modaldialog/AppModalPresenter.java

@@ -13,6 +13,7 @@ import android.view.WindowManager;
 
 import androidx.annotation.VisibleForTesting;
 
+import org.chromium.base.StrictModeContext;
 import org.chromium.ui.modaldialog.DialogDismissalCause;
 import org.chromium.ui.modaldialog.ModalDialogManager;
 import org.chromium.ui.modaldialog.ModalDialogProperties;
@@ -46,6 +47,14 @@ public class AppModalPresenter extends ModalDialogManager.Presenter {
         mContext = context;
     }
 
+    private ModalDialogView loadDialogView() {
+        // LayoutInflater may access the disk.
+        try (StrictModeContext ignored = StrictModeContext.allowDiskReads()) {
+            return (ModalDialogView) LayoutInflater.from(mDialog.getContext())
+                    .inflate(R.layout.modal_dialog_view, null);
+        }
+    }
+
     @Override
     protected void addDialogView(PropertyModel model) {
         int style = model.get(ModalDialogProperties.PRIMARY_BUTTON_FILLED)
@@ -57,11 +66,13 @@ public class AppModalPresenter extends ModalDialogManager.Presenter {
         // Cancel on touch outside should be disabled by default. The ModelChangeProcessor wouldn't
         // notify change if the property is not set during initialization.
         mDialog.setCanceledOnTouchOutside(false);
-        ModalDialogView dialogView = (ModalDialogView) LayoutInflater.from(mDialog.getContext())
-                                             .inflate(R.layout.modal_dialog_view, null);
+        ModalDialogView dialogView = loadDialogView();
         mModelChangeProcessor =
                 PropertyModelChangeProcessor.create(model, dialogView, new ViewBinder());
-        mDialog.setContentView(dialogView);
+        // setContentView() can trigger using LayoutInflater, which may read from disk.
+        try (StrictModeContext ignored = StrictModeContext.allowDiskReads()) {
+            mDialog.setContentView(dialogView);
+        }
 
         try {
             mDialog.show();

components/browser_ui/modaldialog/android/java/src/org/chromium/components/browser_ui/modaldialog/TabModalPresenter.java

@@ -18,6 +18,7 @@ import androidx.annotation.NonNull;
 import androidx.annotation.VisibleForTesting;
 import androidx.core.view.ViewCompat;
 
+import org.chromium.base.StrictModeContext;
 import org.chromium.content_public.browser.SelectionPopupController;
 import org.chromium.content_public.browser.WebContents;
 import org.chromium.ui.interpolators.BakedBezierInterpolator;
@@ -105,6 +106,14 @@ public abstract class TabModalPresenter extends ModalDialogManager.Presenter {
         return mDialogContainer;
     }
 
+    private ModalDialogView loadDialogView(int style) {
+        // LayoutInflater may access the disk.
+        try (StrictModeContext ignored = StrictModeContext.allowDiskReads()) {
+            return (ModalDialogView) LayoutInflater.from(new ContextThemeWrapper(mContext, style))
+                    .inflate(R.layout.modal_dialog_view, null);
+        }
+    }
+
     @Override
     protected void addDialogView(PropertyModel model) {
         if (mDialogContainer == null) mDialogContainer = createDialogContainer();
@@ -112,9 +121,7 @@ public abstract class TabModalPresenter extends ModalDialogManager.Presenter {
         int style = model.get(ModalDialogProperties.PRIMARY_BUTTON_FILLED)
                 ? R.style.Theme_Chromium_ModalDialog_FilledPrimaryButton
                 : R.style.Theme_Chromium_ModalDialog_TextPrimaryButton;
-        mDialogView =
-                (ModalDialogView) LayoutInflater.from(new ContextThemeWrapper(mContext, style))
-                        .inflate(R.layout.modal_dialog_view, null);
+        mDialogView = loadDialogView(style);
         mModelChangeProcessor =
                 PropertyModelChangeProcessor.create(model, mDialogView, new ViewBinder());
 

components/embedder_support/android/java/src/org/chromium/components/embedder_support/delegate/ColorChooserAndroid.java

@@ -7,6 +7,7 @@ package org.chromium.components.embedder_support.delegate;
 import android.content.Context;
 
 import org.chromium.base.ContextUtils;
+import org.chromium.base.StrictModeContext;
 import org.chromium.base.annotations.CalledByNative;
 import org.chromium.base.annotations.JNINamespace;
 import org.chromium.base.annotations.NativeMethods;
@@ -37,7 +38,10 @@ public class ColorChooserAndroid {
     }
 
     private void openColorChooser() {
-        mDialog.show();
+        // This triggers LayoutInflater, which may access the disk.
+        try (StrictModeContext ignored = StrictModeContext.allowDiskReads()) {
+            mDialog.show();
+        }
     }
 
     @CalledByNative

components/embedder_support/android/java/src/org/chromium/components/embedder_support/delegate/ColorPickerDialog.java

@@ -13,6 +13,8 @@ import android.view.View;
 import android.widget.Button;
 import android.widget.TextView;
 
+import org.chromium.base.StrictModeContext;
+
 /**
  * UI for the color chooser that shows on the Android platform as a result of
  * <input type=color > form element.
@@ -33,6 +35,15 @@ public class ColorPickerDialog extends AlertDialog implements OnColorChangedList
 
     private int mCurrentColor;
 
+    View inflateView(Context context, int id) {
+        // LayoutInflater may trigger accessing the disk.
+        try (StrictModeContext ignored = StrictModeContext.allowDiskReads()) {
+            LayoutInflater inflater =
+                    (LayoutInflater) context.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
+            return inflater.inflate(id, null);
+        }
+    }
+
     /**
      * @param context The context the dialog is to run in.
      * @param listener The object to notify when the color is set.
@@ -48,9 +59,7 @@ public class ColorPickerDialog extends AlertDialog implements OnColorChangedList
         mCurrentColor = mInitialColor;
 
         // Initialize title
-        LayoutInflater inflater =
-                (LayoutInflater) context.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
-        View title = inflater.inflate(R.layout.color_picker_dialog_title, null);
+        View title = inflateView(context, R.layout.color_picker_dialog_title);
         setCustomTitle(title);
 
         mCurrentColorView = title.findViewById(R.id.selected_color_view);
@@ -90,7 +99,7 @@ public class ColorPickerDialog extends AlertDialog implements OnColorChangedList
         });
 
         // Initialize main content view
-        View content = inflater.inflate(R.layout.color_picker_dialog_content, null);
+        View content = inflateView(context, R.layout.color_picker_dialog_content);
         setView(content);
 
         // Initialize More button.

components/javascript_dialogs/android/java/src/org/chromium/components/javascript_dialogs/JavascriptModalDialog.java

@@ -11,6 +11,7 @@ import android.view.LayoutInflater;
 import androidx.annotation.StringRes;
 
 import org.chromium.base.Log;
+import org.chromium.base.StrictModeContext;
 import org.chromium.ui.modaldialog.DialogDismissalCause;
 import org.chromium.ui.modaldialog.ModalDialogManager;
 import org.chromium.ui.modaldialog.ModalDialogProperties;
@@ -52,8 +53,11 @@ public abstract class JavascriptModalDialog implements ModalDialogProperties.Con
     protected void show(Context context, ModalDialogManager manager,
             @ModalDialogManager.ModalDialogType int dialogType) {
         assert manager != null;
-        mDialogCustomView = (JavascriptDialogCustomView) LayoutInflater.from(context).inflate(
-                R.layout.js_modal_dialog, null);
+        // LayoutInflater may trigger accessing the disk.
+        try (StrictModeContext ignored = StrictModeContext.allowDiskReads()) {
+            mDialogCustomView = (JavascriptDialogCustomView) LayoutInflater.from(context).inflate(
+                    R.layout.js_modal_dialog, null);
+        }
         mDialogCustomView.setPromptText(mDefaultPromptText);
         mDialogCustomView.setSuppressCheckBoxVisibility(mShouldShowSuppressCheckBox);
 

weblayer/browser/java/org/chromium/weblayer_private/WebLayerImpl.java

@@ -405,17 +405,20 @@ public final class WebLayerImpl extends IWebLayer.Stub {
     }
 
     private void loadNativeLibrary(String packageName) {
-        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) {
-            WebViewFactory.loadWebViewNativeLibraryFromPackage(
-                    packageName, getClass().getClassLoader());
-        } else {
-            try {
-                Method loadNativeLibrary =
-                        WebViewFactory.class.getDeclaredMethod("loadNativeLibrary");
-                loadNativeLibrary.setAccessible(true);
-                loadNativeLibrary.invoke(null);
-            } catch (ReflectiveOperationException e) {
-                Log.e(TAG, "Failed to load native library.", e);
+        // Loading the library triggers disk access.
+        try (StrictModeContext ignored = StrictModeContext.allowDiskReads()) {
+            if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) {
+                WebViewFactory.loadWebViewNativeLibraryFromPackage(
+                        packageName, getClass().getClassLoader());
+            } else {
+                try {
+                    Method loadNativeLibrary =
+                            WebViewFactory.class.getDeclaredMethod("loadNativeLibrary");
+                    loadNativeLibrary.setAccessible(true);
+                    loadNativeLibrary.invoke(null);
+                } catch (ReflectiveOperationException e) {
+                    Log.e(TAG, "Failed to load native library.", e);
+                }
             }
         }
     }

weblayer/browser/java/org/chromium/weblayer_private/WebLayerTabModalPresenter.java

@@ -10,6 +10,7 @@ import android.view.View;
 import android.view.ViewGroup;
 import android.widget.FrameLayout;
 
+import org.chromium.base.StrictModeContext;
 import org.chromium.components.browser_ui.modaldialog.R;
 import org.chromium.components.browser_ui.modaldialog.TabModalPresenter;
 import org.chromium.content_public.browser.WebContents;
@@ -43,10 +44,17 @@ public class WebLayerTabModalPresenter extends TabModalPresenter {
         runEnterAnimation();
     }
 
+    private FrameLayout loadDialogContainer() {
+        // LayoutInflater may trigger accessing the disk.
+        try (StrictModeContext ignored = StrictModeContext.allowDiskReads()) {
+            return (FrameLayout) LayoutInflater.from(mContext).inflate(
+                    R.layout.modal_dialog_container, null);
+        }
+    }
+
     @Override
     protected ViewGroup createDialogContainer() {
-        FrameLayout dialogContainer = (FrameLayout) LayoutInflater.from(mContext).inflate(
-                R.layout.modal_dialog_container, null);
+        FrameLayout dialogContainer = loadDialogContainer();
         dialogContainer.setVisibility(View.GONE);
         dialogContainer.setClickable(true);
 

weblayer/shell/android/shell_apk/src/org/chromium/weblayer/shell/InstrumentationActivity.java

@@ -8,6 +8,9 @@ import android.content.Context;
 import android.content.Intent;
 import android.net.Uri;
 import android.os.Bundle;
+import android.os.StrictMode;
+import android.os.StrictMode.ThreadPolicy;
+import android.os.StrictMode.VmPolicy;
 import android.support.v4.app.Fragment;
 import android.support.v4.app.FragmentActivity;
 import android.support.v4.app.FragmentManager;
@@ -58,6 +61,17 @@ public class InstrumentationActivity extends FragmentActivity {
     private Bundle mSavedInstanceState;
     private TabCallback mTabCallback;
 
+    private static boolean isJaCoCoEnabled() {
+        // Nothing is set at runtime indicating jacoco is being used. This looks for the existence
+        // of a javacoco class to determine if jacoco is enabled.
+        try {
+            Class.forName("org.jacoco.agent.rt.RT");
+            return true;
+        } catch (LinkageError | ClassNotFoundException e) {
+        }
+        return false;
+    }
+
     public Tab getTab() {
         return mTab;
     }
@@ -95,6 +109,19 @@ public class InstrumentationActivity extends FragmentActivity {
 
     @Override
     protected void onCreate(final Bundle savedInstanceState) {
+        // JaCoCo injects code that does file access, which doesn't work well with strict mode.
+        if (!isJaCoCoEnabled()) {
+            StrictMode.setThreadPolicy(
+                    new ThreadPolicy.Builder().detectAll().penaltyLog().penaltyDeath().build());
+            // This doesn't use detectAll() as the untagged sockets policy is encountered in tests
+            // using TestServer.
+            StrictMode.setVmPolicy(new VmPolicy.Builder()
+                                           .detectLeakedSqlLiteObjects()
+                                           .detectLeakedClosableObjects()
+                                           .penaltyLog()
+                                           .penaltyDeath()
+                                           .build());
+        }
         super.onCreate(savedInstanceState);
         mSavedInstanceState = savedInstanceState;
         LinearLayout mainView = new LinearLayout(this);