moonbridge

annotate moonbridge_io.c @ 257:77a7ca00ad34

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

Impressum / About Us