|
Packit |
f0b94e |
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
Packit |
f0b94e |
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
Packit |
f0b94e |
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
Packit |
f0b94e |
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
Packit |
f0b94e |
* You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
#ifndef StackArena_h
|
|
Packit |
f0b94e |
#define StackArena_h
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
#include "nsError.h"
|
|
Packit |
f0b94e |
#include "mozilla/Assertions.h"
|
|
Packit |
f0b94e |
#include "mozilla/MemoryReporting.h"
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
namespace mozilla {
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
struct StackBlock;
|
|
Packit |
f0b94e |
struct StackMark;
|
|
Packit |
f0b94e |
class AutoStackArena;
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// Private helper class for AutoStackArena.
|
|
Packit |
f0b94e |
class StackArena {
|
|
Packit |
f0b94e |
private:
|
|
Packit |
f0b94e |
friend class AutoStackArena;
|
|
Packit |
f0b94e |
StackArena();
|
|
Packit |
f0b94e |
~StackArena();
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// Memory management functions.
|
|
Packit |
f0b94e |
void* Allocate(size_t aSize);
|
|
Packit |
f0b94e |
void Push();
|
|
Packit |
f0b94e |
void Pop();
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
size_t SizeOfExcludingThis(mozilla::MallocSizeOf aMallocSizeOf) const;
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// Our current position in memory.
|
|
Packit |
f0b94e |
size_t mPos;
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// A list of memory blocks. Usually there is only one
|
|
Packit |
f0b94e |
// but if we overrun our stack size we can get more memory.
|
|
Packit |
f0b94e |
StackBlock* mBlocks;
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// The current block.
|
|
Packit |
f0b94e |
StackBlock* mCurBlock;
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// Our stack of mark where push has been called.
|
|
Packit |
f0b94e |
StackMark* mMarks;
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// The current top of the mark list.
|
|
Packit |
f0b94e |
uint32_t mStackTop;
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// The size of the mark array.
|
|
Packit |
f0b94e |
uint32_t mMarkLength;
|
|
Packit |
f0b94e |
};
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
// Class for stack scoped arena memory allocations.
|
|
Packit |
f0b94e |
//
|
|
Packit |
f0b94e |
// Callers who wish to allocate memory whose lifetime corresponds to the
|
|
Packit |
f0b94e |
// lifetime of a stack-allocated object can use this class. First,
|
|
Packit |
f0b94e |
// declare an AutoStackArena object on the stack. Then all subsequent
|
|
Packit |
f0b94e |
// calls to Allocate will allocate memory from an arena pool that will
|
|
Packit |
f0b94e |
// be freed when that variable goes out of scope. Nesting is allowed.
|
|
Packit |
f0b94e |
//
|
|
Packit |
f0b94e |
// Individual allocations cannot exceed StackBlock::MAX_USABLE_SIZE
|
|
Packit |
f0b94e |
// bytes.
|
|
Packit |
f0b94e |
//
|
|
Packit |
f0b94e |
class MOZ_RAII AutoStackArena {
|
|
Packit |
f0b94e |
public:
|
|
Packit |
f0b94e |
AutoStackArena() : mOwnsStackArena(false) {
|
|
Packit |
f0b94e |
if (!gStackArena) {
|
|
Packit |
f0b94e |
gStackArena = new StackArena();
|
|
Packit |
f0b94e |
mOwnsStackArena = true;
|
|
Packit |
f0b94e |
}
|
|
Packit |
f0b94e |
gStackArena->Push();
|
|
Packit |
f0b94e |
}
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
~AutoStackArena() {
|
|
Packit |
f0b94e |
gStackArena->Pop();
|
|
Packit |
f0b94e |
if (mOwnsStackArena) {
|
|
Packit |
f0b94e |
delete gStackArena;
|
|
Packit |
f0b94e |
gStackArena = nullptr;
|
|
Packit |
f0b94e |
}
|
|
Packit |
f0b94e |
}
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
static void* Allocate(size_t aSize) { return gStackArena->Allocate(aSize); }
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
private:
|
|
Packit |
f0b94e |
static StackArena* gStackArena;
|
|
Packit |
f0b94e |
bool mOwnsStackArena;
|
|
Packit |
f0b94e |
};
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
} // namespace mozilla
|
|
Packit |
f0b94e |
|
|
Packit |
f0b94e |
#endif
|