moonbridge

annotate moonbridge_io.c @ 87:1d91c6eedf18

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

Impressum / About Us