diff options
Diffstat (limited to 'src/lua/modules/net')
-rw-r--r-- | src/lua/modules/net/Buffer.cpp | 14 | ||||
-rw-r--r-- | src/lua/modules/net/socket.cpp | 18 |
2 files changed, 18 insertions, 14 deletions
diff --git a/src/lua/modules/net/Buffer.cpp b/src/lua/modules/net/Buffer.cpp index 3354518..cc9f2b4 100644 --- a/src/lua/modules/net/Buffer.cpp +++ b/src/lua/modules/net/Buffer.cpp @@ -11,7 +11,9 @@ namespace lua namespace net { - static inline Ref<Buffer>& checkNetBuffer(lua_State* L) + typedef Ref<Buffer>& BufferRef; + + static inline BufferRef checkNetBuffer(lua_State* L) { Proxy* proxy = (Proxy*)luax_checktype(L, 1, JIN_NETWORK_BUFFER); return proxy->getRef<Buffer>(); @@ -20,7 +22,7 @@ namespace net // net.Buffer:append(value) -> value_length static int l_append(lua_State* L) { - Ref<Buffer>& ref = checkNetBuffer(L); + BufferRef ref = checkNetBuffer(L); const int vp = 2; if (luax_isintegerstrict(L, vp)) { @@ -64,7 +66,7 @@ namespace net // net.Buffer:grabString(offset) -> string, length static int l_grabString(lua_State* L) { - Ref<Buffer>& ref = checkNetBuffer(L); + BufferRef ref = checkNetBuffer(L); int offset = luax_checkinteger(L, 2); int len; const char* str = ref->grabString(&len, offset); @@ -76,7 +78,7 @@ namespace net // net.Buffer:grabInteger(offset) -> integer, length static int l_grabInteger(lua_State* L) { - Ref<Buffer>& ref = checkNetBuffer(L); + BufferRef ref = checkNetBuffer(L); int offset = luax_checkinteger(L, 2); int len; int integer = ref->grabInteger(&len, offset); @@ -87,7 +89,7 @@ namespace net static int l_grabFloat(lua_State* L) { - Ref<Buffer>& ref = checkNetBuffer(L); + BufferRef ref = checkNetBuffer(L); int offset = luax_checkinteger(L, 2); int len; float floatv = ref->grabFloat(&len, offset); @@ -98,7 +100,7 @@ namespace net static int l_grabBoolean(lua_State* L) { - Ref<Buffer>& ref = checkNetBuffer(L); + BufferRef ref = checkNetBuffer(L); int offset = luax_checkinteger(L, 2); int len; bool boolean = ref->grabBoolean(&len, offset); diff --git a/src/lua/modules/net/socket.cpp b/src/lua/modules/net/socket.cpp index 6d3fdfb..ef6d4ac 100644 --- a/src/lua/modules/net/socket.cpp +++ b/src/lua/modules/net/socket.cpp @@ -12,9 +12,11 @@ namespace lua using namespace jin::net; using namespace lua::net; + typedef Ref<Socket>& SocketRef; + const int BUFFER_SIZE = 1024; - static inline Ref<Socket>& checkSocket(lua_State* L, int pos = 1) + static inline SocketRef checkSocket(lua_State* L, int pos = 1) { Proxy* proxy = (Proxy*)luax_checktype(L, pos, JIN_NETWORK_SOCKET); return proxy->getRef<Socket>(); @@ -29,7 +31,7 @@ namespace lua // return net.Socket static int l_accept(lua_State* L) { - Ref<Socket>& socket = checkSocket(L); + SocketRef socket = checkSocket(L); Socket* client = socket->accept(); Proxy* proxy = (Proxy*)luax_newinstance(L, JIN_NETWORK_SOCKET, sizeof(Proxy)); proxy->bind(new Ref<Socket>(client, JIN_NETWORK_SOCKET)); @@ -39,7 +41,7 @@ namespace lua // return net.Buffer static int l_receive(lua_State* L) { - Ref<Socket>& socket = checkSocket(L); + SocketRef socket = checkSocket(L); char buffer[BUFFER_SIZE] = {0}; int size = socket->receive(buffer, BUFFER_SIZE); Proxy* proxy = (Proxy*)luax_newinstance(L, JIN_NETWORK_BUFFER, sizeof(Proxy)); @@ -51,7 +53,7 @@ namespace lua // Socket:receiveFrom(address, port) static int l_receiveFrom(lua_State* L) { - Ref<Socket>& socket = checkSocket(L); + SocketRef socket = checkSocket(L); int address = luax_checkinteger(L, 2); int port = luax_checkinteger(L, 3); char buffer[BUFFER_SIZE]; @@ -65,7 +67,7 @@ namespace lua // Socket:send(net.Buffer) -> data_length static int l_send(lua_State* L) { - Ref<Socket>& socket = checkSocket(L); + SocketRef socket = checkSocket(L); Ref<Buffer>& ref = checkNetBuffer(L, 2); int len = socket->send(ref->buffer, ref->size); luax_pushinteger(L, len); @@ -75,7 +77,7 @@ namespace lua // Socket:sendTo(address, port, net.Buffer) static int l_sendTo(lua_State* L) { - Ref<Socket>& socket = checkSocket(L); + SocketRef socket = checkSocket(L); int address = luax_checkinteger(L, 2); int port = luax_checkinteger(L, 3); Ref<Buffer>& buffer = checkNetBuffer(L, 4); @@ -85,14 +87,14 @@ namespace lua static int l_close(lua_State* L) { - Ref<Socket>& socket = checkSocket(L); + SocketRef socket = checkSocket(L); socket->close(); return 0; } static int l_configBlocking(lua_State* L) { - Ref<Socket>& socket = checkSocket(L); + SocketRef socket = checkSocket(L); bool blocking = luax_checkbool(L, 2); socket->configureBlocking(blocking); return 0; |