moonbridge

annotate moonbridge_io.c @ 260:1dc729343830

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

Impressum / About Us