X-Git-Url: https://git.distorted.org.uk/u/mdw/catacomb/blobdiff_plain/c9110b350af5f22c32dc7523999a2aeae6e51517..d47453544cb2ee0b357a94a7aafbb8904ce60ad7:/mparena.c diff --git a/mparena.c b/mparena.c index e7c12cd..95f88d5 100644 --- a/mparena.c +++ b/mparena.c @@ -1,6 +1,6 @@ /* -*-c-*- * - * $Id: mparena.c,v 1.4 1999/12/10 23:28:52 mdw Exp $ + * $Id: mparena.c,v 1.5 2000/06/17 11:35:48 mdw Exp $ * * Allocation and freeing of MP buffers * @@ -30,6 +30,9 @@ /*----- Revision history --------------------------------------------------* * * $Log: mparena.c,v $ + * Revision 1.5 2000/06/17 11:35:48 mdw + * Overhaul to use mLib's arena system underneath. + * * Revision 1.4 1999/12/10 23:28:52 mdw * Memory allocation counting. * @@ -50,7 +53,8 @@ #include #include -#include +#include +#include #include #include "mparena.h" @@ -74,13 +78,6 @@ /* #define MPARENA_DEBUG "mparena.out" */ -/*----- Default allocator -------------------------------------------------*/ - -static void *defalloc(mparena *a, size_t sz) { return xmalloc(sz); } -static void deffree(mparena *a, void *p) { free(p); } - -mparena_ops mparena_defaultops = { defalloc, deffree }; - /*----- Static variables --------------------------------------------------*/ #ifdef MPARENA_DEBUG @@ -97,12 +94,10 @@ mparena_ops mparena_defaultops = { defalloc, deffree }; #endif -static mparena arena = MPARENA_INIT; +/*----- Standard arenas ---------------------------------------------------*/ -#define MPARENA_RESOLVE(a) do { \ - if ((a) == MPARENA_GLOBAL) \ - (a) = &arena; \ -} while (0) +mparena mparena_global = MPARENA_INIT; +mparena mparena_secure = MPARENA_INIT; /*----- Main code ---------------------------------------------------------*/ @@ -146,28 +141,20 @@ void mparena_create(mparena *a) { a->root = 0; a->n = 0; - a->ops = &mparena_defaultops; + a->a = &arena_stdlib; } -/* --- @mparena_setops@ --- * +/* --- @mparena_setarena@ --- * * - * Arguments: @mparena *a@ = pointer to arena block - * @mparena_ops *ops@ = pointer to operations block or null + * Arguments: @mparena *a@ = pointer to MP arena block + * @arena *aa@ = pointer to arena * - * Returns: The previous operations block. + * Returns: --- * - * Use: Sets or queries the operations attached to an arena. + * Use: Sets the underlying arena for an MP arena. */ -mparena_ops *mparena_setops(mparena *a, mparena_ops *ops) -{ - mparena_ops *o; - MPARENA_RESOLVE(a); - o = a->ops; - if (ops) - a->ops = ops; - return (0); -} +extern void mparena_setarena(mparena *a, arena *aa) { a->a = aa; } /* --- @mparena_destroy@ --- * * @@ -182,7 +169,7 @@ mparena_ops *mparena_setops(mparena *a, mparena_ops *ops) static void tfree(mparena *a, mparena_node *n) { - a->ops->free(a, n->v); + A_FREE(a->a, n->v); if (n->left) tfree(a, n->left); if (n->right) @@ -208,7 +195,6 @@ void mparena_destroy(mparena *a) unsigned mparena_count(mparena *a) { - MPARENA_RESOLVE(a); return (a->n); } @@ -227,8 +213,11 @@ unsigned mparena_count(mparena *a) mpw *mpalloc(mparena *a, size_t sz) { - MPARENA_RESOLVE(a); - return (a->ops->alloc(a, MPWS(sz))); + mpw *v; + v = A_ALLOC(a->a, MPWS(sz)); + if (!v) + THROW(EXC_NOMEM); + return (v); } #else @@ -238,7 +227,6 @@ mpw *mpalloc(mparena *a, size_t sz) mparena_node **nn, *n; mpw *v; - MPARENA_RESOLVE(a); nn = &a->root; #ifdef MPARENA_DEBUG @@ -255,7 +243,8 @@ again: #ifdef MPARENA_DEBUG fputs(" failed\n", debugfp); #endif - v = a->ops->alloc(a, MPWS(sz + 1)); + if ((v = A_ALLOC(a->a, MPWS(sz + 1))) == 0) + THROW(EXC_NOMEM); v[0] = sz; a->n++; return (v + 1); @@ -325,8 +314,7 @@ again: void mpfree(mparena *a, mpw *v) { - MPARENA_RESOLVE(a); - a->ops->free(a, v); + A_FREE(a->a, v); } #else @@ -336,8 +324,6 @@ void mpfree(mparena *a, mpw *v) mparena_node **nn, *n; size_t sz = *--v; - MPARENA_RESOLVE(a); - #ifdef MPARENA_DEBUG MPARENA_OPENFILE; fprintf(debugfp, "free %u\n before: ", sz);