moonbridge

annotate moonbridge_io.c @ 314:1b459e9c12c9

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

Impressum / About Us