Giving lua-zmq new API semantics.

Functions now return nil followed by an error message on failure.
They also fail properly in all cases, this removes asserts from
the code base.
lua-zmq
Joshua Simmons 16 years ago
parent 8b961acd05
commit 03a21804bb

1
.gitignore vendored

@ -1,2 +1,3 @@
*.lo *.lo
*.so *.so
*.swp

94
zmq.c

@ -58,6 +58,12 @@ static int Lzmq_version(lua_State *L)
return 1; return 1;
} }
#define zmq_return_error() \
const char *error = zmq_strerror(zmq_errno()); \
lua_pushnil(L); \
lua_pushlstring(L, error, strlen(error)); \
return 2
static int Lzmq_init(lua_State *L) static int Lzmq_init(lua_State *L)
{ {
int io_threads = luaL_checkint(L, 1); int io_threads = luaL_checkint(L, 1);
@ -65,7 +71,7 @@ static int Lzmq_init(lua_State *L)
ctx->ptr = zmq_init(io_threads); ctx->ptr = zmq_init(io_threads);
if (!ctx->ptr) { if (!ctx->ptr) {
return luaL_error(L, zmq_strerror(zmq_errno())); zmq_return_error();
} }
luaL_getmetatable(L, MT_ZMQ_CONTEXT); luaL_getmetatable(L, MT_ZMQ_CONTEXT);
@ -79,11 +85,16 @@ static int Lzmq_term(lua_State *L)
zmq_ptr *ctx = luaL_checkudata(L, 1, MT_ZMQ_CONTEXT); zmq_ptr *ctx = luaL_checkudata(L, 1, MT_ZMQ_CONTEXT);
if(ctx->ptr != NULL) { if(ctx->ptr != NULL) {
assert(zmq_term(ctx->ptr) == 0); if(zmq_term(ctx->ptr) == 0) {
ctx->ptr = NULL; ctx->ptr = NULL;
} else {
zmq_return_error();
}
} }
return 0; lua_pushboolean(L, 1);
return 1;
} }
static int Lzmq_socket(lua_State *L) static int Lzmq_socket(lua_State *L)
@ -96,7 +107,7 @@ static int Lzmq_socket(lua_State *L)
s->ptr = zmq_socket(ctx->ptr, type); s->ptr = zmq_socket(ctx->ptr, type);
if (!s->ptr) { if (!s->ptr) {
return luaL_error(L, zmq_strerror(zmq_errno())); zmq_return_error();
} }
luaL_getmetatable(L, MT_ZMQ_SOCKET); luaL_getmetatable(L, MT_ZMQ_SOCKET);
@ -110,11 +121,16 @@ static int Lzmq_close(lua_State *L)
zmq_ptr *s = luaL_checkudata(L, 1, MT_ZMQ_SOCKET); zmq_ptr *s = luaL_checkudata(L, 1, MT_ZMQ_SOCKET);
if(s->ptr != NULL) { if(s->ptr != NULL) {
assert(zmq_close(s->ptr) == 0); if(zmq_close(s->ptr) == 0) {
s->ptr = NULL; s->ptr = NULL;
} else {
zmq_return_error();
}
} }
return 0; lua_pushboolean(L, 1);
return 1;
} }
static int Lzmq_setsockopt(lua_State *L) static int Lzmq_setsockopt(lua_State *L)
@ -158,9 +174,12 @@ static int Lzmq_setsockopt(lua_State *L)
} }
if (rc != 0) { if (rc != 0) {
return luaL_error(L, zmq_strerror(zmq_errno())); zmq_return_error();
} }
return 0;
lua_pushboolean(L, 1);
return 1;
} }
static int Lzmq_getsockopt(lua_State *L) static int Lzmq_getsockopt(lua_State *L)
@ -221,9 +240,12 @@ static int Lzmq_getsockopt(lua_State *L)
} }
if (rc != 0) { if (rc != 0) {
return luaL_error(L, zmq_strerror(zmq_errno())); zmq_return_error();
} }
return 0;
lua_pushboolean(L, 1);
return 1;
} }
static int Lzmq_bind(lua_State *L) static int Lzmq_bind(lua_State *L)
@ -232,10 +254,12 @@ static int Lzmq_bind(lua_State *L)
const char *addr = luaL_checkstring(L, 2); const char *addr = luaL_checkstring(L, 2);
if (zmq_bind(s->ptr, addr) != 0) { if (zmq_bind(s->ptr, addr) != 0) {
return luaL_error(L, zmq_strerror(zmq_errno())); zmq_return_error();
} }
return 0; lua_pushboolean(L, 1);
return 1;
} }
static int Lzmq_connect(lua_State *L) static int Lzmq_connect(lua_State *L)
@ -244,10 +268,12 @@ static int Lzmq_connect(lua_State *L)
const char *addr = luaL_checkstring(L, 2); const char *addr = luaL_checkstring(L, 2);
if (zmq_connect(s->ptr, addr) != 0) { if (zmq_connect(s->ptr, addr) != 0) {
return luaL_error(L, zmq_strerror(zmq_errno())); zmq_return_error();
} }
return 0; lua_pushboolean(L, 1);
return 1;
} }
static int Lzmq_send(lua_State *L) static int Lzmq_send(lua_State *L)
@ -258,20 +284,19 @@ static int Lzmq_send(lua_State *L)
int flags = luaL_optint(L, 3, 0); int flags = luaL_optint(L, 3, 0);
zmq_msg_t msg; zmq_msg_t msg;
assert(zmq_msg_init_size(&msg, msg_size) == 0); if(zmq_msg_init_size(&msg, msg_size) != 0) {
zmq_return_error();
}
memcpy(zmq_msg_data(&msg), data, msg_size); memcpy(zmq_msg_data(&msg), data, msg_size);
int rc = zmq_send(s->ptr, &msg, flags); int rc = zmq_send(s->ptr, &msg, flags);
assert(zmq_msg_close(&msg) == 0); if(zmq_msg_close(&msg) != 0) {
zmq_return_error();
if (rc != 0 && zmq_errno() == EAGAIN) {
lua_pushboolean(L, 0);
return 1;
} }
if (rc != 0) { if (rc != 0) {
return luaL_error(L, zmq_strerror(zmq_errno())); zmq_return_error();
} }
lua_pushboolean(L, 1); lua_pushboolean(L, 1);
@ -285,24 +310,23 @@ static int Lzmq_recv(lua_State *L)
int flags = luaL_optint(L, 2, 0); int flags = luaL_optint(L, 2, 0);
zmq_msg_t msg; zmq_msg_t msg;
assert(zmq_msg_init(&msg) == 0); if(zmq_msg_init(&msg) != 0) {
zmq_return_error();
int rc = zmq_recv(s->ptr, &msg, flags);
if (rc != 0 && zmq_errno() == EAGAIN) {
assert(zmq_msg_close(&msg) == 0);
lua_pushnil(L);
return 1;
} }
if (rc != 0) { if(zmq_recv(s->ptr, &msg, flags) != 0) {
assert(zmq_msg_close(&msg) == 0); // Best we can do in this case is try to close and hope for the best.
return luaL_error(L, zmq_strerror(zmq_errno())); zmq_msg_close(&msg);
zmq_return_error();
} }
lua_pushlstring(L, zmq_msg_data(&msg), zmq_msg_size(&msg)); lua_pushlstring(L, zmq_msg_data(&msg), zmq_msg_size(&msg));
assert(zmq_msg_close(&msg) == 0); if(zmq_msg_close(&msg) != 0) {
// Above string will be poped from the stack by the normalising code
// upon sucessful return.
zmq_return_error();
}
return 1; return 1;
} }

Loading…
Cancel
Save