|
Packit Service |
95ac19 |
/*-
|
|
Packit Service |
95ac19 |
* Copyright (c) 2009, 2010, 2011, 2013, 2014, 2016
|
|
Packit Service |
95ac19 |
* mirabilos <m@mirbsd.org>
|
|
Packit Service |
95ac19 |
*
|
|
Packit Service |
95ac19 |
* Provided that these terms and disclaimer and all copyright notices
|
|
Packit Service |
95ac19 |
* are retained or reproduced in an accompanying document, permission
|
|
Packit Service |
95ac19 |
* is granted to deal in this work without restriction, including un-
|
|
Packit Service |
95ac19 |
* limited rights to use, publicly perform, distribute, sell, modify,
|
|
Packit Service |
95ac19 |
* merge, give away, or sublicence.
|
|
Packit Service |
95ac19 |
*
|
|
Packit Service |
95ac19 |
* This work is provided "AS IS" and WITHOUT WARRANTY of any kind, to
|
|
Packit Service |
95ac19 |
* the utmost extent permitted by applicable law, neither express nor
|
|
Packit Service |
95ac19 |
* implied; without malicious intent or gross negligence. In no event
|
|
Packit Service |
95ac19 |
* may a licensor, author or contributor be held liable for indirect,
|
|
Packit Service |
95ac19 |
* direct, other damage, loss, or other issues arising in any way out
|
|
Packit Service |
95ac19 |
* of dealing in the work, even if advised of the possibility of such
|
|
Packit Service |
95ac19 |
* damage or existence of a defect, except proven that it results out
|
|
Packit Service |
95ac19 |
* of said person's immediate fault when using the work as intended.
|
|
Packit Service |
95ac19 |
*/
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
#include "sh.h"
|
|
Packit Service |
95ac19 |
#ifdef MKSH_ALLOC_CATCH_UNDERRUNS
|
|
Packit Service |
95ac19 |
#include <err.h>
|
|
Packit Service |
95ac19 |
#endif
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
__RCSID("$MirOS: src/bin/mksh/lalloc.c,v 1.26 2016/02/26 21:53:36 tg Exp $");
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
/* build with CPPFLAGS+= -DUSE_REALLOC_MALLOC=0 on ancient systems */
|
|
Packit Service |
95ac19 |
#if defined(USE_REALLOC_MALLOC) && (USE_REALLOC_MALLOC == 0)
|
|
Packit Service |
95ac19 |
#define remalloc(p,n) ((p) == NULL ? malloc_osi(n) : realloc_osi((p), (n)))
|
|
Packit Service |
95ac19 |
#else
|
|
Packit Service |
95ac19 |
#define remalloc(p,n) realloc_osi((p), (n))
|
|
Packit Service |
95ac19 |
#endif
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
static struct lalloc_common *findptr(struct lalloc_common **, char *, Area *);
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
#ifndef MKSH_ALLOC_CATCH_UNDERRUNS
|
|
Packit Service |
95ac19 |
#define ALLOC_ISUNALIGNED(p) (((size_t)(p)) % sizeof(struct lalloc_common))
|
|
Packit Service |
95ac19 |
#else
|
|
Packit Service |
95ac19 |
#define ALLOC_ISUNALIGNED(p) (((size_t)(p)) & 4095)
|
|
Packit Service |
95ac19 |
#undef remalloc
|
|
Packit Service |
95ac19 |
#undef free_osimalloc
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
static void
|
|
Packit Service |
95ac19 |
free_osimalloc(void *ptr)
|
|
Packit Service |
95ac19 |
{
|
|
Packit Service |
95ac19 |
struct lalloc_item *lp = ptr;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
if (munmap(lp, lp->len))
|
|
Packit Service |
95ac19 |
err(1, "free_osimalloc");
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
static void *
|
|
Packit Service |
95ac19 |
remalloc(void *ptr, size_t size)
|
|
Packit Service |
95ac19 |
{
|
|
Packit Service |
95ac19 |
struct lalloc_item *lp, *lold = ptr;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
size = (size + 4095) & ~(size_t)4095;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
if (lold && lold->len >= size)
|
|
Packit Service |
95ac19 |
return (ptr);
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
if ((lp = mmap(NULL, size, PROT_READ | PROT_WRITE,
|
|
Packit Service |
95ac19 |
MAP_ANON | MAP_PRIVATE, -1, (off_t)0)) == MAP_FAILED)
|
|
Packit Service |
95ac19 |
err(1, "remalloc: mmap(%zu)", size);
|
|
Packit Service |
95ac19 |
if (ALLOC_ISUNALIGNED(lp))
|
|
Packit Service |
95ac19 |
errx(1, "remalloc: unaligned(%p)", lp);
|
|
Packit Service |
95ac19 |
if (mprotect(((char *)lp) + 4096, 4096, PROT_NONE))
|
|
Packit Service |
95ac19 |
err(1, "remalloc: mprotect");
|
|
Packit Service |
95ac19 |
lp->len = size;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
if (lold) {
|
|
Packit Service |
95ac19 |
memcpy(((char *)lp) + 8192, ((char *)lold) + 8192,
|
|
Packit Service |
95ac19 |
lold->len - 8192);
|
|
Packit Service |
95ac19 |
if (munmap(lold, lold->len))
|
|
Packit Service |
95ac19 |
err(1, "remalloc: munmap");
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
return (lp);
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
#endif
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
void
|
|
Packit Service |
95ac19 |
ainit(Area *ap)
|
|
Packit Service |
95ac19 |
{
|
|
Packit Service |
95ac19 |
#ifdef MKSH_ALLOC_CATCH_UNDERRUNS
|
|
Packit Service |
95ac19 |
if (sysconf(_SC_PAGESIZE) != 4096) {
|
|
Packit Service |
95ac19 |
fprintf(stderr, "mksh: fatal: pagesize %lu not 4096!\n",
|
|
Packit Service |
95ac19 |
sysconf(_SC_PAGESIZE));
|
|
Packit Service |
95ac19 |
fflush(stderr);
|
|
Packit Service |
95ac19 |
abort();
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
#endif
|
|
Packit Service |
95ac19 |
/* area pointer and items share struct lalloc_common */
|
|
Packit Service |
95ac19 |
ap->next = NULL;
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
static struct lalloc_common *
|
|
Packit Service |
95ac19 |
findptr(struct lalloc_common **lpp, char *ptr, Area *ap)
|
|
Packit Service |
95ac19 |
{
|
|
Packit Service |
95ac19 |
void *lp;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
#ifndef MKSH_SMALL
|
|
Packit Service |
95ac19 |
if (ALLOC_ISUNALIGNED(ptr))
|
|
Packit Service |
95ac19 |
goto fail;
|
|
Packit Service |
95ac19 |
#endif
|
|
Packit Service |
95ac19 |
/* get address of ALLOC_ITEM from user item */
|
|
Packit Service |
95ac19 |
/*
|
|
Packit Service |
95ac19 |
* note: the alignment of "ptr" to ALLOC_ITEM is checked
|
|
Packit Service |
95ac19 |
* above; the "void *" gets us rid of a gcc 2.95 warning
|
|
Packit Service |
95ac19 |
*/
|
|
Packit Service |
95ac19 |
*lpp = (lp = ptr - sizeof(ALLOC_ITEM));
|
|
Packit Service |
95ac19 |
/* search for allocation item in group list */
|
|
Packit Service |
95ac19 |
while (ap->next != lp)
|
|
Packit Service |
95ac19 |
if ((ap = ap->next) == NULL) {
|
|
Packit Service |
95ac19 |
#ifndef MKSH_SMALL
|
|
Packit Service |
95ac19 |
fail:
|
|
Packit Service |
95ac19 |
#endif
|
|
Packit Service |
95ac19 |
#ifdef DEBUG
|
|
Packit Service |
95ac19 |
internal_warningf("rogue pointer %zX in ap %zX",
|
|
Packit Service |
95ac19 |
(size_t)ptr, (size_t)ap);
|
|
Packit Service |
95ac19 |
/* try to get a coredump */
|
|
Packit Service |
95ac19 |
abort();
|
|
Packit Service |
95ac19 |
#else
|
|
Packit Service |
95ac19 |
internal_errorf("rogue pointer %zX", (size_t)ptr);
|
|
Packit Service |
95ac19 |
#endif
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
return (ap);
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
void *
|
|
Packit Service |
95ac19 |
aresize2(void *ptr, size_t fac1, size_t fac2, Area *ap)
|
|
Packit Service |
95ac19 |
{
|
|
Packit Service |
95ac19 |
if (notoktomul(fac1, fac2))
|
|
Packit Service |
95ac19 |
internal_errorf(Tintovfl, fac1, '*', fac2);
|
|
Packit Service |
95ac19 |
return (aresize(ptr, fac1 * fac2, ap));
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
void *
|
|
Packit Service |
95ac19 |
aresize(void *ptr, size_t numb, Area *ap)
|
|
Packit Service |
95ac19 |
{
|
|
Packit Service |
95ac19 |
struct lalloc_common *lp = NULL;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
/* resizing (true) or newly allocating? */
|
|
Packit Service |
95ac19 |
if (ptr != NULL) {
|
|
Packit Service |
95ac19 |
struct lalloc_common *pp;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
pp = findptr(&lp, ptr, ap);
|
|
Packit Service |
95ac19 |
pp->next = lp->next;
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
if (notoktoadd(numb, sizeof(ALLOC_ITEM)) ||
|
|
Packit Service |
95ac19 |
(lp = remalloc(lp, numb + sizeof(ALLOC_ITEM))) == NULL
|
|
Packit Service |
95ac19 |
#ifndef MKSH_SMALL
|
|
Packit Service |
95ac19 |
|| ALLOC_ISUNALIGNED(lp)
|
|
Packit Service |
95ac19 |
#endif
|
|
Packit Service |
95ac19 |
)
|
|
Packit Service |
95ac19 |
internal_errorf(Toomem, numb);
|
|
Packit Service |
95ac19 |
/* area pointer and items share struct lalloc_common */
|
|
Packit Service |
95ac19 |
lp->next = ap->next;
|
|
Packit Service |
95ac19 |
ap->next = lp;
|
|
Packit Service |
95ac19 |
/* return user item address */
|
|
Packit Service |
95ac19 |
return ((char *)lp + sizeof(ALLOC_ITEM));
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
void
|
|
Packit Service |
95ac19 |
afree(void *ptr, Area *ap)
|
|
Packit Service |
95ac19 |
{
|
|
Packit Service |
95ac19 |
if (ptr != NULL) {
|
|
Packit Service |
95ac19 |
struct lalloc_common *lp, *pp;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
pp = findptr(&lp, ptr, ap);
|
|
Packit Service |
95ac19 |
/* unhook */
|
|
Packit Service |
95ac19 |
pp->next = lp->next;
|
|
Packit Service |
95ac19 |
/* now free ALLOC_ITEM */
|
|
Packit Service |
95ac19 |
free_osimalloc(lp);
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
void
|
|
Packit Service |
95ac19 |
afreeall(Area *ap)
|
|
Packit Service |
95ac19 |
{
|
|
Packit Service |
95ac19 |
struct lalloc_common *lp;
|
|
Packit Service |
95ac19 |
|
|
Packit Service |
95ac19 |
/* traverse group (linked list) */
|
|
Packit Service |
95ac19 |
while ((lp = ap->next) != NULL) {
|
|
Packit Service |
95ac19 |
/* make next ALLOC_ITEM head of list */
|
|
Packit Service |
95ac19 |
ap->next = lp->next;
|
|
Packit Service |
95ac19 |
/* free old head */
|
|
Packit Service |
95ac19 |
free_osimalloc(lp);
|
|
Packit Service |
95ac19 |
}
|
|
Packit Service |
95ac19 |
}
|