moonbridge

annotate moonbridge_io.c @ 265:37aaca00941c

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

Impressum / About Us