1 files changed, 112 insertions, 0 deletions
diff --git a/webkit/plugins/ppapi/host_var_tracker.cc b/webkit/plugins/ppapi/host_var_tracker.cc
new file mode 100644
index 0000000..0508d2c
--- /dev/null
+++ b/webkit/plugins/ppapi/host_var_tracker.cc
@@ -0,0 +1,112 @@
+// Copyright (c) 2011 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 "webkit/plugins/ppapi/host_var_tracker.h"
+
+#include "base/logging.h"
+#include "ppapi/c/pp_var.h"
+#include "webkit/plugins/ppapi/npobject_var.h"
+#include "webkit/plugins/ppapi/ppapi_plugin_instance.h"
+
+using ppapi::NPObjectVar;
+
+namespace webkit {
+namespace ppapi {
+
+HostVarTracker::HostVarTracker() {
+}
+
+HostVarTracker::~HostVarTracker() {
+}
+
+void HostVarTracker::AddNPObjectVar(NPObjectVar* object_var) {
+  InstanceMap::iterator found_instance = instance_map_.find(
+      object_var->pp_instance());
+  if (found_instance == instance_map_.end()) {
+    // Lazily create the instance map.
+    DCHECK(object_var->pp_instance() != 0);
+    found_instance = instance_map_.insert(std::make_pair(
+        object_var->pp_instance(),
+        linked_ptr<NPObjectToNPObjectVarMap>(new NPObjectToNPObjectVarMap))).
+            first;
+  }
+  NPObjectToNPObjectVarMap* np_object_map = found_instance->second.get();
+
+  DCHECK(np_object_map->find(object_var->np_object()) ==
+         np_object_map->end()) << "NPObjectVar already in map";
+  np_object_map->insert(
+      std::make_pair(object_var->np_object(), object_var));
+}
+
+void HostVarTracker::RemoveNPObjectVar(NPObjectVar* object_var) {
+  InstanceMap::iterator found_instance = instance_map_.find(
+      object_var->pp_instance());
+  if (found_instance == instance_map_.end()) {
+    NOTREACHED() << "NPObjectVar has invalid instance.";
+    return;
+  }
+  NPObjectToNPObjectVarMap* np_object_map = found_instance->second.get();
+
+  NPObjectToNPObjectVarMap::iterator found_object =
+      np_object_map->find(object_var->np_object());
+  if (found_object == np_object_map->end()) {
+    NOTREACHED() << "NPObjectVar not registered.";
+    return;
+  }
+  if (found_object->second != object_var) {
+    NOTREACHED() << "NPObjectVar doesn't match.";
+    return;
+  }
+  np_object_map->erase(found_object);
+
+  // Clean up when the map is empty.
+  if (np_object_map->empty())
+    instance_map_.erase(found_instance);
+}
+
+NPObjectVar* HostVarTracker::NPObjectVarForNPObject(PP_Instance instance,
+                                                    NPObject* np_object) {
+  InstanceMap::iterator found_instance = instance_map_.find(instance);
+  if (found_instance == instance_map_.end())
+    return NULL;  // No such instance.
+  NPObjectToNPObjectVarMap* np_object_map = found_instance->second.get();
+
+  NPObjectToNPObjectVarMap::iterator found_object =
+      np_object_map->find(np_object);
+  if (found_object == np_object_map->end())
+    return NULL;  // No such object.
+  return found_object->second;
+}
+
+int HostVarTracker::GetLiveNPObjectVarsForInstance(PP_Instance instance) const {
+  InstanceMap::const_iterator found = instance_map_.find(instance);
+  if (found == instance_map_.end())
+    return 0;
+  return static_cast<int>(found->second->size());
+}
+
+void HostVarTracker::ForceFreeNPObjectsForInstance(PP_Instance instance) {
+  InstanceMap::iterator found_instance = instance_map_.find(instance);
+  if (found_instance == instance_map_.end())
+    return;  // Nothing to do.
+  NPObjectToNPObjectVarMap* np_object_map = found_instance->second.get();
+
+  // Force delete all var references. Need to make a copy so we can iterate over
+  // the map while deleting stuff from it.
+  NPObjectToNPObjectVarMap np_object_map_copy = *np_object_map;
+  NPObjectToNPObjectVarMap::iterator cur_var =
+      np_object_map_copy.begin();
+  while (cur_var != np_object_map_copy.end()) {
+    NPObjectToNPObjectVarMap::iterator current = cur_var++;
+    current->second->InstanceDeleted();
+    np_object_map->erase(current->first);
+  }
+
+  // Remove the record for this instance since it should be empty.
+  DCHECK(np_object_map->empty());
+  instance_map_.erase(found_instance);
+}
+
+}  // namespace ppapi
+}  // namespace webkit