dany/luasocket
1
0
Fork 0
mirror of https://github.com/lunarmodules/luasocket.git synced 2024-07-03 11:49:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

Added support to UDP socket options.

Browse Source
This commit is contained in:
Diego Nehab 2001-04-23 22:37:55 +00:00
parent 561177a1dd
commit 8c6473577d
1 changed files with 139 additions and 77 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

216
src/luasocket.c
View File

@ -159,7 +159,7 @@ static int global_tohostname(lua_State *L);
static int global_udpsocket(lua_State *L);
#ifndef LUASOCKET_NOGLOBALS
static int global_calltable(lua_State *L);
static int global_callfromtable(lua_State *L);
#endif
/* luasocket table method API functions */
@ -219,6 +219,7 @@ static const char *udp_setpeername(p_sock sock, const char *address,
unsigned short port);
static const char *udp_setsockname(p_sock sock, const char *address,
unsigned short port);
static int set_option(lua_State *L, p_sock sock);
static void set_reuseaddr(p_sock sock);
static void set_blocking(p_sock sock);
static void set_nonblocking(p_sock sock);
@ -305,8 +306,6 @@ static int global_tcpconnect(lua_State *L)
/*-------------------------------------------------------------------------*\
* Creates a udp socket object and returns it to the Lua script.
* The timeout values are initialized as -1 so that the socket will block
* at any IO operation.
* Lua Returns
* On success: udp socket
* On error: nil, followed by an error message
@ -314,8 +313,16 @@ static int global_tcpconnect(lua_State *L)
static int global_udpsocket(lua_State *L)
{
p_tags tags = pop_tags(L);
int top = lua_gettop(L);
p_sock sock = push_udptable(L, tags);
if (!sock) return 2;
if (top >= 1 && lua_istable(L, 1)) {
lua_pushnil(L);
while (lua_next(L, 1)) {
if (!set_option(L, sock)) lua_error(L, "invalid socket option");
lua_pop(L, 1);
}
}
return 1;
}
@ -483,8 +490,8 @@ static int table_tcpsend(lua_State *L)
for (arg = 2; arg <= top; arg++) { /* skip self table */
int sent;
size_t wanted;
const char *data = luaL_opt_lstr(L, arg, NULL, &wanted);
if (!data || err != NET_DONE) break;
const char *data = luaL_opt_lstr(L, arg, NULL, &wanted);
if (!data || err != NET_DONE) break;
err = send_raw(sock, data, wanted, &sent);
total += sent;
}
@ -542,14 +549,14 @@ static int table_udpsendto(lua_State *L)
* Global function that calls corresponding table method.
\*-------------------------------------------------------------------------*/
#ifndef LUASOCKET_NOGLOBALS
int global_calltable(lua_State *L)
int global_callfromtable(lua_State *L)
{
p_tags tags = pop_tags(L);
if (lua_tag(L, 1) != tags->table) lua_error(L, "invalid socket object");
lua_gettable(L, 1);
lua_insert(L, 1);
lua_call(L, lua_gettop(L)-1, LUA_MULTRET);
return lua_gettop(L);
p_tags tags = pop_tags(L);
if (lua_tag(L, 1) != tags->table) lua_error(L, "invalid socket object");
lua_gettable(L, 1);
lua_insert(L, 1);
lua_call(L, lua_gettop(L)-1, LUA_MULTRET);
return lua_gettop(L);
}
#endif
@ -571,8 +578,8 @@ int global_select(lua_State *L)
fd_set readfds, *prfds = NULL, writefds, *pwfds = NULL;
struct timeval tm, *ptm = NULL;
int ret;
unsigned max = 0;
SOCKET s;
unsigned max = 0;
SOCKET s;
int byfds, canread, canwrite;
/* reset the file descriptor sets */
FD_ZERO(&readfds); FD_ZERO(&writefds);
@ -588,25 +595,25 @@ int global_select(lua_State *L)
while (lua_next(L, 1)) {
if (lua_tag(L, -1) == tags->table) { /* skip strange fields */
p_sock sock = get_sock(L, -1, tags, NULL);
if (sock->sock != INVALID_SOCKET) { /* skip closed sockets */
lua_pushnumber(L, sock->sock);
lua_pushvalue(L, -2);
lua_settable(L, byfds);
if (sock->sock > max) max = sock->sock;
/* a socket can have unread data in our internal
buffer. in that case, we only call select to find
out which of the other sockets can be written to
or read from immediately. */
if (!bf_isempty(sock)) {
ms = 0;
lua_pushnumber(L, lua_getn(L, canread) + 1);
lua_pushvalue(L, -2);
lua_settable(L, canread);
} else {
FD_SET(sock->sock, &readfds);
prfds = &readfds;
}
}
if (sock->sock != INVALID_SOCKET) { /* skip closed sockets */
lua_pushnumber(L, sock->sock);
lua_pushvalue(L, -2);
lua_settable(L, byfds);
if (sock->sock > max) max = sock->sock;
/* a socket can have unread data in our internal
buffer. in that case, we only call select to find
out which of the other sockets can be written to
or read from immediately. */
if (!bf_isempty(sock)) {
ms = 0;
lua_pushnumber(L, lua_getn(L, canread) + 1);
lua_pushvalue(L, -2);
lua_settable(L, canread);
} else {
FD_SET(sock->sock, &readfds);
prfds = &readfds;
}
}
}
/* get rid of lua_next value and expose index */
lua_pop(L, 1);
@ -618,14 +625,14 @@ int global_select(lua_State *L)
while (lua_next(L, 2)) {
if (lua_tag(L, -1) == tags->table) { /* skip strange fields */
p_sock sock = get_sock(L, -1, tags, NULL);
if (sock->sock != INVALID_SOCKET) { /* skip closed sockets */
lua_pushnumber(L, sock->sock);
lua_pushvalue(L, -2);
lua_settable(L, byfds);
if (sock->sock > max) max = sock->sock;
FD_SET(sock->sock, &writefds);
pwfds = &writefds;
}
if (sock->sock != INVALID_SOCKET) { /* skip closed sockets */
lua_pushnumber(L, sock->sock);
lua_pushvalue(L, -2);
lua_settable(L, byfds);
if (sock->sock > max) max = sock->sock;
FD_SET(sock->sock, &writefds);
pwfds = &writefds;
}
}
/* get rid of lua_next value and expose index */
lua_pop(L, 1);
@ -1049,6 +1056,61 @@ void set_reuseaddr(p_sock sock)
setsockopt(sock->sock, SOL_SOCKET, SO_REUSEADDR, (char *)&val, sizeof(val));
}
/*-------------------------------------------------------------------------*\
* 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 to set option
* Returns
* 1 if successful, 0 otherwise
\*-------------------------------------------------------------------------*/
static int set_option(lua_State *L, p_sock sock)
{
static const char *const optionnames[] = {
"SO_KEEPALIVE", "SO_DONTROUTE", "SO_BROADCAST", "SO_LINGER", NULL
};
const char *option = lua_tostring(L, -2);
int err;
switch (luaL_findstring(option, optionnames)) {
case 0: {
int bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_KEEPALIVE, &bool,
sizeof(bool));
return err >= 0;
}
case 1: {
int bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_DONTROUTE, &bool,
sizeof(bool));
return err >= 0;
}
case 2: {
int bool = (int) lua_tonumber(L, -1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_BROADCAST, &bool,
sizeof(bool));
return err >= 0;
}
case 3: {
struct linger linger;
if (!lua_istable(L, -1)) return 0;
lua_pushstring(L, "l_onoff");
lua_gettable(L, -2);
linger.l_onoff = lua_tonumber(L, -1);
lua_pop(L, 1);
lua_pushstring(L, "l_linger");
lua_gettable(L, -2);
linger.l_linger = lua_tonumber(L, -1);
lua_pop(L, 1);
err = setsockopt(sock->sock, SOL_SOCKET, SO_LINGER, &linger,
sizeof(linger));
return err >= 0;
}
default: return 0;
}
}
/*-------------------------------------------------------------------------*\
* Tries to create a TCP socket and bind it to (address, port)
* Input
@ -1462,20 +1524,20 @@ static int receive_dosline(lua_State *L, p_sock sock)
}
buffer = bf_receive(sock, &got);
if (got <= 0) {
luaL_pushresult(&b);
luaL_pushresult(&b);
return NET_CLOSED;
}
pos = 0;
}
pos = 0;
while (pos < got && buffer[pos] != '\n') {
/* we ignore all \r's */
if (buffer[pos] != '\r') luaL_putchar(&b, buffer[pos]);
pos++;
}
if (pos < got) {
luaL_pushresult(&b);
bf_skip(sock, pos+1); /* skip '\n' too */
return NET_DONE;
} else bf_skip(sock, pos);
/* we ignore all \r's */
if (buffer[pos] != '\r') luaL_putchar(&b, buffer[pos]);
pos++;
}
if (pos < got) {
luaL_pushresult(&b);
bf_skip(sock, pos+1); /* skip '\n' too */
return NET_DONE;
} else bf_skip(sock, pos);
}
}
@ -1501,17 +1563,17 @@ static int receive_unixline(lua_State *L, p_sock sock)
}
buffer = bf_receive(sock, &got);
if (got <= 0) {
luaL_pushresult(&b);
luaL_pushresult(&b);
return NET_CLOSED;
}
pos = 0;
}
pos = 0;
while (pos < got && buffer[pos] != '\n') pos++;
luaL_addlstring(&b, buffer, pos);
if (pos < got) {
luaL_pushresult(&b);
bf_skip(sock, pos+1); /* skip '\n' too */
return NET_DONE;
} else bf_skip(sock, pos);
luaL_addlstring(&b, buffer, pos);
if (pos < got) {
luaL_pushresult(&b);
bf_skip(sock, pos+1); /* skip '\n' too */
return NET_DONE;
} else bf_skip(sock, pos);
}
}
@ -1545,7 +1607,7 @@ static int receive_word(lua_State *L, p_sock sock)
bf_skip(sock, pos);
if (pos < got) {
buffer += pos;
got -= pos;
got -= pos;
pos = 0;
break;
}
@ -1620,20 +1682,20 @@ void lua_socketlibopen(lua_State *L)
lua_pushcfunction(L, global_time); lua_setglobal(L, "time");
#endif
#ifndef LUASOCKET_NOGLOBALS
{
char *global[] = {
"accept", "close", "getpeername",
"getsockname", "receive", "send",
"receivefrom", "sendto"
};
unsigned int i;
for (i = 0; i < sizeof(global)/sizeof(char *); i++) {
lua_pushstring(L, global[i]);
lua_pushuserdata(L, tags);
lua_pushcclosure(L, global_calltable, 2);
lua_setglobal(L, global[i]);
}
}
{
char *global[] = {
"accept", "close", "getpeername",
"getsockname", "receive", "send",
"receivefrom", "sendto"
};
unsigned int i;
for (i = 0; i < sizeof(global)/sizeof(char *); i++) {
lua_pushstring(L, global[i]);
lua_pushuserdata(L, tags);
lua_pushcclosure(L, global_callfromtable, 2);
lua_setglobal(L, global[i]);
}
}
#endif
}