96 lines
1.8 KiB
C
96 lines
1.8 KiB
C
#include "sh.h"
|
|
|
|
__RCSID("$MirOS: src/bin/mksh/lalloc.c,v 1.3 2009/03/23 10:31:15 tg Exp $");
|
|
|
|
#ifndef SIZE_MAX
|
|
#ifdef SIZE_T_MAX
|
|
#define SIZE_MAX SIZE_T_MAX
|
|
#else
|
|
#define SIZE_MAX ((size_t)-1)
|
|
#endif
|
|
#endif
|
|
|
|
struct lalloc {
|
|
struct lalloc *next; /* entry pointer, must be first */
|
|
Area *group; /* group backpointer */
|
|
};
|
|
|
|
static void findptr(struct lalloc **, struct lalloc **, char *, Area *);
|
|
|
|
void
|
|
ainit(Area *ap)
|
|
{
|
|
ap->ent = NULL;
|
|
}
|
|
|
|
static void
|
|
findptr(struct lalloc **lpp, struct lalloc **ppp, char *ptr, Area *ap)
|
|
{
|
|
*lpp = (struct lalloc *)(ptr - sizeof (struct lalloc));
|
|
if ((*lpp)->group != ap)
|
|
notfound:
|
|
internal_errorf("pointer %p not in group %p", ptr, ap);
|
|
*ppp = (struct lalloc *)ap;
|
|
while ((*ppp)->next != *lpp)
|
|
if (((*ppp) = (*ppp)->next) == NULL)
|
|
goto notfound;
|
|
}
|
|
|
|
void *
|
|
aresize(void *ptr, size_t numb, Area *ap)
|
|
{
|
|
struct lalloc *lp, *pp;
|
|
|
|
if (numb >= SIZE_MAX - sizeof (struct lalloc))
|
|
goto failure;
|
|
|
|
if (ptr == NULL) {
|
|
pp = (struct lalloc *)ap;
|
|
lp = malloc(numb + sizeof (struct lalloc));
|
|
} else {
|
|
findptr(&lp, &pp, ptr, ap);
|
|
lp = realloc(lp, numb + sizeof (struct lalloc));
|
|
}
|
|
if (lp == NULL) {
|
|
failure:
|
|
internal_errorf("cannot allocate %lu data bytes",
|
|
(unsigned long)numb);
|
|
}
|
|
if (ptr == NULL) {
|
|
lp->group = ap;
|
|
lp->next = ap->ent;
|
|
}
|
|
pp->next = lp;
|
|
return ((char *)lp + sizeof (struct lalloc));
|
|
}
|
|
|
|
void
|
|
afree(void *ptr, Area *ap)
|
|
{
|
|
struct lalloc *lp, *pp;
|
|
|
|
if (ptr == NULL)
|
|
return;
|
|
|
|
findptr(&lp, &pp, ptr, ap);
|
|
pp->next = lp->next;
|
|
/* lp->group = NULL; */
|
|
free(lp);
|
|
}
|
|
|
|
void
|
|
afreeall(Area *ap)
|
|
{
|
|
struct lalloc *lp;
|
|
|
|
while ((lp = ap->ent) != NULL) {
|
|
#ifndef MKSH_SMALL
|
|
if (lp->group != ap)
|
|
internal_errorf("rogue pointer in group %p", ap);
|
|
#endif
|
|
ap->ent = lp->next;
|
|
/* lp->group = NULL; */
|
|
free(lp);
|
|
}
|
|
}
|