| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "mojo/system/platform_channel.h" | 5 #include "mojo/system/platform_channel.h" |
| 6 | 6 |
| 7 #include <fcntl.h> | 7 #include <fcntl.h> |
| 8 #include <sys/socket.h> | 8 #include <sys/socket.h> |
| 9 #include <sys/types.h> | 9 #include <sys/types.h> |
| 10 #include <unistd.h> | 10 #include <unistd.h> |
| 11 | 11 |
| 12 #include "base/command_line.h" | 12 #include "base/command_line.h" |
| 13 #include "base/compiler_specific.h" | |
| 14 #include "base/logging.h" | 13 #include "base/logging.h" |
| 15 #include "base/posix/global_descriptors.h" | 14 #include "base/posix/global_descriptors.h" |
| 16 #include "base/strings/string_number_conversions.h" | 15 #include "base/strings/string_number_conversions.h" |
| 17 | 16 |
| 18 namespace mojo { | 17 namespace mojo { |
| 19 namespace system { | 18 namespace system { |
| 20 | 19 |
| 21 namespace { | 20 namespace { |
| 22 | 21 |
| 23 const char kMojoChannelDescriptorSwitch[] = "mojo-channel-descriptor"; | 22 const char kMojoChannelDescriptorSwitch[] = "mojo-channel-descriptor"; |
| 24 | 23 |
| 25 bool IsTargetDescriptorUsed( | 24 bool IsTargetDescriptorUsed( |
| 26 const base::FileHandleMappingVector& file_handle_mapping, | 25 const base::FileHandleMappingVector& file_handle_mapping, |
| 27 int target_fd) { | 26 int target_fd) { |
| 28 for (size_t i = 0; i < file_handle_mapping.size(); i++) { | 27 for (size_t i = 0; i < file_handle_mapping.size(); i++) { |
| 29 if (file_handle_mapping[i].second == target_fd) | 28 if (file_handle_mapping[i].second == target_fd) |
| 30 return true; | 29 return true; |
| 31 } | 30 } |
| 32 return false; | 31 return false; |
| 33 } | 32 } |
| 34 | 33 |
| 35 class PlatformServerChannelPosix : public PlatformServerChannel { | 34 } // namespace |
| 36 public: | |
| 37 PlatformServerChannelPosix(const std::string& name); | |
| 38 virtual ~PlatformServerChannelPosix(); | |
| 39 | 35 |
| 40 // |PlatformServerChannel| implementation: | 36 PlatformChannelPair::PlatformChannelPair() { |
| 41 virtual scoped_ptr<PlatformClientChannel> CreateClientChannel() OVERRIDE; | |
| 42 virtual void GetDataNeededToPassClientChannelToChildProcess( | |
| 43 CommandLine* command_line, | |
| 44 base::FileHandleMappingVector* file_handle_mapping) const OVERRIDE; | |
| 45 virtual void ChildProcessLaunched() OVERRIDE; | |
| 46 | |
| 47 private: | |
| 48 PlatformChannelHandle client_handle_; | |
| 49 | |
| 50 DISALLOW_COPY_AND_ASSIGN(PlatformServerChannelPosix); | |
| 51 }; | |
| 52 | |
| 53 PlatformServerChannelPosix::PlatformServerChannelPosix( | |
| 54 const std::string& name) | |
| 55 : PlatformServerChannel(name) { | |
| 56 // Create the Unix domain socket and set the ends to nonblocking. | 37 // Create the Unix domain socket and set the ends to nonblocking. |
| 57 int fds[2]; | 38 int fds[2]; |
| 39 // TODO(vtl): Maybe fail gracefully if |socketpair()| fails. |
| 58 PCHECK(socketpair(AF_UNIX, SOCK_STREAM, 0, fds) == 0); | 40 PCHECK(socketpair(AF_UNIX, SOCK_STREAM, 0, fds) == 0); |
| 59 PCHECK(fcntl(fds[0], F_SETFL, O_NONBLOCK) == 0); | 41 PCHECK(fcntl(fds[0], F_SETFL, O_NONBLOCK) == 0); |
| 60 PCHECK(fcntl(fds[1], F_SETFL, O_NONBLOCK) == 0); | 42 PCHECK(fcntl(fds[1], F_SETFL, O_NONBLOCK) == 0); |
| 61 | 43 |
| 62 mutable_handle()->fd = fds[0]; | 44 server_handle_.fd = fds[0]; |
| 63 DCHECK(is_valid()); | 45 DCHECK(server_handle_.is_valid()); |
| 64 client_handle_.fd = fds[1]; | 46 client_handle_.fd = fds[1]; |
| 65 DCHECK(client_handle_.is_valid()); | 47 DCHECK(client_handle_.is_valid()); |
| 66 } | 48 } |
| 67 | 49 |
| 68 PlatformServerChannelPosix::~PlatformServerChannelPosix() { | 50 // static |
| 69 client_handle_.CloseIfNecessary(); | 51 scoped_ptr<PlatformChannel> |
| 52 PlatformChannelPair::CreateClientChannelFromParentProcess( |
| 53 const CommandLine& command_line) { |
| 54 std::string client_fd_string = |
| 55 command_line.GetSwitchValueASCII(kMojoChannelDescriptorSwitch); |
| 56 int client_fd = -1; |
| 57 if (client_fd_string.empty() || |
| 58 !base::StringToInt(client_fd_string, &client_fd) || |
| 59 client_fd < base::GlobalDescriptors::kBaseDescriptor) { |
| 60 LOG(ERROR) << "Missing or invalid --" << kMojoChannelDescriptorSwitch; |
| 61 return scoped_ptr<PlatformChannel>(); |
| 62 } |
| 63 |
| 64 return PlatformChannel::CreateFromHandle(PlatformChannelHandle(client_fd)); |
| 70 } | 65 } |
| 71 | 66 |
| 72 scoped_ptr<PlatformClientChannel> | 67 void PlatformChannelPair::PrepareToPassClientChannelToChildProcess( |
| 73 PlatformServerChannelPosix::CreateClientChannel() { | |
| 74 if (!client_handle_.is_valid()) { | |
| 75 NOTREACHED(); | |
| 76 return scoped_ptr<PlatformClientChannel>(); | |
| 77 } | |
| 78 | |
| 79 scoped_ptr<PlatformClientChannel> rv = | |
| 80 PlatformClientChannel::CreateFromHandle(client_handle_); | |
| 81 DCHECK(rv->is_valid()); | |
| 82 client_handle_ = PlatformChannelHandle(); | |
| 83 return rv.Pass(); | |
| 84 } | |
| 85 | |
| 86 void PlatformServerChannelPosix::GetDataNeededToPassClientChannelToChildProcess( | |
| 87 CommandLine* command_line, | 68 CommandLine* command_line, |
| 88 base::FileHandleMappingVector* file_handle_mapping) const { | 69 base::FileHandleMappingVector* file_handle_mapping) const { |
| 89 DCHECK(command_line); | 70 DCHECK(command_line); |
| 90 DCHECK(file_handle_mapping); | 71 DCHECK(file_handle_mapping); |
| 91 // This is an arbitrary sanity check. (Note that this guarantees that the loop | 72 // This is an arbitrary sanity check. (Note that this guarantees that the loop |
| 92 // below will terminate sanely.) | 73 // below will terminate sanely.) |
| 93 CHECK_LT(file_handle_mapping->size(), 1000u); | 74 CHECK_LT(file_handle_mapping->size(), 1000u); |
| 94 | 75 |
| 95 DCHECK(client_handle_.is_valid()); | 76 DCHECK(client_handle_.is_valid()); |
| 96 | 77 |
| (...skipping 12 matching lines...) Expand all Loading... |
| 109 LOG_IF(WARNING, command_line->HasSwitch(kMojoChannelDescriptorSwitch)) | 90 LOG_IF(WARNING, command_line->HasSwitch(kMojoChannelDescriptorSwitch)) |
| 110 << "Child command line already has switch --" | 91 << "Child command line already has switch --" |
| 111 << kMojoChannelDescriptorSwitch << "=" | 92 << kMojoChannelDescriptorSwitch << "=" |
| 112 << command_line->GetSwitchValueASCII(kMojoChannelDescriptorSwitch); | 93 << command_line->GetSwitchValueASCII(kMojoChannelDescriptorSwitch); |
| 113 // (Any existing switch won't actually be removed from the command line, but | 94 // (Any existing switch won't actually be removed from the command line, but |
| 114 // the last one appended takes precedence.) | 95 // the last one appended takes precedence.) |
| 115 command_line->AppendSwitchASCII(kMojoChannelDescriptorSwitch, | 96 command_line->AppendSwitchASCII(kMojoChannelDescriptorSwitch, |
| 116 base::IntToString(target_fd)); | 97 base::IntToString(target_fd)); |
| 117 } | 98 } |
| 118 | 99 |
| 119 void PlatformServerChannelPosix::ChildProcessLaunched() { | 100 void PlatformChannelPair::ChildProcessLaunched() { |
| 120 DCHECK(client_handle_.is_valid()); | 101 DCHECK(client_handle_.is_valid()); |
| 121 client_handle_.CloseIfNecessary(); | 102 client_handle_.CloseIfNecessary(); |
| 122 } | 103 } |
| 123 | 104 |
| 124 } // namespace | |
| 125 | |
| 126 // ----------------------------------------------------------------------------- | |
| 127 | |
| 128 // Static factory method declared in platform_channel.h. | |
| 129 // static | |
| 130 scoped_ptr<PlatformServerChannel> PlatformServerChannel::Create( | |
| 131 const std::string& name) { | |
| 132 return scoped_ptr<PlatformServerChannel>( | |
| 133 new PlatformServerChannelPosix(name)); | |
| 134 } | |
| 135 | |
| 136 // ----------------------------------------------------------------------------- | |
| 137 | |
| 138 // Static factory method declared in platform_channel.h. | |
| 139 // static | |
| 140 scoped_ptr<PlatformClientChannel> | |
| 141 PlatformClientChannel::CreateFromParentProcess( | |
| 142 const CommandLine& command_line) { | |
| 143 std::string client_fd_string = | |
| 144 command_line.GetSwitchValueASCII(kMojoChannelDescriptorSwitch); | |
| 145 int client_fd = -1; | |
| 146 if (client_fd_string.empty() || | |
| 147 !base::StringToInt(client_fd_string, &client_fd) || | |
| 148 client_fd < base::GlobalDescriptors::kBaseDescriptor) { | |
| 149 LOG(ERROR) << "Missing or invalid --" << kMojoChannelDescriptorSwitch; | |
| 150 return scoped_ptr<PlatformClientChannel>(); | |
| 151 } | |
| 152 | |
| 153 return CreateFromHandle(PlatformChannelHandle(client_fd)); | |
| 154 } | |
| 155 | |
| 156 } // namespace system | 105 } // namespace system |
| 157 } // namespace mojo | 106 } // namespace mojo |
| OLD | NEW |