moonbridge

annotate moonbridge_io.c @ 254:cda7988daafe

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

Impressum / About Us