[Cronet] Enable persistence mode for Sdch

components/cronet/android/test/sdch_test_util.cc

+// 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.
+
+#include "sdch_test_util.h"
+
+#include <string>
+
+#include "base/android/jni_android.h"
+#include "base/android/jni_string.h"
+#include "base/android/scoped_java_ref.h"
+#include "base/bind.h"
+#include "base/macros.h"
+#include "components/cronet/android/cronet_url_request_context_adapter.h"
+#include "components/cronet/android/url_request_context_adapter.h"
+#include "jni/SdchObserver_jni.h"
+#include "net/base/sdch_manager.h"
+#include "net/base/sdch_observer.h"
+#include "net/url_request/url_request_context.h"
+#include "url/gurl.h"
+
+namespace cronet {
+
+namespace {
+
+class TestSdchObserver : public net::SdchObserver {
+ public:
+  TestSdchObserver(
+      const GURL& target_url,
+      net::SdchManager* manager,
+      const base::android::ScopedJavaGlobalRef<jobject>& jsdch_observer_ref)
+      : target_url_(target_url), manager_(manager) {
+    jsdch_observer_ref_.Reset(jsdch_observer_ref);
+  }
+
+  // SdchObserver implementation
+  void OnDictionaryAdded(const GURL& dictionary_url,
+                         const std::string& server_hash) override {
+    // Only notify if the dictionary for the |target_url_| has been added.
+    if (manager_->GetDictionarySet(target_url_)) {
+      JNIEnv* env = base::android::AttachCurrentThread();
+      Java_SdchObserver_onDictionaryAdded(env, jsdch_observer_ref_.obj());
+      manager_->RemoveObserver(this);
+      delete this;
+    }
+  }
+
+  void OnDictionaryRemoved(const std::string& server_hash) override {}
+
+  void OnDictionaryUsed(const std::string& server_hash) override {}
+
+  void OnGetDictionary(const GURL& request_url,
+                       const GURL& dictionary_url) override {}
+
+  void OnClearDictionaries() override {}
+
+ private:
+  GURL target_url_;
+  net::SdchManager* manager_;
+  base::android::ScopedJavaGlobalRef<jobject> jsdch_observer_ref_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestSdchObserver);
+};
+
+void AddSdchObserverHelper(
+    const GURL& target_url,
+    const base::android::ScopedJavaGlobalRef<jobject>& jsdch_observer_ref,
+    net::URLRequestContext* url_request_context) {
+  JNIEnv* env = base::android::AttachCurrentThread();
+  // If dictionaries for |target_url| are already added, skip adding the
+  // observer.
+  if (url_request_context->sdch_manager()->GetDictionarySet(target_url)) {
+    Java_SdchObserver_onDictionarySetAlreadyPresent(env,
+                                                    jsdch_observer_ref.obj());
+    return;
+  }
+
+  url_request_context->sdch_manager()->AddObserver(new TestSdchObserver(
+      target_url, url_request_context->sdch_manager(), jsdch_observer_ref));
+  Java_SdchObserver_onAddSdchObserverCompleted(env, jsdch_observer_ref.obj());
+}
+
+void AddSdchObserverOnNetworkThread(
+    const GURL& target_url,
+    const base::android::ScopedJavaGlobalRef<jobject>& jsdch_observer_ref,
+    CronetURLRequestContextAdapter* context_adapter) {
+  AddSdchObserverHelper(target_url, jsdch_observer_ref,
+                        context_adapter->GetURLRequestContext());
+}
+
+// TODO(xunjieli): Delete this once legacy API is removed.
+void AddSdchObserverOnNetworkThreadLegacyAPI(
+    const GURL& target_url,
+    const base::android::ScopedJavaGlobalRef<jobject>& jsdch_observer_ref,
+    URLRequestContextAdapter* context_adapter) {
+  AddSdchObserverHelper(target_url, jsdch_observer_ref,
+                        context_adapter->GetURLRequestContext());
+}
+
+}  // namespace
+
+void AddSdchObserver(JNIEnv* env,
+                     jobject jsdch_observer,
+                     jstring jtarget_url,
+                     jlong jadapter) {
+  base::android::ScopedJavaGlobalRef<jobject> jsdch_observer_ref;
+  // ScopedJavaGlobalRef do not hold onto the env reference, so it is safe to
+  // use it across threads. |AddSdchObserverHelper| will acquire a new
+  // JNIEnv before calling into Java.
+  jsdch_observer_ref.Reset(env, jsdch_observer);
+
+  GURL target_url(base::android::ConvertJavaStringToUTF8(env, jtarget_url));
+  CronetURLRequestContextAdapter* context_adapter =
+      reinterpret_cast<CronetURLRequestContextAdapter*>(jadapter);
+  context_adapter->PostTaskToNetworkThread(
+      FROM_HERE,
+      base::Bind(&AddSdchObserverOnNetworkThread, target_url,
+                 jsdch_observer_ref, base::Unretained(context_adapter)));
+}
+
+void AddSdchObserverLegacyAPI(JNIEnv* env,
+                              jobject jsdch_observer,
+                              jstring jtarget_url,
+                              jlong jadapter) {
+  base::android::ScopedJavaGlobalRef<jobject> jsdch_observer_ref;
+  // ScopedJavaGlobalRef do not hold onto the env reference, so it is safe to
+  // use it across threads. |AddSdchObserverHelper| will acquire a new
+  // JNIEnv before calling into Java.
+  jsdch_observer_ref.Reset(env, jsdch_observer);
+  GURL target_url(base::android::ConvertJavaStringToUTF8(env, jtarget_url));
+  URLRequestContextAdapter* context_adapter =
+      reinterpret_cast<URLRequestContextAdapter*>(jadapter);
+  context_adapter->PostTaskToNetworkThread(
+      FROM_HERE,
+      base::Bind(&AddSdchObserverOnNetworkThreadLegacyAPI, target_url,
+                 jsdch_observer_ref, base::Unretained(context_adapter)));
+}
+
+bool RegisterSdchTestUtil(JNIEnv* env) {
+  return RegisterNativesImpl(env);
+}
+
+}  // namespace cronet