X-Git-Url: http://git.archive.openwrt.org/?a=blobdiff_plain;f=libs%2Fluci-lib-nixio%2Fsrc%2Fpoll.c;fp=libs%2Fluci-lib-nixio%2Fsrc%2Fpoll.c;h=1211bc72ac392471a70586d10a2c4279773f6f07;hb=1bb4822dca6113f73e3bc89e2acf15935e6f8e92;hp=0000000000000000000000000000000000000000;hpb=9edd0e46c3f880727738ce8ca6ff1c8b85f99ef4;p=project%2Fluci.git

diff --git a/libs/luci-lib-nixio/src/poll.c b/libs/luci-lib-nixio/src/poll.c
new file mode 100644
index 000000000..1211bc72a
--- /dev/null
+++ b/libs/luci-lib-nixio/src/poll.c
@@ -0,0 +1,210 @@
+/*
+ * nixio - Linux I/O library for lua
+ *
+ *   Copyright (C) 2009 Steven Barth <steven@midlink.org>
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+
+#include "nixio.h"
+#include <time.h>
+#include <errno.h>
+#include <string.h>
+#include <stdlib.h>
+#include <sys/time.h>
+
+
+static int nixio_gettimeofday(lua_State *L) {
+	struct timeval tv;
+	gettimeofday(&tv, NULL);
+	nixio__pushnumber(L, tv.tv_sec);
+	nixio__pushnumber(L, tv.tv_usec);
+	return 2;
+}
+
+
+/**
+ * nanosleep()
+ */
+static int nixio_nanosleep(lua_State *L) {
+	struct timespec req, rem;
+	req.tv_sec = luaL_optint(L, 1, 0);
+	req.tv_nsec = luaL_optint(L, 2, 0);
+
+	int status = nanosleep(&req, &rem);
+	if (!status) {
+		lua_pushboolean(L, 1);
+		return 1;
+	} else {
+		if (errno == EINTR) {
+			lua_pushboolean(L, 0);
+			lua_pushinteger(L, rem.tv_sec);
+			lua_pushinteger(L, rem.tv_nsec);
+			return 3;
+		} else {
+			return nixio__perror(L);
+		}
+	}
+}
+
+/**
+ * Checks whether a flag is set in the bitmap and sets the matching table value
+ */
+static void nixio_poll_flags__r(lua_State *L, int *map, int f, const char *t) {
+	lua_pushstring(L, t);
+	if (*map & f) {
+		lua_pushboolean(L, 1);
+	} else {
+		lua_pushnil(L);
+	}
+	lua_rawset(L, -3);
+}
+
+/**
+ * Translate integer to poll flags and vice versa
+ */
+static int nixio_poll_flags(lua_State *L) {
+	int flags;
+	if (lua_isnumber(L, 1)) {
+		flags = luaL_checkinteger(L, 1);
+		lua_newtable(L);
+		nixio_poll_flags__r(L, &flags, POLLIN, "in");
+		nixio_poll_flags__r(L, &flags, POLLOUT, "out");
+		nixio_poll_flags__r(L, &flags, POLLERR, "err");
+#ifndef __WINNT__
+		nixio_poll_flags__r(L, &flags, POLLPRI, "pri");
+		nixio_poll_flags__r(L, &flags, POLLHUP, "hup");
+		nixio_poll_flags__r(L, &flags, POLLNVAL, "nval");
+#endif
+	 } else {
+		flags = 0;
+		const int j = lua_gettop(L);
+		for (int i=1; i<=j; i++) {
+			const char *flag = luaL_checkstring(L, i);
+			if (!strcmp(flag, "in")) {
+				flags |= POLLIN;
+			} else if (!strcmp(flag, "out")) {
+				flags |= POLLOUT;
+			} else if (!strcmp(flag, "err")) {
+				flags |= POLLERR;
+			} else if (!strcmp(flag, "pri")) {
+#ifndef __WINNT__
+				flags |= POLLPRI;
+#endif
+			} else if (!strcmp(flag, "hup")) {
+#ifndef __WINNT__
+				flags |= POLLHUP;
+#endif
+			} else if (!strcmp(flag, "nval")) {
+#ifndef __WINNT__
+				flags |= POLLNVAL;
+#endif
+			} else {
+				return luaL_argerror(L, i,
+				 "supported values: in, pri, out, err, hup, nval");
+			}
+		}
+		lua_pushinteger(L, flags);
+	}
+	return 1;
+}
+
+/**
+ * poll({{fd = socket, events = FLAGS}, ...}, timeout)
+ */
+static int nixio_poll(lua_State *L) {
+	int len = lua_objlen(L, 1);
+	int i, fd;
+	int timeout = luaL_optint(L, 2, 0);
+	int status = -1;
+
+	/* we are being abused as sleep() replacement... */
+	if (lua_isnoneornil(L, 1) || len < 1) {
+		if (!poll(NULL, 0, timeout)) {
+			lua_pushinteger(L, 0);
+			return 1;
+		} else {
+			return nixio__perror(L);
+		}
+	}
+
+	luaL_checktype(L, 1, LUA_TTABLE);
+	struct pollfd *fds = calloc(len, sizeof(struct pollfd));
+	if (!fds) {
+		return luaL_error(L, NIXIO_OOM);
+	}
+
+	for (i = 0; i < len; i++) {
+		lua_rawgeti(L, 1, i+1);
+		if (!lua_istable(L, -1)) {
+			free(fds);
+			return luaL_argerror(L, 1, "invalid datastructure");
+		}
+
+		lua_pushliteral(L, "fd");
+		lua_rawget(L, -2);
+		fd = nixio__tofd(L, -1);
+		if (fd == -1) {
+			free(fds);
+			return luaL_argerror(L, 1, "invalid fd in datastructure");
+		}
+		fds[i].fd = fd;
+
+		lua_pushliteral(L, "events");
+		lua_rawget(L, -3);
+		fds[i].events = (short)lua_tointeger(L, -1);
+
+		lua_pop(L, 3);
+	}
+
+	status = poll(fds, (nfds_t)len, timeout);
+
+	if (status == 0) {
+		free(fds);
+		lua_pushboolean(L, 0);
+		return 1;
+	} else if (status < 0) {
+		free(fds);
+		return nixio__perror(L);
+	}
+
+	for (i = 0; i < len; i++) {
+		lua_rawgeti(L, 1, i+1);
+
+		lua_pushliteral(L, "revents");
+		lua_pushinteger(L, fds[i].revents);
+		lua_rawset(L, -3);
+
+		lua_pop(L, 1);
+	}
+
+	free(fds);
+
+	lua_pushinteger(L, status);
+	lua_pushvalue(L, 1);
+
+	return 2;
+}
+
+/* module table */
+static const luaL_reg R[] = {
+	{"gettimeofday", nixio_gettimeofday},
+	{"nanosleep",	nixio_nanosleep},
+	{"poll",		nixio_poll},
+	{"poll_flags",	nixio_poll_flags},
+	{NULL,			NULL}
+};
+
+void nixio_open_poll(lua_State *L) {
+	luaL_register(L, NULL, R);
+}