1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#include <lua.h>
#include <lauxlib.h>
#include <honeysuckle.h>
#include "util.h"
void create_table(lua_State *L, struct honey_tbl_t *tbl)
{
lua_createtable(L, 0, 0);
int t = lua_gettop(L);
for (struct honey_tbl_t *pair = tbl; pair->key != NULL; pair += 1) {
if (pair->type == LUA_TNUMBER) {
lua_pushinteger(L, pair->value.integer);
lua_setfield(L, t, pair->key);
}
else if (pair->type == LUA_TFUNCTION) {
lua_pushcfunction(L, pair->value.function);
lua_setfield(L, t, pair->key);
}
else {
/* bad type, ignore */
}
}
}
void append_table(lua_State *L, int tbl_a, int tbl_b)
{
lua_pushnil(L);
while(lua_next(L, tbl_b) != 0) {
lua_pushvalue(L, -2); // key
lua_pushvalue(L, -2); // value
lua_settable(L, tbl_a);
lua_pop(L, 1);
}
}
const char *canary_tname = "honey.canary";
int gc_canary(lua_State *L);
int gc_canary_collect(lua_State *L);
void setup_util(lua_State *L, int honey_tbl)
{
/* create gc_canary metatable */
luaL_newmetatable(L, canary_tname);
lua_pushcfunction(L, gc_canary_collect);
lua_setfield(L, -2, "__gc");
lua_pop(L, 1);
hs_create_table(L,
hs_str_cfunc("gc_canary", gc_canary),
);
lua_setfield(L, honey_tbl, "util");
}
int gc_canary(lua_State *L)
{
luaL_checktype(L, 1, LUA_TFUNCTION);
int ref = luaL_ref(L, LUA_REGISTRYINDEX);
int *canary = lua_newuserdata(L, sizeof(int));
*canary = ref;
luaL_getmetatable(L, canary_tname);
lua_setmetatable(L, -2);
return 1;
}
int gc_canary_collect(lua_State *L)
{
int *canary = luaL_checkudata(L, 1, canary_tname);
lua_rawgeti(L, LUA_REGISTRYINDEX, *canary);
lua_call(L, 0, 0);
luaL_unref(L, LUA_REGISTRYINDEX, *canary);
return 0;
}
|
