moonbridge

annotate moonbridge_io.c @ 86:9fa3a36733ff

find changesets by author, revision, files, or words in the commit message
New I/O methods drain and drain_nb
author jbe
date Tue Apr 07 00:47:30 2015 +0200 (2015-04-07)
parents ef552870f1b6
children 1d91c6eedf18
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@79 8 #include <sys/select.h>
jbe@81 9 #include <fcntl.h>
jbe@79 10
jbe@79 11 #include <lua.h>
jbe@79 12 #include <lauxlib.h>
jbe@79 13 #include <lualib.h>
jbe@79 14
jbe@80 15 #define MOONBR_IO_MAXSTRERRORLEN 80
jbe@85 16 #define MOONBR_IO_READBUFLEN 4096
jbe@80 17 #define MOONBR_IO_WRITEBUFLEN 4096
jbe@80 18
jbe@80 19 #define moonbr_io_errmsg() \
jbe@80 20 char errmsg[MOONBR_IO_MAXSTRERRORLEN]; \
jbe@80 21 strerror_r(errno, errmsg, MOONBR_IO_MAXSTRERRORLEN)
jbe@80 22
jbe@79 23 #define MOONBR_IO_HANDLE_MT_REGKEY "moonbridge_io_handle"
jbe@79 24 #define MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY "moonbridge_io_handle_public"
jbe@79 25
jbe@79 26 typedef struct {
jbe@79 27 int fd;
jbe@84 28 int issocket;
jbe@81 29 int nonblocking;
jbe@85 30 int readerr;
jbe@85 31 int readbufcnt;
jbe@81 32 int writeerr;
jbe@83 33 size_t writeleft;
jbe@83 34 #if LUA_VERSION_NUM >= 503
jbe@83 35 lua_Integer writeqin;
jbe@83 36 lua_Integer writeqout;
jbe@83 37 #else
jbe@83 38 int writeqin;
jbe@83 39 int writeqout;
jbe@83 40 #endif
jbe@83 41 size_t writeqoff;
jbe@81 42 int writebufcnt;
jbe@85 43 char readbuf[MOONBR_IO_READBUFLEN];
jbe@80 44 char writebuf[MOONBR_IO_WRITEBUFLEN];
jbe@79 45 } moonbr_io_handle_t;
jbe@79 46
jbe@81 47 static void moonbr_io_handle_set_nonblocking(lua_State *L, moonbr_io_handle_t *handle, int nonblocking) {
jbe@81 48 if (handle->nonblocking != nonblocking) {
jbe@81 49 int flags;
jbe@81 50 flags = fcntl(handle->fd, F_GETFL, 0);
jbe@81 51 if (flags == -1) {
jbe@81 52 moonbr_io_errmsg();
jbe@81 53 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@81 54 }
jbe@81 55 if (nonblocking) flags |= O_NONBLOCK;
jbe@81 56 else flags &= ~O_NONBLOCK;
jbe@81 57 if (fcntl(handle->fd, F_SETFL, flags) == -1) {
jbe@81 58 moonbr_io_errmsg();
jbe@81 59 luaL_error(L, "Unexpected error in fcntl call: %s", errmsg);
jbe@81 60 }
jbe@81 61 }
jbe@81 62 }
jbe@81 63
jbe@86 64 static int moonbr_io_read_impl(lua_State *L, int nonblocking, int drain) {
jbe@85 65 moonbr_io_handle_t *handle;
jbe@85 66 lua_Integer maxread;
jbe@85 67 const char *terminatorstr;
jbe@85 68 size_t terminatorlen;
jbe@85 69 char terminator;
jbe@85 70 luaL_Buffer luabuf;
jbe@85 71 size_t luabufcnt = 0;
jbe@86 72 int endcnt;
jbe@85 73 char *terminatorpos;
jbe@85 74 ssize_t result;
jbe@85 75 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@85 76 maxread = luaL_optinteger(L, 2, 0);
jbe@85 77 terminatorstr = luaL_optlstring(L, 3, "", &terminatorlen);
jbe@85 78 if (terminatorlen) {
jbe@85 79 luaL_argcheck(L, terminatorlen == 1, 3, "single byte expected");
jbe@85 80 terminator = terminatorstr[0];
jbe@85 81 }
jbe@86 82 lua_settop(L, 1); /* return handle on drain, terminator string may be garbage collected */
jbe@85 83 if (handle->fd < 0) luaL_error(L, "Attempt to read from a closed I/O handle");
jbe@85 84 if (handle->readerr) {
jbe@85 85 lua_pushnil(L);
jbe@85 86 lua_pushliteral(L, "Previous read error");
jbe@85 87 return 2;
jbe@85 88 }
jbe@85 89 moonbr_io_handle_set_nonblocking(L, handle, nonblocking);
jbe@86 90 if (!drain) luaL_buffinit(L, &luabuf);
jbe@85 91 while (1) {
jbe@86 92 endcnt = -1;
jbe@85 93 if (maxread > 0 && handle->readbufcnt >= maxread - luabufcnt) {
jbe@86 94 endcnt = maxread - luabufcnt;
jbe@85 95 } else if (terminatorlen) {
jbe@85 96 terminatorpos = memchr(handle->readbuf, terminator, handle->readbufcnt);
jbe@86 97 if (terminatorpos) endcnt = 1 + (terminatorpos - handle->readbuf);
jbe@85 98 }
jbe@86 99 if (endcnt >= 0) {
jbe@86 100 if (!drain) {
jbe@86 101 luaL_addlstring(&luabuf, handle->readbuf, endcnt);
jbe@86 102 luaL_pushresult(&luabuf);
jbe@86 103 }
jbe@86 104 handle->readbufcnt -= endcnt;
jbe@86 105 memmove(handle->readbuf, handle->readbuf + endcnt, handle->readbufcnt);
jbe@85 106 return 1;
jbe@85 107 }
jbe@86 108 if (!drain) luaL_addlstring(&luabuf, handle->readbuf, handle->readbufcnt);
jbe@85 109 luabufcnt += handle->readbufcnt;
jbe@85 110 handle->readbufcnt = 0;
jbe@85 111 do {
jbe@85 112 result = read(handle->fd, handle->readbuf, MOONBR_IO_READBUFLEN);
jbe@85 113 } while (result < 0 && (errno == EINTR));
jbe@85 114 if (result == 0 || (nonblocking && result < 0 && (errno == EAGAIN || errno == EWOULDBLOCK))) break;
jbe@85 115 if (result < 0) {
jbe@85 116 moonbr_io_errmsg();
jbe@85 117 handle->readerr = 1;
jbe@85 118 lua_pushnil(L);
jbe@85 119 lua_pushstring(L, errmsg);
jbe@85 120 return 2;
jbe@85 121 }
jbe@85 122 handle->readbufcnt += result;
jbe@85 123 }
jbe@86 124 if (!drain) {
jbe@86 125 luaL_addlstring(&luabuf, handle->readbuf, handle->readbufcnt);
jbe@86 126 luaL_pushresult(&luabuf);
jbe@86 127 }
jbe@85 128 handle->readbufcnt = 0;
jbe@85 129 return 1;
jbe@85 130 }
jbe@85 131
jbe@85 132 static int moonbr_io_read(lua_State *L) {
jbe@86 133 return moonbr_io_read_impl(L, 0, 0);
jbe@85 134 }
jbe@85 135
jbe@85 136 static int moonbr_io_read_nb(lua_State *L) {
jbe@86 137 return moonbr_io_read_impl(L, 1, 0);
jbe@86 138 }
jbe@86 139
jbe@86 140 static int moonbr_io_drain(lua_State *L) {
jbe@86 141 return moonbr_io_read_impl(L, 0, 1);
jbe@86 142 }
jbe@86 143
jbe@86 144 static int moonbr_io_drain_nb(lua_State *L) {
jbe@86 145 return moonbr_io_read_impl(L, 1, 1);
jbe@85 146 }
jbe@85 147
jbe@81 148 static int moonbr_io_write_impl(lua_State *L, int nonblocking, int flush) {
jbe@80 149 moonbr_io_handle_t *handle;
jbe@80 150 int i, top;
jbe@80 151 const char *str;
jbe@81 152 size_t strlen, strpos;
jbe@80 153 size_t written;
jbe@80 154 ssize_t result;
jbe@80 155 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@83 156 if (handle->fd < 0) luaL_error(L, "Attempt to write to a closed I/O handle");
jbe@81 157 if (handle->writeerr) {
jbe@80 158 lua_pushnil(L);
jbe@80 159 lua_pushliteral(L, "Previous write error");
jbe@80 160 return 2;
jbe@80 161 }
jbe@81 162 moonbr_io_handle_set_nonblocking(L, handle, nonblocking);
jbe@84 163 top = lua_gettop(L);
jbe@81 164 lua_getuservalue(L, 1);
jbe@81 165 lua_getfield(L, -1, "writebuf");
jbe@84 166 for (i=2; i<=top; i++) {
jbe@84 167 luaL_checklstring(L, i, &strlen);
jbe@84 168 lua_pushvalue(L, i);
jbe@84 169 lua_rawseti(L, -2, handle->writeqin++);
jbe@84 170 handle->writeleft += strlen;
jbe@81 171 }
jbe@83 172 while (handle->writeqout != handle->writeqin) {
jbe@83 173 lua_rawgeti(L, -1, handle->writeqout);
jbe@81 174 str = lua_tolstring(L, -1, &strlen);
jbe@81 175 strpos = handle->writeqoff;
jbe@81 176 while (strpos < strlen) {
jbe@81 177 if (strlen - strpos < MOONBR_IO_WRITEBUFLEN - handle->writebufcnt) {
jbe@81 178 memcpy(handle->writebuf + handle->writebufcnt, str + strpos, strlen - strpos);
jbe@81 179 handle->writebufcnt += strlen - strpos;
jbe@80 180 break;
jbe@80 181 } else {
jbe@80 182 written = 0;
jbe@81 183 memcpy(handle->writebuf + handle->writebufcnt, str + strpos, MOONBR_IO_WRITEBUFLEN - handle->writebufcnt);
jbe@81 184 strpos += MOONBR_IO_WRITEBUFLEN - handle->writebufcnt;
jbe@80 185 while (written < MOONBR_IO_WRITEBUFLEN) {
jbe@80 186 result = write(handle->fd, handle->writebuf + written, MOONBR_IO_WRITEBUFLEN - written);
jbe@80 187 if (result < 0) {
jbe@81 188 if (nonblocking && (errno == EAGAIN || errno == EWOULDBLOCK)) {
jbe@81 189 if (written) {
jbe@81 190 handle->writebufcnt -= written;
jbe@81 191 memmove(handle->writebuf, handle->writebuf + written, handle->writebufcnt);
jbe@81 192 }
jbe@81 193 handle->writeqoff = strpos;
jbe@81 194 goto moonbr_io_write_impl_block;
jbe@81 195 } else if (errno != EINTR) {
jbe@80 196 moonbr_io_errmsg();
jbe@81 197 handle->writeerr = 1;
jbe@80 198 lua_pushnil(L);
jbe@80 199 lua_pushstring(L, errmsg);
jbe@80 200 return 2;
jbe@80 201 }
jbe@80 202 } else {
jbe@80 203 written += result;
jbe@81 204 handle->writeleft -= result;
jbe@80 205 }
jbe@80 206 }
jbe@81 207 handle->writebufcnt = 0;
jbe@80 208 }
jbe@80 209 }
jbe@81 210 handle->writeqoff = 0;
jbe@81 211 lua_pop(L, 1);
jbe@81 212 lua_pushnil(L);
jbe@83 213 lua_rawseti(L, -2, handle->writeqout++);
jbe@80 214 }
jbe@81 215 if (flush) {
jbe@81 216 written = 0;
jbe@81 217 while (written < handle->writebufcnt) {
jbe@81 218 result = write(handle->fd, handle->writebuf + written, handle->writebufcnt - written);
jbe@81 219 if (result < 0) {
jbe@81 220 if (nonblocking && (errno == EAGAIN || errno == EWOULDBLOCK)) {
jbe@81 221 if (written) {
jbe@81 222 handle->writebufcnt -= written;
jbe@81 223 memmove(handle->writebuf, handle->writebuf + written, handle->writebufcnt);
jbe@81 224 }
jbe@81 225 goto moonbr_io_write_impl_block;
jbe@81 226 } else if (errno != EINTR) {
jbe@81 227 moonbr_io_errmsg();
jbe@81 228 handle->writeerr = -1;
jbe@81 229 lua_pushnil(L);
jbe@81 230 lua_pushstring(L, errmsg);
jbe@81 231 return 2;
jbe@81 232 }
jbe@81 233 } else {
jbe@81 234 written += result;
jbe@81 235 handle->writeleft -= result;
jbe@81 236 }
jbe@81 237 }
jbe@81 238 handle->writebufcnt = 0;
jbe@82 239 if (nonblocking) lua_pushinteger(L, 0);
jbe@81 240 } else {
jbe@82 241 if (nonblocking) lua_pushinteger(L, handle->writeleft - handle->writebufcnt);
jbe@81 242 }
jbe@82 243 if (!nonblocking) lua_pushvalue(L, 1);
jbe@80 244 return 1;
jbe@81 245 moonbr_io_write_impl_block:
jbe@81 246 if (flush) lua_pushinteger(L, handle->writeleft);
jbe@81 247 else lua_pushinteger(L, handle->writeleft - handle->writebufcnt);
jbe@81 248 return 1;
jbe@81 249 }
jbe@81 250
jbe@81 251 static int moonbr_io_write(lua_State *L) {
jbe@81 252 return moonbr_io_write_impl(L, 0, 0);
jbe@81 253 }
jbe@81 254
jbe@81 255 static int moonbr_io_write_nb(lua_State *L) {
jbe@81 256 return moonbr_io_write_impl(L, 1, 0);
jbe@80 257 }
jbe@80 258
jbe@80 259 static int moonbr_io_flush(lua_State *L) {
jbe@81 260 return moonbr_io_write_impl(L, 0, 1);
jbe@81 261 }
jbe@81 262
jbe@81 263 static int moonbr_io_flush_nb(lua_State *L) {
jbe@81 264 return moonbr_io_write_impl(L, 1, 1);
jbe@80 265 }
jbe@80 266
jbe@84 267 static int moonbr_io_close_impl(lua_State *L, int clean) {
jbe@83 268 moonbr_io_handle_t *handle;
jbe@83 269 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@83 270 if (handle->fd < 0) luaL_error(L, "Attempt to close a closed I/O handle");
jbe@84 271 if (clean && handle->writeleft) {
jbe@83 272 lua_pushcfunction(L, moonbr_io_flush);
jbe@83 273 lua_pushvalue(L, 1);
jbe@83 274 lua_call(L, 1, 2);
jbe@83 275 if (!lua_toboolean(L, -2)) {
jbe@83 276 close(handle->fd);
jbe@83 277 handle->fd = -1;
jbe@83 278 return 2;
jbe@83 279 }
jbe@83 280 }
jbe@83 281 if (close(handle->fd)) {
jbe@83 282 moonbr_io_errmsg();
jbe@83 283 handle->fd = -1;
jbe@83 284 lua_pushnil(L);
jbe@83 285 lua_pushstring(L, errmsg);
jbe@83 286 return 2;
jbe@83 287 }
jbe@83 288 handle->fd = -1;
jbe@83 289 lua_pushboolean(L, 1);
jbe@83 290 return 1;
jbe@84 291
jbe@83 292 }
jbe@83 293
jbe@84 294 static int moonbr_io_reset(lua_State *L) {
jbe@84 295 return moonbr_io_close_impl(L, 0);
jbe@84 296 }
jbe@84 297
jbe@84 298 static int moonbr_io_close(lua_State *L) {
jbe@84 299 return moonbr_io_close_impl(L, 1);
jbe@84 300 }
jbe@84 301
jbe@84 302 void moonbr_io_pushhandle(lua_State *L, int fd, int issocket) {
jbe@79 303 moonbr_io_handle_t *handle;
jbe@79 304 handle = lua_newuserdata(L, sizeof(moonbr_io_handle_t));
jbe@79 305 handle->fd = fd;
jbe@84 306 handle->issocket = issocket;
jbe@81 307 handle->nonblocking = -1;
jbe@85 308 handle->readerr = 0;
jbe@85 309 handle->readbufcnt = 0;
jbe@81 310 handle->writeerr = 0;
jbe@81 311 handle->writeleft = 0;
jbe@83 312 handle->writeqin = 0;
jbe@83 313 handle->writeqout = 0;
jbe@81 314 handle->writeqoff = 0;
jbe@81 315 handle->writebufcnt = 0;
jbe@79 316 luaL_getmetatable(L, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79 317 lua_setmetatable(L, -2);
jbe@79 318 lua_newtable(L); // uservalue
jbe@81 319 lua_newtable(L);
jbe@81 320 lua_setfield(L, -2, "writebuf");
jbe@79 321 lua_newtable(L); // public
jbe@79 322 luaL_getmetatable(L, MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY);
jbe@79 323 lua_setmetatable(L, -2);
jbe@79 324 lua_setfield(L, -2, "public");
jbe@79 325 lua_setuservalue(L, -2);
jbe@79 326 }
jbe@79 327
jbe@79 328 static int moonbr_io_handleindex(lua_State *L) {
jbe@80 329 luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79 330 lua_getuservalue(L, 1);
jbe@79 331 lua_getfield(L, -1, "public");
jbe@79 332 lua_pushvalue(L, 2);
jbe@79 333 lua_gettable(L, -2);
jbe@79 334 return 1;
jbe@79 335 }
jbe@79 336
jbe@79 337 static int moonbr_io_handlenewindex(lua_State *L) {
jbe@80 338 luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79 339 lua_getuservalue(L, 1);
jbe@79 340 lua_getfield(L, -1, "public");
jbe@79 341 lua_pushvalue(L, 2);
jbe@79 342 lua_pushvalue(L, 3);
jbe@79 343 lua_settable(L, -3);
jbe@79 344 return 0;
jbe@79 345 }
jbe@79 346
jbe@80 347 static int moonbr_io_handlegc(lua_State *L) {
jbe@80 348 moonbr_io_handle_t *handle;
jbe@80 349 handle = luaL_checkudata(L, 1, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@80 350 if (handle->fd >= 0) {
jbe@80 351 lua_getuservalue(L, 1);
jbe@80 352 lua_getfield(L, -1, "gc");
jbe@80 353 lua_pushvalue(L, 1);
jbe@80 354 lua_call(L, 1, 0);
jbe@80 355 }
jbe@80 356 return 0;
jbe@80 357 }
jbe@80 358
jbe@79 359 static int moonbr_io_getdummy(lua_State *L) {
jbe@85 360 moonbr_io_pushhandle(L, 0, 0);
jbe@79 361 return 1;
jbe@79 362 }
jbe@79 363
jbe@79 364 static const struct luaL_Reg moonbr_io_handle_methods[] = {
jbe@85 365 {"read", moonbr_io_read},
jbe@85 366 {"read_nb", moonbr_io_read_nb},
jbe@86 367 {"drain", moonbr_io_drain},
jbe@86 368 {"drain_nb", moonbr_io_drain_nb},
jbe@80 369 {"write", moonbr_io_write},
jbe@81 370 {"write_nb", moonbr_io_write_nb},
jbe@80 371 {"flush", moonbr_io_flush},
jbe@81 372 {"flush_nb", moonbr_io_flush_nb},
jbe@85 373 {"reset", moonbr_io_reset},
jbe@85 374 {"close", moonbr_io_close},
jbe@79 375 {NULL, NULL}
jbe@79 376 };
jbe@79 377
jbe@79 378 static const struct luaL_Reg moonbr_io_handle_metamethods[] = {
jbe@79 379 {"__index", moonbr_io_handleindex},
jbe@79 380 {"__newindex", moonbr_io_handlenewindex},
jbe@80 381 {"__gc", moonbr_io_handlegc},
jbe@79 382 {NULL, NULL}
jbe@79 383 };
jbe@79 384
jbe@79 385 static const struct luaL_Reg moonbr_io_module_funcs[] = {
jbe@79 386 {"getdummy", moonbr_io_getdummy},
jbe@79 387 {NULL, NULL}
jbe@79 388 };
jbe@79 389
jbe@79 390 int luaopen_moonbridge_io(lua_State *L) {
jbe@79 391
jbe@80 392 lua_newtable(L); // module
jbe@80 393
jbe@79 394 lua_newtable(L); // public metatable
jbe@79 395 lua_newtable(L); // handle methods
jbe@79 396 luaL_setfuncs(L, moonbr_io_handle_methods, 0);
jbe@80 397 lua_pushvalue(L, -1);
jbe@80 398 lua_setfield(L, -4, "handle");
jbe@79 399 lua_setfield(L, -2, "__index");
jbe@79 400 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_HANDLE_PUBLIC_MT_REGKEY);
jbe@79 401
jbe@79 402 lua_newtable(L); // handle metatable
jbe@79 403 luaL_setfuncs(L, moonbr_io_handle_metamethods, 0);
jbe@79 404 lua_setfield(L, LUA_REGISTRYINDEX, MOONBR_IO_HANDLE_MT_REGKEY);
jbe@79 405
jbe@79 406 luaL_setfuncs(L, moonbr_io_module_funcs, 0);
jbe@79 407 return 1;
jbe@79 408
jbe@79 409 }
jbe@79 410

Impressum / About Us