nixio linux IO/socket api (incomplete / WIP)

1 files changed, 163 insertions, 0 deletions

diff --git a/libs/nixio/src/socket.c b/libs/nixio/src/socket.c
new file mode 100644
index 0000000000..acfa9aeb22
--- /dev/null
+++ b/libs/nixio/src/socket.c
@@ -0,0 +1,163 @@
+/*
+ * 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 <lua.h>
+#include <lualib.h>
+#include <lauxlib.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <unistd.h>
+#include <string.h>
+#include <errno.h>
+#include "nixio.h"
+
+
+/**
+ * create new socket
+ */
+static int nixio_socket(lua_State *L) {
+	const char *domain = luaL_optlstring(L, 1, "", NULL);
+	const char *type   = luaL_optlstring(L, 2, "", NULL);
+	const char *proto  = lua_tolstring(L, 3, NULL);
+
+	nixio_sock *sock = lua_newuserdata(L, sizeof(nixio_sock));
+
+	if (!strcmp(domain, "inet")) {
+		sock->domain = AF_INET;
+	} else if (!strcmp(domain, "inet6")) {
+		sock->domain = AF_INET6;
+	} else if (!strcmp(domain, "unix")) {
+		sock->domain = AF_UNIX;
+	} else if (!strcmp(domain, "packet")) {
+		sock->domain = AF_PACKET;
+	} else {
+		return luaL_argerror(L, 1,
+		 "supported values: inet, inet6, unix, packet"
+		);
+	}
+
+	if (!strcmp(type, "stream")) {
+		sock->type = SOCK_STREAM;
+	} else if (!strcmp(type, "dgram")) {
+		sock->type = SOCK_DGRAM;
+	} else if (!strcmp(type, "raw")) {
+		sock->type = SOCK_RAW;
+	} else {
+		return luaL_argerror(L, 2, "supported values: stream, dgram, raw");
+	}
+
+	if (!proto) {
+		sock->protocol = 0;
+	} else if (!strcmp(proto, "icmp")) {
+		sock->protocol = IPPROTO_ICMP;
+	} else if (!strcmp(proto, "icmpv6")) {
+		sock->protocol = IPPROTO_ICMPV6;
+	} else {
+		return luaL_argerror(L, 3, "supported values: [empty], icmp, icmpv6");
+	}
+
+	/* create userdata */
+	luaL_getmetatable(L, NIXIO_META);
+	lua_setmetatable(L, -2);
+
+	sock->fd = socket(sock->domain, sock->type, sock->protocol);
+
+	if (sock->fd < 0) {
+		return nixio__perror(L);
+	}
+
+	return 1;
+}
+
+/**
+ * close a socket
+ */
+static int nixio_sock_close(lua_State *L) {
+	nixio_sock *sock = nixio__checksock(L);
+	int sockfd = sock->fd;
+	sock->fd = -1;
+	return nixio__pstatus(L, !close(sockfd));
+}
+
+/**
+ * garbage collector
+ */
+static int nixio_sock__gc(lua_State *L) {
+	nixio_sock *sock = (nixio_sock*)luaL_checkudata(L, 1, NIXIO_META);
+	if (sock && sock->fd != -1) {
+		close(sock->fd);
+	}
+	return 0;
+}
+
+/**
+ * string representation
+ */
+static int nixio_sock__tostring(lua_State *L) {
+	lua_pushfstring(L, "nixio socket %d", nixio__checksockfd(L));
+	return 1;
+}
+
+/**
+ * shutdown a socket
+ */
+static int nixio_sock_shutdown(lua_State *L) {
+	int sockfd = nixio__checksockfd(L);
+	const char *what = luaL_optlstring(L, 2, "rdwr", NULL);
+	int how;
+
+	if (!strcmp(what, "rdwr") || !strcmp(what, "both")) {
+		how = SHUT_RDWR;
+	} else if (!strcmp(what, "rd") || !strcmp(what, "read")) {
+		how = SHUT_RD;
+	} else if (!strcmp(what, "wr") || !strcmp(what, "write")) {
+		how = SHUT_WR;
+	} else {
+		return luaL_argerror(L, 2, "supported values: both, read, write");
+	}
+
+	return nixio__pstatus(L, !shutdown(sockfd, how));
+}
+
+/* module table */
+static const luaL_reg R[] = {
+	{"socket",		nixio_socket},
+	{NULL,			NULL}
+};
+
+/* object table */
+static const luaL_reg M[] = {
+	{"close",		nixio_sock_close},
+	{"shutdown",	nixio_sock_shutdown},
+	{NULL,			NULL}
+};
+
+void nixio_open_socket(lua_State *L) {
+	luaL_getmetatable(L, NIXIO_META);
+	lua_pushcfunction(L, nixio_sock__gc);
+	lua_setfield(L, -2, "__gc");
+	lua_pushcfunction(L, nixio_sock__tostring);
+	lua_setfield(L, -2, "__tostring");
+	lua_pop(L, 1);
+
+	luaL_register(L, NULL, R);
+
+	lua_pushvalue(L, -2);
+	luaL_register(L, NULL, M);
+	lua_pop(L, 1);
+}