| 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 "base/memory/shared_memory.h" | 5 #include "base/memory/shared_memory.h" |
| 6 | 6 |
| 7 #include <errno.h> | 7 #include <errno.h> |
| 8 #include <fcntl.h> | 8 #include <fcntl.h> |
| 9 #include <stddef.h> | 9 #include <stddef.h> |
| 10 #include <sys/mman.h> | 10 #include <sys/mman.h> |
| 11 #include <sys/stat.h> | 11 #include <sys/stat.h> |
| 12 #include <unistd.h> | 12 #include <unistd.h> |
| 13 | 13 |
| 14 #include "base/files/file_util.h" | 14 #include "base/files/file_util.h" |
| 15 #include "base/files/scoped_file.h" | 15 #include "base/files/scoped_file.h" |
| 16 #include "base/logging.h" | 16 #include "base/logging.h" |
| 17 #include "base/memory/shared_memory_helper.h" | 17 #include "base/memory/shared_memory_helper.h" |
| 18 #include "base/memory/shared_memory_tracker.h" |
| 18 #include "base/posix/eintr_wrapper.h" | 19 #include "base/posix/eintr_wrapper.h" |
| 19 #include "base/posix/safe_strerror.h" | 20 #include "base/posix/safe_strerror.h" |
| 20 #include "base/process/process_metrics.h" | 21 #include "base/process/process_metrics.h" |
| 21 #include "base/scoped_generic.h" | 22 #include "base/scoped_generic.h" |
| 22 #include "base/strings/utf_string_conversions.h" | 23 #include "base/strings/utf_string_conversions.h" |
| 23 #include "base/threading/thread_restrictions.h" | 24 #include "base/threading/thread_restrictions.h" |
| 24 #include "build/build_config.h" | 25 #include "build/build_config.h" |
| 25 | 26 |
| 26 #if defined(OS_ANDROID) | 27 #if defined(OS_ANDROID) |
| 27 #include "base/os_compat_android.h" | 28 #include "base/os_compat_android.h" |
| (...skipping 249 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 277 } | 278 } |
| 278 #endif | 279 #endif |
| 279 | 280 |
| 280 memory_ = mmap(NULL, bytes, PROT_READ | (read_only_ ? 0 : PROT_WRITE), | 281 memory_ = mmap(NULL, bytes, PROT_READ | (read_only_ ? 0 : PROT_WRITE), |
| 281 MAP_SHARED, mapped_file_, offset); | 282 MAP_SHARED, mapped_file_, offset); |
| 282 | 283 |
| 283 bool mmap_succeeded = memory_ != (void*)-1 && memory_ != NULL; | 284 bool mmap_succeeded = memory_ != (void*)-1 && memory_ != NULL; |
| 284 if (mmap_succeeded) { | 285 if (mmap_succeeded) { |
| 285 mapped_size_ = bytes; | 286 mapped_size_ = bytes; |
| 286 DCHECK_EQ(0U, reinterpret_cast<uintptr_t>(memory_) & | 287 DCHECK_EQ(0U, reinterpret_cast<uintptr_t>(memory_) & |
| 287 (SharedMemory::MAP_MINIMUM_ALIGNMENT - 1)); | 288 (SharedMemory::MAP_MINIMUM_ALIGNMENT - 1)); |
| 289 if (!SharedMemoryTracker::GetInstance()->IncrementMemoryUsage(*this)) |
| 290 return false; |
| 288 } else { | 291 } else { |
| 289 memory_ = NULL; | 292 memory_ = NULL; |
| 290 } | 293 } |
| 291 | 294 |
| 292 return mmap_succeeded; | 295 return mmap_succeeded; |
| 293 } | 296 } |
| 294 | 297 |
| 295 bool SharedMemory::Unmap() { | 298 bool SharedMemory::Unmap() { |
| 296 if (memory_ == NULL) | 299 if (memory_ == NULL) |
| 297 return false; | 300 return false; |
| 298 | 301 |
| 299 munmap(memory_, mapped_size_); | 302 munmap(memory_, mapped_size_); |
| 303 SharedMemoryTracker::GetInstance()->DecrementMemoryUsage(*this); |
| 300 memory_ = NULL; | 304 memory_ = NULL; |
| 301 mapped_size_ = 0; | 305 mapped_size_ = 0; |
| 302 return true; | 306 return true; |
| 303 } | 307 } |
| 304 | 308 |
| 305 SharedMemoryHandle SharedMemory::handle() const { | 309 SharedMemoryHandle SharedMemory::handle() const { |
| 306 return FileDescriptor(mapped_file_, false); | 310 return FileDescriptor(mapped_file_, false); |
| 307 } | 311 } |
| 308 | 312 |
| 309 SharedMemoryHandle SharedMemory::TakeHandle() { | 313 SharedMemoryHandle SharedMemory::TakeHandle() { |
| 310 FileDescriptor handle(mapped_file_, true); | 314 FileDescriptor handle(mapped_file_, true); |
| 311 mapped_file_ = -1; | 315 mapped_file_ = -1; |
| 312 memory_ = nullptr; | 316 memory_ = nullptr; |
| 313 mapped_size_ = 0; | 317 mapped_size_ = 0; |
| 314 return handle; | 318 return handle; |
| 315 } | 319 } |
| 316 | 320 |
| 317 void SharedMemory::Close() { | 321 void SharedMemory::Close() { |
| 318 if (mapped_file_ > 0) { | 322 if (mapped_file_ > 0) { |
| 319 if (IGNORE_EINTR(close(mapped_file_)) < 0) | 323 if (IGNORE_EINTR(close(mapped_file_)) < 0) |
| 320 PLOG(ERROR) << "close"; | 324 PLOG(ERROR) << "close"; |
| 325 SharedMemoryTracker::GetInstance()->DecrementMemoryUsage(*this); |
| 321 mapped_file_ = -1; | 326 mapped_file_ = -1; |
| 322 } | 327 } |
| 323 if (readonly_mapped_file_ > 0) { | 328 if (readonly_mapped_file_ > 0) { |
| 324 if (IGNORE_EINTR(close(readonly_mapped_file_)) < 0) | 329 if (IGNORE_EINTR(close(readonly_mapped_file_)) < 0) |
| 325 PLOG(ERROR) << "close"; | 330 PLOG(ERROR) << "close"; |
| 326 readonly_mapped_file_ = -1; | 331 readonly_mapped_file_ = -1; |
| 327 } | 332 } |
| 328 } | 333 } |
| 329 | 334 |
| 330 #if !defined(OS_ANDROID) | 335 #if !defined(OS_ANDROID) |
| (...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 384 | 389 |
| 385 if (close_self) { | 390 if (close_self) { |
| 386 Unmap(); | 391 Unmap(); |
| 387 Close(); | 392 Close(); |
| 388 } | 393 } |
| 389 | 394 |
| 390 return true; | 395 return true; |
| 391 } | 396 } |
| 392 | 397 |
| 393 } // namespace base | 398 } // namespace base |
| OLD | NEW |