Hạn chế sử dụng bộ nhớ của tập lệnh Lua?

Question

Tôi đã nhìn thấy nó nói nhiều lần rằng không có cách nào để hạn chế sử dụng bộ nhớ của Lua script, bao gồm cả những người nhảy qua hoops để ngăn chặn Lua script tạo các hàm và bảng. Nhưng cho rằng lua_newstate cho phép bạn vượt qua một cấp phát tùy chỉnh, không ai có thể sử dụng nó để giới hạn mức tiêu thụ bộ nhớ? Tại tồi tệ nhất, người ta có thể sử dụng một phân bổ dựa trên đấu trường và đặt một giới hạn cứng ngay cả trên số lượng bộ nhớ có thể được sử dụng bởi phân mảnh.

Tôi có thiếu gì đó ở đây không?

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

Để làm Lua sử dụng cấp phát, bạn có thể sử dụng

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

Lưu ý: Tôi đã không kiểm tra nguồn gốc, nhưng nó phải làm việc.