components/gcm_driver/gcm_driver_desktop.cc

   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 "components/gcm_driver/gcm_driver_desktop.h"
   6
   7 #include <utility>
   8
   9 #include "base/bind.h"
  10 #include "base/bind_helpers.h"
  11 #include "base/files/file_path.h"
  12 #include "base/location.h"
  13 #include "base/logging.h"
  14 #include "base/metrics/histogram_macros.h"
  15 #include "base/profiler/scoped_tracker.h"
  16 #include "base/sequenced_task_runner.h"
  17 #include "base/threading/sequenced_worker_pool.h"
  18 #include "components/gcm_driver/gcm_account_mapper.h"
  19 #include "components/gcm_driver/gcm_app_handler.h"
  20 #include "components/gcm_driver/gcm_channel_status_syncer.h"
  21 #include "components/gcm_driver/gcm_client_factory.h"
  22 #include "components/gcm_driver/gcm_delayed_task_controller.h"
  23 #include "components/gcm_driver/instance_id/instance_id_impl.h"
  24 #include "components/gcm_driver/system_encryptor.h"
  25 #include "google_apis/gcm/engine/account_mapping.h"
  26 #include "net/base/ip_endpoint.h"
  27 #include "net/url_request/url_request_context_getter.h"
  28
  29 #if defined(OS_CHROMEOS)
  30 #include "components/timers/alarm_timer_chromeos.h"
  31 #endif
  32
  33 namespace gcm {
  34
  35 class GCMDriverDesktop::IOWorker : public GCMClient::Delegate {
  36  public:
  37   // Called on UI thread.
  38   IOWorker(const scoped_refptr<base::SequencedTaskRunner>& ui_thread,
  39            const scoped_refptr<base::SequencedTaskRunner>& io_thread);
  40   virtual ~IOWorker();
  41
  42   // Overridden from GCMClient::Delegate:
  43   // Called on IO thread.
  44   void OnRegisterFinished(const linked_ptr<RegistrationInfo>& registration_info,
  45                           const std::string& registration_id,
  46                           GCMClient::Result result) override;
  47   void OnUnregisterFinished(
  48       const linked_ptr<RegistrationInfo>& registration_info,
  49       GCMClient::Result result) override;
  50   void OnSendFinished(const std::string& app_id,
  51                       const std::string& message_id,
  52                       GCMClient::Result result) override;
  53   void OnMessageReceived(const std::string& app_id,
  54                          const GCMClient::IncomingMessage& message) override;
  55   void OnMessagesDeleted(const std::string& app_id) override;
  56   void OnMessageSendError(
  57       const std::string& app_id,
  58       const GCMClient::SendErrorDetails& send_error_details) override;
  59   void OnSendAcknowledged(const std::string& app_id,
  60                           const std::string& message_id) override;
  61   void OnGCMReady(const std::vector<AccountMapping>& account_mappings,
  62                   const base::Time& last_token_fetch_time) override;
  63   void OnActivityRecorded() override;
  64   void OnConnected(const net::IPEndPoint& ip_endpoint) override;
  65   void OnDisconnected() override;
  66
  67   // Called on IO thread.
  68   void Initialize(
  69       scoped_ptr<GCMClientFactory> gcm_client_factory,
  70       const GCMClient::ChromeBuildInfo& chrome_build_info,
  71       const base::FilePath& store_path,
  72       const scoped_refptr<net::URLRequestContextGetter>& request_context,
  73       const scoped_refptr<base::SequencedTaskRunner> blocking_task_runner);
  74   void Start(GCMClient::StartMode start_mode,
  75              const base::WeakPtr<GCMDriverDesktop>& service);
  76   void Stop();
  77   void Register(const std::string& app_id,
  78                 const std::vector<std::string>& sender_ids);
  79   void Unregister(const std::string& app_id);
  80   void Send(const std::string& app_id,
  81             const std::string& receiver_id,
  82             const GCMClient::OutgoingMessage& message);
  83   void GetGCMStatistics(bool clear_logs);
  84   void SetGCMRecording(bool recording);
  85
  86   void SetAccountTokens(
  87       const std::vector<GCMClient::AccountTokenInfo>& account_tokens);
  88   void UpdateAccountMapping(const AccountMapping& account_mapping);
  89   void RemoveAccountMapping(const std::string& account_id);
  90   void SetLastTokenFetchTime(const base::Time& time);
  91   void WakeFromSuspendForHeartbeat(bool wake);
  92   void AddHeartbeatInterval(const std::string& scope, int interval_ms);
  93   void RemoveHeartbeatInterval(const std::string& scope);
  94
  95   void AddInstanceIDData(const std::string& app_id,
  96                          const std::string& instance_id,
  97                          const std::string& extra_data);
  98   void RemoveInstanceIDData(const std::string& app_id);
  99   void GetInstanceIDData(const std::string& app_id);
 100   void GetToken(const std::string& app_id,
 101                 const std::string& authorized_entity,
 102                 const std::string& scope,
 103                 const std::map<std::string, std::string>& options);
 104   void DeleteToken(const std::string& app_id,
 105                    const std::string& authorized_entity,
 106                    const std::string& scope);
 107
 108   // For testing purpose. Can be called from UI thread. Use with care.
 109   GCMClient* gcm_client_for_testing() const { return gcm_client_.get(); }
 110
 111  private:
 112   scoped_refptr<base::SequencedTaskRunner> ui_thread_;
 113   scoped_refptr<base::SequencedTaskRunner> io_thread_;
 114
 115   base::WeakPtr<GCMDriverDesktop> service_;
 116
 117   scoped_ptr<GCMClient> gcm_client_;
 118
 119   DISALLOW_COPY_AND_ASSIGN(IOWorker);
 120 };
 121
 122 GCMDriverDesktop::IOWorker::IOWorker(
 123     const scoped_refptr<base::SequencedTaskRunner>& ui_thread,
 124     const scoped_refptr<base::SequencedTaskRunner>& io_thread)
 125     : ui_thread_(ui_thread),
 126       io_thread_(io_thread) {
 127   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 128 }
 129
 130 GCMDriverDesktop::IOWorker::~IOWorker() {
 131   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 132 }
 133
 134 void GCMDriverDesktop::IOWorker::Initialize(
 135     scoped_ptr<GCMClientFactory> gcm_client_factory,
 136     const GCMClient::ChromeBuildInfo& chrome_build_info,
 137     const base::FilePath& store_path,
 138     const scoped_refptr<net::URLRequestContextGetter>& request_context,
 139     const scoped_refptr<base::SequencedTaskRunner> blocking_task_runner) {
 140   // TODO(pkasting): Remove ScopedTracker below once crbug.com/477117 is fixed.
 141   tracked_objects::ScopedTracker tracking_profile(
 142       FROM_HERE_WITH_EXPLICIT_FUNCTION(
 143           "477117 GCMDriverDesktop::IOWorker::Initialize"));
 144   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 145
 146   gcm_client_ = gcm_client_factory->BuildInstance();
 147
 148   gcm_client_->Initialize(chrome_build_info,
 149                           store_path,
 150                           blocking_task_runner,
 151                           request_context,
 152                           make_scoped_ptr<Encryptor>(new SystemEncryptor),
 153                           this);
 154 }
 155
 156 void GCMDriverDesktop::IOWorker::OnRegisterFinished(
 157     const linked_ptr<RegistrationInfo>& registration_info,
 158     const std::string& registration_id,
 159     GCMClient::Result result) {
 160   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 161
 162   const GCMRegistrationInfo* gcm_registration_info =
 163       GCMRegistrationInfo::FromRegistrationInfo(registration_info.get());
 164   if (gcm_registration_info) {
 165     ui_thread_->PostTask(
 166         FROM_HERE,
 167         base::Bind(&GCMDriverDesktop::RegisterFinished,
 168                    service_,
 169                    gcm_registration_info->app_id,
 170                    registration_id,
 171                    result));
 172   }
 173
 174   const InstanceIDTokenInfo* instance_id_token_info =
 175       InstanceIDTokenInfo::FromRegistrationInfo(registration_info.get());
 176   if (instance_id_token_info) {
 177     ui_thread_->PostTask(
 178       FROM_HERE,
 179       base::Bind(&GCMDriverDesktop::GetTokenFinished,
 180                  service_,
 181                  instance_id_token_info->app_id,
 182                  instance_id_token_info->authorized_entity,
 183                  instance_id_token_info->scope,
 184                  registration_id,
 185                  result));
 186   }
 187 }
 188
 189 void GCMDriverDesktop::IOWorker::OnUnregisterFinished(
 190     const linked_ptr<RegistrationInfo>& registration_info,
 191     GCMClient::Result result) {
 192   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 193
 194   const GCMRegistrationInfo* gcm_registration_info =
 195       GCMRegistrationInfo::FromRegistrationInfo(registration_info.get());
 196   if (gcm_registration_info) {
 197     ui_thread_->PostTask(
 198         FROM_HERE,
 199         base::Bind(&GCMDriverDesktop::UnregisterFinished,
 200                    service_,
 201                    gcm_registration_info->app_id,
 202                    result));
 203   }
 204
 205   const InstanceIDTokenInfo* instance_id_token_info =
 206       InstanceIDTokenInfo::FromRegistrationInfo(registration_info.get());
 207   if (instance_id_token_info) {
 208     ui_thread_->PostTask(
 209       FROM_HERE,
 210       base::Bind(&GCMDriverDesktop::DeleteTokenFinished,
 211                  service_,
 212                  instance_id_token_info->app_id,
 213                  instance_id_token_info->authorized_entity,
 214                  instance_id_token_info->scope,
 215                  result));
 216   }
 217 }
 218
 219 void GCMDriverDesktop::IOWorker::OnSendFinished(const std::string& app_id,
 220                                                 const std::string& message_id,
 221                                                 GCMClient::Result result) {
 222   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 223
 224   ui_thread_->PostTask(
 225       FROM_HERE,
 226       base::Bind(&GCMDriverDesktop::SendFinished, service_, app_id, message_id,
 227                  result));
 228 }
 229
 230 void GCMDriverDesktop::IOWorker::OnMessageReceived(
 231     const std::string& app_id,
 232     const GCMClient::IncomingMessage& message) {
 233   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 234
 235   ui_thread_->PostTask(
 236       FROM_HERE,
 237       base::Bind(&GCMDriverDesktop::MessageReceived,
 238                  service_,
 239                  app_id,
 240                  message));
 241 }
 242
 243 void GCMDriverDesktop::IOWorker::OnMessagesDeleted(const std::string& app_id) {
 244   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 245
 246   ui_thread_->PostTask(
 247       FROM_HERE,
 248       base::Bind(&GCMDriverDesktop::MessagesDeleted, service_, app_id));
 249 }
 250
 251 void GCMDriverDesktop::IOWorker::OnMessageSendError(
 252     const std::string& app_id,
 253     const GCMClient::SendErrorDetails& send_error_details) {
 254   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 255
 256   ui_thread_->PostTask(
 257       FROM_HERE,
 258       base::Bind(&GCMDriverDesktop::MessageSendError, service_, app_id,
 259                  send_error_details));
 260 }
 261
 262 void GCMDriverDesktop::IOWorker::OnSendAcknowledged(
 263     const std::string& app_id,
 264     const std::string& message_id) {
 265   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 266
 267   ui_thread_->PostTask(
 268       FROM_HERE,
 269       base::Bind(
 270           &GCMDriverDesktop::SendAcknowledged, service_, app_id, message_id));
 271 }
 272
 273 void GCMDriverDesktop::IOWorker::OnGCMReady(
 274     const std::vector<AccountMapping>& account_mappings,
 275     const base::Time& last_token_fetch_time) {
 276   ui_thread_->PostTask(FROM_HERE,
 277                        base::Bind(&GCMDriverDesktop::GCMClientReady,
 278                                   service_,
 279                                   account_mappings,
 280                                   last_token_fetch_time));
 281 }
 282
 283 void GCMDriverDesktop::IOWorker::OnActivityRecorded() {
 284   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 285   // When an activity is recorded, get all the stats and refresh the UI of
 286   // gcm-internals page.
 287   GetGCMStatistics(false);
 288 }
 289
 290 void GCMDriverDesktop::IOWorker::OnConnected(
 291     const net::IPEndPoint& ip_endpoint) {
 292   ui_thread_->PostTask(FROM_HERE,
 293                        base::Bind(&GCMDriverDesktop::OnConnected,
 294                                   service_,
 295                                   ip_endpoint));
 296 }
 297
 298 void GCMDriverDesktop::IOWorker::OnDisconnected() {
 299   ui_thread_->PostTask(FROM_HERE,
 300                        base::Bind(&GCMDriverDesktop::OnDisconnected, service_));
 301 }
 302
 303 void GCMDriverDesktop::IOWorker::Start(
 304     GCMClient::StartMode start_mode,
 305     const base::WeakPtr<GCMDriverDesktop>& service) {
 306   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 307
 308   service_ = service;
 309   gcm_client_->Start(start_mode);
 310 }
 311
 312 void GCMDriverDesktop::IOWorker::Stop() {
 313   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 314
 315   gcm_client_->Stop();
 316 }
 317
 318 void GCMDriverDesktop::IOWorker::Register(
 319     const std::string& app_id,
 320     const std::vector<std::string>& sender_ids) {
 321   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 322
 323   scoped_ptr<GCMRegistrationInfo> gcm_info(new GCMRegistrationInfo);
 324   gcm_info->app_id = app_id;
 325   gcm_info->sender_ids = sender_ids;
 326   gcm_client_->Register(make_linked_ptr<RegistrationInfo>(gcm_info.release()));
 327 }
 328
 329 void GCMDriverDesktop::IOWorker::Unregister(const std::string& app_id) {
 330   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 331
 332   scoped_ptr<GCMRegistrationInfo> gcm_info(new GCMRegistrationInfo);
 333   gcm_info->app_id = app_id;
 334   gcm_client_->Unregister(
 335       make_linked_ptr<RegistrationInfo>(gcm_info.release()));
 336 }
 337
 338 void GCMDriverDesktop::IOWorker::Send(
 339     const std::string& app_id,
 340     const std::string& receiver_id,
 341     const GCMClient::OutgoingMessage& message) {
 342   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 343
 344   gcm_client_->Send(app_id, receiver_id, message);
 345 }
 346
 347 void GCMDriverDesktop::IOWorker::GetGCMStatistics(bool clear_logs) {
 348   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 349   gcm::GCMClient::GCMStatistics stats;
 350
 351   if (gcm_client_.get()) {
 352     if (clear_logs)
 353       gcm_client_->ClearActivityLogs();
 354     stats = gcm_client_->GetStatistics();
 355   }
 356
 357   ui_thread_->PostTask(
 358       FROM_HERE,
 359       base::Bind(&GCMDriverDesktop::GetGCMStatisticsFinished, service_, stats));
 360 }
 361
 362 void GCMDriverDesktop::IOWorker::SetGCMRecording(bool recording) {
 363   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 364   gcm::GCMClient::GCMStatistics stats;
 365
 366   if (gcm_client_.get()) {
 367     gcm_client_->SetRecording(recording);
 368     stats = gcm_client_->GetStatistics();
 369     stats.gcm_client_created = true;
 370   }
 371
 372   ui_thread_->PostTask(
 373       FROM_HERE,
 374       base::Bind(&GCMDriverDesktop::GetGCMStatisticsFinished, service_, stats));
 375 }
 376
 377 void GCMDriverDesktop::IOWorker::SetAccountTokens(
 378     const std::vector<GCMClient::AccountTokenInfo>& account_tokens) {
 379   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 380
 381   if (gcm_client_.get())
 382     gcm_client_->SetAccountTokens(account_tokens);
 383 }
 384
 385 void GCMDriverDesktop::IOWorker::UpdateAccountMapping(
 386     const AccountMapping& account_mapping) {
 387   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 388
 389   if (gcm_client_.get())
 390     gcm_client_->UpdateAccountMapping(account_mapping);
 391 }
 392
 393 void GCMDriverDesktop::IOWorker::RemoveAccountMapping(
 394     const std::string& account_id) {
 395   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 396
 397   if (gcm_client_.get())
 398     gcm_client_->RemoveAccountMapping(account_id);
 399 }
 400
 401 void GCMDriverDesktop::IOWorker::SetLastTokenFetchTime(const base::Time& time) {
 402   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 403
 404   if (gcm_client_.get())
 405     gcm_client_->SetLastTokenFetchTime(time);
 406 }
 407
 408 void GCMDriverDesktop::IOWorker::AddInstanceIDData(
 409     const std::string& app_id,
 410     const std::string& instance_id,
 411     const std::string& extra_data) {
 412   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 413
 414   if (gcm_client_.get())
 415     gcm_client_->AddInstanceIDData(app_id, instance_id, extra_data);
 416 }
 417
 418 void GCMDriverDesktop::IOWorker::RemoveInstanceIDData(
 419     const std::string& app_id) {
 420   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 421
 422   if (gcm_client_.get())
 423     gcm_client_->RemoveInstanceIDData(app_id);
 424 }
 425
 426 void GCMDriverDesktop::IOWorker::GetInstanceIDData(
 427     const std::string& app_id) {
 428   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 429
 430   std::string instance_id;
 431   std::string extra_data;
 432   if (gcm_client_.get())
 433     gcm_client_->GetInstanceIDData(app_id, &instance_id, &extra_data);
 434
 435   ui_thread_->PostTask(
 436       FROM_HERE,
 437       base::Bind(&GCMDriverDesktop::GetInstanceIDDataFinished,
 438                  service_, app_id, instance_id, extra_data));
 439 }
 440
 441 void GCMDriverDesktop::IOWorker::GetToken(
 442     const std::string& app_id,
 443     const std::string& authorized_entity,
 444     const std::string& scope,
 445     const std::map<std::string, std::string>& options) {
 446   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 447
 448   scoped_ptr<InstanceIDTokenInfo> instance_id_token_info(
 449       new InstanceIDTokenInfo);
 450   instance_id_token_info->app_id = app_id;
 451   instance_id_token_info->authorized_entity = authorized_entity;
 452   instance_id_token_info->scope = scope;
 453   instance_id_token_info->options = options;
 454   gcm_client_->Register(
 455       make_linked_ptr<RegistrationInfo>(instance_id_token_info.release()));
 456 }
 457
 458 void GCMDriverDesktop::IOWorker::DeleteToken(
 459     const std::string& app_id,
 460     const std::string& authorized_entity,
 461     const std::string& scope) {
 462   scoped_ptr<InstanceIDTokenInfo> instance_id_token_info(
 463       new InstanceIDTokenInfo);
 464   instance_id_token_info->app_id = app_id;
 465   instance_id_token_info->authorized_entity = authorized_entity;
 466   instance_id_token_info->scope = scope;
 467   gcm_client_->Unregister(
 468       make_linked_ptr<RegistrationInfo>(instance_id_token_info.release()));
 469 }
 470
 471 void GCMDriverDesktop::IOWorker::WakeFromSuspendForHeartbeat(bool wake) {
 472 #if defined(OS_CHROMEOS)
 473   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 474
 475   scoped_ptr<base::Timer> timer;
 476   if (wake)
 477     timer.reset(new timers::SimpleAlarmTimer());
 478   else
 479     timer.reset(new base::Timer(true, false));
 480
 481   gcm_client_->UpdateHeartbeatTimer(timer.Pass());
 482 #endif
 483 }
 484
 485 void GCMDriverDesktop::IOWorker::AddHeartbeatInterval(const std::string& scope,
 486                                                       int interval_ms) {
 487   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 488   gcm_client_->AddHeartbeatInterval(scope, interval_ms);
 489 }
 490
 491 void GCMDriverDesktop::IOWorker::RemoveHeartbeatInterval(
 492     const std::string& scope) {
 493   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 494   gcm_client_->RemoveHeartbeatInterval(scope);
 495 }
 496
 497 GCMDriverDesktop::GCMDriverDesktop(
 498     scoped_ptr<GCMClientFactory> gcm_client_factory,
 499     const GCMClient::ChromeBuildInfo& chrome_build_info,
 500     const std::string& channel_status_request_url,
 501     const std::string& user_agent,
 502     PrefService* prefs,
 503     const base::FilePath& store_path,
 504     const scoped_refptr<net::URLRequestContextGetter>& request_context,
 505     const scoped_refptr<base::SequencedTaskRunner>& ui_thread,
 506     const scoped_refptr<base::SequencedTaskRunner>& io_thread,
 507     const scoped_refptr<base::SequencedTaskRunner>& blocking_task_runner)
 508     : gcm_channel_status_syncer_(
 509           new GCMChannelStatusSyncer(this,
 510                                      prefs,
 511                                      channel_status_request_url,
 512                                      user_agent,
 513                                      request_context)),
 514       signed_in_(false),
 515       gcm_started_(false),
 516       gcm_enabled_(true),
 517       connected_(false),
 518       account_mapper_(new GCMAccountMapper(this)),
 519       // Setting to max, to make sure it does not prompt for token reporting
 520       // Before reading a reasonable value from the DB, which might be never,
 521       // in which case the fetching will be triggered.
 522       last_token_fetch_time_(base::Time::Max()),
 523       ui_thread_(ui_thread),
 524       io_thread_(io_thread),
 525       wake_from_suspend_enabled_(false),
 526       weak_ptr_factory_(this) {
 527   gcm_enabled_ = gcm_channel_status_syncer_->gcm_enabled();
 528
 529   // Create and initialize the GCMClient. Note that this does not initiate the
 530   // GCM check-in.
 531   io_worker_.reset(new IOWorker(ui_thread, io_thread));
 532   io_thread_->PostTask(
 533       FROM_HERE,
 534       base::Bind(&GCMDriverDesktop::IOWorker::Initialize,
 535                  base::Unretained(io_worker_.get()),
 536                  base::Passed(&gcm_client_factory),
 537                  chrome_build_info,
 538                  store_path,
 539                  request_context,
 540                  blocking_task_runner));
 541 }
 542
 543 GCMDriverDesktop::~GCMDriverDesktop() {
 544 }
 545
 546 void GCMDriverDesktop::Shutdown() {
 547   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 548
 549   Stop();
 550   GCMDriver::Shutdown();
 551
 552   // Dispose the syncer in order to release the reference to
 553   // URLRequestContextGetter that needs to be done before IOThread gets
 554   // deleted.
 555   gcm_channel_status_syncer_.reset();
 556
 557   io_thread_->DeleteSoon(FROM_HERE, io_worker_.release());
 558 }
 559
 560 void GCMDriverDesktop::OnSignedIn() {
 561   signed_in_ = true;
 562 }
 563
 564 void GCMDriverDesktop::OnSignedOut() {
 565   signed_in_ = false;
 566 }
 567
 568 void GCMDriverDesktop::AddAppHandler(const std::string& app_id,
 569                                      GCMAppHandler* handler) {
 570   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 571   GCMDriver::AddAppHandler(app_id, handler);
 572
 573    // Ensures that the GCM service is started when there is an interest.
 574   EnsureStarted(GCMClient::DELAYED_START);
 575 }
 576
 577 void GCMDriverDesktop::RemoveAppHandler(const std::string& app_id) {
 578   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 579   GCMDriver::RemoveAppHandler(app_id);
 580
 581   // Stops the GCM service when no app intends to consume it. Stop function will
 582   // remove the last app handler - account mapper.
 583   if (app_handlers().size() == 1) {
 584     Stop();
 585     gcm_channel_status_syncer_->Stop();
 586   }
 587 }
 588
 589 void GCMDriverDesktop::AddConnectionObserver(GCMConnectionObserver* observer) {
 590   connection_observer_list_.AddObserver(observer);
 591 }
 592
 593 void GCMDriverDesktop::RemoveConnectionObserver(
 594     GCMConnectionObserver* observer) {
 595   connection_observer_list_.RemoveObserver(observer);
 596 }
 597
 598 void GCMDriverDesktop::Enable() {
 599   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 600
 601   if (gcm_enabled_)
 602     return;
 603   gcm_enabled_ = true;
 604
 605   EnsureStarted(GCMClient::DELAYED_START);
 606 }
 607
 608 void GCMDriverDesktop::Disable() {
 609   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 610
 611   if (!gcm_enabled_)
 612     return;
 613   gcm_enabled_ = false;
 614
 615   Stop();
 616 }
 617
 618 void GCMDriverDesktop::Stop() {
 619   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 620
 621   // No need to stop GCM service if not started yet.
 622   if (!gcm_started_)
 623     return;
 624
 625   account_mapper_->ShutdownHandler();
 626   GCMDriver::RemoveAppHandler(kGCMAccountMapperAppId);
 627
 628   RemoveCachedData();
 629
 630   io_thread_->PostTask(
 631       FROM_HERE,
 632       base::Bind(&GCMDriverDesktop::IOWorker::Stop,
 633                  base::Unretained(io_worker_.get())));
 634 }
 635
 636 void GCMDriverDesktop::RegisterImpl(
 637     const std::string& app_id,
 638     const std::vector<std::string>& sender_ids) {
 639   // Delay the register operation until GCMClient is ready.
 640   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
 641     delayed_task_controller_->AddTask(base::Bind(&GCMDriverDesktop::DoRegister,
 642                                                  weak_ptr_factory_.GetWeakPtr(),
 643                                                  app_id,
 644                                                  sender_ids));
 645     return;
 646   }
 647
 648   DoRegister(app_id, sender_ids);
 649 }
 650
 651 void GCMDriverDesktop::DoRegister(const std::string& app_id,
 652                                   const std::vector<std::string>& sender_ids) {
 653   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 654   if (!HasRegisterCallback(app_id)) {
 655     // The callback could have been removed when the app is uninstalled.
 656     return;
 657   }
 658
 659   io_thread_->PostTask(
 660       FROM_HERE,
 661       base::Bind(&GCMDriverDesktop::IOWorker::Register,
 662                  base::Unretained(io_worker_.get()),
 663                  app_id,
 664                  sender_ids));
 665 }
 666
 667 void GCMDriverDesktop::UnregisterImpl(const std::string& app_id) {
 668   // Delay the unregister operation until GCMClient is ready.
 669   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
 670     delayed_task_controller_->AddTask(
 671         base::Bind(&GCMDriverDesktop::DoUnregister,
 672                    weak_ptr_factory_.GetWeakPtr(),
 673                    app_id));
 674     return;
 675   }
 676
 677   DoUnregister(app_id);
 678 }
 679
 680 void GCMDriverDesktop::DoUnregister(const std::string& app_id) {
 681   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 682
 683   // Ask the server to unregister it. There could be a small chance that the
 684   // unregister request fails. If this occurs, it does not bring any harm since
 685   // we simply reject the messages/events received from the server.
 686   io_thread_->PostTask(
 687       FROM_HERE,
 688       base::Bind(&GCMDriverDesktop::IOWorker::Unregister,
 689                  base::Unretained(io_worker_.get()),
 690                  app_id));
 691 }
 692
 693 void GCMDriverDesktop::SendImpl(const std::string& app_id,
 694                                 const std::string& receiver_id,
 695                                 const GCMClient::OutgoingMessage& message) {
 696   // Delay the send operation until all GCMClient is ready.
 697   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
 698     delayed_task_controller_->AddTask(base::Bind(&GCMDriverDesktop::DoSend,
 699                                                  weak_ptr_factory_.GetWeakPtr(),
 700                                                  app_id,
 701                                                  receiver_id,
 702                                                  message));
 703     return;
 704   }
 705
 706   DoSend(app_id, receiver_id, message);
 707 }
 708
 709 void GCMDriverDesktop::DoSend(const std::string& app_id,
 710                               const std::string& receiver_id,
 711                               const GCMClient::OutgoingMessage& message) {
 712   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 713   io_thread_->PostTask(
 714       FROM_HERE,
 715       base::Bind(&GCMDriverDesktop::IOWorker::Send,
 716                  base::Unretained(io_worker_.get()),
 717                  app_id,
 718                  receiver_id,
 719                  message));
 720 }
 721
 722 GCMClient* GCMDriverDesktop::GetGCMClientForTesting() const {
 723   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 724   return io_worker_ ? io_worker_->gcm_client_for_testing() : NULL;
 725 }
 726
 727 bool GCMDriverDesktop::IsStarted() const {
 728   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 729   return gcm_started_;
 730 }
 731
 732 bool GCMDriverDesktop::IsConnected() const {
 733   return connected_;
 734 }
 735
 736 void GCMDriverDesktop::GetGCMStatistics(
 737     const GetGCMStatisticsCallback& callback,
 738     bool clear_logs) {
 739   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 740   DCHECK(!callback.is_null());
 741
 742   request_gcm_statistics_callback_ = callback;
 743   io_thread_->PostTask(
 744       FROM_HERE,
 745       base::Bind(&GCMDriverDesktop::IOWorker::GetGCMStatistics,
 746                  base::Unretained(io_worker_.get()),
 747                  clear_logs));
 748 }
 749
 750 void GCMDriverDesktop::SetGCMRecording(const GetGCMStatisticsCallback& callback,
 751                                        bool recording) {
 752   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 753
 754   request_gcm_statistics_callback_ = callback;
 755   io_thread_->PostTask(
 756       FROM_HERE,
 757       base::Bind(&GCMDriverDesktop::IOWorker::SetGCMRecording,
 758                  base::Unretained(io_worker_.get()),
 759                  recording));
 760 }
 761
 762 void GCMDriverDesktop::UpdateAccountMapping(
 763     const AccountMapping& account_mapping) {
 764   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 765
 766   io_thread_->PostTask(
 767       FROM_HERE,
 768       base::Bind(&GCMDriverDesktop::IOWorker::UpdateAccountMapping,
 769                  base::Unretained(io_worker_.get()),
 770                  account_mapping));
 771 }
 772
 773 void GCMDriverDesktop::RemoveAccountMapping(const std::string& account_id) {
 774   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 775
 776   io_thread_->PostTask(
 777       FROM_HERE,
 778       base::Bind(&GCMDriverDesktop::IOWorker::RemoveAccountMapping,
 779                  base::Unretained(io_worker_.get()),
 780                  account_id));
 781 }
 782
 783 base::Time GCMDriverDesktop::GetLastTokenFetchTime() {
 784   return last_token_fetch_time_;
 785 }
 786
 787 void GCMDriverDesktop::SetLastTokenFetchTime(const base::Time& time) {
 788   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 789
 790   last_token_fetch_time_ = time;
 791
 792   io_thread_->PostTask(
 793       FROM_HERE,
 794       base::Bind(&GCMDriverDesktop::IOWorker::SetLastTokenFetchTime,
 795                  base::Unretained(io_worker_.get()),
 796                  time));
 797 }
 798
 799 InstanceIDHandler* GCMDriverDesktop::GetInstanceIDHandler() {
 800   return this;
 801 }
 802
 803 void GCMDriverDesktop::GetToken(
 804     const std::string& app_id,
 805     const std::string& authorized_entity,
 806     const std::string& scope,
 807     const std::map<std::string, std::string>& options,
 808     const GetTokenCallback& callback) {
 809   DCHECK(!app_id.empty());
 810   DCHECK(!authorized_entity.empty());
 811   DCHECK(!scope.empty());
 812   DCHECK(!callback.is_null());
 813   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 814
 815   GCMClient::Result result = EnsureStarted(GCMClient::IMMEDIATE_START);
 816   if (result != GCMClient::SUCCESS) {
 817     callback.Run(std::string(), result);
 818     return;
 819   }
 820
 821   // If previous GetToken operation is still in progress, bail out.
 822   TokenTuple tuple_key(app_id, authorized_entity, scope);
 823   if (get_token_callbacks_.find(tuple_key) != get_token_callbacks_.end()) {
 824     callback.Run(std::string(), GCMClient::ASYNC_OPERATION_PENDING);
 825     return;
 826   }
 827
 828   get_token_callbacks_[tuple_key] = callback;
 829
 830   // Delay the GetToken operation until GCMClient is ready.
 831   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
 832     delayed_task_controller_->AddTask(
 833         base::Bind(&GCMDriverDesktop::DoGetToken,
 834                    weak_ptr_factory_.GetWeakPtr(),
 835                    app_id,
 836                    authorized_entity,
 837                    scope,
 838                    options));
 839     return;
 840   }
 841
 842   DoGetToken(app_id, authorized_entity, scope, options);
 843 }
 844
 845 void GCMDriverDesktop::DoGetToken(
 846     const std::string& app_id,
 847     const std::string& authorized_entity,
 848     const std::string& scope,
 849     const std::map<std::string, std::string>& options) {
 850   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 851
 852   TokenTuple tuple_key(app_id, authorized_entity, scope);
 853   auto callback_iter = get_token_callbacks_.find(tuple_key);
 854   if (callback_iter == get_token_callbacks_.end()) {
 855     // The callback could have been removed when the app is uninstalled.
 856     return;
 857   }
 858
 859   io_thread_->PostTask(
 860       FROM_HERE,
 861       base::Bind(&GCMDriverDesktop::IOWorker::GetToken,
 862                  base::Unretained(io_worker_.get()),
 863                  app_id,
 864                  authorized_entity,
 865                  scope,
 866                  options));
 867 }
 868
 869 void GCMDriverDesktop::DeleteToken(const std::string& app_id,
 870                                    const std::string& authorized_entity,
 871                                    const std::string& scope,
 872                                    const DeleteTokenCallback& callback) {
 873   DCHECK(!app_id.empty());
 874   DCHECK(!authorized_entity.empty());
 875   DCHECK(!scope.empty());
 876   DCHECK(!callback.is_null());
 877   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 878
 879   GCMClient::Result result = EnsureStarted(GCMClient::IMMEDIATE_START);
 880   if (result != GCMClient::SUCCESS) {
 881     callback.Run(result);
 882     return;
 883   }
 884
 885   // If previous GetToken operation is still in progress, bail out.
 886   TokenTuple tuple_key(app_id, authorized_entity, scope);
 887   if (delete_token_callbacks_.find(tuple_key) !=
 888       delete_token_callbacks_.end()) {
 889     callback.Run(GCMClient::ASYNC_OPERATION_PENDING);
 890     return;
 891   }
 892
 893   delete_token_callbacks_[tuple_key] = callback;
 894
 895   // Delay the DeleteToken operation until GCMClient is ready.
 896   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
 897     delayed_task_controller_->AddTask(
 898         base::Bind(&GCMDriverDesktop::DoDeleteToken,
 899                    weak_ptr_factory_.GetWeakPtr(),
 900                    app_id,
 901                    authorized_entity,
 902                    scope));
 903     return;
 904   }
 905
 906   DoDeleteToken(app_id, authorized_entity, scope);
 907 }
 908
 909 void GCMDriverDesktop::DoDeleteToken(const std::string& app_id,
 910                                      const std::string& authorized_entity,
 911                                      const std::string& scope) {
 912   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 913
 914   io_thread_->PostTask(
 915       FROM_HERE,
 916       base::Bind(&GCMDriverDesktop::IOWorker::DeleteToken,
 917                  base::Unretained(io_worker_.get()),
 918                  app_id,
 919                  authorized_entity,
 920                  scope));
 921 }
 922
 923 void GCMDriverDesktop::AddInstanceIDData(
 924     const std::string& app_id,
 925     const std::string& instance_id,
 926     const std::string& extra_data) {
 927   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 928
 929   GCMClient::Result result = EnsureStarted(GCMClient::IMMEDIATE_START);
 930   if (result != GCMClient::SUCCESS)
 931     return;
 932
 933   // Delay the operation until GCMClient is ready.
 934   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
 935     delayed_task_controller_->AddTask(
 936         base::Bind(&GCMDriverDesktop::DoAddInstanceIDData,
 937                    weak_ptr_factory_.GetWeakPtr(),
 938                    app_id,
 939                    instance_id,
 940                    extra_data));
 941     return;
 942   }
 943
 944   DoAddInstanceIDData(app_id, instance_id, extra_data);
 945 }
 946
 947 void GCMDriverDesktop::DoAddInstanceIDData(
 948     const std::string& app_id,
 949     const std::string& instance_id,
 950     const std::string& extra_data) {
 951   io_thread_->PostTask(
 952       FROM_HERE,
 953       base::Bind(&GCMDriverDesktop::IOWorker::AddInstanceIDData,
 954                  base::Unretained(io_worker_.get()),
 955                  app_id,
 956                  instance_id,
 957                  extra_data));
 958 }
 959
 960 void GCMDriverDesktop::RemoveInstanceIDData(const std::string& app_id) {
 961   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 962
 963   GCMClient::Result result = EnsureStarted(GCMClient::IMMEDIATE_START);
 964   if (result != GCMClient::SUCCESS)
 965     return;
 966
 967   // Delay the operation until GCMClient is ready.
 968   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
 969     delayed_task_controller_->AddTask(
 970         base::Bind(&GCMDriverDesktop::DoRemoveInstanceIDData,
 971                    weak_ptr_factory_.GetWeakPtr(),
 972                    app_id));
 973     return;
 974   }
 975
 976   DoRemoveInstanceIDData(app_id);
 977 }
 978
 979 void GCMDriverDesktop::DoRemoveInstanceIDData(const std::string& app_id) {
 980   io_thread_->PostTask(
 981       FROM_HERE,
 982       base::Bind(&GCMDriverDesktop::IOWorker::RemoveInstanceIDData,
 983                  base::Unretained(io_worker_.get()),
 984                  app_id));
 985 }
 986
 987 void GCMDriverDesktop::GetInstanceIDData(
 988     const std::string& app_id,
 989     const GetInstanceIDDataCallback& callback) {
 990   DCHECK(!get_instance_id_data_callbacks_.count(app_id));
 991   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 992
 993   GCMClient::Result result = EnsureStarted(GCMClient::IMMEDIATE_START);
 994   if (result != GCMClient::SUCCESS) {
 995     callback.Run(std::string(), std::string());
 996     return;
 997   }
 998
 999   get_instance_id_data_callbacks_[app_id] = callback;
1000
1001   // Delay the operation until GCMClient is ready.
1002   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
1003     delayed_task_controller_->AddTask(
1004         base::Bind(&GCMDriverDesktop::DoGetInstanceIDData,
1005                    weak_ptr_factory_.GetWeakPtr(),
1006                    app_id));
1007     return;
1008   }
1009
1010   DoGetInstanceIDData(app_id);
1011 }
1012
1013 void GCMDriverDesktop::DoGetInstanceIDData(const std::string& app_id) {
1014   io_thread_->PostTask(
1015       FROM_HERE,
1016       base::Bind(&GCMDriverDesktop::IOWorker::GetInstanceIDData,
1017                  base::Unretained(io_worker_.get()),
1018                  app_id));
1019 }
1020
1021 void GCMDriverDesktop::GetInstanceIDDataFinished(
1022     const std::string& app_id,
1023     const std::string& instance_id,
1024     const std::string& extra_data) {
1025   DCHECK(get_instance_id_data_callbacks_.count(app_id));
1026   get_instance_id_data_callbacks_[app_id].Run(instance_id, extra_data);
1027   get_instance_id_data_callbacks_.erase(app_id);
1028 }
1029
1030 void GCMDriverDesktop::GetTokenFinished(const std::string& app_id,
1031                                         const std::string& authorized_entity,
1032                                         const std::string& scope,
1033                                         const std::string& token,
1034                                         GCMClient::Result result) {
1035   TokenTuple tuple_key(app_id, authorized_entity, scope);
1036   auto callback_iter = get_token_callbacks_.find(tuple_key);
1037   if (callback_iter == get_token_callbacks_.end()) {
1038     // The callback could have been removed when the app is uninstalled.
1039     return;
1040   }
1041
1042   GetTokenCallback callback = callback_iter->second;
1043   get_token_callbacks_.erase(callback_iter);
1044   callback.Run(token, result);
1045 }
1046
1047 void GCMDriverDesktop::DeleteTokenFinished(const std::string& app_id,
1048                                            const std::string& authorized_entity,
1049                                            const std::string& scope,
1050                                            GCMClient::Result result) {
1051   TokenTuple tuple_key(app_id, authorized_entity, scope);
1052   auto callback_iter = delete_token_callbacks_.find(tuple_key);
1053   if (callback_iter == delete_token_callbacks_.end()) {
1054     // The callback could have been removed when the app is uninstalled.
1055     return;
1056   }
1057
1058   DeleteTokenCallback callback = callback_iter->second;
1059   delete_token_callbacks_.erase(callback_iter);
1060   callback.Run(result);
1061 }
1062
1063 void GCMDriverDesktop::WakeFromSuspendForHeartbeat(bool wake) {
1064   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1065
1066   wake_from_suspend_enabled_ = wake;
1067
1068   // The GCM service has not been initialized.
1069   if (!delayed_task_controller_)
1070     return;
1071
1072   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
1073     // The GCM service was initialized but has not started yet.
1074     delayed_task_controller_->AddTask(
1075         base::Bind(&GCMDriverDesktop::WakeFromSuspendForHeartbeat,
1076                    weak_ptr_factory_.GetWeakPtr(),
1077                    wake_from_suspend_enabled_));
1078     return;
1079   }
1080
1081   // The GCMClient is ready so we can go ahead and post this task to the
1082   // IOWorker.
1083   io_thread_->PostTask(
1084       FROM_HERE,
1085       base::Bind(&GCMDriverDesktop::IOWorker::WakeFromSuspendForHeartbeat,
1086                  base::Unretained(io_worker_.get()),
1087                  wake_from_suspend_enabled_));
1088 }
1089
1090 void GCMDriverDesktop::AddHeartbeatInterval(const std::string& scope,
1091                                             int interval_ms) {
1092   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1093
1094   // The GCM service has not been initialized.
1095   if (!delayed_task_controller_)
1096     return;
1097
1098   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
1099     // The GCM service was initialized but has not started yet.
1100     delayed_task_controller_->AddTask(
1101         base::Bind(&GCMDriverDesktop::AddHeartbeatInterval,
1102                    weak_ptr_factory_.GetWeakPtr(), scope, interval_ms));
1103     return;
1104   }
1105
1106   io_thread_->PostTask(
1107       FROM_HERE,
1108       base::Bind(&GCMDriverDesktop::IOWorker::AddHeartbeatInterval,
1109                  base::Unretained(io_worker_.get()), scope, interval_ms));
1110 }
1111
1112 void GCMDriverDesktop::RemoveHeartbeatInterval(const std::string& scope) {
1113   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1114
1115   // The GCM service has not been initialized.
1116   if (!delayed_task_controller_)
1117     return;
1118
1119   if (!delayed_task_controller_->CanRunTaskWithoutDelay()) {
1120     // The GCM service was initialized but has not started yet.
1121     delayed_task_controller_->AddTask(
1122         base::Bind(&GCMDriverDesktop::RemoveHeartbeatInterval,
1123                    weak_ptr_factory_.GetWeakPtr(), scope));
1124     return;
1125   }
1126
1127   io_thread_->PostTask(
1128       FROM_HERE,
1129       base::Bind(&GCMDriverDesktop::IOWorker::RemoveHeartbeatInterval,
1130                  base::Unretained(io_worker_.get()), scope));
1131 }
1132
1133 void GCMDriverDesktop::SetAccountTokens(
1134     const std::vector<GCMClient::AccountTokenInfo>& account_tokens) {
1135   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1136
1137   account_mapper_->SetAccountTokens(account_tokens);
1138
1139   io_thread_->PostTask(
1140       FROM_HERE,
1141       base::Bind(&GCMDriverDesktop::IOWorker::SetAccountTokens,
1142                  base::Unretained(io_worker_.get()),
1143                  account_tokens));
1144 }
1145
1146 GCMClient::Result GCMDriverDesktop::EnsureStarted(
1147     GCMClient::StartMode start_mode) {
1148   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1149
1150   if (gcm_started_)
1151     return GCMClient::SUCCESS;
1152
1153   // Have any app requested the service?
1154   if (app_handlers().empty())
1155     return GCMClient::UNKNOWN_ERROR;
1156
1157   // Polling for channel status should be invoked when GCM is being requested,
1158   // no matter whether GCM is enabled or nor.
1159   gcm_channel_status_syncer_->EnsureStarted();
1160
1161   if (!gcm_enabled_)
1162     return GCMClient::GCM_DISABLED;
1163
1164   if (!delayed_task_controller_)
1165     delayed_task_controller_.reset(new GCMDelayedTaskController);
1166
1167   // Note that we need to pass weak pointer again since the existing weak
1168   // pointer in IOWorker might have been invalidated when GCM is stopped.
1169   io_thread_->PostTask(
1170       FROM_HERE,
1171       base::Bind(&GCMDriverDesktop::IOWorker::Start,
1172                  base::Unretained(io_worker_.get()),
1173                  start_mode,
1174                  weak_ptr_factory_.GetWeakPtr()));
1175
1176   return GCMClient::SUCCESS;
1177 }
1178
1179 void GCMDriverDesktop::RemoveCachedData() {
1180   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1181   // Remove all the queued tasks since they no longer make sense after
1182   // GCM service is stopped.
1183   weak_ptr_factory_.InvalidateWeakPtrs();
1184
1185   gcm_started_ = false;
1186   delayed_task_controller_.reset();
1187   ClearCallbacks();
1188 }
1189
1190 void GCMDriverDesktop::MessageReceived(
1191     const std::string& app_id,
1192     const GCMClient::IncomingMessage& message) {
1193   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1194
1195   // Drop the event if the service has been stopped.
1196   if (!gcm_started_)
1197     return;
1198
1199   GetAppHandler(app_id)->OnMessage(app_id, message);
1200 }
1201
1202 void GCMDriverDesktop::MessagesDeleted(const std::string& app_id) {
1203   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1204
1205   // Drop the event if the service has been stopped.
1206   if (!gcm_started_)
1207     return;
1208
1209   GetAppHandler(app_id)->OnMessagesDeleted(app_id);
1210 }
1211
1212 void GCMDriverDesktop::MessageSendError(
1213     const std::string& app_id,
1214     const GCMClient::SendErrorDetails& send_error_details) {
1215   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1216
1217   // Drop the event if the service has been stopped.
1218   if (!gcm_started_)
1219     return;
1220
1221   GetAppHandler(app_id)->OnSendError(app_id, send_error_details);
1222 }
1223
1224 void GCMDriverDesktop::SendAcknowledged(const std::string& app_id,
1225                                         const std::string& message_id) {
1226   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1227
1228   // Drop the event if the service has been stopped.
1229   if (!gcm_started_)
1230     return;
1231
1232   GetAppHandler(app_id)->OnSendAcknowledged(app_id, message_id);
1233 }
1234
1235 void GCMDriverDesktop::GCMClientReady(
1236     const std::vector<AccountMapping>& account_mappings,
1237     const base::Time& last_token_fetch_time) {
1238   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1239
1240   UMA_HISTOGRAM_BOOLEAN("GCM.UserSignedIn", signed_in_);
1241
1242   gcm_started_ = true;
1243   if (wake_from_suspend_enabled_)
1244     WakeFromSuspendForHeartbeat(wake_from_suspend_enabled_);
1245
1246   last_token_fetch_time_ = last_token_fetch_time;
1247
1248   GCMDriver::AddAppHandler(kGCMAccountMapperAppId, account_mapper_.get());
1249   account_mapper_->Initialize(account_mappings,
1250                               base::Bind(&GCMDriverDesktop::MessageReceived,
1251                                          weak_ptr_factory_.GetWeakPtr()));
1252
1253   delayed_task_controller_->SetReady();
1254 }
1255
1256 void GCMDriverDesktop::OnConnected(const net::IPEndPoint& ip_endpoint) {
1257   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1258
1259   connected_ = true;
1260
1261   // Drop the event if the service has been stopped.
1262   if (!gcm_started_)
1263     return;
1264
1265   FOR_EACH_OBSERVER(GCMConnectionObserver,
1266                     connection_observer_list_,
1267                     OnConnected(ip_endpoint));
1268 }
1269
1270 void GCMDriverDesktop::OnDisconnected() {
1271   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1272
1273   connected_ = false;
1274
1275   // Drop the event if the service has been stopped.
1276   if (!gcm_started_)
1277     return;
1278
1279   FOR_EACH_OBSERVER(
1280       GCMConnectionObserver, connection_observer_list_, OnDisconnected());
1281 }
1282
1283 void GCMDriverDesktop::GetGCMStatisticsFinished(
1284     const GCMClient::GCMStatistics& stats) {
1285   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
1286
1287   // request_gcm_statistics_callback_ could be null when an activity, i.e.
1288   // network activity, is triggered while gcm-intenals page is not open.
1289   if (!request_gcm_statistics_callback_.is_null())
1290     request_gcm_statistics_callback_.Run(stats);
1291 }
1292
1293 bool GCMDriverDesktop::TokenTupleComparer::operator()(
1294     const TokenTuple& a, const TokenTuple& b) const {
1295   if (base::get<0>(a) < base::get<0>(b))
1296     return true;
1297   if (base::get<0>(a) > base::get<0>(b))
1298     return false;
1299
1300   if (base::get<1>(a) < base::get<1>(b))
1301     return true;
1302   if (base::get<1>(a) > base::get<1>(b))
1303     return false;
1304
1305   return base::get<2>(a) < base::get<2>(b);
1306 }
1307
1308 }  // namespace gcm