OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 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 "base/memory/discardable_memory.h" | 5 #include "base/memory/discardable_memory.h" |
6 | 6 |
7 #include <mach/mach.h> | 7 #include <mach/mach.h> |
8 #include <mach/mach_vm.h> | 8 #include <mach/mach_vm.h> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
(...skipping 15 matching lines...) Expand all Loading... |
26 | 26 |
27 class DiscardableMemoryMac : public DiscardableMemory { | 27 class DiscardableMemoryMac : public DiscardableMemory { |
28 public: | 28 public: |
29 explicit DiscardableMemoryMac(size_t size) | 29 explicit DiscardableMemoryMac(size_t size) |
30 : memory_(0, 0), | 30 : memory_(0, 0), |
31 size_(mach_vm_round_page(size)) { | 31 size_(mach_vm_round_page(size)) { |
32 } | 32 } |
33 | 33 |
34 bool Initialize() { | 34 bool Initialize() { |
35 DCHECK_EQ(memory_.size(), 0u); | 35 DCHECK_EQ(memory_.size(), 0u); |
36 mach_vm_address_t address = 0; | 36 vm_address_t address = 0; |
37 kern_return_t ret = mach_vm_allocate(mach_task_self(), | 37 kern_return_t ret = vm_allocate(mach_task_self(), |
38 &address, | 38 &address, |
39 size_, | 39 size_, |
40 VM_FLAGS_PURGABLE | | 40 VM_FLAGS_PURGABLE | |
41 VM_FLAGS_ANYWHERE | | 41 VM_FLAGS_ANYWHERE | |
42 kDiscardableMemoryTag); | 42 kDiscardableMemoryTag); |
43 if (ret != KERN_SUCCESS) { | 43 if (ret != KERN_SUCCESS) { |
44 MACH_DLOG(ERROR, ret) << "mach_vm_allocate"; | 44 MACH_DLOG(ERROR, ret) << "vm_allocate"; |
45 return false; | 45 return false; |
46 } | 46 } |
47 | 47 |
48 memory_.reset(address, size_); | 48 memory_.reset(address, size_); |
49 | 49 |
50 return true; | 50 return true; |
51 } | 51 } |
52 | 52 |
53 virtual ~DiscardableMemoryMac() { | 53 virtual ~DiscardableMemoryMac() { |
54 } | 54 } |
55 | 55 |
56 virtual DiscardableMemoryLockStatus Lock() OVERRIDE { | 56 virtual DiscardableMemoryLockStatus Lock() OVERRIDE { |
57 kern_return_t ret; | 57 kern_return_t ret; |
58 MACH_DCHECK((ret = mach_vm_protect(mach_task_self(), | 58 MACH_DCHECK((ret = vm_protect(mach_task_self(), |
59 memory_.address(), | 59 memory_.address(), |
60 memory_.size(), | 60 memory_.size(), |
61 FALSE, | 61 FALSE, |
62 VM_PROT_DEFAULT)) == KERN_SUCCESS, ret); | 62 VM_PROT_DEFAULT)) == KERN_SUCCESS, ret); |
63 int state = VM_PURGABLE_NONVOLATILE; | 63 int state = VM_PURGABLE_NONVOLATILE; |
64 ret = mach_vm_purgable_control(mach_task_self(), | 64 ret = vm_purgable_control(mach_task_self(), |
65 memory_.address(), | 65 memory_.address(), |
66 VM_PURGABLE_SET_STATE, | 66 VM_PURGABLE_SET_STATE, |
67 &state); | 67 &state); |
68 if (ret != KERN_SUCCESS) | 68 if (ret != KERN_SUCCESS) |
69 return DISCARDABLE_MEMORY_LOCK_STATUS_FAILED; | 69 return DISCARDABLE_MEMORY_LOCK_STATUS_FAILED; |
70 | 70 |
71 return state & VM_PURGABLE_EMPTY ? DISCARDABLE_MEMORY_LOCK_STATUS_PURGED | 71 return state & VM_PURGABLE_EMPTY ? DISCARDABLE_MEMORY_LOCK_STATUS_PURGED |
72 : DISCARDABLE_MEMORY_LOCK_STATUS_SUCCESS; | 72 : DISCARDABLE_MEMORY_LOCK_STATUS_SUCCESS; |
73 } | 73 } |
74 | 74 |
75 virtual void Unlock() OVERRIDE { | 75 virtual void Unlock() OVERRIDE { |
76 int state = VM_PURGABLE_VOLATILE | VM_VOLATILE_GROUP_DEFAULT; | 76 int state = VM_PURGABLE_VOLATILE | VM_VOLATILE_GROUP_DEFAULT; |
77 kern_return_t ret = mach_vm_purgable_control(mach_task_self(), | 77 kern_return_t ret = vm_purgable_control(mach_task_self(), |
78 memory_.address(), | 78 memory_.address(), |
79 VM_PURGABLE_SET_STATE, | 79 VM_PURGABLE_SET_STATE, |
80 &state); | 80 &state); |
81 MACH_DLOG_IF(ERROR, ret != KERN_SUCCESS, ret) << "mach_vm_purgable_control"; | 81 MACH_DLOG_IF(ERROR, ret != KERN_SUCCESS, ret) << "vm_purgable_control"; |
82 MACH_DCHECK((ret = mach_vm_protect(mach_task_self(), | 82 MACH_DCHECK((ret = vm_protect(mach_task_self(), |
83 memory_.address(), | 83 memory_.address(), |
84 memory_.size(), | 84 memory_.size(), |
85 FALSE, | 85 FALSE, |
86 VM_PROT_NONE)) == KERN_SUCCESS, ret); | 86 VM_PROT_NONE)) == KERN_SUCCESS, ret); |
87 } | 87 } |
88 | 88 |
89 virtual void* Memory() const OVERRIDE { | 89 virtual void* Memory() const OVERRIDE { |
90 return reinterpret_cast<void*>(memory_.address()); | 90 return reinterpret_cast<void*>(memory_.address()); |
91 } | 91 } |
92 | 92 |
93 private: | 93 private: |
94 mac::ScopedMachVM memory_; | 94 mac::ScopedMachVM memory_; |
95 size_t size_; | 95 size_t size_; |
96 | 96 |
(...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
152 } | 152 } |
153 } | 153 } |
154 | 154 |
155 NOTREACHED(); | 155 NOTREACHED(); |
156 return scoped_ptr<DiscardableMemory>(); | 156 return scoped_ptr<DiscardableMemory>(); |
157 } | 157 } |
158 | 158 |
159 // static | 159 // static |
160 void DiscardableMemory::PurgeForTesting() { | 160 void DiscardableMemory::PurgeForTesting() { |
161 int state = 0; | 161 int state = 0; |
162 mach_vm_purgable_control(mach_task_self(), 0, VM_PURGABLE_PURGE_ALL, &state); | 162 vm_purgable_control(mach_task_self(), 0, VM_PURGABLE_PURGE_ALL, &state); |
163 internal::DiscardableMemoryEmulated::PurgeForTesting(); | 163 internal::DiscardableMemoryEmulated::PurgeForTesting(); |
164 } | 164 } |
165 | 165 |
166 } // namespace base | 166 } // namespace base |
OLD | NEW |