mirror of
https://git.proxmox.com/git/mirror_zfs.git
synced 2024-12-27 03:19:35 +03:00
3eabed74c0
The `zfs program` subcommand invokes a LUA interpreter to run ZFS "channel programs". This interpreter runs in a constrained environment, with defined memory limits. The LUA stack (used for LUA functions that call each other) is allocated in the kernel's heap, and is limited by the `-m MEMORY-LIMIT` flag and the `zfs_lua_max_memlimit` module parameter. The C stack is used by certain LUA features that are implemented in C. The C stack is limited by `LUAI_MAXCCALLS=20`, which limits call depth. Some LUA C calls use more stack space than others, and `gsub()` uses an unusually large amount. With a programming trick, it can be invoked recursively using the C stack (rather than the LUA stack). This overflows the 16KB Linux kernel stack after about 11 iterations, less than the limit of 20. One solution would be to decrease `LUAI_MAXCCALLS`. This could be made to work, but it has a few drawbacks: 1. The existing test suite does not pass with `LUAI_MAXCCALLS=10`. 2. There may be other LUA functions that use a lot of stack space, and the stack space may change depending on compiler version and options. This commit addresses the problem by adding a new limit on the amount of free space (in bytes) remaining on the C stack while running the LUA interpreter: `LUAI_MINCSTACK=4096`. If there is less than this amount of stack space remaining, a LUA runtime error is generated. Reviewed-by: George Wilson <gwilson@delphix.com> Reviewed-by: Brian Behlendorf <behlendorf1@llnl.gov> Reviewed-by: Ryan Moeller <ryan@ixsystems.com> Reviewed-by: Allan Jude <allanjude@freebsd.org> Reviewed-by: Serapheim Dimitropoulos <serapheim@delphix.com> Signed-off-by: Matthew Ahrens <mahrens@delphix.com> Closes #10611 Closes #10613
231 lines
7.5 KiB
C
231 lines
7.5 KiB
C
/* BEGIN CSTYLED */
|
|
/*
|
|
** $Id: lstate.h,v 2.82.1.1 2013/04/12 18:48:47 roberto Exp $
|
|
** Global State
|
|
** See Copyright Notice in lua.h
|
|
*/
|
|
|
|
#ifndef lstate_h
|
|
#define lstate_h
|
|
|
|
#include <sys/lua/lua.h>
|
|
|
|
#include "lobject.h"
|
|
#include "ltm.h"
|
|
#include "lzio.h"
|
|
|
|
|
|
/*
|
|
|
|
** Some notes about garbage-collected objects: All objects in Lua must
|
|
** be kept somehow accessible until being freed.
|
|
**
|
|
** Lua keeps most objects linked in list g->allgc. The link uses field
|
|
** 'next' of the CommonHeader.
|
|
**
|
|
** Strings are kept in several lists headed by the array g->strt.hash.
|
|
**
|
|
** Open upvalues are not subject to independent garbage collection. They
|
|
** are collected together with their respective threads. Lua keeps a
|
|
** double-linked list with all open upvalues (g->uvhead) so that it can
|
|
** mark objects referred by them. (They are always gray, so they must
|
|
** be remarked in the atomic step. Usually their contents would be marked
|
|
** when traversing the respective threads, but the thread may already be
|
|
** dead, while the upvalue is still accessible through closures.)
|
|
**
|
|
** Objects with finalizers are kept in the list g->finobj.
|
|
**
|
|
** The list g->tobefnz links all objects being finalized.
|
|
|
|
*/
|
|
|
|
|
|
struct lua_longjmp; /* defined in ldo.c */
|
|
|
|
|
|
|
|
/* extra stack space to handle TM calls and some other extras */
|
|
#define EXTRA_STACK 5
|
|
|
|
|
|
#define BASIC_STACK_SIZE (2*LUA_MINSTACK)
|
|
|
|
|
|
/* kinds of Garbage Collection */
|
|
#define KGC_NORMAL 0
|
|
#define KGC_EMERGENCY 1 /* gc was forced by an allocation failure */
|
|
#define KGC_GEN 2 /* generational collection */
|
|
|
|
|
|
typedef struct stringtable {
|
|
GCObject **hash;
|
|
lu_int32 nuse; /* number of elements */
|
|
int size;
|
|
} stringtable;
|
|
|
|
|
|
/*
|
|
** information about a call
|
|
*/
|
|
typedef struct CallInfo {
|
|
StkId func; /* function index in the stack */
|
|
StkId top; /* top for this function */
|
|
struct CallInfo *previous, *next; /* dynamic call link */
|
|
short nresults; /* expected number of results from this function */
|
|
lu_byte callstatus;
|
|
ptrdiff_t extra;
|
|
union {
|
|
struct { /* only for Lua functions */
|
|
StkId base; /* base for this function */
|
|
const Instruction *savedpc;
|
|
} l;
|
|
struct { /* only for C functions */
|
|
int ctx; /* context info. in case of yields */
|
|
lua_CFunction k; /* continuation in case of yields */
|
|
ptrdiff_t old_errfunc;
|
|
lu_byte old_allowhook;
|
|
lu_byte status;
|
|
} c;
|
|
} u;
|
|
} CallInfo;
|
|
|
|
|
|
/*
|
|
** Bits in CallInfo status
|
|
*/
|
|
#define CIST_LUA (1<<0) /* call is running a Lua function */
|
|
#define CIST_HOOKED (1<<1) /* call is running a debug hook */
|
|
#define CIST_REENTRY (1<<2) /* call is running on same invocation of
|
|
luaV_execute of previous call */
|
|
#define CIST_YIELDED (1<<3) /* call reentered after suspension */
|
|
#define CIST_YPCALL (1<<4) /* call is a yieldable protected call */
|
|
#define CIST_STAT (1<<5) /* call has an error status (pcall) */
|
|
#define CIST_TAIL (1<<6) /* call was tail called */
|
|
#define CIST_HOOKYIELD (1<<7) /* last hook called yielded */
|
|
|
|
|
|
#define isLua(ci) ((ci)->callstatus & CIST_LUA)
|
|
|
|
|
|
/*
|
|
** `global state', shared by all threads of this state
|
|
*/
|
|
typedef struct global_State {
|
|
lua_Alloc frealloc; /* function to reallocate memory */
|
|
void *ud; /* auxiliary data to `frealloc' */
|
|
lu_mem totalbytes; /* number of bytes currently allocated - GCdebt */
|
|
l_mem GCdebt; /* bytes allocated not yet compensated by the collector */
|
|
lu_mem GCmemtrav; /* memory traversed by the GC */
|
|
lu_mem GCestimate; /* an estimate of the non-garbage memory in use */
|
|
stringtable strt; /* hash table for strings */
|
|
TValue l_registry;
|
|
unsigned int seed; /* randomized seed for hashes */
|
|
lu_byte currentwhite;
|
|
lu_byte gcstate; /* state of garbage collector */
|
|
lu_byte gckind; /* kind of GC running */
|
|
lu_byte gcrunning; /* true if GC is running */
|
|
int sweepstrgc; /* position of sweep in `strt' */
|
|
GCObject *allgc; /* list of all collectable objects */
|
|
GCObject *finobj; /* list of collectable objects with finalizers */
|
|
GCObject **sweepgc; /* current position of sweep in list 'allgc' */
|
|
GCObject **sweepfin; /* current position of sweep in list 'finobj' */
|
|
GCObject *gray; /* list of gray objects */
|
|
GCObject *grayagain; /* list of objects to be traversed atomically */
|
|
GCObject *weak; /* list of tables with weak values */
|
|
GCObject *ephemeron; /* list of ephemeron tables (weak keys) */
|
|
GCObject *allweak; /* list of all-weak tables */
|
|
GCObject *tobefnz; /* list of userdata to be GC */
|
|
UpVal uvhead; /* head of double-linked list of all open upvalues */
|
|
Mbuffer buff; /* temporary buffer for string concatenation */
|
|
int gcpause; /* size of pause between successive GCs */
|
|
int gcmajorinc; /* pause between major collections (only in gen. mode) */
|
|
int gcstepmul; /* GC `granularity' */
|
|
lua_CFunction panic; /* to be called in unprotected errors */
|
|
struct lua_State *mainthread;
|
|
const lua_Number *version; /* pointer to version number */
|
|
TString *memerrmsg; /* memory-error message */
|
|
TString *tmname[TM_N]; /* array with tag-method names */
|
|
struct Table *mt[LUA_NUMTAGS]; /* metatables for basic types */
|
|
} global_State;
|
|
|
|
|
|
/*
|
|
** `per thread' state
|
|
*/
|
|
struct lua_State {
|
|
CommonHeader;
|
|
lu_byte status;
|
|
StkId top; /* first free slot in the stack */
|
|
global_State *l_G;
|
|
CallInfo *ci; /* call info for current function */
|
|
const Instruction *oldpc; /* last pc traced */
|
|
StkId stack_last; /* last free slot in the stack */
|
|
StkId stack; /* stack base */
|
|
int stacksize;
|
|
unsigned short nny; /* number of non-yieldable calls in stack */
|
|
unsigned short nCcalls; /* number of nested C calls */
|
|
lu_byte hookmask;
|
|
lu_byte allowhook;
|
|
lu_byte runerror; /* handling a runtime error */
|
|
int basehookcount;
|
|
int hookcount;
|
|
lua_Hook hook;
|
|
GCObject *openupval; /* list of open upvalues in this stack */
|
|
GCObject *gclist;
|
|
struct lua_longjmp *errorJmp; /* current error recover point */
|
|
ptrdiff_t errfunc; /* current error handling function (stack index) */
|
|
CallInfo base_ci; /* CallInfo for first level (C calling Lua) */
|
|
};
|
|
|
|
|
|
#define G(L) (L->l_G)
|
|
|
|
|
|
/*
|
|
** Union of all collectable objects
|
|
*/
|
|
union GCObject {
|
|
GCheader gch; /* common header */
|
|
union TString ts;
|
|
union Udata u;
|
|
union Closure cl;
|
|
struct Table h;
|
|
struct Proto p;
|
|
struct UpVal uv;
|
|
struct lua_State th; /* thread */
|
|
};
|
|
|
|
|
|
#define gch(o) (&(o)->gch)
|
|
|
|
/* macros to convert a GCObject into a specific value */
|
|
#define rawgco2ts(o) \
|
|
check_exp(novariant((o)->gch.tt) == LUA_TSTRING, &((o)->ts))
|
|
#define gco2ts(o) (&rawgco2ts(o)->tsv)
|
|
#define rawgco2u(o) check_exp((o)->gch.tt == LUA_TUSERDATA, &((o)->u))
|
|
#define gco2u(o) (&rawgco2u(o)->uv)
|
|
#define gco2lcl(o) check_exp((o)->gch.tt == LUA_TLCL, &((o)->cl.l))
|
|
#define gco2ccl(o) check_exp((o)->gch.tt == LUA_TCCL, &((o)->cl.c))
|
|
#define gco2cl(o) \
|
|
check_exp(novariant((o)->gch.tt) == LUA_TFUNCTION, &((o)->cl))
|
|
#define gco2t(o) check_exp((o)->gch.tt == LUA_TTABLE, &((o)->h))
|
|
#define gco2p(o) check_exp((o)->gch.tt == LUA_TPROTO, &((o)->p))
|
|
#define gco2uv(o) check_exp((o)->gch.tt == LUA_TUPVAL, &((o)->uv))
|
|
#define gco2th(o) check_exp((o)->gch.tt == LUA_TTHREAD, &((o)->th))
|
|
|
|
/* macro to convert any Lua object into a GCObject */
|
|
#define obj2gco(v) (cast(GCObject *, (v)))
|
|
|
|
|
|
/* actual number of total bytes allocated */
|
|
#define gettotalbytes(g) ((g)->totalbytes + (g)->GCdebt)
|
|
|
|
LUAI_FUNC void luaE_setdebt (global_State *g, l_mem debt);
|
|
LUAI_FUNC void luaE_freethread (lua_State *L, lua_State *L1);
|
|
LUAI_FUNC CallInfo *luaE_extendCI (lua_State *L);
|
|
LUAI_FUNC void luaE_freeCI (lua_State *L);
|
|
|
|
|
|
#endif
|
|
/* END CSTYLED */
|