Extract ExtensionFunctionRegistry from ExtensionFunctionDispatcher.

chrome/browser/extensions/extension_function_registry.cc

Index: chrome/browser/extensions/extension_function_registry.cc
diff --git a/chrome/browser/extensions/extension_function_dispatcher.cc b/chrome/browser/extensions/extension_function_registry.cc
similarity index 60%
copy from chrome/browser/extensions/extension_function_dispatcher.cc
copy to chrome/browser/extensions/extension_function_registry.cc
index 7d04788345ced7eb334f0db04fdbebced4b4e3c8..99d525cfeb545428c89bdf9c93b232cefb51648e 100644
--- a/chrome/browser/extensions/extension_function_dispatcher.cc
+++ b/chrome/browser/extensions/extension_function_registry.cc
@@ -2,16 +2,8 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-#include "chrome/browser/extensions/extension_function_dispatcher.h"
+#include "chrome/browser/extensions/extension_function_registry.h"
 
-#include <map>
-
-#include "base/json/json_string_value_serializer.h"
-#include "base/memory/ref_counted.h"
-#include "base/memory/singleton.h"
-#include "base/process_util.h"
-#include "base/values.h"
-#include "build/build_config.h"
 #include "chrome/browser/accessibility/accessibility_extension_api.h"
 #include "chrome/browser/bookmarks/bookmark_extension_api.h"
 #include "chrome/browser/bookmarks/bookmark_manager_extension_api.h"
@@ -19,7 +11,6 @@
 #include "chrome/browser/extensions/api/app/app_api.h"
 #include "chrome/browser/extensions/api/browsing_data/browsing_data_api.h"
 #include "chrome/browser/extensions/api/declarative/declarative_api.h"
-#include "chrome/browser/extensions/api/dns/dns_api.h"
 #include "chrome/browser/extensions/api/extension_action/extension_browser_actions_api.h"
 #include "chrome/browser/extensions/api/extension_action/extension_page_actions_api.h"
 #include "chrome/browser/extensions/api/permissions/permissions_api.h"
@@ -27,14 +18,12 @@
 #include "chrome/browser/extensions/api/socket/socket_api.h"
 #include "chrome/browser/extensions/api/web_request/web_request_api.h"
 #include "chrome/browser/extensions/execute_code_in_tab_function.h"
-#include "chrome/browser/extensions/extension_activity_log.h"
 #include "chrome/browser/extensions/extension_chrome_auth_private_api.h"
 #include "chrome/browser/extensions/extension_content_settings_api.h"
 #include "chrome/browser/extensions/extension_context_menu_api.h"
 #include "chrome/browser/extensions/extension_cookies_api.h"
 #include "chrome/browser/extensions/extension_debugger_api.h"
 #include "chrome/browser/extensions/extension_font_settings_api.h"
-#include "chrome/browser/extensions/extension_function.h"
 #include "chrome/browser/extensions/extension_i18n_api.h"
 #include "chrome/browser/extensions/extension_idle_api.h"
 #include "chrome/browser/extensions/extension_managed_mode_api.h"
@@ -45,39 +34,21 @@
 #include "chrome/browser/extensions/extension_page_capture_api.h"
 #include "chrome/browser/extensions/extension_preference_api.h"
 #include "chrome/browser/extensions/extension_processes_api.h"
-#include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_tabs_module.h"
 #include "chrome/browser/extensions/extension_test_api.h"
 #include "chrome/browser/extensions/extension_tts_api.h"
 #include "chrome/browser/extensions/extension_tts_engine_api.h"
 #include "chrome/browser/extensions/extension_web_socket_proxy_private_api.h"
-#include "chrome/browser/extensions/extension_web_ui.h"
 #include "chrome/browser/extensions/extension_webnavigation_api.h"
 #include "chrome/browser/extensions/extension_webstore_private_api.h"
