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 | 9 |
10 #include <memory> | 10 #include <memory> |
(...skipping 740 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
751 }; | 751 }; |
752 | 752 |
753 class IPCChannelProxyMojoTest : public IPCChannelMojoTestBase { | 753 class IPCChannelProxyMojoTest : public IPCChannelMojoTestBase { |
754 public: | 754 public: |
755 void InitWithMojo(const std::string& client_name) { | 755 void InitWithMojo(const std::string& client_name) { |
756 IPCChannelMojoTestBase::InitWithMojo(client_name); | 756 IPCChannelMojoTestBase::InitWithMojo(client_name); |
757 runner_.reset(new ChannelProxyRunner(TakeHandle(), true)); | 757 runner_.reset(new ChannelProxyRunner(TakeHandle(), true)); |
758 } | 758 } |
759 void CreateProxy(IPC::Listener* listener) { runner_->CreateProxy(listener); } | 759 void CreateProxy(IPC::Listener* listener) { runner_->CreateProxy(listener); } |
760 void RunProxy() { runner_->RunProxy(); } | 760 void RunProxy() { runner_->RunProxy(); } |
761 void DestroyProxy() { | |
762 runner_.reset(); | |
763 base::RunLoop().RunUntilIdle(); | |
764 } | |
765 | 761 |
766 IPC::ChannelProxy* proxy() { return runner_->proxy(); } | 762 IPC::ChannelProxy* proxy() { return runner_->proxy(); } |
767 | 763 |
768 private: | 764 private: |
769 base::MessageLoop message_loop_; | 765 base::MessageLoop message_loop_; |
770 std::unique_ptr<ChannelProxyRunner> runner_; | 766 std::unique_ptr<ChannelProxyRunner> runner_; |
771 }; | 767 }; |
772 | 768 |
773 class ListenerWithSimpleProxyAssociatedInterface | 769 class ListenerWithSimpleProxyAssociatedInterface |
774 : public IPC::Listener, | 770 : public IPC::Listener, |
(...skipping 30 matching lines...) Expand all Loading... |
805 | 801 |
806 private: | 802 private: |
807 // IPC::mojom::SimpleTestDriver: | 803 // IPC::mojom::SimpleTestDriver: |
808 void ExpectString(const mojo::String& str) override { | 804 void ExpectString(const mojo::String& str) override { |
809 next_expected_string_ = str; | 805 next_expected_string_ = str; |
810 } | 806 } |
811 | 807 |
812 void RequestQuit(const RequestQuitCallback& callback) override { | 808 void RequestQuit(const RequestQuitCallback& callback) override { |
813 received_quit_ = true; | 809 received_quit_ = true; |
814 callback.Run(); | 810 callback.Run(); |
815 binding_.Close(); | |
816 base::MessageLoop::current()->QuitWhenIdle(); | 811 base::MessageLoop::current()->QuitWhenIdle(); |
817 } | 812 } |
818 | 813 |
819 void BindRequest(IPC::mojom::SimpleTestDriverAssociatedRequest request) { | 814 void BindRequest(IPC::mojom::SimpleTestDriverAssociatedRequest request) { |
820 DCHECK(!binding_.is_bound()); | 815 DCHECK(!binding_.is_bound()); |
821 binding_.Bind(std::move(request)); | 816 binding_.Bind(std::move(request)); |
822 } | 817 } |
823 | 818 |
824 std::string next_expected_string_; | 819 std::string next_expected_string_; |
825 int num_messages_received_ = 0; | 820 int num_messages_received_ = 0; |
(...skipping 10 matching lines...) Expand all Loading... |
836 ListenerWithSimpleProxyAssociatedInterface listener; | 831 ListenerWithSimpleProxyAssociatedInterface listener; |
837 CreateProxy(&listener); | 832 CreateProxy(&listener); |
838 listener.RegisterInterfaceFactory(proxy()); | 833 listener.RegisterInterfaceFactory(proxy()); |
839 RunProxy(); | 834 RunProxy(); |
840 | 835 |
841 base::RunLoop().Run(); | 836 base::RunLoop().Run(); |
842 | 837 |
843 EXPECT_TRUE(WaitForClientShutdown()); | 838 EXPECT_TRUE(WaitForClientShutdown()); |
844 EXPECT_TRUE(listener.received_all_messages()); | 839 EXPECT_TRUE(listener.received_all_messages()); |
845 | 840 |
846 DestroyProxy(); | 841 base::RunLoop().RunUntilIdle(); |
847 } | 842 } |
848 | 843 |
849 class ChannelProxyClient { | 844 class ChannelProxyClient { |
850 public: | 845 public: |
851 void Init(mojo::ScopedMessagePipeHandle handle) { | 846 void Init(mojo::ScopedMessagePipeHandle handle) { |
852 runner_.reset(new ChannelProxyRunner(std::move(handle), false)); | 847 runner_.reset(new ChannelProxyRunner(std::move(handle), false)); |
853 } | 848 } |
854 void CreateProxy(IPC::Listener* listener) { runner_->CreateProxy(listener); } | 849 void CreateProxy(IPC::Listener* listener) { runner_->CreateProxy(listener); } |
855 void RunProxy() { runner_->RunProxy(); } | 850 void RunProxy() { runner_->RunProxy(); } |
856 void DestroyProxy() { | |
857 runner_.reset(); | |
858 base::RunLoop().RunUntilIdle(); | |
859 } | |
860 | 851 |
861 IPC::ChannelProxy* proxy() { return runner_->proxy(); } | 852 IPC::ChannelProxy* proxy() { return runner_->proxy(); } |
862 | 853 |
863 private: | 854 private: |
864 base::MessageLoop message_loop_; | 855 base::MessageLoop message_loop_; |
865 std::unique_ptr<ChannelProxyRunner> runner_; | 856 std::unique_ptr<ChannelProxyRunner> runner_; |
866 }; | 857 }; |
867 | 858 |
868 class DummyListener : public IPC::Listener { | 859 class ListenerThatWaitsForConnect : public IPC::Listener { |
869 public: | 860 public: |
| 861 explicit ListenerThatWaitsForConnect(const base::Closure& connect_handler) |
| 862 : connect_handler_(connect_handler) {} |
| 863 |
870 // IPC::Listener | 864 // IPC::Listener |
871 bool OnMessageReceived(const IPC::Message& message) override { return true; } | 865 bool OnMessageReceived(const IPC::Message& message) override { return true; } |
| 866 void OnChannelConnected(int32_t) override { connect_handler_.Run(); } |
| 867 |
| 868 private: |
| 869 base::Closure connect_handler_; |
872 }; | 870 }; |
873 | 871 |
874 DEFINE_IPC_CHANNEL_MOJO_TEST_CLIENT(ProxyThreadAssociatedInterfaceClient, | 872 DEFINE_IPC_CHANNEL_MOJO_TEST_CLIENT(ProxyThreadAssociatedInterfaceClient, |
875 ChannelProxyClient) { | 873 ChannelProxyClient) { |
876 DummyListener listener; | 874 base::RunLoop connect_loop; |
| 875 ListenerThatWaitsForConnect listener(connect_loop.QuitClosure()); |
877 CreateProxy(&listener); | 876 CreateProxy(&listener); |
878 RunProxy(); | 877 RunProxy(); |
| 878 connect_loop.Run(); |
879 | 879 |
880 // Send a bunch of interleaved messages, alternating between the associated | 880 // Send a bunch of interleaved messages, alternating between the associated |
881 // interface and a legacy IPC::Message. | 881 // interface and a legacy IPC::Message. |
882 IPC::mojom::SimpleTestDriverAssociatedPtr driver; | 882 IPC::mojom::SimpleTestDriverAssociatedPtr driver; |
883 proxy()->GetRemoteAssociatedInterface(&driver); | 883 proxy()->GetRemoteAssociatedInterface(&driver); |
884 for (int i = 0; i < ListenerWithSimpleProxyAssociatedInterface::kNumMessages; | 884 for (int i = 0; i < ListenerWithSimpleProxyAssociatedInterface::kNumMessages; |
885 ++i) { | 885 ++i) { |
886 std::string str = base::StringPrintf("Hello! %d", i); | 886 std::string str = base::StringPrintf("Hello! %d", i); |
887 driver->ExpectString(str); | 887 driver->ExpectString(str); |
888 SendString(proxy(), str); | 888 SendString(proxy(), str); |
889 } | 889 } |
890 driver->RequestQuit(base::MessageLoop::QuitWhenIdleClosure()); | 890 driver->RequestQuit(base::MessageLoop::QuitWhenIdleClosure()); |
891 base::RunLoop().Run(); | 891 base::RunLoop().Run(); |
892 | |
893 DestroyProxy(); | |
894 } | 892 } |
895 | 893 |
896 #if defined(OS_POSIX) | 894 #if defined(OS_POSIX) |
897 | 895 |
898 class ListenerThatExpectsFile : public IPC::Listener { | 896 class ListenerThatExpectsFile : public IPC::Listener { |
899 public: | 897 public: |
900 ListenerThatExpectsFile() : sender_(NULL) {} | 898 ListenerThatExpectsFile() : sender_(NULL) {} |
901 | 899 |
902 ~ListenerThatExpectsFile() override {} | 900 ~ListenerThatExpectsFile() override {} |
903 | 901 |
(...skipping 144 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1048 Connect(&listener); | 1046 Connect(&listener); |
1049 | 1047 |
1050 base::MessageLoop::current()->Run(); | 1048 base::MessageLoop::current()->Run(); |
1051 | 1049 |
1052 Close(); | 1050 Close(); |
1053 } | 1051 } |
1054 | 1052 |
1055 #endif // OS_LINUX | 1053 #endif // OS_LINUX |
1056 | 1054 |
1057 } // namespace | 1055 } // namespace |
OLD | NEW |