OLD | NEW |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 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 "base/memory/shared_memory_handle.h" | 5 #include "base/memory/shared_memory_handle.h" |
6 | 6 |
7 #include <unistd.h> | 7 #include <unistd.h> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/posix/eintr_wrapper.h" | 10 #include "base/posix/eintr_wrapper.h" |
11 #include "base/unguessable_token.h" | 11 #include "base/unguessable_token.h" |
12 | 12 |
13 namespace base { | 13 namespace base { |
14 | 14 |
15 SharedMemoryHandle::SharedMemoryHandle() = default; | 15 SharedMemoryHandle::SharedMemoryHandle() = default; |
16 | 16 |
17 SharedMemoryHandle::SharedMemoryHandle( | 17 SharedMemoryHandle::SharedMemoryHandle( |
18 const base::FileDescriptor& file_descriptor, | 18 const base::FileDescriptor& file_descriptor, |
| 19 size_t size, |
19 const base::UnguessableToken& guid) | 20 const base::UnguessableToken& guid) |
20 : file_descriptor_(file_descriptor), guid_(guid) {} | 21 : file_descriptor_(file_descriptor), guid_(guid), size_(size) {} |
21 | 22 |
22 // static | 23 // static |
23 SharedMemoryHandle SharedMemoryHandle::ImportHandle(int fd) { | 24 SharedMemoryHandle SharedMemoryHandle::ImportHandle(int fd, size_t size) { |
24 SharedMemoryHandle handle; | 25 SharedMemoryHandle handle; |
25 handle.file_descriptor_.fd = fd; | 26 handle.file_descriptor_.fd = fd; |
26 handle.file_descriptor_.auto_close = false; | 27 handle.file_descriptor_.auto_close = false; |
27 handle.guid_ = UnguessableToken::Create(); | 28 handle.guid_ = UnguessableToken::Create(); |
| 29 handle.size_ = size; |
28 return handle; | 30 return handle; |
29 } | 31 } |
30 | 32 |
31 int SharedMemoryHandle::GetHandle() const { | 33 int SharedMemoryHandle::GetHandle() const { |
32 return file_descriptor_.fd; | 34 return file_descriptor_.fd; |
33 } | 35 } |
34 | 36 |
35 void SharedMemoryHandle::SetHandle(int handle) { | |
36 file_descriptor_.fd = handle; | |
37 } | |
38 | |
39 bool SharedMemoryHandle::IsValid() const { | 37 bool SharedMemoryHandle::IsValid() const { |
40 return file_descriptor_.fd >= 0; | 38 return file_descriptor_.fd >= 0; |
41 } | 39 } |
42 | 40 |
43 void SharedMemoryHandle::Close() const { | 41 void SharedMemoryHandle::Close() const { |
44 if (IGNORE_EINTR(close(file_descriptor_.fd)) < 0) | 42 if (IGNORE_EINTR(close(file_descriptor_.fd)) < 0) |
45 PLOG(ERROR) << "close"; | 43 PLOG(ERROR) << "close"; |
46 } | 44 } |
47 | 45 |
48 int SharedMemoryHandle::Release() { | 46 int SharedMemoryHandle::Release() { |
49 int old_fd = file_descriptor_.fd; | 47 int old_fd = file_descriptor_.fd; |
50 file_descriptor_.fd = -1; | 48 file_descriptor_.fd = -1; |
51 return old_fd; | 49 return old_fd; |
52 } | 50 } |
53 | 51 |
54 SharedMemoryHandle SharedMemoryHandle::Duplicate() const { | 52 SharedMemoryHandle SharedMemoryHandle::Duplicate() const { |
55 if (!IsValid()) | 53 if (!IsValid()) |
56 return SharedMemoryHandle(); | 54 return SharedMemoryHandle(); |
57 | 55 |
58 int duped_handle = HANDLE_EINTR(dup(file_descriptor_.fd)); | 56 int duped_handle = HANDLE_EINTR(dup(file_descriptor_.fd)); |
59 if (duped_handle < 0) | 57 if (duped_handle < 0) |
60 return SharedMemoryHandle(); | 58 return SharedMemoryHandle(); |
61 return SharedMemoryHandle(FileDescriptor(duped_handle, true), GetGUID()); | 59 return SharedMemoryHandle(FileDescriptor(duped_handle, true), GetSize(), |
| 60 GetGUID()); |
62 } | 61 } |
63 | 62 |
64 void SharedMemoryHandle::SetOwnershipPassesToIPC(bool ownership_passes) { | 63 void SharedMemoryHandle::SetOwnershipPassesToIPC(bool ownership_passes) { |
65 file_descriptor_.auto_close = ownership_passes; | 64 file_descriptor_.auto_close = ownership_passes; |
66 } | 65 } |
67 | 66 |
68 bool SharedMemoryHandle::OwnershipPassesToIPC() const { | 67 bool SharedMemoryHandle::OwnershipPassesToIPC() const { |
69 return file_descriptor_.auto_close; | 68 return file_descriptor_.auto_close; |
70 } | 69 } |
71 | 70 |
72 } // namespace base | 71 } // namespace base |
OLD | NEW |