diff options
Diffstat (limited to 'src/lj_vmevent.c')
-rw-r--r-- | src/lj_vmevent.c | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/src/lj_vmevent.c b/src/lj_vmevent.c new file mode 100644 index 00000000..657eb8d7 --- /dev/null +++ b/src/lj_vmevent.c | |||
@@ -0,0 +1,56 @@ | |||
1 | /* | ||
2 | ** VM event handling. | ||
3 | ** Copyright (C) 2005-2009 Mike Pall. See Copyright Notice in luajit.h | ||
4 | */ | ||
5 | |||
6 | #include <stdio.h> | ||
7 | |||
8 | #define lj_vmevent_c | ||
9 | #define LUA_CORE | ||
10 | |||
11 | #include "lj_obj.h" | ||
12 | #include "lj_str.h" | ||
13 | #include "lj_tab.h" | ||
14 | #include "lj_state.h" | ||
15 | #include "lj_dispatch.h" | ||
16 | #include "lj_vm.h" | ||
17 | #include "lj_vmevent.h" | ||
18 | |||
19 | ptrdiff_t lj_vmevent_prepare(lua_State *L, VMEvent ev) | ||
20 | { | ||
21 | global_State *g = G(L); | ||
22 | GCstr *s = lj_str_newlit(L, LJ_VMEVENTS_REGKEY); | ||
23 | cTValue *tv = lj_tab_getstr(tabV(registry(L)), s); | ||
24 | if (tvistab(tv)) { | ||
25 | int hash = VMEVENT_HASH(ev); | ||
26 | tv = lj_tab_getint(tabV(tv), hash); | ||
27 | if (tv && tvisfunc(tv)) { | ||
28 | lj_state_checkstack(L, LUA_MINSTACK); | ||
29 | setfuncV(L, L->top++, funcV(tv)); | ||
30 | return savestack(L, L->top); | ||
31 | } | ||
32 | } | ||
33 | g->vmevmask &= ~VMEVENT_MASK(ev); /* No handler: cache this fact. */ | ||
34 | return 0; | ||
35 | } | ||
36 | |||
37 | void lj_vmevent_call(lua_State *L, ptrdiff_t argbase) | ||
38 | { | ||
39 | global_State *g = G(L); | ||
40 | uint8_t oldmask = g->vmevmask; | ||
41 | uint8_t oldh = hook_save(g); | ||
42 | int status; | ||
43 | g->vmevmask = 0; /* Disable all events. */ | ||
44 | hook_vmevent(g); | ||
45 | status = lj_vm_pcall(L, restorestack(L, argbase), 0+1, 0); | ||
46 | if (LJ_UNLIKELY(status)) { | ||
47 | /* Really shouldn't use stderr here, but where else to complain? */ | ||
48 | L->top--; | ||
49 | fprintf(stderr, "VM handler failed: %s\n", | ||
50 | tvisstr(L->top) ? strVdata(L->top) : "?"); | ||
51 | } | ||
52 | hook_restore(g, oldh); | ||
53 | if (g->vmevmask != VMEVENT_NOCACHE) | ||
54 | g->vmevmask = oldmask; /* Restore event mask, but not if not modified. */ | ||
55 | } | ||
56 | |||