¿Limitar el uso de memoria de un guion Lua?

Question

Lo he visto decir muchas veces que no hay forma de limitar el uso de la memoria de un guion Lua, incluidas las personas saltando a través de aros para evitar que los guiones de Lua creen funciones y tablas. Pero dado que lua_newstate le permite pasar un asignador personalizado, ¿no podría uno simplemente usar eso para limitar el consumo de memoria? En el peor de los casos, uno podría usar un asignador basado en arena y poner un límite estricto incluso en la cantidad de memoria que podría ser utilizada por la fragmentación.

¿Falta algo aquí?

Answer 1

static void *l_alloc_restricted (void *ud, void *ptr, size_t osize, size_t nsize) 
{ 
    const int MAX_SIZE = 1024; /* set limit here */ 
    int *used = (int *)ud; 

    if(ptr == NULL) { 
    /* 
     * <http://www.lua.org/manual/5.2/manual.html#lua_Alloc>: 
     * When ptr is NULL, osize encodes the kind of object that Lua is 
     * allocating. 
     * 
     * Since we don’t care about that, just mark it as 0. 
     */ 
    osize = 0; 
    } 

    if (nsize == 0) 
    { 
    free(ptr); 
    *used -= osize; /* substract old size from used memory */ 
    return NULL; 
    } 
    else 
    { 
    if (*used + (nsize - osize) > MAX_SIZE) /* too much memory in use */ 
     return NULL; 
    ptr = realloc(ptr, nsize); 
    if (ptr) /* reallocation successful? */ 
     *used += (nsize - osize); 
    return ptr; 
    } 
} 

Para hacer uso de su Lua asignador, puede utilizar

int *ud = malloc(sizeof(int)); *ud = 0; 
lua_State *L = lua_State *lua_newstate (l_alloc_restricted, ud); 

Nota: No he probado la fuente, pero debería funcionar.