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 "chrome/browser/extensions/extension_gcm_app_handler.h" 6 7#include "base/bind.h" 8#include "base/lazy_instance.h" 9#include "base/location.h" 10#include "chrome/browser/chrome_notification_types.h" 11#include "chrome/browser/profiles/profile.h" 12#include "chrome/browser/services/gcm/gcm_profile_service.h" 13#include "chrome/browser/services/gcm/gcm_profile_service_factory.h" 14#include "components/gcm_driver/gcm_driver.h" 15#include "extensions/browser/extension_registry.h" 16#include "extensions/browser/extension_system.h" 17#include "extensions/common/extension.h" 18#include "extensions/common/permissions/permissions_data.h" 19 20#if !defined(OS_ANDROID) 21#include "chrome/browser/extensions/api/gcm/gcm_api.h" 22#endif 23 24namespace extensions { 25 26namespace { 27 28const char kDummyAppId[] = "extension.guard.dummy.id"; 29 30base::LazyInstance<BrowserContextKeyedAPIFactory<ExtensionGCMAppHandler> > 31 g_factory = LAZY_INSTANCE_INITIALIZER; 32 33bool IsGCMPermissionEnabled(const Extension* extension) { 34 return extension->permissions_data()->HasAPIPermission(APIPermission::kGcm); 35} 36 37} // namespace 38 39 40// static 41BrowserContextKeyedAPIFactory<ExtensionGCMAppHandler>* 42ExtensionGCMAppHandler::GetFactoryInstance() { 43 return g_factory.Pointer(); 44} 45 46ExtensionGCMAppHandler::ExtensionGCMAppHandler(content::BrowserContext* context) 47 : profile_(Profile::FromBrowserContext(context)), 48 extension_registry_observer_(this), 49 weak_factory_(this) { 50 extension_registry_observer_.Add(ExtensionRegistry::Get(profile_)); 51#if !defined(OS_ANDROID) 52 js_event_router_.reset(new extensions::GcmJsEventRouter(profile_)); 53#endif 54} 55 56ExtensionGCMAppHandler::~ExtensionGCMAppHandler() { 57 const ExtensionSet& enabled_extensions = 58 ExtensionRegistry::Get(profile_)->enabled_extensions(); 59 for (ExtensionSet::const_iterator extension = enabled_extensions.begin(); 60 extension != enabled_extensions.end(); 61 ++extension) { 62 if (IsGCMPermissionEnabled(extension->get())) 63 GetGCMDriver()->RemoveAppHandler((*extension)->id()); 64 } 65} 66 67void ExtensionGCMAppHandler::ShutdownHandler() { 68#if !defined(OS_ANDROID) 69 js_event_router_.reset(); 70#endif 71} 72 73void ExtensionGCMAppHandler::OnMessage( 74 const std::string& app_id, 75 const gcm::GCMClient::IncomingMessage& message) { 76#if !defined(OS_ANDROID) 77 js_event_router_->OnMessage(app_id, message); 78#endif 79} 80 81void ExtensionGCMAppHandler::OnMessagesDeleted(const std::string& app_id) { 82#if !defined(OS_ANDROID) 83 js_event_router_->OnMessagesDeleted(app_id); 84#endif 85} 86 87void ExtensionGCMAppHandler::OnSendError( 88 const std::string& app_id, 89 const gcm::GCMClient::SendErrorDetails& send_error_details) { 90#if !defined(OS_ANDROID) 91 js_event_router_->OnSendError(app_id, send_error_details); 92#endif 93} 94 95void ExtensionGCMAppHandler::OnExtensionLoaded( 96 content::BrowserContext* browser_context, 97 const Extension* extension) { 98 if (IsGCMPermissionEnabled(extension)) 99 AddAppHandler(extension->id()); 100} 101 102void ExtensionGCMAppHandler::OnExtensionUnloaded( 103 content::BrowserContext* browser_context, 104 const Extension* extension, 105 UnloadedExtensionInfo::Reason reason) { 106 if (!IsGCMPermissionEnabled(extension)) 107 return; 108 109 if (reason == UnloadedExtensionInfo::REASON_UPDATE && 110 GetGCMDriver()->app_handlers().size() == 1) { 111 // When the extension is being updated, it will be first unloaded and then 112 // loaded again by ExtensionService::AddExtension. If the app handler for 113 // this extension is the only handler, removing it and adding it again will 114 // cause the GCM service being stopped and restarted unnecessarily. To work 115 // around this, we add a dummy app handler to guard against it. This dummy 116 // app handler will be removed once the extension loading logic is done. 117 // 118 // Also note that the GCM message routing will not be interruptted during 119 // the update process since unloading and reloading extension are done in 120 // the single function ExtensionService::AddExtension. 121 AddDummyAppHandler(); 122 123 base::MessageLoop::current()->PostTask( 124 FROM_HERE, 125 base::Bind(&ExtensionGCMAppHandler::RemoveDummyAppHandler, 126 weak_factory_.GetWeakPtr())); 127 } 128 129 RemoveAppHandler(extension->id()); 130} 131 132void ExtensionGCMAppHandler::OnExtensionUninstalled( 133 content::BrowserContext* browser_context, 134 const Extension* extension) { 135 if (IsGCMPermissionEnabled(extension)) { 136 GetGCMDriver()->Unregister( 137 extension->id(), 138 base::Bind(&ExtensionGCMAppHandler::OnUnregisterCompleted, 139 weak_factory_.GetWeakPtr(), 140 extension->id())); 141 RemoveAppHandler(extension->id()); 142 } 143} 144 145void ExtensionGCMAppHandler::AddDummyAppHandler() { 146 AddAppHandler(kDummyAppId); 147} 148 149void ExtensionGCMAppHandler::RemoveDummyAppHandler() { 150 RemoveAppHandler(kDummyAppId); 151} 152 153gcm::GCMDriver* ExtensionGCMAppHandler::GetGCMDriver() const { 154 return gcm::GCMProfileServiceFactory::GetForProfile(profile_)->driver(); 155} 156 157void ExtensionGCMAppHandler::OnUnregisterCompleted( 158 const std::string& app_id, gcm::GCMClient::Result result) { 159 // Nothing to do. 160} 161 162void ExtensionGCMAppHandler::AddAppHandler(const std::string& app_id) { 163 GetGCMDriver()->AddAppHandler(app_id, this); 164} 165 166void ExtensionGCMAppHandler::RemoveAppHandler(const std::string& app_id) { 167 GetGCMDriver()->RemoveAppHandler(app_id); 168} 169 170} // namespace extensions 171