Update prerender policy for custom tabs

chrome/android/java/src/org/chromium/chrome/browser/customtabs/ClientManager.java

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.chrome.browser.customtabs;
 
 import android.content.ComponentName;
 import android.content.Context;
 import android.content.Intent;
 import android.content.ServiceConnection;
@@ skipping 36 matching lines @@
     @VisibleForTesting static final int SESSION_WARMUP_COUNT = 5;
 
     /** Per-session values. */
     private static class SessionParams {
         public final int uid;
         public final String packageName;
         public final ICustomTabsCallback callback;
         public final IBinder.DeathRecipient deathRecipient;
         public boolean mIgnoreFragments;
         private boolean mShouldHideDomain;
+        private boolean mShouldPrerenderOnCellular;
         private ServiceConnection mKeepAliveConnection;
         private String mPredictedUrl;
         private long mLastMayLaunchUrlTimestamp;
 
         public SessionParams(Context context, int uid, ICustomTabsCallback callback,
                 IBinder.DeathRecipient deathRecipient) {
             this.uid = uid;
             packageName = getPackageName(context, uid);
             this.callback = callback;
             this.deathRecipient = deathRecipient;
@@ skipping 196 matching lines @@
      * @return Whether the fragment should be ignored for prerender matching.
      */
     public synchronized boolean getIgnoreFragmentsForSession(IBinder session) {
         SessionParams params = mSessionParams.get(session);
         return params == null ? false : params.mIgnoreFragments;
     }
 
     /** Sets whether the fragment should be ignored for prerender matching. */
     public synchronized void setIgnoreFragmentsForSession(IBinder session, boolean value) {
         SessionParams params = mSessionParams.get(session);
-        params.mIgnoreFragments = value;
+        if (params != null) params.mIgnoreFragments = value;
+    }
+
+    /**
+     * @return Whether prerender should be turned on for cellular networks for given session.
+     */
+    public synchronized boolean shouldPrerenderOnCellularForSession(IBinder session) {
+        SessionParams params = mSessionParams.get(session);
+        return params != null ? params.mShouldPrerenderOnCellular : false;
+    }
+
+    /**
+     * Sets whether prerender should be turned on for mobile networks for given session.
+     */
+    public synchronized void setPrerenderCellularForSession(IBinder session, boolean prerender) {
+        SessionParams params = mSessionParams.get(session);
+        if (params != null) params.mShouldPrerenderOnCellular = prerender;
     }
 
     /** Tries to bind to a client to keep it alive, and returns true for success. */
     public synchronized boolean keepAliveForSession(IBinder session, Intent intent) {
         // When an application is bound to a service, its priority is raised to
         // be at least equal to the application's one. This binds to a dummy
         // service (no calls to this service are made).
         if (intent == null || intent.getComponent() == null) return false;
         SessionParams params = mSessionParams.get(session);
         if (params == null) return false;
@@ skipping 58 matching lines @@
 
     private synchronized void cleanupSession(IBinder session) {
         SessionParams params = mSessionParams.get(session);
         if (params == null) return;
         mSessionParams.remove(session);
         mUidHasCalledWarmup.delete(params.uid);
         IBinder binder = params.callback.asBinder();
         binder.unlinkToDeath(params.deathRecipient, 0);
     }
 }