1 /*- 2 * Copyright (c) 2009, 2010, 2011, 2013 3 * Thorsten Glaser <tg (at) mirbsd.org> 4 * 5 * Provided that these terms and disclaimer and all copyright notices 6 * are retained or reproduced in an accompanying document, permission 7 * is granted to deal in this work without restriction, including un- 8 * limited rights to use, publicly perform, distribute, sell, modify, 9 * merge, give away, or sublicence. 10 * 11 * This work is provided "AS IS" and WITHOUT WARRANTY of any kind, to 12 * the utmost extent permitted by applicable law, neither express nor 13 * implied; without malicious intent or gross negligence. In no event 14 * may a licensor, author or contributor be held liable for indirect, 15 * direct, other damage, loss, or other issues arising in any way out 16 * of dealing in the work, even if advised of the possibility of such 17 * damage or existence of a defect, except proven that it results out 18 * of said person's immediate fault when using the work as intended. 19 */ 20 21 #include "sh.h" 22 23 __RCSID("$MirOS: src/bin/mksh/lalloc.c,v 1.20 2013/06/03 22:28:33 tg Exp $"); 24 25 /* build with CPPFLAGS+= -DUSE_REALLOC_MALLOC=0 on ancient systems */ 26 #if defined(USE_REALLOC_MALLOC) && (USE_REALLOC_MALLOC == 0) 27 #define remalloc(p,n) ((p) == NULL ? malloc_osi(n) : realloc_osi((p), (n))) 28 #else 29 #define remalloc(p,n) realloc_osi((p), (n)) 30 #endif 31 32 #define ALLOC_ISUNALIGNED(p) (((ptrdiff_t)(p)) % ALLOC_SIZE) 33 34 static ALLOC_ITEM *findptr(ALLOC_ITEM **, char *, Area *); 35 36 void 37 ainit(Area *ap) 38 { 39 /* area pointer is an ALLOC_ITEM, just the head of the list */ 40 ap->next = NULL; 41 } 42 43 static ALLOC_ITEM * 44 findptr(ALLOC_ITEM **lpp, char *ptr, Area *ap) 45 { 46 void *lp; 47 48 #ifndef MKSH_SMALL 49 if (ALLOC_ISUNALIGNED(ptr)) 50 goto fail; 51 #endif 52 /* get address of ALLOC_ITEM from user item */ 53 /* 54 * note: the alignment of "ptr" to ALLOC_SIZE is checked 55 * above; the "void *" gets us rid of a gcc 2.95 warning 56 */ 57 *lpp = (lp = ptr - ALLOC_SIZE); 58 /* search for allocation item in group list */ 59 while (ap->next != lp) 60 if ((ap = ap->next) == NULL) { 61 #ifndef MKSH_SMALL 62 fail: 63 #endif 64 #ifdef DEBUG 65 internal_warningf("rogue pointer %zX in ap %zX", 66 (size_t)ptr, (size_t)ap); 67 /* try to get a coredump */ 68 abort(); 69 #else 70 internal_errorf("rogue pointer %zX", (size_t)ptr); 71 #endif 72 } 73 return (ap); 74 } 75 76 void * 77 aresize2(void *ptr, size_t fac1, size_t fac2, Area *ap) 78 { 79 if (notoktomul(fac1, fac2)) 80 internal_errorf(Tintovfl, fac1, '*', fac2); 81 return (aresize(ptr, fac1 * fac2, ap)); 82 } 83 84 void * 85 aresize(void *ptr, size_t numb, Area *ap) 86 { 87 ALLOC_ITEM *lp = NULL; 88 89 /* resizing (true) or newly allocating? */ 90 if (ptr != NULL) { 91 ALLOC_ITEM *pp; 92 93 pp = findptr(&lp, ptr, ap); 94 pp->next = lp->next; 95 } 96 97 if (notoktoadd(numb, ALLOC_SIZE) || 98 (lp = remalloc(lp, numb + ALLOC_SIZE)) == NULL 99 #ifndef MKSH_SMALL 100 || ALLOC_ISUNALIGNED(lp) 101 #endif 102 ) 103 internal_errorf(Toomem, numb); 104 /* this only works because Area is an ALLOC_ITEM */ 105 lp->next = ap->next; 106 ap->next = lp; 107 /* return user item address */ 108 return ((char *)lp + ALLOC_SIZE); 109 } 110 111 void 112 afree(void *ptr, Area *ap) 113 { 114 if (ptr != NULL) { 115 ALLOC_ITEM *lp, *pp; 116 117 pp = findptr(&lp, ptr, ap); 118 /* unhook */ 119 pp->next = lp->next; 120 /* now free ALLOC_ITEM */ 121 free_osimalloc(lp); 122 } 123 } 124 125 void 126 afreeall(Area *ap) 127 { 128 ALLOC_ITEM *lp; 129 130 /* traverse group (linked list) */ 131 while ((lp = ap->next) != NULL) { 132 /* make next ALLOC_ITEM head of list */ 133 ap->next = lp->next; 134 /* free old head */ 135 free_osimalloc(lp); 136 } 137 } 138
