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

chrome/browser/extensions/extension_tab_helper.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_tab_helper.h"
 
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/webstore_inline_installer.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/sessions/restore_tab_helper.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper_delegate.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/extensions/extension_action.h"
 #include "chrome/common/extensions/extension_constants.h"
 #include "chrome/common/extensions/extension_icon_set.h"
 #include "chrome/common/extensions/extension_messages.h"
 #include "chrome/common/extensions/extension_resource.h"
+#include "content/browser/renderer_host/render_view_host.h"
 #include "content/browser/renderer_host/render_widget_host_view.h"
 #include "content/browser/tab_contents/navigation_details.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "content/common/notification_service.h"
 
 ExtensionTabHelper::ExtensionTabHelper(TabContentsWrapper* wrapper)
     : TabContentsObserver(wrapper->tab_contents()),
       extension_app_(NULL),
       ALLOW_THIS_IN_INITIALIZER_LIST(
           extension_function_dispatcher_(wrapper->profile(), this)),
@@ skipping 89 matching lines @@
 
 bool ExtensionTabHelper::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(ExtensionTabHelper, message)
     IPC_MESSAGE_HANDLER(ExtensionHostMsg_DidGetApplicationInfo,
                         OnDidGetApplicationInfo)
     IPC_MESSAGE_HANDLER(ExtensionHostMsg_InstallApplication,
                         OnInstallApplication)
     IPC_MESSAGE_HANDLER(ExtensionHostMsg_InlineWebstoreInstall,
                         OnInlineWebstoreInstall)
+    IPC_MESSAGE_HANDLER(ExtensionHostMsg_GetAppNotifyChannel,
+                        OnGetAppNotifyChannel)
     IPC_MESSAGE_HANDLER(ExtensionHostMsg_Request, OnRequest)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void ExtensionTabHelper::OnDidGetApplicationInfo(
     int32 page_id, const WebApplicationInfo& info) {
   web_app_info_ = info;
 
   if (wrapper_->delegate())
     wrapper_->delegate()->OnDidGetApplicationInfo(wrapper_, page_id);
 }
 
 void ExtensionTabHelper::OnInstallApplication(const WebApplicationInfo& info) {
   if (wrapper_->delegate())
     wrapper_->delegate()->OnInstallApplication(wrapper_, info);
 }
 
 void ExtensionTabHelper::OnInlineWebstoreInstall(
     int install_id,
     const std::string& webstore_item_id,
     const GURL& requestor_url) {
   scoped_refptr<WebstoreInlineInstaller> installer(new WebstoreInlineInstaller(
       tab_contents(), install_id, webstore_item_id, requestor_url, this));
   installer->BeginInstall();
 }
 
+void ExtensionTabHelper::OnGetAppNotifyChannel(
+    int request_id,
+    const GURL& requestor_url,
+    const std::string& client_id) {
+
+  // Check for permission first.
+  Profile* profile =
+      Profile::FromBrowserContext(tab_contents()->browser_context());
+  ExtensionService* extension_service = profile->GetExtensionService();
+  ExtensionProcessManager* process_manager =
+      profile->GetExtensionProcessManager();
+  RenderProcessHost* process =
+      tab_contents_wrapper()->render_view_host()->process();
+  const Extension* extension =
+      extension_service->GetInstalledApp(requestor_url);

[Aaron Boodman, 2011/10/06 20:29:28]

Are you sure this works correctly for hosted apps? It seems like allowed will be
false for all hosted apps (the process check will fail).

[asargent_no_longer_on_chrome, 2011/10/06 22:37:38]

The process check below is:

(extension->is_hosted_app() ||
 process_manager->GetExtensionProcess(requestor_url) == process)

So we don't do the process check for packaged apps. I guess I could rewrite it
to be something like:

 bool allowed =
  ...
  <all the other stuff except the process check>
  ...;

 if (allowed && extension->is_packaged_app()) {
    RenderProcessHost* expected_process =
        process_manager->GetExtensionProcess(requestor_url);
   allowed = (expected_process == process);
 }

Let me know if you think it reads better this way. 

[Aaron Boodman, 2011/10/06 22:40:26]

I see, ok.

+  bool allowed =
+      extension &&
+      extension->is_app() &&
+      extension->HasAPIPermission(
+          ExtensionAPIPermission::kExperimental) &&
+      (extension->is_hosted_app() ||
+       process_manager->GetExtensionProcess(requestor_url) == process);
+  if (!allowed) {
+    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 ExtensionTabHelper::AppNotifyChannelSetupComplete(
+    int request_id, const std::string& client_id, const std::string& error) {
+  Send(new ExtensionMsg_GetAppNotifyChannelResponse(
+      routing_id(), request_id, client_id, error));
+}
+
 void ExtensionTabHelper::OnRequest(
     const ExtensionHostMsg_Request_Params& request) {
   extension_function_dispatcher_.Dispatch(request,
                                           tab_contents()->render_view_host());
 }
 
 void ExtensionTabHelper::UpdateExtensionAppIcon(const Extension* extension) {
   extension_app_icon_.reset();
 
   if (extension) {
@@ skipping 47 matching lines @@
 }
 
 TabContents* ExtensionTabHelper::GetAssociatedTabContents() const {
   return tab_contents();
 }
 
 gfx::NativeView ExtensionTabHelper::GetNativeViewOfHost() {
   RenderWidgetHostView* rwhv = tab_contents()->GetRenderWidgetHostView();
   return rwhv ? rwhv->GetNativeView() : NULL;
 }