1 /* This Source Code Form is subject to the terms of the Mozilla Public
2 * License, v. 2.0. If a copy of the MPL was not distributed with this
3 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
5 #include "VolatileBuffer.h"
6 #include "mozilla/Assertions.h"
7 #include "mozilla/DebugOnly.h"
8 #include "mozilla/mozalloc.h"
10 #include <mach/mach.h>
14 #define MIN_VOLATILE_ALLOC_SIZE 8192
18 VolatileBuffer::VolatileBuffer()
19 : mMutex("VolatileBuffer")
28 VolatileBuffer::Init(size_t aSize
, size_t aAlignment
)
30 MOZ_ASSERT(!mSize
&& !mBuf
, "Init called twice");
31 MOZ_ASSERT(!(aAlignment
% sizeof(void *)),
32 "Alignment must be multiple of pointer size");
36 kern_return_t ret
= 0;
37 if (aSize
< MIN_VOLATILE_ALLOC_SIZE
) {
41 ret
= vm_allocate(mach_task_self(),
44 VM_FLAGS_PURGABLE
| VM_FLAGS_ANYWHERE
);
45 if (ret
== KERN_SUCCESS
) {
50 (void)moz_posix_memalign(&mBuf
, aAlignment
, aSize
);
55 VolatileBuffer::~VolatileBuffer()
57 MOZ_ASSERT(mLockCount
== 0, "Being destroyed with non-zero lock count?");
62 vm_deallocate(mach_task_self(), (vm_address_t
)mBuf
, mSize
);
67 VolatileBuffer::Lock(void** aBuf
)
69 MutexAutoLock
lock(mMutex
);
71 MOZ_ASSERT(mBuf
, "Attempting to lock an uninitialized VolatileBuffer");
74 if (++mLockCount
> 1 || OnHeap()) {
78 int state
= VM_PURGABLE_NONVOLATILE
;
80 vm_purgable_control(mach_task_self(),
82 VM_PURGABLE_SET_STATE
,
84 return ret
== KERN_SUCCESS
&& !(state
& VM_PURGABLE_EMPTY
);
88 VolatileBuffer::Unlock()
90 MutexAutoLock
lock(mMutex
);
92 MOZ_ASSERT(mLockCount
> 0, "VolatileBuffer unlocked too many times!");
93 if (--mLockCount
|| OnHeap()) {
97 int state
= VM_PURGABLE_VOLATILE
| VM_VOLATILE_GROUP_DEFAULT
;
98 DebugOnly
<kern_return_t
> ret
=
99 vm_purgable_control(mach_task_self(),
101 VM_PURGABLE_SET_STATE
,
103 MOZ_ASSERT(ret
== KERN_SUCCESS
, "Failed to set buffer as purgable");
107 VolatileBuffer::OnHeap() const
113 VolatileBuffer::HeapSizeOfExcludingThis(MallocSizeOf aMallocSizeOf
) const
115 return OnHeap() ? aMallocSizeOf(mBuf
) : 0;
119 VolatileBuffer::NonHeapSizeOfExcludingThis() const
125 unsigned long pagemask
= getpagesize() - 1;
126 return (mSize
+ pagemask
) & ~pagemask
;
129 } // namespace mozilla