moonbridge

annotate moonbridge_io.c @ 250:28bffa2af1ec

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

Impressum / About Us