dany/luasocket
1
0
Fork 0
mirror of https://github.com/lunarmodules/luasocket.git synced 2025-02-13 15:32:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

Almost ready for Lua 4.1

Browse Source
This commit is contained in:
Diego Nehab 2001-09-26 20:31:59 +00:00
parent 6eb7f22c4b
commit 6f02d1444b
1 changed files with 87 additions and 97 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

184
src/luasocket.c
View File

@ -291,13 +291,7 @@ static int global_tcpconnect(lua_State *L)
const char *address = luaL_check_string(L, 1); const char *address = luaL_check_string(L, 1);
unsigned short port = (unsigned short) luaL_check_number(L, 2); unsigned short port = (unsigned short) luaL_check_number(L, 2);
p_sock sock = push_clienttable(L, tags); p_sock sock = push_clienttable(L, tags);
const char *err; const char *err = tcp_tryconnect(sock, address, port);
if (!sock) {
lua_pushnil(L);
lua_pushstring(L, "out of memory");
return 2;
}
err = tcp_tryconnect(sock, address, port);
if (err) { if (err) {
lua_pushnil(L); lua_pushnil(L);
lua_pushstring(L, err); lua_pushstring(L, err);
@ -316,18 +310,18 @@ static int global_tcpconnect(lua_State *L)
static int global_udpsocket(lua_State *L) static int global_udpsocket(lua_State *L)
{ {
p_tags tags = pop_tags(L); p_tags tags = pop_tags(L);
int top = lua_gettop(L); int top = lua_gettop(L);
p_sock sock = push_udptable(L, tags); p_sock sock = push_udptable(L, tags);
if (!sock) return 2; if (!sock) return 2;
if (top >= 1 ) { if (top >= 1 ) {
if (lua_istable(L, 1)) { if (lua_istable(L, 1)) {
lua_pushnil(L); lua_pushnil(L);
while (lua_next(L, 1)) { while (lua_next(L, 1)) {
if (!set_option(L, sock)) lua_error(L, "error setting option"); if (!set_option(L, sock)) lua_error(L, "error setting option");
lua_pop(L, 1); lua_pop(L, 1);
} }
} else luaL_argerror(L, 1, "invalid options"); } else luaL_argerror(L, 1, "invalid options");
} }
return 1; return 1;
} }
@ -1072,63 +1066,63 @@ void set_reuseaddr(p_sock sock)
\*-------------------------------------------------------------------------*/ \*-------------------------------------------------------------------------*/
static int set_option(lua_State *L, p_sock sock) static int set_option(lua_State *L, p_sock sock)
{ {
static const char *const optionnames[] = { static const char *const optionnames[] = {
"SO_KEEPALIVE", "SO_DONTROUTE", "SO_BROADCAST", "SO_LINGER", NULL "SO_KEEPALIVE", "SO_DONTROUTE", "SO_BROADCAST", "SO_LINGER", NULL
}; };
const char *option; const char *option;
int err; int err;
if (!lua_isstring(L, -2)) return 0; if (!lua_isstring(L, -2)) return 0;
option = lua_tostring(L, -2); option = lua_tostring(L, -2);
switch (luaL_findstring(option, optionnames)) { switch (luaL_findstring(option, optionnames)) {
case 0: { case 0: {
int bool; int bool;
if (!lua_isnumber(L, -1)) if (!lua_isnumber(L, -1))
lua_error(L, "invalid SO_KEEPALIVE value"); lua_error(L, "invalid SO_KEEPALIVE value");
bool = (int) lua_tonumber(L, -1); bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_KEEPALIVE, err = setsockopt(sock->sock, SOL_SOCKET, SO_KEEPALIVE,
(char *) &bool, sizeof(bool)); (char *) &bool, sizeof(bool));
return err >= 0; return err >= 0;
} }
case 1: { case 1: {
int bool; int bool;
if (!lua_isnumber(L, -1)) if (!lua_isnumber(L, -1))
lua_error(L, "invalid SO_DONTROUTE value"); lua_error(L, "invalid SO_DONTROUTE value");
bool = (int) lua_tonumber(L, -1); bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_DONTROUTE, err = setsockopt(sock->sock, SOL_SOCKET, SO_DONTROUTE,
(char *) &bool, sizeof(bool)); (char *) &bool, sizeof(bool));
return err >= 0; return err >= 0;
} }
case 2: { case 2: {
int bool; int bool;
if (!lua_isnumber(L, -1)) if (!lua_isnumber(L, -1))
lua_error(L, "invalid SO_BROADCAST value"); lua_error(L, "invalid SO_BROADCAST value");
bool = (int) lua_tonumber(L, -1); bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_BROADCAST, err = setsockopt(sock->sock, SOL_SOCKET, SO_BROADCAST,
(char *) &bool, sizeof(bool)); (char *) &bool, sizeof(bool));
return err >= 0; return err >= 0;
} }
case 3: { case 3: {
struct linger linger; struct linger linger;
if (!lua_istable(L, -1)) if (!lua_istable(L, -1))
lua_error(L, "invalid SO_LINGER value"); lua_error(L, "invalid SO_LINGER value");
lua_pushstring(L, "l_onoff"); lua_pushstring(L, "l_onoff");
lua_gettable(L, -2); lua_gettable(L, -2);
if (!lua_isnumber(L, -1)) if (!lua_isnumber(L, -1))
lua_error(L, "invalid SO_LINGER (l_onoff) value"); lua_error(L, "invalid SO_LINGER (l_onoff) value");
linger.l_onoff = (int) lua_tonumber(L, -1); linger.l_onoff = (int) lua_tonumber(L, -1);
lua_pop(L, 1); lua_pop(L, 1);
lua_pushstring(L, "l_linger"); lua_pushstring(L, "l_linger");
lua_gettable(L, -2); lua_gettable(L, -2);
if (!lua_isnumber(L, -1)) if (!lua_isnumber(L, -1))
lua_error(L, "invalid SO_LINGER (l_linger) value"); lua_error(L, "invalid SO_LINGER (l_linger) value");
linger.l_linger = (int) lua_tonumber(L, -1); linger.l_linger = (int) lua_tonumber(L, -1);
lua_pop(L, 1); lua_pop(L, 1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_LINGER, err = setsockopt(sock->sock, SOL_SOCKET, SO_LINGER,
(char *) &linger, sizeof(linger)); (char *) &linger, sizeof(linger));
return err >= 0; return err >= 0;
} }
default: return 0; default: return 0;
} }
} }
@ -1675,21 +1669,18 @@ LUASOCKET_API int lua_socketlibopen(lua_State *L)
unsigned int i; unsigned int i;
/* declare new Lua tags for used userdata values */ /* declare new Lua tags for used userdata values */
p_tags tags = (p_tags) lua_newuserdata(L, sizeof(t_tags)); p_tags tags = (p_tags) lua_newuserdata(L, sizeof(t_tags));
if (!tags) lua_error(L, "out of memory");
/* remove tags userdatum from stack but avoid garbage collection */
lua_ref(L, 1);
tags->client = lua_newtag(L); tags->client = lua_newtag(L);
tags->server = lua_newtag(L); tags->server = lua_newtag(L);
tags->table = lua_newtag(L); tags->table = lua_newtag(L);
tags->udp = lua_newtag(L); tags->udp = lua_newtag(L);
/* global functions exported */ /* global functions exported */
for (i = 0; i < sizeof(funcs)/sizeof(funcs[0]); i++) { for (i = 0; i < sizeof(funcs)/sizeof(funcs[0]); i++) {
lua_pushuserdata(L, tags); lua_pushvalue(L, -1);
lua_pushcclosure(L, funcs[i].func, 1); lua_pushcclosure(L, funcs[i].func, 1);
lua_setglobal(L, funcs[i].name); lua_setglobal(L, funcs[i].name);
} }
/* socket garbage collection */ /* socket garbage collection */
lua_pushuserdata(L, tags); lua_pushvalue(L, -1);
lua_pushcclosure(L, gc_table, 1); lua_pushcclosure(L, gc_table, 1);
lua_settagmethod(L, tags->table, "gc"); lua_settagmethod(L, tags->table, "gc");
#ifdef WIN32 #ifdef WIN32
@ -1714,13 +1705,15 @@ LUASOCKET_API int lua_socketlibopen(lua_State *L)
unsigned int i; unsigned int i;
for (i = 0; i < sizeof(global)/sizeof(char *); i++) { for (i = 0; i < sizeof(global)/sizeof(char *); i++) {
lua_pushstring(L, global[i]); lua_pushstring(L, global[i]);
lua_pushuserdata(L, tags); lua_pushvalue(L, -2);
lua_pushcclosure(L, global_callfromtable, 2); lua_pushcclosure(L, global_callfromtable, 2);
lua_setglobal(L, global[i]); lua_setglobal(L, global[i]);
} }
} }
#endif #endif
return 1; /* remove tags userdatum from stack */
lua_pop(L, 1);
return 1;
} }
/*=========================================================================*\ /*=========================================================================*\
@ -1745,12 +1738,11 @@ static p_sock push_clienttable(lua_State *L, p_tags tags)
{"timeout", table_timeout}, {"timeout", table_timeout},
}; };
unsigned int i; unsigned int i;
p_sock sock; p_sock sock = (p_sock) lua_newuserdata(L, sizeof(t_sock));
lua_settag(L, tags->client);
lua_newtable(L); lua_settag(L, tags->table); lua_newtable(L); lua_settag(L, tags->table);
lua_pushstring(L, P_SOCK); lua_pushstring(L, P_SOCK);
sock = (p_sock) lua_newuserdata(L, sizeof(t_sock)); lua_pushvalue(L, -3);
if (!sock) return NULL;
lua_settag(L, tags->client);
lua_settable(L, -3); lua_settable(L, -3);
sock->sock = INVALID_SOCKET; sock->sock = INVALID_SOCKET;
sock->is_connected = 0; sock->is_connected = 0;
@ -1759,7 +1751,7 @@ static p_sock push_clienttable(lua_State *L, p_tags tags)
sock->bf_first = sock->bf_last = 0; sock->bf_first = sock->bf_last = 0;
for (i = 0; i < sizeof(funcs)/sizeof(funcs[0]); i++) { for (i = 0; i < sizeof(funcs)/sizeof(funcs[0]); i++) {
lua_pushstring(L, funcs[i].name); lua_pushstring(L, funcs[i].name);
lua_pushusertag(L, sock, tags->client); lua_pushvalue(L, -3);
lua_pushcclosure(L, funcs[i].func, 1); lua_pushcclosure(L, funcs[i].func, 1);
lua_settable(L, -3); lua_settable(L, -3);
} }
@ -1782,12 +1774,11 @@ static p_sock push_servertable(lua_State *L, p_tags tags)
{"timeout", table_timeout}, {"timeout", table_timeout},
}; };
unsigned int i; unsigned int i;
p_sock sock; p_sock sock = (p_sock) lua_newuserdata(L, sizeof(t_sock));
lua_settag(L, tags->server);
lua_newtable(L); lua_settag(L, tags->table); lua_newtable(L); lua_settag(L, tags->table);
lua_pushstring(L, P_SOCK); lua_pushstring(L, P_SOCK);
sock = (p_sock) lua_newuserdata(L, sizeof(t_sock)); lua_pushvalue(L, -3);
if (!sock) return NULL;
lua_settag(L, tags->server);
lua_settable(L, -3); lua_settable(L, -3);
sock->sock = INVALID_SOCKET; sock->sock = INVALID_SOCKET;
sock->tm_block = -1; sock->tm_block = -1;
@ -1795,14 +1786,18 @@ static p_sock push_servertable(lua_State *L, p_tags tags)
sock->bf_first = sock->bf_last = 0; sock->bf_first = sock->bf_last = 0;
for (i = 0; i < sizeof(funcs)/sizeof(funcs[0]); i++) { for (i = 0; i < sizeof(funcs)/sizeof(funcs[0]); i++) {
lua_pushstring(L, funcs[i].name); lua_pushstring(L, funcs[i].name);
lua_pushusertag(L, sock, tags->client); lua_pushvalue(L, -3);
lua_pushcclosure(L, funcs[i].func, 1); lua_pushcclosure(L, funcs[i].func, 1);
lua_settable(L, -3); lua_settable(L, -3);
} }
/* the accept method is different, it needs the tags closure too */ /* the accept method is different, it needs the tags closure too */
lua_pushstring(L, "accept"); lua_pushstring(L, "accept");
#ifdef LUASOCKET_41FRIENDLY
lua_newuserdatabox(L, tags);
#else
lua_pushuserdata(L, tags); lua_pushuserdata(L, tags);
lua_pushusertag(L, sock, tags->client); #endif
lua_pushvalue(L, -4);
lua_pushcclosure(L, table_tcpaccept, 2); lua_pushcclosure(L, table_tcpaccept, 2);
lua_settable(L, -3); lua_settable(L, -3);
return sock; return sock;
@ -1831,16 +1826,11 @@ static p_sock push_udptable(lua_State *L, p_tags tags)
{"timeout", table_timeout}, {"timeout", table_timeout},
}; };
unsigned int i; unsigned int i;
p_sock sock; p_sock sock = (p_sock) lua_newuserdata(L, sizeof(t_sock));
lua_settag(L, tags->udp);
lua_newtable(L); lua_settag(L, tags->table); lua_newtable(L); lua_settag(L, tags->table);
lua_pushstring(L, P_SOCK); lua_pushstring(L, P_SOCK);
sock = (p_sock) lua_newuserdata(L, sizeof(t_sock)); lua_pushvalue(L, -3);
if (!sock) {
lua_pushnil(L);
lua_pushstring(L, "out of memory");
return NULL;
}
lua_settag(L, tags->udp);
lua_settable(L, -3); lua_settable(L, -3);
sock->sock = socket(AF_INET, SOCK_DGRAM, 0); sock->sock = socket(AF_INET, SOCK_DGRAM, 0);
if (sock->sock == INVALID_SOCKET) { if (sock->sock == INVALID_SOCKET) {
@ -1854,7 +1844,7 @@ static p_sock push_udptable(lua_State *L, p_tags tags)
sock->bf_first = sock->bf_last = 0; sock->bf_first = sock->bf_last = 0;
for (i = 0; i < sizeof(funcs)/sizeof(funcs[0]); i++) { for (i = 0; i < sizeof(funcs)/sizeof(funcs[0]); i++) {
lua_pushstring(L, funcs[i].name); lua_pushstring(L, funcs[i].name);
lua_pushusertag(L, sock, tags->udp); lua_pushvalue(L, -3);
lua_pushcclosure(L, funcs[i].func, 1); lua_pushcclosure(L, funcs[i].func, 1);
lua_settable(L, -3); lua_settable(L, -3);
} }