moonbridge: moonbridge

moonbridge

annotate moonbridge_io.c @ 306:25f40c8d5082

Added tag v1.0.2 for changeset e4e1ad8d5ab8

author	jbe
date	Sun Nov 19 02:21:26 2017 +0100 (2017-11-19)
parents	e4e1ad8d5ab8
children	930a43a0099a

rev	line source
jbe@209	1
jbe@209	2 #ifndef __has_include
jbe@209	3 #define __has_include(x) 0
jbe@209	4 #endif
jbe@79	5
jbe@79	6 #include <stdlib.h>
jbe@135	7 #include <stdint.h>
jbe@135	8 #include <string.h>
jbe@135	9 #include <errno.h>
jbe@79	10 #include <unistd.h>
jbe@135	11 #include <signal.h>
jbe@135	12 #include <fcntl.h>
jbe@135	13 #include <sys/stat.h>
jbe@79	14 #include <sys/socket.h>
jbe@111	15 #include <sys/un.h>
jbe@95	16 #include <netinet/in.h>
jbe@95	17 #include <netinet/tcp.h>
jbe@135	18 #include <sys/select.h>
jbe@136	19 #include <time.h>
jbe@135	20 #include <netdb.h>
jbe@107	21 #include <arpa/inet.h>
jbe@205	22 #include <sys/types.h>
jbe@205	23 #include <sys/wait.h>
jbe@209	24 #if defined(__linux__) \|\| __has_include(<bsd/unistd.h>)
jbe@209	25 #include <bsd/unistd.h>
jbe@209	26 #endif
jbe@211	27
jbe@243	28 #ifdef MOONBR_IO_USE_TLS
jbe@243	29 #include <tls.h>
jbe@243	30 #endif
jbe@243	31
jbe@79	32 #include <lua.h>
jbe@79	33 #include <lauxlib.h>
jbe@79	34 #include <lualib.h>
jbe@79	35
jbe@149	36 #include <assert.h>
jbe@149	37
jbe@80	38 #define MOONBR_IO_MAXSTRERRORLEN 80
jbe@85	39 #define MOONBR_IO_READBUFLEN 4096
jbe@80	40 #define MOONBR_IO_WRITEBUFLEN 4096
jbe@80	41
jbe@108	42 #define MOONBR_IO_LISTEN_BACKLOG 1024
jbe@108	43
jbe@213	44 #define MOONBR_IO_STRERROR_R_MSG "Error detail unavailable due to noncompliant strerror_r() implementation"
jbe@257	45 #define moonbr_io_prepare_errmsg() \
jbe@213	46 char errmsg[MOONBR_IO_MAXSTRERRORLEN] = MOONBR_IO_STRERROR_R_MSG; \
jbe@80	47 strerror_r(errno, errmsg, MOONBR_IO_MAXSTRERRORLEN)
jbe@257	48 #define moonbr_io_return_prepared_errmsg() \
jbe@257	49 lua_pushnil(L); \
jbe@257	50 lua_pushstring(L, errmsg); \
jbe@257	51 return 2
jbe@257	52 #define moonbr_io_return_errmsg() \
jbe@257	53 do { \
jbe@257	54 moonbr_io_prepare_errmsg(); \
jbe@257	55 moonbr_io_return_prepared_errmsg(); \
jbe@257	56 } while (0)
jbe@80	57
jbe@79	58 #define MOONBR_IO_HANDLE_MT_REGKEY "moonbridge_io_handle"
jbe@79	59 #define MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY "moonbridge_io_handle_public"
jbe@108	60 #define MOONBR_IO_LISTENER_MT_REGKEY "moonbridge_io_listener"
jbe@205	61 #define MOONBR_IO_CHILD_MT_REGKEY "moonbridge_io_child"
jbe@205	62 #define MOONBR_IO_CHILD_PT_REGKEY "moonbridge_io_child_pt"
jbe@79	63
jbe@243	64 #ifdef MOONBR_IO_USE_TLS
jbe@258	65
jbe@243	66 #define MOONBR_IO_TLSCONF_MT_REGKEY "moonbridge_io_tlsconf"
jbe@252	67
jbe@252	68 typedef struct {
jbe@252	69 struct tls_config *config;
jbe@252	70 int server;
jbe@252	71 } moonbr_io_tlsconf_t;
jbe@258	72
jbe@258	73 #endif /* MOONBR_IO_USE_TLS */
jbe@243	74
jbe@79	75 typedef struct {
jbe@79	76 int fd;
jbe@107	77 int issock;
jbe@107	78 sa_family_t addrfam;
jbe@94	79 int finished;
jbe@94	80 int closed;
jbe@81	81 int nonblocking;
jbe@95	82 int nopush;
jbe@85	83 int readerr;
jbe@105	84 int readbufin;
jbe@105	85 int readbufout;
jbe@81	86 int writeerr;
jbe@83	87 size_t writeleft;
jbe@132	88 size_t flushedleft;
jbe@83	89 #if LUA_VERSION_NUM >= 503
jbe@83	90 lua_Integer writeqin;
jbe@83	91 lua_Integer writeqout;
jbe@83	92 #else
jbe@83	93 int writeqin;
jbe@83	94 int writeqout;
jbe@83	95 #endif
jbe@83	96 size_t writeqoff;
jbe@103	97 int writebufin;
jbe@103	98 int writebufout;
jbe@85	99 char readbuf[MOONBR_IO_READBUFLEN];
jbe@80	100 char writebuf[MOONBR_IO_WRITEBUFLEN];
jbe@246	101 #ifdef MOONBR_IO_USE_TLS
jbe@250	102 struct tls *tls;
jbe@250	103 struct tls *servertls;
jbe@250	104 int tlshandshake;
jbe@261	105 int tlsclosing;
jbe@246	106 #endif
jbe@79	107 } moonbr_io_handle_t;
jbe@79	108
jbe@108	109 typedef struct {
jbe@108	110 int fd;
jbe@118	111 sa_family_t addrfam;
jbe@108	112 int nonblocking;
jbe@108	113 } moonbr_io_listener_t;
jbe@108	114
jbe@205	115 typedef struct {
jbe@205	116 pid_t pid;
jbe@303	117 int status;
jbe@303	118 int status_valid;
jbe@205	119 } moonbr_io_child_t;
jbe@205	120
jbe@293	121 volatile sig_atomic_t moonbr_io_sigterm_flag = 0;
jbe@297	122 volatile sig_atomic_t moonbr_io_sigchld_flag = 0;
jbe@284	123
jbe@148	124 static int moonbr_io_yield(lua_State *L) {
jbe@149	125 return lua_yield(L, lua_gettop(L));
jbe@140	126 }
jbe@140	127
jbe@140	128 #if LUA_VERSION_NUM >= 503
jbe@140	129 static int moonbr_io_cont_returnall(lua_State *L, int status, lua_KContext ctx) {
jbe@140	130 #else
jbe@140	131 static int moonbr_io_cont_returnall(lua_State *L) {
jbe@140	132 #endif
jbe@140	133 return lua_gettop(L);
jbe@140	134 }
jbe@140	135
jbe@145	136 #define moonbr_io_yield_wrapper(yieldfunc, callfunc) \
jbe@145	137 static int yieldfunc(lua_State *L) { \
jbe@145	138 int args; \
jbe@145	139 lua_pushcfunction(L, callfunc); \
jbe@145	140 lua_insert(L, 1); \
jbe@145	141 args = lua_gettop(L); \
jbe@148	142 lua_pushcfunction(L, moonbr_io_yield); \
jbe@145	143 lua_insert(L, 3); \
jbe@145	144 lua_callk(L, args, LUA_MULTRET, 0, moonbr_io_cont_returnall); \
jbe@145	145 return lua_gettop(L); \
jbe@145	146 }
jbe@145	147
jbe@256	148 static int moonbr_io_handle_set_nonblocking(lua_State L, moonbr_io_handle_t handle, int nonblocking) {
jbe@95	149 int flags;
jbe@256	150 if (handle->nonblocking == nonblocking) return 0;
jbe@95	151 flags = fcntl(handle->fd, F_GETFL, 0);
jbe@256	152 if (flags == -1) return -1;
jbe@95	153 if (nonblocking) flags \|= O_NONBLOCK;
jbe@95	154 else flags &= ~O_NONBLOCK;
jbe@256	155 if (fcntl(handle->fd, F_SETFL, flags) == -1) return -1;
jbe@95	156 handle->nonblocking = nonblocking;
jbe@256	157 return 0;
jbe@81	158 }
jbe@81	159
jbe@256	160 static int moonbr_io_handle_set_linger(lua_State L, moonbr_io_handle_t handle, int timeout) {
jbe@87	161 struct linger lingerval = { 0, };
jbe@256	162 if (!handle->issock) return 0;
jbe@87	163 if (timeout >= 0) {
jbe@87	164 lingerval.l_onoff = 1;
jbe@87	165 lingerval.l_linger = timeout;
jbe@87	166 }
jbe@256	167 if (setsockopt(handle->fd, SOL_SOCKET, SO_LINGER, &lingerval, sizeof(lingerval))) return -1;
jbe@256	168 return 0;
jbe@87	169 }
jbe@87	170
jbe@256	171 static inline int moonbr_io_handle_set_nopush(lua_State L, moonbr_io_handle_t handle, int nopush) {
jbe@96	172 #if defined(TCP_NOPUSH) \|\| defined(TCP_CORK)
jbe@107	173 if (
jbe@107	174 !(handle->addrfam == AF_INET6 \|\| handle->addrfam == AF_INET) \|\|
jbe@107	175 handle->nopush == nopush
jbe@256	176 ) return 0;
jbe@96	177 #if defined(TCP_NOPUSH)
jbe@256	178 if (setsockopt(handle->fd, IPPROTO_TCP, TCP_NOPUSH, &nopush, sizeof(nopush))) return -1;
jbe@104	179 #elif defined(TCP_CORK)
jbe@256	180 if (setsockopt(handle->fd, IPPROTO_TCP, TCP_CORK, &nopush, sizeof(nopush))) return -1;
jbe@251	181 #endif
jbe@95	182 handle->nopush = nopush;
jbe@96	183 #else
jbe@96	184 #warning Neither TCP_NOPUSH nor TCP_CORK is available
jbe@96	185 #endif
jbe@256	186 return 0;
jbe@95	187 }
jbe@95	188
jbe@86	189 static int moonbr_io_read_impl(lua_State *L, int nonblocking, int drain) {
jbe@85	190 moonbr_io_handle_t *handle;
jbe@85	191 lua_Integer maxread;
jbe@85	192 const char *terminatorstr;
jbe@210	193 size_t terminatorlen;
jbe@210	194 char terminator = 0; /* initialize to avoid compiler warning */
jbe@85	195 luaL_Buffer luabuf;
jbe@85	196 size_t luabufcnt = 0;
jbe@105	197 int remaining;
jbe@85	198 char *terminatorpos;
jbe@103	199 ssize_t bytesread;
jbe@85	200 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@153	201 maxread = luaL_optinteger(L, 2, -1);
jbe@85	202 terminatorstr = luaL_optlstring(L, 3, "", &terminatorlen);
jbe@85	203 if (terminatorlen) {
jbe@85	204 luaL_argcheck(L, terminatorlen == 1, 3, "single byte expected");
jbe@85	205 terminator = terminatorstr[0];
jbe@85	206 }
jbe@86	207 lua_settop(L, 1); /* return handle on drain, terminator string may be garbage collected */
jbe@94	208 if (handle->closed) luaL_error(L, "Attempt to read from a closed I/O handle");
jbe@85	209 if (handle->readerr) {
jbe@85	210 lua_pushnil(L);
jbe@85	211 lua_pushliteral(L, "Previous read error");
jbe@85	212 return 2;
jbe@85	213 }
jbe@143	214 if (handle->fd < 0) {
jbe@143	215 /* fake EOF to simulate shutdown */
jbe@143	216 if (!drain) lua_pushliteral(L, "");
jbe@143	217 else lua_pushinteger(L, 0);
jbe@143	218 lua_pushliteral(L, "eof");
jbe@143	219 return 2;
jbe@143	220 }
jbe@105	221 handle->readerr = 1;
jbe@257	222 if (moonbr_io_handle_set_nonblocking(L, handle, nonblocking)) moonbr_io_return_errmsg();
jbe@86	223 if (!drain) luaL_buffinit(L, &luabuf);
jbe@85	224 while (1) {
jbe@105	225 remaining = -1;
jbe@142	226 terminatorpos = NULL;
jbe@105	227 if (
jbe@153	228 maxread >= 0 &&
jbe@143	229 handle->readbufin - handle->readbufout >= (size_t)maxread - luabufcnt
jbe@105	230 ) {
jbe@142	231 remaining = (size_t)maxread - luabufcnt;
jbe@193	232 if (terminatorlen) {
jbe@193	233 terminatorpos = memchr(
jbe@193	234 handle->readbuf + handle->readbufout,
jbe@193	235 terminator,
jbe@193	236 remaining
jbe@193	237 );
jbe@193	238 }
jbe@85	239 } else if (terminatorlen) {
jbe@105	240 terminatorpos = memchr(
jbe@105	241 handle->readbuf + handle->readbufout,
jbe@105	242 terminator,
jbe@105	243 handle->readbufin - handle->readbufout
jbe@105	244 );
jbe@85	245 }
jbe@142	246 if (terminatorpos) remaining = 1 + (
jbe@142	247 terminatorpos - (handle->readbuf + handle->readbufout)
jbe@142	248 );
jbe@105	249 if (remaining >= 0) {
jbe@86	250 if (!drain) {
jbe@105	251 luaL_addlstring(
jbe@105	252 &luabuf,
jbe@105	253 handle->readbuf + handle->readbufout,
jbe@105	254 remaining
jbe@105	255 );
jbe@86	256 luaL_pushresult(&luabuf);
jbe@90	257 } else {
jbe@105	258 lua_pushinteger(L, luabufcnt + remaining);
jbe@86	259 }
jbe@143	260 if (terminatorpos) lua_pushliteral(L, "term");
jbe@143	261 else lua_pushliteral(L, "maxlen");
jbe@105	262 handle->readbufout += remaining;
jbe@105	263 if (handle->readbufout == handle->readbufin) {
jbe@105	264 handle->readbufin = 0;
jbe@143	265 handle->readbufout = 0;
jbe@105	266 }
jbe@104	267 handle->readerr = 0;
jbe@143	268 return 2;
jbe@85	269 }
jbe@105	270 if (!drain) luaL_addlstring(
jbe@105	271 &luabuf,
jbe@105	272 handle->readbuf + handle->readbufout,
jbe@105	273 handle->readbufin - handle->readbufout
jbe@105	274 );
jbe@105	275 luabufcnt += handle->readbufin - handle->readbufout;
jbe@143	276 handle->readbufout = 0;
jbe@249	277 #ifdef MOONBR_IO_USE_TLS
jbe@249	278 if (handle->tls) {
jbe@250	279 do {
jbe@250	280 if (!handle->tlshandshake) {
jbe@250	281 do bytesread = tls_handshake(handle->tls);
jbe@250	282 while (!nonblocking && (bytesread == TLS_WANT_POLLIN \|\| bytesread == TLS_WANT_POLLOUT));
jbe@250	283 if (bytesread == TLS_WANT_POLLIN \|\| bytesread == TLS_WANT_POLLOUT) {
jbe@250	284 handle->tlshandshake = bytesread;
jbe@250	285 errno = EAGAIN;
jbe@250	286 break;
jbe@250	287 }
jbe@250	288 if (bytesread < 0) {
jbe@250	289 lua_pushnil(L);
jbe@250	290 lua_pushstring(L, tls_error(handle->tls));
jbe@250	291 return 2;
jbe@250	292 }
jbe@250	293 handle->tlshandshake = 1;
jbe@250	294 }
jbe@250	295 do bytesread = tls_read(handle->tls, handle->readbuf, MOONBR_IO_READBUFLEN);
jbe@250	296 while (!nonblocking && (bytesread == TLS_WANT_POLLIN \|\| bytesread == TLS_WANT_POLLOUT));
jbe@250	297 if (bytesread == TLS_WANT_POLLIN \|\| bytesread == TLS_WANT_POLLOUT) {
jbe@250	298 errno = EAGAIN;
jbe@250	299 break;
jbe@250	300 }
jbe@250	301 if (bytesread < 0) {
jbe@250	302 lua_pushnil(L);
jbe@250	303 lua_pushstring(L, tls_error(handle->tls));
jbe@250	304 return 2;
jbe@250	305 }
jbe@250	306 } while (0);
jbe@249	307 }
jbe@249	308 else
jbe@249	309 #endif
jbe@249	310 do bytesread = read(handle->fd, handle->readbuf, MOONBR_IO_READBUFLEN);
jbe@249	311 while (bytesread < 0 && (errno == EINTR));
jbe@105	312 if (
jbe@105	313 bytesread == 0 \|\| (
jbe@105	314 nonblocking &&
jbe@105	315 bytesread < 0 && (errno == EAGAIN \|\| errno == EWOULDBLOCK)
jbe@105	316 )
jbe@105	317 ) {
jbe@105	318 handle->readbufin = 0;
jbe@143	319 if (!drain) luaL_pushresult(&luabuf);
jbe@143	320 else lua_pushinteger(L, luabufcnt);
jbe@143	321 if (bytesread == 0) lua_pushliteral(L, "eof");
jbe@143	322 else lua_pushliteral(L, "block");
jbe@105	323 handle->readerr = 0;
jbe@143	324 return 2;
jbe@105	325 }
jbe@257	326 if (bytesread < 0) moonbr_io_return_errmsg();
jbe@105	327 handle->readbufin = bytesread;
jbe@86	328 }
jbe@85	329 }
jbe@85	330
jbe@85	331 static int moonbr_io_read(lua_State *L) {
jbe@86	332 return moonbr_io_read_impl(L, 0, 0);
jbe@85	333 }
jbe@85	334
jbe@85	335 static int moonbr_io_read_nb(lua_State *L) {
jbe@86	336 return moonbr_io_read_impl(L, 1, 0);
jbe@86	337 }
jbe@86	338
jbe@86	339 static int moonbr_io_drain(lua_State *L) {
jbe@86	340 return moonbr_io_read_impl(L, 0, 1);
jbe@86	341 }
jbe@86	342
jbe@86	343 static int moonbr_io_drain_nb(lua_State *L) {
jbe@86	344 return moonbr_io_read_impl(L, 1, 1);
jbe@85	345 }
jbe@85	346
jbe@140	347 #if LUA_VERSION_NUM >= 503
jbe@140	348 static int moonbr_io_read_cont(lua_State *L, int status, lua_KContext ctx) {
jbe@140	349 #else
jbe@140	350 static int moonbr_io_read_cont(lua_State *L) {
jbe@140	351 #endif
jbe@143	352 lua_Integer remaining;
jbe@143	353 size_t len;
jbe@140	354 #if !(LUA_VERSION_NUM >= 503)
jbe@140	355 int ctx = 0;
jbe@140	356 lua_getctx(L, &ctx);
jbe@140	357 #endif
jbe@140	358 remaining = lua_tointeger(L, 3);
jbe@140	359 while (1) {
jbe@140	360 lua_pushcfunction(L, moonbr_io_read_nb);
jbe@140	361 lua_pushvalue(L, 1);
jbe@140	362 lua_pushvalue(L, 3);
jbe@140	363 lua_pushvalue(L, 4);
jbe@140	364 lua_call(L, 3, 2);
jbe@143	365 if (lua_isnil(L, -2)) return 2;
jbe@143	366 lua_insert(L, -2);
jbe@143	367 len = lua_rawlen(L, -1);
jbe@143	368 if (ctx == 0) {
jbe@143	369 lua_replace(L, 5);
jbe@143	370 ctx = 1;
jbe@143	371 } else if (ctx == 1) {
jbe@143	372 lua_pushvalue(L, 5);
jbe@143	373 lua_newtable(L);
jbe@143	374 lua_replace(L, 5);
jbe@304	375 lua_rawseti(L, 5, 1);
jbe@143	376 lua_rawseti(L, 5, 2);
jbe@143	377 ctx = 2;
jbe@140	378 } else {
jbe@143	379 lua_rawseti(L, 5, lua_rawlen(L, 5) + 1);
jbe@140	380 }
jbe@144	381 if (strcmp(lua_tostring(L, -1), "block") != 0) break;
jbe@144	382 lua_pop(L, 1);
jbe@144	383 if (remaining >= 0 && len) {
jbe@143	384 remaining -= len;
jbe@143	385 lua_pushinteger(L, remaining);
jbe@143	386 lua_replace(L, 3);
jbe@143	387 }
jbe@140	388 lua_pushvalue(L, 2);
jbe@152	389 lua_callk(L, 0, 0, ctx, moonbr_io_read_cont);
jbe@140	390 }
jbe@305	391 if (ctx != 1) {
jbe@140	392 luaL_Buffer buf;
jbe@140	393 lua_Integer i, chunkcount;
jbe@140	394 chunkcount = lua_rawlen(L, 5);
jbe@140	395 luaL_buffinit(L, &buf);
jbe@140	396 for (i=1; i<=chunkcount && i>0; i++) {
jbe@140	397 lua_rawgeti(L, 5, i);
jbe@140	398 luaL_addvalue(&buf);
jbe@140	399 }
jbe@140	400 luaL_pushresult(&buf);
jbe@305	401 lua_pushvalue(L, -2);
jbe@140	402 }
jbe@143	403 return 2;
jbe@140	404 }
jbe@140	405
jbe@140	406 static int moonbr_io_read_call(lua_State *L) {
jbe@140	407 lua_settop(L, 4);
jbe@140	408 lua_pushnil(L);
jbe@140	409 #if LUA_VERSION_NUM >= 503
jbe@140	410 return moonbr_io_read_cont(L, 0, 0);
jbe@140	411 #else
jbe@140	412 return moonbr_io_read_cont(L);
jbe@140	413 #endif
jbe@140	414 }
jbe@140	415
jbe@145	416 moonbr_io_yield_wrapper(moonbr_io_read_yield, moonbr_io_read_call);
jbe@140	417
jbe@144	418 #if LUA_VERSION_NUM >= 503
jbe@144	419 static int moonbr_io_drain_cont(lua_State *L, int status, lua_KContext ctx) {
jbe@144	420 #else
jbe@144	421 static int moonbr_io_drain_cont(lua_State *L) {
jbe@144	422 #endif
jbe@144	423 lua_Integer remaining, len;
jbe@144	424 size_t totallen = 0;
jbe@144	425 #if !(LUA_VERSION_NUM >= 503)
jbe@144	426 int ctx = 0;
jbe@144	427 lua_getctx(L, &ctx);
jbe@144	428 #endif
jbe@144	429 remaining = lua_tointeger(L, 3);
jbe@144	430 while (1) {
jbe@144	431 lua_pushcfunction(L, moonbr_io_drain_nb);
jbe@144	432 lua_pushvalue(L, 1);
jbe@144	433 lua_pushvalue(L, 3);
jbe@144	434 lua_pushvalue(L, 4);
jbe@144	435 lua_call(L, 3, 2);
jbe@144	436 if (lua_isnil(L, -2)) return 2;
jbe@144	437 lua_insert(L, -2);
jbe@144	438 len = lua_tointeger(L, -1);
jbe@144	439 lua_pop(L, 1);
jbe@144	440 totallen += len;
jbe@144	441 if (strcmp(lua_tostring(L, -1), "block") != 0) break;
jbe@144	442 lua_pop(L, 1);
jbe@144	443 if (remaining >= 0 && len) {
jbe@144	444 remaining -= len;
jbe@144	445 lua_pushinteger(L, remaining);
jbe@144	446 lua_replace(L, 3);
jbe@144	447 }
jbe@144	448 lua_pushvalue(L, 2);
jbe@152	449 lua_callk(L, 0, 0, ctx, moonbr_io_drain_cont);
jbe@144	450 }
jbe@144	451 lua_pushinteger(L, totallen);
jbe@144	452 lua_pushvalue(L, -2);
jbe@144	453 return 2;
jbe@144	454 }
jbe@144	455
jbe@144	456 static int moonbr_io_drain_call(lua_State *L) {
jbe@144	457 #if LUA_VERSION_NUM >= 503
jbe@144	458 return moonbr_io_drain_cont(L, 0, 0);
jbe@144	459 #else
jbe@144	460 return moonbr_io_drain_cont(L);
jbe@144	461 #endif
jbe@144	462 }
jbe@144	463
jbe@145	464 moonbr_io_yield_wrapper(moonbr_io_drain_yield, moonbr_io_drain_call);
jbe@144	465
jbe@250	466 #ifdef MOONBR_IO_USE_TLS
jbe@258	467
jbe@250	468 #define moonbr_io_write_tls(buf, buflen) \
jbe@250	469 if (handle->tls) { \
jbe@250	470 do { \
jbe@250	471 if (!handle->tlshandshake) { \
jbe@250	472 do written = tls_handshake(handle->tls); \
jbe@250	473 while (!nonblocking && (written == TLS_WANT_POLLIN \|\| written == TLS_WANT_POLLOUT)); \
jbe@250	474 if (written == TLS_WANT_POLLIN \|\| written == TLS_WANT_POLLOUT) { \
jbe@250	475 handle->tlshandshake = written; \
jbe@250	476 errno = EAGAIN; \
jbe@250	477 break; \
jbe@250	478 } \
jbe@250	479 if (written < 0) { \
jbe@250	480 lua_pushnil(L); \
jbe@250	481 lua_pushstring(L, tls_error(handle->tls)); \
jbe@250	482 return 2; \
jbe@250	483 } \
jbe@250	484 handle->tlshandshake = 1; \
jbe@250	485 } \
jbe@250	486 do written = tls_write(handle->tls, (buf), (buflen)); \
jbe@250	487 while (!nonblocking && (written == TLS_WANT_POLLIN \|\| written == TLS_WANT_POLLOUT)); \
jbe@250	488 if (written == TLS_WANT_POLLIN \|\| written == TLS_WANT_POLLOUT) { \
jbe@250	489 errno = EAGAIN; \
jbe@250	490 break; \
jbe@250	491 } \
jbe@250	492 if (written < 0) { \
jbe@250	493 lua_pushnil(L); \
jbe@250	494 lua_pushstring(L, tls_error(handle->tls)); \
jbe@250	495 return 2; \
jbe@250	496 } \
jbe@250	497 } while (0); \
jbe@250	498 } \
jbe@250	499 else
jbe@258	500
jbe@258	501 #endif /* MOONBR_IO_USE_TLS */
jbe@250	502
jbe@81	503 static int moonbr_io_write_impl(lua_State *L, int nonblocking, int flush) {
jbe@80	504 moonbr_io_handle_t *handle;
jbe@80	505 int i, top;
jbe@80	506 const char *str;
jbe@92	507 size_t strlen;
jbe@103	508 ssize_t written;
jbe@80	509 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@94	510 if (handle->closed) luaL_error(L, "Attempt to write to a closed I/O handle");
jbe@94	511 if (handle->finished) luaL_error(L, "Attempt to write to a finished I/O handle");
jbe@81	512 if (handle->writeerr) {
jbe@80	513 lua_pushnil(L);
jbe@80	514 lua_pushliteral(L, "Previous write error");
jbe@80	515 return 2;
jbe@80	516 }
jbe@103	517 handle->writeerr = 1;
jbe@257	518 if (moonbr_io_handle_set_nonblocking(L, handle, nonblocking)) moonbr_io_return_errmsg();
jbe@84	519 top = lua_gettop(L);
jbe@81	520 lua_getuservalue(L, 1);
jbe@103	521 lua_getfield(L, -1, "writequeue");
jbe@84	522 for (i=2; i<=top; i++) {
jbe@84	523 luaL_checklstring(L, i, &strlen);
jbe@84	524 lua_pushvalue(L, i);
jbe@84	525 lua_rawseti(L, -2, handle->writeqin++);
jbe@84	526 handle->writeleft += strlen;
jbe@81	527 }
jbe@132	528 if (flush) handle->flushedleft = handle->writeleft;
jbe@83	529 while (handle->writeqout != handle->writeqin) {
jbe@83	530 lua_rawgeti(L, -1, handle->writeqout);
jbe@81	531 str = lua_tolstring(L, -1, &strlen);
jbe@92	532 while (handle->writeqoff < strlen) {
jbe@103	533 if (
jbe@132	534 strlen - handle->writeqoff <
jbe@103	535 MOONBR_IO_WRITEBUFLEN - handle->writebufin
jbe@103	536 ) {
jbe@103	537 memcpy(
jbe@103	538 handle->writebuf + handle->writebufin,
jbe@103	539 str + handle->writeqoff,
jbe@103	540 strlen - handle->writeqoff
jbe@103	541 );
jbe@103	542 handle->writebufin += strlen - handle->writeqoff;
jbe@80	543 break;
jbe@80	544 } else {
jbe@103	545 memcpy(
jbe@103	546 handle->writebuf + handle->writebufin,
jbe@103	547 str + handle->writeqoff,
jbe@103	548 MOONBR_IO_WRITEBUFLEN - handle->writebufin
jbe@103	549 );
jbe@103	550 handle->writeqoff += MOONBR_IO_WRITEBUFLEN - handle->writebufin;
jbe@228	551 handle->writebufin = MOONBR_IO_WRITEBUFLEN;
jbe@103	552 while (handle->writebufout < MOONBR_IO_WRITEBUFLEN) {
jbe@257	553 if (moonbr_io_handle_set_nopush(L, handle, 1)) moonbr_io_return_errmsg();
jbe@249	554 #ifdef MOONBR_IO_USE_TLS
jbe@250	555 moonbr_io_write_tls(
jbe@250	556 handle->writebuf + handle->writebufout,
jbe@250	557 MOONBR_IO_WRITEBUFLEN - handle->writebufout
jbe@250	558 )
jbe@249	559 #endif
jbe@103	560 written = write(
jbe@103	561 handle->fd,
jbe@103	562 handle->writebuf + handle->writebufout,
jbe@103	563 MOONBR_IO_WRITEBUFLEN - handle->writebufout
jbe@103	564 );
jbe@103	565 if (written < 0) {
jbe@81	566 if (nonblocking && (errno == EAGAIN \|\| errno == EWOULDBLOCK)) {
jbe@81	567 goto moonbr_io_write_impl_block;
jbe@257	568 } else if (errno != EINTR) moonbr_io_return_errmsg();
jbe@103	569 } else {
jbe@103	570 handle->writebufout += written;
jbe@103	571 handle->writeleft -= written;
jbe@133	572 if (handle->flushedleft) {
jbe@133	573 if (written >= handle->flushedleft) {
jbe@133	574 handle->flushedleft = 0;
jbe@257	575 if (moonbr_io_handle_set_nopush(L, handle, 0)) moonbr_io_return_errmsg();
jbe@133	576 } else {
jbe@133	577 handle->flushedleft -= written;
jbe@133	578 }
jbe@133	579 }
jbe@80	580 }
jbe@80	581 }
jbe@103	582 handle->writebufin = 0;
jbe@103	583 handle->writebufout = 0;
jbe@80	584 }
jbe@80	585 }
jbe@81	586 handle->writeqoff = 0;
jbe@81	587 lua_pop(L, 1);
jbe@81	588 lua_pushnil(L);
jbe@83	589 lua_rawseti(L, -2, handle->writeqout++);
jbe@80	590 }
jbe@132	591 while (handle->flushedleft) {
jbe@257	592 if (moonbr_io_handle_set_nopush(L, handle, 1)) moonbr_io_return_errmsg();
jbe@249	593 #ifdef MOONBR_IO_USE_TLS
jbe@250	594 moonbr_io_write_tls(
jbe@250	595 handle->writebuf + handle->writebufout,
jbe@250	596 handle->writebufin - handle->writebufout
jbe@250	597 )
jbe@249	598 #endif
jbe@132	599 written = write(
jbe@132	600 handle->fd,
jbe@132	601 handle->writebuf + handle->writebufout,
jbe@132	602 handle->writebufin - handle->writebufout
jbe@132	603 );
jbe@132	604 if (written < 0) {
jbe@132	605 if (nonblocking && (errno == EAGAIN \|\| errno == EWOULDBLOCK)) {
jbe@132	606 goto moonbr_io_write_impl_block;
jbe@257	607 } else if (errno != EINTR) moonbr_io_return_errmsg();
jbe@132	608 } else {
jbe@132	609 handle->writebufout += written;
jbe@132	610 handle->writeleft -= written;
jbe@133	611 if (handle->flushedleft) {
jbe@133	612 if (written >= handle->flushedleft) {
jbe@133	613 handle->flushedleft = 0;
jbe@257	614 if (moonbr_io_handle_set_nopush(L, handle, 0)) moonbr_io_return_errmsg();
jbe@133	615 } else {
jbe@133	616 handle->flushedleft -= written;
jbe@133	617 }
jbe@133	618 }
jbe@81	619 }
jbe@132	620 }
jbe@132	621 if (handle->writebufout == handle->writebufin) {
jbe@103	622 handle->writebufin = 0;
jbe@103	623 handle->writebufout = 0;
jbe@81	624 }
jbe@103	625 if (nonblocking) lua_pushinteger(L, 0);
jbe@103	626 else lua_pushvalue(L, 1);
jbe@103	627 handle->writeerr = 0;
jbe@80	628 return 1;
jbe@81	629 moonbr_io_write_impl_block:
jbe@91	630 lua_pushinteger(L, handle->writeleft);
jbe@103	631 handle->writeerr = 0;
jbe@81	632 return 1;
jbe@81	633 }
jbe@81	634
jbe@81	635 static int moonbr_io_write(lua_State *L) {
jbe@81	636 return moonbr_io_write_impl(L, 0, 0);
jbe@81	637 }
jbe@81	638
jbe@81	639 static int moonbr_io_write_nb(lua_State *L) {
jbe@81	640 return moonbr_io_write_impl(L, 1, 0);
jbe@80	641 }
jbe@80	642
jbe@80	643 static int moonbr_io_flush(lua_State *L) {
jbe@81	644 return moonbr_io_write_impl(L, 0, 1);
jbe@81	645 }
jbe@81	646
jbe@81	647 static int moonbr_io_flush_nb(lua_State *L) {
jbe@81	648 return moonbr_io_write_impl(L, 1, 1);
jbe@80	649 }
jbe@80	650
jbe@145	651 #if LUA_VERSION_NUM >= 503
jbe@145	652 static int moonbr_io_write_cont(lua_State *L, int status, lua_KContext ctx) {
jbe@145	653 #else
jbe@145	654 static int moonbr_io_write_cont(lua_State *L) {
jbe@145	655 #endif
jbe@145	656 while (1) {
jbe@145	657 lua_pushcfunction(L, moonbr_io_write_nb);
jbe@145	658 lua_pushvalue(L, 1);
jbe@145	659 lua_call(L, 1, 2);
jbe@145	660 if (lua_isnil(L, -2)) return 2;
jbe@145	661 if (!lua_tointeger(L, -2)) {
jbe@145	662 lua_pushvalue(L, 1);
jbe@145	663 return 1;
jbe@145	664 }
jbe@145	665 lua_pop(L, 2);
jbe@145	666 lua_pushvalue(L, 2);
jbe@152	667 lua_callk(L, 0, 0, 0, moonbr_io_write_cont);
jbe@145	668 }
jbe@145	669 }
jbe@145	670
jbe@145	671 static int moonbr_io_write_call(lua_State *L) {
jbe@145	672 lua_pushcfunction(L, moonbr_io_write_nb);
jbe@145	673 lua_insert(L, 3);
jbe@145	674 lua_pushvalue(L, 1);
jbe@145	675 lua_insert(L, 4);
jbe@145	676 lua_call(L, lua_gettop(L) - 3, 2);
jbe@145	677 if (lua_isnil(L, -2)) return 2;
jbe@145	678 if (!lua_tointeger(L, -2)) {
jbe@145	679 lua_pushvalue(L, 1);
jbe@145	680 return 1;
jbe@145	681 }
jbe@145	682 #if LUA_VERSION_NUM >= 503
jbe@145	683 return moonbr_io_write_cont(L, 0, 0);
jbe@145	684 #else
jbe@145	685 return moonbr_io_write_cont(L);
jbe@145	686 #endif
jbe@145	687 }
jbe@145	688
jbe@145	689 moonbr_io_yield_wrapper(moonbr_io_write_yield, moonbr_io_write_call);
jbe@145	690
jbe@145	691 static int moonbr_io_flush_call(lua_State *L) {
jbe@145	692 lua_pushcfunction(L, moonbr_io_flush_nb);
jbe@145	693 lua_insert(L, 3);
jbe@145	694 lua_pushvalue(L, 1);
jbe@145	695 lua_insert(L, 4);
jbe@145	696 lua_call(L, lua_gettop(L) - 3, 2);
jbe@145	697 if (lua_isnil(L, -2)) return 2;
jbe@145	698 if (!lua_tointeger(L, -2)) {
jbe@145	699 lua_pushvalue(L, 1);
jbe@145	700 return 1;
jbe@145	701 }
jbe@145	702 #if LUA_VERSION_NUM >= 503
jbe@145	703 return moonbr_io_write_cont(L, 0, 0);
jbe@145	704 #else
jbe@145	705 return moonbr_io_write_cont(L);
jbe@145	706 #endif
jbe@145	707 }
jbe@145	708
jbe@145	709 moonbr_io_yield_wrapper(moonbr_io_flush_yield, moonbr_io_flush_call);
jbe@145	710
jbe@88	711 static int moonbr_io_finish(lua_State *L) {
jbe@88	712 moonbr_io_handle_t *handle;
jbe@88	713 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@94	714 if (handle->closed) luaL_error(L, "Attempt to finish a closed I/O handle");
jbe@94	715 if (handle->finished) luaL_error(L, "Attempt to finish a finished I/O handle");
jbe@94	716 if (handle->writeleft) {
jbe@94	717 lua_pushcfunction(L, moonbr_io_flush);
jbe@94	718 lua_pushvalue(L, 1);
jbe@116	719 if (lua_pcall(L, 1, 2, 0)) {
jbe@116	720 handle->finished = 1;
jbe@116	721 lua_error(L);
jbe@116	722 }
jbe@94	723 if (!lua_toboolean(L, -2)) {
jbe@94	724 handle->finished = 1;
jbe@94	725 return 2;
jbe@88	726 }
jbe@94	727 }
jbe@94	728 handle->finished = 1;
jbe@260	729 #ifdef MOONBR_IO_USE_TLS
jbe@260	730 if ((handle->addrfam == AF_INET6 \|\| handle->addrfam == AF_INET) && !handle->tls) {
jbe@260	731 #else
jbe@107	732 if (handle->addrfam == AF_INET6 \|\| handle->addrfam == AF_INET) {
jbe@260	733 #endif
jbe@257	734 if (shutdown(handle->fd, SHUT_WR)) moonbr_io_return_errmsg();
jbe@94	735 } else {
jbe@251	736 #ifdef MOONBR_IO_USE_TLS
jbe@260	737 if (handle->tls) {
jbe@260	738 int status;
jbe@260	739 if (moonbr_io_handle_set_nonblocking(L, handle, 1)) moonbr_io_return_errmsg();
jbe@260	740 do status = tls_close(handle->tls);
jbe@260	741 while (status == TLS_WANT_POLLIN \|\| status == TLS_WANT_POLLOUT);
jbe@260	742 if (status) {
jbe@260	743 close(handle->fd);
jbe@260	744 handle->fd = -1;
jbe@260	745 lua_pushnil(L);
jbe@260	746 lua_pushstring(L, tls_error(handle->tls));
jbe@260	747 return 2;
jbe@260	748 }
jbe@260	749 }
jbe@251	750 #endif
jbe@94	751 if (close(handle->fd)) {
jbe@94	752 handle->fd = -1;
jbe@257	753 moonbr_io_return_errmsg();
jbe@94	754 }
jbe@94	755 handle->fd = -1; /* fake EOF on read */
jbe@88	756 }
jbe@88	757 lua_pushboolean(L, 1);
jbe@88	758 return 1;
jbe@88	759 }
jbe@88	760
jbe@261	761 static int moonbr_io_close_impl(lua_State *L, int nonblocking, int reset) {
jbe@83	762 moonbr_io_handle_t *handle;
jbe@83	763 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@94	764 if (handle->closed) luaL_error(L, "Attempt to close a closed I/O handle");
jbe@259	765 if (!reset && handle->fd >= 0) {
jbe@87	766 if (handle->writeleft) {
jbe@261	767 lua_pushcfunction(L, nonblocking ? moonbr_io_flush_nb : moonbr_io_flush);
jbe@87	768 lua_pushvalue(L, 1);
jbe@116	769 if (lua_pcall(L, 1, 2, 0)) {
jbe@129	770 handle->closed = 1;
jbe@116	771 close(handle->fd);
jbe@116	772 handle->fd = -1;
jbe@116	773 lua_error(L);
jbe@116	774 }
jbe@261	775 if (!nonblocking) handle->closed = 1; /* TODO: handle nonblocking case */
jbe@87	776 if (!lua_toboolean(L, -2)) {
jbe@87	777 close(handle->fd);
jbe@87	778 handle->fd = -1;
jbe@87	779 return 2;
jbe@87	780 }
jbe@261	781 #if LUA_VERSION_NUM >= 503
jbe@261	782 if (nonblocking && lua_tointeger(L, -2)) {
jbe@261	783 #else
jbe@261	784 if (nonblocking && lua_tonumber(L, -2)) {
jbe@261	785 #endif
jbe@261	786 lua_pushliteral(L, "flush");
jbe@261	787 lua_pushvalue(L, -3);
jbe@261	788 return 2;
jbe@261	789 }
jbe@129	790 } else {
jbe@129	791 handle->closed = 1;
jbe@83	792 }
jbe@260	793 #ifdef MOONBR_IO_USE_TLS
jbe@260	794 if (handle->tls) {
jbe@260	795 int status;
jbe@261	796 if (moonbr_io_handle_set_nonblocking(L, handle, nonblocking)) moonbr_io_return_errmsg();
jbe@260	797 do status = tls_close(handle->tls);
jbe@261	798 while (!nonblocking && (status == TLS_WANT_POLLIN \|\| status == TLS_WANT_POLLOUT));
jbe@261	799 if (status == TLS_WANT_POLLIN \|\| status == TLS_WANT_POLLOUT) {
jbe@261	800 handle->tlsclosing = status; /* TODO: handle polling */
jbe@261	801 lua_pushliteral(L, "close");
jbe@261	802 return 1;
jbe@261	803 }
jbe@260	804 if (status) {
jbe@260	805 close(handle->fd);
jbe@260	806 handle->fd = -1;
jbe@260	807 lua_pushnil(L);
jbe@260	808 lua_pushstring(L, tls_error(handle->tls));
jbe@260	809 return 2;
jbe@260	810 }
jbe@260	811 }
jbe@260	812 #endif
jbe@256	813 if (moonbr_io_handle_set_linger(L, handle, -1)) {
jbe@257	814 moonbr_io_prepare_errmsg();
jbe@256	815 close(handle->fd);
jbe@256	816 handle->fd = -1;
jbe@257	817 moonbr_io_return_prepared_errmsg();
jbe@256	818 }
jbe@129	819 } else {
jbe@129	820 handle->closed = 1;
jbe@83	821 }
jbe@94	822 if (handle->fd >= 0) {
jbe@94	823 if (close(handle->fd)) {
jbe@94	824 handle->fd = -1;
jbe@257	825 moonbr_io_return_errmsg();
jbe@94	826 }
jbe@104	827 handle->fd = -1;
jbe@83	828 }
jbe@83	829 lua_pushboolean(L, 1);
jbe@83	830 return 1;
jbe@84	831
jbe@83	832 }
jbe@83	833
jbe@94	834 static int moonbr_io_close(lua_State *L) {
jbe@261	835 return moonbr_io_close_impl(L, 0, 0);
jbe@94	836 }
jbe@94	837
jbe@84	838 static int moonbr_io_reset(lua_State *L) {
jbe@261	839 return moonbr_io_close_impl(L, 0, 1);
jbe@84	840 }
jbe@84	841
jbe@108	842 static int moonbr_io_handlegc(lua_State *L) {
jbe@88	843 moonbr_io_handle_t *handle;
jbe@88	844 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@88	845 if (handle->fd >= 0) {
jbe@218	846 lua_pushcfunction(L, moonbr_io_reset);
jbe@88	847 lua_pushvalue(L, 1);
jbe@88	848 lua_pushinteger(L, 0);
jbe@88	849 lua_call(L, 2, 0);
jbe@88	850 }
jbe@246	851 #ifdef MOONBR_IO_USE_TLS
jbe@246	852 if (handle->tls) {
jbe@246	853 tls_free(handle->tls);
jbe@246	854 handle->tls = NULL;
jbe@246	855 }
jbe@246	856 if (handle->servertls) {
jbe@246	857 tls_free(handle->servertls);
jbe@246	858 handle->servertls = NULL;
jbe@246	859 }
jbe@246	860 #endif
jbe@88	861 return 0;
jbe@88	862 }
jbe@88	863
jbe@100	864 void moonbr_io_closehandle(lua_State *L, int idx, int reset) {
jbe@88	865 moonbr_io_handle_t *handle;
jbe@88	866 handle = luaL_checkudata(L, idx, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@100	867 if (!handle->closed) {
jbe@100	868 lua_pushcfunction(L, reset ? moonbr_io_reset : moonbr_io_close);
jbe@100	869 lua_pushvalue(L, idx < 0 ? idx-1 : idx);
jbe@100	870 lua_call(L, 1, 0);
jbe@88	871 }
jbe@88	872 }
jbe@88	873
jbe@208	874 static int moonbr_io_pushhandle_impl(lua_State *L) {
jbe@208	875 int *fd;
jbe@79	876 moonbr_io_handle_t *handle;
jbe@107	877 struct sockaddr addr;
jbe@107	878 socklen_t addrlen;
jbe@208	879 fd = lua_touserdata(L, 1);
jbe@79	880 handle = lua_newuserdata(L, sizeof(moonbr_io_handle_t));
jbe@208	881 handle->fd = -1; /* avoid closing incomplete handle */
jbe@107	882 addrlen = sizeof(addr);
jbe@208	883 if (getsockname(*fd, &addr, &addrlen)) {
jbe@107	884 if (errno != ENOTSOCK) {
jbe@257	885 moonbr_io_prepare_errmsg();
jbe@107	886 luaL_error(L, "Unexpected error when examining socket: %s", errmsg);
jbe@107	887 }
jbe@107	888 handle->issock = 0;
jbe@107	889 } else {
jbe@107	890 handle->issock = 1;
jbe@107	891 handle->addrfam = addr.sa_family;
jbe@107	892 }
jbe@94	893 handle->finished = 0;
jbe@94	894 handle->closed = 0;
jbe@81	895 handle->nonblocking = -1;
jbe@95	896 handle->nopush = -1;
jbe@85	897 handle->readerr = 0;
jbe@105	898 handle->readbufin = 0;
jbe@105	899 handle->readbufout = 0;
jbe@81	900 handle->writeerr = 0;
jbe@81	901 handle->writeleft = 0;
jbe@132	902 handle->flushedleft = 0;
jbe@83	903 handle->writeqin = 0;
jbe@83	904 handle->writeqout = 0;
jbe@81	905 handle->writeqoff = 0;
jbe@103	906 handle->writebufin = 0;
jbe@103	907 handle->writebufout = 0;
jbe@246	908 #ifdef MOONBR_IO_USE_TLS
jbe@246	909 handle->tls = NULL;
jbe@246	910 handle->servertls = NULL;
jbe@250	911 handle->tlshandshake = 0;
jbe@261	912 handle->tlsclosing = 0;
jbe@246	913 #endif
jbe@208	914 handle->fd = fd; / required for set_linger call */
jbe@256	915 if (moonbr_io_handle_set_linger(L, handle, 0)) {
jbe@257	916 moonbr_io_prepare_errmsg();
jbe@256	917 handle->fd = -1;
jbe@256	918 luaL_error(L, "Unexpected error while setting SO_LINGER with setsockopt: %s", errmsg);
jbe@256	919 }
jbe@208	920 handle->fd = -1; /* avoid closing incomplete handle */
jbe@238	921 luaL_setmetatable(L, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79	922 lua_newtable(L); // uservalue
jbe@81	923 lua_newtable(L);
jbe@103	924 lua_setfield(L, -2, "writequeue");
jbe@79	925 lua_newtable(L); // public
jbe@107	926 if (handle->addrfam == AF_INET6) {
jbe@107	927 struct sockaddr_in6 addr_in6;
jbe@107	928 char addrstrbuf[INET6_ADDRSTRLEN];
jbe@107	929 const char *addrstr;
jbe@107	930 addrlen = sizeof(addr_in6);
jbe@208	931 if (getsockname(fd, (struct sockaddr )&addr_in6, &addrlen)) {
jbe@257	932 moonbr_io_prepare_errmsg();
jbe@107	933 luaL_error(L, "Could not determine local IP address/port: %s", errmsg);
jbe@107	934 }
jbe@107	935 if (addrlen > sizeof(addr_in6)) {
jbe@107	936 luaL_error(L, "Could not determine local IP address/port: buffer size exceeded");
jbe@107	937 }
jbe@107	938 addrstr = inet_ntop(AF_INET6, addr_in6.sin6_addr.s6_addr, addrstrbuf, sizeof(addrstrbuf));
jbe@107	939 if (!addrstr) {
jbe@257	940 moonbr_io_prepare_errmsg();
jbe@107	941 luaL_error(L, "Could not format local IP address: %s", errmsg);
jbe@107	942 } else {
jbe@107	943 lua_pushstring(L, addrstr);
jbe@107	944 lua_setfield(L, -2, "local_ip6");
jbe@107	945 }
jbe@107	946 lua_pushinteger(L, ntohs(addr_in6.sin6_port));
jbe@107	947 lua_setfield(L, -2, "local_tcpport");
jbe@208	948 if (getpeername(fd, (struct sockaddr )&addr_in6, &addrlen)) {
jbe@257	949 moonbr_io_prepare_errmsg();
jbe@107	950 luaL_error(L, "Could not determine remote IP address/port: %s", errmsg);
jbe@107	951 }
jbe@107	952 if (addrlen > sizeof(addr_in6)) {
jbe@107	953 luaL_error(L, "Could not determine remote IP address/port: buffer size exceeded");
jbe@107	954 }
jbe@107	955 addrstr = inet_ntop(AF_INET6, addr_in6.sin6_addr.s6_addr, addrstrbuf, sizeof(addrstrbuf));
jbe@107	956 if (!addrstr) {
jbe@257	957 moonbr_io_prepare_errmsg();
jbe@107	958 luaL_error(L, "Could not format remote IP address: %s", errmsg);
jbe@107	959 } else {
jbe@107	960 lua_pushstring(L, addrstr);
jbe@107	961 lua_setfield(L, -2, "remote_ip6");
jbe@107	962 }
jbe@107	963 lua_pushinteger(L, ntohs(addr_in6.sin6_port));
jbe@107	964 lua_setfield(L, -2, "remote_tcpport");
jbe@107	965 } else if (handle->addrfam == AF_INET) {
jbe@107	966 struct sockaddr_in addr_in;
jbe@107	967 char addrstrbuf[INET_ADDRSTRLEN];
jbe@107	968 const char *addrstr;
jbe@107	969 addrlen = sizeof(addr_in);
jbe@208	970 if (getsockname(fd, (struct sockaddr )&addr_in, &addrlen)) {
jbe@257	971 moonbr_io_prepare_errmsg();
jbe@107	972 luaL_error(L, "Could not determine local IP address/port: %s", errmsg);
jbe@107	973 }
jbe@107	974 if (addrlen > sizeof(addr_in)) {
jbe@107	975 luaL_error(L, "Could not determine local IP address/port: buffer size exceeded");
jbe@107	976 }
jbe@107	977 addrstr = inet_ntop(AF_INET, &addr_in.sin_addr.s_addr, addrstrbuf, sizeof(addrstrbuf));
jbe@107	978 if (!addrstr) {
jbe@257	979 moonbr_io_prepare_errmsg();
jbe@107	980 luaL_error(L, "Could not format local IP address: %s", errmsg);
jbe@107	981 } else {
jbe@107	982 lua_pushstring(L, addrstr);
jbe@107	983 lua_setfield(L, -2, "local_ip4");
jbe@107	984 }
jbe@107	985 lua_pushinteger(L, ntohs(addr_in.sin_port));
jbe@107	986 lua_setfield(L, -2, "local_tcpport");
jbe@208	987 if (getpeername(fd, (struct sockaddr )&addr_in, &addrlen)) {
jbe@257	988 moonbr_io_prepare_errmsg();
jbe@107	989 luaL_error(L, "Could not determine remote IP address/port: %s", errmsg);
jbe@107	990 }
jbe@107	991 if (addrlen > sizeof(addr_in)) {
jbe@107	992 luaL_error(L, "Could not determine remote IP address/port: buffer size exceeded");
jbe@107	993 }
jbe@107	994 addrstr = inet_ntop(AF_INET, &addr_in.sin_addr.s_addr, addrstrbuf, sizeof(addrstrbuf));
jbe@107	995 if (!addrstr) {
jbe@257	996 moonbr_io_prepare_errmsg();
jbe@107	997 luaL_error(L, "Could not format remote IP address: %s", errmsg);
jbe@107	998 } else {
jbe@107	999 lua_pushstring(L, addrstr);
jbe@107	1000 lua_setfield(L, -2, "remote_ip4");
jbe@107	1001 }
jbe@107	1002 lua_pushinteger(L, ntohs(addr_in.sin_port));
jbe@107	1003 lua_setfield(L, -2, "remote_tcpport");
jbe@107	1004 }
jbe@238	1005 luaL_setmetatable(L, MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY);
jbe@79	1006 lua_setfield(L, -2, "public");
jbe@79	1007 lua_setuservalue(L, -2);
jbe@208	1008 handle->fd = *fd;
jbe@208	1009 fd = -1; / closing is now handled by garbage collection */
jbe@208	1010 return 1;
jbe@208	1011 }
jbe@208	1012
jbe@208	1013 void moonbr_io_pushhandle(lua_State *L, int fd) {
jbe@208	1014 lua_pushcfunction(L, moonbr_io_pushhandle_impl);
jbe@208	1015 lua_pushlightuserdata(L, &fd);
jbe@208	1016 if (lua_pcall(L, 1, 1, 0)) {
jbe@251	1017 if (fd != -1) close(fd); // TODO: correct to close file descriptor here?
jbe@208	1018 lua_error(L);
jbe@208	1019 }
jbe@79	1020 }
jbe@79	1021
jbe@79	1022 static int moonbr_io_handleindex(lua_State *L) {
jbe@80	1023 luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@205	1024 luaL_checkany(L, 2);
jbe@79	1025 lua_getuservalue(L, 1);
jbe@79	1026 lua_getfield(L, -1, "public");
jbe@79	1027 lua_pushvalue(L, 2);
jbe@79	1028 lua_gettable(L, -2);
jbe@79	1029 return 1;
jbe@79	1030 }
jbe@79	1031
jbe@79	1032 static int moonbr_io_handlenewindex(lua_State *L) {
jbe@80	1033 luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@205	1034 luaL_checkany(L, 2);
jbe@205	1035 luaL_checkany(L, 3);
jbe@79	1036 lua_getuservalue(L, 1);
jbe@79	1037 lua_getfield(L, -1, "public");
jbe@79	1038 lua_pushvalue(L, 2);
jbe@79	1039 lua_pushvalue(L, 3);
jbe@79	1040 lua_settable(L, -3);
jbe@79	1041 return 0;
jbe@79	1042 }
jbe@79	1043
jbe@111	1044 static int moonbr_io_localconnect_impl(lua_State *L, int nonblocking) {
jbe@111	1045 const char *path;
jbe@111	1046 struct sockaddr_un sockaddr = { .sun_family = AF_LOCAL };
jbe@111	1047 const int path_maxlen = sizeof(struct sockaddr_un) - (
jbe@111	1048 (void )sockaddr.sun_path - (void )&sockaddr
jbe@111	1049 ) - 1; /* one byte for termination */
jbe@111	1050 int sock;
jbe@111	1051 path = luaL_checkstring(L, 1);
jbe@111	1052 if (strlen(path) > path_maxlen) luaL_error(L, "Path too long; only %i characters allowed", path_maxlen);
jbe@111	1053 strcpy(sockaddr.sun_path, path);
jbe@111	1054 sock = socket(
jbe@111	1055 PF_LOCAL,
jbe@111	1056 SOCK_STREAM \| SOCK_CLOEXEC \| (nonblocking ? SOCK_NONBLOCK : 0),
jbe@111	1057 0
jbe@111	1058 );
jbe@257	1059 if (sock < 0) moonbr_io_return_errmsg();
jbe@111	1060 if (connect(sock, (struct sockaddr *)&sockaddr, sizeof(sockaddr))) {
jbe@111	1061 if (!nonblocking && errno == EINTR) {
jbe@257	1062 moonbr_io_prepare_errmsg();
jbe@111	1063 close(sock);
jbe@257	1064 moonbr_io_return_prepared_errmsg();
jbe@257	1065 } else if (!(nonblocking && (errno == EINPROGRESS \|\| errno == EINTR))) moonbr_io_return_errmsg();
jbe@111	1066 }
jbe@111	1067 moonbr_io_pushhandle(L, sock);
jbe@111	1068 return 1;
jbe@111	1069 }
jbe@111	1070
jbe@111	1071 static int moonbr_io_localconnect(lua_State *L) {
jbe@111	1072 return moonbr_io_localconnect_impl(L, 0);
jbe@111	1073 }
jbe@111	1074
jbe@111	1075 static int moonbr_io_localconnect_nb(lua_State *L) {
jbe@111	1076 return moonbr_io_localconnect_impl(L, 1);
jbe@111	1077 }
jbe@111	1078
jbe@99	1079 static int moonbr_io_tcpconnect_impl(lua_State *L, int nonblocking) {
jbe@98	1080 const char host, port;
jbe@98	1081 struct addrinfo hints = { 0, };
jbe@98	1082 struct addrinfo res, addrinfo;
jbe@98	1083 int errcode;
jbe@98	1084 int sock;
jbe@98	1085 host = luaL_checkstring(L, 1);
jbe@98	1086 port = luaL_checkstring(L, 2);
jbe@98	1087 hints.ai_family = AF_UNSPEC;
jbe@98	1088 hints.ai_socktype = SOCK_STREAM;
jbe@98	1089 hints.ai_protocol = IPPROTO_TCP;
jbe@98	1090 hints.ai_flags = AI_ADDRCONFIG;
jbe@98	1091 errcode = getaddrinfo(host, port, &hints, &res);
jbe@98	1092 if (errcode) {
jbe@102	1093 freeaddrinfo(res);
jbe@98	1094 if (errcode == EAI_SYSTEM) {
jbe@257	1095 moonbr_io_prepare_errmsg();
jbe@98	1096 lua_pushnil(L);
jbe@98	1097 lua_pushfstring(L, "%s: %s", gai_strerror(errcode), errmsg);
jbe@98	1098 } else {
jbe@98	1099 lua_pushnil(L);
jbe@98	1100 lua_pushstring(L, gai_strerror(errcode));
jbe@98	1101 }
jbe@98	1102 return 2;
jbe@98	1103 }
jbe@98	1104 for (addrinfo=res; addrinfo; addrinfo=addrinfo->ai_next) {
jbe@134	1105 if (addrinfo->ai_family == AF_INET6) goto moonbr_io_tcpconnect_found;
jbe@98	1106 }
jbe@98	1107 for (addrinfo=res; addrinfo; addrinfo=addrinfo->ai_next) {
jbe@134	1108 if (addrinfo->ai_family == AF_INET) goto moonbr_io_tcpconnect_found;
jbe@98	1109 }
jbe@98	1110 addrinfo = res;
jbe@98	1111 moonbr_io_tcpconnect_found:
jbe@99	1112 sock = socket(
jbe@134	1113 addrinfo->ai_family, /* NOTE: not correctly using PF_* but AF_* constants here */
jbe@99	1114 addrinfo->ai_socktype \| SOCK_CLOEXEC \| (nonblocking ? SOCK_NONBLOCK : 0),
jbe@99	1115 addrinfo->ai_protocol
jbe@99	1116 );
jbe@98	1117 if (sock < 0) {
jbe@257	1118 moonbr_io_prepare_errmsg();
jbe@108	1119 freeaddrinfo(res);
jbe@257	1120 moonbr_io_return_prepared_errmsg();
jbe@98	1121 }
jbe@98	1122 if (connect(sock, addrinfo->ai_addr, addrinfo->ai_addrlen)) {
jbe@108	1123 freeaddrinfo(res);
jbe@99	1124 if (!nonblocking && errno == EINTR) {
jbe@257	1125 moonbr_io_prepare_errmsg();
jbe@99	1126 close(sock);
jbe@257	1127 moonbr_io_return_prepared_errmsg();
jbe@257	1128 } else if (!(nonblocking && (errno == EINPROGRESS \|\| errno == EINTR))) moonbr_io_return_errmsg();
jbe@108	1129 } else {
jbe@108	1130 freeaddrinfo(res);
jbe@98	1131 }
jbe@107	1132 moonbr_io_pushhandle(L, sock);
jbe@98	1133 return 1;
jbe@98	1134 }
jbe@98	1135
jbe@99	1136 static int moonbr_io_tcpconnect(lua_State *L) {
jbe@99	1137 return moonbr_io_tcpconnect_impl(L, 0);
jbe@99	1138 }
jbe@99	1139
jbe@99	1140 static int moonbr_io_tcpconnect_nb(lua_State *L) {
jbe@99	1141 return moonbr_io_tcpconnect_impl(L, 1);
jbe@99	1142 }
jbe@99	1143
jbe@112	1144 static int moonbr_io_locallisten(lua_State *L) {
jbe@112	1145 moonbr_io_listener_t *listener;
jbe@112	1146 const char *path;
jbe@118	1147 struct stat sb;
jbe@112	1148 struct sockaddr_un sockaddr = { .sun_family = AF_LOCAL };
jbe@112	1149 const int path_maxlen = sizeof(struct sockaddr_un) - (
jbe@112	1150 (void )sockaddr.sun_path - (void )&sockaddr
jbe@112	1151 ) - 1; /* one byte for termination */
jbe@112	1152 int sock;
jbe@112	1153 path = luaL_checkstring(L, 1);
jbe@112	1154 if (strlen(path) > path_maxlen) luaL_error(L, "Path too long; only %i characters allowed", path_maxlen);
jbe@112	1155 strcpy(sockaddr.sun_path, path);
jbe@118	1156 if (stat(path, &sb) == 0) {
jbe@118	1157 if (S_ISSOCK(sb.st_mode)) unlink(path);
jbe@118	1158 }
jbe@112	1159 listener = lua_newuserdata(L, sizeof(moonbr_io_listener_t));
jbe@117	1160 listener->fd = -1;
jbe@112	1161 luaL_setmetatable(L, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@112	1162 sock = socket(
jbe@112	1163 PF_LOCAL,
jbe@112	1164 SOCK_STREAM \| SOCK_CLOEXEC,
jbe@112	1165 0
jbe@112	1166 );
jbe@257	1167 if (sock < 0) moonbr_io_return_errmsg();
jbe@112	1168 if (bind(sock, (struct sockaddr *)&sockaddr, sizeof(sockaddr))) {
jbe@257	1169 moonbr_io_prepare_errmsg();
jbe@112	1170 close(sock);
jbe@257	1171 moonbr_io_return_prepared_errmsg();
jbe@112	1172 }
jbe@112	1173 if (listen(sock, MOONBR_IO_LISTEN_BACKLOG)) {
jbe@257	1174 moonbr_io_prepare_errmsg();
jbe@112	1175 close(sock);
jbe@257	1176 moonbr_io_return_prepared_errmsg();
jbe@112	1177 }
jbe@112	1178 listener->fd = sock;
jbe@118	1179 listener->addrfam = AF_LOCAL;
jbe@112	1180 listener->nonblocking = -1;
jbe@112	1181 return 1;
jbe@112	1182 }
jbe@112	1183
jbe@108	1184 static int moonbr_io_tcplisten(lua_State *L) {
jbe@108	1185 moonbr_io_listener_t *listener;
jbe@108	1186 const char host, port;
jbe@108	1187 struct addrinfo hints = { 0, };
jbe@108	1188 struct addrinfo res, addrinfo;
jbe@108	1189 int errcode;
jbe@108	1190 int sock;
jbe@108	1191 host = luaL_optstring(L, 1, NULL);
jbe@108	1192 port = luaL_checkstring(L, 2);
jbe@108	1193 listener = lua_newuserdata(L, sizeof(moonbr_io_listener_t));
jbe@117	1194 listener->fd = -1;
jbe@108	1195 luaL_setmetatable(L, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108	1196 hints.ai_family = AF_UNSPEC;
jbe@108	1197 hints.ai_socktype = SOCK_STREAM;
jbe@108	1198 hints.ai_protocol = IPPROTO_TCP;
jbe@108	1199 hints.ai_flags = AI_ADDRCONFIG \| AI_PASSIVE;
jbe@108	1200 errcode = getaddrinfo(host, port, &hints, &res);
jbe@108	1201 if (errcode) {
jbe@108	1202 freeaddrinfo(res);
jbe@108	1203 if (errcode == EAI_SYSTEM) {
jbe@257	1204 moonbr_io_prepare_errmsg();
jbe@108	1205 lua_pushnil(L);
jbe@108	1206 lua_pushfstring(L, "%s: %s", gai_strerror(errcode), errmsg);
jbe@108	1207 } else {
jbe@108	1208 lua_pushnil(L);
jbe@108	1209 lua_pushstring(L, gai_strerror(errcode));
jbe@108	1210 }
jbe@108	1211 return 2;
jbe@108	1212 }
jbe@108	1213 for (addrinfo=res; addrinfo; addrinfo=addrinfo->ai_next) {
jbe@134	1214 if (addrinfo->ai_family == AF_INET6) goto moonbr_io_tcpconnect_found;
jbe@108	1215 }
jbe@108	1216 for (addrinfo=res; addrinfo; addrinfo=addrinfo->ai_next) {
jbe@134	1217 if (addrinfo->ai_family == AF_INET) goto moonbr_io_tcpconnect_found;
jbe@108	1218 }
jbe@108	1219 addrinfo = res;
jbe@108	1220 moonbr_io_tcpconnect_found:
jbe@118	1221 listener->addrfam = addrinfo->ai_family;
jbe@108	1222 sock = socket(
jbe@134	1223 addrinfo->ai_family, /* NOTE: not correctly using PF_* but AF_* constants here */
jbe@108	1224 addrinfo->ai_socktype \| SOCK_CLOEXEC,
jbe@108	1225 addrinfo->ai_protocol
jbe@108	1226 );
jbe@108	1227 if (sock < 0) {
jbe@257	1228 moonbr_io_prepare_errmsg();
jbe@108	1229 freeaddrinfo(res);
jbe@257	1230 moonbr_io_return_prepared_errmsg();
jbe@108	1231 }
jbe@120	1232 {
jbe@120	1233 static const int reuseval = 1;
jbe@120	1234 if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &reuseval, sizeof(reuseval))) {
jbe@257	1235 moonbr_io_prepare_errmsg();
jbe@120	1236 freeaddrinfo(res);
jbe@120	1237 close(sock);
jbe@120	1238 lua_pushnil(L);
jbe@120	1239 lua_pushfstring(L, "Error while setting SO_REUSEADDR with setsockopt: %s", errmsg);
jbe@120	1240 return 2;
jbe@120	1241 }
jbe@120	1242 }
jbe@108	1243 if (bind(sock, addrinfo->ai_addr, addrinfo->ai_addrlen)) {
jbe@257	1244 moonbr_io_prepare_errmsg();
jbe@108	1245 freeaddrinfo(res);
jbe@108	1246 close(sock);
jbe@257	1247 moonbr_io_return_prepared_errmsg();
jbe@108	1248 }
jbe@108	1249 freeaddrinfo(res);
jbe@108	1250 if (listen(sock, MOONBR_IO_LISTEN_BACKLOG)) {
jbe@257	1251 moonbr_io_prepare_errmsg();
jbe@108	1252 close(sock);
jbe@257	1253 moonbr_io_return_prepared_errmsg();
jbe@108	1254 }
jbe@108	1255 listener->fd = sock;
jbe@108	1256 listener->nonblocking = -1;
jbe@108	1257 return 1;
jbe@108	1258 }
jbe@108	1259
jbe@108	1260 static int moonbr_io_accept_impl(lua_State *L, int nonblocking) {
jbe@108	1261 moonbr_io_listener_t *listener;
jbe@108	1262 int fd;
jbe@108	1263 listener = luaL_checkudata(L, 1, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108	1264 if (listener->fd < 0) luaL_error(L, "Attempt to use a closed listener");
jbe@108	1265 if (listener->nonblocking != nonblocking) {
jbe@108	1266 int flags;
jbe@108	1267 flags = fcntl(listener->fd, F_GETFL, 0);
jbe@108	1268 if (flags == -1) {
jbe@257	1269 moonbr_io_prepare_errmsg();
jbe@108	1270 close(listener->fd);
jbe@108	1271 listener->fd = -1;
jbe@108	1272 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@108	1273 }
jbe@108	1274 if (nonblocking) flags \|= O_NONBLOCK;
jbe@108	1275 else flags &= ~O_NONBLOCK;
jbe@108	1276 if (fcntl(listener->fd, F_SETFL, flags) == -1) {
jbe@257	1277 moonbr_io_prepare_errmsg();
jbe@108	1278 close(listener->fd);
jbe@108	1279 listener->fd = -1;
jbe@108	1280 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@108	1281 }
jbe@108	1282 listener->nonblocking = nonblocking;
jbe@108	1283 }
jbe@108	1284 while (1) {
jbe@214	1285 #if defined(__linux__) && !defined(_GNU_SOURCE)
jbe@215	1286 fd = accept(listener->fd, NULL, NULL);
jbe@214	1287 if (fd != -1) {
jbe@214	1288 if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1) {
jbe@257	1289 moonbr_io_prepare_errmsg();
jbe@214	1290 close(listener->fd);
jbe@214	1291 listener->fd = -1;
jbe@214	1292 close(fd);
jbe@214	1293 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@214	1294 }
jbe@214	1295 }
jbe@214	1296 #else
jbe@108	1297 fd = accept4(listener->fd, NULL, NULL, SOCK_CLOEXEC);
jbe@214	1298 #endif
jbe@108	1299 if (fd < 0) {
jbe@108	1300 if (nonblocking && (errno == EAGAIN \|\| errno == EWOULDBLOCK)) {
jbe@108	1301 lua_pushboolean(L, 0);
jbe@108	1302 lua_pushliteral(L, "No incoming connection pending");
jbe@108	1303 return 2;
jbe@257	1304 } else if (errno != EINTR) moonbr_io_return_errmsg();
jbe@108	1305 } else {
jbe@108	1306 moonbr_io_pushhandle(L, fd);
jbe@108	1307 return 1;
jbe@108	1308 }
jbe@108	1309 }
jbe@108	1310 }
jbe@108	1311
jbe@108	1312 static int moonbr_io_accept(lua_State *L) {
jbe@108	1313 return moonbr_io_accept_impl(L, 0);
jbe@108	1314 }
jbe@108	1315
jbe@108	1316 static int moonbr_io_accept_nb(lua_State *L) {
jbe@108	1317 return moonbr_io_accept_impl(L, 1);
jbe@108	1318 }
jbe@108	1319
jbe@108	1320 static int moonbr_io_unlisten(lua_State *L) {
jbe@108	1321 moonbr_io_listener_t *listener;
jbe@118	1322 struct sockaddr_un addr;
jbe@118	1323 socklen_t addrlen;
jbe@118	1324 struct stat sb;
jbe@108	1325 listener = luaL_checkudata(L, 1, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108	1326 if (listener->fd < 0) luaL_error(L, "Attempt to close a closed listener");
jbe@118	1327 addrlen = sizeof(addr);
jbe@118	1328 if (getsockname(listener->fd, (struct sockaddr *)&addr, &addrlen)) addrlen = 0;
jbe@108	1329 if (close(listener->fd)) {
jbe@257	1330 moonbr_io_prepare_errmsg();
jbe@108	1331 listener->fd = -1;
jbe@118	1332 if (addrlen && addrlen <= sizeof(addr)) {
jbe@118	1333 if (stat(addr.sun_path, &sb) == 0) {
jbe@118	1334 if (S_ISSOCK(sb.st_mode)) unlink(addr.sun_path);
jbe@118	1335 }
jbe@118	1336 }
jbe@257	1337 moonbr_io_return_prepared_errmsg();
jbe@108	1338 }
jbe@108	1339 listener->fd = -1;
jbe@118	1340 if (addrlen && addrlen <= sizeof(addr)) {
jbe@118	1341 if (stat(addr.sun_path, &sb) == 0) {
jbe@118	1342 if (S_ISSOCK(sb.st_mode)) unlink(addr.sun_path);
jbe@118	1343 }
jbe@118	1344 }
jbe@108	1345 lua_pushboolean(L, 1);
jbe@108	1346 return 1;
jbe@108	1347 }
jbe@108	1348
jbe@108	1349 static int moonbr_io_listenergc(lua_State *L) {
jbe@108	1350 moonbr_io_listener_t *listener;
jbe@108	1351 listener = luaL_checkudata(L, 1, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@117	1352 if (listener->fd >= 0) close(listener->fd);
jbe@108	1353 listener->fd = -1;
jbe@108	1354 return 0;
jbe@108	1355 }
jbe@108	1356
jbe@205	1357 static int moonbr_io_exec(lua_State *L) {
jbe@205	1358 char **argv;
jbe@205	1359 int i, argc;
jbe@205	1360 int sockin[2], sockout[2], sockerr[2];
jbe@205	1361 volatile int errorcond = 0;
jbe@213	1362 volatile char errmsgbuf[MOONBR_IO_MAXSTRERRORLEN] = MOONBR_IO_STRERROR_R_MSG;
jbe@205	1363 moonbr_io_child_t *child;
jbe@205	1364 argc = lua_gettop(L);
jbe@205	1365 argv = lua_newuserdata(L, (argc + 1) * sizeof(char *));
jbe@205	1366 for (i=0; i<argc; i++) argv[i] = (char *)luaL_checkstring(L, i+1);
jbe@205	1367 argv[argc] = NULL;
jbe@208	1368 child = lua_newuserdata(L, sizeof(moonbr_io_child_t));
jbe@208	1369 child->pid = 0;
jbe@303	1370 child->status_valid = 0;
jbe@208	1371 lua_newtable(L);
jbe@208	1372 lua_setuservalue(L, -2);
jbe@238	1373 luaL_setmetatable(L, MOONBR_IO_CHILD_MT_REGKEY);
jbe@206	1374 if (socketpair(PF_LOCAL, SOCK_STREAM \| SOCK_CLOEXEC, 0, sockin)) {
jbe@257	1375 moonbr_io_prepare_errmsg();
jbe@205	1376 lua_pushnil(L);
jbe@205	1377 lua_pushfstring(L, "Could not create socket pair: %s", errmsg);
jbe@205	1378 return 2;
jbe@205	1379 }
jbe@206	1380 if (socketpair(PF_LOCAL, SOCK_STREAM \| SOCK_CLOEXEC, 0, sockout)) {
jbe@257	1381 moonbr_io_prepare_errmsg();
jbe@205	1382 close(sockin[0]);
jbe@205	1383 close(sockin[1]);
jbe@205	1384 lua_pushnil(L);
jbe@205	1385 lua_pushfstring(L, "Could not create socket pair: %s", errmsg);
jbe@205	1386 return 2;
jbe@205	1387 }
jbe@206	1388 if (socketpair(PF_LOCAL, SOCK_STREAM \| SOCK_CLOEXEC, 0, sockerr)) {
jbe@257	1389 moonbr_io_prepare_errmsg();
jbe@205	1390 close(sockin[0]);
jbe@205	1391 close(sockin[1]);
jbe@205	1392 close(sockout[0]);
jbe@205	1393 close(sockout[1]);
jbe@205	1394 lua_pushnil(L);
jbe@205	1395 lua_pushfstring(L, "Could not create socket pair: %s", errmsg);
jbe@205	1396 return 2;
jbe@205	1397 }
jbe@205	1398 child->pid = vfork();
jbe@205	1399 if (child->pid == -1) {
jbe@257	1400 moonbr_io_prepare_errmsg();
jbe@205	1401 close(sockin[0]);
jbe@205	1402 close(sockin[1]);
jbe@205	1403 close(sockout[0]);
jbe@205	1404 close(sockout[1]);
jbe@205	1405 close(sockerr[0]);
jbe@205	1406 close(sockerr[1]);
jbe@205	1407 lua_pushnil(L);
jbe@205	1408 lua_pushfstring(L, "Could not fork: %s", errmsg);
jbe@205	1409 return 2;
jbe@205	1410 }
jbe@205	1411 if (!child->pid) {
jbe@208	1412 if (dup2(sockin[1], 0) == -1) goto moonbr_io_exec_error1;
jbe@208	1413 if (dup2(sockout[1], 1) == -1) goto moonbr_io_exec_error1;
jbe@208	1414 if (dup2(sockerr[1], 2) == -1) goto moonbr_io_exec_error1;
jbe@232	1415 closefrom(3);
jbe@208	1416 if (fcntl(0, F_SETFD, 0) == -1) goto moonbr_io_exec_error1;
jbe@208	1417 if (fcntl(1, F_SETFD, 0) == -1) goto moonbr_io_exec_error1;
jbe@208	1418 if (fcntl(2, F_SETFD, 0) == -1) goto moonbr_io_exec_error1;
jbe@205	1419 if (execvp(argv[0], argv)) {
jbe@205	1420 errorcond = 2;
jbe@205	1421 strerror_r(errno, (char *)errmsgbuf, MOONBR_IO_MAXSTRERRORLEN);
jbe@205	1422 _exit(0);
jbe@205	1423 }
jbe@208	1424 moonbr_io_exec_error1:
jbe@206	1425 errorcond = 1;
jbe@206	1426 strerror_r(errno, (char *)errmsgbuf, MOONBR_IO_MAXSTRERRORLEN);
jbe@206	1427 _exit(0);
jbe@205	1428 }
jbe@205	1429 close(sockin[1]);
jbe@205	1430 close(sockout[1]);
jbe@205	1431 close(sockerr[1]);
jbe@205	1432 if (errorcond) {
jbe@205	1433 int status;
jbe@205	1434 close(sockin[0]);
jbe@205	1435 close(sockout[0]);
jbe@205	1436 close(sockerr[0]);
jbe@205	1437 while (waitpid(child->pid, &status, 0) == -1) {
jbe@205	1438 if (errno != EINTR) {
jbe@257	1439 moonbr_io_prepare_errmsg();
jbe@205	1440 luaL_error(L, "Error in waitpid call after unsuccessful exec: %s", errmsg);
jbe@205	1441 }
jbe@205	1442 }
jbe@208	1443 child->pid = 0;
jbe@205	1444 lua_pushnil(L);
jbe@205	1445 if (errorcond == 2) lua_pushfstring(L, "Could not execute: %s", errmsgbuf);
jbe@205	1446 else lua_pushfstring(L, "Error in fork: %s", errmsgbuf);
jbe@205	1447 return 2;
jbe@205	1448 }
jbe@208	1449 lua_pushcfunction(L, moonbr_io_pushhandle_impl);
jbe@208	1450 lua_pushlightuserdata(L, &sockin[0]);
jbe@208	1451 if (lua_pcall(L, 1, 1, 0)) {
jbe@208	1452 if (sockin[0] != -1) close(sockin[0]);
jbe@208	1453 close(sockout[0]);
jbe@208	1454 close(sockerr[0]);
jbe@208	1455 goto moonbr_io_exec_error2;
jbe@208	1456 }
jbe@205	1457 lua_setfield(L, -2, "stdin");
jbe@208	1458 lua_pushcfunction(L, moonbr_io_pushhandle_impl);
jbe@208	1459 lua_pushlightuserdata(L, &sockout[0]);
jbe@208	1460 if (lua_pcall(L, 1, 1, 0)) {
jbe@208	1461 if (sockout[0] != -1) close(sockout[0]);
jbe@208	1462 close(sockerr[0]);
jbe@208	1463 goto moonbr_io_exec_error2;
jbe@208	1464 }
jbe@205	1465 lua_setfield(L, -2, "stdout");
jbe@208	1466 lua_pushcfunction(L, moonbr_io_pushhandle_impl);
jbe@208	1467 lua_pushlightuserdata(L, &sockerr[0]);
jbe@208	1468 if (lua_pcall(L, 1, 1, 0)) {
jbe@208	1469 if (sockerr[0] != -1) close(sockerr[0]);
jbe@208	1470 goto moonbr_io_exec_error2;
jbe@208	1471 }
jbe@205	1472 lua_setfield(L, -2, "stderr");
jbe@205	1473 return 1;
jbe@208	1474 moonbr_io_exec_error2:
jbe@208	1475 {
jbe@208	1476 int status;
jbe@208	1477 while (waitpid(child->pid, &status, 0) == -1) {
jbe@208	1478 if (errno != EINTR) {
jbe@257	1479 moonbr_io_prepare_errmsg();
jbe@208	1480 luaL_error(L, "Error in waitpid call after error creating socket handles: %s", errmsg);
jbe@208	1481 }
jbe@208	1482 }
jbe@208	1483 }
jbe@208	1484 child->pid = 0;
jbe@208	1485 return lua_error(L);
jbe@205	1486 }
jbe@205	1487
jbe@205	1488 static int moonbr_io_childindex(lua_State *L) {
jbe@205	1489 luaL_checkudata(L, 1, MOONBR_IO_CHILD_MT_REGKEY);
jbe@205	1490 luaL_checkany(L, 2);
jbe@205	1491 lua_getuservalue(L, 1);
jbe@205	1492 lua_pushvalue(L, 2);
jbe@205	1493 lua_gettable(L, -2);
jbe@205	1494 if (lua_isnil(L, -1)) {
jbe@205	1495 luaL_getmetatable(L, MOONBR_IO_CHILD_PT_REGKEY);
jbe@205	1496 lua_pushvalue(L, 2);
jbe@205	1497 lua_gettable(L, -2);
jbe@205	1498 }
jbe@205	1499 return 1;
jbe@205	1500 }
jbe@205	1501
jbe@205	1502 static int moonbr_io_childnewindex(lua_State *L) {
jbe@205	1503 luaL_checkudata(L, 1, MOONBR_IO_CHILD_MT_REGKEY);
jbe@205	1504 luaL_checkany(L, 2);
jbe@205	1505 luaL_checkany(L, 3);
jbe@205	1506 lua_getuservalue(L, 1);
jbe@205	1507 lua_pushvalue(L, 2);
jbe@205	1508 lua_pushvalue(L, 3);
jbe@205	1509 lua_settable(L, -3);
jbe@205	1510 return 0;
jbe@205	1511 }
jbe@205	1512
jbe@205	1513 static int moonbr_io_childgc(lua_State *L) {
jbe@205	1514 moonbr_io_child_t *child;
jbe@205	1515 child = luaL_checkudata(L, 1, MOONBR_IO_CHILD_MT_REGKEY);
jbe@205	1516 if (child->pid) {
jbe@205	1517 int status;
jbe@281	1518 int pid = child->pid;
jbe@281	1519 child->pid = 0;
jbe@281	1520 if (kill(pid, SIGKILL)) {
jbe@257	1521 moonbr_io_prepare_errmsg();
jbe@205	1522 luaL_error(L, "Error in kill call during garbage collection: %s", errmsg);
jbe@205	1523 }
jbe@281	1524 while (waitpid(pid, &status, 0) == -1) {
jbe@205	1525 if (errno != EINTR) {
jbe@257	1526 moonbr_io_prepare_errmsg();
jbe@205	1527 luaL_error(L, "Error in waitpid call during garbage collection: %s", errmsg);
jbe@205	1528 }
jbe@205	1529 }
jbe@205	1530 }
jbe@205	1531 return 0;
jbe@205	1532 }
jbe@205	1533
jbe@205	1534 static int moonbr_io_kill(lua_State *L) {
jbe@205	1535 moonbr_io_child_t *child;
jbe@205	1536 int sig;
jbe@205	1537 child = luaL_checkudata(L, 1, MOONBR_IO_CHILD_MT_REGKEY);
jbe@285	1538 sig = luaL_optinteger(L, 2, SIGKILL);
jbe@303	1539 if (!child->pid) {
jbe@303	1540 if (!child->status_valid) luaL_error(L, "Attempt to kill an already collected child process");
jbe@303	1541 } else {
jbe@303	1542 if (kill(child->pid, sig)) {
jbe@303	1543 moonbr_io_prepare_errmsg();
jbe@303	1544 luaL_error(L, "Error in kill call: %s", errmsg);
jbe@303	1545 }
jbe@205	1546 }
jbe@205	1547 lua_settop(L, 1);
jbe@205	1548 return 1;
jbe@205	1549 }
jbe@205	1550
jbe@205	1551 static int moonbr_io_wait_impl(lua_State *L, int nonblocking) {
jbe@205	1552 moonbr_io_child_t *child;
jbe@205	1553 int status;
jbe@205	1554 child = luaL_checkudata(L, 1, MOONBR_IO_CHILD_MT_REGKEY);
jbe@303	1555 if (!child->pid) {
jbe@303	1556 if (!child->status_valid) luaL_error(L, "Attempt to wait for an already collected child process");
jbe@303	1557 status = child->status;
jbe@303	1558 child->status_valid = 0;
jbe@303	1559 } else {
jbe@303	1560 pid_t waitedpid;
jbe@303	1561 while ((waitedpid = waitpid(child->pid, &status, nonblocking ? WNOHANG : 0)) == -1) {
jbe@303	1562 if (errno != EINTR) {
jbe@303	1563 moonbr_io_prepare_errmsg();
jbe@303	1564 luaL_error(L, "Error in waitpid call: %s", errmsg);
jbe@303	1565 }
jbe@205	1566 }
jbe@303	1567 if (!waitedpid) {
jbe@303	1568 lua_pushboolean(L, 0);
jbe@303	1569 lua_pushliteral(L, "Process is still running");
jbe@303	1570 return 2;
jbe@303	1571 }
jbe@303	1572 child->pid = 0;
jbe@205	1573 }
jbe@303	1574 if (WIFEXITED(status)) {
jbe@303	1575 lua_pushinteger(L, WEXITSTATUS(status));
jbe@303	1576 } else if (WIFSIGNALED(status)) {
jbe@303	1577 lua_pushinteger(L, -WTERMSIG(status));
jbe@205	1578 } else {
jbe@303	1579 luaL_error(L, "Unexpected status value returned by waitpid call");
jbe@205	1580 }
jbe@303	1581 return 1;
jbe@205	1582 }
jbe@205	1583
jbe@205	1584 static int moonbr_io_wait(lua_State *L) {
jbe@205	1585 return moonbr_io_wait_impl(L, 0);
jbe@205	1586 }
jbe@205	1587
jbe@205	1588 static int moonbr_io_wait_nb(lua_State *L) {
jbe@205	1589 return moonbr_io_wait_impl(L, 1);
jbe@205	1590 }
jbe@205	1591
jbe@205	1592 #if LUA_VERSION_NUM >= 503
jbe@205	1593 static int moonbr_io_wait_cont(lua_State *L, int status, lua_KContext ctx) {
jbe@205	1594 #else
jbe@205	1595 static int moonbr_io_wait_cont(lua_State *L) {
jbe@205	1596 #endif
jbe@205	1597 #if !(LUA_VERSION_NUM >= 503)
jbe@205	1598 int ctx = 0;
jbe@205	1599 lua_getctx(L, &ctx);
jbe@205	1600 #endif
jbe@205	1601 while (1) {
jbe@205	1602 lua_pushcfunction(L, moonbr_io_wait_nb);
jbe@205	1603 lua_pushvalue(L, 1);
jbe@205	1604 lua_call(L, 1, 1);
jbe@205	1605 if (!lua_isnil(L, -1)) break;
jbe@205	1606 lua_pushvalue(L, 2);
jbe@205	1607 lua_callk(L, 0, 0, ctx, moonbr_io_wait_cont);
jbe@205	1608 }
jbe@205	1609 return 1;
jbe@205	1610 }
jbe@205	1611
jbe@205	1612 static int moonbr_io_wait_call(lua_State *L) {
jbe@205	1613 lua_settop(L, 2);
jbe@205	1614 #if LUA_VERSION_NUM >= 503
jbe@205	1615 return moonbr_io_wait_cont(L, 0, 0);
jbe@205	1616 #else
jbe@205	1617 return moonbr_io_wait_cont(L);
jbe@205	1618 #endif
jbe@205	1619 }
jbe@205	1620
jbe@205	1621 moonbr_io_yield_wrapper(moonbr_io_wait_yield, moonbr_io_wait_call);
jbe@205	1622
jbe@284	1623 static void moonbr_io_sigterm_handler(int sig) {
jbe@284	1624 moonbr_io_sigterm_flag = 1;
jbe@266	1625 }
jbe@266	1626
jbe@297	1627 static void moonbr_io_sigchld_handler(int sig) {
jbe@297	1628 moonbr_io_sigchld_flag = 1;
jbe@297	1629 }
jbe@297	1630
jbe@288	1631 int moonbr_io_catch_sigterm(lua_State *L) {
jbe@284	1632 signal(SIGTERM, moonbr_io_sigterm_handler);
jbe@284	1633 return 0;
jbe@284	1634 }
jbe@284	1635
jbe@266	1636 static int moonbr_io_getpid(lua_State *L) {
jbe@266	1637 lua_pushinteger(L, getpid());
jbe@266	1638 return 1;
jbe@266	1639 }
jbe@266	1640
jbe@250	1641 #ifdef MOONBR_IO_USE_TLS
jbe@258	1642
jbe@250	1643 #define moonbr_io_poll_tls() \
jbe@250	1644 if (!handle->tlshandshake) { \
jbe@295	1645 force_wakeup = 1; \
jbe@295	1646 continue; \
jbe@250	1647 } \
jbe@250	1648 if (handle->tlshandshake == TLS_WANT_POLLIN) { \
jbe@250	1649 if (fd < 0) { \
jbe@295	1650 force_wakeup = 1; \
jbe@295	1651 continue; \
jbe@250	1652 } \
jbe@250	1653 FD_SET(fd, &readfds); \
jbe@250	1654 if (fd+1 > nfds) nfds = fd+1; \
jbe@250	1655 continue; \
jbe@250	1656 } \
jbe@250	1657 if (handle->tlshandshake == TLS_WANT_POLLOUT) { \
jbe@250	1658 if (fd < 0) { \
jbe@295	1659 force_wakeup = 1; \
jbe@295	1660 continue; \
jbe@250	1661 } \
jbe@250	1662 FD_SET(fd, &writefds); \
jbe@250	1663 if (fd+1 > nfds) nfds = fd+1; \
jbe@250	1664 continue; \
jbe@250	1665 } \
jbe@250	1666 while (0)
jbe@258	1667
jbe@258	1668 #endif /* MOONBR_IO_USE_TLS */
jbe@250	1669
jbe@106	1670 static int moonbr_io_poll(lua_State *L) {
jbe@106	1671 moonbr_io_handle_t *handle;
jbe@108	1672 moonbr_io_listener_t *listener;
jbe@296	1673 moonbr_io_child_t *child;
jbe@106	1674 int fd, isnum;
jbe@106	1675 int nfds = 0;
jbe@106	1676 fd_set readfds, writefds, exceptfds;
jbe@284	1677 struct timespec timeout = {0, };
jbe@295	1678 int force_wakeup = 0;
jbe@288	1679 int use_timeout = 0; // negative for negative timeout
jbe@284	1680 int check_sigterm = 0;
jbe@297	1681 int check_sigchld = 0;
jbe@298	1682 pid_t waitedpid;
jbe@284	1683 sigset_t mask, orig_mask;
jbe@106	1684 int status;
jbe@106	1685 FD_ZERO(&readfds);
jbe@106	1686 FD_ZERO(&writefds);
jbe@106	1687 FD_ZERO(&exceptfds);
jbe@106	1688 if (!lua_isnoneornil(L, 1)) {
jbe@106	1689 luaL_checktype(L, 1, LUA_TTABLE);
jbe@106	1690 for (lua_pushnil(L); lua_next(L, 1); lua_pop(L, 1)) {
jbe@106	1691 if (lua_toboolean(L, -1)) {
jbe@106	1692 handle = luaL_testudata(L, -2, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@106	1693 if (handle) {
jbe@109	1694 if (handle->closed) luaL_error(L, "Attempt to poll a closed connection");
jbe@106	1695 fd = handle->fd;
jbe@247	1696 #if MOONBR_IO_USE_TLS
jbe@250	1697 moonbr_io_poll_tls();
jbe@247	1698 #endif
jbe@122	1699 if (
jbe@122	1700 fd < 0 \|\| /* fake EOF to simulate shutdown if fd < 0 */
jbe@122	1701 handle->readbufin != handle->readbufout /* data pending in buffer */
jbe@122	1702 ) {
jbe@295	1703 force_wakeup = 1;
jbe@295	1704 continue;
jbe@109	1705 }
jbe@106	1706 } else {
jbe@108	1707 listener = luaL_testudata(L, -2, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108	1708 if (listener) {
jbe@108	1709 fd = listener->fd;
jbe@108	1710 if (fd < 0) luaL_error(L, "Attempt to poll a closed listener");
jbe@108	1711 } else {
jbe@296	1712 child = luaL_testudata(L, -2, MOONBR_IO_CHILD_MT_REGKEY);
jbe@296	1713 if (child) {
jbe@299	1714 if (!child->pid) luaL_error(L, "Attemt to poll an already collected child process");
jbe@297	1715 if (!check_sigchld) {
jbe@297	1716 check_sigchld = 1;
jbe@297	1717 moonbr_io_sigchld_flag = 0;
jbe@297	1718 signal(SIGCHLD, moonbr_io_sigchld_handler);
jbe@297	1719 }
jbe@303	1720 if (child->status_valid) {
jbe@303	1721 force_wakeup = 1;
jbe@303	1722 } else {
jbe@303	1723 while ((waitedpid = waitpid(child->pid, &status, WNOHANG)) == -1) {
jbe@303	1724 if (errno != EINTR) {
jbe@303	1725 moonbr_io_prepare_errmsg();
jbe@303	1726 luaL_error(L, "Error in waitpid call: %s", errmsg);
jbe@303	1727 }
jbe@303	1728 }
jbe@303	1729 if (waitedpid) {
jbe@303	1730 child->pid = 0;
jbe@303	1731 child->status = status;
jbe@303	1732 child->status_valid = 1;
jbe@303	1733 force_wakeup = 1;
jbe@298	1734 }
jbe@298	1735 }
jbe@296	1736 continue;
jbe@296	1737 } else {
jbe@296	1738 fd = lua_tointegerx(L, -2, &isnum);
jbe@296	1739 if (!isnum) luaL_error(L, "Expected integer (file descriptor), I/O handle, or listener in table key");
jbe@296	1740 }
jbe@108	1741 }
jbe@106	1742 }
jbe@138	1743 if (fd < 0 \|\| fd >= FD_SETSIZE) luaL_error(L, "File descriptor out of valid range");
jbe@106	1744 FD_SET(fd, &readfds);
jbe@106	1745 if (fd+1 > nfds) nfds = fd+1;
jbe@106	1746 }
jbe@106	1747 }
jbe@106	1748 }
jbe@106	1749 if (!lua_isnoneornil(L, 2)) {
jbe@106	1750 luaL_checktype(L, 2, LUA_TTABLE);
jbe@106	1751 for (lua_pushnil(L); lua_next(L, 2); lua_pop(L, 1)) {
jbe@106	1752 if (lua_toboolean(L, -1)) {
jbe@106	1753 handle = luaL_testudata(L, -2, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@106	1754 if (handle) {
jbe@109	1755 if (handle->closed) luaL_error(L, "Attempt to poll a closed connection");
jbe@109	1756 if (handle->finished) luaL_error(L, "Attempt to write-poll a finished connection");
jbe@106	1757 fd = handle->fd;
jbe@247	1758 #if MOONBR_IO_USE_TLS
jbe@250	1759 moonbr_io_poll_tls();
jbe@247	1760 #endif
jbe@106	1761 } else {
jbe@108	1762 listener = luaL_testudata(L, -2, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@109	1763 if (listener) luaL_error(L, "Attempt to write-poll a listener");
jbe@109	1764 fd = lua_tointegerx(L, -2, &isnum);
jbe@109	1765 if (!isnum) luaL_error(L, "Expected integer (file descriptor) or I/O handle in table key");
jbe@106	1766 }
jbe@138	1767 if (fd < 0 \|\| fd >= FD_SETSIZE) luaL_error(L, "File descriptor out of valid range");
jbe@106	1768 FD_SET(fd, &writefds);
jbe@106	1769 if (fd+1 > nfds) nfds = fd+1;
jbe@106	1770 }
jbe@106	1771 }
jbe@106	1772 }
jbe@106	1773 if (!lua_isnoneornil(L, 3)) {
jbe@106	1774 lua_Number n;
jbe@106	1775 n = lua_tonumberx(L, 3, &isnum);
jbe@146	1776 if (isnum && n<0) {
jbe@288	1777 use_timeout = -1;
jbe@146	1778 } else if (isnum && n>=0 && n<100000000) {
jbe@288	1779 use_timeout = 1;
jbe@106	1780 timeout.tv_sec = n;
jbe@284	1781 timeout.tv_nsec = 1e9 * (n - timeout.tv_sec);
jbe@106	1782 } else {
jbe@106	1783 luaL_argcheck(L, 0, 3, "not a valid timeout");
jbe@106	1784 }
jbe@284	1785 }
jbe@295	1786 if (use_timeout < 0) force_wakeup = 1;
jbe@284	1787 if (!lua_isnoneornil(L, 4)) luaL_checktype(L, 4, LUA_TBOOLEAN);
jbe@284	1788 check_sigterm = lua_toboolean(L, 4);
jbe@297	1789 if ((check_sigterm \|\| check_sigchld) && !force_wakeup) {
jbe@296	1790 sigemptyset(&mask);
jbe@296	1791 if (check_sigterm) sigaddset(&mask, SIGTERM);
jbe@297	1792 if (check_sigchld) sigaddset(&mask, SIGCHLD);
jbe@296	1793 if (sigprocmask(SIG_BLOCK, &mask, &orig_mask)) abort();
jbe@296	1794 }
jbe@296	1795 if (check_sigterm && moonbr_io_sigterm_flag) {
jbe@295	1796 if (!force_wakeup) {
jbe@296	1797 if (sigprocmask(SIG_SETMASK, &orig_mask, NULL)) abort();
jbe@295	1798 }
jbe@296	1799 lua_pushboolean(L, 0);
jbe@296	1800 lua_pushliteral(L, "SIGTERM received");
jbe@296	1801 lua_pushboolean(L, 1);
jbe@296	1802 return 3;
jbe@296	1803 }
jbe@297	1804 if (check_sigchld && !force_wakeup && moonbr_io_sigchld_flag) {
jbe@297	1805 if (sigprocmask(SIG_SETMASK, &orig_mask, NULL)) abort();
jbe@297	1806 force_wakeup = 1;
jbe@288	1807 }
jbe@288	1808 if (use_timeout < 0) {
jbe@288	1809 lua_pushboolean(L, 0);
jbe@290	1810 lua_pushliteral(L, "Timeout");
jbe@288	1811 if (check_sigterm) {
jbe@288	1812 lua_pushboolean(L, 0);
jbe@288	1813 return 3;
jbe@288	1814 } else {
jbe@284	1815 return 2;
jbe@284	1816 }
jbe@284	1817 }
jbe@295	1818 if (!force_wakeup) {
jbe@295	1819 status = pselect(
jbe@295	1820 nfds, &readfds, &writefds, &exceptfds,
jbe@295	1821 use_timeout ? &timeout : NULL,
jbe@301	1822 (check_sigterm \|\| check_sigchld) ? &orig_mask : NULL
jbe@295	1823 );
jbe@301	1824 if (check_sigterm \|\| check_sigchld) {
jbe@295	1825 if (sigprocmask(SIG_SETMASK, &orig_mask, NULL)) abort();
jbe@301	1826 if (check_sigterm && moonbr_io_sigterm_flag) {
jbe@295	1827 lua_pushboolean(L, 0);
jbe@295	1828 lua_pushliteral(L, "SIGTERM received");
jbe@295	1829 lua_pushboolean(L, 1);
jbe@295	1830 return 3;
jbe@295	1831 }
jbe@295	1832 }
jbe@295	1833 if (status == -1) {
jbe@295	1834 if (errno == EINTR) {
jbe@295	1835 lua_pushboolean(L, 1);
jbe@295	1836 return 1;
jbe@295	1837 } else {
jbe@295	1838 moonbr_io_prepare_errmsg();
jbe@295	1839 return luaL_error(L, "Unexpected error during \"select\" system call: %s", errmsg);
jbe@295	1840 }
jbe@295	1841 } else if (status == 0) {
jbe@292	1842 lua_pushboolean(L, 0);
jbe@295	1843 lua_pushliteral(L, "Timeout");
jbe@295	1844 if (check_sigterm) {
jbe@295	1845 lua_pushboolean(L, 0);
jbe@295	1846 return 3;
jbe@295	1847 } else {
jbe@295	1848 return 2;
jbe@295	1849 }
jbe@292	1850 }
jbe@106	1851 }
jbe@295	1852 lua_pushboolean(L, 1);
jbe@295	1853 return 1;
jbe@106	1854 }
jbe@106	1855
jbe@115	1856 static int moonbr_io_timeref(lua_State *L) {
jbe@115	1857 lua_Number sub;
jbe@115	1858 struct timespec tp;
jbe@115	1859 sub = luaL_optnumber(L, 1, 0);
jbe@115	1860 if (clock_gettime(CLOCK_MONOTONIC, &tp)) {
jbe@115	1861 return luaL_error(L, "Could not access CLOCK_MONOTONIC");
jbe@115	1862 }
jbe@115	1863 lua_pushnumber(L, tp.tv_sec + tp.tv_nsec / 1.0e9 - sub);
jbe@115	1864 return 1;
jbe@115	1865 }
jbe@115	1866
jbe@243	1867 #ifdef MOONBR_IO_USE_TLS
jbe@243	1868
jbe@243	1869 #define moonbr_io_tlsconf_string(name, field, func) \
jbe@243	1870 /* NOTE: use valuetype = lua_getfield(...) for LUA_VERSION_NUM >= 503 */ \
jbe@243	1871 lua_getfield(L, 1, (field)); \
jbe@243	1872 valuetype = lua_type(L, -1); \
jbe@243	1873 if (valuetype != LUA_TNIL) { \
jbe@243	1874 luaL_argcheck(L, valuetype == LUA_TSTRING, 1, "field \"" field "\" is not a string"); \
jbe@243	1875 value = lua_tostring(L, -1); \
jbe@252	1876 if (func(tlsconf->config, value)) { \
jbe@243	1877 lua_pushnil(L); \
jbe@243	1878 lua_pushfstring(L, "Could not set " name " \"%s\"", value); \
jbe@243	1879 return 2; \
jbe@243	1880 } \
jbe@243	1881 } \
jbe@243	1882 lua_pop(L, 1);
jbe@243	1883
jbe@243	1884 #define moonbr_io_tlsconf_binary(name, field, func) \
jbe@243	1885 /* NOTE: use valuetype = lua_getfield(...) for LUA_VERSION_NUM >= 503 */ \
jbe@243	1886 lua_getfield(L, 1, (field)); \
jbe@243	1887 valuetype = lua_type(L, -1); \
jbe@243	1888 if (valuetype != LUA_TNIL) { \
jbe@243	1889 luaL_argcheck(L, valuetype == LUA_TSTRING, 1, "field \"" field "\" is not a string"); \
jbe@243	1890 value = lua_tolstring(L, -1, &valuelen); \
jbe@252	1891 if (func(tlsconf->config, (void *)value, valuelen)) { \
jbe@243	1892 lua_pushnil(L); \
jbe@243	1893 lua_pushliteral(L, "Could not set " name); \
jbe@243	1894 return 2; \
jbe@243	1895 } \
jbe@243	1896 } \
jbe@243	1897 lua_pop(L, 1);
jbe@243	1898
jbe@243	1899 static int moonbr_io_tlsconf(lua_State *L) {
jbe@252	1900 moonbr_io_tlsconf_t *tlsconf;
jbe@243	1901 int valuetype;
jbe@243	1902 const char *value;
jbe@243	1903 size_t valuelen;
jbe@243	1904 luaL_checktype(L, 1, LUA_TTABLE);
jbe@252	1905 tlsconf = lua_newuserdata(L, sizeof(moonbr_io_tlsconf_t));
jbe@252	1906 tlsconf->config = tls_config_new();
jbe@252	1907 if (!tlsconf->config) {
jbe@243	1908 return luaL_error(L, "Could not allocate memory for TLS configuration");
jbe@243	1909 }
jbe@243	1910 luaL_setmetatable(L, MOONBR_IO_TLSCONF_MT_REGKEY);
jbe@245	1911 lua_getfield(L, 1, "mode");
jbe@252	1912 value = lua_tostring(L, -1);
jbe@252	1913 if (value && !strcmp(value, "server")) tlsconf->server = 1;
jbe@252	1914 else if (value && !strcmp(value, "client")) tlsconf->server = 0;
jbe@252	1915 else luaL_argcheck(L, 0, 1, "field \"mode\" must be set to \"server\" or \"client\"");
jbe@245	1916 lua_pop(L, 1);
jbe@243	1917 moonbr_io_tlsconf_string("CA file", "ca_file", tls_config_set_ca_file);
jbe@243	1918 moonbr_io_tlsconf_string("CA path", "ca_path", tls_config_set_ca_path);
jbe@243	1919 moonbr_io_tlsconf_binary("CA", "ca_mem", tls_config_set_ca_mem);
jbe@243	1920 moonbr_io_tlsconf_string("certificate file", "cert_file", tls_config_set_cert_file);
jbe@243	1921 moonbr_io_tlsconf_binary("certificate", "cert_mem", tls_config_set_cert_mem);
jbe@243	1922 moonbr_io_tlsconf_string("key file", "key_file", tls_config_set_key_file);
jbe@243	1923 moonbr_io_tlsconf_binary("key", "key_mem", tls_config_set_key_mem);
jbe@245	1924 #if LUA_VERSION_NUM >= 503
jbe@245	1925 valuetype = lua_getfield(L, 1, "verify_client");
jbe@245	1926 #else
jbe@244	1927 lua_getfield(L, 1, "verify_client");
jbe@245	1928 #endif
jbe@244	1929 if (lua_toboolean(L, -1)) {
jbe@252	1930 value = lua_tostring(L, -1);
jbe@252	1931 if (value && !strcmp(value, "required")) {
jbe@252	1932 tls_config_verify_client(tlsconf->config);
jbe@252	1933 } else if (value && !strcmp(value, "optional")) {
jbe@252	1934 tls_config_verify_client_optional(tlsconf->config);
jbe@244	1935 } else {
jbe@244	1936 luaL_argcheck(L, 0, 1, "field \"verify_client\" must be set to \"required\", \"optional\", or be false or nil");
jbe@244	1937 }
jbe@244	1938 }
jbe@244	1939 lua_pop(L, 1);
jbe@254	1940 // TODO: configurable legacy support
jbe@254	1941 // tls_config_set_protocols(tlsconf->config, TLS_PROTOCOLS_ALL);
jbe@254	1942 // tls_config_set_ciphers(tlsconf->config, "legacy");
jbe@243	1943 return 1;
jbe@243	1944 }
jbe@243	1945
jbe@252	1946 static int moonbr_io_tlsconfgc(lua_State *L) {
jbe@252	1947 moonbr_io_tlsconf_t *tlsconf;
jbe@243	1948 tlsconf = luaL_checkudata(L, 1, MOONBR_IO_TLSCONF_MT_REGKEY);
jbe@252	1949 if (tlsconf->config) tls_config_free(tlsconf->config);
jbe@252	1950 tlsconf->config = NULL;
jbe@243	1951 return 0;
jbe@243	1952 }
jbe@243	1953
jbe@246	1954 static int moonbr_io_starttls(lua_State *L) {
jbe@246	1955 moonbr_io_handle_t *handle;
jbe@252	1956 moonbr_io_tlsconf_t *tlsconf;
jbe@246	1957 const char *servername;
jbe@246	1958 struct tls tls, tls2;
jbe@246	1959 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@246	1960 if (lua_type(L, 2) == LUA_TTABLE) {
jbe@246	1961 lua_pushcfunction(L, moonbr_io_tlsconf);
jbe@246	1962 lua_pushvalue(L, 2);
jbe@246	1963 lua_call(L, 1, 2);
jbe@246	1964 if (lua_isnil(L, -2)) return 2;
jbe@246	1965 lua_pop(L, 1);
jbe@246	1966 lua_replace(L, 2);
jbe@246	1967 }
jbe@246	1968 tlsconf = luaL_checkudata(L, 2, MOONBR_IO_TLSCONF_MT_REGKEY);
jbe@246	1969 if (handle->closed) return luaL_error(L, "Attempt to start TLS on a closed I/O handle");
jbe@246	1970 if (handle->finished) return luaL_error(L, "Attempt to start TLS on a finished I/O handle");
jbe@247	1971 if (handle->tls) return luaL_error(L, "Attempt to start TLS twice");
jbe@246	1972 if (handle->readbufin \|\| handle->writebufin) {
jbe@246	1973 return luaL_error(L, "Attempt to start TLS on an I/O handle with non-empty buffers");
jbe@246	1974 }
jbe@252	1975 if (tlsconf->server) tls = tls_server();
jbe@252	1976 else {
jbe@246	1977 servername = luaL_checkstring(L, 3);
jbe@246	1978 tls = tls_client();
jbe@246	1979 }
jbe@246	1980 if (!tls) {
jbe@246	1981 return luaL_error(L, "Could not allocate memory for TLS context");
jbe@246	1982 }
jbe@252	1983 if (tls_configure(tls, tlsconf->config)) goto moonbr_io_starttls_error;
jbe@252	1984 if (tlsconf->server) {
jbe@246	1985 if (tls_accept_socket(tls, &tls2, handle->fd)) goto moonbr_io_starttls_error;
jbe@246	1986 handle->servertls = tls;
jbe@246	1987 handle->tls = tls2;
jbe@246	1988 } else {
jbe@246	1989 if (tls_connect_socket(tls, handle->fd, servername)) goto moonbr_io_starttls_error;
jbe@246	1990 handle->tls = tls;
jbe@246	1991 }
jbe@246	1992 lua_settop(L, 1);
jbe@246	1993 return 1;
jbe@246	1994 moonbr_io_starttls_error:
jbe@246	1995 lua_pushnil(L);
jbe@246	1996 lua_pushstring(L, tls_error(tls));
jbe@246	1997 tls_free(tls);
jbe@246	1998 return 2;
jbe@246	1999 }
jbe@246	2000
jbe@258	2001 #endif /* MOONBR_IO_USE_TLS */
jbe@243	2002
jbe@79	2003 static const struct luaL_Reg moonbr_io_handle_methods[] = {
jbe@85	2004 {"read", moonbr_io_read},
jbe@85	2005 {"read_nb", moonbr_io_read_nb},
jbe@140	2006 {"read_call", moonbr_io_read_call},
jbe@140	2007 {"read_yield", moonbr_io_read_yield},
jbe@86	2008 {"drain", moonbr_io_drain},
jbe@86	2009 {"drain_nb", moonbr_io_drain_nb},
jbe@144	2010 {"drain_call", moonbr_io_drain_call},
jbe@144	2011 {"drain_yield", moonbr_io_drain_yield},
jbe@80	2012 {"write", moonbr_io_write},
jbe@81	2013 {"write_nb", moonbr_io_write_nb},
jbe@145	2014 {"write_call", moonbr_io_write_call},
jbe@145	2015 {"write_yield", moonbr_io_write_yield},
jbe@80	2016 {"flush", moonbr_io_flush},
jbe@81	2017 {"flush_nb", moonbr_io_flush_nb},
jbe@145	2018 {"flush_call", moonbr_io_flush_call},
jbe@145	2019 {"flush_yield", moonbr_io_flush_yield},
jbe@88	2020 {"finish", moonbr_io_finish},
jbe@87	2021 {"close", moonbr_io_close},
jbe@85	2022 {"reset", moonbr_io_reset},
jbe@246	2023 #ifdef MOONBR_IO_USE_TLS
jbe@246	2024 {"starttls", moonbr_io_starttls},
jbe@246	2025 #endif
jbe@79	2026 {NULL, NULL}
jbe@79	2027 };
jbe@79	2028
jbe@79	2029 static const struct luaL_Reg moonbr_io_handle_metamethods[] = {
jbe@79	2030 {"__index", moonbr_io_handleindex},
jbe@79	2031 {"__newindex", moonbr_io_handlenewindex},
jbe@108	2032 {"__gc", moonbr_io_handlegc},
jbe@108	2033 {NULL, NULL}
jbe@108	2034 };
jbe@108	2035
jbe@108	2036 static const struct luaL_Reg moonbr_io_listener_methods[] = {
jbe@108	2037 {"accept", moonbr_io_accept},
jbe@108	2038 {"accept_nb", moonbr_io_accept_nb},
jbe@108	2039 {"close", moonbr_io_unlisten},
jbe@108	2040 {NULL, NULL}
jbe@108	2041 };
jbe@108	2042
jbe@108	2043 static const struct luaL_Reg moonbr_io_listener_metamethods[] = {
jbe@108	2044 {"__gc", moonbr_io_listenergc},
jbe@79	2045 {NULL, NULL}
jbe@79	2046 };
jbe@79	2047
jbe@205	2048 static const struct luaL_Reg moonbr_io_child_methods[] = {
jbe@205	2049 {"kill", moonbr_io_kill},
jbe@205	2050 {"wait", moonbr_io_wait},
jbe@205	2051 {"wait_nb", moonbr_io_wait_nb},
jbe@205	2052 {"wait_call", moonbr_io_wait_call},
jbe@205	2053 {"wait_yield", moonbr_io_wait_yield},
jbe@205	2054 {NULL, NULL}
jbe@205	2055 };
jbe@205	2056
jbe@205	2057 static const struct luaL_Reg moonbr_io_child_metamethods[] = {
jbe@205	2058 {"__index", moonbr_io_childindex},
jbe@205	2059 {"__newindex", moonbr_io_childnewindex},
jbe@205	2060 {"__gc", moonbr_io_childgc},
jbe@205	2061 {NULL, NULL}
jbe@205	2062 };
jbe@205	2063
jbe@79	2064 static const struct luaL_Reg moonbr_io_module_funcs[] = {
jbe@111	2065 {"localconnect", moonbr_io_localconnect},
jbe@111	2066 {"localconnect_nb", moonbr_io_localconnect_nb},
jbe@98	2067 {"tcpconnect", moonbr_io_tcpconnect},
jbe@99	2068 {"tcpconnect_nb", moonbr_io_tcpconnect_nb},
jbe@112	2069 {"locallisten", moonbr_io_locallisten},
jbe@108	2070 {"tcplisten", moonbr_io_tcplisten},
jbe@205	2071 {"exec", moonbr_io_exec},
jbe@288	2072 {"catch_sigterm", moonbr_io_catch_sigterm},
jbe@266	2073 {"getpid", moonbr_io_getpid},
jbe@106	2074 {"poll", moonbr_io_poll},
jbe@115	2075 {"timeref", moonbr_io_timeref},
jbe@243	2076 #ifdef MOONBR_IO_USE_TLS
jbe@243	2077 {"tlsconf", moonbr_io_tlsconf},
jbe@243	2078 #endif
jbe@79	2079 {NULL, NULL}
jbe@79	2080 };
jbe@79	2081
jbe@243	2082 #ifdef MOONBR_IO_USE_TLS
jbe@258	2083
jbe@243	2084 static const struct luaL_Reg moonbr_io_tlsconf_metamethods[] = {
jbe@243	2085 {"__gc", moonbr_io_tlsconfgc},
jbe@243	2086 {NULL, NULL}
jbe@243	2087 };
jbe@258	2088
jbe@258	2089 #endif /* MOONBR_IO_USE_TLS */
jbe@243	2090
jbe@79	2091 int luaopen_moonbridge_io(lua_State *L) {
jbe@79	2092
jbe@113	2093 signal(SIGPIPE, SIG_IGN); /* generate I/O errors instead of signal 13 */
jbe@113	2094
jbe@80	2095 lua_newtable(L); // module
jbe@80	2096
jbe@79	2097 lua_newtable(L); // public metatable
jbe@79	2098 lua_newtable(L); // handle methods
jbe@79	2099 luaL_setfuncs(L, moonbr_io_handle_methods, 0);
jbe@80	2100 lua_pushvalue(L, -1);
jbe@146	2101 lua_setfield(L, -4, "handle_pt");
jbe@79	2102 lua_setfield(L, -2, "__index");
jbe@79	2103 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY);
jbe@79	2104
jbe@79	2105 lua_newtable(L); // handle metatable
jbe@79	2106 luaL_setfuncs(L, moonbr_io_handle_metamethods, 0);
jbe@146	2107 lua_pushvalue(L, -1);
jbe@146	2108 lua_setfield(L, -3, "handle_mt");
jbe@79	2109 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79	2110
jbe@108	2111 lua_newtable(L); // listener metatable
jbe@108	2112 luaL_setfuncs(L, moonbr_io_listener_metamethods, 0);
jbe@108	2113 lua_newtable(L); // listener methods
jbe@108	2114 luaL_setfuncs(L, moonbr_io_listener_methods, 0);
jbe@108	2115 lua_pushvalue(L, -1);
jbe@146	2116 lua_setfield(L, -4, "listener_pt");
jbe@108	2117 lua_setfield(L, -2, "__index");
jbe@146	2118 lua_pushvalue(L, -1);
jbe@146	2119 lua_setfield(L, -3, "listener_mt");
jbe@108	2120 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108	2121
jbe@205	2122 lua_newtable(L); // child methods
jbe@205	2123 luaL_setfuncs(L, moonbr_io_child_methods, 0);
jbe@205	2124 lua_pushvalue(L, -1);
jbe@205	2125 lua_setfield(L, -3, "child_pt");
jbe@205	2126 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_CHILD_PT_REGKEY);
jbe@205	2127 lua_newtable(L); // child metatable
jbe@205	2128 luaL_setfuncs(L, moonbr_io_child_metamethods, 0);
jbe@205	2129 lua_pushvalue(L, -1);
jbe@205	2130 lua_setfield(L, -3, "child_mt");
jbe@205	2131 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_CHILD_MT_REGKEY);
jbe@205	2132
jbe@243	2133 #ifdef MOONBR_IO_USE_TLS
jbe@243	2134 if(tls_init()) {
jbe@243	2135 return luaL_error(L, "Could not initialize TLS library");
jbe@243	2136 }
jbe@243	2137 lua_newtable(L); // tlsconf metatable
jbe@243	2138 luaL_setfuncs(L, moonbr_io_tlsconf_metamethods, 0);
jbe@243	2139 lua_pushvalue(L, -1);
jbe@243	2140 lua_setfield(L, -3, "tlsconf_mt");
jbe@243	2141 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_TLSCONF_MT_REGKEY);
jbe@243	2142 #endif
jbe@243	2143
jbe@205	2144 moonbr_io_pushhandle(L, 0);
jbe@205	2145 lua_setfield(L, -2, "stdin");
jbe@205	2146 moonbr_io_pushhandle(L, 1);
jbe@205	2147 lua_setfield(L, -2, "stdout");
jbe@205	2148 moonbr_io_pushhandle(L, 2);
jbe@205	2149 lua_setfield(L, -2, "stderr");
jbe@270	2150
jbe@79	2151 luaL_setfuncs(L, moonbr_io_module_funcs, 0);
jbe@79	2152 return 1;
jbe@79	2153
jbe@79	2154 }
jbe@79	2155