morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 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 | |
amistry | d4aa70d | 2016-06-23 07:52:37 | [diff] [blame] | 5 | #include "ipc/ipc_mojo_bootstrap.h" |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 6 | |
avi | 246998d8 | 2015-12-22 02:39:04 | [diff] [blame] | 7 | #include <stdint.h> |
danakj | 03de39b2 | 2016-04-23 04:21:09 | [diff] [blame] | 8 | #include <memory> |
avi | 246998d8 | 2015-12-22 02:39:04 | [diff] [blame] | 9 | |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 10 | #include "base/base_paths.h" |
| 11 | #include "base/files/file.h" |
| 12 | #include "base/message_loop/message_loop.h" |
rockot | cf1d7d0 | 2016-11-22 05:27:27 | [diff] [blame] | 13 | #include "base/run_loop.h" |
gab | f08ccc0 | 2016-05-11 18:51:11 | [diff] [blame] | 14 | #include "base/threading/thread_task_runner_handle.h" |
avi | 246998d8 | 2015-12-22 02:39:04 | [diff] [blame] | 15 | #include "build/build_config.h" |
amistry | d4aa70d | 2016-06-23 07:52:37 | [diff] [blame] | 16 | #include "ipc/ipc.mojom.h" |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 17 | #include "ipc/ipc_test_base.h" |
sammc | e4d0abd | 2016-03-07 22:38:04 | [diff] [blame] | 18 | #include "mojo/edk/embedder/embedder.h" |
| 19 | #include "mojo/edk/test/mojo_test_base.h" |
| 20 | #include "mojo/edk/test/multiprocess_test_helper.h" |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 21 | #include "mojo/public/cpp/bindings/associated_binding.h" |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 22 | |
| 23 | #if defined(OS_POSIX) |
| 24 | #include "base/file_descriptor_posix.h" |
| 25 | #endif |
| 26 | |
| 27 | namespace { |
| 28 | |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 29 | constexpr int32_t kTestServerPid = 42; |
| 30 | constexpr int32_t kTestClientPid = 4242; |
| 31 | |
| 32 | class PeerPidReceiver : public IPC::mojom::Channel { |
| 33 | public: |
| 34 | PeerPidReceiver(IPC::mojom::ChannelAssociatedRequest request, |
| 35 | const base::Closure& on_peer_pid_set) |
| 36 | : binding_(this, std::move(request)), on_peer_pid_set_(on_peer_pid_set) {} |
Chris Watkins | 2d879af | 2017-11-30 02:11:59 | [diff] [blame^] | 37 | ~PeerPidReceiver() override = default; |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 38 | |
| 39 | // mojom::Channel: |
| 40 | void SetPeerPid(int32_t pid) override { |
| 41 | peer_pid_ = pid; |
| 42 | on_peer_pid_set_.Run(); |
| 43 | } |
| 44 | |
Yuzhu Shen | a0a2b36 | 2017-08-25 22:31:39 | [diff] [blame] | 45 | void Receive(base::span<const uint8_t> data, |
Ken Rockot | fd90763 | 2017-09-14 04:23:41 | [diff] [blame] | 46 | base::Optional<std::vector<mojo::native::SerializedHandlePtr>> |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 47 | handles) override {} |
| 48 | |
| 49 | void GetAssociatedInterface( |
| 50 | const std::string& name, |
| 51 | IPC::mojom::GenericInterfaceAssociatedRequest request) override {} |
| 52 | |
| 53 | int32_t peer_pid() const { return peer_pid_; } |
| 54 | |
| 55 | private: |
| 56 | mojo::AssociatedBinding<IPC::mojom::Channel> binding_; |
| 57 | const base::Closure on_peer_pid_set_; |
| 58 | int32_t peer_pid_ = -1; |
| 59 | |
| 60 | DISALLOW_COPY_AND_ASSIGN(PeerPidReceiver); |
| 61 | }; |
| 62 | |
sammc | 57ed9f98 | 2016-03-10 06:28:35 | [diff] [blame] | 63 | class IPCMojoBootstrapTest : public testing::Test { |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 64 | protected: |
sammc | 57ed9f98 | 2016-03-10 06:28:35 | [diff] [blame] | 65 | mojo::edk::test::MultiprocessTestHelper helper_; |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 66 | }; |
| 67 | |
amistry | 6de2ee4f | 2016-05-05 05:12:09 | [diff] [blame] | 68 | TEST_F(IPCMojoBootstrapTest, Connect) { |
sammc | e4d0abd | 2016-03-07 22:38:04 | [diff] [blame] | 69 | base::MessageLoop message_loop; |
danakj | 03de39b2 | 2016-04-23 04:21:09 | [diff] [blame] | 70 | std::unique_ptr<IPC::MojoBootstrap> bootstrap = IPC::MojoBootstrap::Create( |
sammc | 57ed9f98 | 2016-03-10 06:28:35 | [diff] [blame] | 71 | helper_.StartChild("IPCMojoBootstrapTestClient"), |
Hajime Hoshi | a98f110 | 2017-11-20 06:34:35 | [diff] [blame] | 72 | IPC::Channel::MODE_SERVER, base::ThreadTaskRunnerHandle::Get(), |
| 73 | base::ThreadTaskRunnerHandle::Get()); |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 74 | |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 75 | IPC::mojom::ChannelAssociatedPtr sender; |
| 76 | IPC::mojom::ChannelAssociatedRequest receiver; |
| 77 | bootstrap->Connect(&sender, &receiver); |
| 78 | sender->SetPeerPid(kTestServerPid); |
| 79 | |
| 80 | base::RunLoop run_loop; |
| 81 | PeerPidReceiver impl(std::move(receiver), run_loop.QuitClosure()); |
sammc | e4d0abd | 2016-03-07 22:38:04 | [diff] [blame] | 82 | run_loop.Run(); |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 83 | |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 84 | EXPECT_EQ(kTestClientPid, impl.peer_pid()); |
| 85 | |
sammc | 57ed9f98 | 2016-03-10 06:28:35 | [diff] [blame] | 86 | EXPECT_TRUE(helper_.WaitForChildTestShutdown()); |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 87 | } |
| 88 | |
sammc | e4d0abd | 2016-03-07 22:38:04 | [diff] [blame] | 89 | // A long running process that connects to us. |
sammc | 57ed9f98 | 2016-03-10 06:28:35 | [diff] [blame] | 90 | MULTIPROCESS_TEST_MAIN_WITH_SETUP( |
| 91 | IPCMojoBootstrapTestClientTestChildMain, |
| 92 | ::mojo::edk::test::MultiprocessTestHelper::ChildSetup) { |
sammc | e4d0abd | 2016-03-07 22:38:04 | [diff] [blame] | 93 | base::MessageLoop message_loop; |
danakj | 03de39b2 | 2016-04-23 04:21:09 | [diff] [blame] | 94 | std::unique_ptr<IPC::MojoBootstrap> bootstrap = IPC::MojoBootstrap::Create( |
sammc | b0a39f8 | 2016-08-10 06:29:54 | [diff] [blame] | 95 | std::move(mojo::edk::test::MultiprocessTestHelper::primordial_pipe), |
Hajime Hoshi | a98f110 | 2017-11-20 06:34:35 | [diff] [blame] | 96 | IPC::Channel::MODE_CLIENT, base::ThreadTaskRunnerHandle::Get(), |
| 97 | base::ThreadTaskRunnerHandle::Get()); |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 98 | |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 99 | IPC::mojom::ChannelAssociatedPtr sender; |
| 100 | IPC::mojom::ChannelAssociatedRequest receiver; |
| 101 | bootstrap->Connect(&sender, &receiver); |
| 102 | sender->SetPeerPid(kTestClientPid); |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 103 | |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 104 | base::RunLoop run_loop; |
| 105 | PeerPidReceiver impl(std::move(receiver), run_loop.QuitClosure()); |
sammc | e4d0abd | 2016-03-07 22:38:04 | [diff] [blame] | 106 | run_loop.Run(); |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 107 | |
rockot | a628d0b | 2017-02-09 08:40:15 | [diff] [blame] | 108 | EXPECT_EQ(kTestServerPid, impl.peer_pid()); |
| 109 | |
| 110 | return 0; |
morrita | 54f6f80c | 2014-09-23 21:16:00 | [diff] [blame] | 111 | } |
| 112 | |
| 113 | } // namespace |