| OLD | NEW | 
 | (Empty) | 
|   1 // Copyright (c) 2006-2008 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/test/automation/automation_handle_tracker.h" |  | 
|   6  |  | 
|   7 #include "chrome/common/automation_messages.h" |  | 
|   8 #include "chrome/test/automation/automation_proxy.h" |  | 
|   9  |  | 
|  10 AutomationResourceProxy::AutomationResourceProxy( |  | 
|  11     AutomationHandleTracker* tracker, AutomationMessageSender* sender, |  | 
|  12     AutomationHandle handle) |  | 
|  13     : handle_(handle), |  | 
|  14       tracker_(tracker), |  | 
|  15       sender_(sender), |  | 
|  16       is_valid_(true) { |  | 
|  17       tracker_->Add(this); |  | 
|  18 } |  | 
|  19  |  | 
|  20 AutomationResourceProxy::~AutomationResourceProxy() { |  | 
|  21   if (tracker_) |  | 
|  22     tracker_->Remove(this); |  | 
|  23 } |  | 
|  24  |  | 
|  25 AutomationHandleTracker::AutomationHandleTracker() : channel_(NULL) {} |  | 
|  26  |  | 
|  27 AutomationHandleTracker::~AutomationHandleTracker() { |  | 
|  28   // Tell any live objects that the tracker is going away so they don't try to |  | 
|  29   // call us when they are being destroyed. |  | 
|  30   for (HandleToObjectMap::iterator iter = handle_to_object_.begin(); |  | 
|  31        iter != handle_to_object_.end(); ++iter) { |  | 
|  32     iter->second->Invalidate(); |  | 
|  33   } |  | 
|  34 } |  | 
|  35  |  | 
|  36 void AutomationHandleTracker::Add(AutomationResourceProxy* proxy) { |  | 
|  37   base::AutoLock lock(map_lock_); |  | 
|  38   handle_to_object_.insert(MapEntry(proxy->handle(), proxy)); |  | 
|  39 } |  | 
|  40  |  | 
|  41 void AutomationHandleTracker::Remove(AutomationResourceProxy* proxy) { |  | 
|  42   base::AutoLock lock(map_lock_); |  | 
|  43   HandleToObjectMap::iterator iter = handle_to_object_.find(proxy->handle()); |  | 
|  44   if (iter != handle_to_object_.end()) { |  | 
|  45     AutomationHandle proxy_handle = proxy->handle(); |  | 
|  46     handle_to_object_.erase(iter); |  | 
|  47     if (channel_) |  | 
|  48       channel_->Send(new AutomationMsg_HandleUnused(proxy_handle)); |  | 
|  49   } |  | 
|  50 } |  | 
|  51  |  | 
|  52 void AutomationHandleTracker::InvalidateHandle(AutomationHandle handle) { |  | 
|  53   // Called in background thread. |  | 
|  54   base::AutoLock lock(map_lock_); |  | 
|  55   HandleToObjectMap::iterator iter = handle_to_object_.find(handle); |  | 
|  56   if (iter != handle_to_object_.end()) { |  | 
|  57     scoped_refptr<AutomationResourceProxy> proxy = iter->second; |  | 
|  58     handle_to_object_.erase(iter); |  | 
|  59     proxy->Invalidate(); |  | 
|  60   } |  | 
|  61 } |  | 
|  62  |  | 
|  63 AutomationResourceProxy* AutomationHandleTracker::GetResource( |  | 
|  64     AutomationHandle handle) { |  | 
|  65   base::AutoLock lock(map_lock_); |  | 
|  66   HandleToObjectMap::iterator iter = handle_to_object_.find(handle); |  | 
|  67   if (iter == handle_to_object_.end()) |  | 
|  68     return NULL; |  | 
|  69  |  | 
|  70   iter->second->AddRef(); |  | 
|  71   return iter->second.get(); |  | 
|  72 } |  | 
| OLD | NEW |