content/browser/renderer_host/java/java_bridge_dispatcher_host_manager.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/browser/renderer_host/java/java_bridge_dispatcher_host_manager.h"
   6
   7 #include "base/android/jni_android.h"
   8 #include "base/android/jni_helper.h"
   9 #include "base/android/scoped_java_ref.h"
  10 #include "base/bind.h"
  11 #include "base/logging.h"
  12 #include "base/strings/utf_string_conversions.h"
  13 #include "content/browser/renderer_host/java/java_bound_object.h"
  14 #include "content/browser/renderer_host/java/java_bridge_dispatcher_host.h"
  15 #include "content/common/android/hash_set.h"
  16 #include "content/public/browser/browser_thread.h"
  17 #include "third_party/WebKit/public/web/WebBindings.h"
  18
  19 namespace content {
  20
  21 JavaBridgeDispatcherHostManager::JavaBridgeDispatcherHostManager(
  22     WebContents* web_contents)
  23     : WebContentsObserver(web_contents) {
  24 }
  25
  26 JavaBridgeDispatcherHostManager::~JavaBridgeDispatcherHostManager() {
  27   for (ObjectMap::iterator iter = objects_.begin(); iter != objects_.end();
  28       ++iter) {
  29     blink::WebBindings::releaseObject(iter->second);
  30   }
  31   DCHECK_EQ(0U, instances_.size());
  32 }
  33
  34 void JavaBridgeDispatcherHostManager::AddNamedObject(const base::string16& name,
  35                                                      NPObject* object) {
  36   // Record this object in a map so that we can add it into RenderViewHosts
  37   // created later. The JavaBridgeDispatcherHost instances will take a
  38   // reference to the object, but we take one too, because this method can be
  39   // called before there are any such instances.
  40   blink::WebBindings::retainObject(object);
  41   objects_[name] = object;
  42
  43   for (InstanceMap::iterator iter = instances_.begin();
  44       iter != instances_.end(); ++iter) {
  45     iter->second->AddNamedObject(name, object);
  46   }
  47 }
  48
  49 void JavaBridgeDispatcherHostManager::SetRetainedObjectSet(
  50     const JavaObjectWeakGlobalRef& retained_object_set) {
  51   // It's an error to replace the retained_object_set_ after it's been set,
  52   // so we check that it hasn't already been here.
  53   // TODO(benm): It'd be better to pass the set in the constructor to avoid
  54   // the chance of this happening; but that's tricky as this get's constructed
  55   // before ContentViewCore (which owns the set). Best solution may be to move
  56   // ownership of the JavaBridgerDispatchHostManager from WebContents to
  57   // ContentViewCore?
  58   JNIEnv* env = base::android::AttachCurrentThread();
  59   base::android::ScopedJavaLocalRef<jobject> new_retained_object_set =
  60       retained_object_set.get(env);
  61   base::android::ScopedJavaLocalRef<jobject> current_retained_object_set =
  62       retained_object_set_.get(env);
  63   if (!env->IsSameObject(new_retained_object_set.obj(),
  64                          current_retained_object_set.obj())) {
  65     DCHECK(current_retained_object_set.is_null());
  66     retained_object_set_ = retained_object_set;
  67   }
  68 }
  69
  70 void JavaBridgeDispatcherHostManager::RemoveNamedObject(
  71     const base::string16& name) {
  72   ObjectMap::iterator iter = objects_.find(name);
  73   if (iter == objects_.end()) {
  74     return;
  75   }
  76
  77   blink::WebBindings::releaseObject(iter->second);
  78   objects_.erase(iter);
  79
  80   for (InstanceMap::iterator iter = instances_.begin();
  81       iter != instances_.end(); ++iter) {
  82     iter->second->RemoveNamedObject(name);
  83   }
  84 }
  85
  86 void JavaBridgeDispatcherHostManager::OnGetChannelHandle(
  87     RenderViewHost* render_view_host, IPC::Message* reply_msg) {
  88   instances_[render_view_host]->OnGetChannelHandle(reply_msg);
  89 }
  90
  91 void JavaBridgeDispatcherHostManager::RenderViewCreated(
  92     RenderViewHost* render_view_host) {
  93   // Creates a JavaBridgeDispatcherHost for the specified RenderViewHost and
  94   // adds all currently registered named objects to the new instance.
  95   scoped_refptr<JavaBridgeDispatcherHost> instance =
  96       new JavaBridgeDispatcherHost(render_view_host);
  97
  98   for (ObjectMap::const_iterator iter = objects_.begin();
  99       iter != objects_.end(); ++iter) {
 100     instance->AddNamedObject(iter->first, iter->second);
 101   }
 102
 103   instances_[render_view_host] = instance;
 104 }
 105
 106 void JavaBridgeDispatcherHostManager::RenderViewDeleted(
 107     RenderViewHost* render_view_host) {
 108   if (!instances_.count(render_view_host))  // Needed for tests.
 109     return;
 110   instances_[render_view_host]->RenderViewDeleted();
 111   instances_.erase(render_view_host);
 112 }
 113
 114 void JavaBridgeDispatcherHostManager::DocumentAvailableInMainFrame() {
 115   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 116   // Called when the window object has been cleared in the main frame.
 117   JNIEnv* env = base::android::AttachCurrentThread();
 118   base::android::ScopedJavaLocalRef<jobject> retained_object_set =
 119       retained_object_set_.get(env);
 120   if (!retained_object_set.is_null()) {
 121     JNI_Java_HashSet_clear(env, retained_object_set);
 122
 123     // We also need to add back the named objects we have so far as they
 124     // should survive navigations.
 125     ObjectMap::iterator it = objects_.begin();
 126     for (; it != objects_.end(); ++it) {
 127       JNI_Java_HashSet_add(env, retained_object_set,
 128                            JavaBoundObject::GetJavaObject(it->second));
 129     }
 130   }
 131 }
 132
 133 void JavaBridgeDispatcherHostManager::JavaBoundObjectCreated(
 134     const base::android::JavaRef<jobject>& object) {
 135   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 136
 137   JNIEnv* env = base::android::AttachCurrentThread();
 138   base::android::ScopedJavaLocalRef<jobject> retained_object_set =
 139       retained_object_set_.get(env);
 140   if (!retained_object_set.is_null()) {
 141     JNI_Java_HashSet_add(env, retained_object_set, object);
 142   }
 143 }
 144
 145 void JavaBridgeDispatcherHostManager::JavaBoundObjectDestroyed(
 146     const base::android::JavaRef<jobject>& object) {
 147   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 148
 149   JNIEnv* env = base::android::AttachCurrentThread();
 150   base::android::ScopedJavaLocalRef<jobject> retained_object_set =
 151       retained_object_set_.get(env);
 152   if (!retained_object_set.is_null()) {
 153     JNI_Java_HashSet_remove(env, retained_object_set, object);
 154   }
 155 }
 156
 157 }  // namespace content