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 |