Update prerender policy for custom tabs

chrome/browser/prerender/external_prerender_handler_android.cc

 // Copyright 2013 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.
 
 #include "chrome/browser/prerender/external_prerender_handler_android.h"
 
 #include "base/android/jni_android.h"
 #include "base/android/jni_string.h"
 #include "base/strings/string16.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/prerender/prerender_handle.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/prerender/prerender_manager_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_android.h"
 #include "content/public/browser/web_contents.h"
-
 #include "jni/ExternalPrerenderHandler_jni.h"
+#include "net/base/network_change_notifier.h"
 
 using base::android::ConvertJavaStringToUTF16;
 
 namespace prerender {
 
 namespace {
 
 static bool CheckAndConvertParams(
     JNIEnv* env,
     const JavaParamRef<jobject>& jprofile,
@@ skipping 20 matching lines @@
 }  // namespace
 
 bool ExternalPrerenderHandlerAndroid::AddPrerender(
     JNIEnv* env,
     const JavaParamRef<jobject>& obj,
     const JavaParamRef<jobject>& jprofile,
     const JavaParamRef<jobject>& jweb_contents,
     const JavaParamRef<jstring>& jurl,
     const JavaParamRef<jstring>& jreferrer,
     jint width,
-    jint height) {
+    jint height,
+    jboolean prerender_on_cellular) {
   Profile* profile = ProfileAndroid::FromProfileAndroid(jprofile);
 
   GURL url = GURL(ConvertJavaStringToUTF16(env, jurl));
   if (!url.is_valid())
     return false;
 
   GURL referrer_url = GURL(ConvertJavaStringToUTF16(env, jreferrer));
   content::Referrer referrer = referrer_url.is_valid() ?
       content::Referrer(referrer_url, blink::WebReferrerPolicyDefault) :
       content::Referrer();
   PrerenderManager* prerender_manager =
       prerender::PrerenderManagerFactory::GetForProfile(profile);
   if (!prerender_manager)
     return false;
   content::WebContents* web_contents =
       content::WebContents::FromJavaWebContents(jweb_contents);
   if (prerender_handle_.get()) {
     prerender_handle_->OnNavigateAway();
   }
-  prerender_handle_.reset(
-      prerender_manager->AddPrerenderFromExternalRequest(
-          url,
-          referrer,
-          web_contents->GetController().GetDefaultSessionStorageNamespace(),
-          gfx::Size(width, height)));
+  if (prerender_on_cellular && net::NetworkChangeNotifier::IsConnectionCellular(
+                   net::NetworkChangeNotifier::GetConnectionType())) {
+    prerender_handle_.reset(
+        prerender_manager->AddPrerenderOnCellularFromExternalRequest(
+            url,
+            referrer,
+            web_contents->GetController().GetDefaultSessionStorageNamespace(),
+            gfx::Size(width, height)));
+  } else {
+    prerender_handle_.reset(
+        prerender_manager->AddPrerenderFromExternalRequest(
+            url,
+            referrer,
+            web_contents->GetController().GetDefaultSessionStorageNamespace(),
+            gfx::Size(width, height)));
+  }
+
   if (!prerender_handle_)
     return false;
   return true;
 }
 
 void ExternalPrerenderHandlerAndroid::CancelCurrentPrerender(
     JNIEnv* env,
     const JavaParamRef<jobject>& object) {
   if (!prerender_handle_)
     return;
@@ skipping 40 matching lines @@
       new ExternalPrerenderHandlerAndroid();
   return reinterpret_cast<intptr_t>(external_handler);
 }
 
 bool ExternalPrerenderHandlerAndroid::RegisterExternalPrerenderHandlerAndroid(
     JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
 
 } // namespace prerender