moonbridge

annotate moonbridge_io.c @ 290:70f047d32a0f

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

Impressum / About Us