Limitare l'utilizzo della memoria di uno script Lua?

Question

Ho visto più volte che non c'è modo di limitare l'utilizzo della memoria di uno script Lua, comprese le persone che saltano attraverso i cerchi per impedire agli script Lua di creare funzioni e tabelle. Ma dato che lua_newstate ti permette di passare un allocatore personalizzato, non è possibile usarlo solo per limitare il consumo di memoria? Nel peggiore dei casi, si potrebbe usare un allocatore basato su arena e porre un limite rigido anche alla quantità di memoria che potrebbe essere utilizzata dalla frammentazione.

Mi manca qualcosa qui?

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; 
    } 
} 

Per rendere Lua utilizzare l'allocator, è possibile utilizzare

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

Nota: non ho ancora testato la fonte, ma dovrebbe funzionare.