| OLD | NEW |
| (Empty) |
| 1 // Copyright 2017 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 "mojo/edk/system/watch.h" | |
| 6 | |
| 7 #include "mojo/edk/system/request_context.h" | |
| 8 #include "mojo/edk/system/watcher_dispatcher.h" | |
| 9 | |
| 10 namespace mojo { | |
| 11 namespace edk { | |
| 12 | |
| 13 Watch::Watch(const scoped_refptr<WatcherDispatcher>& watcher, | |
| 14 const scoped_refptr<Dispatcher>& dispatcher, | |
| 15 uintptr_t context, | |
| 16 MojoHandleSignals signals) | |
| 17 : watcher_(watcher), | |
| 18 dispatcher_(dispatcher), | |
| 19 context_(context), | |
| 20 signals_(signals) {} | |
| 21 | |
| 22 bool Watch::NotifyState(const HandleSignalsState& state, | |
| 23 bool allowed_to_call_callback) { | |
| 24 AssertWatcherLockAcquired(); | |
| 25 | |
| 26 // NOTE: This method must NEVER call into |dispatcher_| directly, because it | |
| 27 // may be called while |dispatcher_| holds a lock. | |
| 28 | |
| 29 MojoResult rv = MOJO_RESULT_SHOULD_WAIT; | |
| 30 RequestContext* const request_context = RequestContext::current(); | |
| 31 if (state.satisfies(signals_)) { | |
| 32 rv = MOJO_RESULT_OK; | |
| 33 if (allowed_to_call_callback && rv != last_known_result_) { | |
| 34 request_context->AddWatchNotifyFinalizer(this, MOJO_RESULT_OK, state); | |
| 35 } | |
| 36 } else if (!state.can_satisfy(signals_)) { | |
| 37 rv = MOJO_RESULT_FAILED_PRECONDITION; | |
| 38 if (allowed_to_call_callback && rv != last_known_result_) { | |
| 39 request_context->AddWatchNotifyFinalizer( | |
| 40 this, MOJO_RESULT_FAILED_PRECONDITION, state); | |
| 41 } | |
| 42 } | |
| 43 | |
| 44 last_known_signals_state_ = | |
| 45 *static_cast<const MojoHandleSignalsState*>(&state); | |
| 46 last_known_result_ = rv; | |
| 47 return ready(); | |
| 48 } | |
| 49 | |
| 50 void Watch::Cancel() { | |
| 51 RequestContext::current()->AddWatchCancelFinalizer(this); | |
| 52 } | |
| 53 | |
| 54 void Watch::InvokeCallback(MojoResult result, | |
| 55 const HandleSignalsState& state, | |
| 56 MojoWatcherNotificationFlags flags) { | |
| 57 // We hold the lock through invocation to ensure that only one notification | |
| 58 // callback runs for this context at any given time. | |
| 59 base::AutoLock lock(notification_lock_); | |
| 60 if (result == MOJO_RESULT_CANCELLED) { | |
| 61 // Make sure cancellation is the last notification we dispatch. | |
| 62 DCHECK(!is_cancelled_); | |
| 63 is_cancelled_ = true; | |
| 64 } else if (is_cancelled_) { | |
| 65 return; | |
| 66 } | |
| 67 | |
| 68 // NOTE: This will acquire |watcher_|'s internal lock. It's safe because a | |
| 69 // thread can only enter InvokeCallback() from within a RequestContext | |
| 70 // destructor where no dispatcher locks are held. | |
| 71 watcher_->InvokeWatchCallback(context_, result, state, flags); | |
| 72 } | |
| 73 | |
| 74 Watch::~Watch() {} | |
| 75 | |
| 76 #if DCHECK_IS_ON() | |
| 77 void Watch::AssertWatcherLockAcquired() const { | |
| 78 watcher_->lock_.AssertAcquired(); | |
| 79 } | |
| 80 #endif | |
| 81 | |
| 82 } // namespace edk | |
| 83 } // namespace mojo | |
| OLD | NEW |