summaryrefslogtreecommitdiff
path: root/l_cmd.c
diff options
context:
space:
mode:
Diffstat (limited to 'l_cmd.c')
-rw-r--r--l_cmd.c140
1 files changed, 140 insertions, 0 deletions
diff --git a/l_cmd.c b/l_cmd.c
new file mode 100644
index 0000000..baf8db7
--- /dev/null
+++ b/l_cmd.c
@@ -0,0 +1,140 @@
+/*
+ * rhctl
+ *
+ * Copyright (C) 2009 Christian Pointner <equinox@spreadspace.org>
+ *
+ * This file is part of rhctl.
+ *
+ * rhctl is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * any later version.
+ *
+ * rhctl is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with rhctl. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <lua.h>
+#include <lauxlib.h>
+
+#include <sys/socket.h>
+#include <errno.h>
+
+#include "datatypes.h"
+
+#include "l_cmd.h"
+
+#include "options.h"
+#include "command_queue.h"
+#include "client_list.h"
+#include "log.h"
+
+int cmd_fd;
+read_buffer_t cmd_buffer;
+
+static int l_cmd_getfd(lua_State *L)
+{
+ if(!lua_istable(L, -1))
+ luaL_error(L, "can't retreive signal fd");
+
+ lua_pushliteral(L, "fd");
+ lua_gettable(L, -2);
+ return 1;
+}
+
+static int l_cmd_dirty(lua_State *L)
+{
+ lua_pushboolean(L, 0);
+ return 1;
+}
+
+static int l_cmd_init(lua_State *L)
+{
+ lua_newtable(L);
+ lua_pushliteral(L, "fd");
+ lua_pushinteger(L, cmd_fd);
+ lua_settable(L, -3);
+ lua_pushliteral(L, "getfd");
+ lua_pushcfunction(L, l_cmd_getfd);
+ lua_settable(L, -3);
+ lua_pushliteral(L, "dirty");
+ lua_pushcfunction(L, l_cmd_dirty);
+ lua_settable(L, -3);
+ return 1;
+}
+
+// gets called by nonblock_recvline for every complete message
+static int cmd_recv_data(lua_State* L)
+{
+ int ret = 0;
+ for(;;) {
+ ret = recv(cmd_fd, &cmd_buffer.buf[cmd_buffer.offset], 1, 0);
+ if(!ret)
+ return 2;
+ if(ret == -1 && errno == EAGAIN)
+ return 0;
+ else if(ret < 0)
+ break;
+
+ if(cmd_buffer.buf[cmd_buffer.offset] == '\n') {
+ cmd_buffer.buf[cmd_buffer.offset] = 0;
+
+ lua_pushstring(L, cmd_buffer.buf);
+
+ ret = lua_pcall(L, 1, 1, 0);
+ if(ret) {
+ const char* err_str = luaL_checkstring(L, -1);
+ switch(ret) {
+ case LUA_ERRRUN: log_printf(ERROR, "lua_pcall(cmd_callback) runtime error: %s", err_str); break;
+ case LUA_ERRMEM: log_printf(ERROR, "lua_pcall(cmd_callback) malloc error: %s", err_str); break;
+ case LUA_ERRERR: log_printf(ERROR, "lua_pcall(cmd_callback) error at error handler function: %s", err_str); break;
+ }
+ return -1;
+ }
+
+ ret = lua_tointeger(L, 1);
+ cmd_buffer.offset = 0;
+ break;
+ }
+
+ cmd_buffer.offset++;
+ if(cmd_buffer.offset >= sizeof(cmd_buffer.buf)) {
+ log_printf(DEBUG, "string too long cmd_fd");
+ cmd_buffer.offset = 0;
+ return 0;
+ }
+ }
+
+ return ret;
+}
+
+static int l_cmd_recv_data(lua_State* L)
+{
+ int ret = cmd_recv_data(L);
+ lua_pushinteger(L, ret);
+ return 1;
+}
+
+static int l_cmd_send_string(lua_State* L)
+{
+ return 0;
+}
+
+static const struct luaL_reg cmd_funcs [] = {
+ { "init", l_cmd_init },
+ { "recv_data", l_cmd_recv_data },
+ { "send_string", l_cmd_send_string },
+ { NULL, NULL }
+};
+
+
+LUALIB_API int luaopen_cmd(lua_State *L)
+{
+ luaL_register(L, LUA_CMDLIBNAME, cmd_funcs);
+ return 1;
+}