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/ipc_channel_mojo.h" | 5 #include "ipc/ipc_channel_mojo.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 #include <stdint.h> | 8 #include <stdint.h> |
9 | |
10 #include <memory> | 9 #include <memory> |
11 #include <utility> | 10 #include <utility> |
12 | 11 |
13 #include "base/base_paths.h" | 12 #include "base/base_paths.h" |
14 #include "base/files/file.h" | 13 #include "base/files/file.h" |
15 #include "base/files/scoped_temp_dir.h" | 14 #include "base/files/scoped_temp_dir.h" |
16 #include "base/location.h" | 15 #include "base/location.h" |
17 #include "base/path_service.h" | 16 #include "base/path_service.h" |
18 #include "base/pickle.h" | 17 #include "base/pickle.h" |
19 #include "base/run_loop.h" | 18 #include "base/run_loop.h" |
20 #include "base/single_thread_task_runner.h" | 19 #include "base/single_thread_task_runner.h" |
21 #include "base/strings/stringprintf.h" | |
22 #include "base/test/test_io_thread.h" | 20 #include "base/test/test_io_thread.h" |
23 #include "base/test/test_timeouts.h" | 21 #include "base/test/test_timeouts.h" |
24 #include "base/threading/thread.h" | 22 #include "base/threading/thread.h" |
25 #include "base/threading/thread_task_runner_handle.h" | 23 #include "base/threading/thread_task_runner_handle.h" |
26 #include "build/build_config.h" | 24 #include "build/build_config.h" |
27 #include "ipc/ipc_message.h" | 25 #include "ipc/ipc_message.h" |
28 #include "ipc/ipc_mojo_handle_attachment.h" | 26 #include "ipc/ipc_mojo_handle_attachment.h" |
29 #include "ipc/ipc_mojo_message_helper.h" | 27 #include "ipc/ipc_mojo_message_helper.h" |
30 #include "ipc/ipc_mojo_param_traits.h" | 28 #include "ipc/ipc_mojo_param_traits.h" |
31 #include "ipc/ipc_test.mojom.h" | |
32 #include "ipc/ipc_test_base.h" | 29 #include "ipc/ipc_test_base.h" |
33 #include "ipc/ipc_test_channel_listener.h" | 30 #include "ipc/ipc_test_channel_listener.h" |
34 #include "mojo/edk/test/mojo_test_base.h" | 31 #include "mojo/edk/test/mojo_test_base.h" |
35 #include "mojo/edk/test/multiprocess_test_helper.h" | 32 #include "mojo/edk/test/multiprocess_test_helper.h" |
36 #include "testing/gtest/include/gtest/gtest.h" | 33 #include "testing/gtest/include/gtest/gtest.h" |
37 | 34 |
38 #if defined(OS_POSIX) | 35 #if defined(OS_POSIX) |
39 #include "base/file_descriptor_posix.h" | 36 #include "base/file_descriptor_posix.h" |
40 #include "ipc/ipc_platform_file_attachment_posix.h" | 37 #include "ipc/ipc_platform_file_attachment_posix.h" |
41 #endif | 38 #endif |
(...skipping 14 matching lines...) Expand all Loading... |
56 test.Main(); \ | 53 test.Main(); \ |
57 return (::testing::Test::HasFatalFailure() || \ | 54 return (::testing::Test::HasFatalFailure() || \ |
58 ::testing::Test::HasNonfatalFailure()) \ | 55 ::testing::Test::HasNonfatalFailure()) \ |
59 ? 1 \ | 56 ? 1 \ |
60 : 0; \ | 57 : 0; \ |
61 } \ | 58 } \ |
62 void client_name##_MainFixture::Main() | 59 void client_name##_MainFixture::Main() |
63 | 60 |
64 namespace { | 61 namespace { |
65 | 62 |
66 void SendString(IPC::Sender* sender, const std::string& str) { | |
67 IPC::Message* message = new IPC::Message(0, 2, IPC::Message::PRIORITY_NORMAL); | |
68 message->WriteString(str); | |
69 ASSERT_TRUE(sender->Send(message)); | |
70 } | |
71 | |
72 class ListenerThatExpectsOK : public IPC::Listener { | 63 class ListenerThatExpectsOK : public IPC::Listener { |
73 public: | 64 public: |
74 ListenerThatExpectsOK() : received_ok_(false) {} | 65 ListenerThatExpectsOK() : received_ok_(false) {} |
75 | 66 |
76 ~ListenerThatExpectsOK() override {} | 67 ~ListenerThatExpectsOK() override {} |
77 | 68 |
78 bool OnMessageReceived(const IPC::Message& message) override { | 69 bool OnMessageReceived(const IPC::Message& message) override { |
79 base::PickleIterator iter(message); | 70 base::PickleIterator iter(message); |
80 std::string should_be_ok; | 71 std::string should_be_ok; |
81 EXPECT_TRUE(iter.ReadString(&should_be_ok)); | 72 EXPECT_TRUE(iter.ReadString(&should_be_ok)); |
82 EXPECT_EQ(should_be_ok, "OK"); | 73 EXPECT_EQ(should_be_ok, "OK"); |
83 received_ok_ = true; | 74 received_ok_ = true; |
84 base::MessageLoop::current()->QuitWhenIdle(); | 75 base::MessageLoop::current()->QuitWhenIdle(); |
85 return true; | 76 return true; |
86 } | 77 } |
87 | 78 |
88 void OnChannelError() override { | 79 void OnChannelError() override { |
89 // The connection should be healthy while the listener is waiting | 80 // The connection should be healthy while the listener is waiting |
90 // message. An error can occur after that because the peer | 81 // message. An error can occur after that because the peer |
91 // process dies. | 82 // process dies. |
92 DCHECK(received_ok_); | 83 DCHECK(received_ok_); |
93 } | 84 } |
94 | 85 |
95 static void SendOK(IPC::Sender* sender) { SendString(sender, "OK"); } | 86 static void SendOK(IPC::Sender* sender) { |
| 87 IPC::Message* message = |
| 88 new IPC::Message(0, 2, IPC::Message::PRIORITY_NORMAL); |
| 89 message->WriteString(std::string("OK")); |
| 90 ASSERT_TRUE(sender->Send(message)); |
| 91 } |
96 | 92 |
97 private: | 93 private: |
98 bool received_ok_; | 94 bool received_ok_; |
99 }; | 95 }; |
100 | 96 |
101 class ChannelClient { | 97 class ChannelClient { |
102 public: | 98 public: |
103 void Init(mojo::ScopedMessagePipeHandle handle) { | 99 void Init(mojo::ScopedMessagePipeHandle handle) { |
104 handle_ = std::move(handle); | 100 handle_ = std::move(handle); |
105 } | 101 } |
(...skipping 15 matching lines...) Expand all Loading... |
121 IPC::ChannelMojo* channel() const { return channel_.get(); } | 117 IPC::ChannelMojo* channel() const { return channel_.get(); } |
122 | 118 |
123 private: | 119 private: |
124 base::MessageLoopForIO main_message_loop_; | 120 base::MessageLoopForIO main_message_loop_; |
125 mojo::ScopedMessagePipeHandle handle_; | 121 mojo::ScopedMessagePipeHandle handle_; |
126 std::unique_ptr<IPC::ChannelMojo> channel_; | 122 std::unique_ptr<IPC::ChannelMojo> channel_; |
127 }; | 123 }; |
128 | 124 |
129 class IPCChannelMojoTest : public testing::Test { | 125 class IPCChannelMojoTest : public testing::Test { |
130 public: | 126 public: |
131 IPCChannelMojoTest() {} | 127 IPCChannelMojoTest() : io_thread_(base::TestIOThread::Mode::kAutoStart) {} |
132 | 128 |
133 void TearDown() override { base::RunLoop().RunUntilIdle(); } | 129 void TearDown() override { base::RunLoop().RunUntilIdle(); } |
134 | 130 |
135 void InitWithMojo(const std::string& test_client_name) { | 131 void InitWithMojo(const std::string& test_client_name) { |
136 handle_ = helper_.StartChild(test_client_name); | 132 handle_ = helper_.StartChild(test_client_name); |
137 } | 133 } |
138 | 134 |
139 void CreateChannel(IPC::Listener* listener) { | 135 void CreateChannel(IPC::Listener* listener) { |
140 channel_ = IPC::ChannelMojo::Create(std::move(handle_), | 136 channel_ = IPC::ChannelMojo::Create(std::move(handle_), |
141 IPC::Channel::MODE_SERVER, listener); | 137 IPC::Channel::MODE_SERVER, listener); |
142 } | 138 } |
143 | 139 |
144 bool ConnectChannel() { return channel_->Connect(); } | 140 bool ConnectChannel() { return channel_->Connect(); } |
145 | 141 |
146 void DestroyChannel() { channel_.reset(); } | 142 void DestroyChannel() { channel_.reset(); } |
147 | 143 |
148 bool WaitForClientShutdown() { return helper_.WaitForChildTestShutdown(); } | 144 bool WaitForClientShutdown() { return helper_.WaitForChildTestShutdown(); } |
149 | 145 |
150 IPC::Sender* sender() { return channel(); } | 146 IPC::Sender* sender() { return channel(); } |
151 IPC::Channel* channel() { return channel_.get(); } | 147 IPC::Channel* channel() { return channel_.get(); } |
152 | 148 |
153 private: | 149 private: |
154 base::MessageLoop message_loop_; | 150 base::MessageLoop message_loop_; |
| 151 base::TestIOThread io_thread_; |
155 mojo::edk::test::MultiprocessTestHelper helper_; | 152 mojo::edk::test::MultiprocessTestHelper helper_; |
156 mojo::ScopedMessagePipeHandle handle_; | 153 mojo::ScopedMessagePipeHandle handle_; |
157 std::unique_ptr<IPC::Channel> channel_; | 154 std::unique_ptr<IPC::Channel> channel_; |
158 }; | 155 }; |
159 | 156 |
160 class TestChannelListenerWithExtraExpectations | 157 class TestChannelListenerWithExtraExpectations |
161 : public IPC::TestChannelListener { | 158 : public IPC::TestChannelListener { |
162 public: | 159 public: |
163 TestChannelListenerWithExtraExpectations() : is_connected_called_(false) {} | 160 TestChannelListenerWithExtraExpectations() : is_connected_called_(false) {} |
164 | 161 |
(...skipping 413 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
578 ChannelClient) { | 575 ChannelClient) { |
579 ListenerSendingOneOk listener; | 576 ListenerSendingOneOk listener; |
580 Connect(&listener); | 577 Connect(&listener); |
581 listener.set_sender(channel()); | 578 listener.set_sender(channel()); |
582 | 579 |
583 base::RunLoop().Run(); | 580 base::RunLoop().Run(); |
584 | 581 |
585 Close(); | 582 Close(); |
586 } | 583 } |
587 | 584 |
588 class ListenerWithSimpleAssociatedInterface | |
589 : public IPC::Listener, | |
590 public IPC::mojom::SimpleTestDriver { | |
591 public: | |
592 static const int kNumMessages; | |
593 | |
594 ListenerWithSimpleAssociatedInterface() : binding_(this) {} | |
595 | |
596 ~ListenerWithSimpleAssociatedInterface() override {} | |
597 | |
598 bool OnMessageReceived(const IPC::Message& message) override { | |
599 base::PickleIterator iter(message); | |
600 std::string should_be_expected; | |
601 EXPECT_TRUE(iter.ReadString(&should_be_expected)); | |
602 EXPECT_EQ(should_be_expected, next_expected_string_); | |
603 num_messages_received_++; | |
604 return true; | |
605 } | |
606 | |
607 void OnChannelError() override { | |
608 DCHECK(received_quit_); | |
609 } | |
610 | |
611 void RegisterInterfaceFactory(IPC::Channel* channel) { | |
612 channel->GetAssociatedInterfaceSupport()->AddAssociatedInterface( | |
613 base::Bind(&ListenerWithSimpleAssociatedInterface::BindRequest, | |
614 base::Unretained(this))); | |
615 } | |
616 | |
617 private: | |
618 // IPC::mojom::SimpleTestDriver: | |
619 void ExpectString(const mojo::String& str) override { | |
620 next_expected_string_ = str; | |
621 } | |
622 | |
623 void RequestQuit(const RequestQuitCallback& callback) override { | |
624 EXPECT_EQ(kNumMessages, num_messages_received_); | |
625 received_quit_ = true; | |
626 callback.Run(); | |
627 base::MessageLoop::current()->QuitWhenIdle(); | |
628 } | |
629 | |
630 void BindRequest(IPC::mojom::SimpleTestDriverAssociatedRequest request) { | |
631 DCHECK(!binding_.is_bound()); | |
632 binding_.Bind(std::move(request)); | |
633 } | |
634 | |
635 std::string next_expected_string_; | |
636 int num_messages_received_ = 0; | |
637 bool received_quit_ = false; | |
638 | |
639 mojo::AssociatedBinding<IPC::mojom::SimpleTestDriver> binding_; | |
640 }; | |
641 | |
642 const int ListenerWithSimpleAssociatedInterface::kNumMessages = 1000; | |
643 | |
644 class ListenerSendingAssociatedMessages : public IPC::Listener { | |
645 public: | |
646 ListenerSendingAssociatedMessages() {} | |
647 | |
648 bool OnMessageReceived(const IPC::Message& message) override { return true; } | |
649 | |
650 void OnChannelConnected(int32_t peer_pid) override { | |
651 DCHECK(channel_); | |
652 channel_->GetAssociatedInterfaceSupport()->GetRemoteAssociatedInterface( | |
653 &driver_); | |
654 | |
655 // Send a bunch of interleaved messages, alternating between the associated | |
656 // interface and a legacy IPC::Message. | |
657 for (int i = 0; i < ListenerWithSimpleAssociatedInterface::kNumMessages; | |
658 ++i) { | |
659 std::string str = base::StringPrintf("Hello! %d", i); | |
660 driver_->ExpectString(str); | |
661 SendString(channel_, str); | |
662 } | |
663 driver_->RequestQuit(base::Bind(&OnQuitAck)); | |
664 } | |
665 | |
666 void set_channel(IPC::Channel* channel) { channel_ = channel; } | |
667 | |
668 private: | |
669 static void OnQuitAck() { base::MessageLoop::current()->QuitWhenIdle(); } | |
670 | |
671 IPC::Channel* channel_ = nullptr; | |
672 IPC::mojom::SimpleTestDriverAssociatedPtr driver_; | |
673 }; | |
674 | |
675 TEST_F(IPCChannelMojoTest, SimpleAssociatedInterface) { | |
676 InitWithMojo("SimpleAssociatedInterfaceClient"); | |
677 | |
678 ListenerWithSimpleAssociatedInterface listener; | |
679 CreateChannel(&listener); | |
680 ASSERT_TRUE(ConnectChannel()); | |
681 | |
682 listener.RegisterInterfaceFactory(channel()); | |
683 | |
684 base::RunLoop().Run(); | |
685 channel()->Close(); | |
686 | |
687 EXPECT_TRUE(WaitForClientShutdown()); | |
688 DestroyChannel(); | |
689 } | |
690 | |
691 DEFINE_IPC_CHANNEL_MOJO_TEST_CLIENT(SimpleAssociatedInterfaceClient, | |
692 ChannelClient) { | |
693 ListenerSendingAssociatedMessages listener; | |
694 Connect(&listener); | |
695 listener.set_channel(channel()); | |
696 | |
697 base::RunLoop().Run(); | |
698 | |
699 Close(); | |
700 } | |
701 | |
702 #if defined(OS_POSIX) | 585 #if defined(OS_POSIX) |
703 class ListenerThatExpectsFile : public IPC::Listener { | 586 class ListenerThatExpectsFile : public IPC::Listener { |
704 public: | 587 public: |
705 ListenerThatExpectsFile() : sender_(NULL) {} | 588 ListenerThatExpectsFile() : sender_(NULL) {} |
706 | 589 |
707 ~ListenerThatExpectsFile() override {} | 590 ~ListenerThatExpectsFile() override {} |
708 | 591 |
709 bool OnMessageReceived(const IPC::Message& message) override { | 592 bool OnMessageReceived(const IPC::Message& message) override { |
710 base::PickleIterator iter(message); | 593 base::PickleIterator iter(message); |
711 HandleSendingHelper::ReadReceivedFile(message, &iter); | 594 HandleSendingHelper::ReadReceivedFile(message, &iter); |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
806 ChannelClient) { | 689 ChannelClient) { |
807 ListenerThatExpectsFileAndPipe listener; | 690 ListenerThatExpectsFileAndPipe listener; |
808 Connect(&listener); | 691 Connect(&listener); |
809 listener.set_sender(channel()); | 692 listener.set_sender(channel()); |
810 | 693 |
811 base::RunLoop().Run(); | 694 base::RunLoop().Run(); |
812 | 695 |
813 Close(); | 696 Close(); |
814 } | 697 } |
815 | 698 |
816 #endif // defined(OS_POSIX) | 699 #endif |
817 | 700 |
818 #if defined(OS_LINUX) | 701 #if defined(OS_LINUX) |
819 | 702 |
820 const base::ProcessId kMagicChildId = 54321; | 703 const base::ProcessId kMagicChildId = 54321; |
821 | 704 |
822 class ListenerThatVerifiesPeerPid : public IPC::Listener { | 705 class ListenerThatVerifiesPeerPid : public IPC::Listener { |
823 public: | 706 public: |
824 void OnChannelConnected(int32_t peer_pid) override { | 707 void OnChannelConnected(int32_t peer_pid) override { |
825 EXPECT_EQ(peer_pid, kMagicChildId); | 708 EXPECT_EQ(peer_pid, kMagicChildId); |
826 base::MessageLoop::current()->QuitWhenIdle(); | 709 base::MessageLoop::current()->QuitWhenIdle(); |
(...skipping 26 matching lines...) Expand all Loading... |
853 Connect(&listener); | 736 Connect(&listener); |
854 | 737 |
855 base::MessageLoop::current()->Run(); | 738 base::MessageLoop::current()->Run(); |
856 | 739 |
857 Close(); | 740 Close(); |
858 } | 741 } |
859 | 742 |
860 #endif // OS_LINUX | 743 #endif // OS_LINUX |
861 | 744 |
862 } // namespace | 745 } // namespace |
OLD | NEW |