[Cronet] Add API to set DNS cache entry

components/cronet/android/cronet_url_request_context_adapter.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 "components/cronet/android/cronet_url_request_context_adapter.h"
 
 #include <limits.h>
 #include <stddef.h>
 #include <stdint.h>
 
@@ skipping 37 matching lines @@
 #include "components/prefs/pref_service_factory.h"
 #include "jni/CronetUrlRequestContext_jni.h"
 #include "net/base/load_flags.h"
 #include "net/base/logging_network_change_observer.h"
 #include "net/base/net_errors.h"
 #include "net/base/network_delegate_impl.h"
 #include "net/base/url_util.h"
 #include "net/cert/caching_cert_verifier.h"
 #include "net/cert/cert_verifier.h"
 #include "net/cookies/cookie_monster.h"
+#include "net/dns/host_cache.h"
 #include "net/http/http_auth_handler_factory.h"
 #include "net/http/http_server_properties_manager.h"
 #include "net/log/file_net_log_observer.h"
 #include "net/log/net_log_util.h"
 #include "net/nqe/external_estimate_provider.h"
 #include "net/nqe/network_qualities_prefs_manager.h"
 #include "net/proxy/proxy_config_service_android.h"
 #include "net/proxy/proxy_service.h"
 #include "net/sdch/sdch_owner.h"
 #include "net/ssl/channel_id_service.h"
@@ skipping 524 matching lines @@
 void CronetURLRequestContextAdapter::ProvideThroughputObservations(
     JNIEnv* env,
     const JavaParamRef<jobject>& jcaller,
     bool should) {
   PostTaskToNetworkThread(
       FROM_HERE, base::Bind(&CronetURLRequestContextAdapter::
                                 ProvideThroughputObservationsOnNetworkThread,
                             base::Unretained(this), should));
 }
 
+void CronetURLRequestContextAdapter::PutHostCacheOnNetworkThread(
+    std::string hostname,
+    net::AddressList address_list,
+    base::TimeDelta ttl) {
+  DCHECK(GetNetworkTaskRunner()->BelongsToCurrentThread());
+  net::HostCache::Key key(hostname, net::ADDRESS_FAMILY_UNSPECIFIED, 0);
+  net::HostCache::Entry entry(net::OK, address_list, ttl);
+  context_->host_resolver()->GetHostCache()->Set(key, entry,
+                                                 base::TimeTicks::Now(), ttl);
+}
+
+void CronetURLRequestContextAdapter::PutHostCache(
+    JNIEnv* env,
+    const base::android::JavaParamRef<jobject>& jcaller,
+    const base::android::JavaParamRef<jstring>& jhostname,
+    const base::android::JavaParamRef<jobjectArray>& jaddresses,
+    jint jseconds_to_live) {
+  size_t address_count = env->GetArrayLength(jaddresses);
+  net::AddressList address_list;
+  for (size_t i = 0; i < address_count; ++i) {
+    ScopedJavaLocalRef<jbyteArray> bytes_array(
+        env,
+        static_cast<jbyteArray>(env->GetObjectArrayElement(jaddresses, i)));
+    size_t address_size = env->GetArrayLength(bytes_array.obj());
+    if (address_size != net::IPAddress::kIPv4AddressSize &&
+        address_size != net::IPAddress::kIPv6AddressSize) {
+      LOG(ERROR) << "Unsupported IP address size: " << address_size;
+      return;
+    }
+    jbyte* bytes = env->GetByteArrayElements(bytes_array.obj(), nullptr);
+    address_list.push_back(net::IPEndPoint(
+        net::IPAddress(reinterpret_cast<uint8_t*>(bytes), address_size), 0));
+    env->ReleaseByteArrayElements(bytes_array.obj(), bytes, JNI_ABORT);
+  }
+  PostTaskToNetworkThread(
+      FROM_HERE,
+      base::Bind(&CronetURLRequestContextAdapter::PutHostCacheOnNetworkThread,
+                 base::Unretained(this),
+                 base::android::ConvertJavaStringToUTF8(env, jhostname),
+                 address_list, base::TimeDelta::FromSeconds(jseconds_to_live)));
+}
+
 void CronetURLRequestContextAdapter::InitializeOnNetworkThread(
     std::unique_ptr<URLRequestContextConfig> config,
     const base::android::ScopedJavaGlobalRef<jobject>&
         jcronet_url_request_context) {
   DCHECK(GetNetworkTaskRunner()->BelongsToCurrentThread());
   DCHECK(!is_context_initialized_);
   DCHECK(proxy_config_service_);
 
   // TODO(mmenke):  Add method to have the builder enable SPDY.
   net::URLRequestContextBuilder context_builder;
@@ skipping 562 matching lines @@
     JNIEnv* env,
     const JavaParamRef<jclass>& jcaller) {
   DCHECK(base::StatisticsRecorder::IsActive());
   std::vector<uint8_t> data;
   if (!HistogramManager::GetInstance()->GetDeltas(&data))
     return ScopedJavaLocalRef<jbyteArray>();
   return base::android::ToJavaByteArray(env, &data[0], data.size());
 }
 
 }  // namespace cronet