diff options
author | Diego Nehab <diego@tecgraf.puc-rio.br> | 2004-01-24 00:18:19 +0000 |
---|---|---|
committer | Diego Nehab <diego@tecgraf.puc-rio.br> | 2004-01-24 00:18:19 +0000 |
commit | 0c9f420a3549df3fb331bb24157b65a3301641d4 (patch) | |
tree | e1b6ce40b55a77ed2bc20493f10f8239b7c00071 | |
parent | 42e0e74487ca62b58a8a1fa06580154c632b4942 (diff) | |
download | luasocket-0c9f420a3549df3fb331bb24157b65a3301641d4.tar.gz luasocket-0c9f420a3549df3fb331bb24157b65a3301641d4.tar.bz2 luasocket-0c9f420a3549df3fb331bb24157b65a3301641d4.zip |
New accept/connect code.
Better error checking.
Better tests.
__tostring implemented.
-rw-r--r-- | TODO | 3 | ||||
-rw-r--r-- | src/auxiliar.c | 30 | ||||
-rw-r--r-- | src/inet.c | 14 | ||||
-rw-r--r-- | src/inet.h | 2 | ||||
-rw-r--r-- | src/socket.h | 10 | ||||
-rw-r--r-- | src/tcp.c | 54 | ||||
-rw-r--r-- | src/udp.c | 18 | ||||
-rw-r--r-- | src/usocket.c | 181 | ||||
-rw-r--r-- | src/wsocket.c | 215 | ||||
-rw-r--r-- | test/testclnt.lua | 89 |
10 files changed, 422 insertions, 194 deletions
@@ -12,6 +12,9 @@ manual | |||
12 | local connect | 12 | local connect |
13 | add thanks to 'carlos cassino' and 'david burgess' | 13 | add thanks to 'carlos cassino' and 'david burgess' |
14 | add new ip- options and reuseaddr option | 14 | add new ip- options and reuseaddr option |
15 | add listen to manual | ||
16 | bind method doesn't do listen anymore | ||
17 | bind doesn't turn an object into a server object: listen does. | ||
15 | 18 | ||
16 | tests | 19 | tests |
17 | checar todos os metodos | 20 | checar todos os metodos |
diff --git a/src/auxiliar.c b/src/auxiliar.c index 6888f9c..9a249b6 100644 --- a/src/auxiliar.c +++ b/src/auxiliar.c | |||
@@ -5,6 +5,7 @@ | |||
5 | * RCS ID: $Id$ | 5 | * RCS ID: $Id$ |
6 | \*=========================================================================*/ | 6 | \*=========================================================================*/ |
7 | #include <string.h> | 7 | #include <string.h> |
8 | #include <stdio.h> | ||
8 | 9 | ||
9 | #include "luasocket.h" | 10 | #include "luasocket.h" |
10 | #include "auxiliar.h" | 11 | #include "auxiliar.h" |
@@ -13,6 +14,27 @@ | |||
13 | * Exported functions | 14 | * Exported functions |
14 | \*=========================================================================*/ | 15 | \*=========================================================================*/ |
15 | /*-------------------------------------------------------------------------*\ | 16 | /*-------------------------------------------------------------------------*\ |
17 | * Prints the value of a class in a nice way | ||
18 | \*-------------------------------------------------------------------------*/ | ||
19 | int aux_meth_tostring(lua_State *L) | ||
20 | { | ||
21 | char buf[32]; | ||
22 | if (!lua_getmetatable(L, 1)) goto error; | ||
23 | lua_pushstring(L, "__index"); | ||
24 | lua_gettable(L, -2); | ||
25 | if (!lua_istable(L, -1)) goto error; | ||
26 | lua_pushstring(L, "class"); | ||
27 | lua_gettable(L, -2); | ||
28 | if (!lua_isstring(L, -1)) goto error; | ||
29 | sprintf(buf, "%p", lua_touserdata(L, 1)); | ||
30 | lua_pushfstring(L, "socket: %s: %s", lua_tostring(L, -1), buf); | ||
31 | return 1; | ||
32 | error: | ||
33 | lua_pushnil(L); | ||
34 | return 1; | ||
35 | } | ||
36 | |||
37 | /*-------------------------------------------------------------------------*\ | ||
16 | * Initializes the module | 38 | * Initializes the module |
17 | \*-------------------------------------------------------------------------*/ | 39 | \*-------------------------------------------------------------------------*/ |
18 | void aux_open(lua_State *L) | 40 | void aux_open(lua_State *L) |
@@ -26,14 +48,18 @@ void aux_open(lua_State *L) | |||
26 | void aux_newclass(lua_State *L, const char *classname, luaL_reg *func) | 48 | void aux_newclass(lua_State *L, const char *classname, luaL_reg *func) |
27 | { | 49 | { |
28 | luaL_newmetatable(L, classname); /* mt */ | 50 | luaL_newmetatable(L, classname); /* mt */ |
51 | /* set __tostring metamethod */ | ||
52 | lua_pushstring(L, "__tostring"); | ||
53 | lua_pushcfunction(L, aux_meth_tostring); | ||
54 | lua_rawset(L, -3); | ||
55 | /* create __index table to place methods */ | ||
29 | lua_pushstring(L, "__index"); /* mt,"__index" */ | 56 | lua_pushstring(L, "__index"); /* mt,"__index" */ |
30 | lua_newtable(L); /* mt,"__index",it */ | 57 | lua_newtable(L); /* mt,"__index",it */ |
31 | luaL_openlib(L, NULL, func, 0); | 58 | luaL_openlib(L, NULL, func, 0); |
32 | #ifdef LUASOCKET_DEBUG | 59 | /* put class name into class metatable */ |
33 | lua_pushstring(L, "class"); /* mt,"__index",it,"class" */ | 60 | lua_pushstring(L, "class"); /* mt,"__index",it,"class" */ |
34 | lua_pushstring(L, classname); /* mt,"__index",it,"class",classname */ | 61 | lua_pushstring(L, classname); /* mt,"__index",it,"class",classname */ |
35 | lua_rawset(L, -3); /* mt,"__index",it */ | 62 | lua_rawset(L, -3); /* mt,"__index",it */ |
36 | #endif | ||
37 | /* get __gc method from class and use it for garbage collection */ | 63 | /* get __gc method from class and use it for garbage collection */ |
38 | lua_pushstring(L, "__gc"); /* mt,"__index",it,"__gc" */ | 64 | lua_pushstring(L, "__gc"); /* mt,"__index",it,"__gc" */ |
39 | lua_pushstring(L, "__gc"); /* mt,"__index",it,"__gc","__gc" */ | 65 | lua_pushstring(L, "__gc"); /* mt,"__index",it,"__gc","__gc" */ |
@@ -213,8 +213,7 @@ const char *inet_tryconnect(p_sock ps, const char *address, | |||
213 | /*-------------------------------------------------------------------------*\ | 213 | /*-------------------------------------------------------------------------*\ |
214 | * Tries to bind socket to (address, port) | 214 | * Tries to bind socket to (address, port) |
215 | \*-------------------------------------------------------------------------*/ | 215 | \*-------------------------------------------------------------------------*/ |
216 | const char *inet_trybind(p_sock ps, const char *address, unsigned short port, | 216 | const char *inet_trybind(p_sock ps, const char *address, unsigned short port) |
217 | int backlog) | ||
218 | { | 217 | { |
219 | struct sockaddr_in local; | 218 | struct sockaddr_in local; |
220 | const char *err; | 219 | const char *err; |
@@ -231,16 +230,9 @@ const char *inet_trybind(p_sock ps, const char *address, unsigned short port, | |||
231 | addr = (struct in_addr **) hp->h_addr_list; | 230 | addr = (struct in_addr **) hp->h_addr_list; |
232 | memcpy(&local.sin_addr, *addr, sizeof(struct in_addr)); | 231 | memcpy(&local.sin_addr, *addr, sizeof(struct in_addr)); |
233 | } | 232 | } |
234 | sock_setblocking(ps); | ||
235 | err = sock_bind(ps, (SA *) &local, sizeof(local)); | 233 | err = sock_bind(ps, (SA *) &local, sizeof(local)); |
236 | if (err) { | 234 | if (err) sock_destroy(ps); |
237 | sock_destroy(ps); | 235 | return err; |
238 | return err; | ||
239 | } else { | ||
240 | sock_setnonblocking(ps); | ||
241 | if (backlog >= 0) sock_listen(ps, backlog); | ||
242 | return NULL; | ||
243 | } | ||
244 | } | 236 | } |
245 | 237 | ||
246 | /*-------------------------------------------------------------------------*\ | 238 | /*-------------------------------------------------------------------------*\ |
@@ -30,7 +30,7 @@ const char *inet_trycreate(p_sock ps, int type); | |||
30 | const char *inet_tryconnect(p_sock ps, const char *address, | 30 | const char *inet_tryconnect(p_sock ps, const char *address, |
31 | unsigned short port, p_tm tm); | 31 | unsigned short port, p_tm tm); |
32 | const char *inet_trybind(p_sock ps, const char *address, | 32 | const char *inet_trybind(p_sock ps, const char *address, |
33 | unsigned short port, int backlog); | 33 | unsigned short port); |
34 | 34 | ||
35 | int inet_meth_getpeername(lua_State *L, p_sock ps); | 35 | int inet_meth_getpeername(lua_State *L, p_sock ps); |
36 | int inet_meth_getsockname(lua_State *L, p_sock ps); | 36 | int inet_meth_getsockname(lua_State *L, p_sock ps); |
diff --git a/src/socket.h b/src/socket.h index 7c84baa..2e7b6f9 100644 --- a/src/socket.h +++ b/src/socket.h | |||
@@ -39,7 +39,6 @@ typedef struct sockaddr SA; | |||
39 | \*=========================================================================*/ | 39 | \*=========================================================================*/ |
40 | int sock_open(void); | 40 | int sock_open(void); |
41 | void sock_destroy(p_sock ps); | 41 | void sock_destroy(p_sock ps); |
42 | void sock_listen(p_sock ps, int backlog); | ||
43 | void sock_shutdown(p_sock ps, int how); | 42 | void sock_shutdown(p_sock ps, int how); |
44 | int sock_send(p_sock ps, const char *data, size_t count, | 43 | int sock_send(p_sock ps, const char *data, size_t count, |
45 | size_t *sent, int timeout); | 44 | size_t *sent, int timeout); |
@@ -51,10 +50,15 @@ int sock_recvfrom(p_sock ps, char *data, size_t count, | |||
51 | size_t *got, SA *addr, socklen_t *addr_len, int timeout); | 50 | size_t *got, SA *addr, socklen_t *addr_len, int timeout); |
52 | void sock_setnonblocking(p_sock ps); | 51 | void sock_setnonblocking(p_sock ps); |
53 | void sock_setblocking(p_sock ps); | 52 | void sock_setblocking(p_sock ps); |
54 | int sock_accept(p_sock ps, p_sock pa, SA *addr, socklen_t *addr_len, p_tm tm); | 53 | int sock_select(int n, fd_set *rfds, fd_set *wfds, fd_set *efds, int timeout); |
54 | |||
55 | const char *sock_connect(p_sock ps, SA *addr, socklen_t addr_len, p_tm tm); | 55 | const char *sock_connect(p_sock ps, SA *addr, socklen_t addr_len, p_tm tm); |
56 | const char *sock_create(p_sock ps, int domain, int type, int protocol); | 56 | const char *sock_create(p_sock ps, int domain, int type, int protocol); |
57 | const char *sock_bind(p_sock ps, SA *addr, socklen_t addr_len); | 57 | const char *sock_bind(p_sock ps, SA *addr, socklen_t addr_len); |
58 | const char *sock_hoststrerror(void); | 58 | const char *sock_listen(p_sock ps, int backlog); |
59 | const char *sock_accept(p_sock ps, p_sock pa, SA *addr, | ||
60 | socklen_t *addr_len, p_tm tm); | ||
61 | |||
62 | const char *sock_hoststrerror(); | ||
59 | 63 | ||
60 | #endif /* SOCK_H */ | 64 | #endif /* SOCK_H */ |
@@ -22,6 +22,7 @@ | |||
22 | \*=========================================================================*/ | 22 | \*=========================================================================*/ |
23 | static int global_create(lua_State *L); | 23 | static int global_create(lua_State *L); |
24 | static int meth_connect(lua_State *L); | 24 | static int meth_connect(lua_State *L); |
25 | static int meth_listen(lua_State *L); | ||
25 | static int meth_bind(lua_State *L); | 26 | static int meth_bind(lua_State *L); |
26 | static int meth_send(lua_State *L); | 27 | static int meth_send(lua_State *L); |
27 | static int meth_getsockname(lua_State *L); | 28 | static int meth_getsockname(lua_State *L); |
@@ -32,7 +33,8 @@ static int meth_accept(lua_State *L); | |||
32 | static int meth_close(lua_State *L); | 33 | static int meth_close(lua_State *L); |
33 | static int meth_setoption(lua_State *L); | 34 | static int meth_setoption(lua_State *L); |
34 | static int meth_settimeout(lua_State *L); | 35 | static int meth_settimeout(lua_State *L); |
35 | static int meth_fd(lua_State *L); | 36 | static int meth_getfd(lua_State *L); |
37 | static int meth_setfd(lua_State *L); | ||
36 | static int meth_dirty(lua_State *L); | 38 | static int meth_dirty(lua_State *L); |
37 | 39 | ||
38 | /* tcp object methods */ | 40 | /* tcp object methods */ |
@@ -41,6 +43,7 @@ static luaL_reg tcp[] = { | |||
41 | {"send", meth_send}, | 43 | {"send", meth_send}, |
42 | {"receive", meth_receive}, | 44 | {"receive", meth_receive}, |
43 | {"bind", meth_bind}, | 45 | {"bind", meth_bind}, |
46 | {"listen", meth_listen}, | ||
44 | {"accept", meth_accept}, | 47 | {"accept", meth_accept}, |
45 | {"setpeername", meth_connect}, | 48 | {"setpeername", meth_connect}, |
46 | {"setsockname", meth_bind}, | 49 | {"setsockname", meth_bind}, |
@@ -51,7 +54,8 @@ static luaL_reg tcp[] = { | |||
51 | {"shutdown", meth_shutdown}, | 54 | {"shutdown", meth_shutdown}, |
52 | {"setoption", meth_setoption}, | 55 | {"setoption", meth_setoption}, |
53 | {"__gc", meth_close}, | 56 | {"__gc", meth_close}, |
54 | {"fd", meth_fd}, | 57 | {"getfd", meth_getfd}, |
58 | {"setfd", meth_setfd}, | ||
55 | {"dirty", meth_dirty}, | 59 | {"dirty", meth_dirty}, |
56 | {NULL, NULL} | 60 | {NULL, NULL} |
57 | }; | 61 | }; |
@@ -124,16 +128,24 @@ static int meth_setoption(lua_State *L) | |||
124 | /*-------------------------------------------------------------------------*\ | 128 | /*-------------------------------------------------------------------------*\ |
125 | * Select support methods | 129 | * Select support methods |
126 | \*-------------------------------------------------------------------------*/ | 130 | \*-------------------------------------------------------------------------*/ |
127 | static int meth_fd(lua_State *L) | 131 | static int meth_getfd(lua_State *L) |
128 | { | 132 | { |
129 | p_tcp tcp = (p_tcp) aux_checkgroup(L, "tcp{client,server}", 1); | 133 | p_tcp tcp = (p_tcp) aux_checkgroup(L, "tcp{any}", 1); |
130 | lua_pushnumber(L, tcp->sock); | 134 | lua_pushnumber(L, tcp->sock); |
131 | return 1; | 135 | return 1; |
132 | } | 136 | } |
133 | 137 | ||
138 | /* this is very dangerous, but can be handy for those that are brave enough */ | ||
139 | static int meth_setfd(lua_State *L) | ||
140 | { | ||
141 | p_tcp tcp = (p_tcp) aux_checkgroup(L, "tcp{any}", 1); | ||
142 | tcp->sock = (t_sock) luaL_checknumber(L, 2); | ||
143 | return 0; | ||
144 | } | ||
145 | |||
134 | static int meth_dirty(lua_State *L) | 146 | static int meth_dirty(lua_State *L) |
135 | { | 147 | { |
136 | p_tcp tcp = (p_tcp) aux_checkgroup(L, "tcp{client,server}", 1); | 148 | p_tcp tcp = (p_tcp) aux_checkgroup(L, "tcp{any}", 1); |
137 | lua_pushboolean(L, !buf_isempty(&tcp->buf)); | 149 | lua_pushboolean(L, !buf_isempty(&tcp->buf)); |
138 | return 1; | 150 | return 1; |
139 | } | 151 | } |
@@ -147,9 +159,9 @@ static int meth_accept(lua_State *L) | |||
147 | p_tcp server = (p_tcp) aux_checkclass(L, "tcp{server}", 1); | 159 | p_tcp server = (p_tcp) aux_checkclass(L, "tcp{server}", 1); |
148 | p_tm tm = tm_markstart(&server->tm); | 160 | p_tm tm = tm_markstart(&server->tm); |
149 | t_sock sock; | 161 | t_sock sock; |
150 | int err = sock_accept(&server->sock, &sock, NULL, NULL, tm); | 162 | const char *err = sock_accept(&server->sock, &sock, NULL, NULL, tm); |
151 | /* if successful, push client socket */ | 163 | /* if successful, push client socket */ |
152 | if (err == IO_DONE) { | 164 | if (!err) { |
153 | p_tcp clnt = lua_newuserdata(L, sizeof(t_tcp)); | 165 | p_tcp clnt = lua_newuserdata(L, sizeof(t_tcp)); |
154 | aux_setclass(L, "tcp{client}", -1); | 166 | aux_setclass(L, "tcp{client}", -1); |
155 | /* initialize structure fields */ | 167 | /* initialize structure fields */ |
@@ -160,28 +172,25 @@ static int meth_accept(lua_State *L) | |||
160 | return 1; | 172 | return 1; |
161 | } else { | 173 | } else { |
162 | lua_pushnil(L); | 174 | lua_pushnil(L); |
163 | io_pusherror(L, err); | 175 | lua_pushstring(L, err); |
164 | return 2; | 176 | return 2; |
165 | } | 177 | } |
166 | } | 178 | } |
167 | 179 | ||
168 | /*-------------------------------------------------------------------------*\ | 180 | /*-------------------------------------------------------------------------*\ |
169 | * Turns a master object into a server object | 181 | * Binds an object to an address |
170 | \*-------------------------------------------------------------------------*/ | 182 | \*-------------------------------------------------------------------------*/ |
171 | static int meth_bind(lua_State *L) | 183 | static int meth_bind(lua_State *L) |
172 | { | 184 | { |
173 | p_tcp tcp = (p_tcp) aux_checkclass(L, "tcp{master}", 1); | 185 | p_tcp tcp = (p_tcp) aux_checkclass(L, "tcp{master}", 1); |
174 | const char *address = luaL_checkstring(L, 2); | 186 | const char *address = luaL_checkstring(L, 2); |
175 | unsigned short port = (unsigned short) luaL_checknumber(L, 3); | 187 | unsigned short port = (unsigned short) luaL_checknumber(L, 3); |
176 | int backlog = (int) luaL_optnumber(L, 4, 1); | 188 | const char *err = inet_trybind(&tcp->sock, address, port); |
177 | const char *err = inet_trybind(&tcp->sock, address, port, backlog); | ||
178 | if (err) { | 189 | if (err) { |
179 | lua_pushnil(L); | 190 | lua_pushnil(L); |
180 | lua_pushstring(L, err); | 191 | lua_pushstring(L, err); |
181 | return 2; | 192 | return 2; |
182 | } | 193 | } |
183 | /* turn master object into a server object if there was a listen */ | ||
184 | if (backlog >= 0) aux_setclass(L, "tcp{server}", 1); | ||
185 | lua_pushnumber(L, 1); | 194 | lua_pushnumber(L, 1); |
186 | return 1; | 195 | return 1; |
187 | } | 196 | } |
@@ -218,6 +227,25 @@ static int meth_close(lua_State *L) | |||
218 | } | 227 | } |
219 | 228 | ||
220 | /*-------------------------------------------------------------------------*\ | 229 | /*-------------------------------------------------------------------------*\ |
230 | * Puts the sockt in listen mode | ||
231 | \*-------------------------------------------------------------------------*/ | ||
232 | static int meth_listen(lua_State *L) | ||
233 | { | ||
234 | p_tcp tcp = (p_tcp) aux_checkclass(L, "tcp{master}", 1); | ||
235 | int backlog = (int) luaL_checknumber(L, 2); | ||
236 | const char *err = sock_listen(&tcp->sock, backlog); | ||
237 | if (err) { | ||
238 | lua_pushnil(L); | ||
239 | lua_pushstring(L, err); | ||
240 | return 2; | ||
241 | } | ||
242 | /* turn master object into a server object */ | ||
243 | aux_setclass(L, "tcp{server}", 1); | ||
244 | lua_pushnumber(L, 1); | ||
245 | return 1; | ||
246 | } | ||
247 | |||
248 | /*-------------------------------------------------------------------------*\ | ||
221 | * Shuts the connection down partially | 249 | * Shuts the connection down partially |
222 | \*-------------------------------------------------------------------------*/ | 250 | \*-------------------------------------------------------------------------*/ |
223 | static int meth_shutdown(lua_State *L) | 251 | static int meth_shutdown(lua_State *L) |
@@ -33,7 +33,8 @@ static int meth_close(lua_State *L); | |||
33 | static int meth_shutdown(lua_State *L); | 33 | static int meth_shutdown(lua_State *L); |
34 | static int meth_setoption(lua_State *L); | 34 | static int meth_setoption(lua_State *L); |
35 | static int meth_settimeout(lua_State *L); | 35 | static int meth_settimeout(lua_State *L); |
36 | static int meth_fd(lua_State *L); | 36 | static int meth_getfd(lua_State *L); |
37 | static int meth_setfd(lua_State *L); | ||
37 | static int meth_dirty(lua_State *L); | 38 | static int meth_dirty(lua_State *L); |
38 | 39 | ||
39 | /* udp object methods */ | 40 | /* udp object methods */ |
@@ -51,7 +52,8 @@ static luaL_reg udp[] = { | |||
51 | {"shutdown", meth_shutdown}, | 52 | {"shutdown", meth_shutdown}, |
52 | {"setoption", meth_setoption}, | 53 | {"setoption", meth_setoption}, |
53 | {"__gc", meth_close}, | 54 | {"__gc", meth_close}, |
54 | {"fd", meth_fd}, | 55 | {"getfd", meth_getfd}, |
56 | {"setfd", meth_setfd}, | ||
55 | {"dirty", meth_dirty}, | 57 | {"dirty", meth_dirty}, |
56 | {NULL, NULL} | 58 | {NULL, NULL} |
57 | }; | 59 | }; |
@@ -194,13 +196,21 @@ static int meth_receivefrom(lua_State *L) | |||
194 | /*-------------------------------------------------------------------------*\ | 196 | /*-------------------------------------------------------------------------*\ |
195 | * Select support methods | 197 | * Select support methods |
196 | \*-------------------------------------------------------------------------*/ | 198 | \*-------------------------------------------------------------------------*/ |
197 | static int meth_fd(lua_State *L) | 199 | static int meth_getfd(lua_State *L) |
198 | { | 200 | { |
199 | p_udp udp = (p_udp) aux_checkgroup(L, "udp{any}", 1); | 201 | p_udp udp = (p_udp) aux_checkgroup(L, "udp{any}", 1); |
200 | lua_pushnumber(L, udp->sock); | 202 | lua_pushnumber(L, udp->sock); |
201 | return 1; | 203 | return 1; |
202 | } | 204 | } |
203 | 205 | ||
206 | /* this is very dangerous, but can be handy for those that are brave enough */ | ||
207 | static int meth_setfd(lua_State *L) | ||
208 | { | ||
209 | p_udp udp = (p_udp) aux_checkgroup(L, "udp{any}", 1); | ||
210 | udp->sock = (t_sock) luaL_checknumber(L, 2); | ||
211 | return 0; | ||
212 | } | ||
213 | |||
204 | static int meth_dirty(lua_State *L) | 214 | static int meth_dirty(lua_State *L) |
205 | { | 215 | { |
206 | p_udp udp = (p_udp) aux_checkgroup(L, "udp{any}", 1); | 216 | p_udp udp = (p_udp) aux_checkgroup(L, "udp{any}", 1); |
@@ -312,7 +322,7 @@ static int meth_setsockname(lua_State *L) | |||
312 | p_udp udp = (p_udp) aux_checkclass(L, "udp{unconnected}", 1); | 322 | p_udp udp = (p_udp) aux_checkclass(L, "udp{unconnected}", 1); |
313 | const char *address = luaL_checkstring(L, 2); | 323 | const char *address = luaL_checkstring(L, 2); |
314 | unsigned short port = (unsigned short) luaL_checknumber(L, 3); | 324 | unsigned short port = (unsigned short) luaL_checknumber(L, 3); |
315 | const char *err = inet_trybind(&udp->sock, address, port, -1); | 325 | const char *err = inet_trybind(&udp->sock, address, port); |
316 | if (err) { | 326 | if (err) { |
317 | lua_pushnil(L); | 327 | lua_pushnil(L); |
318 | lua_pushstring(L, err); | 328 | lua_pushstring(L, err); |
diff --git a/src/usocket.c b/src/usocket.c index 5afa1bf..bece354 100644 --- a/src/usocket.c +++ b/src/usocket.c | |||
@@ -20,9 +20,11 @@ | |||
20 | 20 | ||
21 | #include "socket.h" | 21 | #include "socket.h" |
22 | 22 | ||
23 | static const char *sock_createstrerror(void); | 23 | static const char *sock_createstrerror(int err); |
24 | static const char *sock_bindstrerror(void); | 24 | static const char *sock_bindstrerror(int err); |
25 | static const char *sock_connectstrerror(void); | 25 | static const char *sock_connectstrerror(int err); |
26 | static const char *sock_acceptstrerror(int err); | ||
27 | static const char *sock_listenstrerror(int err); | ||
26 | 28 | ||
27 | /*-------------------------------------------------------------------------*\ | 29 | /*-------------------------------------------------------------------------*\ |
28 | * Initializes module | 30 | * Initializes module |
@@ -46,19 +48,31 @@ int sock_open(void) | |||
46 | void sock_destroy(p_sock ps) | 48 | void sock_destroy(p_sock ps) |
47 | { | 49 | { |
48 | if (*ps != SOCK_INVALID) { | 50 | if (*ps != SOCK_INVALID) { |
51 | sock_setblocking(ps); | ||
49 | close(*ps); | 52 | close(*ps); |
50 | *ps = SOCK_INVALID; | 53 | *ps = SOCK_INVALID; |
51 | } | 54 | } |
52 | } | 55 | } |
53 | 56 | ||
54 | /*-------------------------------------------------------------------------*\ | 57 | /*-------------------------------------------------------------------------*\ |
58 | * Select with int timeout in ms | ||
59 | \*-------------------------------------------------------------------------*/ | ||
60 | int sock_select(int n, fd_set *rfds, fd_set *wfds, fd_set *efds, int timeout) | ||
61 | { | ||
62 | struct timeval tv; | ||
63 | tv.tv_sec = timeout / 1000; | ||
64 | tv.tv_usec = (timeout % 1000) * 1000; | ||
65 | return select(n, rfds, wfds, efds, timeout >= 0? &tv: NULL); | ||
66 | } | ||
67 | |||
68 | /*-------------------------------------------------------------------------*\ | ||
55 | * Creates and sets up a socket | 69 | * Creates and sets up a socket |
56 | \*-------------------------------------------------------------------------*/ | 70 | \*-------------------------------------------------------------------------*/ |
57 | const char *sock_create(p_sock ps, int domain, int type, int protocol) | 71 | const char *sock_create(p_sock ps, int domain, int type, int protocol) |
58 | { | 72 | { |
59 | int val = 1; | 73 | int val = 1; |
60 | t_sock sock = socket(domain, type, protocol); | 74 | t_sock sock = socket(domain, type, protocol); |
61 | if (sock == SOCK_INVALID) return sock_createstrerror(); | 75 | if (sock == SOCK_INVALID) return sock_createstrerror(errno); |
62 | *ps = sock; | 76 | *ps = sock; |
63 | sock_setnonblocking(ps); | 77 | sock_setnonblocking(ps); |
64 | setsockopt(*ps, SOL_SOCKET, SO_REUSEADDR, (char *) &val, sizeof(val)); | 78 | setsockopt(*ps, SOL_SOCKET, SO_REUSEADDR, (char *) &val, sizeof(val)); |
@@ -79,27 +93,22 @@ const char *sock_connect(p_sock ps, SA *addr, socklen_t addr_len, p_tm tm) | |||
79 | /* if no error, we're done */ | 93 | /* if no error, we're done */ |
80 | if (err == 0) return NULL; | 94 | if (err == 0) return NULL; |
81 | /* make sure the system is trying to connect */ | 95 | /* make sure the system is trying to connect */ |
82 | if (errno != EINPROGRESS) return io_strerror(IO_ERROR); | 96 | if (errno != EINPROGRESS) return sock_connectstrerror(errno); |
83 | /* wait for a timeout or for the system's answer */ | 97 | /* wait for a timeout or for the system's answer */ |
84 | for ( ;; ) { | 98 | for ( ;; ) { |
85 | struct timeval tv; | ||
86 | fd_set rfds, wfds, efds; | 99 | fd_set rfds, wfds, efds; |
87 | int timeout = tm_getretry(tm); | ||
88 | tv.tv_sec = timeout / 1000; | ||
89 | tv.tv_usec = (timeout % 1000) * 1000; | ||
90 | FD_ZERO(&rfds); FD_SET(sock, &rfds); | 100 | FD_ZERO(&rfds); FD_SET(sock, &rfds); |
91 | FD_ZERO(&wfds); FD_SET(sock, &wfds); | 101 | FD_ZERO(&wfds); FD_SET(sock, &wfds); |
92 | FD_ZERO(&efds); FD_SET(sock, &efds); | 102 | FD_ZERO(&efds); FD_SET(sock, &efds); |
93 | /* we run select to avoid busy waiting */ | 103 | /* we run select to avoid busy waiting */ |
94 | err = select(sock+1, &rfds, &wfds, &efds, timeout >= 0? &tv: NULL); | 104 | do err = sock_select(sock+1, &rfds, &wfds, &efds, tm_getretry(tm)); |
95 | /* if select was interrupted, try again */ | 105 | while (err < 0 && errno == EINTR); |
96 | if (err < 0 && errno == EINTR) continue; | ||
97 | /* if selects readable, try reading */ | 106 | /* if selects readable, try reading */ |
98 | if (err > 0) { | 107 | if (err > 0) { |
99 | char dummy; | 108 | char dummy; |
100 | /* recv will set errno to the value a blocking connect would set */ | 109 | /* recv will set errno to the value a blocking connect would set */ |
101 | if (recv(sock, &dummy, 0, 0) < 0 && errno != EAGAIN) | 110 | if (recv(sock, &dummy, 0, 0) < 0 && errno != EWOULDBLOCK) |
102 | return sock_connectstrerror(); | 111 | return sock_connectstrerror(errno); |
103 | else | 112 | else |
104 | return NULL; | 113 | return NULL; |
105 | /* if no event happened, there was a timeout */ | 114 | /* if no event happened, there was a timeout */ |
@@ -113,16 +122,24 @@ const char *sock_connect(p_sock ps, SA *addr, socklen_t addr_len, p_tm tm) | |||
113 | \*-------------------------------------------------------------------------*/ | 122 | \*-------------------------------------------------------------------------*/ |
114 | const char *sock_bind(p_sock ps, SA *addr, socklen_t addr_len) | 123 | const char *sock_bind(p_sock ps, SA *addr, socklen_t addr_len) |
115 | { | 124 | { |
116 | if (bind(*ps, addr, addr_len) < 0) return sock_bindstrerror(); | 125 | const char *err = NULL; |
117 | else return NULL; | 126 | sock_setblocking(ps); |
127 | if (bind(*ps, addr, addr_len) < 0) err = sock_bindstrerror(errno); | ||
128 | sock_setnonblocking(ps); | ||
129 | return err; | ||
118 | } | 130 | } |
119 | 131 | ||
120 | /*-------------------------------------------------------------------------*\ | 132 | /*-------------------------------------------------------------------------*\ |
121 | * | 133 | * |
122 | \*-------------------------------------------------------------------------*/ | 134 | \*-------------------------------------------------------------------------*/ |
123 | void sock_listen(p_sock ps, int backlog) | 135 | const char* sock_listen(p_sock ps, int backlog) |
124 | { | 136 | { |
125 | listen(*ps, backlog); | 137 | const char *err = NULL; |
138 | sock_setblocking(ps); | ||
139 | if (listen(*ps, backlog)) | ||
140 | err = sock_listenstrerror(errno); | ||
141 | sock_setnonblocking(ps); | ||
142 | return err; | ||
126 | } | 143 | } |
127 | 144 | ||
128 | /*-------------------------------------------------------------------------*\ | 145 | /*-------------------------------------------------------------------------*\ |
@@ -130,35 +147,40 @@ void sock_listen(p_sock ps, int backlog) | |||
130 | \*-------------------------------------------------------------------------*/ | 147 | \*-------------------------------------------------------------------------*/ |
131 | void sock_shutdown(p_sock ps, int how) | 148 | void sock_shutdown(p_sock ps, int how) |
132 | { | 149 | { |
150 | sock_setblocking(ps); | ||
133 | shutdown(*ps, how); | 151 | shutdown(*ps, how); |
152 | sock_setnonblocking(ps); | ||
134 | } | 153 | } |
135 | 154 | ||
136 | /*-------------------------------------------------------------------------*\ | 155 | /*-------------------------------------------------------------------------*\ |
137 | * Accept with timeout | 156 | * Accept with timeout |
138 | \*-------------------------------------------------------------------------*/ | 157 | \*-------------------------------------------------------------------------*/ |
139 | int sock_accept(p_sock ps, p_sock pa, SA *addr, socklen_t *addr_len, p_tm tm) | 158 | const char *sock_accept(p_sock ps, p_sock pa, SA *addr, |
159 | socklen_t *addr_len, p_tm tm) | ||
140 | { | 160 | { |
141 | t_sock sock = *ps; | 161 | t_sock sock = *ps; |
142 | SA dummy_addr; | 162 | SA dummy_addr; |
143 | socklen_t dummy_len = sizeof(dummy_addr); | 163 | socklen_t dummy_len = sizeof(dummy_addr); |
144 | if (sock == SOCK_INVALID) return IO_CLOSED; | 164 | if (sock == SOCK_INVALID) return io_strerror(IO_CLOSED); |
145 | if (!addr) addr = &dummy_addr; | 165 | if (!addr) addr = &dummy_addr; |
146 | if (!addr_len) addr_len = &dummy_len; | 166 | if (!addr_len) addr_len = &dummy_len; |
147 | for (;;) { | 167 | for (;;) { |
148 | int timeout = tm_getretry(tm); | 168 | int err; |
149 | struct timeval tv; | ||
150 | fd_set fds; | 169 | fd_set fds; |
151 | *pa = accept(sock, addr, addr_len); | 170 | *pa = accept(sock, addr, addr_len); |
152 | if (*pa != SOCK_INVALID) return IO_DONE; | 171 | /* if result is valid, we are done */ |
153 | if (timeout == 0) return IO_TIMEOUT; | 172 | if (*pa != SOCK_INVALID) return NULL; |
154 | tv.tv_sec = timeout / 1000; | 173 | /* find out if we failed for a fatal reason */ |
155 | tv.tv_usec = (timeout % 1000) * 1000; | 174 | if (errno != EWOULDBLOCK && errno != ECONNABORTED) |
175 | return sock_acceptstrerror(errno); | ||
176 | /* call select just to avoid busy-wait. */ | ||
156 | FD_ZERO(&fds); | 177 | FD_ZERO(&fds); |
157 | FD_SET(sock, &fds); | 178 | FD_SET(sock, &fds); |
158 | /* call select just to avoid busy-wait. */ | 179 | do err = sock_select(sock+1, &fds, NULL, NULL, tm_getretry(tm)); |
159 | select(sock+1, &fds, NULL, NULL, timeout >= 0? &tv: NULL); | 180 | while (err < 0 && errno == EINTR); |
181 | if (err == 0) return io_strerror(IO_TIMEOUT); | ||
160 | } | 182 | } |
161 | return IO_TIMEOUT; /* can't get here */ | 183 | return io_strerror(IO_TIMEOUT); /* can't get here */ |
162 | } | 184 | } |
163 | 185 | ||
164 | /*-------------------------------------------------------------------------*\ | 186 | /*-------------------------------------------------------------------------*\ |
@@ -176,18 +198,15 @@ int sock_send(p_sock ps, const char *data, size_t count, size_t *sent, | |||
176 | while (put < 0 && errno == EINTR); | 198 | while (put < 0 && errno == EINTR); |
177 | /* deal with failure */ | 199 | /* deal with failure */ |
178 | if (put <= 0) { | 200 | if (put <= 0) { |
179 | struct timeval tv; | ||
180 | fd_set fds; | 201 | fd_set fds; |
181 | /* in any case, nothing has been sent */ | 202 | /* in any case, nothing has been sent */ |
182 | *sent = 0; | 203 | *sent = 0; |
183 | /* here we know the connection has been closed */ | 204 | /* here we know the connection has been closed */ |
184 | if (errno == EPIPE) return IO_CLOSED; | 205 | if (errno == EPIPE) return IO_CLOSED; |
185 | /* run select to avoid busy wait */ | 206 | /* run select to avoid busy wait */ |
186 | tv.tv_sec = timeout / 1000; | ||
187 | tv.tv_usec = (timeout % 1000) * 1000; | ||
188 | FD_ZERO(&fds); | 207 | FD_ZERO(&fds); |
189 | FD_SET(sock, &fds); | 208 | FD_SET(sock, &fds); |
190 | if (select(sock+1, NULL, &fds, NULL, timeout >= 0? &tv: NULL) <= 0) { | 209 | if (sock_select(sock+1, NULL, &fds, NULL, timeout) <= 0) { |
191 | /* here the call was interrupted. calling again might work */ | 210 | /* here the call was interrupted. calling again might work */ |
192 | if (errno == EINTR) return IO_RETRY; | 211 | if (errno == EINTR) return IO_RETRY; |
193 | /* here there was no data before timeout */ | 212 | /* here there was no data before timeout */ |
@@ -213,15 +232,12 @@ int sock_sendto(p_sock ps, const char *data, size_t count, size_t *sent, | |||
213 | do put = sendto(sock, data, count, 0, addr, addr_len); | 232 | do put = sendto(sock, data, count, 0, addr, addr_len); |
214 | while (put < 0 && errno == EINTR); | 233 | while (put < 0 && errno == EINTR); |
215 | if (put <= 0) { | 234 | if (put <= 0) { |
216 | struct timeval tv; | ||
217 | fd_set fds; | 235 | fd_set fds; |
218 | *sent = 0; | 236 | *sent = 0; |
219 | if (errno == EPIPE) return IO_CLOSED; | 237 | if (errno == EPIPE) return IO_CLOSED; |
220 | tv.tv_sec = timeout / 1000; | ||
221 | tv.tv_usec = (timeout % 1000) * 1000; | ||
222 | FD_ZERO(&fds); | 238 | FD_ZERO(&fds); |
223 | FD_SET(sock, &fds); | 239 | FD_SET(sock, &fds); |
224 | if (select(sock+1, NULL, &fds, NULL, timeout >= 0? &tv: NULL) <= 0) { | 240 | if (sock_select(sock+1, NULL, &fds, NULL, timeout) <= 0) { |
225 | if (errno == EINTR) return IO_RETRY; | 241 | if (errno == EINTR) return IO_RETRY; |
226 | else return IO_TIMEOUT; | 242 | else return IO_TIMEOUT; |
227 | } else return IO_DONE; | 243 | } else return IO_DONE; |
@@ -242,16 +258,13 @@ int sock_recv(p_sock ps, char *data, size_t count, size_t *got, int timeout) | |||
242 | do taken = read(sock, data, count); | 258 | do taken = read(sock, data, count); |
243 | while (taken < 0 && errno == EINTR); | 259 | while (taken < 0 && errno == EINTR); |
244 | if (taken <= 0) { | 260 | if (taken <= 0) { |
245 | struct timeval tv; | ||
246 | fd_set fds; | 261 | fd_set fds; |
247 | int ret; | 262 | int ret; |
248 | *got = 0; | 263 | *got = 0; |
249 | if (taken == 0) return IO_CLOSED; | 264 | if (taken == 0) return IO_CLOSED; |
250 | tv.tv_sec = timeout / 1000; | ||
251 | tv.tv_usec = (timeout % 1000) * 1000; | ||
252 | FD_ZERO(&fds); | 265 | FD_ZERO(&fds); |
253 | FD_SET(sock, &fds); | 266 | FD_SET(sock, &fds); |
254 | ret = select(sock+1, &fds, NULL, NULL, timeout >= 0 ? &tv : NULL); | 267 | ret = sock_select(sock+1, &fds, NULL, NULL, timeout); |
255 | if (ret < 0 && errno == EINTR) return IO_RETRY; | 268 | if (ret < 0 && errno == EINTR) return IO_RETRY; |
256 | if (ret == 0) return IO_TIMEOUT; | 269 | if (ret == 0) return IO_TIMEOUT; |
257 | else return IO_DONE; | 270 | else return IO_DONE; |
@@ -273,16 +286,13 @@ int sock_recvfrom(p_sock ps, char *data, size_t count, size_t *got, | |||
273 | do taken = recvfrom(sock, data, count, 0, addr, addr_len); | 286 | do taken = recvfrom(sock, data, count, 0, addr, addr_len); |
274 | while (taken < 0 && errno == EINTR); | 287 | while (taken < 0 && errno == EINTR); |
275 | if (taken <= 0) { | 288 | if (taken <= 0) { |
276 | struct timeval tv; | ||
277 | fd_set fds; | 289 | fd_set fds; |
278 | int ret; | 290 | int ret; |
279 | *got = 0; | 291 | *got = 0; |
280 | if (taken == 0) return IO_CLOSED; | 292 | if (taken == 0) return IO_CLOSED; |
281 | tv.tv_sec = timeout / 1000; | ||
282 | tv.tv_usec = (timeout % 1000) * 1000; | ||
283 | FD_ZERO(&fds); | 293 | FD_ZERO(&fds); |
284 | FD_SET(sock, &fds); | 294 | FD_SET(sock, &fds); |
285 | ret = select(sock+1, &fds, NULL, NULL, timeout >= 0 ? &tv : NULL); | 295 | ret = sock_select(sock+1, &fds, NULL, NULL, timeout); |
286 | if (ret < 0 && errno == EINTR) return IO_RETRY; | 296 | if (ret < 0 && errno == EINTR) return IO_RETRY; |
287 | if (ret == 0) return IO_TIMEOUT; | 297 | if (ret == 0) return IO_TIMEOUT; |
288 | else return IO_DONE; | 298 | else return IO_DONE; |
@@ -315,53 +325,98 @@ void sock_setnonblocking(p_sock ps) | |||
315 | /*-------------------------------------------------------------------------*\ | 325 | /*-------------------------------------------------------------------------*\ |
316 | * Error translation functions | 326 | * Error translation functions |
317 | \*-------------------------------------------------------------------------*/ | 327 | \*-------------------------------------------------------------------------*/ |
328 | /* return error messages for the known errors reported by gethostbyname */ | ||
318 | const char *sock_hoststrerror(void) | 329 | const char *sock_hoststrerror(void) |
319 | { | 330 | { |
320 | switch (h_errno) { | 331 | switch (h_errno) { |
321 | case HOST_NOT_FOUND: return "host not found"; | 332 | case HOST_NOT_FOUND: return "host not found"; |
322 | case NO_ADDRESS: return "unable to resolve host name"; | 333 | case NO_ADDRESS: return "valid host but no ip found"; |
323 | case NO_RECOVERY: return "name server error"; | 334 | case NO_RECOVERY: return "name server error"; |
324 | case TRY_AGAIN: return "name server unavailable, try again later"; | 335 | case TRY_AGAIN: return "name server unavailable, try again later"; |
325 | default: return "unknown error"; | 336 | default: return "unknown error"; |
326 | } | 337 | } |
327 | } | 338 | } |
328 | 339 | ||
329 | static const char *sock_createstrerror(void) | 340 | /* return error messages for the known errors reported by socket */ |
341 | static const char *sock_createstrerror(int err) | ||
330 | { | 342 | { |
331 | switch (errno) { | 343 | switch (err) { |
344 | case EPROTONOSUPPORT: return "protocol not supported"; | ||
332 | case EACCES: return "access denied"; | 345 | case EACCES: return "access denied"; |
333 | case EMFILE: return "descriptor table is full"; | 346 | case EMFILE: return "process file table is full"; |
334 | case ENFILE: return "too many open files"; | 347 | case ENFILE: return "kernel file table is full"; |
348 | case EINVAL: return "unknown protocol or family"; | ||
335 | case ENOBUFS: return "insuffucient buffer space"; | 349 | case ENOBUFS: return "insuffucient buffer space"; |
336 | default: return "unknown error"; | 350 | default: return "unknown error"; |
337 | } | 351 | } |
338 | } | 352 | } |
339 | 353 | ||
340 | static const char *sock_bindstrerror(void) | 354 | /* return error messages for the known errors reported by accept */ |
355 | static const char *sock_acceptstrerror(int err) | ||
341 | { | 356 | { |
342 | switch (errno) { | 357 | switch (err) { |
358 | case EWOULDBLOCK: return io_strerror(IO_RETRY); | ||
343 | case EBADF: return "invalid descriptor"; | 359 | case EBADF: return "invalid descriptor"; |
344 | case EINVAL: return "socket already bound"; | 360 | case ENOBUFS: case ENOMEM: return "insuffucient buffer space"; |
345 | case EACCES: return "access denied"; | 361 | case ENOTSOCK: return "descriptor not a socket"; |
346 | case ENOTSOCK: return "not a socket descriptor"; | 362 | case EOPNOTSUPP: return "not supported"; |
363 | case EINTR: return "call interrupted"; | ||
364 | case ECONNABORTED: return "connection aborted"; | ||
365 | case EINVAL: return "not listening"; | ||
366 | case EMFILE: return "process file table is full"; | ||
367 | case ENFILE: return "kernel file table is full"; | ||
368 | case EFAULT: return "invalid memory address"; | ||
369 | default: return "unknown error"; | ||
370 | } | ||
371 | } | ||
372 | |||
373 | |||
374 | /* return error messages for the known errors reported by bind */ | ||
375 | static const char *sock_bindstrerror(int err) | ||
376 | { | ||
377 | switch (err) { | ||
378 | case EBADF: return "invalid descriptor"; | ||
379 | case ENOTSOCK: return "descriptor not a socket"; | ||
380 | case EADDRNOTAVAIL: return "address unavailable in local host"; | ||
347 | case EADDRINUSE: return "address already in use"; | 381 | case EADDRINUSE: return "address already in use"; |
348 | case EADDRNOTAVAIL: return "address unavailable"; | 382 | case EINVAL: return "already bound"; |
383 | case EACCES: return "access denied"; | ||
384 | case EFAULT: return "invalid memory address"; | ||
349 | case ENOMEM: return "out of memory"; | 385 | case ENOMEM: return "out of memory"; |
350 | default: return "unknown error"; | 386 | default: return "unknown error"; |
351 | } | 387 | } |
352 | } | 388 | } |
353 | 389 | ||
354 | static const char *sock_connectstrerror(void) | 390 | /* return error messages for the known errors reported by listen */ |
391 | static const char *sock_listenstrerror(int err) | ||
355 | { | 392 | { |
356 | switch (errno) { | 393 | switch (err) { |
394 | case EADDRINUSE: return "local address already in use"; | ||
357 | case EBADF: return "invalid descriptor"; | 395 | case EBADF: return "invalid descriptor"; |
358 | case ENOTSOCK: return "not a socket descriptor"; | 396 | case ENOTSOCK: return "descriptor not a socket"; |
359 | case EADDRNOTAVAIL: return "address not availabe"; | 397 | case EOPNOTSUPP: return "not supported"; |
360 | case ETIMEDOUT: return "connection timed out"; | 398 | default: return "unknown error"; |
399 | } | ||
400 | } | ||
401 | |||
402 | /* return error messages for the known errors reported by connect */ | ||
403 | static const char *sock_connectstrerror(int err) | ||
404 | { | ||
405 | switch (err) { | ||
406 | case EBADF: return "invalid descriptor"; | ||
407 | case EFAULT: return "invalid memory address"; | ||
408 | case ENOTSOCK: return "descriptor not a socket"; | ||
409 | case EADDRNOTAVAIL: return "address not available in local host"; | ||
410 | case EISCONN: return "already connected"; | ||
361 | case ECONNREFUSED: return "connection refused"; | 411 | case ECONNREFUSED: return "connection refused"; |
362 | case EACCES: return "access denied"; | 412 | case ETIMEDOUT: return io_strerror(IO_TIMEOUT); |
363 | case ENETUNREACH: return "network is unreachable"; | 413 | case ENETUNREACH: return "network is unreachable"; |
364 | case EADDRINUSE: return "address already in use"; | 414 | case EADDRINUSE: return "local address already in use"; |
415 | case EINPROGRESS: return "would block"; | ||
416 | case EALREADY: return "connect already in progress"; | ||
417 | case EAGAIN: return "not enough free ports"; | ||
418 | case EAFNOSUPPORT: return "address family not supported"; | ||
419 | case EPERM: return "broadcast not enabled or firewall block"; | ||
365 | default: return "unknown error"; | 420 | default: return "unknown error"; |
366 | } | 421 | } |
367 | } | 422 | } |
diff --git a/src/wsocket.c b/src/wsocket.c index f834503..f33e154 100644 --- a/src/wsocket.c +++ b/src/wsocket.c | |||
@@ -15,9 +15,11 @@ | |||
15 | 15 | ||
16 | #include "socket.h" | 16 | #include "socket.h" |
17 | 17 | ||
18 | static const char *sock_createstrerror(void); | 18 | static const char *sock_createstrerror(int err); |
19 | static const char *sock_bindstrerror(void); | 19 | static const char *sock_bindstrerror(int err); |
20 | static const char *sock_connectstrerror(void); | 20 | static const char *sock_connectstrerror(int err); |
21 | static const char *sock_acceptstrerror(int err); | ||
22 | static const char *sock_listenstrerror(int err); | ||
21 | 23 | ||
22 | /*-------------------------------------------------------------------------*\ | 24 | /*-------------------------------------------------------------------------*\ |
23 | * Initializes module | 25 | * Initializes module |
@@ -36,11 +38,23 @@ int sock_open(void) | |||
36 | } | 38 | } |
37 | 39 | ||
38 | /*-------------------------------------------------------------------------*\ | 40 | /*-------------------------------------------------------------------------*\ |
41 | * Select with int timeout in ms | ||
42 | \*-------------------------------------------------------------------------*/ | ||
43 | int sock_select(int n, fd_set *rfds, fd_set *wfds, fd_set *efds, int timeout) | ||
44 | { | ||
45 | struct timeval tv; | ||
46 | tv.tv_sec = timeout / 1000; | ||
47 | tv.tv_usec = (timeout % 1000) * 1000; | ||
48 | return select(n, rfds, wfds, efds, timeout >= 0? &tv: NULL); | ||
49 | } | ||
50 | |||
51 | /*-------------------------------------------------------------------------*\ | ||
39 | * Close and inutilize socket | 52 | * Close and inutilize socket |
40 | \*-------------------------------------------------------------------------*/ | 53 | \*-------------------------------------------------------------------------*/ |
41 | void sock_destroy(p_sock ps) | 54 | void sock_destroy(p_sock ps) |
42 | { | 55 | { |
43 | if (*ps != SOCK_INVALID) { | 56 | if (*ps != SOCK_INVALID) { |
57 | sock_setblocking(ps); /* close can take a long time on WIN32 */ | ||
44 | closesocket(*ps); | 58 | closesocket(*ps); |
45 | *ps = SOCK_INVALID; | 59 | *ps = SOCK_INVALID; |
46 | } | 60 | } |
@@ -51,7 +65,9 @@ void sock_destroy(p_sock ps) | |||
51 | \*-------------------------------------------------------------------------*/ | 65 | \*-------------------------------------------------------------------------*/ |
52 | void sock_shutdown(p_sock ps, int how) | 66 | void sock_shutdown(p_sock ps, int how) |
53 | { | 67 | { |
68 | sock_setblocking(ps); | ||
54 | shutdown(*ps, how); | 69 | shutdown(*ps, how); |
70 | sock_setnonblocking(ps); | ||
55 | } | 71 | } |
56 | 72 | ||
57 | /*-------------------------------------------------------------------------*\ | 73 | /*-------------------------------------------------------------------------*\ |
@@ -61,10 +77,11 @@ const char *sock_create(p_sock ps, int domain, int type, int protocol) | |||
61 | { | 77 | { |
62 | int val = 1; | 78 | int val = 1; |
63 | t_sock sock = socket(domain, type, protocol); | 79 | t_sock sock = socket(domain, type, protocol); |
64 | if (sock == SOCK_INVALID) return sock_createstrerror(); | 80 | if (sock == SOCK_INVALID) |
81 | return sock_createstrerror(WSAGetLastError()); | ||
65 | *ps = sock; | 82 | *ps = sock; |
66 | sock_setnonblocking(ps); | ||
67 | setsockopt(*ps, SOL_SOCKET, SO_REUSEADDR, (char *) &val, sizeof(val)); | 83 | setsockopt(*ps, SOL_SOCKET, SO_REUSEADDR, (char *) &val, sizeof(val)); |
84 | sock_setnonblocking(ps); | ||
68 | return NULL; | 85 | return NULL; |
69 | } | 86 | } |
70 | 87 | ||
@@ -75,7 +92,6 @@ const char *sock_connect(p_sock ps, SA *addr, socklen_t addr_len, p_tm tm) | |||
75 | { | 92 | { |
76 | t_sock sock = *ps; | 93 | t_sock sock = *ps; |
77 | int err, timeout = tm_getretry(tm); | 94 | int err, timeout = tm_getretry(tm); |
78 | struct timeval tv; | ||
79 | fd_set efds, wfds; | 95 | fd_set efds, wfds; |
80 | /* don't call on closed socket */ | 96 | /* don't call on closed socket */ |
81 | if (sock == SOCK_INVALID) return io_strerror(IO_CLOSED); | 97 | if (sock == SOCK_INVALID) return io_strerror(IO_CLOSED); |
@@ -84,27 +100,24 @@ const char *sock_connect(p_sock ps, SA *addr, socklen_t addr_len, p_tm tm) | |||
84 | /* if no error, we're done */ | 100 | /* if no error, we're done */ |
85 | if (err == 0) return NULL; | 101 | if (err == 0) return NULL; |
86 | /* make sure the system is trying to connect */ | 102 | /* make sure the system is trying to connect */ |
87 | if (WSAGetLastError() != WSAEWOULDBLOCK) return sock_connectstrerror(); | 103 | err = WSAGetLastError(); |
104 | if (err != WSAEWOULDBLOCK) return sock_connectstrerror(err); | ||
88 | /* wait for a timeout or for the system's answer */ | 105 | /* wait for a timeout or for the system's answer */ |
89 | tv.tv_sec = timeout / 1000; | ||
90 | tv.tv_usec = (timeout % 1000) * 1000; | ||
91 | FD_ZERO(&wfds); FD_SET(sock, &wfds); | 106 | FD_ZERO(&wfds); FD_SET(sock, &wfds); |
92 | FD_ZERO(&efds); FD_SET(sock, &efds); | 107 | FD_ZERO(&efds); FD_SET(sock, &efds); |
93 | /* we run select to wait */ | 108 | /* we run select to wait */ |
94 | err = select(0, NULL, &wfds, &efds, timeout >= 0? &tv: NULL); | 109 | err = sock_select(0, NULL, &wfds, &efds, timeout); |
95 | /* if select returned due to an event */ | 110 | /* if select returned due to an event */ |
96 | if (err > 0 ) { | 111 | if (err > 0 ) { |
97 | /* if was in efds, we failed */ | 112 | /* if was in efds, we failed */ |
98 | if (FD_ISSET(sock,&efds) || !FD_ISSET(sock,&wfds)) { | 113 | if (FD_ISSET(sock, &efds)) { |
99 | int why; | 114 | int why, len = sizeof(why); |
100 | int len = sizeof(why); | ||
101 | /* find out why we failed */ | 115 | /* find out why we failed */ |
102 | getsockopt(sock, SOL_SOCKET, SO_ERROR, (char *)&why, &len); | 116 | getsockopt(sock, SOL_SOCKET, SO_ERROR, (char *)&why, &len); |
103 | WSASetLastError(why); | 117 | return sock_connectstrerror(why); |
104 | return sock_connectstrerror(); | 118 | /* otherwise it must be in wfds, so we succeeded */ |
105 | /* if was in wfds, we succeeded */ | ||
106 | } else return NULL; | 119 | } else return NULL; |
107 | /* if nothing happened, we timed out */ | 120 | /* if no event happened, we timed out */ |
108 | } else return io_strerror(IO_TIMEOUT); | 121 | } else return io_strerror(IO_TIMEOUT); |
109 | } | 122 | } |
110 | 123 | ||
@@ -113,44 +126,60 @@ const char *sock_connect(p_sock ps, SA *addr, socklen_t addr_len, p_tm tm) | |||
113 | \*-------------------------------------------------------------------------*/ | 126 | \*-------------------------------------------------------------------------*/ |
114 | const char *sock_bind(p_sock ps, SA *addr, socklen_t addr_len) | 127 | const char *sock_bind(p_sock ps, SA *addr, socklen_t addr_len) |
115 | { | 128 | { |
116 | if (bind(*ps, addr, addr_len) < 0) return sock_bindstrerror(); | 129 | const char *err = NULL; |
117 | else return NULL; | 130 | sock_setblocking(ps); |
131 | if (bind(*ps, addr, addr_len) < 0) | ||
132 | err = sock_bindstrerror(WSAGetLastError()); | ||
133 | sock_setnonblocking(ps); | ||
134 | return err; | ||
118 | } | 135 | } |
119 | 136 | ||
120 | /*-------------------------------------------------------------------------*\ | 137 | /*-------------------------------------------------------------------------*\ |
121 | * | 138 | * |
122 | \*-------------------------------------------------------------------------*/ | 139 | \*-------------------------------------------------------------------------*/ |
123 | void sock_listen(p_sock ps, int backlog) | 140 | const char *sock_listen(p_sock ps, int backlog) |
124 | { | 141 | { |
125 | listen(*ps, backlog); | 142 | const char *err = NULL; |
143 | sock_setblocking(ps); | ||
144 | if (listen(*ps, backlog) < 0) | ||
145 | err = sock_listenstrerror(WSAGetLastError()); | ||
146 | sock_setnonblocking(ps); | ||
147 | return err; | ||
126 | } | 148 | } |
127 | 149 | ||
128 | /*-------------------------------------------------------------------------*\ | 150 | /*-------------------------------------------------------------------------*\ |
129 | * Accept with timeout | 151 | * Accept with timeout |
130 | \*-------------------------------------------------------------------------*/ | 152 | \*-------------------------------------------------------------------------*/ |
131 | int sock_accept(p_sock ps, p_sock pa, SA *addr, socklen_t *addr_len, p_tm tm) | 153 | const char *sock_accept(p_sock ps, p_sock pa, SA *addr, |
154 | socklen_t *addr_len, p_tm tm) | ||
132 | { | 155 | { |
133 | t_sock sock = *ps; | 156 | t_sock sock = *ps; |
134 | SA dummy_addr; | 157 | SA dummy_addr; |
135 | socklen_t dummy_len = sizeof(dummy_addr); | 158 | socklen_t dummy_len = sizeof(dummy_addr); |
136 | if (sock == SOCK_INVALID) return IO_CLOSED; | 159 | if (sock == SOCK_INVALID) return io_strerror(IO_CLOSED); |
137 | if (!addr) addr = &dummy_addr; | 160 | if (!addr) addr = &dummy_addr; |
138 | if (!addr_len) addr_len = &dummy_len; | 161 | if (!addr_len) addr_len = &dummy_len; |
139 | for (;;) { | 162 | for (;;) { |
163 | fd_set rfds; | ||
140 | int timeout = tm_getretry(tm); | 164 | int timeout = tm_getretry(tm); |
141 | struct timeval tv; | 165 | int err; |
142 | fd_set fds; | 166 | /* try to get client socket */ |
143 | *pa = accept(sock, addr, addr_len); | 167 | *pa = accept(sock, addr, addr_len); |
144 | if (*pa != SOCK_INVALID) return IO_DONE; | 168 | /* if return is valid, we are done */ |
145 | if (timeout == 0) return IO_TIMEOUT; | 169 | if (*pa != SOCK_INVALID) return NULL; |
146 | tv.tv_sec = timeout / 1000; | 170 | /* optimization */ |
147 | tv.tv_usec = (timeout % 1000) * 1000; | 171 | if (timeout == 0) return io_strerror(IO_TIMEOUT); |
148 | FD_ZERO(&fds); | 172 | /* otherwise find out why we failed */ |
149 | FD_SET(sock, &fds); | 173 | err = WSAGetLastError(); |
150 | /* call select just to avoid busy-wait. */ | 174 | /* if we failed because there was no connectoin, keep trying*/ |
151 | select(0, &fds, NULL, NULL, timeout >= 0? &tv: NULL); | 175 | if (err != WSAEWOULDBLOCK) return sock_acceptstrerror(err); |
176 | /* call select to avoid busy wait */ | ||
177 | FD_ZERO(&rfds); | ||
178 | FD_SET(sock, &rfds); | ||
179 | err = sock_select(0, &rfds, NULL, NULL, timeout); | ||
180 | if (err <= 0) return io_strerror(IO_TIMEOUT); | ||
152 | } | 181 | } |
153 | return IO_TIMEOUT; /* can't get here */ | 182 | return io_strerror(IO_TIMEOUT); /* can't get here */ |
154 | } | 183 | } |
155 | 184 | ||
156 | /*-------------------------------------------------------------------------*\ | 185 | /*-------------------------------------------------------------------------*\ |
@@ -172,13 +201,10 @@ int sock_send(p_sock ps, const char *data, size_t count, size_t *sent, | |||
172 | *sent = 0; | 201 | *sent = 0; |
173 | /* run select to avoid busy wait */ | 202 | /* run select to avoid busy wait */ |
174 | if (WSAGetLastError() == WSAEWOULDBLOCK) { | 203 | if (WSAGetLastError() == WSAEWOULDBLOCK) { |
175 | struct timeval tv; | ||
176 | fd_set fds; | 204 | fd_set fds; |
177 | tv.tv_sec = timeout / 1000; | ||
178 | tv.tv_usec = (timeout % 1000) * 1000; | ||
179 | FD_ZERO(&fds); | 205 | FD_ZERO(&fds); |
180 | FD_SET(sock, &fds); | 206 | FD_SET(sock, &fds); |
181 | ret = select(0, NULL, &fds, NULL, timeout >= 0 ? &tv : NULL); | 207 | ret = sock_select(0, NULL, &fds, NULL, timeout); |
182 | /* tell the caller to call us again because there is more data */ | 208 | /* tell the caller to call us again because there is more data */ |
183 | if (ret > 0) return IO_DONE; | 209 | if (ret > 0) return IO_DONE; |
184 | /* tell the caller there was no data before timeout */ | 210 | /* tell the caller there was no data before timeout */ |
@@ -211,13 +237,10 @@ int sock_sendto(p_sock ps, const char *data, size_t count, size_t *sent, | |||
211 | *sent = 0; | 237 | *sent = 0; |
212 | /* run select to avoid busy wait */ | 238 | /* run select to avoid busy wait */ |
213 | if (WSAGetLastError() == WSAEWOULDBLOCK) { | 239 | if (WSAGetLastError() == WSAEWOULDBLOCK) { |
214 | struct timeval tv; | ||
215 | fd_set fds; | 240 | fd_set fds; |
216 | tv.tv_sec = timeout / 1000; | ||
217 | tv.tv_usec = (timeout % 1000) * 1000; | ||
218 | FD_ZERO(&fds); | 241 | FD_ZERO(&fds); |
219 | FD_SET(sock, &fds); | 242 | FD_SET(sock, &fds); |
220 | ret = select(0, NULL, &fds, NULL, timeout >= 0 ? &tv : NULL); | 243 | ret = sock_select(0, NULL, &fds, NULL, timeout); |
221 | /* tell the caller to call us again because there is more data */ | 244 | /* tell the caller to call us again because there is more data */ |
222 | if (ret > 0) return IO_DONE; | 245 | if (ret > 0) return IO_DONE; |
223 | /* tell the caller there was no data before timeout */ | 246 | /* tell the caller there was no data before timeout */ |
@@ -241,16 +264,13 @@ int sock_recv(p_sock ps, char *data, size_t count, size_t *got, int timeout) | |||
241 | if (sock == SOCK_INVALID) return IO_CLOSED; | 264 | if (sock == SOCK_INVALID) return IO_CLOSED; |
242 | taken = recv(sock, data, (int) count, 0); | 265 | taken = recv(sock, data, (int) count, 0); |
243 | if (taken <= 0) { | 266 | if (taken <= 0) { |
244 | struct timeval tv; | ||
245 | fd_set fds; | 267 | fd_set fds; |
246 | int ret; | 268 | int ret; |
247 | *got = 0; | 269 | *got = 0; |
248 | if (taken == 0) return IO_CLOSED; | 270 | if (taken == 0) return IO_CLOSED; |
249 | tv.tv_sec = timeout / 1000; | ||
250 | tv.tv_usec = (timeout % 1000) * 1000; | ||
251 | FD_ZERO(&fds); | 271 | FD_ZERO(&fds); |
252 | FD_SET(sock, &fds); | 272 | FD_SET(sock, &fds); |
253 | ret = select(0, &fds, NULL, NULL, timeout >= 0 ? &tv : NULL); | 273 | ret = sock_select(0, &fds, NULL, NULL, timeout); |
254 | if (ret > 0) return IO_DONE; | 274 | if (ret > 0) return IO_DONE; |
255 | else return IO_TIMEOUT; | 275 | else return IO_TIMEOUT; |
256 | } else { | 276 | } else { |
@@ -270,16 +290,13 @@ int sock_recvfrom(p_sock ps, char *data, size_t count, size_t *got, | |||
270 | if (sock == SOCK_INVALID) return IO_CLOSED; | 290 | if (sock == SOCK_INVALID) return IO_CLOSED; |
271 | taken = recvfrom(sock, data, (int) count, 0, addr, addr_len); | 291 | taken = recvfrom(sock, data, (int) count, 0, addr, addr_len); |
272 | if (taken <= 0) { | 292 | if (taken <= 0) { |
273 | struct timeval tv; | ||
274 | fd_set fds; | 293 | fd_set fds; |
275 | int ret; | 294 | int ret; |
276 | *got = 0; | 295 | *got = 0; |
277 | if (taken == 0) return IO_CLOSED; | 296 | if (taken == 0) return IO_CLOSED; |
278 | tv.tv_sec = timeout / 1000; | ||
279 | tv.tv_usec = (timeout % 1000) * 1000; | ||
280 | FD_ZERO(&fds); | 297 | FD_ZERO(&fds); |
281 | FD_SET(sock, &fds); | 298 | FD_SET(sock, &fds); |
282 | ret = select(0, &fds, NULL, NULL, timeout >= 0 ? &tv : NULL); | 299 | ret = sock_select(0, &fds, NULL, NULL, timeout); |
283 | if (ret > 0) return IO_DONE; | 300 | if (ret > 0) return IO_DONE; |
284 | else return IO_TIMEOUT; | 301 | else return IO_TIMEOUT; |
285 | } else { | 302 | } else { |
@@ -309,51 +326,117 @@ void sock_setnonblocking(p_sock ps) | |||
309 | /*-------------------------------------------------------------------------*\ | 326 | /*-------------------------------------------------------------------------*\ |
310 | * Error translation functions | 327 | * Error translation functions |
311 | \*-------------------------------------------------------------------------*/ | 328 | \*-------------------------------------------------------------------------*/ |
329 | /* return error messages for the known errors reported by gethostbyname */ | ||
312 | const char *sock_hoststrerror(void) | 330 | const char *sock_hoststrerror(void) |
313 | { | 331 | { |
314 | switch (WSAGetLastError()) { | 332 | switch (WSAGetLastError()) { |
315 | case HOST_NOT_FOUND: return "host not found"; | 333 | case WSANOTINITIALISED: return "not initialized"; |
316 | case NO_ADDRESS: return "unable to resolve host name"; | 334 | case WSAENETDOWN: return "network is down"; |
317 | case NO_RECOVERY: return "name server error"; | 335 | case WSAHOST_NOT_FOUND: return "host not found"; |
318 | case TRY_AGAIN: return "name server unavailable, try again later."; | 336 | case WSATRY_AGAIN: return "name server unavailable, try again later"; |
337 | case WSANO_RECOVERY: return "name server error"; | ||
338 | case WSANO_DATA: return "host not found"; | ||
339 | case WSAEINPROGRESS: return "another call in progress"; | ||
340 | case WSAEFAULT: return "invalid memory address"; | ||
341 | case WSAEINTR: return "call interrupted"; | ||
319 | default: return "unknown error"; | 342 | default: return "unknown error"; |
320 | } | 343 | } |
321 | } | 344 | } |
322 | 345 | ||
323 | static const char *sock_createstrerror(void) | 346 | /* return error messages for the known errors reported by socket */ |
347 | static const char *sock_createstrerror(int err) | ||
324 | { | 348 | { |
325 | switch (WSAGetLastError()) { | 349 | switch (err) { |
326 | case WSANOTINITIALISED: return "not initialized"; | 350 | case WSANOTINITIALISED: return "not initialized"; |
327 | case WSAENETDOWN: return "network is down"; | 351 | case WSAENETDOWN: return "network is down"; |
352 | case WSAEAFNOSUPPORT: return "address family not supported"; | ||
353 | case WSAEINPROGRESS: return "another call in progress"; | ||
328 | case WSAEMFILE: return "descriptor table is full"; | 354 | case WSAEMFILE: return "descriptor table is full"; |
329 | case WSAENOBUFS: return "insufficient buffer space"; | 355 | case WSAENOBUFS: return "insufficient buffer space"; |
356 | case WSAEPROTONOSUPPORT: return "protocol not supported"; | ||
357 | case WSAEPROTOTYPE: return "wrong protocol type"; | ||
358 | case WSAESOCKTNOSUPPORT: return "socket type not supported by family"; | ||
330 | default: return "unknown error"; | 359 | default: return "unknown error"; |
331 | } | 360 | } |
332 | } | 361 | } |
333 | 362 | ||
334 | static const char *sock_bindstrerror(void) | 363 | /* return error messages for the known errors reported by accept */ |
364 | static const char *sock_acceptstrerror(int err) | ||
335 | { | 365 | { |
336 | switch (WSAGetLastError()) { | 366 | switch (err) { |
337 | case WSANOTINITIALISED: return "not initialized"; | 367 | case WSANOTINITIALISED: return "not initialized"; |
338 | case WSAENETDOWN: return "network is down"; | 368 | case WSAENETDOWN: return "network is down"; |
339 | case WSAEADDRINUSE: return "address already in use"; | 369 | case WSAEFAULT: return "invalid memory address"; |
340 | case WSAEINVAL: return "socket already bound"; | 370 | case WSAEINTR: return "call interrupted"; |
341 | case WSAENOBUFS: return "too many connections"; | 371 | case WSAEINPROGRESS: return "another call in progress"; |
342 | case WSAEFAULT: return "invalid address"; | 372 | case WSAEINVAL: return "not listening"; |
343 | case WSAENOTSOCK: return "not a socket descriptor"; | 373 | case WSAEMFILE: return "descriptor table is full"; |
374 | case WSAENOBUFS: return "insufficient buffer space"; | ||
375 | case WSAENOTSOCK: return "descriptor not a socket"; | ||
376 | case WSAEOPNOTSUPP: return "not supported"; | ||
377 | case WSAEWOULDBLOCK: return "call would block"; | ||
344 | default: return "unknown error"; | 378 | default: return "unknown error"; |
345 | } | 379 | } |
346 | } | 380 | } |
347 | 381 | ||
348 | static const char *sock_connectstrerror(void) | 382 | /* return error messages for the known errors reported by bind */ |
383 | static const char *sock_bindstrerror(int err) | ||
349 | { | 384 | { |
350 | switch (WSAGetLastError()) { | 385 | switch (err) { |
351 | case WSANOTINITIALISED: return "not initialized"; | 386 | case WSANOTINITIALISED: return "not initialized"; |
352 | case WSAENETDOWN: return "network is down"; | 387 | case WSAENETDOWN: return "network is down"; |
388 | case WSAEACCES: return "broadcast not enabled for socket"; | ||
353 | case WSAEADDRINUSE: return "address already in use"; | 389 | case WSAEADDRINUSE: return "address already in use"; |
354 | case WSAEADDRNOTAVAIL: return "address unavailable"; | 390 | case WSAEADDRNOTAVAIL: return "address not available in local host"; |
391 | case WSAEFAULT: return "invalid memory address"; | ||
392 | case WSAEINPROGRESS: return "another call in progress"; | ||
393 | case WSAEINVAL: return "already bound"; | ||
394 | case WSAENOBUFS: return "insuficient buffer space"; | ||
395 | case WSAENOTSOCK: return "descriptor not a socket"; | ||
396 | default: return "unknown error"; | ||
397 | } | ||
398 | |||
399 | } | ||
400 | |||
401 | /* return error messages for the known errors reported by listen */ | ||
402 | static const char *sock_listenstrerror(int err) | ||
403 | { | ||
404 | switch (err) { | ||
405 | case WSANOTINITIALISED: return "not initialized"; | ||
406 | case WSAENETDOWN: return "network is down"; | ||
407 | case WSAEADDRINUSE: return "local address already in use"; | ||
408 | case WSAEINPROGRESS: return "another call in progress"; | ||
409 | case WSAEINVAL: return "not bound"; | ||
410 | case WSAEISCONN: return "already connected"; | ||
411 | case WSAEMFILE: return "descriptor table is full"; | ||
412 | case WSAENOBUFS: return "insuficient buffer space"; | ||
413 | case WSAENOTSOCK: return "descriptor not a socket"; | ||
414 | case WSAEOPNOTSUPP: return "not supported"; | ||
415 | default: return "unknown error"; | ||
416 | } | ||
417 | } | ||
418 | |||
419 | /* return error messages for the known errors reported by connect */ | ||
420 | static const char *sock_connectstrerror(int err) | ||
421 | { | ||
422 | switch (err) { | ||
423 | case WSANOTINITIALISED: return "not initialized"; | ||
424 | case WSAENETDOWN: return "network is down"; | ||
425 | case WSAEADDRINUSE: return "local address already in use"; | ||
426 | case WSAEINTR: return "call interrupted"; | ||
427 | case WSAEINPROGRESS: return "another call in progress"; | ||
428 | case WSAEALREADY: return "connect already in progress"; | ||
429 | case WSAEADDRNOTAVAIL: return "invalid remote address"; | ||
430 | case WSAEAFNOSUPPORT: return "address family not supported"; | ||
355 | case WSAECONNREFUSED: return "connection refused"; | 431 | case WSAECONNREFUSED: return "connection refused"; |
432 | case WSAEFAULT: return "invalid memory address"; | ||
433 | case WSAEINVAL: return "socket is listening"; | ||
434 | case WSAEISCONN: return "socket already connected"; | ||
356 | case WSAENETUNREACH: return "network is unreachable"; | 435 | case WSAENETUNREACH: return "network is unreachable"; |
436 | case WSAENOTSOCK: return "descriptor not a socket"; | ||
437 | case WSAETIMEDOUT: return io_strerror(IO_TIMEOUT); | ||
438 | case WSAEWOULDBLOCK: return "would block"; | ||
439 | case WSAEACCES: return "broadcast not enabled"; | ||
357 | default: return "unknown error"; | 440 | default: return "unknown error"; |
358 | } | 441 | } |
359 | } | 442 | } |
diff --git a/test/testclnt.lua b/test/testclnt.lua index 6dccd0c..beb0157 100644 --- a/test/testclnt.lua +++ b/test/testclnt.lua | |||
@@ -117,7 +117,6 @@ function test_mixed(len) | |||
117 | local p3 = "raw " .. string.rep("z", inter) .. "bytes" | 117 | local p3 = "raw " .. string.rep("z", inter) .. "bytes" |
118 | local p4 = "end" .. string.rep("w", inter) .. "bytes" | 118 | local p4 = "end" .. string.rep("w", inter) .. "bytes" |
119 | local bp1, bp2, bp3, bp4 | 119 | local bp1, bp2, bp3, bp4 |
120 | pass(len .. " byte(s) patterns") | ||
121 | remote (string.format("str = data:receive(%d)", | 120 | remote (string.format("str = data:receive(%d)", |
122 | string.len(p1)+string.len(p2)+string.len(p3)+string.len(p4))) | 121 | string.len(p1)+string.len(p2)+string.len(p3)+string.len(p4))) |
123 | sent, err = data:send(p1, p2, p3, p4) | 122 | sent, err = data:send(p1, p2, p3, p4) |
@@ -137,7 +136,6 @@ function test_asciiline(len) | |||
137 | str = string.rep("x", math.mod(len, 10)) | 136 | str = string.rep("x", math.mod(len, 10)) |
138 | str10 = string.rep("aZb.c#dAe?", math.floor(len/10)) | 137 | str10 = string.rep("aZb.c#dAe?", math.floor(len/10)) |
139 | str = str .. str10 | 138 | str = str .. str10 |
140 | pass(len .. " byte(s) line") | ||
141 | remote "str = data:receive()" | 139 | remote "str = data:receive()" |
142 | sent, err = data:send(str, "\n") | 140 | sent, err = data:send(str, "\n") |
143 | if err then fail(err) end | 141 | if err then fail(err) end |
@@ -156,7 +154,6 @@ function test_rawline(len) | |||
156 | str10 = string.rep(string.char(120,21,77,4,5,0,7,36,44,100), | 154 | str10 = string.rep(string.char(120,21,77,4,5,0,7,36,44,100), |
157 | math.floor(len/10)) | 155 | math.floor(len/10)) |
158 | str = str .. str10 | 156 | str = str .. str10 |
159 | pass(len .. " byte(s) line") | ||
160 | remote "str = data:receive()" | 157 | remote "str = data:receive()" |
161 | sent, err = data:send(str, "\n") | 158 | sent, err = data:send(str, "\n") |
162 | if err then fail(err) end | 159 | if err then fail(err) end |
@@ -174,7 +171,6 @@ function test_raw(len) | |||
174 | local s1, s2, back, err | 171 | local s1, s2, back, err |
175 | s1 = string.rep("x", half) | 172 | s1 = string.rep("x", half) |
176 | s2 = string.rep("y", len-half) | 173 | s2 = string.rep("y", len-half) |
177 | pass(len .. " byte(s) block") | ||
178 | remote (string.format("str = data:receive(%d)", len)) | 174 | remote (string.format("str = data:receive(%d)", len)) |
179 | sent, err = data:send(s1) | 175 | sent, err = data:send(s1) |
180 | if err then fail(err) end | 176 | if err then fail(err) end |
@@ -271,12 +267,10 @@ end | |||
271 | function empty_connect() | 267 | function empty_connect() |
272 | reconnect() | 268 | reconnect() |
273 | if data then data:close() data = nil end | 269 | if data then data:close() data = nil end |
274 | print("before remote") | ||
275 | remote [[ | 270 | remote [[ |
276 | if data then data:close() data = nil end | 271 | if data then data:close() data = nil end |
277 | data = server:accept() | 272 | data = server:accept() |
278 | ]] | 273 | ]] |
279 | print("after remote") | ||
280 | data, err = socket.connect("", port) | 274 | data, err = socket.connect("", port) |
281 | if not data then | 275 | if not data then |
282 | pass("ok") | 276 | pass("ok") |
@@ -286,7 +280,7 @@ end | |||
286 | 280 | ||
287 | ------------------------------------------------------------------------ | 281 | ------------------------------------------------------------------------ |
288 | function isclosed(c) | 282 | function isclosed(c) |
289 | return c:fd() == -1 or c:fd() == (2^32-1) | 283 | return c:getfd() == -1 or c:getfd() == (2^32-1) |
290 | end | 284 | end |
291 | 285 | ||
292 | function active_close() | 286 | function active_close() |
@@ -354,13 +348,14 @@ end | |||
354 | 348 | ||
355 | ------------------------------------------------------------------------ | 349 | ------------------------------------------------------------------------ |
356 | function accept_timeout() | 350 | function accept_timeout() |
351 | io.write("accept with timeout (if it hangs, it failed): ") | ||
357 | local s, e = socket.bind("*", 0, 0) | 352 | local s, e = socket.bind("*", 0, 0) |
358 | assert(s, e) | 353 | assert(s, e) |
359 | local t = socket.time() | 354 | local t = socket.time() |
360 | s:settimeout(1) | 355 | s:settimeout(1) |
361 | local c, e = s:accept() | 356 | local c, e = s:accept() |
362 | assert(not c, "should not accept") | 357 | assert(not c, "should not accept") |
363 | assert(e == "timeout", "wrong error message") | 358 | assert(e == "timeout", string.format("wrong error message (%s)", e)) |
364 | t = socket.time() - t | 359 | t = socket.time() - t |
365 | assert(t < 2, string.format("took to long to give up (%gs)", t)) | 360 | assert(t < 2, string.format("took to long to give up (%gs)", t)) |
366 | s:close() | 361 | s:close() |
@@ -369,22 +364,51 @@ end | |||
369 | 364 | ||
370 | ------------------------------------------------------------------------ | 365 | ------------------------------------------------------------------------ |
371 | function connect_timeout() | 366 | function connect_timeout() |
367 | io.write("connect with timeout (if it hangs, it failed): ") | ||
372 | local t = socket.time() | 368 | local t = socket.time() |
373 | local c, e = socket.tcp() | 369 | local c, e = socket.tcp() |
374 | assert(c, e) | 370 | assert(c, e) |
375 | c:settimeout(0.1) | 371 | c:settimeout(0.1) |
376 | local r, e = c:connect("ibere.tecgraf.puc-rio.br", 80) | 372 | local r, e = c:connect("ibere.tecgraf.puc-rio.br", 80) |
377 | if r or e ~= "timeout" then | 373 | assert(not r, "should not connect") |
378 | print("wrong error message (this test is flaky anyways)") | 374 | assert(e == "timeout", e) |
379 | end | 375 | assert(socket.time() - t < 2, "took to long to give up") |
380 | if socket.time() - t > 1 then | ||
381 | print("took to long to give up") | ||
382 | end | ||
383 | print("whatever") | ||
384 | c:close() | 376 | c:close() |
385 | end | 377 | end |
386 | 378 | ||
387 | ------------------------------------------------------------------------ | 379 | ------------------------------------------------------------------------ |
380 | function accept_errors() | ||
381 | io.write("not listenning: ") | ||
382 | local d, e = socket.bind("*", 0) | ||
383 | assert(d, e); | ||
384 | local c, e = socket.tcp(); | ||
385 | assert(c, e); | ||
386 | d:setfd(c:getfd()) | ||
387 | local r, e = d:accept() | ||
388 | assert(not r and e == "not listening", e) | ||
389 | print("ok") | ||
390 | io.write("not supported: ") | ||
391 | local c, e = socket.udp() | ||
392 | assert(c, e); | ||
393 | d:setfd(c:getfd()) | ||
394 | local r, e = d:accept() | ||
395 | assert(not r and e == "not supported", e) | ||
396 | print("ok") | ||
397 | end | ||
398 | |||
399 | ------------------------------------------------------------------------ | ||
400 | function connect_errors() | ||
401 | io.write("connection refused: ") | ||
402 | local c, e = socket.connect("localhost", 1); | ||
403 | assert(not c and e == "connection refused", e) | ||
404 | print("ok") | ||
405 | io.write("host not found: ") | ||
406 | local c, e = socket.connect("not.exist.com", 1); | ||
407 | assert(not c and e == "host not found", e) | ||
408 | print("ok") | ||
409 | end | ||
410 | |||
411 | ------------------------------------------------------------------------ | ||
388 | function rebind_test() | 412 | function rebind_test() |
389 | local c = socket.bind("localhost", 0) | 413 | local c = socket.bind("localhost", 0) |
390 | local i, p = c:getsockname() | 414 | local i, p = c:getsockname() |
@@ -400,40 +424,44 @@ end | |||
400 | ------------------------------------------------------------------------ | 424 | ------------------------------------------------------------------------ |
401 | test("method registration") | 425 | test("method registration") |
402 | test_methods(socket.tcp(), { | 426 | test_methods(socket.tcp(), { |
403 | "connect", | 427 | "accept", |
404 | "send", | ||
405 | "receive", | ||
406 | "bind", | 428 | "bind", |
407 | "accept", | 429 | "close", |
408 | "setpeername", | 430 | "connect", |
409 | "setsockname", | ||
410 | "getpeername", | 431 | "getpeername", |
411 | "getsockname", | 432 | "getsockname", |
433 | "listen", | ||
434 | "receive", | ||
435 | "send", | ||
412 | "setoption", | 436 | "setoption", |
437 | "setpeername", | ||
438 | "setsockname", | ||
413 | "settimeout", | 439 | "settimeout", |
414 | "shutdown", | 440 | "shutdown", |
415 | "close", | ||
416 | }) | 441 | }) |
442 | |||
417 | test_methods(socket.udp(), { | 443 | test_methods(socket.udp(), { |
444 | "close", | ||
418 | "getpeername", | 445 | "getpeername", |
419 | "getsockname", | 446 | "getsockname", |
420 | "setsockname", | ||
421 | "setpeername", | ||
422 | "send", | ||
423 | "sendto", | ||
424 | "receive", | 447 | "receive", |
425 | "receivefrom", | 448 | "receivefrom", |
449 | "send", | ||
450 | "sendto", | ||
426 | "setoption", | 451 | "setoption", |
452 | "setpeername", | ||
453 | "setsockname", | ||
427 | "settimeout", | 454 | "settimeout", |
428 | "shutdown", | 455 | "shutdown", |
429 | "close", | ||
430 | }) | 456 | }) |
431 | 457 | ||
432 | test("select function") | 458 | test("select function") |
433 | test_selectbugs() | 459 | test_selectbugs() |
434 | 460 | ||
435 | test("empty host connect: ") | 461 | test("connect function") |
462 | connect_timeout() | ||
436 | empty_connect() | 463 | empty_connect() |
464 | connect_errors() | ||
437 | 465 | ||
438 | test("rebinding: ") | 466 | test("rebinding: ") |
439 | rebind_test() | 467 | rebind_test() |
@@ -444,11 +472,10 @@ active_close() | |||
444 | test("closed connection detection: ") | 472 | test("closed connection detection: ") |
445 | test_closed() | 473 | test_closed() |
446 | 474 | ||
447 | test("accept with timeout (if it hangs, it failed:)") | 475 | test("accept function: ") |
448 | accept_timeout() | 476 | accept_timeout() |
477 | accept_errors() | ||
449 | 478 | ||
450 | test("connect with timeout (if it hangs, it failed:)") | ||
451 | connect_timeout() | ||
452 | 479 | ||
453 | test("mixed patterns") | 480 | test("mixed patterns") |
454 | test_mixed(1) | 481 | test_mixed(1) |