diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/options.c | 70 | ||||
-rw-r--r-- | src/options.h | 8 | ||||
-rw-r--r-- | src/tcp.c | 16 |
3 files changed, 94 insertions, 0 deletions
diff --git a/src/options.c b/src/options.c index a464a4b..2085fdc 100644 --- a/src/options.c +++ b/src/options.c | |||
@@ -18,8 +18,11 @@ | |||
18 | \*=========================================================================*/ | 18 | \*=========================================================================*/ |
19 | static int opt_setmembership(lua_State *L, p_socket ps, int level, int name); | 19 | static int opt_setmembership(lua_State *L, p_socket ps, int level, int name); |
20 | static int opt_setboolean(lua_State *L, p_socket ps, int level, int name); | 20 | static int opt_setboolean(lua_State *L, p_socket ps, int level, int name); |
21 | static int opt_getboolean(lua_State *L, p_socket ps, int level, int name); | ||
21 | static int opt_set(lua_State *L, p_socket ps, int level, int name, | 22 | static int opt_set(lua_State *L, p_socket ps, int level, int name, |
22 | void *val, int len); | 23 | void *val, int len); |
24 | static int opt_get(lua_State *L, p_socket ps, int level, int name, | ||
25 | void *val, int* len); | ||
23 | 26 | ||
24 | /*=========================================================================*\ | 27 | /*=========================================================================*\ |
25 | * Exported functions | 28 | * Exported functions |
@@ -40,23 +43,51 @@ int opt_meth_setoption(lua_State *L, p_opt opt, p_socket ps) | |||
40 | return opt->func(L, ps); | 43 | return opt->func(L, ps); |
41 | } | 44 | } |
42 | 45 | ||
46 | int opt_meth_getoption(lua_State *L, p_opt opt, p_socket ps) | ||
47 | { | ||
48 | const char *name = luaL_checkstring(L, 2); /* obj, name, ... */ | ||
49 | while (opt->name && strcmp(name, opt->name)) | ||
50 | opt++; | ||
51 | if (!opt->func) { | ||
52 | char msg[45]; | ||
53 | sprintf(msg, "unsupported option `%.35s'", name); | ||
54 | luaL_argerror(L, 2, msg); | ||
55 | } | ||
56 | return opt->func(L, ps); | ||
57 | } | ||
58 | |||
43 | /* enables reuse of local address */ | 59 | /* enables reuse of local address */ |
44 | int opt_reuseaddr(lua_State *L, p_socket ps) | 60 | int opt_reuseaddr(lua_State *L, p_socket ps) |
45 | { | 61 | { |
46 | return opt_setboolean(L, ps, SOL_SOCKET, SO_REUSEADDR); | 62 | return opt_setboolean(L, ps, SOL_SOCKET, SO_REUSEADDR); |
47 | } | 63 | } |
48 | 64 | ||
65 | int opt_get_reuseaddr(lua_State *L, p_socket ps) | ||
66 | { | ||
67 | return opt_getboolean(L, ps, SOL_SOCKET, SO_REUSEADDR); | ||
68 | } | ||
69 | |||
49 | /* disables the Naggle algorithm */ | 70 | /* disables the Naggle algorithm */ |
50 | int opt_tcp_nodelay(lua_State *L, p_socket ps) | 71 | int opt_tcp_nodelay(lua_State *L, p_socket ps) |
51 | { | 72 | { |
52 | return opt_setboolean(L, ps, IPPROTO_TCP, TCP_NODELAY); | 73 | return opt_setboolean(L, ps, IPPROTO_TCP, TCP_NODELAY); |
53 | } | 74 | } |
54 | 75 | ||
76 | int opt_get_tcp_nodelay(lua_State *L, p_socket ps) | ||
77 | { | ||
78 | return opt_getboolean(L, ps, IPPROTO_TCP, TCP_NODELAY); | ||
79 | } | ||
80 | |||
55 | int opt_keepalive(lua_State *L, p_socket ps) | 81 | int opt_keepalive(lua_State *L, p_socket ps) |
56 | { | 82 | { |
57 | return opt_setboolean(L, ps, SOL_SOCKET, SO_KEEPALIVE); | 83 | return opt_setboolean(L, ps, SOL_SOCKET, SO_KEEPALIVE); |
58 | } | 84 | } |
59 | 85 | ||
86 | int opt_get_keepalive(lua_State *L, p_socket ps) | ||
87 | { | ||
88 | return opt_getboolean(L, ps, SOL_SOCKET, SO_KEEPALIVE); | ||
89 | } | ||
90 | |||
60 | int opt_dontroute(lua_State *L, p_socket ps) | 91 | int opt_dontroute(lua_State *L, p_socket ps) |
61 | { | 92 | { |
62 | return opt_setboolean(L, ps, SOL_SOCKET, SO_DONTROUTE); | 93 | return opt_setboolean(L, ps, SOL_SOCKET, SO_DONTROUTE); |
@@ -105,6 +136,21 @@ int opt_ip_drop_membersip(lua_State *L, p_socket ps) | |||
105 | return opt_setmembership(L, ps, IPPROTO_IP, IP_DROP_MEMBERSHIP); | 136 | return opt_setmembership(L, ps, IPPROTO_IP, IP_DROP_MEMBERSHIP); |
106 | } | 137 | } |
107 | 138 | ||
139 | int opt_get_linger(lua_State *L, p_socket ps) | ||
140 | { | ||
141 | struct linger li; /* obj, name */ | ||
142 | int len = sizeof(li); | ||
143 | int err = opt_get(L, ps, SOL_SOCKET, SO_LINGER, (char *) &li, &len); | ||
144 | if (err) | ||
145 | return err; | ||
146 | lua_newtable(L); | ||
147 | lua_pushboolean(L, li.l_onoff); | ||
148 | lua_setfield(L, -2, "on"); | ||
149 | lua_pushinteger(L, li.l_linger); | ||
150 | lua_setfield(L, -2, "timeout"); | ||
151 | return 1; | ||
152 | } | ||
153 | |||
108 | /*=========================================================================*\ | 154 | /*=========================================================================*\ |
109 | * Auxiliar functions | 155 | * Auxiliar functions |
110 | \*=========================================================================*/ | 156 | \*=========================================================================*/ |
@@ -130,6 +176,19 @@ static int opt_setmembership(lua_State *L, p_socket ps, int level, int name) | |||
130 | } | 176 | } |
131 | 177 | ||
132 | static | 178 | static |
179 | int opt_get(lua_State *L, p_socket ps, int level, int name, void *val, int* len) | ||
180 | { | ||
181 | socklen_t socklen = *len; | ||
182 | if (getsockopt(*ps, level, name, (char *) val, &socklen) < 0) { | ||
183 | lua_pushnil(L); | ||
184 | lua_pushstring(L, "getsockopt failed"); | ||
185 | return 2; | ||
186 | } | ||
187 | *len = socklen; | ||
188 | return 0; | ||
189 | } | ||
190 | |||
191 | static | ||
133 | int opt_set(lua_State *L, p_socket ps, int level, int name, void *val, int len) | 192 | int opt_set(lua_State *L, p_socket ps, int level, int name, void *val, int len) |
134 | { | 193 | { |
135 | if (setsockopt(*ps, level, name, (char *) val, len) < 0) { | 194 | if (setsockopt(*ps, level, name, (char *) val, len) < 0) { |
@@ -141,6 +200,17 @@ int opt_set(lua_State *L, p_socket ps, int level, int name, void *val, int len) | |||
141 | return 1; | 200 | return 1; |
142 | } | 201 | } |
143 | 202 | ||
203 | static int opt_getboolean(lua_State *L, p_socket ps, int level, int name) | ||
204 | { | ||
205 | int val = 0; | ||
206 | int len = sizeof(val); | ||
207 | int err = opt_get(L, ps, level, name, (char *) &val, &len); | ||
208 | if (err) | ||
209 | return err; | ||
210 | lua_pushboolean(L, val); | ||
211 | return 1; | ||
212 | } | ||
213 | |||
144 | static int opt_setboolean(lua_State *L, p_socket ps, int level, int name) | 214 | static int opt_setboolean(lua_State *L, p_socket ps, int level, int name) |
145 | { | 215 | { |
146 | int val = auxiliar_checkboolean(L, 3); /* obj, name, bool */ | 216 | int val = auxiliar_checkboolean(L, 3); /* obj, name, bool */ |
diff --git a/src/options.h b/src/options.h index 900761e..c9b2e47 100644 --- a/src/options.h +++ b/src/options.h | |||
@@ -17,10 +17,17 @@ | |||
17 | typedef struct t_opt { | 17 | typedef struct t_opt { |
18 | const char *name; | 18 | const char *name; |
19 | int (*func)(lua_State *L, p_socket ps); | 19 | int (*func)(lua_State *L, p_socket ps); |
20 | int (*get)(lua_State *L, p_socket ps); | ||
20 | } t_opt; | 21 | } t_opt; |
21 | typedef t_opt *p_opt; | 22 | typedef t_opt *p_opt; |
22 | 23 | ||
23 | /* supported options */ | 24 | /* supported options */ |
25 | int opt_get_reuseaddr(lua_State *L, p_socket ps); | ||
26 | int opt_get_tcp_nodelay(lua_State *L, p_socket ps); | ||
27 | int opt_get_keepalive(lua_State *L, p_socket ps); | ||
28 | int opt_get_linger(lua_State *L, p_socket ps); | ||
29 | int opt_get_reuseaddr(lua_State *L, p_socket ps); | ||
30 | |||
24 | int opt_dontroute(lua_State *L, p_socket ps); | 31 | int opt_dontroute(lua_State *L, p_socket ps); |
25 | int opt_broadcast(lua_State *L, p_socket ps); | 32 | int opt_broadcast(lua_State *L, p_socket ps); |
26 | int opt_reuseaddr(lua_State *L, p_socket ps); | 33 | int opt_reuseaddr(lua_State *L, p_socket ps); |
@@ -35,5 +42,6 @@ int opt_ip_drop_membersip(lua_State *L, p_socket ps); | |||
35 | 42 | ||
36 | /* invokes the appropriate option handler */ | 43 | /* invokes the appropriate option handler */ |
37 | int opt_meth_setoption(lua_State *L, p_opt opt, p_socket ps); | 44 | int opt_meth_setoption(lua_State *L, p_opt opt, p_socket ps); |
45 | int opt_meth_getoption(lua_State *L, p_opt opt, p_socket ps); | ||
38 | 46 | ||
39 | #endif | 47 | #endif |
@@ -31,6 +31,7 @@ static int meth_shutdown(lua_State *L); | |||
31 | static int meth_receive(lua_State *L); | 31 | static int meth_receive(lua_State *L); |
32 | static int meth_accept(lua_State *L); | 32 | static int meth_accept(lua_State *L); |
33 | static int meth_close(lua_State *L); | 33 | static int meth_close(lua_State *L); |
34 | static int meth_getoption(lua_State *L); | ||
34 | static int meth_setoption(lua_State *L); | 35 | static int meth_setoption(lua_State *L); |
35 | static int meth_settimeout(lua_State *L); | 36 | static int meth_settimeout(lua_State *L); |
36 | static int meth_getfd(lua_State *L); | 37 | static int meth_getfd(lua_State *L); |
@@ -47,6 +48,7 @@ static luaL_reg tcp[] = { | |||
47 | {"connect", meth_connect}, | 48 | {"connect", meth_connect}, |
48 | {"dirty", meth_dirty}, | 49 | {"dirty", meth_dirty}, |
49 | {"getfd", meth_getfd}, | 50 | {"getfd", meth_getfd}, |
51 | {"getoption", meth_getoption}, | ||
50 | {"getpeername", meth_getpeername}, | 52 | {"getpeername", meth_getpeername}, |
51 | {"getsockname", meth_getsockname}, | 53 | {"getsockname", meth_getsockname}, |
52 | {"getstats", meth_getstats}, | 54 | {"getstats", meth_getstats}, |
@@ -64,6 +66,14 @@ static luaL_reg tcp[] = { | |||
64 | }; | 66 | }; |
65 | 67 | ||
66 | /* socket option handlers */ | 68 | /* socket option handlers */ |
69 | static t_opt optget[] = { | ||
70 | {"keepalive", opt_get_keepalive}, | ||
71 | {"reuseaddr", opt_get_reuseaddr}, | ||
72 | {"tcp-nodelay", opt_get_tcp_nodelay}, | ||
73 | {"linger", opt_get_linger}, | ||
74 | {NULL, NULL} | ||
75 | }; | ||
76 | |||
67 | static t_opt opt[] = { | 77 | static t_opt opt[] = { |
68 | {"keepalive", opt_keepalive}, | 78 | {"keepalive", opt_keepalive}, |
69 | {"reuseaddr", opt_reuseaddr}, | 79 | {"reuseaddr", opt_reuseaddr}, |
@@ -125,6 +135,12 @@ static int meth_setstats(lua_State *L) { | |||
125 | /*-------------------------------------------------------------------------*\ | 135 | /*-------------------------------------------------------------------------*\ |
126 | * Just call option handler | 136 | * Just call option handler |
127 | \*-------------------------------------------------------------------------*/ | 137 | \*-------------------------------------------------------------------------*/ |
138 | static int meth_getoption(lua_State *L) | ||
139 | { | ||
140 | p_tcp tcp = (p_tcp) auxiliar_checkgroup(L, "tcp{any}", 1); | ||
141 | return opt_meth_getoption(L, optget, &tcp->sock); | ||
142 | } | ||
143 | |||
128 | static int meth_setoption(lua_State *L) | 144 | static int meth_setoption(lua_State *L) |
129 | { | 145 | { |
130 | p_tcp tcp = (p_tcp) auxiliar_checkgroup(L, "tcp{any}", 1); | 146 | p_tcp tcp = (p_tcp) auxiliar_checkgroup(L, "tcp{any}", 1); |