-#include "chrome/browser/extensions/extensions_quota_service.h"
-#include "chrome/browser/extensions/process_map.h"
 #include "chrome/browser/extensions/settings/settings_api.h"
 #include "chrome/browser/extensions/system/system_api.h"
-#include "chrome/browser/external_protocol/external_protocol_handler.h"
 #include "chrome/browser/history/history_extension_api.h"
 #include "chrome/browser/history/top_sites_extension_api.h"
 #include "chrome/browser/infobars/infobar_extension_api.h"
-#include "chrome/browser/profiles/profile.h"
-#include "chrome/browser/renderer_host/chrome_render_message_filter.h"
 #include "chrome/browser/rlz/rlz_extension_api.h"
 #include "chrome/browser/speech/speech_input_extension_api.h"
-#include "chrome/browser/ui/browser_list.h"
-#include "chrome/browser/ui/browser_window.h"
-#include "chrome/common/extensions/api/extension_api.h"
-#include "chrome/common/extensions/extension_messages.h"
-#include "chrome/common/extensions/extension_set.h"
-#include "chrome/common/url_constants.h"
-#include "content/public/browser/render_process_host.h"
-#include "content/public/browser/render_view_host.h"
-#include "ipc/ipc_message.h"
-#include "ipc/ipc_message_macros.h"
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebSecurityOrigin.h"
-#include "third_party/skia/include/core/SkBitmap.h"
+#include "chrome/common/extensions/api/generated_api.h"
 
 #if defined(TOOLKIT_VIEWS)
 #include "chrome/browser/extensions/extension_input_api.h"
@@ -96,56 +67,19 @@
 #include "chrome/browser/extensions/extension_input_method_api.h"
 #endif
 
