Chromium Code Reviews| Index: components/cronet/android/test/native_test_server.cc |
| diff --git a/components/cronet/android/test/native_test_server.cc b/components/cronet/android/test/native_test_server.cc |
| index aab2d32599cf9510126cb7fab08b790680bf35b4..d62a1c6fe38f35e34700e709ff5be8e82d0f43ae 100644 |
| --- a/components/cronet/android/test/native_test_server.cc |
| +++ b/components/cronet/android/test/native_test_server.cc |
| @@ -168,14 +168,14 @@ scoped_ptr<net::test_server::HttpResponse> SdchRequestHandler( |
| return scoped_ptr<net::test_server::BasicHttpResponse>(); |
| } |
| -void RegisterHostResolverProcHelper( |
| - net::URLRequestContext* url_request_context) { |
| +void RegisterHostResolverProcHelper(net::URLRequestContext* url_request_context, |
| + const std::string& destination) { |
|
xunjieli
2015/12/17 21:42:23
nit: document what |destination| is.
pauljensen
2015/12/18 14:26:08
Done.
|
| 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(); |
| @@ -183,14 +183,18 @@ void RegisterHostResolverProcHelper( |
| } |
| 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 |
| @@ -217,19 +221,22 @@ jboolean StartNativeTestServer(JNIEnv* env, |
| 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 = |
| 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)); |
| } |
| } |