Initial revision

2 files changed, 394 insertions, 0 deletions

diff --git a/src/unix.c b/src/unix.c
new file mode 100644
index 0000000..d50d98c
--- /dev/null
+++ b/src/unix.c
@@ -0,0 +1,325 @@
+/*=========================================================================*\
+* Network compatibilization module
+\*=========================================================================*/
+#include <string.h>
+
+#include <lua.h>
+#include <lauxlib.h>
+
+#include "lscompat.h"
+
+/*=========================================================================*\
+* Internal function prototypes
+\*=========================================================================*/
+static cchar *try_setoption(lua_State *L, COMPAT_FD sock);
+static cchar *try_setbooloption(lua_State *L, COMPAT_FD sock, int name);
+
+/*=========================================================================*\
+* Exported functions.
+\*=========================================================================*/
+void compat_open(lua_State *L)
+{
+    /* Instals a handler to ignore sigpipe. This function is not 
+       needed on the WinSock2, since it's sockets don't raise signals. */
+    struct sigaction new;
+    memset(&new, 0, sizeof(new));
+    new.sa_handler = SIG_IGN;
+    sigaction(SIGPIPE, &new, NULL);
+}
+
+COMPAT_FD compat_accept(COMPAT_FD s, struct  sockaddr  *addr,  
+        socklen_t *len, int deadline)
+{
+    struct timeval tv;
+    fd_set fds;
+    tv.tv_sec = deadline / 1000;
+    tv.tv_usec = (deadline % 1000) * 1000;
+    FD_ZERO(&fds);
+    FD_SET(s, &fds);
+    select(s+1, &fds, NULL, NULL, deadline >= 0 ? &tv : NULL);
+    return accept(s, addr, len);
+}
+
+int compat_send(COMPAT_FD c, cchar *data, size_t count, size_t *sent, 
+        int deadline)
+{
+    struct timeval tv;
+    fd_set fds;
+    ssize_t put = 0;
+    int err;
+    int ret;
+    tv.tv_sec = deadline / 1000;
+    tv.tv_usec = (deadline % 1000) * 1000;
+    FD_ZERO(&fds);
+    FD_SET(c, &fds);
+    ret = select(c+1, NULL, &fds, NULL, deadline >= 0 ? &tv : NULL);
+    if (ret > 0) {
+       put = write(c, data, count);  
+       if (put <= 0) {
+           err = PRIV_CLOSED;
+#ifdef __CYGWIN__
+           /* this is for CYGWIN, which is like Unix but has Win32 bugs */
+           if (sent < 0 && errno == EWOULDBLOCK) err = PRIV_DONE;
+#endif
+           *sent = 0;
+       } else {
+           *sent = put;
+           err = PRIV_DONE;
+       }
+       return err;
+    } else {
+        *sent = 0;
+        return PRIV_TIMEOUT;
+    }
+}
+
+int compat_sendto(COMPAT_FD c, cchar *data, size_t count, size_t *sent, 
+        int deadline, SA *addr, socklen_t len)
+{
+    struct timeval tv;
+    fd_set fds;
+    ssize_t put = 0;
+    int err;
+    int ret;
+    tv.tv_sec = deadline / 1000;
+    tv.tv_usec = (deadline % 1000) * 1000;
+    FD_ZERO(&fds);
+    FD_SET(c, &fds);
+    ret = select(c+1, NULL, &fds, NULL, deadline >= 0 ? &tv : NULL);
+    if (ret > 0) {
+       put = sendto(c, data, count, 0, addr, len);  
+       if (put <= 0) {
+           err = PRIV_CLOSED;
+#ifdef __CYGWIN__
+           /* this is for CYGWIN, which is like Unix but has Win32 bugs */
+           if (sent < 0 && errno == EWOULDBLOCK) err = PRIV_DONE;
+#endif
+           *sent = 0;
+       } else {
+           *sent = put;
+           err = PRIV_DONE;
+       }
+       return err;
+    } else {
+        *sent = 0;
+        return PRIV_TIMEOUT;
+    }
+}
+
+int compat_recv(COMPAT_FD c, uchar *data, size_t count, size_t *got, 
+        int deadline)
+{
+    struct timeval tv;
+    fd_set fds;
+    int ret;
+    ssize_t taken = 0;
+    tv.tv_sec = deadline / 1000;
+    tv.tv_usec = (deadline % 1000) * 1000;
+    FD_ZERO(&fds);
+    FD_SET(c, &fds);
+    ret = select(c+1, &fds, NULL, NULL, deadline >= 0 ? &tv : NULL);
+    if (ret > 0) {
+       taken = read(c, data, count);  
+       if (taken <= 0) {
+           *got = 0;
+           return PRIV_CLOSED;
+       } else {
+           *got = taken;
+           return PRIV_DONE;
+       }
+    } else {
+        *got = 0;
+        return PRIV_TIMEOUT;
+    }
+}
+
+int compat_recvfrom(COMPAT_FD c, uchar *data, size_t count, size_t *got, 
+        int deadline, SA *addr, socklen_t *len)
+{
+    struct timeval tv;
+    fd_set fds;
+    int ret;
+    ssize_t taken = 0;
+    tv.tv_sec = deadline / 1000;
+    tv.tv_usec = (deadline % 1000) * 1000;
+    FD_ZERO(&fds);
+    FD_SET(c, &fds);
+    ret = select(c+1, &fds, NULL, NULL, deadline >= 0 ? &tv : NULL);
+    if (ret > 0) {
+       taken = recvfrom(c, data, count, 0, addr, len);  
+       if (taken <= 0) {
+           *got = 0;
+           return PRIV_CLOSED;
+       } else {
+           *got = taken;
+           return PRIV_DONE;
+       }
+    } else {
+        *got = 0;
+        return PRIV_TIMEOUT;
+    }
+}
+
+/*-------------------------------------------------------------------------*\
+* Returns a string describing the last host manipulation error.
+\*-------------------------------------------------------------------------*/
+const char *compat_hoststrerror(void)
+{
+    switch (h_errno) {
+        case HOST_NOT_FOUND: return "host not found";
+        case NO_ADDRESS: return "unable to resolve host name";
+        case NO_RECOVERY: return "name server error";
+        case TRY_AGAIN: return "name server unavailable, try again later";
+        default: return "unknown error";
+    }
+}
+
+/*-------------------------------------------------------------------------*\
+* Returns a string describing the last socket manipulation error.
+\*-------------------------------------------------------------------------*/
+const char *compat_socketstrerror(void)
+{
+    switch (errno) {
+        case EACCES: return "access denied";
+        case EMFILE: return "descriptor table is full";
+        case ENFILE: return "too many open files";
+        case ENOBUFS: return "insuffucient buffer space";
+        default: return "unknown error";
+    }
+}
+
+/*-------------------------------------------------------------------------*\
+* Returns a string describing the last bind command error.
+\*-------------------------------------------------------------------------*/
+const char *compat_bindstrerror(void)
+{
+    switch (errno) {
+        case EBADF: return "invalid descriptor";
+        case EINVAL: return "socket already bound";
+        case EACCES: return "access denied";
+        case ENOTSOCK: return "not a socket descriptor";
+        case EADDRINUSE: return "address already in use";
+        case EADDRNOTAVAIL: return "address unavailable";
+        case ENOMEM: return "out of memory";
+        default: return "unknown error";
+    }
+}
+
+/*-------------------------------------------------------------------------*\
+* Returns a string describing the last connect error.
+\*-------------------------------------------------------------------------*/
+const char *compat_connectstrerror(void)
+{
+    switch (errno) {
+        case EBADF: return "invalid descriptor";
+        case ENOTSOCK: return "not a socket descriptor";
+        case EADDRNOTAVAIL: return "address not availabe";
+        case ETIMEDOUT: return "connection timed out";
+        case ECONNREFUSED: return "connection refused";
+        case EACCES: return "access denied";
+        case ENETUNREACH: return "network is unreachable";
+        case EADDRINUSE: return "address already in use";
+        default: return "unknown error";
+    }
+}
+
+/*-------------------------------------------------------------------------*\
+* Sets the SO_REUSEADDR socket option
+* Input
+*   sock: socket descriptor
+\*-------------------------------------------------------------------------*/
+void compat_setreuseaddr(COMPAT_FD sock)
+{
+    int val = 1;
+    setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char *)&val, sizeof(val));
+}
+
+COMPAT_FD compat_socket(int domain, int type, int protocol)
+{
+    COMPAT_FD sock = socket(domain, type, protocol);
+    if (sock != COMPAT_INVALIDFD) {
+        compat_setnonblocking(sock);
+        compat_setreuseaddr(sock);
+    }
+    return sock;
+}
+
+/*-------------------------------------------------------------------------*\
+* Put socket into blocking mode.
+\*-------------------------------------------------------------------------*/
+void compat_setblocking(COMPAT_FD sock)
+{
+    int flags = fcntl(sock, F_GETFL, 0);
+    flags &= (~(O_NONBLOCK));
+    fcntl(sock, F_SETFL, flags);
+}
+
+/*-------------------------------------------------------------------------*\
+* Put socket into non-blocking mode.
+\*-------------------------------------------------------------------------*/
+void compat_setnonblocking(COMPAT_FD sock)
+{
+    int flags = fcntl(sock, F_GETFL, 0);
+    flags |= O_NONBLOCK;
+    fcntl(sock, F_SETFL, flags);
+}
+
+/*-------------------------------------------------------------------------*\
+* Tries to set extended udp socket options
+* Input
+*   udp: udp structure
+*   oldtop: top of stack
+* Returns
+*   NULL if successfull, error message on error
+\*-------------------------------------------------------------------------*/
+cchar *compat_trysetoptions(lua_State *L, COMPAT_FD sock)
+{
+    if (!lua_istable(L, 1)) luaL_argerror(L, 1, "invalid options table");
+    lua_pushnil(L);
+    while (lua_next(L, 1)) {
+        cchar *err = try_setoption(L, sock);
+        lua_pop(L, 1);
+        if (err) return err;
+    }
+    return NULL;
+}
+
+/*=========================================================================*\
+* Internal functions.
+\*=========================================================================*/
+static cchar *try_setbooloption(lua_State *L, COMPAT_FD sock, int name)
+{
+    int bool, res;
+    if (!lua_isnumber(L, -1)) lua_error(L, "invalid option value");
+    bool = (int) lua_tonumber(L, -1);
+    res = setsockopt(sock, SOL_SOCKET, name, (char *) &bool, sizeof(bool));
+    if (res < 0) return "error setting option";
+    else return NULL;
+}
+
+
+/*-------------------------------------------------------------------------*\
+* Set socket options from a table on top of Lua stack.
+* Supports SO_KEEPALIVE, SO_DONTROUTE, SO_BROADCAST, and SO_LINGER options.
+* Input
+*   L: Lua state to use
+*   sock: socket descriptor
+* Returns
+*   1 if successful, 0 otherwise
+\*-------------------------------------------------------------------------*/
+static cchar *try_setoption(lua_State *L, COMPAT_FD sock)
+{
+    static cchar *options[] = {
+        "SO_KEEPALIVE", "SO_DONTROUTE", "SO_BROADCAST", "SO_LINGER", NULL
+    };
+    cchar *option = lua_tostring(L, -2);
+    if (!lua_isstring(L, -2)) return "invalid option";
+    switch (luaL_findstring(option, options)) {
+        case 0: return try_setbooloption(L, sock, SO_KEEPALIVE);
+        case 1: return try_setbooloption(L, sock, SO_DONTROUTE);
+        case 2: return try_setbooloption(L, sock, SO_BROADCAST);
+        case 3: return "SO_LINGER is deprecated";
+        default: return "unsupported option";
+    }
+}
+
diff --git a/src/unix.h b/src/unix.h
new file mode 100644
index 0000000..e317b06
--- /dev/null
+++ b/src/unix.h
@@ -0,0 +1,69 @@
+#ifndef COMPAT_H_
+#define COMPAT_H_
+
+#include "lspriv.h"
+
+/*=========================================================================*\
+* BSD include files
+\*=========================================================================*/
+/* error codes */
+#include <errno.h>
+/* close function */
+#include <unistd.h>
+/* fnctnl function and associated constants */
+#include <fcntl.h>
+/* struct timeval and CLK_TCK */
+#include <sys/time.h>
+/* times function and struct tms */
+#include <sys/times.h>
+/* struct sockaddr */
+#include <sys/types.h>
+/* socket function */
+#include <sys/socket.h>
+/* gethostbyname and gethostbyaddr functions */
+#include <netdb.h>
+/* sigpipe handling */
+#include <signal.h>
+
+#include <netinet/in.h>
+#include <arpa/inet.h>
+
+#define COMPAT_FD int
+#define COMPAT_INVALIDFD (-1)
+
+/* we are lazy... */
+typedef struct sockaddr SA;
+
+/*=========================================================================*\
+* Exported functions
+\*=========================================================================*/
+void compat_open(lua_State *L);
+
+#define compat_bind bind
+#define compat_connect connect
+#define compat_listen listen
+#define compat_close close
+#define compat_select select
+
+COMPAT_FD compat_socket(int domain, int type, int protocol);
+COMPAT_FD compat_accept(COMPAT_FD s, SA *addr, socklen_t *len, int deadline);
+int compat_send(COMPAT_FD c, cchar *data, size_t count, size_t *done, 
+        int deadline);
+int compat_recv(COMPAT_FD c, uchar *data, size_t count, size_t *done, 
+        int deadline);
+int compat_sendto(COMPAT_FD c, cchar *data, size_t count, size_t *done,
+        int deadline, SA *addr, socklen_t len);
+int compat_recvfrom(COMPAT_FD c, uchar *data, size_t count, size_t *got, 
+        int deadline, SA *addr, socklen_t *len);
+void compat_setnonblocking(COMPAT_FD sock);
+void compat_setblocking(COMPAT_FD sock);
+void compat_setreuseaddr(COMPAT_FD sock);
+
+const char *compat_hoststrerror(void);
+const char *compat_socketstrerror(void);
+const char *compat_bindstrerror(void);
+const char *compat_connectstrerror(void);
+
+cchar *compat_trysetoptions(lua_State *L, COMPAT_FD sock);
+
+#endif /* COMPAT_H_ */