#include "H5HLmodule.h"
#include "H5private.h"
#include "H5Eprivate.h"
#include "H5Fprivate.h"
#include "H5HLpkg.h"
#include "H5MFprivate.h"
#define H5HL_MIN_HEAP 128
static H5HL_free_t *H5HL__remove_free(H5HL_t *heap, H5HL_free_t *fl);
static herr_t H5HL__minimize_heap_space(H5F_t *f, H5HL_t *heap);
static herr_t H5HL__dirty(H5HL_t *heap);
hbool_t H5_PKG_INIT_VAR = FALSE;
H5FL_DEFINE(H5HL_free_t);
H5FL_BLK_DEFINE(lheap_chunk);
BEGIN_FUNC(PRIV, ERR,
herr_t, SUCCEED, FAIL,
H5HL_create(H5F_t *f, size_t size_hint, haddr_t *addr_p))
H5HL_t *heap = NULL;
H5HL_prfx_t *prfx = NULL;
hsize_t total_size = 0;
HDassert(f);
HDassert(addr_p);
if(size_hint && size_hint < H5HL_SIZEOF_FREE(f))
size_hint = H5HL_SIZEOF_FREE(f);
size_hint = H5HL_ALIGN(size_hint);
if(NULL == (heap = H5HL__new(H5F_SIZEOF_SIZE(f), H5F_SIZEOF_ADDR(f), H5HL_SIZEOF_HDR(f))))
H5E_THROW(H5E_CANTALLOC, "can't allocate new heap struct");
total_size = heap->prfx_size + size_hint;
if(HADDR_UNDEF == (heap->prfx_addr = H5MF_alloc(f, H5FD_MEM_LHEAP, total_size)))
H5E_THROW(H5E_CANTALLOC, "unable to allocate file memory");
heap->single_cache_obj = TRUE;
heap->dblk_addr = heap->prfx_addr + (hsize_t)heap->prfx_size;
heap->dblk_size = size_hint;
if(size_hint)
if(NULL == (heap->dblk_image = H5FL_BLK_CALLOC(lheap_chunk, size_hint)))
H5E_THROW(H5E_CANTALLOC, "memory allocation failed");
if(size_hint) {
if(NULL == (heap->freelist = H5FL_MALLOC(H5HL_free_t)))
H5E_THROW(H5E_CANTALLOC, "memory allocation failed");
heap->freelist->offset = 0;
heap->freelist->size = size_hint;
heap->freelist->prev = heap->freelist->next = NULL;
heap->free_block = 0;
}
else {
heap->freelist = NULL;
heap->free_block = H5HL_FREE_NULL;
}
if(NULL == (prfx = H5HL__prfx_new(heap)))
H5E_THROW(H5E_CANTALLOC, "memory allocation failed");
if(FAIL == H5AC_insert_entry(f, H5AC_LHEAP_PRFX, heap->prfx_addr, prfx, H5AC__NO_FLAGS_SET))
H5E_THROW(H5E_CANTINIT, "unable to cache local heap prefix");
*addr_p = heap->prfx_addr;
CATCH
if(ret_value < 0) {
*addr_p = HADDR_UNDEF;
if(prfx) {
if(FAIL == H5HL__prfx_dest(prfx))
H5E_THROW(H5E_CANTFREE, "unable to destroy local heap prefix");
}
else {
if(heap) {
if(H5F_addr_defined(heap->prfx_addr))
if(FAIL == H5MF_xfree(f, H5FD_MEM_LHEAP, heap->prfx_addr, total_size))
H5E_THROW(H5E_CANTFREE, "can't release heap data?");
if(FAIL == H5HL__dest(heap))
H5E_THROW(H5E_CANTFREE, "unable to destroy local heap");
}
}
}
END_FUNC(PRIV)
BEGIN_FUNC(STATIC, ERR,
herr_t, SUCCEED, FAIL,
H5HL__minimize_heap_space(H5F_t *f, H5HL_t *heap))
size_t new_heap_size = heap->dblk_size;
HDassert(f);
HDassert(heap);
if(heap->freelist) {
H5HL_free_t *tmp_fl;
H5HL_free_t *last_fl = NULL;
for(tmp_fl = heap->freelist; tmp_fl; tmp_fl = tmp_fl->next)
if(tmp_fl->offset + tmp_fl->size == heap->dblk_size) {
last_fl = tmp_fl;
break;
}
if(last_fl) {
if(last_fl->size >= (heap->dblk_size / 2) && heap->dblk_size > H5HL_MIN_HEAP) {
while(new_heap_size > H5HL_MIN_HEAP &&
new_heap_size >= (last_fl->offset + H5HL_SIZEOF_FREE(f)))
new_heap_size /= 2;
if(new_heap_size < (last_fl->offset + H5HL_SIZEOF_FREE(f))) {
if(last_fl->prev == NULL && last_fl->next == NULL) {
new_heap_size *= 2;
last_fl->size = H5HL_ALIGN(new_heap_size - last_fl->offset);
new_heap_size = last_fl->offset + last_fl->size;
HDassert(last_fl->size >= H5HL_SIZEOF_FREE(f));
}
else {
new_heap_size = last_fl->offset;
last_fl = H5HL__remove_free(heap, last_fl);
}
}
else {
last_fl->size = H5HL_ALIGN(new_heap_size - last_fl->offset);
new_heap_size = last_fl->offset + last_fl->size;
HDassert(last_fl->size >= H5HL_SIZEOF_FREE(f));
HDassert(last_fl->size == H5HL_ALIGN(last_fl->size));
}
}
}
}
if(new_heap_size != heap->dblk_size) {
HDassert(new_heap_size < heap->dblk_size);
if(NULL == (heap->dblk_image = H5FL_BLK_REALLOC(lheap_chunk, heap->dblk_image, new_heap_size)))
H5E_THROW(H5E_CANTALLOC, "memory allocation failed");
if(FAIL == H5HL__dblk_realloc(f, heap, new_heap_size))
H5E_THROW(H5E_CANTRESIZE, "reallocating data block failed");
}
CATCH
END_FUNC(STATIC)
BEGIN_FUNC(PRIV, ERR,
H5HL_t *, NULL, NULL,
H5HL_protect(H5F_t *f, haddr_t addr, unsigned flags))
H5HL_cache_prfx_ud_t prfx_udata;
H5HL_prfx_t *prfx = NULL;
H5HL_dblk_t *dblk = NULL;
H5HL_t *heap = NULL;
unsigned prfx_cache_flags = H5AC__NO_FLAGS_SET;
unsigned dblk_cache_flags = H5AC__NO_FLAGS_SET;
HDassert(f);
HDassert(H5F_addr_defined(addr));
HDassert((flags & (unsigned)(~H5AC__READ_ONLY_FLAG)) == 0);
prfx_udata.sizeof_size = H5F_SIZEOF_SIZE(f);
prfx_udata.sizeof_addr = H5F_SIZEOF_ADDR(f);
prfx_udata.prfx_addr = addr;
prfx_udata.sizeof_prfx = H5HL_SIZEOF_HDR(f);
if(NULL == (prfx = (H5HL_prfx_t *)H5AC_protect(f, H5AC_LHEAP_PRFX, addr, &prfx_udata, flags)))
H5E_THROW(H5E_CANTPROTECT, "unable to load heap prefix");
heap = prfx->heap;
if(heap->prots == 0) {
if(heap->single_cache_obj)
prfx_cache_flags |= H5AC__PIN_ENTRY_FLAG;
else {
if(NULL == (dblk = (H5HL_dblk_t *)H5AC_protect(f, H5AC_LHEAP_DBLK, heap->dblk_addr, heap, flags)))
H5E_THROW(H5E_CANTPROTECT, "unable to load heap data block");
dblk_cache_flags |= H5AC__PIN_ENTRY_FLAG;
}
}
heap->prots++;
ret_value = heap;
CATCH
if(prfx && heap && H5AC_unprotect(f, H5AC_LHEAP_PRFX, heap->prfx_addr, prfx, prfx_cache_flags) < 0)
H5E_THROW(H5E_CANTUNPROTECT, "unable to release local heap prefix");
if(dblk && heap && H5AC_unprotect(f, H5AC_LHEAP_DBLK, heap->dblk_addr, dblk, dblk_cache_flags) < 0)
H5E_THROW(H5E_CANTUNPROTECT, "unable to release local heap data block");
END_FUNC(PRIV)
BEGIN_FUNC(PRIV, ERR,
void *, NULL, NULL,
H5HL_offset_into(const H5HL_t *heap, size_t offset))
HDassert(heap);
if(offset >= heap->dblk_size)
H5E_THROW(H5E_CANTGET, "unable to offset into local heap data block");
ret_value = heap->dblk_image + offset;
CATCH
END_FUNC(PRIV)
BEGIN_FUNC(PRIV, ERR,
herr_t, SUCCEED, FAIL,
H5HL_unprotect(H5HL_t *heap))
HDassert(heap);
heap->prots--;
if(heap->prots == 0) {
if(heap->single_cache_obj) {
if(FAIL == H5AC_unpin_entry(heap->prfx))
H5E_THROW(H5E_CANTUNPIN, "unable to unpin local heap data block");
}
else {
HDassert(heap->dblk);
if(FAIL == H5AC_unpin_entry(heap->dblk))
H5E_THROW(H5E_CANTUNPIN, "unable to unpin local heap data block");
}
}
CATCH
END_FUNC(PRIV)
BEGIN_FUNC(STATIC, NOERR,
H5HL_free_t *, NULL, -,
H5HL__remove_free(H5HL_t *heap, H5HL_free_t *fl))
if(fl->prev)
fl->prev->next = fl->next;
if(fl->next)
fl->next->prev = fl->prev;
if(!fl->prev)
heap->freelist = fl->next;
ret_value = (H5HL_free_t *)H5FL_FREE(H5HL_free_t, fl);
END_FUNC(STATIC)
BEGIN_FUNC(STATIC, ERR,
herr_t, SUCCEED, FAIL,
H5HL__dirty(H5HL_t *heap))
HDassert(heap);
HDassert(heap->prfx);
if(!heap->single_cache_obj) {
HDassert(heap->dblk);
if(FAIL == H5AC_mark_entry_dirty(heap->dblk))
H5E_THROW(H5E_CANTMARKDIRTY, "unable to mark heap data block as dirty");
}
if(FAIL == H5AC_mark_entry_dirty(heap->prfx))
H5E_THROW(H5E_CANTMARKDIRTY, "unable to mark heap prefix as dirty");
CATCH
END_FUNC(STATIC)
BEGIN_FUNC(PRIV, ERR,
size_t, UFAIL, UFAIL,
H5HL_insert(H5F_t *f, H5HL_t *heap, size_t buf_size, const void *buf))
H5HL_free_t *fl = NULL, *last_fl = NULL;
size_t offset = 0;
size_t need_size;
hbool_t found;
HDassert(f);
HDassert(heap);
HDassert(buf_size > 0);
HDassert(buf);
if(FAIL == H5HL__dirty(heap))
H5E_THROW(H5E_CANTMARKDIRTY, "unable to mark heap as dirty");
need_size = H5HL_ALIGN(buf_size);
for(fl = heap->freelist, found = FALSE; fl; fl = fl->next) {
if(fl->size > need_size && fl->size - need_size >= H5HL_SIZEOF_FREE(f)) {
offset = fl->offset;
fl->offset += need_size;
fl->size -= need_size;
HDassert(fl->offset == H5HL_ALIGN(fl->offset));
HDassert(fl->size == H5HL_ALIGN(fl->size));
found = TRUE;
break;
} else if(fl->size == need_size) {
offset = fl->offset;
fl = H5HL__remove_free(heap, fl);
found = TRUE;
break;
} else if(!last_fl || last_fl->offset < fl->offset) {
last_fl = fl;
}
}
if(found == FALSE) {
size_t need_more;
size_t new_dblk_size;
size_t old_dblk_size;
htri_t was_extended;
need_more = MAX(need_size, heap->dblk_size);
if(!(last_fl && last_fl->offset + last_fl->size == heap->dblk_size)
&& (need_more < (need_size + H5HL_SIZEOF_FREE(f))))
need_more = need_size;
new_dblk_size = heap->dblk_size + need_more;
HDassert(heap->dblk_size < new_dblk_size);
old_dblk_size = heap->dblk_size;
H5_CHECK_OVERFLOW(heap->dblk_size, size_t, hsize_t);
H5_CHECK_OVERFLOW(new_dblk_size, size_t, hsize_t);
was_extended = H5MF_try_extend(f, H5FD_MEM_LHEAP, heap->dblk_addr, (hsize_t)(heap->dblk_size), (hsize_t)need_more);
if(FAIL == was_extended)
H5E_THROW(H5E_CANTEXTEND, "error trying to extend heap");
if(was_extended == TRUE) {
if(heap->single_cache_obj) {
if(FAIL == H5AC_resize_entry(heap->prfx, (size_t)(heap->prfx_size + new_dblk_size)))
H5E_THROW(H5E_CANTRESIZE, "unable to resize heap prefix in cache");
}
else {
if(FAIL == H5AC_resize_entry(heap->dblk, (size_t)new_dblk_size))
H5E_THROW(H5E_CANTRESIZE, "unable to resize heap data block in cache");
}
heap->dblk_size = new_dblk_size;
}
else {
if(FAIL == H5HL__dblk_realloc(f, heap, new_dblk_size))
H5E_THROW(H5E_CANTRESIZE, "reallocating data block failed");
}
if(last_fl && last_fl->offset + last_fl->size == old_dblk_size) {
offset = last_fl->offset;
last_fl->offset += need_size;
last_fl->size += need_more - need_size;
HDassert(last_fl->offset == H5HL_ALIGN(last_fl->offset));
HDassert(last_fl->size == H5HL_ALIGN(last_fl->size));
if (last_fl->size < H5HL_SIZEOF_FREE(f)) {
#ifdef H5HL_DEBUG
if (H5DEBUG(HL) && last_fl->size) {
HDfprintf(H5DEBUG(HL), "H5HL: lost %lu bytes at line %d\n",
(unsigned long)(last_fl->size), __LINE__);
}
#endif
last_fl = H5HL__remove_free(heap, last_fl);
}
}
else {
offset = old_dblk_size;
if(need_more - need_size >= H5HL_SIZEOF_FREE(f)) {
if(NULL == (fl = H5FL_MALLOC(H5HL_free_t)))
H5E_THROW(H5E_CANTALLOC, "memory allocation failed");
fl->offset = old_dblk_size + need_size;
fl->size = need_more - need_size;
HDassert(fl->offset == H5HL_ALIGN(fl->offset));
HDassert(fl->size == H5HL_ALIGN(fl->size));
fl->prev = NULL;
fl->next = heap->freelist;
if(heap->freelist)
heap->freelist->prev = fl;
heap->freelist = fl;
#ifdef H5HL_DEBUG
} else if (H5DEBUG(HL) && need_more > need_size) {
HDfprintf(H5DEBUG(HL), "H5HL_insert: lost %lu bytes at line %d\n",
(unsigned long)(need_more - need_size), __LINE__);
#endif
}
}
#ifdef H5HL_DEBUG
if (H5DEBUG(HL)) {
HDfprintf(H5DEBUG(HL), "H5HL: resize mem buf from %lu to %lu bytes\n",
(unsigned long)(old_dblk_size),
(unsigned long)(old_dblk_size + need_more));
}
#endif
if(NULL == (heap->dblk_image = H5FL_BLK_REALLOC(lheap_chunk, heap->dblk_image, heap->dblk_size)))
H5E_THROW(H5E_CANTALLOC, "memory allocation failed");
HDmemset(heap->dblk_image + offset + buf_size, 0, (new_dblk_size - (offset + buf_size)));
}
H5MM_memcpy(heap->dblk_image + offset, buf, buf_size);
ret_value = offset;
CATCH
END_FUNC(PRIV)
BEGIN_FUNC(PRIV, ERR,
herr_t, SUCCEED, FAIL,
H5HL_remove(H5F_t *f, H5HL_t *heap, size_t offset, size_t size))
H5HL_free_t *fl = NULL;
HDassert(f);
HDassert(heap);
HDassert(size > 0);
HDassert(offset == H5HL_ALIGN(offset));
size = H5HL_ALIGN(size);
HDassert(offset < heap->dblk_size);
HDassert(offset + size <= heap->dblk_size);
if(FAIL == H5HL__dirty(heap))
H5E_THROW(H5E_CANTMARKDIRTY, "unable to mark heap as dirty");
fl = heap->freelist;
while(fl) {
H5HL_free_t *fl2 = NULL;
if((offset + size) == fl->offset) {
fl->offset = offset;
fl->size += size;
HDassert(fl->offset == H5HL_ALIGN(fl->offset));
HDassert(fl->size == H5HL_ALIGN(fl->size));
fl2 = fl->next;
while(fl2) {
if((fl2->offset + fl2->size) == fl->offset) {
fl->offset = fl2->offset;
fl->size += fl2->size;
HDassert(fl->offset == H5HL_ALIGN(fl->offset));
HDassert(fl->size == H5HL_ALIGN(fl->size));
fl2 = H5HL__remove_free(heap, fl2);
if(((fl->offset + fl->size) == heap->dblk_size) &&
((2 * fl->size) > heap->dblk_size)) {
if(FAIL == H5HL__minimize_heap_space(f, heap))
H5E_THROW(H5E_CANTFREE, "heap size minimization failed");
}
H5_LEAVE(SUCCEED);
}
fl2 = fl2->next;
}
if(((fl->offset + fl->size) == heap->dblk_size) &&
((2 * fl->size) > heap->dblk_size)) {
if(FAIL == H5HL__minimize_heap_space(f, heap))
H5E_THROW(H5E_CANTFREE, "heap size minimization failed");
}
H5_LEAVE(SUCCEED);
} else if(fl->offset + fl->size == offset) {
fl->size += size;
fl2 = fl->next;
HDassert(fl->size == H5HL_ALIGN(fl->size));
while(fl2) {
if(fl->offset + fl->size == fl2->offset) {
fl->size += fl2->size;
HDassert(fl->size == H5HL_ALIGN(fl->size));
fl2 = H5HL__remove_free(heap, fl2);
if(((fl->offset + fl->size) == heap->dblk_size) &&
((2 * fl->size) > heap->dblk_size)) {
if(FAIL == H5HL__minimize_heap_space(f, heap))
H5E_THROW(H5E_CANTFREE, "heap size minimization failed");
}
H5_LEAVE(SUCCEED);
}
fl2 = fl2->next;
}
if(((fl->offset + fl->size) == heap->dblk_size) &&
((2 * fl->size) > heap->dblk_size)) {
if(FAIL == H5HL__minimize_heap_space(f, heap))
H5E_THROW(H5E_CANTFREE, "heap size minimization failed");
}
H5_LEAVE(SUCCEED);
}
fl = fl->next;
}
if(size < H5HL_SIZEOF_FREE(f)) {
#ifdef H5HL_DEBUG
if(H5DEBUG(HL)) {
HDfprintf(H5DEBUG(HL), "H5HL: lost %lu bytes\n", (unsigned long) size);
}
#endif
H5_LEAVE(SUCCEED);
}
if(NULL == (fl = H5FL_MALLOC(H5HL_free_t)))
H5E_THROW(H5E_CANTALLOC, "memory allocation failed");
fl->offset = offset;
fl->size = size;
HDassert(fl->offset == H5HL_ALIGN(fl->offset));
HDassert(fl->size == H5HL_ALIGN(fl->size));
fl->prev = NULL;
fl->next = heap->freelist;
if(heap->freelist)
heap->freelist->prev = fl;
heap->freelist = fl;
if(((fl->offset + fl->size) == heap->dblk_size) && ((2 * fl->size) > heap->dblk_size))
if(FAIL == H5HL__minimize_heap_space(f, heap))
H5E_THROW(H5E_CANTFREE, "heap size minimization failed");
CATCH
END_FUNC(PRIV)
BEGIN_FUNC(PRIV, ERR,
herr_t, SUCCEED, FAIL,
H5HL_delete(H5F_t *f, haddr_t addr))
H5HL_t *heap = NULL;
H5HL_cache_prfx_ud_t prfx_udata;
H5HL_prfx_t *prfx = NULL;
H5HL_dblk_t *dblk = NULL;
unsigned cache_flags = H5AC__NO_FLAGS_SET;
HDassert(f);
HDassert(H5F_addr_defined(addr));
prfx_udata.sizeof_size = H5F_SIZEOF_SIZE(f);
prfx_udata.sizeof_addr = H5F_SIZEOF_ADDR(f);
prfx_udata.prfx_addr = addr;
prfx_udata.sizeof_prfx = H5HL_SIZEOF_HDR(f);
if(NULL == (prfx = (H5HL_prfx_t *)H5AC_protect(f, H5AC_LHEAP_PRFX, addr, &prfx_udata, H5AC__NO_FLAGS_SET)))
H5E_THROW(H5E_CANTPROTECT, "unable to load heap prefix");
heap = prfx->heap;
if(!heap->single_cache_obj)
if(NULL == (dblk = (H5HL_dblk_t *)H5AC_protect(f, H5AC_LHEAP_DBLK, heap->dblk_addr, heap, H5AC__NO_FLAGS_SET)))
H5E_THROW(H5E_CANTPROTECT, "unable to load heap data block");
cache_flags |= H5AC__DIRTIED_FLAG | H5AC__DELETED_FLAG | H5AC__FREE_FILE_SPACE_FLAG;
CATCH
if(dblk && heap && H5AC_unprotect(f, H5AC_LHEAP_DBLK, heap->dblk_addr, dblk, cache_flags) < 0)
H5E_THROW(H5E_CANTUNPROTECT, "unable to release local heap data block");
if(prfx && heap && H5AC_unprotect(f, H5AC_LHEAP_PRFX, heap->prfx_addr, prfx, cache_flags) < 0)
H5E_THROW(H5E_CANTUNPROTECT, "unable to release local heap prefix");
END_FUNC(PRIV)
BEGIN_FUNC(PRIV, ERR,
herr_t, SUCCEED, FAIL,
H5HL_get_size(H5F_t *f, haddr_t addr, size_t *size))
H5HL_cache_prfx_ud_t prfx_udata;
H5HL_prfx_t *prfx = NULL;
H5HL_t *heap;
HDassert(f);
HDassert(H5F_addr_defined(addr));
HDassert(size);
prfx_udata.sizeof_size = H5F_SIZEOF_SIZE(f);
prfx_udata.sizeof_addr = H5F_SIZEOF_ADDR(f);
prfx_udata.prfx_addr = addr;
prfx_udata.sizeof_prfx = H5HL_SIZEOF_HDR(f);
if(NULL == (prfx = (H5HL_prfx_t *)H5AC_protect(f, H5AC_LHEAP_PRFX, addr, &prfx_udata, H5AC__READ_ONLY_FLAG)))
H5E_THROW(H5E_CANTPROTECT, "unable to load heap prefix");
heap = prfx->heap;
*size = heap->dblk_size;
CATCH
if(prfx && FAIL == H5AC_unprotect(f, H5AC_LHEAP_PRFX, heap->prfx_addr, prfx, H5AC__NO_FLAGS_SET))
H5E_THROW(H5E_CANTUNPROTECT, "unable to release local heap prefix");
END_FUNC(PRIV)
BEGIN_FUNC(PRIV, ERR,
herr_t, SUCCEED, FAIL,
H5HL_heapsize(H5F_t *f, haddr_t addr, hsize_t *heap_size))
H5HL_cache_prfx_ud_t prfx_udata;
H5HL_prfx_t *prfx = NULL;
H5HL_t *heap;
HDassert(f);
HDassert(H5F_addr_defined(addr));
HDassert(heap_size);
prfx_udata.sizeof_size = H5F_SIZEOF_SIZE(f);
prfx_udata.sizeof_addr = H5F_SIZEOF_ADDR(f);
prfx_udata.prfx_addr = addr;
prfx_udata.sizeof_prfx = H5HL_SIZEOF_HDR(f);
if(NULL == (prfx = (H5HL_prfx_t *)H5AC_protect(f, H5AC_LHEAP_PRFX, addr, &prfx_udata, H5AC__READ_ONLY_FLAG)))
H5E_THROW(H5E_CANTPROTECT, "unable to load heap prefix");
heap = prfx->heap;
*heap_size += (hsize_t)(heap->prfx_size + heap->dblk_size);
CATCH
if(prfx && FAIL == H5AC_unprotect(f, H5AC_LHEAP_PRFX, heap->prfx_addr, prfx, H5AC__NO_FLAGS_SET))
H5E_THROW(H5E_CANTUNPROTECT, "unable to release local heap prefix");
END_FUNC(PRIV)