| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "mojo/public/cpp/bindings/connector.h" | 5 #include "mojo/public/cpp/bindings/connector.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 #include <utility> | 8 #include <utility> |
| 9 | 9 |
| 10 #include "base/bind.h" | 10 #include "base/bind.h" |
| 11 #include "base/location.h" | 11 #include "base/location.h" |
| 12 #include "base/logging.h" | 12 #include "base/logging.h" |
| 13 #include "base/macros.h" | 13 #include "base/macros.h" |
| 14 #include "base/synchronization/lock.h" | 14 #include "base/synchronization/lock.h" |
| 15 #include "mojo/public/cpp/bindings/lib/may_auto_lock.h" |
| 15 #include "mojo/public/cpp/bindings/sync_handle_watcher.h" | 16 #include "mojo/public/cpp/bindings/sync_handle_watcher.h" |
| 16 | 17 |
| 17 namespace mojo { | 18 namespace mojo { |
| 18 | 19 |
| 19 namespace { | |
| 20 | |
| 21 // Similar to base::AutoLock, except that it does nothing if |lock| passed into | |
| 22 // the constructor is null. | |
| 23 class MayAutoLock { | |
| 24 public: | |
| 25 explicit MayAutoLock(base::Lock* lock) : lock_(lock) { | |
| 26 if (lock_) | |
| 27 lock_->Acquire(); | |
| 28 } | |
| 29 | |
| 30 ~MayAutoLock() { | |
| 31 if (lock_) { | |
| 32 lock_->AssertAcquired(); | |
| 33 lock_->Release(); | |
| 34 } | |
| 35 } | |
| 36 | |
| 37 private: | |
| 38 base::Lock* lock_; | |
| 39 DISALLOW_COPY_AND_ASSIGN(MayAutoLock); | |
| 40 }; | |
| 41 | |
| 42 } // namespace | |
| 43 | |
| 44 // ---------------------------------------------------------------------------- | |
| 45 | |
| 46 Connector::Connector(ScopedMessagePipeHandle message_pipe, | 20 Connector::Connector(ScopedMessagePipeHandle message_pipe, |
| 47 ConnectorConfig config, | 21 ConnectorConfig config, |
| 48 scoped_refptr<base::SingleThreadTaskRunner> runner) | 22 scoped_refptr<base::SingleThreadTaskRunner> runner) |
| 49 : message_pipe_(std::move(message_pipe)), | 23 : message_pipe_(std::move(message_pipe)), |
| 50 task_runner_(std::move(runner)), | 24 task_runner_(std::move(runner)), |
| 51 handle_watcher_(task_runner_), | 25 handle_watcher_(task_runner_), |
| 52 lock_(config == MULTI_THREADED_SEND ? new base::Lock : nullptr), | 26 lock_(config == MULTI_THREADED_SEND ? new base::Lock : nullptr), |
| 53 weak_factory_(this) { | 27 weak_factory_(this) { |
| 54 weak_self_ = weak_factory_.GetWeakPtr(); | 28 weak_self_ = weak_factory_.GetWeakPtr(); |
| 55 // Even though we don't have an incoming receiver, we still want to monitor | 29 // Even though we don't have an incoming receiver, we still want to monitor |
| (...skipping 10 matching lines...) Expand all Loading... |
| 66 } | 40 } |
| 67 | 41 |
| 68 DCHECK(thread_checker_.CalledOnValidThread()); | 42 DCHECK(thread_checker_.CalledOnValidThread()); |
| 69 CancelWait(); | 43 CancelWait(); |
| 70 } | 44 } |
| 71 | 45 |
| 72 void Connector::CloseMessagePipe() { | 46 void Connector::CloseMessagePipe() { |
| 73 DCHECK(thread_checker_.CalledOnValidThread()); | 47 DCHECK(thread_checker_.CalledOnValidThread()); |
| 74 | 48 |
| 75 CancelWait(); | 49 CancelWait(); |
| 76 MayAutoLock locker(lock_.get()); | 50 internal::MayAutoLock locker(lock_.get()); |
| 77 message_pipe_.reset(); | 51 message_pipe_.reset(); |
| 78 | 52 |
| 79 base::AutoLock lock(connected_lock_); | 53 base::AutoLock lock(connected_lock_); |
| 80 connected_ = false; | 54 connected_ = false; |
| 81 } | 55 } |
| 82 | 56 |
| 83 ScopedMessagePipeHandle Connector::PassMessagePipe() { | 57 ScopedMessagePipeHandle Connector::PassMessagePipe() { |
| 84 DCHECK(thread_checker_.CalledOnValidThread()); | 58 DCHECK(thread_checker_.CalledOnValidThread()); |
| 85 | 59 |
| 86 CancelWait(); | 60 CancelWait(); |
| 87 MayAutoLock locker(lock_.get()); | 61 internal::MayAutoLock locker(lock_.get()); |
| 88 ScopedMessagePipeHandle message_pipe = std::move(message_pipe_); | 62 ScopedMessagePipeHandle message_pipe = std::move(message_pipe_); |
| 89 | 63 |
| 90 base::AutoLock lock(connected_lock_); | 64 base::AutoLock lock(connected_lock_); |
| 91 connected_ = false; | 65 connected_ = false; |
| 92 return message_pipe; | 66 return message_pipe; |
| 93 } | 67 } |
| 94 | 68 |
| 95 void Connector::RaiseError() { | 69 void Connector::RaiseError() { |
| 96 DCHECK(thread_checker_.CalledOnValidThread()); | 70 DCHECK(thread_checker_.CalledOnValidThread()); |
| 97 | 71 |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 142 | 116 |
| 143 bool Connector::Accept(Message* message) { | 117 bool Connector::Accept(Message* message) { |
| 144 DCHECK(lock_ || thread_checker_.CalledOnValidThread()); | 118 DCHECK(lock_ || thread_checker_.CalledOnValidThread()); |
| 145 | 119 |
| 146 // It shouldn't hurt even if |error_| may be changed by a different thread at | 120 // It shouldn't hurt even if |error_| may be changed by a different thread at |
| 147 // the same time. The outcome is that we may write into |message_pipe_| after | 121 // the same time. The outcome is that we may write into |message_pipe_| after |
| 148 // encountering an error, which should be fine. | 122 // encountering an error, which should be fine. |
| 149 if (error_) | 123 if (error_) |
| 150 return false; | 124 return false; |
| 151 | 125 |
| 152 MayAutoLock locker(lock_.get()); | 126 internal::MayAutoLock locker(lock_.get()); |
| 153 | 127 |
| 154 if (!message_pipe_.is_valid() || drop_writes_) | 128 if (!message_pipe_.is_valid() || drop_writes_) |
| 155 return true; | 129 return true; |
| 156 | 130 |
| 157 MojoResult rv = | 131 MojoResult rv = |
| 158 WriteMessageNew(message_pipe_.get(), message->TakeMojoMessage(), | 132 WriteMessageNew(message_pipe_.get(), message->TakeMojoMessage(), |
| 159 MOJO_WRITE_MESSAGE_FLAG_NONE); | 133 MOJO_WRITE_MESSAGE_FLAG_NONE); |
| 160 | 134 |
| 161 switch (rv) { | 135 switch (rv) { |
| 162 case MOJO_RESULT_OK: | 136 case MOJO_RESULT_OK: |
| (...skipping 161 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 324 // receiving messages. We need to wait until the user starts receiving | 298 // receiving messages. We need to wait until the user starts receiving |
| 325 // messages again. | 299 // messages again. |
| 326 force_async_handler = true; | 300 force_async_handler = true; |
| 327 } | 301 } |
| 328 | 302 |
| 329 if (!force_pipe_reset && force_async_handler) | 303 if (!force_pipe_reset && force_async_handler) |
| 330 force_pipe_reset = true; | 304 force_pipe_reset = true; |
| 331 | 305 |
| 332 if (force_pipe_reset) { | 306 if (force_pipe_reset) { |
| 333 CancelWait(); | 307 CancelWait(); |
| 334 MayAutoLock locker(lock_.get()); | 308 internal::MayAutoLock locker(lock_.get()); |
| 335 message_pipe_.reset(); | 309 message_pipe_.reset(); |
| 336 MessagePipe dummy_pipe; | 310 MessagePipe dummy_pipe; |
| 337 message_pipe_ = std::move(dummy_pipe.handle0); | 311 message_pipe_ = std::move(dummy_pipe.handle0); |
| 338 } else { | 312 } else { |
| 339 CancelWait(); | 313 CancelWait(); |
| 340 } | 314 } |
| 341 | 315 |
| 342 if (force_async_handler) { | 316 if (force_async_handler) { |
| 343 if (!paused_) | 317 if (!paused_) |
| 344 WaitToReadMore(); | 318 WaitToReadMore(); |
| 345 } else { | 319 } else { |
| 346 error_ = true; | 320 error_ = true; |
| 347 if (!connection_error_handler_.is_null()) | 321 if (!connection_error_handler_.is_null()) |
| 348 connection_error_handler_.Run(); | 322 connection_error_handler_.Run(); |
| 349 } | 323 } |
| 350 } | 324 } |
| 351 | 325 |
| 352 void Connector::EnsureSyncWatcherExists() { | 326 void Connector::EnsureSyncWatcherExists() { |
| 353 if (sync_watcher_) | 327 if (sync_watcher_) |
| 354 return; | 328 return; |
| 355 sync_watcher_.reset(new SyncHandleWatcher( | 329 sync_watcher_.reset(new SyncHandleWatcher( |
| 356 message_pipe_.get(), MOJO_HANDLE_SIGNAL_READABLE, | 330 message_pipe_.get(), MOJO_HANDLE_SIGNAL_READABLE, |
| 357 base::Bind(&Connector::OnSyncHandleWatcherHandleReady, | 331 base::Bind(&Connector::OnSyncHandleWatcherHandleReady, |
| 358 base::Unretained(this)))); | 332 base::Unretained(this)))); |
| 359 } | 333 } |
| 360 | 334 |
| 361 } // namespace mojo | 335 } // namespace mojo |
| OLD | NEW |