extensions/renderer/object_backed_native_handler.cc

   1 // Copyright 2014 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 "extensions/renderer/object_backed_native_handler.h"
   6
   7 #include "base/logging.h"
   8 #include "base/memory/linked_ptr.h"
   9 #include "extensions/renderer/console.h"
  10 #include "extensions/renderer/module_system.h"
  11 #include "extensions/renderer/script_context.h"
  12 #include "v8/include/v8.h"
  13
  14 namespace extensions {
  15
  16 namespace {
  17 // Key for the base::Bound routed function.
  18 const char* kHandlerFunction = "handler_function";
  19 }  // namespace
  20
  21 ObjectBackedNativeHandler::ObjectBackedNativeHandler(ScriptContext* context)
  22     : router_data_(context->isolate()),
  23       context_(context),
  24       object_template_(context->isolate(),
  25                        v8::ObjectTemplate::New(context->isolate())) {
  26 }
  27
  28 ObjectBackedNativeHandler::~ObjectBackedNativeHandler() { Invalidate(); }
  29
  30 v8::Handle<v8::Object> ObjectBackedNativeHandler::NewInstance() {
  31   return v8::Local<v8::ObjectTemplate>::New(GetIsolate(), object_template_)
  32       ->NewInstance();
  33 }
  34
  35 // static
  36 void ObjectBackedNativeHandler::Router(
  37     const v8::FunctionCallbackInfo<v8::Value>& args) {
  38   v8::HandleScope handle_scope(args.GetIsolate());
  39   v8::Handle<v8::Object> data = args.Data().As<v8::Object>();
  40
  41   v8::Handle<v8::Value> handler_function_value =
  42       data->Get(v8::String::NewFromUtf8(args.GetIsolate(), kHandlerFunction));
  43   // See comment in header file for why we do this.
  44   if (handler_function_value.IsEmpty() ||
  45       handler_function_value->IsUndefined()) {
  46     console::Error(args.GetIsolate()->GetCallingContext(),
  47                    "Extension view no longer exists");
  48     return;
  49   }
  50   DCHECK(handler_function_value->IsExternal());
  51   static_cast<HandlerFunction*>(
  52       handler_function_value.As<v8::External>()->Value())->Run(args);
  53 }
  54
  55 void ObjectBackedNativeHandler::RouteFunction(
  56     const std::string& name,
  57     const HandlerFunction& handler_function) {
  58   v8::Isolate* isolate = v8::Isolate::GetCurrent();
  59   v8::HandleScope handle_scope(isolate);
  60   v8::Context::Scope context_scope(context_->v8_context());
  61
  62   v8::Local<v8::Object> data = v8::Object::New(isolate);
  63   data->Set(
  64       v8::String::NewFromUtf8(isolate, kHandlerFunction),
  65       v8::External::New(isolate, new HandlerFunction(handler_function)));
  66   v8::Handle<v8::FunctionTemplate> function_template =
  67       v8::FunctionTemplate::New(isolate, Router, data);
  68   v8::Local<v8::ObjectTemplate>::New(isolate, object_template_)
  69       ->Set(isolate, name.c_str(), function_template);
  70   router_data_.Append(data);
  71 }
  72
  73 v8::Isolate* ObjectBackedNativeHandler::GetIsolate() const {
  74   return context_->isolate();
  75 }
  76
  77 void ObjectBackedNativeHandler::Invalidate() {
  78   if (!is_valid())
  79     return;
  80   v8::Isolate* isolate = v8::Isolate::GetCurrent();
  81   v8::HandleScope handle_scope(isolate);
  82   v8::Context::Scope context_scope(context_->v8_context());
  83
  84   for (size_t i = 0; i < router_data_.Size(); i++) {
  85     v8::Handle<v8::Object> data = router_data_.Get(i);
  86     v8::Handle<v8::Value> handler_function_value =
  87         data->Get(v8::String::NewFromUtf8(isolate, kHandlerFunction));
  88     CHECK(!handler_function_value.IsEmpty());
  89     delete static_cast<HandlerFunction*>(
  90         handler_function_value.As<v8::External>()->Value());
  91     data->Delete(v8::String::NewFromUtf8(isolate, kHandlerFunction));
  92   }
  93   router_data_.Clear();
  94   object_template_.Reset();
  95   context_ = NULL;
  96   NativeHandler::Invalidate();
  97 }
  98
  99 }  // namespace extensions