[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 "base/android/jni_android.h"
+#include "base/android/jni_string.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/SdchTestUtil_jni.h"
+#include "net/base/sdch_manager.h"
+#include "net/base/sdch_observer.h"
+#include "url/gurl.h"
+
+namespace cronet {
+
+namespace {
+
+net::SdchObserver* g_sdch_observer = nullptr;
+
+class TestSdchObserver : public net::SdchObserver {
+ public:
+  TestSdchObserver() {}
+
+  // SdchObserver implementation
+  void OnDictionaryAdded(net::SdchManager* manager,
+                         const GURL& dictionary_url,
+                         const std::string& server_hash) override {
+    JNIEnv* env = base::android::AttachCurrentThread();
+    Java_SdchTestUtil_onDictionaryAdded(
+        env, base::android::ConvertUTF8ToJavaString(env, dictionary_url.spec())
+                 .Release());
+  }
+
+  void OnDictionaryRemoved(net::SdchManager* manager,
+                           const std::string& server_hash){};
+
+  void OnDictionaryUsed(net::SdchManager* manager,
+                        const std::string& server_hash){};
+
+  void OnGetDictionary(net::SdchManager* manager,
+                       const GURL& request_url,
+                       const GURL& dictionary_url){};
+
+  void OnClearDictionaries(net::SdchManager* manager){};
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(TestSdchObserver);
+};
+
+void AddRemoveSdchObserverHelper(net::URLRequestContext* url_request_context,
+                                 bool add) {
+  // Add a SdchObserver for testing.
+  if (add) {
+    DCHECK(!g_sdch_observer);
+    g_sdch_observer = new TestSdchObserver();
+    url_request_context->sdch_manager()->AddObserver(g_sdch_observer);
+  } else {
+    DCHECK(g_sdch_observer);
+    url_request_context->sdch_manager()->RemoveObserver(g_sdch_observer);
+    g_sdch_observer = nullptr;

[mef, 2015/05/13 17:57:13]

do we intentionally leak |g_sdch_observer| here?

[xunjieli, 2015/05/13 19:32:47]

yes, the reason I did this is because we are required to remove the observer
that we registered, otherwise sdch_manager will complain during shutdown. So
making the observer as a global seems the easy way to doing it, although it
might not be the best way. I am open to suggestions.

+  }
+
+  JNIEnv* env = base::android::AttachCurrentThread();
+  Java_SdchTestUtil_onAddRemoveSdchObserver(env);
+}
+
+void AddRemoveSdchObserverOnNetworkThread(
+    CronetURLRequestContextAdapter* context_adapter,
+    bool add) {
+  AddRemoveSdchObserverHelper(context_adapter->GetURLRequestContext(), add);
+}
+
+// TODO(xunjieli): Delete this once legacy API is removed.
+void AddRemoveSdchObserverOnNetworkThreadLegacyAPI(
+    URLRequestContextAdapter* context_adapter,
+    bool add) {
+  AddRemoveSdchObserverHelper(context_adapter->GetURLRequestContext(), add);
+}
+
+}  // namespace
+
+void AddRemoveSdchObserver(JNIEnv* env,
+                           jclass jcaller,
+                           jlong jadapter,
+                           jboolean jlegacy_api,
+                           jboolean add) {
+  if (jlegacy_api == JNI_TRUE) {
+    URLRequestContextAdapter* context_adapter =
+        reinterpret_cast<URLRequestContextAdapter*>(jadapter);
+    context_adapter->PostTaskToNetworkThread(
+        FROM_HERE, base::Bind(&AddRemoveSdchObserverOnNetworkThreadLegacyAPI,
+                              base::Unretained(context_adapter),
+                              add ? JNI_TRUE : JNI_FALSE));
+    return;
+  }
+  CronetURLRequestContextAdapter* context_adapter =
+      reinterpret_cast<CronetURLRequestContextAdapter*>(jadapter);
+  context_adapter->PostTaskToNetworkThread(
+      FROM_HERE, base::Bind(&AddRemoveSdchObserverOnNetworkThread,
+                            base::Unretained(context_adapter),
+                            add ? JNI_TRUE : JNI_FALSE));
+}
+
+bool RegisterSdchTestUtil(JNIEnv* env) {
+  return RegisterNativesImpl(env);
+}
+
+}  // namespace cronet