Add js api for hosted/pacakged apps to request notification authorization

chrome/browser/extensions/extension_message_handler.cc

 // 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 "chrome/browser/extensions/extension_message_handler.h"
 
+#include "chrome/browser/extensions/app_notify_channel_setup.h"
 #include "chrome/browser/extensions/extension_message_service.h"
+#include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/extensions/extension_messages.h"
 #include "content/browser/child_process_security_policy.h"
 #include "content/browser/renderer_host/render_process_host.h"
 #include "content/browser/renderer_host/render_view_host.h"
 #include "content/browser/renderer_host/render_view_host_delegate.h"
 
 ExtensionMessageHandler::ExtensionMessageHandler(
     RenderViewHost* render_view_host)
     : RenderViewHostObserver(render_view_host) {
 }
 
 ExtensionMessageHandler::~ExtensionMessageHandler() {
 }
 
 bool ExtensionMessageHandler::OnMessageReceived(
     const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(ExtensionMessageHandler, message)
     IPC_MESSAGE_HANDLER(ExtensionHostMsg_PostMessage, OnPostMessage)
+    IPC_MESSAGE_HANDLER(ExtensionHostMsg_GetAppNotifyChannel,
+                        OnGetAppNotifyChannel)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void ExtensionMessageHandler::RenderViewHostInitialized() {
   Send(new ExtensionMsg_NotifyRenderViewType(
       routing_id(), render_view_host()->delegate()->GetRenderViewType()));
 }
 
 void ExtensionMessageHandler::OnPostMessage(int port_id,
                                             const std::string& message) {
   Profile* profile = Profile::FromBrowserContext(
       render_view_host()->process()->browser_context());
   if (profile->GetExtensionMessageService()) {
     profile->GetExtensionMessageService()->PostMessageFromRenderer(
         port_id, message);
   }
 }
+
+void ExtensionMessageHandler::OnGetAppNotifyChannel(
+    int request_id,
+    const GURL& requestor_url,
+    const std::string& client_id) {
+
+  // Check for permission first.
+  Profile* profile = Profile::FromBrowserContext(
+      render_view_host()->process()->browser_context());
+  ExtensionService* extension_service = profile->GetExtensionService();
+  if (!extension_service || !extension_service->is_ready())

[Aaron Boodman, 2011/10/04 23:40:36]

I don't think that it is possible for either of these to be false.

[asargent_no_longer_on_chrome, 2011/10/06 03:35:32]

Ok, removed this. 

+    return;
+  const Extension* extension =
+      extension_service->GetInstalledApp(requestor_url);

[Aaron Boodman, 2011/10/04 23:40:36]

Can you also check that the app is running in the requesting process (via
ExtensionProcessManager) ?

[asargent_no_longer_on_chrome, 2011/10/06 03:35:32]

As discussed in IRC, I've added the process check for packaged apps but not
hosted ones. 

+  if (!extension ||
+      !extension->HasAPIPermission(ExtensionAPIPermission::kExperimental)) {
+    AppNotifyChannelSetupComplete(request_id, "", "permission_error");
+    return;
+  }
+
+  scoped_refptr<AppNotifyChannelSetup> channel_setup(
+      new AppNotifyChannelSetup(request_id,
+                                client_id,
+                                requestor_url,
+                                this->AsWeakPtr()));
+  channel_setup->Start();
+  // We'll get called back in AppNotifyChannelSetupComplete.
+}
+
+void ExtensionMessageHandler::AppNotifyChannelSetupComplete(
+    int request_id, const std::string& client_id, const std::string& error) {
+  Send(new ExtensionMsg_GetAppNotifyChannelResponse(
+      routing_id(), request_id, client_id, error));
+}