| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "webkit/plugins/ppapi/callbacks.h" | 5 #include "ppapi/shared_impl/callback_tracker.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/compiler_specific.h" | 10 #include "base/compiler_specific.h" |
| 11 #include "base/logging.h" | 11 #include "base/logging.h" |
| 12 #include "base/message_loop.h" | 12 #include "base/message_loop.h" |
| 13 #include "ppapi/c/pp_completion_callback.h" | 13 #include "ppapi/c/pp_completion_callback.h" |
| 14 #include "ppapi/c/pp_errors.h" | 14 #include "ppapi/shared_impl/tracked_callback.h" |
| 15 | 15 |
| 16 namespace webkit { | |
| 17 namespace ppapi { | 16 namespace ppapi { |
| 18 | 17 |
| 19 // CallbackTracker ------------------------------------------------------------- | 18 // CallbackTracker ------------------------------------------------------------- |
| 20 | 19 |
| 21 CallbackTracker::CallbackTracker() { | 20 CallbackTracker::CallbackTracker() { |
| 22 } | 21 } |
| 23 | 22 |
| 24 void CallbackTracker::AbortAll() { | 23 void CallbackTracker::AbortAll() { |
| 25 // Iterate over a copy since |Abort()| calls |Remove()| (indirectly). | 24 // Iterate over a copy since |Abort()| calls |Remove()| (indirectly). |
| 26 // TODO(viettrungluu): This obviously isn't so efficient. | 25 // TODO(viettrungluu): This obviously isn't so efficient. |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 66 DCHECK(map_it != pending_callbacks_.end()); | 65 DCHECK(map_it != pending_callbacks_.end()); |
| 67 CallbackSet::iterator it = map_it->second.find(tracked_callback); | 66 CallbackSet::iterator it = map_it->second.find(tracked_callback); |
| 68 DCHECK(it != map_it->second.end()); | 67 DCHECK(it != map_it->second.end()); |
| 69 map_it->second.erase(it); | 68 map_it->second.erase(it); |
| 70 | 69 |
| 71 // If there are no pending callbacks left for this ID, get rid of the entry. | 70 // If there are no pending callbacks left for this ID, get rid of the entry. |
| 72 if (map_it->second.empty()) | 71 if (map_it->second.empty()) |
| 73 pending_callbacks_.erase(map_it); | 72 pending_callbacks_.erase(map_it); |
| 74 } | 73 } |
| 75 | 74 |
| 76 // TrackedCallback ------------------------------------------------------------- | |
| 77 | |
| 78 TrackedCallback::TrackedCallback(const scoped_refptr<CallbackTracker>& tracker, | |
| 79 PP_Resource resource_id) | |
| 80 : ALLOW_THIS_IN_INITIALIZER_LIST(abort_impl_factory_(this)), | |
| 81 tracker_(tracker), | |
| 82 resource_id_(resource_id), | |
| 83 completed_(false), | |
| 84 aborted_(false) { | |
| 85 tracker_->Add(make_scoped_refptr(this)); | |
| 86 } | |
| 87 | |
| 88 void TrackedCallback::Abort() { | |
| 89 if (!completed()) { | |
| 90 aborted_ = true; | |
| 91 AbortImpl(); | |
| 92 } | |
| 93 } | |
| 94 | |
| 95 void TrackedCallback::PostAbort() { | |
| 96 if (!completed()) { | |
| 97 aborted_ = true; | |
| 98 // Post a task for the abort (only if necessary). | |
| 99 if (!abort_impl_factory_.HasWeakPtrs()) { | |
| 100 MessageLoop::current()->PostTask( | |
| 101 FROM_HERE, | |
| 102 base::Bind(&TrackedCallback::AbortImpl, | |
| 103 abort_impl_factory_.GetWeakPtr())); | |
| 104 } | |
| 105 } | |
| 106 } | |
| 107 | |
| 108 TrackedCallback::~TrackedCallback() { | |
| 109 // The tracker must ensure that the callback is completed (maybe abortively). | |
| 110 DCHECK(completed_); | |
| 111 } | |
| 112 | |
| 113 void TrackedCallback::MarkAsCompleted() { | |
| 114 DCHECK(!completed()); | |
| 115 | |
| 116 // We will be removed; maintain a reference to ensure we won't be deleted | |
| 117 // until we're done. | |
| 118 scoped_refptr<TrackedCallback> thiz = this; | |
| 119 completed_ = true; | |
| 120 tracker_->Remove(thiz); | |
| 121 tracker_ = NULL; | |
| 122 } | |
| 123 | |
| 124 // TrackedCompletionCallback --------------------------------------------------- | |
| 125 | |
| 126 TrackedCompletionCallback::TrackedCompletionCallback( | |
| 127 const scoped_refptr<CallbackTracker>& tracker, | |
| 128 PP_Resource resource_id, | |
| 129 const PP_CompletionCallback& callback) | |
| 130 : TrackedCallback(tracker, resource_id), | |
| 131 callback_(callback) { | |
| 132 } | |
| 133 | |
| 134 void TrackedCompletionCallback::Run(int32_t result) { | |
| 135 if (!completed()) { | |
| 136 // Cancel any pending calls. | |
| 137 abort_impl_factory_.InvalidateWeakPtrs(); | |
| 138 | |
| 139 // Copy |callback_| and look at |aborted()| now, since |MarkAsCompleted()| | |
| 140 // may delete us. | |
| 141 PP_CompletionCallback callback = callback_; | |
| 142 if (aborted()) | |
| 143 result = PP_ERROR_ABORTED; | |
| 144 | |
| 145 // Do this before running the callback in case of reentrancy (which | |
| 146 // shouldn't happen, but avoid strange failures). | |
| 147 MarkAsCompleted(); | |
| 148 PP_RunCompletionCallback(&callback, result); | |
| 149 } | |
| 150 } | |
| 151 | |
| 152 void TrackedCompletionCallback::AbortImpl() { | |
| 153 Run(PP_ERROR_ABORTED); | |
| 154 } | |
| 155 | |
| 156 } // namespace ppapi | 75 } // namespace ppapi |
| 157 } // namespace webkit | |
| OLD | NEW |