1// Copyright (c) 2010 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 "ppapi/shared_impl/callback_tracker.h" 6 7#include <algorithm> 8 9#include "base/bind.h" 10#include "base/compiler_specific.h" 11#include "base/logging.h" 12#include "base/message_loop/message_loop.h" 13#include "ppapi/c/pp_completion_callback.h" 14#include "ppapi/shared_impl/tracked_callback.h" 15 16namespace ppapi { 17 18// CallbackTracker ------------------------------------------------------------- 19 20CallbackTracker::CallbackTracker() { 21} 22 23void CallbackTracker::AbortAll() { 24 // Iterate over a copy since |Abort()| calls |Remove()| (indirectly). 25 // TODO(viettrungluu): This obviously isn't so efficient. 26 CallbackSetMap pending_callbacks_copy = pending_callbacks_; 27 for (CallbackSetMap::iterator it1 = pending_callbacks_copy.begin(); 28 it1 != pending_callbacks_copy.end(); ++it1) { 29 for (CallbackSet::iterator it2 = it1->second.begin(); 30 it2 != it1->second.end(); ++it2) { 31 (*it2)->Abort(); 32 } 33 } 34} 35 36void CallbackTracker::PostAbortForResource(PP_Resource resource_id) { 37 CHECK(resource_id != 0); 38 CallbackSetMap::iterator it1 = pending_callbacks_.find(resource_id); 39 if (it1 == pending_callbacks_.end()) 40 return; 41 for (CallbackSet::iterator it2 = it1->second.begin(); 42 it2 != it1->second.end(); ++it2) { 43 // Post the abort. 44 (*it2)->PostAbort(); 45 } 46} 47 48CallbackTracker::~CallbackTracker() { 49 // All callbacks must be aborted before destruction. 50 CHECK_EQ(0u, pending_callbacks_.size()); 51} 52 53void CallbackTracker::Add( 54 const scoped_refptr<TrackedCallback>& tracked_callback) { 55 PP_Resource resource_id = tracked_callback->resource_id(); 56 DCHECK(pending_callbacks_[resource_id].find(tracked_callback) == 57 pending_callbacks_[resource_id].end()); 58 pending_callbacks_[resource_id].insert(tracked_callback); 59} 60 61void CallbackTracker::Remove( 62 const scoped_refptr<TrackedCallback>& tracked_callback) { 63 CallbackSetMap::iterator map_it = 64 pending_callbacks_.find(tracked_callback->resource_id()); 65 DCHECK(map_it != pending_callbacks_.end()); 66 CallbackSet::iterator it = map_it->second.find(tracked_callback); 67 DCHECK(it != map_it->second.end()); 68 map_it->second.erase(it); 69 70 // If there are no pending callbacks left for this ID, get rid of the entry. 71 if (map_it->second.empty()) 72 pending_callbacks_.erase(map_it); 73} 74 75} // namespace ppapi 76