chrome/browser/automation/automation_resource_tracker.cc

   1 // Copyright (c) 2011 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/automation/automation_resource_tracker.h"
   6
   7 #include "chrome/common/automation_messages.h"
   8
   9 AutomationResourceTrackerImpl::AutomationResourceTrackerImpl(
  10     IPC::Sender* sender)
  11     : sender_(sender) {
  12 }
  13
  14 AutomationResourceTrackerImpl::~AutomationResourceTrackerImpl() {
  15 }
  16
  17 int AutomationResourceTrackerImpl::AddImpl(const void* resource) {
  18   if (ContainsResourceImpl(resource))
  19     return resource_to_handle_[resource];
  20
  21   int handle = GenerateHandle();
  22   DCHECK(!ContainsHandleImpl(handle));
  23
  24   resource_to_handle_[resource] = handle;
  25   handle_to_resource_[handle] = resource;
  26
  27   AddObserverTypeProxy(resource);
  28
  29   return handle;
  30 }
  31
  32 void AutomationResourceTrackerImpl::RemoveImpl(const void* resource) {
  33   if (!ContainsResourceImpl(resource))
  34     return;
  35
  36   int handle = resource_to_handle_[resource];
  37   DCHECK_EQ(resource, handle_to_resource_[handle]);
  38
  39   RemoveObserverTypeProxy(resource);
  40
  41   resource_to_handle_.erase(resource);
  42   handle_to_resource_.erase(handle);
  43 }
  44
  45 int AutomationResourceTrackerImpl::GenerateHandle() {
  46   static int handle = 0;
  47   return ++handle;
  48 }
  49
  50 bool AutomationResourceTrackerImpl::ContainsResourceImpl(const void* resource) {
  51   return resource_to_handle_.find(resource) != resource_to_handle_.end();
  52 }
  53
  54 bool AutomationResourceTrackerImpl::ContainsHandleImpl(int handle) {
  55   return handle_to_resource_.find(handle) != handle_to_resource_.end();
  56 }
  57
  58 const void* AutomationResourceTrackerImpl::GetResourceImpl(int handle) {
  59   HandleToResourceMap::const_iterator iter = handle_to_resource_.find(handle);
  60   if (iter == handle_to_resource_.end())
  61     return NULL;
  62
  63   return iter->second;
  64 }
  65
  66 int AutomationResourceTrackerImpl::GetHandleImpl(const void* resource) {
  67   ResourceToHandleMap::const_iterator iter =
  68     resource_to_handle_.find(resource);
  69   if (iter == resource_to_handle_.end())
  70     return 0;
  71
  72   return iter->second;
  73 }
  74
  75 void AutomationResourceTrackerImpl::HandleCloseNotification(
  76     const void* resource) {
  77   if (!ContainsResourceImpl(resource))
  78     return;
  79
  80   sender_->Send(
  81     new AutomationMsg_InvalidateHandle(resource_to_handle_[resource]));
  82
  83   RemoveImpl(resource);
  84 }