dany/luasocket
1
0
Fork 0
mirror of https://github.com/lunarmodules/luasocket.git synced 2024-12-26 04:28:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Select was also crashing on non-table parameters.

Browse Source 
Changed time to _time and sleep to _sleep to avoid name clashes.
This commit is contained in:
Diego Nehab 2001-06-04 20:44:39 +00:00
parent f1ae9db45e
commit 791c9f3168
1 changed files with 26 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/luasocket.c
View File

@ -316,12 +316,14 @@ static int global_udpsocket(lua_State *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 && lua_istable(L, 1)) { if (top >= 1 ) {
lua_pushnil(L); if (lua_istable(L, 1)) {
while (lua_next(L, 1)) { lua_pushnil(L);
if (!set_option(L, sock)) lua_error(L, "invalid socket option"); while (lua_next(L, 1)) {
lua_pop(L, 1); if (!set_option(L, sock)) lua_error(L, "error setting option");
} lua_pop(L, 1);
}
} else luaL_argerror(L, 1, "invalid options");
} }
return 1; return 1;
} }
@ -590,7 +592,7 @@ int global_select(lua_State *L)
/* writable sockets table to be returned */ /* writable sockets table to be returned */
lua_newtable(L); canwrite = lua_gettop(L); lua_newtable(L); canwrite = lua_gettop(L);
/* get sockets we will test for readability into fd_set */ /* get sockets we will test for readability into fd_set */
if (!lua_isnil(L, 1)) { if (lua_istable(L, 1)) {
lua_pushnil(L); lua_pushnil(L);
while (lua_next(L, 1)) { while (lua_next(L, 1)) {
if (lua_tag(L, -1) == tags->table) { /* skip strange fields */ if (lua_tag(L, -1) == tags->table) { /* skip strange fields */
@ -620,7 +622,7 @@ int global_select(lua_State *L)
} }
} }
/* get sockets we will test for writability into fd_set */ /* get sockets we will test for writability into fd_set */
if (!lua_isnil(L, 2)) { if (lua_istable(L, 2)) {
lua_pushnil(L); lua_pushnil(L);
while (lua_next(L, 2)) { while (lua_next(L, 2)) {
if (lua_tag(L, -1) == tags->table) { /* skip strange fields */ if (lua_tag(L, -1) == tags->table) { /* skip strange fields */
@ -1070,23 +1072,31 @@ 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 = lua_tostring(L, -2); const char *option;
int err; int err;
if (!lua_isstring(L, -2)) return 0;
option = lua_tostring(L, -2);
switch (luaL_findstring(option, optionnames)) { switch (luaL_findstring(option, optionnames)) {
case 0: { case 0: {
int bool = (int) lua_tonumber(L, -1); int bool;
if (!lua_isnumber(L, -1)) return 0;
bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_KEEPALIVE, &bool, err = setsockopt(sock->sock, SOL_SOCKET, SO_KEEPALIVE, &bool,
sizeof(bool)); sizeof(bool));
return err >= 0; return err >= 0;
} }
case 1: { case 1: {
int bool = (int) lua_tonumber(L, -1); int bool;
if (!lua_isnumber(L, -1)) return 0;
bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_DONTROUTE, &bool, err = setsockopt(sock->sock, SOL_SOCKET, SO_DONTROUTE, &bool,
sizeof(bool)); sizeof(bool));
return err >= 0; return err >= 0;
} }
case 2: { case 2: {
int bool = (int) lua_tonumber(L, -1); int bool;
if (!lua_isnumber(L, -1)) return 0;
bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_BROADCAST, &bool, err = setsockopt(sock->sock, SOL_SOCKET, SO_BROADCAST, &bool,
sizeof(bool)); sizeof(bool));
return err >= 0; return err >= 0;
@ -1096,10 +1106,12 @@ static int set_option(lua_State *L, p_sock sock)
if (!lua_istable(L, -1)) return 0; if (!lua_istable(L, -1)) return 0;
lua_pushstring(L, "l_onoff"); lua_pushstring(L, "l_onoff");
lua_gettable(L, -2); lua_gettable(L, -2);
if (!lua_isnumber(L, -1)) return 0;
linger.l_onoff = lua_tonumber(L, -1); linger.l_onoff = 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)) return 0;
linger.l_linger = lua_tonumber(L, -1); linger.l_linger = lua_tonumber(L, -1);
lua_pop(L, 1); lua_pop(L, 1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_LINGER, &linger, err = setsockopt(sock->sock, SOL_SOCKET, SO_LINGER, &linger,
@ -1678,8 +1690,8 @@ void lua_socketlibopen(lua_State *L)
#endif #endif
#ifdef _DEBUG #ifdef _DEBUG
/* test support functions */ /* test support functions */
lua_pushcfunction(L, global_sleep); lua_setglobal(L, "sleep"); lua_pushcfunction(L, global_sleep); lua_setglobal(L, "_sleep");
lua_pushcfunction(L, global_time); lua_setglobal(L, "time"); lua_pushcfunction(L, global_time); lua_setglobal(L, "_time");
#endif #endif
#ifndef LUASOCKET_NOGLOBALS #ifndef LUASOCKET_NOGLOBALS
{ {