-using extensions::ExtensionAPI;
-using content::RenderViewHost;
-using WebKit::WebSecurityOrigin;
-
-// FactoryRegistry -------------------------------------------------------------
-
-namespace {
-
-// Template for defining ExtensionFunctionFactory.
-template<class T>
-ExtensionFunction* NewExtensionFunction() {
-  return new T();
+// static
+ExtensionFunctionRegistry* ExtensionFunctionRegistry::GetInstance() {
+  return Singleton<ExtensionFunctionRegistry>::get();
 }
 
-// Contains a list of all known extension functions and allows clients to
-// create instances of them.
-class FactoryRegistry {
- public:
-  static FactoryRegistry* GetInstance();
-  FactoryRegistry() { ResetFunctions(); }
-
-  // Resets all functions to their default values.
-  void ResetFunctions();
-
-  // Adds all function names to 'names'.
-  void GetAllNames(std::vector<std::string>* names);
-
-  // Allows overriding of specific functions (e.g. for testing).  Functions
-  // must be previously registered.  Returns true if successful.
-  bool OverrideFunction(const std::string& name,
-                        ExtensionFunctionFactory factory);
-
-  // Factory method for the ExtensionFunction registered as 'name'.
-  ExtensionFunction* NewFunction(const std::string& name);
-
- private:
-  template<class T>
-  void RegisterFunction() {
-    factories_[T::function_name()] = &NewExtensionFunction<T>;
-  }
-
-  typedef std::map<std::string, ExtensionFunctionFactory> FactoryMap;
-  FactoryMap factories_;
-};
+ExtensionFunctionRegistry::ExtensionFunctionRegistry() {
+  ResetFunctions();
+}
 
-FactoryRegistry* FactoryRegistry::GetInstance() {
-  return Singleton<FactoryRegistry>::get();
+ExtensionFunctionRegistry::~ExtensionFunctionRegistry() {
 }
 
-void FactoryRegistry::ResetFunctions() {
+void ExtensionFunctionRegistry::ResetFunctions() {
   // Register all functions here.
 
   // Windows
@@ -523,9 +457,6 @@ void FactoryRegistry::ResetFunctions() {
   RegisterFunction<extensions::SocketReadFunction>();
   RegisterFunction<extensions::SocketWriteFunction>();
 
-  // DNS
-  RegisterFunction<extensions::DNSResolveFunction>();
-
   // System
   RegisterFunction<extensions::GetIncognitoModeAvailabilityFunction>();
   RegisterFunction<extensions::GetUpdateStatusFunction>();
@@ -534,17 +465,21 @@ void FactoryRegistry::ResetFunctions() {
   RegisterFunction<extensions::AddRulesFunction>();
   RegisterFunction<extensions::RemoveRulesFunction>();
   RegisterFunction<extensions::GetRulesFunction>();
+
+  // Generated APIs
+  extensions::api::GeneratedFunctionRegistry::RegisterAll(this);
 }
 
-void FactoryRegistry::GetAllNames(std::vector<std::string>* names) {
+void ExtensionFunctionRegistry::GetAllNames(std::vector<std::string>* names) {
   for (FactoryMap::iterator iter = factories_.begin();
        iter != factories_.end(); ++iter) {
     names->push_back(iter->first);
   }
 }
 
-bool FactoryRegistry::OverrideFunction(const std::string& name,
-                                       ExtensionFunctionFactory factory) {
+bool ExtensionFunctionRegistry::OverrideFunction(
+    const std::string& name,
+    ExtensionFunctionFactory factory) {
   FactoryMap::iterator iter = factories_.find(name);
   if (iter == factories_.end()) {
     return false;
@@ -554,261 +489,11 @@ bool FactoryRegistry::OverrideFunction(const std::string& name,
   }
 }
 
-ExtensionFunction* FactoryRegistry::NewFunction(const std::string& name) {
+ExtensionFunction* ExtensionFunctionRegistry::NewFunction(
+    const std::string& name) {
   FactoryMap::iterator iter = factories_.find(name);
   DCHECK(iter != factories_.end());
   ExtensionFunction* function = iter->second();
   function->set_name(name);
   return function;
 }
-
-const char kAccessDenied[] = "access denied";
-const char kQuotaExceeded[] = "quota exceeded";
-
-void LogSuccess(const Extension* extension,
-                const ExtensionHostMsg_Request_Params& params) {
-  ExtensionActivityLog* extension_activity_log =
-      ExtensionActivityLog::GetInstance();
-  if (extension_activity_log->HasObservers(extension)) {
-    std::string call_signature = params.name + "(";
-    ListValue::const_iterator it = params.arguments.begin();
-    for (; it != params.arguments.end(); ++it) {
-      std::string arg;
-      JSONStringValueSerializer serializer(&arg);
-      if (serializer.SerializeAndOmitBinaryValues(**it)) {
-        if (it != params.arguments.begin())
-          call_signature += ", ";
-        call_signature += arg;
-      }
-    }
-    call_signature += ")";
-
-    extension_activity_log->Log(
-        extension,
-        ExtensionActivityLog::ACTIVITY_EXTENSION_API_CALL,
-        call_signature);
-  }
-}
-
-void LogFailure(const Extension* extension,
-                const std::string& func_name,
-                const char* reason) {
-  ExtensionActivityLog* extension_activity_log =
-      ExtensionActivityLog::GetInstance();
-  if (extension_activity_log->HasObservers(extension)) {
-    extension_activity_log->Log(
-        extension,
-        ExtensionActivityLog::ACTIVITY_EXTENSION_API_BLOCK,
-        func_name + ": " + reason);
-  }
-}
-
-};  // namespace
-
-// ExtensionFunctionDispatcher -------------------------------------------------
-
-void ExtensionFunctionDispatcher::GetAllFunctionNames(
-    std::vector<std::string>* names) {
-  FactoryRegistry::GetInstance()->GetAllNames(names);
-}
-
-bool ExtensionFunctionDispatcher::OverrideFunction(
-    const std::string& name, ExtensionFunctionFactory factory) {
-  return FactoryRegistry::GetInstance()->OverrideFunction(name, factory);
-}
-
-void ExtensionFunctionDispatcher::ResetFunctions() {
-  FactoryRegistry::GetInstance()->ResetFunctions();
-}
-
-// static
-void ExtensionFunctionDispatcher::DispatchOnIOThread(
-    ExtensionInfoMap* extension_info_map,
-    void* profile,
-    int render_process_id,
-    base::WeakPtr<ChromeRenderMessageFilter> ipc_sender,
-    int routing_id,
-    const ExtensionHostMsg_Request_Params& params) {
-  const Extension* extension =
-      extension_info_map->extensions().GetByID(params.extension_id);
-
-  scoped_refptr<ExtensionFunction> function(
-      CreateExtensionFunction(params, extension, render_process_id,
-                              extension_info_map->process_map(), profile,
-                              ipc_sender, routing_id));
-  if (!function) {
-    LogFailure(extension, params.name, kAccessDenied);
-    return;
-  }
-
-  IOThreadExtensionFunction* function_io =
-      function->AsIOThreadExtensionFunction();
-  if (!function_io) {
-    NOTREACHED();
-    return;
-  }
-  function_io->set_ipc_sender(ipc_sender, routing_id);
-  function_io->set_extension_info_map(extension_info_map);
-  function->set_include_incognito(
-      extension_info_map->IsIncognitoEnabled(extension->id()));
-
-  ExtensionsQuotaService* quota = extension_info_map->quota_service();
-  if (quota->Assess(extension->id(), function, &params.arguments,
-                    base::TimeTicks::Now())) {
-    function->Run();
-    LogSuccess(extension, params);
-  } else {
-    function->OnQuotaExceeded();
-    LogFailure(extension, params.name, kQuotaExceeded);
-  }
-}
-
-ExtensionFunctionDispatcher::ExtensionFunctionDispatcher(Profile* profile,
-                                                         Delegate* delegate)
-  : profile_(profile),
-    delegate_(delegate) {
-}
-
-ExtensionFunctionDispatcher::~ExtensionFunctionDispatcher() {
-}
-
-Browser* ExtensionFunctionDispatcher::GetCurrentBrowser(
-    RenderViewHost* render_view_host, bool include_incognito) {
-  Browser* browser = delegate_->GetBrowser();
-
-  // If the delegate has an associated browser, that is always the right answer.
-  if (browser)
-    return browser;
-
-  // Otherwise, try to default to a reasonable browser. If |include_incognito|
-  // is true, we will also search browsers in the incognito version of this
-  // profile. Note that the profile may already be incognito, in which case
-  // we will search the incognito version only, regardless of the value of
-  // |include_incognito|.
-  Profile* profile = Profile::FromBrowserContext(
-      render_view_host->GetProcess()->GetBrowserContext());
-  browser = BrowserList::FindAnyBrowser(profile, include_incognito);
-
-  // NOTE(rafaelw): This can return NULL in some circumstances. In particular,
-  // a background_page onload chrome.tabs api call can make it into here
-  // before the browser is sufficiently initialized to return here.
-  // A similar situation may arise during shutdown.
-  // TODO(rafaelw): Delay creation of background_page until the browser
-  // is available. http://code.google.com/p/chromium/issues/detail?id=13284
-  return browser;
-}
-
-void ExtensionFunctionDispatcher::Dispatch(
-    const ExtensionHostMsg_Request_Params& params,
-    RenderViewHost* render_view_host) {
-  ExtensionService* service = profile()->GetExtensionService();
-  extensions::ProcessMap* process_map = service->process_map();
-  if (!service || !process_map)
-    return;
-
-  const Extension* extension = service->extensions()->GetByID(
-      params.extension_id);
-  if (!extension)
-    extension = service->extensions()->GetHostedAppByURL(ExtensionURLInfo(
-        WebSecurityOrigin::createFromString(params.source_origin),
-        params.source_url));
-
-  scoped_refptr<ExtensionFunction> function(
-      CreateExtensionFunction(params, extension,
-                              render_view_host->GetProcess()->GetID(),
-                              *(service->process_map()),
-                              profile(), render_view_host,
-                              render_view_host->GetRoutingID()));
-  if (!function) {
-    LogFailure(extension, params.name, kAccessDenied);
-    return;
-  }
-
-  UIThreadExtensionFunction* function_ui =
-      function->AsUIThreadExtensionFunction();
-  if (!function_ui) {
-    NOTREACHED();
-    return;
-  }
-  function_ui->SetRenderViewHost(render_view_host);
-  function_ui->set_dispatcher(AsWeakPtr());
-  function_ui->set_profile(profile_);
-  function->set_include_incognito(service->CanCrossIncognito(extension));
-
-  ExtensionsQuotaService* quota = service->quota_service();
-  if (quota->Assess(extension->id(), function, &params.arguments,
-                    base::TimeTicks::Now())) {
-    // See crbug.com/39178.
-    ExternalProtocolHandler::PermitLaunchUrl();
-
-    function->Run();
-    LogSuccess(extension, params);
-  } else {
-    function->OnQuotaExceeded();
-    LogFailure(extension, params.name, kQuotaExceeded);
-  }
-
-  // We only adjust the keepalive count for UIThreadExtensionFunction for
-  // now, largely for simplicity's sake. This is OK because currently, only
-  // the webRequest API uses IOThreadExtensionFunction, and that API is not
-  // compatible with lazy background pages.
-  profile()->GetExtensionProcessManager()->IncrementLazyKeepaliveCount(
-      extension);
-}
-
-void ExtensionFunctionDispatcher::OnExtensionFunctionCompleted(
-    const Extension* extension) {
-  profile()->GetExtensionProcessManager()->DecrementLazyKeepaliveCount(
-      extension);
-}
-
-// static
-ExtensionFunction* ExtensionFunctionDispatcher::CreateExtensionFunction(
-    const ExtensionHostMsg_Request_Params& params,
-    const Extension* extension,
-    int requesting_process_id,
-    const extensions::ProcessMap& process_map,
-    void* profile,
-    IPC::Message::Sender* ipc_sender,
-    int routing_id) {
-  if (!extension) {
-    LOG(ERROR) << "Specified extension does not exist.";
-    SendAccessDenied(ipc_sender, routing_id, params.request_id);
-    return NULL;
-  }
-
-  if (ExtensionAPI::GetInstance()->IsPrivileged(params.name) &&
-      !process_map.Contains(extension->id(), requesting_process_id)) {
-    LOG(ERROR) << "Extension API called from incorrect process "
-               << requesting_process_id
-               << " from URL " << params.source_url.spec();
-    SendAccessDenied(ipc_sender, routing_id, params.request_id);
-    return NULL;
-  }
-
-  if (!extension->HasAPIPermission(params.name)) {
-    LOG(ERROR) << "Extension " << extension->id() << " does not have "
-               << "permission to function: " << params.name;
-    SendAccessDenied(ipc_sender, routing_id, params.request_id);
-    return NULL;
-  }
-
-  ExtensionFunction* function =
-      FactoryRegistry::GetInstance()->NewFunction(params.name);
-  function->SetArgs(&params.arguments);
-  function->set_source_url(params.source_url);
-  function->set_request_id(params.request_id);
-  function->set_has_callback(params.has_callback);
-  function->set_user_gesture(params.user_gesture);
-  function->set_extension(extension);
-  function->set_profile_id(profile);
-  return function;
-}
-
-// static
-void ExtensionFunctionDispatcher::SendAccessDenied(
-    IPC::Message::Sender* ipc_sender, int routing_id, int request_id) {
-  ipc_sender->Send(new ExtensionMsg_Response(
-      routing_id, request_id, false, std::string(),
-      "Access to extension API denied."));
-}