content/child/notifications/notification_dispatcher.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 "content/child/notifications/notification_dispatcher.h"
   6
   7 #include <limits>
   8
   9 #include "content/child/notifications/notification_manager.h"
  10
  11 namespace content {
  12
  13 NotificationDispatcher::NotificationDispatcher(
  14     ThreadSafeSender* thread_safe_sender)
  15     : WorkerThreadMessageFilter(thread_safe_sender) {
  16 }
  17
  18 NotificationDispatcher::~NotificationDispatcher() {}
  19
  20 int NotificationDispatcher::GenerateNotificationId(int thread_id) {
  21   base::AutoLock lock(notification_id_map_lock_);
  22   CHECK_LT(next_notification_id_, std::numeric_limits<int>::max());
  23
  24   notification_id_map_[next_notification_id_] = thread_id;
  25   return next_notification_id_++;
  26 }
  27
  28 bool NotificationDispatcher::ShouldHandleMessage(
  29     const IPC::Message& msg) const {
  30   return IPC_MESSAGE_CLASS(msg) == PlatformNotificationMsgStart;
  31 }
  32
  33 void NotificationDispatcher::OnFilteredMessageReceived(
  34     const IPC::Message& msg) {
  35   NotificationManager::ThreadSpecificInstance(thread_safe_sender(),
  36                                               main_thread_task_runner(),
  37                                               this)->OnMessageReceived(msg);
  38 }
  39
  40 bool NotificationDispatcher::GetWorkerThreadIdForMessage(
  41     const IPC::Message& msg,
  42     int* ipc_thread_id) {
  43   int notification_id = -1;
  44   const bool success = base::PickleIterator(msg).ReadInt(&notification_id);
  45   DCHECK(success);
  46
  47   base::AutoLock lock(notification_id_map_lock_);
  48   auto iterator = notification_id_map_.find(notification_id);
  49   if (iterator != notification_id_map_.end()) {
  50     *ipc_thread_id = iterator->second;
  51     return true;
  52   }
  53   return false;
  54 }
  55
  56 }  // namespace content