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