moonbridge

annotate moonbridge_io.c @ 117:d8469c1039e6

find changesets by author, revision, files, or words in the commit message
Bugfixes regarding treatment of memory allocation errors and closing file descriptors
author jbe
date Fri Apr 10 02:40:43 2015 +0200 (2015-04-10)
parents 2ab19e15aa6b
children 118e320a7812
rev   line source
jbe@79 1
jbe@79 2 #include <stdlib.h>
jbe@79 3 #include <unistd.h>
jbe@79 4 #include <stdint.h>
jbe@79 5 #include <errno.h>
jbe@79 6 #include <string.h>
jbe@79 7 #include <sys/socket.h>
jbe@111 8 #include <sys/un.h>
jbe@79 9 #include <sys/select.h>
jbe@81 10 #include <fcntl.h>
jbe@95 11 #include <netinet/in.h>
jbe@95 12 #include <netinet/tcp.h>
jbe@107 13 #include <arpa/inet.h>
jbe@98 14 #include <sys/types.h>
jbe@98 15 #include <netdb.h>
jbe@113 16 #include <signal.h>
jbe@115 17 #include <time.h>
jbe@79 18
jbe@79 19 #include <lua.h>
jbe@79 20 #include <lauxlib.h>
jbe@79 21 #include <lualib.h>
jbe@79 22
jbe@80 23 #define MOONBR_IO_MAXSTRERRORLEN 80
jbe@85 24 #define MOONBR_IO_READBUFLEN 4096
jbe@80 25 #define MOONBR_IO_WRITEBUFLEN 4096
jbe@80 26
jbe@108 27 #define MOONBR_IO_LISTEN_BACKLOG 1024
jbe@108 28
jbe@80 29 #define moonbr_io_errmsg() \
jbe@80 30 char errmsg[MOONBR_IO_MAXSTRERRORLEN]; \
jbe@80 31 strerror_r(errno, errmsg, MOONBR_IO_MAXSTRERRORLEN)
jbe@80 32
jbe@79 33 #define MOONBR_IO_HANDLE_MT_REGKEY "moonbridge_io_handle"
jbe@79 34 #define MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY "moonbridge_io_handle_public"
jbe@108 35 #define MOONBR_IO_LISTENER_MT_REGKEY "moonbridge_io_listener"
jbe@79 36
jbe@79 37 typedef struct {
jbe@79 38 int fd;
jbe@107 39 int issock;
jbe@107 40 sa_family_t addrfam;
jbe@94 41 int finished;
jbe@94 42 int closed;
jbe@81 43 int nonblocking;
jbe@95 44 int nopush;
jbe@85 45 int readerr;
jbe@105 46 int readbufin;
jbe@105 47 int readbufout;
jbe@81 48 int writeerr;
jbe@83 49 size_t writeleft;
jbe@83 50 #if LUA_VERSION_NUM >= 503
jbe@83 51 lua_Integer writeqin;
jbe@83 52 lua_Integer writeqout;
jbe@83 53 #else
jbe@83 54 int writeqin;
jbe@83 55 int writeqout;
jbe@83 56 #endif
jbe@83 57 size_t writeqoff;
jbe@103 58 int writebufin;
jbe@103 59 int writebufout;
jbe@85 60 char readbuf[MOONBR_IO_READBUFLEN];
jbe@80 61 char writebuf[MOONBR_IO_WRITEBUFLEN];
jbe@79 62 } moonbr_io_handle_t;
jbe@79 63
jbe@108 64 typedef struct {
jbe@108 65 int fd;
jbe@108 66 int nonblocking;
jbe@108 67 } moonbr_io_listener_t;
jbe@108 68
jbe@81 69 static void moonbr_io_handle_set_nonblocking(lua_State *L, moonbr_io_handle_t *handle, int nonblocking) {
jbe@95 70 int flags;
jbe@95 71 if (handle->nonblocking == nonblocking) return;
jbe@95 72 flags = fcntl(handle->fd, F_GETFL, 0);
jbe@95 73 if (flags == -1) {
jbe@95 74 moonbr_io_errmsg();
jbe@96 75 close(handle->fd);
jbe@96 76 handle->fd = -1;
jbe@96 77 handle->closed = 1;
jbe@95 78 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@81 79 }
jbe@95 80 if (nonblocking) flags |= O_NONBLOCK;
jbe@95 81 else flags &= ~O_NONBLOCK;
jbe@95 82 if (fcntl(handle->fd, F_SETFL, flags) == -1) {
jbe@95 83 moonbr_io_errmsg();
jbe@96 84 close(handle->fd);
jbe@96 85 handle->fd = -1;
jbe@96 86 handle->closed = 1;
jbe@95 87 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@95 88 }
jbe@95 89 handle->nonblocking = nonblocking;
jbe@81 90 }
jbe@81 91
jbe@87 92 static void moonbr_io_handle_set_linger(lua_State *L, moonbr_io_handle_t *handle, int timeout) {
jbe@87 93 struct linger lingerval = { 0, };
jbe@107 94 if (!handle->issock) return;
jbe@87 95 if (timeout >= 0) {
jbe@87 96 lingerval.l_onoff = 1;
jbe@87 97 lingerval.l_linger = timeout;
jbe@87 98 }
jbe@87 99 if (setsockopt(handle->fd, SOL_SOCKET, SO_LINGER, &lingerval, sizeof(lingerval))) {
jbe@87 100 moonbr_io_errmsg();
jbe@96 101 close(handle->fd);
jbe@96 102 handle->fd = -1;
jbe@96 103 handle->closed = 1;
jbe@95 104 luaL_error(L, "Unexpected error while setting SO_LINGER with setsockopt: %s", errmsg);
jbe@87 105 }
jbe@87 106 }
jbe@87 107
jbe@95 108 static void moonbr_io_handle_set_nopush(lua_State *L, moonbr_io_handle_t *handle, int nopush) {
jbe@96 109 #if defined(TCP_NOPUSH) || defined(TCP_CORK)
jbe@107 110 if (
jbe@107 111 !(handle->addrfam == AF_INET6 || handle->addrfam == AF_INET) ||
jbe@107 112 handle->nopush == nopush
jbe@107 113 ) return;
jbe@96 114 #if defined(TCP_NOPUSH)
jbe@96 115 if (setsockopt(handle->fd, IPPROTO_TCP, TCP_NOPUSH, &nopush, sizeof(nopush))) {
jbe@104 116 #elif defined(TCP_CORK)
jbe@104 117 if (setsockopt(handle->fd, IPPROTO_TCP, TCP_CORK, &nopush, sizeof(nopush))) {
jbe@104 118 #endif
jbe@96 119 moonbr_io_errmsg();
jbe@96 120 close(handle->fd);
jbe@96 121 handle->fd = -1;
jbe@96 122 handle->closed = 1;
jbe@104 123 #if defined(TCP_NOPUSH)
jbe@96 124 luaL_error(L, "Unexpected error while setting TCP_NOPUSH with setsockopt: %s", errmsg);
jbe@96 125 #elif defined(TCP_CORK)
jbe@95 126 luaL_error(L, "Unexpected error while setting TCP_CORK with setsockopt: %s", errmsg);
jbe@104 127 #endif
jbe@95 128 }
jbe@95 129 handle->nopush = nopush;
jbe@96 130 #else
jbe@96 131 #warning Neither TCP_NOPUSH nor TCP_CORK is available
jbe@96 132 #endif
jbe@95 133 }
jbe@95 134
jbe@86 135 static int moonbr_io_read_impl(lua_State *L, int nonblocking, int drain) {
jbe@85 136 moonbr_io_handle_t *handle;
jbe@85 137 lua_Integer maxread;
jbe@85 138 const char *terminatorstr;
jbe@85 139 size_t terminatorlen;
jbe@85 140 char terminator;
jbe@85 141 luaL_Buffer luabuf;
jbe@85 142 size_t luabufcnt = 0;
jbe@105 143 int remaining;
jbe@85 144 char *terminatorpos;
jbe@103 145 ssize_t bytesread;
jbe@85 146 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@85 147 maxread = luaL_optinteger(L, 2, 0);
jbe@85 148 terminatorstr = luaL_optlstring(L, 3, "", &terminatorlen);
jbe@85 149 if (terminatorlen) {
jbe@85 150 luaL_argcheck(L, terminatorlen == 1, 3, "single byte expected");
jbe@85 151 terminator = terminatorstr[0];
jbe@85 152 }
jbe@86 153 lua_settop(L, 1); /* return handle on drain, terminator string may be garbage collected */
jbe@94 154 if (handle->closed) luaL_error(L, "Attempt to read from a closed I/O handle");
jbe@85 155 if (handle->readerr) {
jbe@85 156 lua_pushnil(L);
jbe@85 157 lua_pushliteral(L, "Previous read error");
jbe@85 158 return 2;
jbe@85 159 }
jbe@105 160 handle->readerr = 1;
jbe@104 161 if (handle->fd < 0) goto moonbr_io_read_impl_eof; /* fake EOF to simulate shutdown */
jbe@85 162 moonbr_io_handle_set_nonblocking(L, handle, nonblocking);
jbe@86 163 if (!drain) luaL_buffinit(L, &luabuf);
jbe@85 164 while (1) {
jbe@105 165 remaining = -1;
jbe@105 166 if (
jbe@105 167 maxread > 0 &&
jbe@105 168 handle->readbufin - handle->readbufout >= (size_t)maxread - luabufcnt
jbe@105 169 ) {
jbe@105 170 remaining = maxread - luabufcnt;
jbe@85 171 } else if (terminatorlen) {
jbe@105 172 terminatorpos = memchr(
jbe@105 173 handle->readbuf + handle->readbufout,
jbe@105 174 terminator,
jbe@105 175 handle->readbufin - handle->readbufout
jbe@105 176 );
jbe@105 177 if (terminatorpos) remaining = 1 + (
jbe@105 178 terminatorpos - (handle->readbuf + handle->readbufout)
jbe@105 179 );
jbe@85 180 }
jbe@105 181 if (remaining >= 0) {
jbe@86 182 if (!drain) {
jbe@105 183 luaL_addlstring(
jbe@105 184 &luabuf,
jbe@105 185 handle->readbuf + handle->readbufout,
jbe@105 186 remaining
jbe@105 187 );
jbe@86 188 luaL_pushresult(&luabuf);
jbe@90 189 } else {
jbe@105 190 luaL_pushresult(&luabuf);
jbe@105 191 lua_pop(L, 1);
jbe@105 192 lua_pushinteger(L, luabufcnt + remaining);
jbe@86 193 }
jbe@105 194 handle->readbufout += remaining;
jbe@105 195 if (handle->readbufout == handle->readbufin) {
jbe@105 196 handle->readbufin = 0;
jbe@105 197 handle->readbufout =0;
jbe@105 198 }
jbe@104 199 handle->readerr = 0;
jbe@85 200 return 1;
jbe@85 201 }
jbe@105 202 if (!drain) luaL_addlstring(
jbe@105 203 &luabuf,
jbe@105 204 handle->readbuf + handle->readbufout,
jbe@105 205 handle->readbufin - handle->readbufout
jbe@105 206 );
jbe@105 207 luabufcnt += handle->readbufin - handle->readbufout;
jbe@85 208 do {
jbe@103 209 bytesread = read(handle->fd, handle->readbuf, MOONBR_IO_READBUFLEN);
jbe@103 210 } while (bytesread < 0 && (errno == EINTR));
jbe@105 211 if (
jbe@105 212 bytesread == 0 || (
jbe@105 213 nonblocking &&
jbe@105 214 bytesread < 0 && (errno == EAGAIN || errno == EWOULDBLOCK)
jbe@105 215 )
jbe@105 216 ) {
jbe@105 217 handle->readbufin = 0;
jbe@105 218 handle->readbufout = 0;
jbe@105 219 if (!drain) {
jbe@105 220 luaL_pushresult(&luabuf);
jbe@105 221 if (!luabufcnt && bytesread == 0) {
jbe@105 222 lua_pop(L, 1);
jbe@105 223 moonbr_io_read_impl_eof:
jbe@105 224 lua_pushboolean(L, 0);
jbe@105 225 lua_pushliteral(L, "End of file");
jbe@105 226 handle->readerr = 0;
jbe@105 227 return 2;
jbe@105 228 }
jbe@105 229 } else {
jbe@105 230 if (!luabufcnt && bytesread == 0) lua_pushboolean(L, 1);
jbe@105 231 else lua_pushboolean(L, luabufcnt);
jbe@105 232 }
jbe@105 233 handle->readerr = 0;
jbe@105 234 return 1;
jbe@105 235 }
jbe@103 236 if (bytesread < 0) {
jbe@85 237 moonbr_io_errmsg();
jbe@85 238 lua_pushnil(L);
jbe@85 239 lua_pushstring(L, errmsg);
jbe@85 240 return 2;
jbe@85 241 }
jbe@105 242 handle->readbufin = bytesread;
jbe@105 243 handle->readbufout = 0;
jbe@86 244 }
jbe@85 245 }
jbe@85 246
jbe@85 247 static int moonbr_io_read(lua_State *L) {
jbe@86 248 return moonbr_io_read_impl(L, 0, 0);
jbe@85 249 }
jbe@85 250
jbe@85 251 static int moonbr_io_read_nb(lua_State *L) {
jbe@86 252 return moonbr_io_read_impl(L, 1, 0);
jbe@86 253 }
jbe@86 254
jbe@86 255 static int moonbr_io_drain(lua_State *L) {
jbe@86 256 return moonbr_io_read_impl(L, 0, 1);
jbe@86 257 }
jbe@86 258
jbe@86 259 static int moonbr_io_drain_nb(lua_State *L) {
jbe@86 260 return moonbr_io_read_impl(L, 1, 1);
jbe@85 261 }
jbe@85 262
jbe@81 263 static int moonbr_io_write_impl(lua_State *L, int nonblocking, int flush) {
jbe@80 264 moonbr_io_handle_t *handle;
jbe@80 265 int i, top;
jbe@80 266 const char *str;
jbe@92 267 size_t strlen;
jbe@103 268 ssize_t written;
jbe@80 269 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@94 270 if (handle->closed) luaL_error(L, "Attempt to write to a closed I/O handle");
jbe@94 271 if (handle->finished) luaL_error(L, "Attempt to write to a finished I/O handle");
jbe@81 272 if (handle->writeerr) {
jbe@80 273 lua_pushnil(L);
jbe@80 274 lua_pushliteral(L, "Previous write error");
jbe@80 275 return 2;
jbe@80 276 }
jbe@103 277 handle->writeerr = 1;
jbe@81 278 moonbr_io_handle_set_nonblocking(L, handle, nonblocking);
jbe@84 279 top = lua_gettop(L);
jbe@81 280 lua_getuservalue(L, 1);
jbe@103 281 lua_getfield(L, -1, "writequeue");
jbe@84 282 for (i=2; i<=top; i++) {
jbe@84 283 luaL_checklstring(L, i, &strlen);
jbe@84 284 lua_pushvalue(L, i);
jbe@84 285 lua_rawseti(L, -2, handle->writeqin++);
jbe@84 286 handle->writeleft += strlen;
jbe@81 287 }
jbe@83 288 while (handle->writeqout != handle->writeqin) {
jbe@83 289 lua_rawgeti(L, -1, handle->writeqout);
jbe@81 290 str = lua_tolstring(L, -1, &strlen);
jbe@92 291 while (handle->writeqoff < strlen) {
jbe@103 292 if (
jbe@103 293 strlen - handle->writeqoff <=
jbe@103 294 MOONBR_IO_WRITEBUFLEN - handle->writebufin
jbe@103 295 ) {
jbe@103 296 memcpy(
jbe@103 297 handle->writebuf + handle->writebufin,
jbe@103 298 str + handle->writeqoff,
jbe@103 299 strlen - handle->writeqoff
jbe@103 300 );
jbe@103 301 handle->writebufin += strlen - handle->writeqoff;
jbe@80 302 break;
jbe@80 303 } else {
jbe@97 304 moonbr_io_handle_set_nopush(L, handle, 1);
jbe@103 305 memcpy(
jbe@103 306 handle->writebuf + handle->writebufin,
jbe@103 307 str + handle->writeqoff,
jbe@103 308 MOONBR_IO_WRITEBUFLEN - handle->writebufin
jbe@103 309 );
jbe@103 310 handle->writeqoff += MOONBR_IO_WRITEBUFLEN - handle->writebufin;
jbe@103 311 while (handle->writebufout < MOONBR_IO_WRITEBUFLEN) {
jbe@103 312 written = write(
jbe@103 313 handle->fd,
jbe@103 314 handle->writebuf + handle->writebufout,
jbe@103 315 MOONBR_IO_WRITEBUFLEN - handle->writebufout
jbe@103 316 );
jbe@103 317 if (written < 0) {
jbe@81 318 if (nonblocking && (errno == EAGAIN || errno == EWOULDBLOCK)) {
jbe@81 319 goto moonbr_io_write_impl_block;
jbe@81 320 } else if (errno != EINTR) {
jbe@80 321 moonbr_io_errmsg();
jbe@80 322 lua_pushnil(L);
jbe@80 323 lua_pushstring(L, errmsg);
jbe@80 324 return 2;
jbe@80 325 }
jbe@103 326 } else {
jbe@103 327 handle->writebufout += written;
jbe@103 328 handle->writeleft -= written;
jbe@80 329 }
jbe@80 330 }
jbe@103 331 handle->writebufin = 0;
jbe@103 332 handle->writebufout = 0;
jbe@80 333 }
jbe@80 334 }
jbe@81 335 handle->writeqoff = 0;
jbe@81 336 lua_pop(L, 1);
jbe@81 337 lua_pushnil(L);
jbe@83 338 lua_rawseti(L, -2, handle->writeqout++);
jbe@80 339 }
jbe@81 340 if (flush) {
jbe@97 341 moonbr_io_handle_set_nopush(L, handle, 0);
jbe@103 342 while (handle->writebufout < handle->writebufin) {
jbe@103 343 written = write(
jbe@103 344 handle->fd,
jbe@103 345 handle->writebuf + handle->writebufout,
jbe@103 346 handle->writebufin - handle->writebufout
jbe@103 347 );
jbe@103 348 if (written < 0) {
jbe@81 349 if (nonblocking && (errno == EAGAIN || errno == EWOULDBLOCK)) {
jbe@81 350 goto moonbr_io_write_impl_block;
jbe@81 351 } else if (errno != EINTR) {
jbe@81 352 moonbr_io_errmsg();
jbe@81 353 lua_pushnil(L);
jbe@81 354 lua_pushstring(L, errmsg);
jbe@81 355 return 2;
jbe@81 356 }
jbe@81 357 } else {
jbe@103 358 handle->writebufout += written;
jbe@103 359 handle->writeleft -= written;
jbe@81 360 }
jbe@81 361 }
jbe@103 362 handle->writebufin = 0;
jbe@103 363 handle->writebufout = 0;
jbe@81 364 }
jbe@103 365 if (nonblocking) lua_pushinteger(L, 0);
jbe@103 366 else lua_pushvalue(L, 1);
jbe@103 367 handle->writeerr = 0;
jbe@80 368 return 1;
jbe@81 369 moonbr_io_write_impl_block:
jbe@91 370 lua_pushinteger(L, handle->writeleft);
jbe@103 371 handle->writeerr = 0;
jbe@81 372 return 1;
jbe@81 373 }
jbe@81 374
jbe@81 375 static int moonbr_io_write(lua_State *L) {
jbe@81 376 return moonbr_io_write_impl(L, 0, 0);
jbe@81 377 }
jbe@81 378
jbe@81 379 static int moonbr_io_write_nb(lua_State *L) {
jbe@81 380 return moonbr_io_write_impl(L, 1, 0);
jbe@80 381 }
jbe@80 382
jbe@80 383 static int moonbr_io_flush(lua_State *L) {
jbe@81 384 return moonbr_io_write_impl(L, 0, 1);
jbe@81 385 }
jbe@81 386
jbe@81 387 static int moonbr_io_flush_nb(lua_State *L) {
jbe@81 388 return moonbr_io_write_impl(L, 1, 1);
jbe@80 389 }
jbe@80 390
jbe@88 391 static int moonbr_io_finish(lua_State *L) {
jbe@88 392 moonbr_io_handle_t *handle;
jbe@88 393 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@94 394 if (handle->closed) luaL_error(L, "Attempt to finish a closed I/O handle");
jbe@94 395 if (handle->finished) luaL_error(L, "Attempt to finish a finished I/O handle");
jbe@94 396 if (handle->writeleft) {
jbe@94 397 lua_pushcfunction(L, moonbr_io_flush);
jbe@94 398 lua_pushvalue(L, 1);
jbe@116 399 if (lua_pcall(L, 1, 2, 0)) {
jbe@116 400 handle->finished = 1;
jbe@116 401 lua_error(L);
jbe@116 402 }
jbe@94 403 if (!lua_toboolean(L, -2)) {
jbe@94 404 handle->finished = 1;
jbe@94 405 return 2;
jbe@88 406 }
jbe@94 407 }
jbe@94 408 handle->finished = 1;
jbe@107 409 if (handle->addrfam == AF_INET6 || handle->addrfam == AF_INET) {
jbe@88 410 if (shutdown(handle->fd, SHUT_WR)) {
jbe@88 411 moonbr_io_errmsg();
jbe@88 412 lua_pushnil(L);
jbe@88 413 lua_pushstring(L, errmsg);
jbe@88 414 return 2;
jbe@88 415 }
jbe@94 416 } else {
jbe@94 417 if (close(handle->fd)) {
jbe@94 418 moonbr_io_errmsg();
jbe@94 419 handle->fd = -1;
jbe@94 420 lua_pushnil(L);
jbe@94 421 lua_pushstring(L, errmsg);
jbe@94 422 return 2;
jbe@94 423 }
jbe@94 424 handle->fd = -1; /* fake EOF on read */
jbe@88 425 }
jbe@88 426 lua_pushboolean(L, 1);
jbe@88 427 return 1;
jbe@88 428 }
jbe@88 429
jbe@94 430 static int moonbr_io_close_impl(lua_State *L, int reset) {
jbe@83 431 moonbr_io_handle_t *handle;
jbe@83 432 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@94 433 if (handle->closed) luaL_error(L, "Attempt to close a closed I/O handle");
jbe@116 434 handle->closed = 1;
jbe@94 435 if (!reset) {
jbe@87 436 if (handle->writeleft) {
jbe@87 437 lua_pushcfunction(L, moonbr_io_flush);
jbe@87 438 lua_pushvalue(L, 1);
jbe@116 439 if (lua_pcall(L, 1, 2, 0)) {
jbe@116 440 close(handle->fd);
jbe@116 441 handle->fd = -1;
jbe@116 442 lua_error(L);
jbe@116 443 }
jbe@87 444 if (!lua_toboolean(L, -2)) {
jbe@87 445 close(handle->fd);
jbe@87 446 handle->fd = -1;
jbe@87 447 return 2;
jbe@87 448 }
jbe@83 449 }
jbe@94 450 moonbr_io_handle_set_linger(L, handle, -1);
jbe@83 451 }
jbe@94 452 if (handle->fd >= 0) {
jbe@94 453 if (close(handle->fd)) {
jbe@94 454 moonbr_io_errmsg();
jbe@94 455 handle->fd = -1;
jbe@94 456 lua_pushnil(L);
jbe@94 457 lua_pushstring(L, errmsg);
jbe@94 458 return 2;
jbe@94 459 }
jbe@104 460 handle->fd = -1;
jbe@83 461 }
jbe@83 462 lua_pushboolean(L, 1);
jbe@83 463 return 1;
jbe@84 464
jbe@83 465 }
jbe@83 466
jbe@94 467 static int moonbr_io_close(lua_State *L) {
jbe@94 468 return moonbr_io_close_impl(L, 0);
jbe@94 469 }
jbe@94 470
jbe@84 471 static int moonbr_io_reset(lua_State *L) {
jbe@94 472 return moonbr_io_close_impl(L, 1);
jbe@84 473 }
jbe@84 474
jbe@108 475 static int moonbr_io_handlegc(lua_State *L) {
jbe@88 476 moonbr_io_handle_t *handle;
jbe@88 477 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@88 478 if (handle->fd >= 0) {
jbe@88 479 lua_pushcfunction(L, moonbr_io_close);
jbe@88 480 lua_pushvalue(L, 1);
jbe@88 481 lua_pushinteger(L, 0);
jbe@88 482 lua_call(L, 2, 0);
jbe@88 483 }
jbe@88 484 return 0;
jbe@88 485 }
jbe@88 486
jbe@100 487 void moonbr_io_closehandle(lua_State *L, int idx, int reset) {
jbe@88 488 moonbr_io_handle_t *handle;
jbe@88 489 handle = luaL_checkudata(L, idx, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@100 490 if (!handle->closed) {
jbe@100 491 lua_pushcfunction(L, reset ? moonbr_io_reset : moonbr_io_close);
jbe@100 492 lua_pushvalue(L, idx < 0 ? idx-1 : idx);
jbe@100 493 lua_call(L, 1, 0);
jbe@88 494 }
jbe@88 495 }
jbe@88 496
jbe@107 497 void moonbr_io_pushhandle(lua_State *L, int fd) {
jbe@79 498 moonbr_io_handle_t *handle;
jbe@107 499 struct sockaddr addr;
jbe@107 500 socklen_t addrlen;
jbe@79 501 handle = lua_newuserdata(L, sizeof(moonbr_io_handle_t));
jbe@79 502 handle->fd = fd;
jbe@107 503 addrlen = sizeof(addr);
jbe@107 504 if (getsockname(fd, &addr, &addrlen)) {
jbe@107 505 if (errno != ENOTSOCK) {
jbe@107 506 moonbr_io_errmsg();
jbe@107 507 luaL_error(L, "Unexpected error when examining socket: %s", errmsg);
jbe@107 508 }
jbe@107 509 handle->issock = 0;
jbe@107 510 } else {
jbe@107 511 handle->issock = 1;
jbe@107 512 handle->addrfam = addr.sa_family;
jbe@107 513 }
jbe@94 514 handle->finished = 0;
jbe@94 515 handle->closed = 0;
jbe@81 516 handle->nonblocking = -1;
jbe@95 517 handle->nopush = -1;
jbe@85 518 handle->readerr = 0;
jbe@105 519 handle->readbufin = 0;
jbe@105 520 handle->readbufout = 0;
jbe@81 521 handle->writeerr = 0;
jbe@81 522 handle->writeleft = 0;
jbe@83 523 handle->writeqin = 0;
jbe@83 524 handle->writeqout = 0;
jbe@81 525 handle->writeqoff = 0;
jbe@103 526 handle->writebufin = 0;
jbe@103 527 handle->writebufout = 0;
jbe@87 528 moonbr_io_handle_set_linger(L, handle, 0);
jbe@79 529 luaL_getmetatable(L, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79 530 lua_setmetatable(L, -2);
jbe@79 531 lua_newtable(L); // uservalue
jbe@81 532 lua_newtable(L);
jbe@103 533 lua_setfield(L, -2, "writequeue");
jbe@79 534 lua_newtable(L); // public
jbe@107 535 if (handle->addrfam == AF_INET6) {
jbe@107 536 struct sockaddr_in6 addr_in6;
jbe@107 537 char addrstrbuf[INET6_ADDRSTRLEN];
jbe@107 538 const char *addrstr;
jbe@107 539 addrlen = sizeof(addr_in6);
jbe@107 540 if (getsockname(fd, (struct sockaddr *)&addr_in6, &addrlen)) {
jbe@107 541 moonbr_io_errmsg();
jbe@107 542 luaL_error(L, "Could not determine local IP address/port: %s", errmsg);
jbe@107 543 }
jbe@107 544 if (addrlen > sizeof(addr_in6)) {
jbe@107 545 luaL_error(L, "Could not determine local IP address/port: buffer size exceeded");
jbe@107 546 }
jbe@107 547 addrstr = inet_ntop(AF_INET6, addr_in6.sin6_addr.s6_addr, addrstrbuf, sizeof(addrstrbuf));
jbe@107 548 if (!addrstr) {
jbe@107 549 moonbr_io_errmsg();
jbe@107 550 luaL_error(L, "Could not format local IP address: %s", errmsg);
jbe@107 551 } else {
jbe@107 552 lua_pushstring(L, addrstr);
jbe@107 553 lua_setfield(L, -2, "local_ip6");
jbe@107 554 }
jbe@107 555 lua_pushinteger(L, ntohs(addr_in6.sin6_port));
jbe@107 556 lua_setfield(L, -2, "local_tcpport");
jbe@107 557 if (getpeername(fd, (struct sockaddr *)&addr_in6, &addrlen)) {
jbe@107 558 moonbr_io_errmsg();
jbe@107 559 luaL_error(L, "Could not determine remote IP address/port: %s", errmsg);
jbe@107 560 }
jbe@107 561 if (addrlen > sizeof(addr_in6)) {
jbe@107 562 luaL_error(L, "Could not determine remote IP address/port: buffer size exceeded");
jbe@107 563 }
jbe@107 564 addrstr = inet_ntop(AF_INET6, addr_in6.sin6_addr.s6_addr, addrstrbuf, sizeof(addrstrbuf));
jbe@107 565 if (!addrstr) {
jbe@107 566 moonbr_io_errmsg();
jbe@107 567 luaL_error(L, "Could not format remote IP address: %s", errmsg);
jbe@107 568 } else {
jbe@107 569 lua_pushstring(L, addrstr);
jbe@107 570 lua_setfield(L, -2, "remote_ip6");
jbe@107 571 }
jbe@107 572 lua_pushinteger(L, ntohs(addr_in6.sin6_port));
jbe@107 573 lua_setfield(L, -2, "remote_tcpport");
jbe@107 574 } else if (handle->addrfam == AF_INET) {
jbe@107 575 struct sockaddr_in addr_in;
jbe@107 576 char addrstrbuf[INET_ADDRSTRLEN];
jbe@107 577 const char *addrstr;
jbe@107 578 addrlen = sizeof(addr_in);
jbe@107 579 if (getsockname(fd, (struct sockaddr *)&addr_in, &addrlen)) {
jbe@107 580 moonbr_io_errmsg();
jbe@107 581 luaL_error(L, "Could not determine local IP address/port: %s", errmsg);
jbe@107 582 }
jbe@107 583 if (addrlen > sizeof(addr_in)) {
jbe@107 584 luaL_error(L, "Could not determine local IP address/port: buffer size exceeded");
jbe@107 585 }
jbe@107 586 addrstr = inet_ntop(AF_INET, &addr_in.sin_addr.s_addr, addrstrbuf, sizeof(addrstrbuf));
jbe@107 587 if (!addrstr) {
jbe@107 588 moonbr_io_errmsg();
jbe@107 589 luaL_error(L, "Could not format local IP address: %s", errmsg);
jbe@107 590 } else {
jbe@107 591 lua_pushstring(L, addrstr);
jbe@107 592 lua_setfield(L, -2, "local_ip4");
jbe@107 593 }
jbe@107 594 lua_pushinteger(L, ntohs(addr_in.sin_port));
jbe@107 595 lua_setfield(L, -2, "local_tcpport");
jbe@107 596 if (getpeername(fd, (struct sockaddr *)&addr_in, &addrlen)) {
jbe@107 597 moonbr_io_errmsg();
jbe@107 598 luaL_error(L, "Could not determine remote IP address/port: %s", errmsg);
jbe@107 599 }
jbe@107 600 if (addrlen > sizeof(addr_in)) {
jbe@107 601 luaL_error(L, "Could not determine remote IP address/port: buffer size exceeded");
jbe@107 602 }
jbe@107 603 addrstr = inet_ntop(AF_INET, &addr_in.sin_addr.s_addr, addrstrbuf, sizeof(addrstrbuf));
jbe@107 604 if (!addrstr) {
jbe@107 605 moonbr_io_errmsg();
jbe@107 606 luaL_error(L, "Could not format remote IP address: %s", errmsg);
jbe@107 607 } else {
jbe@107 608 lua_pushstring(L, addrstr);
jbe@107 609 lua_setfield(L, -2, "remote_ip4");
jbe@107 610 }
jbe@107 611 lua_pushinteger(L, ntohs(addr_in.sin_port));
jbe@107 612 lua_setfield(L, -2, "remote_tcpport");
jbe@107 613 }
jbe@79 614 luaL_getmetatable(L, MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY);
jbe@79 615 lua_setmetatable(L, -2);
jbe@79 616 lua_setfield(L, -2, "public");
jbe@79 617 lua_setuservalue(L, -2);
jbe@79 618 }
jbe@79 619
jbe@79 620 static int moonbr_io_handleindex(lua_State *L) {
jbe@80 621 luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79 622 lua_getuservalue(L, 1);
jbe@79 623 lua_getfield(L, -1, "public");
jbe@79 624 lua_pushvalue(L, 2);
jbe@79 625 lua_gettable(L, -2);
jbe@79 626 return 1;
jbe@79 627 }
jbe@79 628
jbe@79 629 static int moonbr_io_handlenewindex(lua_State *L) {
jbe@80 630 luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79 631 lua_getuservalue(L, 1);
jbe@79 632 lua_getfield(L, -1, "public");
jbe@79 633 lua_pushvalue(L, 2);
jbe@79 634 lua_pushvalue(L, 3);
jbe@79 635 lua_settable(L, -3);
jbe@79 636 return 0;
jbe@79 637 }
jbe@79 638
jbe@111 639 static int moonbr_io_localconnect_impl(lua_State *L, int nonblocking) {
jbe@111 640 const char *path;
jbe@111 641 struct sockaddr_un sockaddr = { .sun_family = AF_LOCAL };
jbe@111 642 const int path_maxlen = sizeof(struct sockaddr_un) - (
jbe@111 643 (void *)sockaddr.sun_path - (void *)&sockaddr
jbe@111 644 ) - 1; /* one byte for termination */
jbe@111 645 int sock;
jbe@111 646 path = luaL_checkstring(L, 1);
jbe@111 647 if (strlen(path) > path_maxlen) luaL_error(L, "Path too long; only %i characters allowed", path_maxlen);
jbe@111 648 strcpy(sockaddr.sun_path, path);
jbe@111 649 sock = socket(
jbe@111 650 PF_LOCAL,
jbe@111 651 SOCK_STREAM | SOCK_CLOEXEC | (nonblocking ? SOCK_NONBLOCK : 0),
jbe@111 652 0
jbe@111 653 );
jbe@111 654 if (sock < 0) {
jbe@111 655 moonbr_io_errmsg();
jbe@111 656 lua_pushnil(L);
jbe@111 657 lua_pushstring(L, errmsg);
jbe@111 658 return 2;
jbe@111 659 }
jbe@111 660 if (connect(sock, (struct sockaddr *)&sockaddr, sizeof(sockaddr))) {
jbe@111 661 if (!nonblocking && errno == EINTR) {
jbe@111 662 moonbr_io_errmsg();
jbe@111 663 close(sock);
jbe@111 664 lua_pushnil(L);
jbe@111 665 lua_pushstring(L, errmsg);
jbe@111 666 return 2;
jbe@111 667 } else if (!(nonblocking && (errno == EINPROGRESS || errno == EINTR))) {
jbe@111 668 moonbr_io_errmsg();
jbe@111 669 lua_pushnil(L);
jbe@111 670 lua_pushstring(L, errmsg);
jbe@111 671 return 2;
jbe@111 672 }
jbe@111 673 }
jbe@111 674 moonbr_io_pushhandle(L, sock);
jbe@111 675 return 1;
jbe@111 676 }
jbe@111 677
jbe@111 678 static int moonbr_io_localconnect(lua_State *L) {
jbe@111 679 return moonbr_io_localconnect_impl(L, 0);
jbe@111 680 }
jbe@111 681
jbe@111 682 static int moonbr_io_localconnect_nb(lua_State *L) {
jbe@111 683 return moonbr_io_localconnect_impl(L, 1);
jbe@111 684 }
jbe@111 685
jbe@99 686 static int moonbr_io_tcpconnect_impl(lua_State *L, int nonblocking) {
jbe@98 687 const char *host, *port;
jbe@98 688 struct addrinfo hints = { 0, };
jbe@98 689 struct addrinfo *res, *addrinfo;
jbe@98 690 int errcode;
jbe@98 691 int sock;
jbe@98 692 host = luaL_checkstring(L, 1);
jbe@98 693 port = luaL_checkstring(L, 2);
jbe@98 694 hints.ai_family = AF_UNSPEC;
jbe@98 695 hints.ai_socktype = SOCK_STREAM;
jbe@98 696 hints.ai_protocol = IPPROTO_TCP;
jbe@98 697 hints.ai_flags = AI_ADDRCONFIG;
jbe@98 698 errcode = getaddrinfo(host, port, &hints, &res);
jbe@98 699 if (errcode) {
jbe@102 700 freeaddrinfo(res);
jbe@98 701 if (errcode == EAI_SYSTEM) {
jbe@98 702 moonbr_io_errmsg();
jbe@98 703 lua_pushnil(L);
jbe@98 704 lua_pushfstring(L, "%s: %s", gai_strerror(errcode), errmsg);
jbe@98 705 } else {
jbe@98 706 lua_pushnil(L);
jbe@98 707 lua_pushstring(L, gai_strerror(errcode));
jbe@98 708 }
jbe@98 709 return 2;
jbe@98 710 }
jbe@98 711 for (addrinfo=res; addrinfo; addrinfo=addrinfo->ai_next) {
jbe@98 712 if (addrinfo->ai_family == PF_INET6) goto moonbr_io_tcpconnect_found;
jbe@98 713 }
jbe@98 714 for (addrinfo=res; addrinfo; addrinfo=addrinfo->ai_next) {
jbe@98 715 if (addrinfo->ai_family == PF_INET) goto moonbr_io_tcpconnect_found;
jbe@98 716 }
jbe@98 717 addrinfo = res;
jbe@98 718 moonbr_io_tcpconnect_found:
jbe@99 719 sock = socket(
jbe@99 720 addrinfo->ai_family,
jbe@99 721 addrinfo->ai_socktype | SOCK_CLOEXEC | (nonblocking ? SOCK_NONBLOCK : 0),
jbe@99 722 addrinfo->ai_protocol
jbe@99 723 );
jbe@98 724 if (sock < 0) {
jbe@98 725 moonbr_io_errmsg();
jbe@108 726 freeaddrinfo(res);
jbe@98 727 lua_pushnil(L);
jbe@98 728 lua_pushstring(L, errmsg);
jbe@108 729 return 2;
jbe@98 730 }
jbe@98 731 if (connect(sock, addrinfo->ai_addr, addrinfo->ai_addrlen)) {
jbe@108 732 freeaddrinfo(res);
jbe@99 733 if (!nonblocking && errno == EINTR) {
jbe@99 734 moonbr_io_errmsg();
jbe@99 735 close(sock);
jbe@99 736 lua_pushnil(L);
jbe@99 737 lua_pushstring(L, errmsg);
jbe@99 738 return 2;
jbe@99 739 } else if (!(nonblocking && (errno == EINPROGRESS || errno == EINTR))) {
jbe@99 740 moonbr_io_errmsg();
jbe@99 741 lua_pushnil(L);
jbe@99 742 lua_pushstring(L, errmsg);
jbe@99 743 return 2;
jbe@99 744 }
jbe@108 745 } else {
jbe@108 746 freeaddrinfo(res);
jbe@98 747 }
jbe@107 748 moonbr_io_pushhandle(L, sock);
jbe@98 749 return 1;
jbe@98 750 }
jbe@98 751
jbe@99 752 static int moonbr_io_tcpconnect(lua_State *L) {
jbe@99 753 return moonbr_io_tcpconnect_impl(L, 0);
jbe@99 754 }
jbe@99 755
jbe@99 756 static int moonbr_io_tcpconnect_nb(lua_State *L) {
jbe@99 757 return moonbr_io_tcpconnect_impl(L, 1);
jbe@99 758 }
jbe@99 759
jbe@112 760 static int moonbr_io_locallisten(lua_State *L) {
jbe@112 761 moonbr_io_listener_t *listener;
jbe@112 762 const char *path;
jbe@112 763 struct sockaddr_un sockaddr = { .sun_family = AF_LOCAL };
jbe@112 764 const int path_maxlen = sizeof(struct sockaddr_un) - (
jbe@112 765 (void *)sockaddr.sun_path - (void *)&sockaddr
jbe@112 766 ) - 1; /* one byte for termination */
jbe@112 767 int sock;
jbe@112 768 path = luaL_checkstring(L, 1);
jbe@112 769 if (strlen(path) > path_maxlen) luaL_error(L, "Path too long; only %i characters allowed", path_maxlen);
jbe@112 770 strcpy(sockaddr.sun_path, path);
jbe@112 771 listener = lua_newuserdata(L, sizeof(moonbr_io_listener_t));
jbe@117 772 listener->fd = -1;
jbe@112 773 luaL_setmetatable(L, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@112 774 sock = socket(
jbe@112 775 PF_LOCAL,
jbe@112 776 SOCK_STREAM | SOCK_CLOEXEC,
jbe@112 777 0
jbe@112 778 );
jbe@112 779 if (sock < 0) {
jbe@112 780 moonbr_io_errmsg();
jbe@112 781 lua_pushnil(L);
jbe@112 782 lua_pushstring(L, errmsg);
jbe@112 783 return 2;
jbe@112 784 }
jbe@112 785 if (bind(sock, (struct sockaddr *)&sockaddr, sizeof(sockaddr))) {
jbe@112 786 moonbr_io_errmsg();
jbe@112 787 close(sock);
jbe@112 788 lua_pushnil(L);
jbe@112 789 lua_pushstring(L, errmsg);
jbe@112 790 return 2;
jbe@112 791 }
jbe@112 792 if (listen(sock, MOONBR_IO_LISTEN_BACKLOG)) {
jbe@112 793 moonbr_io_errmsg();
jbe@112 794 close(sock);
jbe@112 795 lua_pushnil(L);
jbe@112 796 lua_pushstring(L, errmsg);
jbe@112 797 return 2;
jbe@112 798 }
jbe@112 799 listener->fd = sock;
jbe@112 800 listener->nonblocking = -1;
jbe@112 801 return 1;
jbe@112 802 }
jbe@112 803
jbe@108 804 static int moonbr_io_tcplisten(lua_State *L) {
jbe@108 805 moonbr_io_listener_t *listener;
jbe@108 806 const char *host, *port;
jbe@108 807 struct addrinfo hints = { 0, };
jbe@108 808 struct addrinfo *res, *addrinfo;
jbe@108 809 int errcode;
jbe@108 810 int sock;
jbe@108 811 host = luaL_optstring(L, 1, NULL);
jbe@108 812 port = luaL_checkstring(L, 2);
jbe@108 813 listener = lua_newuserdata(L, sizeof(moonbr_io_listener_t));
jbe@117 814 listener->fd = -1;
jbe@108 815 luaL_setmetatable(L, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108 816 hints.ai_family = AF_UNSPEC;
jbe@108 817 hints.ai_socktype = SOCK_STREAM;
jbe@108 818 hints.ai_protocol = IPPROTO_TCP;
jbe@108 819 hints.ai_flags = AI_ADDRCONFIG | AI_PASSIVE;
jbe@108 820 errcode = getaddrinfo(host, port, &hints, &res);
jbe@108 821 if (errcode) {
jbe@108 822 freeaddrinfo(res);
jbe@108 823 if (errcode == EAI_SYSTEM) {
jbe@108 824 moonbr_io_errmsg();
jbe@108 825 lua_pushnil(L);
jbe@108 826 lua_pushfstring(L, "%s: %s", gai_strerror(errcode), errmsg);
jbe@108 827 } else {
jbe@108 828 lua_pushnil(L);
jbe@108 829 lua_pushstring(L, gai_strerror(errcode));
jbe@108 830 }
jbe@108 831 return 2;
jbe@108 832 }
jbe@108 833 for (addrinfo=res; addrinfo; addrinfo=addrinfo->ai_next) {
jbe@108 834 if (addrinfo->ai_family == PF_INET6) goto moonbr_io_tcpconnect_found;
jbe@108 835 }
jbe@108 836 for (addrinfo=res; addrinfo; addrinfo=addrinfo->ai_next) {
jbe@108 837 if (addrinfo->ai_family == PF_INET) goto moonbr_io_tcpconnect_found;
jbe@108 838 }
jbe@108 839 addrinfo = res;
jbe@108 840 moonbr_io_tcpconnect_found:
jbe@108 841 sock = socket(
jbe@108 842 addrinfo->ai_family,
jbe@108 843 addrinfo->ai_socktype | SOCK_CLOEXEC,
jbe@108 844 addrinfo->ai_protocol
jbe@108 845 );
jbe@108 846 if (sock < 0) {
jbe@108 847 moonbr_io_errmsg();
jbe@108 848 freeaddrinfo(res);
jbe@108 849 lua_pushnil(L);
jbe@108 850 lua_pushstring(L, errmsg);
jbe@108 851 return 2;
jbe@108 852 }
jbe@108 853 if (bind(sock, addrinfo->ai_addr, addrinfo->ai_addrlen)) {
jbe@108 854 moonbr_io_errmsg();
jbe@108 855 freeaddrinfo(res);
jbe@108 856 close(sock);
jbe@108 857 lua_pushnil(L);
jbe@108 858 lua_pushstring(L, errmsg);
jbe@108 859 return 2;
jbe@108 860 }
jbe@108 861 freeaddrinfo(res);
jbe@108 862 if (listen(sock, MOONBR_IO_LISTEN_BACKLOG)) {
jbe@108 863 moonbr_io_errmsg();
jbe@108 864 close(sock);
jbe@108 865 lua_pushnil(L);
jbe@108 866 lua_pushstring(L, errmsg);
jbe@108 867 return 2;
jbe@108 868 }
jbe@108 869 listener->fd = sock;
jbe@108 870 listener->nonblocking = -1;
jbe@108 871 return 1;
jbe@108 872 }
jbe@108 873
jbe@108 874 static int moonbr_io_accept_impl(lua_State *L, int nonblocking) {
jbe@108 875 moonbr_io_listener_t *listener;
jbe@108 876 int fd;
jbe@108 877 listener = luaL_checkudata(L, 1, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108 878 if (listener->fd < 0) luaL_error(L, "Attempt to use a closed listener");
jbe@108 879 if (listener->nonblocking != nonblocking) {
jbe@108 880 int flags;
jbe@108 881 flags = fcntl(listener->fd, F_GETFL, 0);
jbe@108 882 if (flags == -1) {
jbe@108 883 moonbr_io_errmsg();
jbe@108 884 close(listener->fd);
jbe@108 885 listener->fd = -1;
jbe@108 886 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@108 887 }
jbe@108 888 if (nonblocking) flags |= O_NONBLOCK;
jbe@108 889 else flags &= ~O_NONBLOCK;
jbe@108 890 if (fcntl(listener->fd, F_SETFL, flags) == -1) {
jbe@108 891 moonbr_io_errmsg();
jbe@108 892 close(listener->fd);
jbe@108 893 listener->fd = -1;
jbe@108 894 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@108 895 }
jbe@108 896 listener->nonblocking = nonblocking;
jbe@108 897 }
jbe@108 898 while (1) {
jbe@108 899 fd = accept4(listener->fd, NULL, NULL, SOCK_CLOEXEC);
jbe@108 900 if (fd < 0) {
jbe@108 901 if (nonblocking && (errno == EAGAIN || errno == EWOULDBLOCK)) {
jbe@108 902 lua_pushboolean(L, 0);
jbe@108 903 lua_pushliteral(L, "No incoming connection pending");
jbe@108 904 return 2;
jbe@108 905 } else if (errno != EINTR) {
jbe@108 906 moonbr_io_errmsg();
jbe@108 907 lua_pushnil(L);
jbe@108 908 lua_pushstring(L, errmsg);
jbe@108 909 return 2;
jbe@108 910 }
jbe@108 911 } else {
jbe@108 912 moonbr_io_pushhandle(L, fd);
jbe@108 913 return 1;
jbe@108 914 }
jbe@108 915 }
jbe@108 916 }
jbe@108 917
jbe@108 918 static int moonbr_io_accept(lua_State *L) {
jbe@108 919 return moonbr_io_accept_impl(L, 0);
jbe@108 920 }
jbe@108 921
jbe@108 922 static int moonbr_io_accept_nb(lua_State *L) {
jbe@108 923 return moonbr_io_accept_impl(L, 1);
jbe@108 924 }
jbe@108 925
jbe@108 926 static int moonbr_io_unlisten(lua_State *L) {
jbe@108 927 moonbr_io_listener_t *listener;
jbe@108 928 listener = luaL_checkudata(L, 1, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108 929 if (listener->fd < 0) luaL_error(L, "Attempt to close a closed listener");
jbe@108 930 if (close(listener->fd)) {
jbe@108 931 moonbr_io_errmsg();
jbe@108 932 listener->fd = -1;
jbe@108 933 lua_pushnil(L);
jbe@108 934 lua_pushstring(L, errmsg);
jbe@108 935 return 2;
jbe@108 936 }
jbe@108 937 listener->fd = -1;
jbe@108 938 lua_pushboolean(L, 1);
jbe@108 939 return 1;
jbe@108 940 }
jbe@108 941
jbe@108 942 static int moonbr_io_listenergc(lua_State *L) {
jbe@108 943 moonbr_io_listener_t *listener;
jbe@108 944 listener = luaL_checkudata(L, 1, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@117 945 if (listener->fd >= 0) close(listener->fd);
jbe@108 946 listener->fd = -1;
jbe@108 947 return 0;
jbe@108 948 }
jbe@108 949
jbe@106 950 static int moonbr_io_poll(lua_State *L) {
jbe@106 951 moonbr_io_handle_t *handle;
jbe@108 952 moonbr_io_listener_t *listener;
jbe@106 953 int fd, isnum;
jbe@106 954 int nfds = 0;
jbe@106 955 fd_set readfds, writefds, exceptfds;
jbe@106 956 struct timeval timeout = {0, };
jbe@106 957 int status;
jbe@106 958 FD_ZERO(&readfds);
jbe@106 959 FD_ZERO(&writefds);
jbe@106 960 FD_ZERO(&exceptfds);
jbe@106 961 if (!lua_isnoneornil(L, 1)) {
jbe@106 962 luaL_checktype(L, 1, LUA_TTABLE);
jbe@106 963 for (lua_pushnil(L); lua_next(L, 1); lua_pop(L, 1)) {
jbe@106 964 if (lua_toboolean(L, -1)) {
jbe@106 965 handle = luaL_testudata(L, -2, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@106 966 if (handle) {
jbe@109 967 if (handle->closed) luaL_error(L, "Attempt to poll a closed connection");
jbe@106 968 fd = handle->fd;
jbe@109 969 if (fd < 0) { /* fake EOF to simulate shutdown */
jbe@109 970 lua_pushboolean(L, 1);
jbe@109 971 return 1;
jbe@109 972 }
jbe@106 973 } else {
jbe@108 974 listener = luaL_testudata(L, -2, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108 975 if (listener) {
jbe@108 976 fd = listener->fd;
jbe@108 977 if (fd < 0) luaL_error(L, "Attempt to poll a closed listener");
jbe@108 978 } else {
jbe@108 979 fd = lua_tointegerx(L, -2, &isnum);
jbe@108 980 if (!isnum) luaL_error(L, "Expected integer (file descriptor), I/O handle, or listener in table key");
jbe@108 981 }
jbe@106 982 }
jbe@106 983 FD_SET(fd, &readfds);
jbe@106 984 if (fd+1 > nfds) nfds = fd+1;
jbe@106 985 }
jbe@106 986 }
jbe@106 987 }
jbe@106 988 if (!lua_isnoneornil(L, 2)) {
jbe@106 989 luaL_checktype(L, 2, LUA_TTABLE);
jbe@106 990 for (lua_pushnil(L); lua_next(L, 2); lua_pop(L, 1)) {
jbe@106 991 if (lua_toboolean(L, -1)) {
jbe@106 992 handle = luaL_testudata(L, -2, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@106 993 if (handle) {
jbe@109 994 if (handle->closed) luaL_error(L, "Attempt to poll a closed connection");
jbe@109 995 if (handle->finished) luaL_error(L, "Attempt to write-poll a finished connection");
jbe@106 996 fd = handle->fd;
jbe@106 997 } else {
jbe@108 998 listener = luaL_testudata(L, -2, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@109 999 if (listener) luaL_error(L, "Attempt to write-poll a listener");
jbe@109 1000 fd = lua_tointegerx(L, -2, &isnum);
jbe@109 1001 if (!isnum) luaL_error(L, "Expected integer (file descriptor) or I/O handle in table key");
jbe@106 1002 }
jbe@106 1003 FD_SET(fd, &writefds);
jbe@106 1004 if (fd+1 > nfds) nfds = fd+1;
jbe@106 1005 }
jbe@106 1006 }
jbe@106 1007 }
jbe@106 1008 if (!lua_isnoneornil(L, 3)) {
jbe@106 1009 lua_Number n;
jbe@106 1010 n = lua_tonumberx(L, 3, &isnum);
jbe@106 1011 if (isnum && n>=0 && n<100000000) {
jbe@106 1012 timeout.tv_sec = n;
jbe@106 1013 timeout.tv_usec = 1e6 * (n - timeout.tv_sec);
jbe@106 1014 } else {
jbe@106 1015 luaL_argcheck(L, 0, 3, "not a valid timeout");
jbe@106 1016 }
jbe@106 1017 status = select(nfds, &readfds, &writefds, &exceptfds, &timeout);
jbe@106 1018 } else {
jbe@106 1019 status = select(nfds, &readfds, &writefds, &exceptfds, NULL);
jbe@106 1020 }
jbe@106 1021 if (status == -1) {
jbe@106 1022 if (errno == EINTR) {
jbe@106 1023 lua_pushboolean(L, 0);
jbe@106 1024 lua_pushliteral(L, "Signal received while polling file descriptors");
jbe@106 1025 return 2;
jbe@106 1026 } else {
jbe@106 1027 moonbr_io_errmsg();
jbe@106 1028 return luaL_error(L, "Unexpected error during \"select\" system call: %s", errmsg);
jbe@106 1029 }
jbe@106 1030 } else if (status == 0) {
jbe@106 1031 lua_pushboolean(L, 0);
jbe@106 1032 lua_pushliteral(L, "Timeout while polling file descriptors");
jbe@106 1033 return 2;
jbe@106 1034 } else {
jbe@106 1035 lua_pushboolean(L, 1);
jbe@106 1036 return 1;
jbe@106 1037 }
jbe@106 1038 }
jbe@106 1039
jbe@115 1040 static int moonbr_io_timeref(lua_State *L) {
jbe@115 1041 lua_Number sub;
jbe@115 1042 struct timespec tp;
jbe@115 1043 sub = luaL_optnumber(L, 1, 0);
jbe@115 1044 if (clock_gettime(CLOCK_MONOTONIC, &tp)) {
jbe@115 1045 return luaL_error(L, "Could not access CLOCK_MONOTONIC");
jbe@115 1046 }
jbe@115 1047 lua_pushnumber(L, tp.tv_sec + tp.tv_nsec / 1.0e9 - sub);
jbe@115 1048 return 1;
jbe@115 1049 }
jbe@115 1050
jbe@79 1051 static const struct luaL_Reg moonbr_io_handle_methods[] = {
jbe@85 1052 {"read", moonbr_io_read},
jbe@85 1053 {"read_nb", moonbr_io_read_nb},
jbe@86 1054 {"drain", moonbr_io_drain},
jbe@86 1055 {"drain_nb", moonbr_io_drain_nb},
jbe@80 1056 {"write", moonbr_io_write},
jbe@81 1057 {"write_nb", moonbr_io_write_nb},
jbe@80 1058 {"flush", moonbr_io_flush},
jbe@81 1059 {"flush_nb", moonbr_io_flush_nb},
jbe@88 1060 {"finish", moonbr_io_finish},
jbe@87 1061 {"close", moonbr_io_close},
jbe@85 1062 {"reset", moonbr_io_reset},
jbe@79 1063 {NULL, NULL}
jbe@79 1064 };
jbe@79 1065
jbe@79 1066 static const struct luaL_Reg moonbr_io_handle_metamethods[] = {
jbe@79 1067 {"__index", moonbr_io_handleindex},
jbe@79 1068 {"__newindex", moonbr_io_handlenewindex},
jbe@108 1069 {"__gc", moonbr_io_handlegc},
jbe@108 1070 {NULL, NULL}
jbe@108 1071 };
jbe@108 1072
jbe@108 1073 static const struct luaL_Reg moonbr_io_listener_methods[] = {
jbe@108 1074 {"accept", moonbr_io_accept},
jbe@108 1075 {"accept_nb", moonbr_io_accept_nb},
jbe@108 1076 {"close", moonbr_io_unlisten},
jbe@108 1077 {NULL, NULL}
jbe@108 1078 };
jbe@108 1079
jbe@108 1080 static const struct luaL_Reg moonbr_io_listener_metamethods[] = {
jbe@108 1081 {"__gc", moonbr_io_listenergc},
jbe@79 1082 {NULL, NULL}
jbe@79 1083 };
jbe@79 1084
jbe@79 1085 static const struct luaL_Reg moonbr_io_module_funcs[] = {
jbe@111 1086 {"localconnect", moonbr_io_localconnect},
jbe@111 1087 {"localconnect_nb", moonbr_io_localconnect_nb},
jbe@98 1088 {"tcpconnect", moonbr_io_tcpconnect},
jbe@99 1089 {"tcpconnect_nb", moonbr_io_tcpconnect_nb},
jbe@112 1090 {"locallisten", moonbr_io_locallisten},
jbe@108 1091 {"tcplisten", moonbr_io_tcplisten},
jbe@106 1092 {"poll", moonbr_io_poll},
jbe@115 1093 {"timeref", moonbr_io_timeref},
jbe@79 1094 {NULL, NULL}
jbe@79 1095 };
jbe@79 1096
jbe@79 1097 int luaopen_moonbridge_io(lua_State *L) {
jbe@79 1098
jbe@113 1099 signal(SIGPIPE, SIG_IGN); /* generate I/O errors instead of signal 13 */
jbe@113 1100
jbe@80 1101 lua_newtable(L); // module
jbe@80 1102
jbe@79 1103 lua_newtable(L); // public metatable
jbe@79 1104 lua_newtable(L); // handle methods
jbe@79 1105 luaL_setfuncs(L, moonbr_io_handle_methods, 0);
jbe@80 1106 lua_pushvalue(L, -1);
jbe@108 1107 lua_setfield(L, -4, "prototype_handle");
jbe@79 1108 lua_setfield(L, -2, "__index");
jbe@79 1109 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY);
jbe@79 1110
jbe@79 1111 lua_newtable(L); // handle metatable
jbe@79 1112 luaL_setfuncs(L, moonbr_io_handle_metamethods, 0);
jbe@79 1113 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79 1114
jbe@108 1115 lua_newtable(L); // listener metatable
jbe@108 1116 luaL_setfuncs(L, moonbr_io_listener_metamethods, 0);
jbe@108 1117 lua_newtable(L); // listener methods
jbe@108 1118 luaL_setfuncs(L, moonbr_io_listener_methods, 0);
jbe@108 1119 lua_pushvalue(L, -1);
jbe@108 1120 lua_setfield(L, -4, "prototype_listener");
jbe@108 1121 lua_setfield(L, -2, "__index");
jbe@108 1122 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_LISTENER_MT_REGKEY);
jbe@108 1123
jbe@79 1124 luaL_setfuncs(L, moonbr_io_module_funcs, 0);
jbe@79 1125 return 1;
jbe@79 1126
jbe@79 1127 }
jbe@79 1128

Impressum / About Us