[Cronet] Get Cronet performance test running again

components/cronet/android/test/native_test_server.cc

 // Copyright 2014 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 "native_test_server.h"
 
 #include <string>
 
 #include "base/android/jni_android.h"
 #include "base/android/jni_string.h"
@@ skipping 150 matching lines @@
         new net::test_server::BasicHttpResponse());
     response->set_content_type("text/plain");
     response->set_content("Sdch is not used.\n");
     return response.Pass();
   }
 
   // Unhandled requests result in the Embedded test server sending a 404.
   return scoped_ptr<net::test_server::BasicHttpResponse>();
 }
 
-void RegisterHostResolverProcHelper(
-    net::URLRequestContext* url_request_context) {
+// Install host resolver rules to map fake domains to |destination|, usually an
+// IP address.
+void RegisterHostResolverProcHelper(net::URLRequestContext* url_request_context,
+                                    const std::string& destination) {
   net::HostResolverImpl* resolver =
       static_cast<net::HostResolverImpl*>(url_request_context->host_resolver());
   scoped_refptr<net::RuleBasedHostResolverProc> proc =
       new net::RuleBasedHostResolverProc(NULL);
-  proc->AddRule(kFakeSdchDomain, "127.0.0.1");
-  proc->AddRule(kFakeQuicDomain, "127.0.0.1");
+  proc->AddRule(kFakeSdchDomain, destination);
+  proc->AddRule(kFakeQuicDomain, destination);
   resolver->set_proc_params_for_test(
       net::HostResolverImpl::ProcTaskParams(proc.get(), 1u));
   JNIEnv* env = base::android::AttachCurrentThread();
   Java_NativeTestServer_onHostResolverProcRegistered(env);
 }
 
 void RegisterHostResolverProcOnNetworkThread(
-    CronetURLRequestContextAdapter* context_adapter) {
-  RegisterHostResolverProcHelper(context_adapter->GetURLRequestContext());
+    CronetURLRequestContextAdapter* context_adapter,
+    const std::string& destination) {
+  RegisterHostResolverProcHelper(context_adapter->GetURLRequestContext(),
+                                 destination);
 }
 
 // TODO(xunjieli): Delete this once legacy API is removed.
 void RegisterHostResolverProcOnNetworkThreadLegacyAPI(
-    URLRequestContextAdapter* context_adapter) {
-  RegisterHostResolverProcHelper(context_adapter->GetURLRequestContext());
+    URLRequestContextAdapter* context_adapter,
+    const std::string& destination) {
+  RegisterHostResolverProcHelper(context_adapter->GetURLRequestContext(),
+                                 destination);
 }
 
 }  // namespace
 
 jboolean StartNativeTestServer(JNIEnv* env,
                                const JavaParamRef<jclass>& jcaller,
                                const JavaParamRef<jstring>& jtest_files_root) {
   // Shouldn't happen.
   if (g_test_server)
     return false;
   g_test_server = new net::EmbeddedTestServer();
   g_test_server->RegisterRequestHandler(
       base::Bind(&NativeTestServerRequestHandler));
   g_test_server->RegisterRequestHandler(base::Bind(&SdchRequestHandler));
   base::FilePath test_files_root(
       base::android::ConvertJavaStringToUTF8(env, jtest_files_root));
 
   // Add a third handler for paths that NativeTestServerRequestHandler does not
   // handle.
   g_test_server->ServeFilesFromDirectory(test_files_root);
   return g_test_server->Start();
 }
 
 void RegisterHostResolverProc(JNIEnv* env,
                               const JavaParamRef<jclass>& jcaller,
                               jlong jadapter,
-                              jboolean jlegacy_api) {
+                              jboolean jlegacy_api,
+                              const JavaParamRef<jstring>& jdestination) {
+  std::string destination(
+      base::android::ConvertJavaStringToUTF8(env, jdestination));
   if (jlegacy_api == JNI_TRUE) {
     URLRequestContextAdapter* context_adapter =

[mef, 2015/12/29 17:27:45]

Does it make sense to call context_adapter->GetURLRequestContext() here and post
RegisterHostResolverProcHelper() call in both legacy and non-legacy case?

[pauljensen, 2016/01/21 03:51:50]

GetURLRequestContext() can only be called from network thread, so we can't do it
here.

         reinterpret_cast<URLRequestContextAdapter*>(jadapter);
     context_adapter->PostTaskToNetworkThread(
         FROM_HERE, base::Bind(&RegisterHostResolverProcOnNetworkThreadLegacyAPI,
-                              base::Unretained(context_adapter)));
+                              base::Unretained(context_adapter), destination));
   } else {
     CronetURLRequestContextAdapter* context_adapter =
         reinterpret_cast<CronetURLRequestContextAdapter*>(jadapter);
     context_adapter->PostTaskToNetworkThread(
         FROM_HERE, base::Bind(&RegisterHostResolverProcOnNetworkThread,
-                              base::Unretained(context_adapter)));
+                              base::Unretained(context_adapter), destination));
   }
 }
 
 void ShutdownNativeTestServer(JNIEnv* env,
                               const JavaParamRef<jclass>& jcaller) {
   if (!g_test_server)
     return;
   delete g_test_server;
   g_test_server = NULL;
 }
@@ skipping 76 matching lines @@
 #else
   return JNI_FALSE;
 #endif
 }
 
 bool RegisterNativeTestServer(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
 
 }  // namespace cronet