| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "ipc/mojo/ipc_mojo_bootstrap.h" | |
| 6 | |
| 7 #include <stdint.h> | |
| 8 #include <utility> | |
| 9 | |
| 10 #include "base/logging.h" | |
| 11 #include "base/macros.h" | |
| 12 #include "base/memory/ptr_util.h" | |
| 13 #include "base/process/process_handle.h" | |
| 14 #include "build/build_config.h" | |
| 15 #include "ipc/ipc_message_utils.h" | |
| 16 #include "ipc/ipc_platform_file.h" | |
| 17 #include "mojo/public/cpp/bindings/binding.h" | |
| 18 | |
| 19 namespace IPC { | |
| 20 | |
| 21 namespace { | |
| 22 | |
| 23 // MojoBootstrap for the server process. You should create the instance | |
| 24 // using MojoBootstrap::Create(). | |
| 25 class MojoServerBootstrap : public MojoBootstrap { | |
| 26 public: | |
| 27 MojoServerBootstrap(); | |
| 28 | |
| 29 private: | |
| 30 // MojoBootstrap implementation. | |
| 31 void Connect() override; | |
| 32 | |
| 33 void OnInitDone(int32_t peer_pid); | |
| 34 | |
| 35 mojom::BootstrapPtr bootstrap_; | |
| 36 IPC::mojom::ChannelAssociatedPtrInfo send_channel_; | |
| 37 IPC::mojom::ChannelAssociatedRequest receive_channel_request_; | |
| 38 | |
| 39 DISALLOW_COPY_AND_ASSIGN(MojoServerBootstrap); | |
| 40 }; | |
| 41 | |
| 42 MojoServerBootstrap::MojoServerBootstrap() = default; | |
| 43 | |
| 44 void MojoServerBootstrap::Connect() { | |
| 45 DCHECK_EQ(state(), STATE_INITIALIZED); | |
| 46 | |
| 47 bootstrap_.Bind(mojom::BootstrapPtrInfo(TakeHandle(), 0)); | |
| 48 bootstrap_.set_connection_error_handler( | |
| 49 base::Bind(&MojoServerBootstrap::Fail, base::Unretained(this))); | |
| 50 | |
| 51 IPC::mojom::ChannelAssociatedRequest send_channel_request; | |
| 52 IPC::mojom::ChannelAssociatedPtrInfo receive_channel; | |
| 53 | |
| 54 bootstrap_.associated_group()->CreateAssociatedInterface( | |
| 55 mojo::AssociatedGroup::WILL_PASS_REQUEST, &send_channel_, | |
| 56 &send_channel_request); | |
| 57 bootstrap_.associated_group()->CreateAssociatedInterface( | |
| 58 mojo::AssociatedGroup::WILL_PASS_PTR, &receive_channel, | |
| 59 &receive_channel_request_); | |
| 60 | |
| 61 bootstrap_->Init( | |
| 62 std::move(send_channel_request), std::move(receive_channel), | |
| 63 GetSelfPID(), | |
| 64 base::Bind(&MojoServerBootstrap::OnInitDone, base::Unretained(this))); | |
| 65 | |
| 66 set_state(STATE_WAITING_ACK); | |
| 67 } | |
| 68 | |
| 69 void MojoServerBootstrap::OnInitDone(int32_t peer_pid) { | |
| 70 if (state() != STATE_WAITING_ACK) { | |
| 71 set_state(STATE_ERROR); | |
| 72 LOG(ERROR) << "Got inconsistent message from client."; | |
| 73 return; | |
| 74 } | |
| 75 | |
| 76 set_state(STATE_READY); | |
| 77 bootstrap_.set_connection_error_handler(mojo::Closure()); | |
| 78 delegate()->OnPipesAvailable(std::move(send_channel_), | |
| 79 std::move(receive_channel_request_), peer_pid); | |
| 80 } | |
| 81 | |
| 82 // MojoBootstrap for client processes. You should create the instance | |
| 83 // using MojoBootstrap::Create(). | |
| 84 class MojoClientBootstrap : public MojoBootstrap, public mojom::Bootstrap { | |
| 85 public: | |
| 86 MojoClientBootstrap(); | |
| 87 | |
| 88 private: | |
| 89 // MojoBootstrap implementation. | |
| 90 void Connect() override; | |
| 91 | |
| 92 // mojom::Bootstrap implementation. | |
| 93 void Init(mojom::ChannelAssociatedRequest receive_channel, | |
| 94 mojom::ChannelAssociatedPtrInfo send_channel, | |
| 95 int32_t peer_pid, | |
| 96 const mojo::Callback<void(int32_t)>& callback) override; | |
| 97 | |
| 98 mojo::Binding<mojom::Bootstrap> binding_; | |
| 99 | |
| 100 DISALLOW_COPY_AND_ASSIGN(MojoClientBootstrap); | |
| 101 }; | |
| 102 | |
| 103 MojoClientBootstrap::MojoClientBootstrap() : binding_(this) {} | |
| 104 | |
| 105 void MojoClientBootstrap::Connect() { | |
| 106 binding_.Bind(TakeHandle()); | |
| 107 binding_.set_connection_error_handler( | |
| 108 base::Bind(&MojoClientBootstrap::Fail, base::Unretained(this))); | |
| 109 } | |
| 110 | |
| 111 void MojoClientBootstrap::Init(mojom::ChannelAssociatedRequest receive_channel, | |
| 112 mojom::ChannelAssociatedPtrInfo send_channel, | |
| 113 int32_t peer_pid, | |
| 114 const mojo::Callback<void(int32_t)>& callback) { | |
| 115 callback.Run(GetSelfPID()); | |
| 116 set_state(STATE_READY); | |
| 117 binding_.set_connection_error_handler(mojo::Closure()); | |
| 118 delegate()->OnPipesAvailable(std::move(send_channel), | |
| 119 std::move(receive_channel), peer_pid); | |
| 120 } | |
| 121 | |
| 122 } // namespace | |
| 123 | |
| 124 // MojoBootstrap | |
| 125 | |
| 126 // static | |
| 127 std::unique_ptr<MojoBootstrap> MojoBootstrap::Create( | |
| 128 mojo::ScopedMessagePipeHandle handle, | |
| 129 Channel::Mode mode, | |
| 130 Delegate* delegate) { | |
| 131 CHECK(mode == Channel::MODE_CLIENT || mode == Channel::MODE_SERVER); | |
| 132 std::unique_ptr<MojoBootstrap> self = | |
| 133 mode == Channel::MODE_CLIENT | |
| 134 ? std::unique_ptr<MojoBootstrap>(new MojoClientBootstrap) | |
| 135 : std::unique_ptr<MojoBootstrap>(new MojoServerBootstrap); | |
| 136 | |
| 137 self->Init(std::move(handle), delegate); | |
| 138 return self; | |
| 139 } | |
| 140 | |
| 141 MojoBootstrap::MojoBootstrap() : delegate_(NULL), state_(STATE_INITIALIZED) { | |
| 142 } | |
| 143 | |
| 144 MojoBootstrap::~MojoBootstrap() {} | |
| 145 | |
| 146 void MojoBootstrap::Init(mojo::ScopedMessagePipeHandle handle, | |
| 147 Delegate* delegate) { | |
| 148 handle_ = std::move(handle); | |
| 149 delegate_ = delegate; | |
| 150 } | |
| 151 | |
| 152 base::ProcessId MojoBootstrap::GetSelfPID() const { | |
| 153 #if defined(OS_LINUX) | |
| 154 if (int global_pid = Channel::GetGlobalPid()) | |
| 155 return global_pid; | |
| 156 #endif // OS_LINUX | |
| 157 return base::GetCurrentProcId(); | |
| 158 } | |
| 159 | |
| 160 void MojoBootstrap::Fail() { | |
| 161 set_state(STATE_ERROR); | |
| 162 delegate()->OnBootstrapError(); | |
| 163 } | |
| 164 | |
| 165 bool MojoBootstrap::HasFailed() const { | |
| 166 return state() == STATE_ERROR; | |
| 167 } | |
| 168 | |
| 169 mojo::ScopedMessagePipeHandle MojoBootstrap::TakeHandle() { | |
| 170 return std::move(handle_); | |
| 171 } | |
| 172 | |
| 173 } // namespace IPC | |
| OLD | NEW |