utils: add thisthread, suspend and resume

1 files changed, 39 insertions, 0 deletions

diff --git a/utils.c b/utils.c
index 1477560..4a07079 100644
--- a/utils.c
+++ b/utils.c
@@ -238,6 +238,35 @@ exit_lua(lua_State *T)
 	return 0;
 }
 
+static int
+thisthread(lua_State *T)
+{
+	lua_pushthread(T);
+	return 1;
+}
+
+static int
+suspend(lua_State *T)
+{
+	return lua_yield(T, 0);
+}
+
+static int
+resume(lua_State *T)
+{
+	int args;
+	lua_State *S;
+
+	luaL_checktype(T, 1, LUA_TTHREAD);
+	S = lua_tothread(T, 1);
+
+	args = lua_gettop(T) - 1;
+	lua_xmove(T, S, args);
+	lem_queue(S, args);
+
+	return 0;
+}
+
 int
 luaopen_lem_utils(lua_State *L)
 {
@@ -288,5 +317,15 @@ luaopen_lem_utils(lua_State *L)
 	lua_pushcfunction(L, exit_lua);
 	lua_setfield(L, -2, "exit");
 
+	/* set thisthread function */
+	lua_pushcfunction(L, thisthread);
+	lua_setfield(L, -2, "thisthread");
+	/* set suspend function */
+	lua_pushcfunction(L, suspend);
+	lua_setfield(L, -2, "suspend");
+	/* set resume function */
+	lua_pushcfunction(L, resume);
+	lua_setfield(L, -2, "resume");
+
 	return 1;
 }