mirror of
https://github.com/lunarmodules/luasocket.git
synced 2024-12-26 12:28:21 +01:00
Initial revision
This commit is contained in:
Diego Nehab
parent
46828c1a7d
commit
cbffc42f29
325
src/unix.c
Normal file
325
src/unix.c
Normal file
@ -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";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
69
src/unix.h
Normal file
69
src/unix.h
Normal file
@ -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_ */
|
||||||
Loading…
Reference in New Issue
Block a user