/*
* heap_get_block_header -- returns the header of the memory block
*/
void *
heap_get_block_header(PMEMobjpool *pop, struct memory_block m,
enum heap_op op, uint64_t *op_result)
{
struct zone *z = &pop->heap->layout->zones[m.zone_id];
struct chunk_header *hdr = &z->chunk_headers[m.chunk_id];
if (hdr->type != CHUNK_TYPE_RUN) {
*op_result = chunk_get_chunk_hdr_value(*hdr,
op == HEAP_OP_ALLOC ? CHUNK_TYPE_USED : CHUNK_TYPE_FREE,
m.size_idx);
heap_chunk_write_footer(hdr, m.size_idx);
return hdr;
}
struct chunk_run *r = (struct chunk_run *)&z->chunks[m.chunk_id];
uint64_t bmask = ((1L << m.size_idx) - 1L) <<
(m.block_off % BITS_PER_VALUE);
int bpos = m.block_off / BITS_PER_VALUE;
if (op == HEAP_OP_FREE)
*op_result = r->bitmap[bpos] & ~bmask;
else
*op_result = r->bitmap[bpos] | bmask;
return &r->bitmap[bpos];
}
/*
* heap_reclaim_zone_garbage -- (internal) creates volatile state of unused runs
*/
static int
heap_reclaim_zone_garbage(struct palloc_heap *heap, uint32_t zone_id, int init)
{
struct zone *z = ZID_TO_ZONE(heap->layout, zone_id);
struct chunk_run *run = NULL;
int rchunks = 0;
/*
* If this is the first time this zone is processed, recreate all
* footers BEFORE any other operation takes place. For example, the
* heap_init_free_chunk call expects the footers to be created.
*/
if (init) {
for (uint32_t i = 0; i < z->header.size_idx; ) {
struct chunk_header *hdr = &z->chunk_headers[i];
switch (hdr->type) {
case CHUNK_TYPE_USED:
heap_chunk_write_footer(hdr,
hdr->size_idx);
break;
}
i += hdr->size_idx;
}
}
for (uint32_t i = 0; i < z->header.size_idx; ) {
struct chunk_header *hdr = &z->chunk_headers[i];
ASSERT(hdr->size_idx != 0);
struct memory_block m = MEMORY_BLOCK_NONE;
m.zone_id = zone_id;
m.chunk_id = i;
m.size_idx = hdr->size_idx;
memblock_rebuild_state(heap, &m);
switch (hdr->type) {
case CHUNK_TYPE_RUN:
run = (struct chunk_run *)&z->chunks[i];
rchunks += heap_reclaim_run(heap, run, &m);
break;
case CHUNK_TYPE_FREE:
if (init)
heap_init_free_chunk(heap, hdr, &m);
break;
case CHUNK_TYPE_USED:
break;
default:
ASSERT(0);
}
i = m.chunk_id + m.size_idx; /* hdr might have changed */
}
return rchunks == 0 ? ENOMEM : 0;
}
/*
* heap_init_free_chunk -- initializes free chunk transient state
*/
static void
heap_init_free_chunk(struct palloc_heap *heap,
struct chunk_header *hdr,
struct memory_block *m)
{
struct operation_context ctx;
operation_init(&ctx, heap->base, NULL, NULL);
ctx.p_ops = &heap->p_ops;
heap_chunk_write_footer(hdr, hdr->size_idx);
/*
* Perform coalescing just in case there
* are any neighbouring free chunks.
*/
struct memory_block nm = heap_coalesce_huge(heap, m);
if (nm.chunk_id != m->chunk_id) {
m->m_ops->prep_hdr(&nm, MEMBLOCK_FREE, &ctx);
operation_process(&ctx);
}
*m = nm;
bucket_insert_block(heap->rt->default_bucket, m);
}
/*
* heap_populate_buckets -- (internal) creates volatile state of memory blocks
*/
static void
heap_populate_buckets(PMEMobjpool *pop)
{
struct pmalloc_heap *h = pop->heap;
if (h->zones_exhausted == h->max_zone)
return;
uint32_t zone_id = h->zones_exhausted++;
struct zone *z = &h->layout->zones[zone_id];
/* ignore zone and chunk headers */
VALGRIND_ADD_TO_GLOBAL_TX_IGNORE(z, sizeof (z->header) +
sizeof (z->chunk_headers));
if (z->header.magic != ZONE_HEADER_MAGIC)
heap_zone_init(pop, zone_id);
struct bucket *def_bucket = h->buckets[DEFAULT_BUCKET];
for (uint32_t i = 0; i < z->header.size_idx; ) {
struct chunk_header *hdr = &z->chunk_headers[i];
heap_chunk_write_footer(hdr, hdr->size_idx);
if (hdr->type == CHUNK_TYPE_RUN) {
struct chunk_run *run =
(struct chunk_run *)&z->chunks[i];
heap_populate_run_bucket(pop,
h->bucket_map[run->block_size], i, zone_id);
} else if (hdr->type == CHUNK_TYPE_FREE) {
struct memory_block m = {i, zone_id, hdr->size_idx, 0};
bucket_insert_block(def_bucket, m);
}
i += hdr->size_idx;
}
}
/*
* heap_chunk_init -- (internal) writes chunk header
*/
static void
heap_chunk_init(struct palloc_heap *heap, struct chunk_header *hdr,
uint16_t type, uint32_t size_idx)
{
struct chunk_header nhdr = {
.type = type,
.flags = 0,
.size_idx = size_idx
};
VALGRIND_DO_MAKE_MEM_UNDEFINED(hdr, sizeof(*hdr));
*hdr = nhdr; /* write the entire header (8 bytes) at once */
pmemops_persist(&heap->p_ops, hdr, sizeof(*hdr));
heap_chunk_write_footer(hdr, size_idx);
}
/*
* heap_zone_init -- (internal) writes zone's first chunk and header
*/
static void
heap_zone_init(struct palloc_heap *heap, uint32_t zone_id)
{
struct zone *z = ZID_TO_ZONE(heap->layout, zone_id);
uint32_t size_idx = get_zone_size_idx(zone_id, heap->rt->max_zone,
heap->size);
heap_chunk_init(heap, &z->chunk_headers[0], CHUNK_TYPE_FREE, size_idx);
struct zone_header nhdr = {
.size_idx = size_idx,
.magic = ZONE_HEADER_MAGIC,
};
z->header = nhdr; /* write the entire header (8 bytes) at once */
pmemops_persist(&heap->p_ops, &z->header, sizeof(z->header));
}
/*
* heap_run_init -- (internal) creates a run based on a chunk
*/
static void
heap_run_init(struct palloc_heap *heap, struct bucket *b,
const struct memory_block *m)
{
struct alloc_class *c = b->aclass;
ASSERTeq(c->type, CLASS_RUN);
struct zone *z = ZID_TO_ZONE(heap->layout, m->zone_id);
struct chunk_run *run = (struct chunk_run *)&z->chunks[m->chunk_id];
ASSERTne(m->size_idx, 0);
size_t runsize = SIZEOF_RUN(run, m->size_idx);
VALGRIND_DO_MAKE_MEM_UNDEFINED(run, runsize);
/* add/remove chunk_run and chunk_header to valgrind transaction */
VALGRIND_ADD_TO_TX(run, runsize);
run->block_size = c->unit_size;
pmemops_persist(&heap->p_ops, &run->block_size,
sizeof(run->block_size));
/* set all the bits */
memset(run->bitmap, 0xFF, sizeof(run->bitmap));
unsigned nval = c->run.bitmap_nval;
ASSERT(nval > 0);
/* clear only the bits available for allocations from this bucket */
memset(run->bitmap, 0, sizeof(uint64_t) * (nval - 1));
run->bitmap[nval - 1] = c->run.bitmap_lastval;
run->incarnation_claim = heap->run_id;
VALGRIND_SET_CLEAN(&run->incarnation_claim,
sizeof(run->incarnation_claim));
VALGRIND_REMOVE_FROM_TX(run, runsize);
pmemops_persist(&heap->p_ops, run->bitmap, sizeof(run->bitmap));
struct chunk_header run_data_hdr;
run_data_hdr.type = CHUNK_TYPE_RUN_DATA;
run_data_hdr.flags = 0;
struct chunk_header *data_hdr;
for (unsigned i = 1; i < m->size_idx; ++i) {
data_hdr = &z->chunk_headers[m->chunk_id + i];
VALGRIND_DO_MAKE_MEM_UNDEFINED(data_hdr, sizeof(*data_hdr));
VALGRIND_ADD_TO_TX(data_hdr, sizeof(*data_hdr));
run_data_hdr.size_idx = i;
*data_hdr = run_data_hdr;
VALGRIND_REMOVE_FROM_TX(data_hdr, sizeof(*data_hdr));
}
pmemops_persist(&heap->p_ops,
&z->chunk_headers[m->chunk_id + 1],
sizeof(struct chunk_header) * (m->size_idx - 1));
struct chunk_header *hdr = &z->chunk_headers[m->chunk_id];
ASSERT(hdr->type == CHUNK_TYPE_FREE);
VALGRIND_ADD_TO_TX(hdr, sizeof(*hdr));
struct chunk_header run_hdr;
run_hdr.size_idx = hdr->size_idx;
run_hdr.type = CHUNK_TYPE_RUN;
run_hdr.flags = header_type_to_flag[c->header_type];
*hdr = run_hdr;
VALGRIND_REMOVE_FROM_TX(hdr, sizeof(*hdr));
pmemops_persist(&heap->p_ops, hdr, sizeof(*hdr));
}
/*
* heap_run_insert -- (internal) inserts and splits a block of memory into a run
*/
static void
heap_run_insert(struct palloc_heap *heap, struct bucket *b,
const struct memory_block *m, uint32_t size_idx, uint16_t block_off)
{
struct alloc_class *c = b->aclass;
ASSERTeq(c->type, CLASS_RUN);
ASSERT(size_idx <= BITS_PER_VALUE);
ASSERT(block_off + size_idx <= c->run.bitmap_nallocs);
uint32_t unit_max = c->run.unit_max;
struct memory_block nm = *m;
nm.size_idx = unit_max - (block_off % unit_max);
nm.block_off = block_off;
if (nm.size_idx > size_idx)
nm.size_idx = size_idx;
do {
bucket_insert_block(b, &nm);
ASSERT(nm.size_idx <= UINT16_MAX);
ASSERT(nm.block_off + nm.size_idx <= UINT16_MAX);
nm.block_off = (uint16_t)(nm.block_off + (uint16_t)nm.size_idx);
size_idx -= nm.size_idx;
nm.size_idx = size_idx > unit_max ? unit_max : size_idx;
} while (size_idx != 0);
}
/*
* heap_process_run_metadata -- (internal) parses the run bitmap
*/
static uint32_t
heap_process_run_metadata(struct palloc_heap *heap, struct bucket *b,
const struct memory_block *m)
{
struct alloc_class *c = b->aclass;
ASSERTeq(c->type, CLASS_RUN);
uint16_t block_off = 0;
uint16_t block_size_idx = 0;
uint32_t inserted_blocks = 0;
struct zone *z = ZID_TO_ZONE(heap->layout, m->zone_id);
struct chunk_run *run = (struct chunk_run *)&z->chunks[m->chunk_id];
for (unsigned i = 0; i < c->run.bitmap_nval; ++i) {
ASSERT(i < MAX_BITMAP_VALUES);
uint64_t v = run->bitmap[i];
ASSERT(BITS_PER_VALUE * i <= UINT16_MAX);
block_off = (uint16_t)(BITS_PER_VALUE * i);
if (v == 0) {
heap_run_insert(heap, b, m, BITS_PER_VALUE, block_off);
inserted_blocks += BITS_PER_VALUE;
continue;
} else if (v == UINT64_MAX) {
continue;
}
for (unsigned j = 0; j < BITS_PER_VALUE; ++j) {
if (BIT_IS_CLR(v, j)) {
block_size_idx++;
} else if (block_size_idx != 0) {
ASSERT(block_off >= block_size_idx);
heap_run_insert(heap, b, m,
block_size_idx,
(uint16_t)(block_off - block_size_idx));
inserted_blocks += block_size_idx;
block_size_idx = 0;
}
if ((block_off++) == c->run.bitmap_nallocs) {
i = MAX_BITMAP_VALUES;
break;
}
}
if (block_size_idx != 0) {
ASSERT(block_off >= block_size_idx);
heap_run_insert(heap, b, m,
block_size_idx,
(uint16_t)(block_off - block_size_idx));
inserted_blocks += block_size_idx;
block_size_idx = 0;
}
}
return inserted_blocks;
}
/*
* heap_create_run -- (internal) initializes a new run on an existing free chunk
*/
static void
heap_create_run(struct palloc_heap *heap, struct bucket *b,
struct memory_block *m)
{
heap_run_init(heap, b, m);
memblock_rebuild_state(heap, m);
heap_process_run_metadata(heap, b, m);
}
/*
* heap_chunk_init -- (internal) writes chunk header
*/
static void
heap_chunk_init(PMEMobjpool *pop, struct chunk_header *hdr,
uint16_t type, uint32_t size_idx)
{
struct chunk_header nhdr = {
.type = type,
.flags = 0,
.size_idx = size_idx
};
*hdr = nhdr; /* write the entire header (8 bytes) at once */
pop->persist(hdr, sizeof (*hdr));
heap_chunk_write_footer(hdr, size_idx);
}
/*
* heap_zone_init -- (internal) writes zone's first chunk and header
*/
static void
heap_zone_init(PMEMobjpool *pop, uint32_t zone_id)
{
struct zone *z = &pop->heap->layout->zones[zone_id];
uint32_t size_idx = get_zone_size_idx(zone_id, pop->heap->max_zone,
pop->heap_size);
heap_chunk_init(pop, &z->chunk_headers[0], CHUNK_TYPE_FREE, size_idx);
struct zone_header nhdr = {
.size_idx = size_idx,
.magic = ZONE_HEADER_MAGIC,
};
z->header = nhdr; /* write the entire header (8 bytes) at once */
pop->persist(&z->header, sizeof (z->header));
}
/*
* heap_init_run -- (internal) creates a run based on a chunk
*/
static void
heap_init_run(PMEMobjpool *pop, struct bucket *b, struct chunk_header *hdr,
struct chunk_run *run)
{
/* add/remove chunk_run and chunk_header to valgrind transaction */
VALGRIND_ADD_TO_TX(run, sizeof (*run));
run->block_size = bucket_unit_size(b);
pop->persist(&run->block_size, sizeof (run->block_size));
ASSERT(hdr->type == CHUNK_TYPE_FREE);
/* set all the bits */
memset(run->bitmap, 0xFF, sizeof (run->bitmap));
/* clear only the bits available for allocations from this bucket */
memset(run->bitmap, 0, sizeof (uint64_t) * (bucket_bitmap_nval(b) - 1));
run->bitmap[bucket_bitmap_nval(b) - 1] = bucket_bitmap_lastval(b);
VALGRIND_REMOVE_FROM_TX(run, sizeof (*run));
pop->persist(run->bitmap, sizeof (run->bitmap));
VALGRIND_ADD_TO_TX(hdr, sizeof (*hdr));
hdr->type = CHUNK_TYPE_RUN;
VALGRIND_REMOVE_FROM_TX(hdr, sizeof (*hdr));
pop->persist(hdr, sizeof (*hdr));
}
