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
24 namespace extensions {
25
26 namespace {
27
28 const char kDummyAppId[] = "extension.guard.dummy.id";
29
30 base::LazyInstance<BrowserContextKeyedAPIFactory<ExtensionGCMAppHandler> >
31 g_factory = LAZY_INSTANCE_INITIALIZER;
32
IsGCMPermissionEnabled(const Extension * extension)33 bool IsGCMPermissionEnabled(const Extension* extension) {
34 return extension->permissions_data()->HasAPIPermission(APIPermission::kGcm);
35 }
36
37 } // namespace
38
39
40 // static
41 BrowserContextKeyedAPIFactory<ExtensionGCMAppHandler>*
GetFactoryInstance()42 ExtensionGCMAppHandler::GetFactoryInstance() {
43 return g_factory.Pointer();
44 }
45
ExtensionGCMAppHandler(content::BrowserContext * context)46 ExtensionGCMAppHandler::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
~ExtensionGCMAppHandler()56 ExtensionGCMAppHandler::~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
ShutdownHandler()67 void ExtensionGCMAppHandler::ShutdownHandler() {
68 #if !defined(OS_ANDROID)
69 js_event_router_.reset();
70 #endif
71 }
72
OnMessage(const std::string & app_id,const gcm::GCMClient::IncomingMessage & message)73 void 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
OnMessagesDeleted(const std::string & app_id)81 void ExtensionGCMAppHandler::OnMessagesDeleted(const std::string& app_id) {
82 #if !defined(OS_ANDROID)
83 js_event_router_->OnMessagesDeleted(app_id);
84 #endif
85 }
86
OnSendError(const std::string & app_id,const gcm::GCMClient::SendErrorDetails & send_error_details)87 void 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
OnExtensionLoaded(content::BrowserContext * browser_context,const Extension * extension)95 void ExtensionGCMAppHandler::OnExtensionLoaded(
96 content::BrowserContext* browser_context,
97 const Extension* extension) {
98 if (IsGCMPermissionEnabled(extension))
99 AddAppHandler(extension->id());
100 }
101
OnExtensionUnloaded(content::BrowserContext * browser_context,const Extension * extension,UnloadedExtensionInfo::Reason reason)102 void 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
OnExtensionUninstalled(content::BrowserContext * browser_context,const Extension * extension)132 void 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
AddDummyAppHandler()145 void ExtensionGCMAppHandler::AddDummyAppHandler() {
146 AddAppHandler(kDummyAppId);
147 }
148
RemoveDummyAppHandler()149 void ExtensionGCMAppHandler::RemoveDummyAppHandler() {
150 RemoveAppHandler(kDummyAppId);
151 }
152
GetGCMDriver() const153 gcm::GCMDriver* ExtensionGCMAppHandler::GetGCMDriver() const {
154 return gcm::GCMProfileServiceFactory::GetForProfile(profile_)->driver();
155 }
156
OnUnregisterCompleted(const std::string & app_id,gcm::GCMClient::Result result)157 void ExtensionGCMAppHandler::OnUnregisterCompleted(
158 const std::string& app_id, gcm::GCMClient::Result result) {
159 // Nothing to do.
160 }
161
AddAppHandler(const std::string & app_id)162 void ExtensionGCMAppHandler::AddAppHandler(const std::string& app_id) {
163 GetGCMDriver()->AddAppHandler(app_id, this);
164 }
165
RemoveAppHandler(const std::string & app_id)166 void ExtensionGCMAppHandler::RemoveAppHandler(const std::string& app_id) {
167 GetGCMDriver()->RemoveAppHandler(app_id);
168 }
169
170 } // namespace extensions
171