123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228 |
- // Copyright (c) 2012 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "build/build_config.h"
- #if BUILDFLAG(IS_MAC)
- extern "C" {
- #include <sandbox.h>
- }
- #endif
- #include <fcntl.h>
- #include <stddef.h>
- #include <sys/socket.h>
- #include <sys/stat.h>
- #include <unistd.h>
- #include <memory>
- #include <queue>
- #include "base/file_descriptor_posix.h"
- #include "base/pickle.h"
- #include "base/posix/eintr_wrapper.h"
- #include "base/run_loop.h"
- #include "base/synchronization/waitable_event.h"
- #include "base/task/single_thread_task_runner.h"
- #include "base/threading/thread.h"
- #include "base/threading/thread_task_runner_handle.h"
- #include "ipc/ipc_message_attachment_set.h"
- #include "ipc/ipc_message_utils.h"
- #include "ipc/ipc_test_base.h"
- #if BUILDFLAG(IS_MAC)
- #include "sandbox/mac/seatbelt.h"
- #elif BUILDFLAG(IS_FUCHSIA)
- #include "base/memory/scoped_refptr.h"
- #include "base/test/scoped_dev_zero_fuchsia.h"
- #endif
- namespace {
- const unsigned kNumFDsToSend = 7; // per message
- const unsigned kNumMessages = 20;
- const char* kDevZeroPath = "/dev/zero";
- static_assert(kNumFDsToSend ==
- IPC::MessageAttachmentSet::kMaxDescriptorsPerMessage,
- "The number of FDs to send must be kMaxDescriptorsPerMessage.");
- class MyChannelDescriptorListenerBase : public IPC::Listener {
- public:
- bool OnMessageReceived(const IPC::Message& message) override {
- base::PickleIterator iter(message);
- base::FileDescriptor descriptor;
- while (IPC::ParamTraits<base::FileDescriptor>::Read(
- &message, &iter, &descriptor)) {
- HandleFD(descriptor.fd);
- }
- return true;
- }
- protected:
- virtual void HandleFD(int fd) = 0;
- };
- class MyChannelDescriptorListener : public MyChannelDescriptorListenerBase {
- public:
- explicit MyChannelDescriptorListener(ino_t expected_inode_num)
- : MyChannelDescriptorListenerBase(),
- expected_inode_num_(expected_inode_num),
- num_fds_received_(0) {
- }
- unsigned num_fds_received() const {
- return num_fds_received_;
- }
- void OnChannelError() override {
- base::RunLoop::QuitCurrentWhenIdleDeprecated();
- }
- protected:
- void HandleFD(int fd) override {
- ASSERT_GE(fd, 0);
- // Check that we can read from the FD.
- char buf;
- ssize_t amt_read = read(fd, &buf, 1);
- ASSERT_EQ(amt_read, 1);
- ASSERT_EQ(buf, 0); // /dev/zero always reads 0 bytes.
- struct stat st;
- ASSERT_EQ(fstat(fd, &st), 0);
- ASSERT_EQ(close(fd), 0);
- // Compare inode numbers to check that the file sent over the wire is
- // actually the one expected.
- ASSERT_EQ(expected_inode_num_, st.st_ino);
- ++num_fds_received_;
- if (num_fds_received_ == kNumFDsToSend * kNumMessages)
- base::RunLoop::QuitCurrentWhenIdleDeprecated();
- }
- private:
- ino_t expected_inode_num_;
- unsigned num_fds_received_;
- };
- class IPCSendFdsTest : public IPCChannelMojoTestBase {
- protected:
- void SetUp() override {
- #if BUILDFLAG(IS_FUCHSIA)
- ASSERT_TRUE(dev_zero_);
- #endif
- }
- void RunServer() {
- // Set up IPC channel and start client.
- MyChannelDescriptorListener listener(-1);
- CreateChannel(&listener);
- ASSERT_TRUE(ConnectChannel());
- for (unsigned i = 0; i < kNumMessages; ++i) {
- IPC::Message* message =
- new IPC::Message(0, 3, IPC::Message::PRIORITY_NORMAL);
- for (unsigned j = 0; j < kNumFDsToSend; ++j) {
- const int fd = open(kDevZeroPath, O_RDONLY);
- ASSERT_GE(fd, 0);
- base::FileDescriptor descriptor(fd, true);
- IPC::ParamTraits<base::FileDescriptor>::Write(message, descriptor);
- }
- ASSERT_TRUE(sender()->Send(message));
- }
- // Run message loop.
- base::RunLoop().Run();
- // Close the channel so the client's OnChannelError() gets fired.
- channel()->Close();
- EXPECT_TRUE(WaitForClientShutdown());
- DestroyChannel();
- }
- private:
- #if BUILDFLAG(IS_FUCHSIA)
- scoped_refptr<base::ScopedDevZero> dev_zero_ = base::ScopedDevZero::Get();
- #endif
- };
- // Disabled on Fuchsia due to failures; see https://crbug.com/1272424.
- #if BUILDFLAG(IS_FUCHSIA)
- #define MAYBE_DescriptorTest DISABLED_DescriptorTest
- #else
- #define MAYBE_DescriptorTest DescriptorTest
- #endif
- TEST_F(IPCSendFdsTest, MAYBE_DescriptorTest) {
- Init("SendFdsClient");
- RunServer();
- }
- class SendFdsTestClientFixture : public IpcChannelMojoTestClient {
- protected:
- void SendFdsClientCommon(const std::string& test_client_name,
- ino_t expected_inode_num) {
- MyChannelDescriptorListener listener(expected_inode_num);
- // Set up IPC channel.
- Connect(&listener);
- // Run message loop.
- base::RunLoop().Run();
- // Verify that the message loop was exited due to getting the correct number
- // of descriptors, and not because of the channel closing unexpectedly.
- EXPECT_EQ(kNumFDsToSend * kNumMessages, listener.num_fds_received());
- Close();
- }
- };
- DEFINE_IPC_CHANNEL_MOJO_TEST_CLIENT_WITH_CUSTOM_FIXTURE(
- SendFdsClient,
- SendFdsTestClientFixture) {
- struct stat st;
- int fd = open(kDevZeroPath, O_RDONLY);
- fstat(fd, &st);
- EXPECT_GE(IGNORE_EINTR(close(fd)), 0);
- SendFdsClientCommon("SendFdsClient", st.st_ino);
- }
- #if BUILDFLAG(IS_MAC)
- // Test that FDs are correctly sent to a sandboxed process.
- // TODO(port): Make this test cross-platform.
- TEST_F(IPCSendFdsTest, DescriptorTestSandboxed) {
- Init("SendFdsSandboxedClient");
- RunServer();
- }
- DEFINE_IPC_CHANNEL_MOJO_TEST_CLIENT_WITH_CUSTOM_FIXTURE(
- SendFdsSandboxedClient,
- SendFdsTestClientFixture) {
- struct stat st;
- const int fd = open(kDevZeroPath, O_RDONLY);
- fstat(fd, &st);
- ASSERT_LE(0, IGNORE_EINTR(close(fd)));
- // Enable the sandbox.
- char* error_buff = NULL;
- int error = sandbox::Seatbelt::Init(
- sandbox::Seatbelt::kProfilePureComputation, SANDBOX_NAMED, &error_buff);
- ASSERT_EQ(0, error);
- ASSERT_FALSE(error_buff);
- sandbox::Seatbelt::FreeError(error_buff);
- // Make sure sandbox is really enabled.
- ASSERT_EQ(-1, open(kDevZeroPath, O_RDONLY))
- << "Sandbox wasn't properly enabled";
- // See if we can receive a file descriptor.
- SendFdsClientCommon("SendFdsSandboxedClient", st.st_ino);
- }
- #endif // BUILDFLAG(IS_MAC)
- } // namespace
|