moonbridge: moonbridge

moonbridge

annotate moonbridge_io.c @ 256:ff7eea59be93

Bugfix: Do not throw (hard) Lua error in moonbr_io_handle_set_nonblocking, moonbr_io_handle_set_linger, and moonbr_io_handle_set_nopush (may cause Lua error on connection reset)

author	jbe
date	Sun Sep 18 00:59:31 2016 +0200 (2016-09-18)
parents	a22a6c19457f
children	77a7ca00ad34

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