From 6551adeca70d4299a99d45245d4e13dbfdfa87e5 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Tue, 23 Oct 2018 15:56:01 +0800
Subject: =?UTF-8?q?*=E4=BF=AE=E6=94=B9lua=E7=BB=91=E5=AE=9A=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E7=9A=84=E6=96=87=E4=BB=B6=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/lua/modules/net/je_lua_buffer.cpp | 142 ++++++++++++++++++++++++++++++++++
 1 file changed, 142 insertions(+)
 create mode 100644 src/lua/modules/net/je_lua_buffer.cpp

(limited to 'src/lua/modules/net/je_lua_buffer.cpp')

diff --git a/src/lua/modules/net/je_lua_buffer.cpp b/src/lua/modules/net/je_lua_buffer.cpp
new file mode 100644
index 0000000..ddfbd6b
--- /dev/null
+++ b/src/lua/modules/net/je_lua_buffer.cpp
@@ -0,0 +1,142 @@
+#include "lua/modules/luax.h"
+#include "lua/modules/types.h"
+#include "lua/common/common.h"
+#include "libjin/jin.h"
+#include "Buffer.h"
+
+namespace JinEngine
+{
+    namespace Lua
+    {
+        namespace Net
+        {
+
+            using namespace JinEngine;
+
+	        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>();
+            }
+
+            // net.Buffer:append(value) -> value_length
+            static int l_append(lua_State* L)
+            {
+		        BufferRef ref = checkNetBuffer(L);
+                const int vp = 2;
+                if (luax_isintegerstrict(L, vp))
+                {
+                    int n = luax_checkinteger(L, vp);
+                    int size = sizeof(n);
+                    ref->append(&n, size);
+                    luax_pushinteger(L, size);
+                    return 1;
+                } 
+                else if (luax_isfloatstrict(L, vp))
+                {
+                    float n = luax_checknumber(L, vp);
+                    int size = sizeof(n);
+			        ref->append(&n, size);
+                    luax_pushinteger(L, size);
+                    return 1;
+                }
+                else if (luax_isbooleanstrict(L, vp))
+                {
+                    bool n = luax_checkbool(L, vp);
+                    int size = sizeof(n);
+			        ref->append(&n, size);
+                    luax_pushinteger(L, size);
+                    return 1;
+                }
+                else if (luax_isstringstrict(L, vp))
+                {
+                    const char* str = luax_checkstring(L, vp);
+                    int size = strlen(str) + 1;
+			        ref->append(str, size);
+                    luax_pushinteger(L, size);
+                    return 1;
+                }
+                else
+                {
+                    luax_typerror(L, vp, "number, bool or string");
+                    return 0;
+                }
+            }
+
+            // net.Buffer:grabString(offset) -> string, length
+            static int l_grabString(lua_State* L)
+            {
+		        BufferRef ref = checkNetBuffer(L);
+                int offset = luax_checkinteger(L, 2);
+                unsigned int len;
+                char* data = ref->grabString(&len, offset);
+                Array<char> str;
+                str.bind(data, len);
+                luax_pushstring(L, &str);
+                luax_pushinteger(L, str.count());
+                return 2;
+            }
+
+            // net.Buffer:grabInteger(offset) -> integer, length
+            static int l_grabInteger(lua_State* L)
+            {
+		        BufferRef ref = checkNetBuffer(L);
+                int offset = luax_checkinteger(L, 2);
+                int len;
+                int integer = ref->grabInteger(&len, offset);
+                luax_pushinteger(L, integer);
+                luax_pushinteger(L, len);
+                return 2;
+            }
+
+            static int l_grabFloat(lua_State* L)
+            {
+		        BufferRef ref = checkNetBuffer(L);
+                int offset = luax_checkinteger(L, 2);
+                int len;
+                float floatv = ref->grabFloat(&len, offset);
+                luax_pushnumber(L, floatv);
+                luax_pushinteger(L, len);
+                return 2;
+            }
+
+            static int l_grabBoolean(lua_State* L)
+            {
+		        BufferRef ref = checkNetBuffer(L);
+                int offset = luax_checkinteger(L, 2);
+                int len;
+                bool boolean = ref->grabBoolean(&len, offset);
+                luax_pushboolean(L, boolean);
+                luax_pushinteger(L, len);
+                return 2;
+            }
+
+            static int l_gc(lua_State* L)
+            {
+                Proxy* proxy = (Proxy*)luax_checktype(L, 1, JIN_NETWORK_BUFFER);
+                proxy->release();
+                return 0;
+            }
+
+            static const luaL_Reg netbuffer_function[] = {
+                { "__gc",           l_gc          },
+                { "append",         l_append      },
+                { "grabString",     l_grabString  },
+                { "grabInteger",    l_grabInteger },
+                { "grabBoolean",    l_grabBoolean },
+                { "grabFloat",      l_grabFloat   },
+                { 0,                0             }
+            };
+
+        } // namespace Net
+
+        int luaopen_Buffer(lua_State* L)
+        {
+            luax_newtype(L, JIN_NETWORK_BUFFER, Net::netbuffer_function);
+            return 0;
+        }
+
+    } // namespace Lua
+} // namespace JinEngine
\ No newline at end of file
-- 
cgit v1.1-26-g67d0