OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "ipc/attachment_broker_privileged.h" | 5 #include "ipc/attachment_broker_privileged.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 | 8 |
9 #include "ipc/ipc_channel.h" | 9 #include "ipc/ipc_endpoint.h" |
10 | 10 |
11 namespace IPC { | 11 namespace IPC { |
12 | 12 |
13 AttachmentBrokerPrivileged::AttachmentBrokerPrivileged() {} | 13 AttachmentBrokerPrivileged::AttachmentBrokerPrivileged() {} |
14 | 14 |
15 AttachmentBrokerPrivileged::~AttachmentBrokerPrivileged() {} | 15 AttachmentBrokerPrivileged::~AttachmentBrokerPrivileged() {} |
16 | 16 |
17 void AttachmentBrokerPrivileged::RegisterCommunicationChannel( | 17 void AttachmentBrokerPrivileged::RegisterCommunicationChannel( |
18 Channel* channel) { | 18 Endpoint* endpoint) { |
19 channel->set_attachment_broker_endpoint(true); | 19 endpoint->SetAttachmentBrokerEndpoint(true); |
20 auto it = std::find(channels_.begin(), channels_.end(), channel); | 20 auto it = std::find(endpoints_.begin(), endpoints_.end(), endpoint); |
21 DCHECK(channels_.end() == it); | 21 DCHECK(endpoints_.end() == it); |
22 channels_.push_back(channel); | 22 endpoints_.push_back(endpoint); |
23 } | 23 } |
24 | 24 |
25 void AttachmentBrokerPrivileged::DeregisterCommunicationChannel( | 25 void AttachmentBrokerPrivileged::DeregisterCommunicationChannel( |
26 Channel* channel) { | 26 Endpoint* endpoint) { |
27 auto it = std::find(channels_.begin(), channels_.end(), channel); | 27 auto it = std::find(endpoints_.begin(), endpoints_.end(), endpoint); |
28 if (it != channels_.end()) | 28 if (it != endpoints_.end()) |
29 channels_.erase(it); | 29 endpoints_.erase(it); |
30 } | 30 } |
31 | 31 |
32 Channel* AttachmentBrokerPrivileged::GetChannelWithProcessId( | 32 Sender* AttachmentBrokerPrivileged::GetSenderWithProcessId(base::ProcessId id) { |
33 base::ProcessId id) { | 33 auto it = std::find_if(endpoints_.begin(), endpoints_.end(), |
34 auto it = std::find_if(channels_.begin(), channels_.end(), | 34 [id](Endpoint* c) { return c->GetPeerPID() == id; }); |
35 [id](Channel* c) { return c->GetPeerPID() == id; }); | 35 if (it == endpoints_.end()) |
36 if (it == channels_.end()) | |
37 return nullptr; | 36 return nullptr; |
38 return *it; | 37 return *it; |
39 } | 38 } |
40 | 39 |
41 } // namespace IPC | 40 } // namespace IPC |
OLD | NEW |