15155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru/*- 203ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra * Copyright (c) 2009, 2010, 2011 35155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * Thorsten Glaser <tg@mirbsd.org> 45155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * 55155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * Provided that these terms and disclaimer and all copyright notices 65155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * are retained or reproduced in an accompanying document, permission 703ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra * is granted to deal in this work without restriction, including un- 85155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * limited rights to use, publicly perform, distribute, sell, modify, 95155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * merge, give away, or sublicence. 105155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * 1103ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra * This work is provided "AS IS" and WITHOUT WARRANTY of any kind, to 125155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * the utmost extent permitted by applicable law, neither express nor 135155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * implied; without malicious intent or gross negligence. In no event 145155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * may a licensor, author or contributor be held liable for indirect, 155155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * direct, other damage, loss, or other issues arising in any way out 165155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * of dealing in the work, even if advised of the possibility of such 175155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * damage or existence of a defect, except proven that it results out 1803ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra * of said person's immediate fault when using the work as intended. 195155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru */ 205155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 215155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#include "sh.h" 225155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 2303ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra__RCSID("$MirOS: src/bin/mksh/lalloc.c,v 1.19 2011/09/07 15:24:16 tg Exp $"); 245155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 255155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru/* build with CPPFLAGS+= -DUSE_REALLOC_MALLOC=0 on ancient systems */ 265155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#if defined(USE_REALLOC_MALLOC) && (USE_REALLOC_MALLOC == 0) 2703ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra#define remalloc(p,n) ((p) == NULL ? malloc_osi(n) : realloc_osi((p), (n))) 285155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#else 2903ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra#define remalloc(p,n) realloc_osi((p), (n)) 305155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#endif 315155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 325155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#define ALLOC_ISUNALIGNED(p) (((ptrdiff_t)(p)) % ALLOC_SIZE) 335155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 345155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Querustatic ALLOC_ITEM *findptr(ALLOC_ITEM **, char *, Area *); 355155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 365155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queruvoid 375155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queruainit(Area *ap) 385155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru{ 395155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* area pointer is an ALLOC_ITEM, just the head of the list */ 405155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru ap->next = NULL; 415155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru} 425155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 435155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Querustatic ALLOC_ITEM * 445155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Querufindptr(ALLOC_ITEM **lpp, char *ptr, Area *ap) 455155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru{ 465155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru void *lp; 475155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 485155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#ifndef MKSH_SMALL 495155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru if (ALLOC_ISUNALIGNED(ptr)) 505155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru goto fail; 515155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#endif 525155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* get address of ALLOC_ITEM from user item */ 535155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* 545155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * note: the alignment of "ptr" to ALLOC_SIZE is checked 555155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru * above; the "void *" gets us rid of a gcc 2.95 warning 565155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru */ 575155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru *lpp = (lp = ptr - ALLOC_SIZE); 585155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* search for allocation item in group list */ 595155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru while (ap->next != lp) 605155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru if ((ap = ap->next) == NULL) { 615155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#ifndef MKSH_SMALL 625155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru fail: 635155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#endif 6403ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra#ifdef DEBUG 6503ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra internal_warningf("rogue pointer %zX in ap %zX", 6603ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra (size_t)ptr, (size_t)ap); 6703ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra /* try to get a coredump */ 6803ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra abort(); 6903ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra#else 7003ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra internal_errorf("rogue pointer %zX", (size_t)ptr); 7103ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra#endif 725155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru } 735155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru return (ap); 745155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru} 755155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 765155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queruvoid * 7703ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condraaresize2(void *ptr, size_t fac1, size_t fac2, Area *ap) 7803ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra{ 7903ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra if (notoktomul(fac1, fac2)) 8003ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra internal_errorf(Tintovfl, fac1, '*', fac2); 8103ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra return (aresize(ptr, fac1 * fac2, ap)); 8203ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra} 8303ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra 8403ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condravoid * 855155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queruaresize(void *ptr, size_t numb, Area *ap) 865155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru{ 875155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru ALLOC_ITEM *lp = NULL; 885155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 895155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* resizing (true) or newly allocating? */ 905155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru if (ptr != NULL) { 915155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru ALLOC_ITEM *pp; 925155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 935155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru pp = findptr(&lp, ptr, ap); 945155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru pp->next = lp->next; 955155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru } 965155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 9703ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra if (notoktoadd(numb, ALLOC_SIZE) || 985155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru (lp = remalloc(lp, numb + ALLOC_SIZE)) == NULL 995155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#ifndef MKSH_SMALL 1005155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru || ALLOC_ISUNALIGNED(lp) 1015155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru#endif 1025155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru ) 10303ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra internal_errorf(Toomem, (unsigned long)numb); 1045155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* this only works because Area is an ALLOC_ITEM */ 1055155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru lp->next = ap->next; 1065155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru ap->next = lp; 1075155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* return user item address */ 1085155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru return ((char *)lp + ALLOC_SIZE); 1095155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru} 1105155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 1115155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queruvoid 1125155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queruafree(void *ptr, Area *ap) 1135155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru{ 1145155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru if (ptr != NULL) { 1155155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru ALLOC_ITEM *lp, *pp; 1165155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 1175155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru pp = findptr(&lp, ptr, ap); 1185155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* unhook */ 1195155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru pp->next = lp->next; 1205155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* now free ALLOC_ITEM */ 12103ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra free_osimalloc(lp); 1225155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru } 1235155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru} 1245155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 1255155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queruvoid 1265155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queruafreeall(Area *ap) 1275155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru{ 1285155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru ALLOC_ITEM *lp; 1295155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru 1305155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* traverse group (linked list) */ 1315155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru while ((lp = ap->next) != NULL) { 1325155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* make next ALLOC_ITEM head of list */ 1335155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru ap->next = lp->next; 1345155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru /* free old head */ 13503ebf06f4e1112a0e9533b93062d169232c4cbfeGeremy Condra free_osimalloc(lp); 1365155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru } 1375155f1c7438ef540d7b25eb70aa1639579795b07Jean-Baptiste Queru} 138