1 files changed, 16 insertions, 19 deletions

diff --git a/src/io.c b/src/io.c
index 2d62147..612454b 100644
--- a/src/io.c
+++ b/src/io.c
@@ -22,27 +22,24 @@ void io_init(p_io io, p_send send, p_recv recv, void *ctx)
 /*-------------------------------------------------------------------------*\
 * Translate error codes to Lua
 \*-------------------------------------------------------------------------*/
-void io_pusherror(lua_State *L, int code)
+const char *io_strerror(int code)
 {
     switch (code) {
-        case IO_DONE:
-            lua_pushnil(L);
-            break;
-        case IO_TIMEOUT:
-            lua_pushstring(L, "timeout");
-            break;
-        case IO_LIMITED:
-            lua_pushstring(L, "limited");
-            break;
-        case IO_CLOSED:
-            lua_pushstring(L, "closed");
-            break;
-        case IO_REFUSED:
-            lua_pushstring(L, "refused");
-            break;
-        default:
-            lua_pushstring(L, "unknown error");
-            break;
+        case IO_DONE: return NULL;
+        case IO_TIMEOUT: return "timeout";
+        case IO_RETRY: return "retry";
+        case IO_CLOSED: return "closed";
+        case IO_REFUSED: return "refused";
+        default: return "unknown error";
     }
 }
 
+/*-------------------------------------------------------------------------*\
+* Translate error codes to Lua
+\*-------------------------------------------------------------------------*/
+void io_pusherror(lua_State *L, int code)
+{
+    const char *err = io_strerror(code);
+    if (err) lua_pushstring(L, err);
+    else lua_pushnil(L);
+}