OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/mojo/ipc_mojo_bootstrap.h" | 5 #include "ipc/mojo/ipc_mojo_bootstrap.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/process/process_handle.h" | 10 #include "base/process/process_handle.h" |
(...skipping 13 matching lines...) Expand all Loading... |
24 | 24 |
25 private: | 25 private: |
26 void SendClientPipe(int32_t peer_pid); | 26 void SendClientPipe(int32_t peer_pid); |
27 | 27 |
28 // Listener implementations | 28 // Listener implementations |
29 bool OnMessageReceived(const Message& message) override; | 29 bool OnMessageReceived(const Message& message) override; |
30 void OnChannelConnected(int32_t peer_pid) override; | 30 void OnChannelConnected(int32_t peer_pid) override; |
31 | 31 |
32 mojo::embedder::ScopedPlatformHandle server_pipe_; | 32 mojo::embedder::ScopedPlatformHandle server_pipe_; |
33 bool connected_; | 33 bool connected_; |
| 34 int32_t peer_pid_; |
34 | 35 |
35 DISALLOW_COPY_AND_ASSIGN(MojoServerBootstrap); | 36 DISALLOW_COPY_AND_ASSIGN(MojoServerBootstrap); |
36 }; | 37 }; |
37 | 38 |
38 MojoServerBootstrap::MojoServerBootstrap() : connected_(false) { | 39 MojoServerBootstrap::MojoServerBootstrap() : connected_(false), peer_pid_(0) { |
39 } | 40 } |
40 | 41 |
41 void MojoServerBootstrap::SendClientPipe(int32_t peer_pid) { | 42 void MojoServerBootstrap::SendClientPipe(int32_t peer_pid) { |
42 DCHECK_EQ(state(), STATE_INITIALIZED); | 43 DCHECK_EQ(state(), STATE_INITIALIZED); |
43 DCHECK(connected_); | 44 DCHECK(connected_); |
44 | 45 |
45 mojo::embedder::PlatformChannelPair channel_pair; | 46 mojo::embedder::PlatformChannelPair channel_pair; |
46 server_pipe_ = channel_pair.PassServerHandle(); | 47 server_pipe_ = channel_pair.PassServerHandle(); |
47 | 48 |
48 base::Process peer_process = | 49 base::Process peer_process = |
(...skipping 22 matching lines...) Expand all Loading... |
71 scoped_ptr<Message> message(new Message()); | 72 scoped_ptr<Message> message(new Message()); |
72 ParamTraits<PlatformFileForTransit>::Write(message.get(), client_pipe); | 73 ParamTraits<PlatformFileForTransit>::Write(message.get(), client_pipe); |
73 Send(message.release()); | 74 Send(message.release()); |
74 | 75 |
75 set_state(STATE_WAITING_ACK); | 76 set_state(STATE_WAITING_ACK); |
76 } | 77 } |
77 | 78 |
78 void MojoServerBootstrap::OnChannelConnected(int32_t peer_pid) { | 79 void MojoServerBootstrap::OnChannelConnected(int32_t peer_pid) { |
79 DCHECK_EQ(state(), STATE_INITIALIZED); | 80 DCHECK_EQ(state(), STATE_INITIALIZED); |
80 connected_ = true; | 81 connected_ = true; |
| 82 peer_pid_ = peer_pid; |
81 SendClientPipe(peer_pid); | 83 SendClientPipe(peer_pid); |
82 } | 84 } |
83 | 85 |
84 bool MojoServerBootstrap::OnMessageReceived(const Message&) { | 86 bool MojoServerBootstrap::OnMessageReceived(const Message&) { |
85 if (state() != STATE_WAITING_ACK) { | 87 if (state() != STATE_WAITING_ACK) { |
86 set_state(STATE_ERROR); | 88 set_state(STATE_ERROR); |
87 LOG(ERROR) << "Got inconsistent message from client."; | 89 LOG(ERROR) << "Got inconsistent message from client."; |
88 return false; | 90 return false; |
89 } | 91 } |
90 | 92 |
91 set_state(STATE_READY); | 93 set_state(STATE_READY); |
92 CHECK(server_pipe_.is_valid()); | 94 CHECK(server_pipe_.is_valid()); |
93 delegate()->OnPipeAvailable( | 95 delegate()->OnPipeAvailable( |
94 mojo::embedder::ScopedPlatformHandle(server_pipe_.release())); | 96 mojo::embedder::ScopedPlatformHandle(server_pipe_.release()), peer_pid_); |
95 | 97 |
96 return true; | 98 return true; |
97 } | 99 } |
98 | 100 |
99 // MojoBootstrap for client processes. You should create the instance | 101 // MojoBootstrap for client processes. You should create the instance |
100 // using MojoBootstrap::Create(). | 102 // using MojoBootstrap::Create(). |
101 class MojoClientBootstrap : public MojoBootstrap { | 103 class MojoClientBootstrap : public MojoBootstrap { |
102 public: | 104 public: |
103 MojoClientBootstrap(); | 105 MojoClientBootstrap(); |
104 | 106 |
105 private: | 107 private: |
106 // Listener implementations | 108 // Listener implementations |
107 bool OnMessageReceived(const Message& message) override; | 109 bool OnMessageReceived(const Message& message) override; |
108 void OnChannelConnected(int32_t peer_pid) override; | 110 void OnChannelConnected(int32_t peer_pid) override; |
109 | 111 |
| 112 int32 peer_pid_; |
| 113 |
110 DISALLOW_COPY_AND_ASSIGN(MojoClientBootstrap); | 114 DISALLOW_COPY_AND_ASSIGN(MojoClientBootstrap); |
111 }; | 115 }; |
112 | 116 |
113 MojoClientBootstrap::MojoClientBootstrap() { | 117 MojoClientBootstrap::MojoClientBootstrap() : peer_pid_(0) { |
114 } | 118 } |
115 | 119 |
116 bool MojoClientBootstrap::OnMessageReceived(const Message& message) { | 120 bool MojoClientBootstrap::OnMessageReceived(const Message& message) { |
117 if (state() != STATE_INITIALIZED) { | 121 if (state() != STATE_INITIALIZED) { |
118 set_state(STATE_ERROR); | 122 set_state(STATE_ERROR); |
119 LOG(ERROR) << "Got inconsistent message from server."; | 123 LOG(ERROR) << "Got inconsistent message from server."; |
120 return false; | 124 return false; |
121 } | 125 } |
122 | 126 |
123 PlatformFileForTransit pipe; | 127 PlatformFileForTransit pipe; |
124 base::PickleIterator iter(message); | 128 base::PickleIterator iter(message); |
125 if (!ParamTraits<PlatformFileForTransit>::Read(&message, &iter, &pipe)) { | 129 if (!ParamTraits<PlatformFileForTransit>::Read(&message, &iter, &pipe)) { |
126 LOG(WARNING) << "Failed to read a file handle from bootstrap channel."; | 130 LOG(WARNING) << "Failed to read a file handle from bootstrap channel."; |
127 message.set_dispatch_error(); | 131 message.set_dispatch_error(); |
128 return false; | 132 return false; |
129 } | 133 } |
130 | 134 |
131 // Sends ACK back. | 135 // Sends ACK back. |
132 Send(new Message()); | 136 Send(new Message()); |
133 set_state(STATE_READY); | 137 set_state(STATE_READY); |
134 delegate()->OnPipeAvailable( | 138 delegate()->OnPipeAvailable( |
135 mojo::embedder::ScopedPlatformHandle(mojo::embedder::PlatformHandle( | 139 mojo::embedder::ScopedPlatformHandle(mojo::embedder::PlatformHandle( |
136 PlatformFileForTransitToPlatformFile(pipe)))); | 140 PlatformFileForTransitToPlatformFile(pipe))), peer_pid_); |
137 | 141 |
138 return true; | 142 return true; |
139 } | 143 } |
140 | 144 |
141 void MojoClientBootstrap::OnChannelConnected(int32_t peer_pid) { | 145 void MojoClientBootstrap::OnChannelConnected(int32_t peer_pid) { |
| 146 peer_pid_ = peer_pid; |
142 } | 147 } |
143 | 148 |
144 } // namespace | 149 } // namespace |
145 | 150 |
146 // MojoBootstrap | 151 // MojoBootstrap |
147 | 152 |
148 // static | 153 // static |
149 scoped_ptr<MojoBootstrap> MojoBootstrap::Create(ChannelHandle handle, | 154 scoped_ptr<MojoBootstrap> MojoBootstrap::Create(ChannelHandle handle, |
150 Channel::Mode mode, | 155 Channel::Mode mode, |
151 Delegate* delegate) { | 156 Delegate* delegate) { |
(...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
208 int MojoBootstrap::GetClientFileDescriptor() const { | 213 int MojoBootstrap::GetClientFileDescriptor() const { |
209 return channel_->GetClientFileDescriptor(); | 214 return channel_->GetClientFileDescriptor(); |
210 } | 215 } |
211 | 216 |
212 base::ScopedFD MojoBootstrap::TakeClientFileDescriptor() { | 217 base::ScopedFD MojoBootstrap::TakeClientFileDescriptor() { |
213 return channel_->TakeClientFileDescriptor(); | 218 return channel_->TakeClientFileDescriptor(); |
214 } | 219 } |
215 #endif // defined(OS_POSIX) && !defined(OS_NACL) | 220 #endif // defined(OS_POSIX) && !defined(OS_NACL) |
216 | 221 |
217 } // namespace IPC | 222 } // namespace IPC |
OLD | NEW |