| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "media/remoting/rpc/rpc_broker.h" | 5 #include "media/remoting/rpc_broker.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/logging.h" | 9 #include "base/logging.h" |
| 10 #include "media/base/bind_to_current_loop.h" | 10 #include "media/base/bind_to_current_loop.h" |
| 11 | 11 |
| 12 namespace media { | 12 namespace media { |
| 13 namespace remoting { | 13 namespace remoting { |
| 14 | 14 |
| 15 namespace { | 15 namespace { |
| (...skipping 19 matching lines...) Expand all Loading... |
| 35 default: | 35 default: |
| 36 out << ", rpc_oneof=" << message.rpc_oneof_case(); | 36 out << ", rpc_oneof=" << message.rpc_oneof_case(); |
| 37 break; | 37 break; |
| 38 } | 38 } |
| 39 return out; | 39 return out; |
| 40 } | 40 } |
| 41 | 41 |
| 42 } // namespace | 42 } // namespace |
| 43 | 43 |
| 44 RpcBroker::RpcBroker(const SendMessageCallback& send_message_cb) | 44 RpcBroker::RpcBroker(const SendMessageCallback& send_message_cb) |
| 45 : next_handle_(kReceiverHandle + 1), | 45 : next_handle_(kFirstHandle), |
| 46 send_message_cb_(send_message_cb), | 46 send_message_cb_(send_message_cb), |
| 47 weak_factory_(this) {} | 47 weak_factory_(this) {} |
| 48 | 48 |
| 49 RpcBroker::~RpcBroker() { | 49 RpcBroker::~RpcBroker() { |
| 50 DCHECK(thread_checker_.CalledOnValidThread()); | 50 DCHECK(thread_checker_.CalledOnValidThread()); |
| 51 receive_callbacks_.clear(); | 51 receive_callbacks_.clear(); |
| 52 } | 52 } |
| 53 | 53 |
| 54 int RpcBroker::GetUniqueHandle() { | 54 int RpcBroker::GetUniqueHandle() { |
| 55 DCHECK(thread_checker_.CalledOnValidThread()); | 55 DCHECK(thread_checker_.CalledOnValidThread()); |
| (...skipping 15 matching lines...) Expand all Loading... |
| 71 receive_callbacks_.erase(handle); | 71 receive_callbacks_.erase(handle); |
| 72 } | 72 } |
| 73 | 73 |
| 74 void RpcBroker::ProcessMessageFromRemote( | 74 void RpcBroker::ProcessMessageFromRemote( |
| 75 std::unique_ptr<pb::RpcMessage> message) { | 75 std::unique_ptr<pb::RpcMessage> message) { |
| 76 DCHECK(message); | 76 DCHECK(message); |
| 77 DCHECK(thread_checker_.CalledOnValidThread()); | 77 DCHECK(thread_checker_.CalledOnValidThread()); |
| 78 VLOG(3) << __func__ << ": " << *message; | 78 VLOG(3) << __func__ << ": " << *message; |
| 79 const auto entry = receive_callbacks_.find(message->handle()); | 79 const auto entry = receive_callbacks_.find(message->handle()); |
| 80 if (entry == receive_callbacks_.end()) { | 80 if (entry == receive_callbacks_.end()) { |
| 81 LOG(ERROR) << "unregistered handle: " << message->handle(); | 81 VLOG(1) << "unregistered handle: " << message->handle(); |
| 82 return; | 82 return; |
| 83 } | 83 } |
| 84 entry->second.Run(std::move(message)); | 84 entry->second.Run(std::move(message)); |
| 85 } | 85 } |
| 86 | 86 |
| 87 void RpcBroker::SendMessageToRemote(std::unique_ptr<pb::RpcMessage> message) { | 87 void RpcBroker::SendMessageToRemote(std::unique_ptr<pb::RpcMessage> message) { |
| 88 DCHECK(thread_checker_.CalledOnValidThread()); | 88 DCHECK(thread_checker_.CalledOnValidThread()); |
| 89 DCHECK(message); | 89 DCHECK(message); |
| 90 VLOG(3) << __func__ << ": " << *message; | 90 VLOG(3) << __func__ << ": " << *message; |
| 91 std::unique_ptr<std::vector<uint8_t>> serialized_message( | 91 std::unique_ptr<std::vector<uint8_t>> serialized_message( |
| 92 new std::vector<uint8_t>(message->ByteSize())); | 92 new std::vector<uint8_t>(message->ByteSize())); |
| 93 CHECK(message->SerializeToArray(serialized_message->data(), | 93 CHECK(message->SerializeToArray(serialized_message->data(), |
| 94 serialized_message->size())); | 94 serialized_message->size())); |
| 95 send_message_cb_.Run(std::move(serialized_message)); | 95 send_message_cb_.Run(std::move(serialized_message)); |
| 96 } | 96 } |
| 97 | 97 |
| 98 base::WeakPtr<RpcBroker> RpcBroker::GetWeakPtr() { | 98 base::WeakPtr<RpcBroker> RpcBroker::GetWeakPtr() { |
| 99 return weak_factory_.GetWeakPtr(); | 99 return weak_factory_.GetWeakPtr(); |
| 100 } | 100 } |
| 101 | 101 |
| 102 void RpcBroker::SetMessageCallbackForTesting( | 102 void RpcBroker::SetMessageCallbackForTesting( |
| 103 const SendMessageCallback& send_message_cb) { | 103 const SendMessageCallback& send_message_cb) { |
| 104 DCHECK(thread_checker_.CalledOnValidThread()); | 104 DCHECK(thread_checker_.CalledOnValidThread()); |
| 105 send_message_cb_ = send_message_cb; | 105 send_message_cb_ = send_message_cb; |
| 106 } | 106 } |
| 107 | 107 |
| 108 } // namespace remoting | 108 } // namespace remoting |
| 109 } // namespace media | 109 } // namespace media |
| OLD | NEW |