| /*=========================================================================*\ |
| * Common option interface |
| * LuaSocket toolkit |
| * |
| * RCS ID: $Id: options.c,v 1.6 2005/11/20 07:20:23 diego Exp $ |
| \*=========================================================================*/ |
| #include <string.h> |
| |
| #include "lauxlib.h" |
| |
| #include "auxiliar.h" |
| #include "options.h" |
| #include "inet.h" |
| |
| |
| /*=========================================================================*\ |
| * Internal functions prototypes |
| \*=========================================================================*/ |
| static int opt_setmembership(lua_State *L, p_socket ps, int level, int name); |
| static int opt_setboolean(lua_State *L, p_socket ps, int level, int name); |
| static int opt_set(lua_State *L, p_socket ps, int level, int name, |
| void *val, int len); |
| |
| /*=========================================================================*\ |
| * Exported functions |
| \*=========================================================================*/ |
| /*-------------------------------------------------------------------------*\ |
| * Calls appropriate option handler |
| \*-------------------------------------------------------------------------*/ |
| int opt_meth_setoption(lua_State *L, p_opt opt, p_socket ps) |
| { |
| const char *name = luaL_checkstring(L, 2); /* obj, name, ... */ |
| while (opt->name && strcmp(name, opt->name)) |
| opt++; |
| if (!opt->func) { |
| char msg[45]; |
| sprintf(msg, "unsupported option `%.35s'", name); |
| luaL_argerror(L, 2, msg); |
| } |
| return opt->func(L, ps); |
| } |
| |
| /* enables reuse of local address */ |
| int opt_reuseaddr(lua_State *L, p_socket ps) |
| { |
| return opt_setboolean(L, ps, SOL_SOCKET, SO_REUSEADDR); |
| } |
| |
| /* disables the Naggle algorithm */ |
| int opt_tcp_nodelay(lua_State *L, p_socket ps) |
| { |
| return opt_setboolean(L, ps, IPPROTO_TCP, TCP_NODELAY); |
| } |
| |
| int opt_keepalive(lua_State *L, p_socket ps) |
| { |
| return opt_setboolean(L, ps, SOL_SOCKET, SO_KEEPALIVE); |
| } |
| |
| int opt_dontroute(lua_State *L, p_socket ps) |
| { |
| return opt_setboolean(L, ps, SOL_SOCKET, SO_DONTROUTE); |
| } |
| |
| int opt_broadcast(lua_State *L, p_socket ps) |
| { |
| return opt_setboolean(L, ps, SOL_SOCKET, SO_BROADCAST); |
| } |
| |
| int opt_ip_multicast_loop(lua_State *L, p_socket ps) |
| { |
| return opt_setboolean(L, ps, IPPROTO_IP, IP_MULTICAST_LOOP); |
| } |
| |
| int opt_linger(lua_State *L, p_socket ps) |
| { |
| struct linger li; /* obj, name, table */ |
| if (!lua_istable(L, 3)) luaL_typerror(L, 3, lua_typename(L, LUA_TTABLE)); |
| lua_pushstring(L, "on"); |
| lua_gettable(L, 3); |
| if (!lua_isboolean(L, -1)) |
| luaL_argerror(L, 3, "boolean 'on' field expected"); |
| li.l_onoff = (u_short) lua_toboolean(L, -1); |
| lua_pushstring(L, "timeout"); |
| lua_gettable(L, 3); |
| if (!lua_isnumber(L, -1)) |
| luaL_argerror(L, 3, "number 'timeout' field expected"); |
| li.l_linger = (u_short) lua_tonumber(L, -1); |
| return opt_set(L, ps, SOL_SOCKET, SO_LINGER, (char *) &li, sizeof(li)); |
| } |
| |
| int opt_ip_multicast_ttl(lua_State *L, p_socket ps) |
| { |
| int val = (int) luaL_checknumber(L, 3); /* obj, name, int */ |
| return opt_set(L, ps, SOL_SOCKET, SO_LINGER, (char *) &val, sizeof(val)); |
| } |
| |
| int opt_ip_add_membership(lua_State *L, p_socket ps) |
| { |
| return opt_setmembership(L, ps, IPPROTO_IP, IP_ADD_MEMBERSHIP); |
| } |
| |
| int opt_ip_drop_membersip(lua_State *L, p_socket ps) |
| { |
| return opt_setmembership(L, ps, IPPROTO_IP, IP_DROP_MEMBERSHIP); |
| } |
| |
| /*=========================================================================*\ |
| * Auxiliar functions |
| \*=========================================================================*/ |
| static int opt_setmembership(lua_State *L, p_socket ps, int level, int name) |
| { |
| struct ip_mreq val; /* obj, name, table */ |
| if (!lua_istable(L, 3)) luaL_typerror(L, 3, lua_typename(L, LUA_TTABLE)); |
| lua_pushstring(L, "multiaddr"); |
| lua_gettable(L, 3); |
| if (!lua_isstring(L, -1)) |
| luaL_argerror(L, 3, "string 'multiaddr' field expected"); |
| if (!inet_aton(lua_tostring(L, -1), &val.imr_multiaddr)) |
| luaL_argerror(L, 3, "invalid 'multiaddr' ip address"); |
| lua_pushstring(L, "interface"); |
| lua_gettable(L, 3); |
| if (!lua_isstring(L, -1)) |
| luaL_argerror(L, 3, "string 'interface' field expected"); |
| val.imr_interface.s_addr = htonl(INADDR_ANY); |
| if (strcmp(lua_tostring(L, -1), "*") && |
| !inet_aton(lua_tostring(L, -1), &val.imr_interface)) |
| luaL_argerror(L, 3, "invalid 'interface' ip address"); |
| return opt_set(L, ps, level, name, (char *) &val, sizeof(val)); |
| } |
| |
| static |
| int opt_set(lua_State *L, p_socket ps, int level, int name, void *val, int len) |
| { |
| if (setsockopt(*ps, level, name, (char *) val, len) < 0) { |
| lua_pushnil(L); |
| lua_pushstring(L, "setsockopt failed"); |
| return 2; |
| } |
| lua_pushnumber(L, 1); |
| return 1; |
| } |
| |
| static int opt_setboolean(lua_State *L, p_socket ps, int level, int name) |
| { |
| int val = auxiliar_checkboolean(L, 3); /* obj, name, bool */ |
| return opt_set(L, ps, level, name, (char *) &val, sizeof(val)); |
| } |
| |