1*dfc11533SChris Williamson /*
2*dfc11533SChris Williamson ** $Id: lmem.c,v 1.84.1.1 2013/04/12 18:48:47 roberto Exp $
3*dfc11533SChris Williamson ** Interface to Memory Manager
4*dfc11533SChris Williamson ** See Copyright Notice in lua.h
5*dfc11533SChris Williamson */
6*dfc11533SChris Williamson
7*dfc11533SChris Williamson
8*dfc11533SChris Williamson #include <sys/zfs_context.h>
9*dfc11533SChris Williamson
10*dfc11533SChris Williamson #define lmem_c
11*dfc11533SChris Williamson #define LUA_CORE
12*dfc11533SChris Williamson
13*dfc11533SChris Williamson #include "lua.h"
14*dfc11533SChris Williamson
15*dfc11533SChris Williamson #include "ldebug.h"
16*dfc11533SChris Williamson #include "ldo.h"
17*dfc11533SChris Williamson #include "lgc.h"
18*dfc11533SChris Williamson #include "lmem.h"
19*dfc11533SChris Williamson #include "lobject.h"
20*dfc11533SChris Williamson #include "lstate.h"
21*dfc11533SChris Williamson
22*dfc11533SChris Williamson
23*dfc11533SChris Williamson
24*dfc11533SChris Williamson /*
25*dfc11533SChris Williamson ** About the realloc function:
26*dfc11533SChris Williamson ** void * frealloc (void *ud, void *ptr, size_t osize, size_t nsize);
27*dfc11533SChris Williamson ** (`osize' is the old size, `nsize' is the new size)
28*dfc11533SChris Williamson **
29*dfc11533SChris Williamson ** * frealloc(ud, NULL, x, s) creates a new block of size `s' (no
30*dfc11533SChris Williamson ** matter 'x').
31*dfc11533SChris Williamson **
32*dfc11533SChris Williamson ** * frealloc(ud, p, x, 0) frees the block `p'
33*dfc11533SChris Williamson ** (in this specific case, frealloc must return NULL);
34*dfc11533SChris Williamson ** particularly, frealloc(ud, NULL, 0, 0) does nothing
35*dfc11533SChris Williamson ** (which is equivalent to free(NULL) in ANSI C)
36*dfc11533SChris Williamson **
37*dfc11533SChris Williamson ** frealloc returns NULL if it cannot create or reallocate the area
38*dfc11533SChris Williamson ** (any reallocation to an equal or smaller size cannot fail!)
39*dfc11533SChris Williamson */
40*dfc11533SChris Williamson
41*dfc11533SChris Williamson
42*dfc11533SChris Williamson
43*dfc11533SChris Williamson #define MINSIZEARRAY 4
44*dfc11533SChris Williamson
45*dfc11533SChris Williamson
luaM_growaux_(lua_State * L,void * block,int * size,size_t size_elems,int limit,const char * what)46*dfc11533SChris Williamson void *luaM_growaux_ (lua_State *L, void *block, int *size, size_t size_elems,
47*dfc11533SChris Williamson int limit, const char *what) {
48*dfc11533SChris Williamson void *newblock;
49*dfc11533SChris Williamson int newsize;
50*dfc11533SChris Williamson if (*size >= limit/2) { /* cannot double it? */
51*dfc11533SChris Williamson if (*size >= limit) /* cannot grow even a little? */
52*dfc11533SChris Williamson luaG_runerror(L, "too many %s (limit is %d)", what, limit);
53*dfc11533SChris Williamson newsize = limit; /* still have at least one free place */
54*dfc11533SChris Williamson }
55*dfc11533SChris Williamson else {
56*dfc11533SChris Williamson newsize = (*size)*2;
57*dfc11533SChris Williamson if (newsize < MINSIZEARRAY)
58*dfc11533SChris Williamson newsize = MINSIZEARRAY; /* minimum size */
59*dfc11533SChris Williamson }
60*dfc11533SChris Williamson newblock = luaM_reallocv(L, block, *size, newsize, size_elems);
61*dfc11533SChris Williamson *size = newsize; /* update only when everything else is OK */
62*dfc11533SChris Williamson return newblock;
63*dfc11533SChris Williamson }
64*dfc11533SChris Williamson
65*dfc11533SChris Williamson
luaM_toobig(lua_State * L)66*dfc11533SChris Williamson l_noret luaM_toobig (lua_State *L) {
67*dfc11533SChris Williamson luaG_runerror(L, "memory allocation error: block too big");
68*dfc11533SChris Williamson }
69*dfc11533SChris Williamson
70*dfc11533SChris Williamson
71*dfc11533SChris Williamson
72*dfc11533SChris Williamson /*
73*dfc11533SChris Williamson ** generic allocation routine.
74*dfc11533SChris Williamson */
luaM_realloc_(lua_State * L,void * block,size_t osize,size_t nsize)75*dfc11533SChris Williamson void *luaM_realloc_ (lua_State *L, void *block, size_t osize, size_t nsize) {
76*dfc11533SChris Williamson void *newblock;
77*dfc11533SChris Williamson global_State *g = G(L);
78*dfc11533SChris Williamson size_t realosize = (block) ? osize : 0;
79*dfc11533SChris Williamson lua_assert((realosize == 0) == (block == NULL));
80*dfc11533SChris Williamson #if defined(HARDMEMTESTS)
81*dfc11533SChris Williamson if (nsize > realosize && g->gcrunning)
82*dfc11533SChris Williamson luaC_fullgc(L, 1); /* force a GC whenever possible */
83*dfc11533SChris Williamson #endif
84*dfc11533SChris Williamson newblock = (*g->frealloc)(g->ud, block, osize, nsize);
85*dfc11533SChris Williamson if (newblock == NULL && nsize > 0) {
86*dfc11533SChris Williamson api_check(L, nsize > realosize,
87*dfc11533SChris Williamson "realloc cannot fail when shrinking a block");
88*dfc11533SChris Williamson if (g->gcrunning) {
89*dfc11533SChris Williamson luaC_fullgc(L, 1); /* try to free some memory... */
90*dfc11533SChris Williamson newblock = (*g->frealloc)(g->ud, block, osize, nsize); /* try again */
91*dfc11533SChris Williamson }
92*dfc11533SChris Williamson if (newblock == NULL)
93*dfc11533SChris Williamson luaD_throw(L, LUA_ERRMEM);
94*dfc11533SChris Williamson }
95*dfc11533SChris Williamson lua_assert((nsize == 0) == (newblock == NULL));
96*dfc11533SChris Williamson g->GCdebt = (g->GCdebt + nsize) - realosize;
97*dfc11533SChris Williamson return newblock;
98*dfc11533SChris Williamson }
99*dfc11533SChris Williamson
100