OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_message.h" | 5 #include "ipc/ipc_message.h" |
6 | 6 |
7 #include "base/atomic_sequence_num.h" | 7 #include "base/atomic_sequence_num.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "build/build_config.h" | 9 #include "build/build_config.h" |
10 | 10 |
11 #if defined(OS_POSIX) | 11 #if defined(OS_POSIX) |
| 12 #include "base/file_descriptor_posix.h" |
12 #include "ipc/file_descriptor_set_posix.h" | 13 #include "ipc/file_descriptor_set_posix.h" |
13 #endif | 14 #endif |
14 | 15 |
15 namespace { | 16 namespace { |
16 | 17 |
17 base::StaticAtomicSequenceNumber g_ref_num; | 18 base::StaticAtomicSequenceNumber g_ref_num; |
18 | 19 |
19 // Create a reference number for identifying IPC messages in traces. The return | 20 // Create a reference number for identifying IPC messages in traces. The return |
20 // values has the reference number stored in the upper 24 bits, leaving the low | 21 // values has the reference number stored in the upper 24 bits, leaving the low |
21 // 8 bits set to 0 for use as flags. | 22 // 8 bits set to 0 for use as flags. |
(...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
115 data -= sizeof(int64); | 116 data -= sizeof(int64); |
116 return *(reinterpret_cast<const int64*>(data)); | 117 return *(reinterpret_cast<const int64*>(data)); |
117 } | 118 } |
118 | 119 |
119 void Message::set_received_time(int64 time) const { | 120 void Message::set_received_time(int64 time) const { |
120 received_time_ = time; | 121 received_time_ = time; |
121 } | 122 } |
122 #endif | 123 #endif |
123 | 124 |
124 #if defined(OS_POSIX) | 125 #if defined(OS_POSIX) |
125 bool Message::WriteFileDescriptor(const base::FileDescriptor& descriptor) { | 126 bool Message::WriteFile(base::ScopedFD descriptor) { |
126 // We write the index of the descriptor so that we don't have to | 127 // We write the index of the descriptor so that we don't have to |
127 // keep the current descriptor as extra decoding state when deserialising. | 128 // keep the current descriptor as extra decoding state when deserialising. |
128 WriteInt(file_descriptor_set()->size()); | 129 WriteInt(file_descriptor_set()->size()); |
129 if (descriptor.auto_close) { | 130 return file_descriptor_set()->AddToOwn(descriptor.Pass()); |
130 return file_descriptor_set()->AddAndAutoClose(descriptor.fd); | |
131 } else { | |
132 return file_descriptor_set()->Add(descriptor.fd); | |
133 } | |
134 } | 131 } |
135 | 132 |
136 bool Message::ReadFileDescriptor(PickleIterator* iter, | 133 bool Message::WriteBorrowingFile(const base::PlatformFile& descriptor) { |
137 base::FileDescriptor* descriptor) const { | 134 // We write the index of the descriptor so that we don't have to |
| 135 // keep the current descriptor as extra decoding state when deserialising. |
| 136 WriteInt(file_descriptor_set()->size()); |
| 137 return file_descriptor_set()->AddToBorrow(descriptor); |
| 138 } |
| 139 |
| 140 bool Message::ReadFile(PickleIterator* iter, base::ScopedFD* descriptor) const { |
138 int descriptor_index; | 141 int descriptor_index; |
139 if (!ReadInt(iter, &descriptor_index)) | 142 if (!ReadInt(iter, &descriptor_index)) |
140 return false; | 143 return false; |
141 | 144 |
142 FileDescriptorSet* file_descriptor_set = file_descriptor_set_.get(); | 145 FileDescriptorSet* file_descriptor_set = file_descriptor_set_.get(); |
143 if (!file_descriptor_set) | 146 if (!file_descriptor_set) |
144 return false; | 147 return false; |
145 | 148 |
146 descriptor->fd = file_descriptor_set->GetDescriptorAt(descriptor_index); | 149 base::PlatformFile file = |
147 descriptor->auto_close = true; | 150 file_descriptor_set->TakeDescriptorAt(descriptor_index); |
| 151 if (file < 0) |
| 152 return false; |
148 | 153 |
149 return descriptor->fd >= 0; | 154 descriptor->reset(file); |
| 155 return true; |
150 } | 156 } |
151 | 157 |
152 bool Message::HasFileDescriptors() const { | 158 bool Message::HasFileDescriptors() const { |
153 return file_descriptor_set_.get() && !file_descriptor_set_->empty(); | 159 return file_descriptor_set_.get() && !file_descriptor_set_->empty(); |
154 } | 160 } |
155 | 161 |
156 void Message::EnsureFileDescriptorSet() { | 162 void Message::EnsureFileDescriptorSet() { |
157 if (file_descriptor_set_.get() == NULL) | 163 if (file_descriptor_set_.get() == NULL) |
158 file_descriptor_set_ = new FileDescriptorSet; | 164 file_descriptor_set_ = new FileDescriptorSet; |
159 } | 165 } |
160 | 166 |
161 #endif | 167 #endif |
162 | 168 |
163 } // namespace IPC | 169 } // namespace IPC |
OLD | NEW |