add stream read and write

6 files changed, 314 insertions, 0 deletions

diff --git a/core/src/xmake/engine.c b/core/src/xmake/engine.c
index a5f767edf..43e073210 100644
--- a/core/src/xmake/engine.c
+++ b/core/src/xmake/engine.c
@@ -197,8 +197,12 @@ tb_int_t xm_lz4_block_decompress(lua_State* lua);
 tb_int_t xm_lz4_compress_file(lua_State* lua);
 tb_int_t xm_lz4_decompress_file(lua_State* lua);
 tb_int_t xm_lz4_compress_stream_open(lua_State* lua);
+tb_int_t xm_lz4_compress_stream_read(lua_State* lua);
+tb_int_t xm_lz4_compress_stream_write(lua_State* lua);
 tb_int_t xm_lz4_compress_stream_close(lua_State* lua);
 tb_int_t xm_lz4_decompress_stream_open(lua_State* lua);
+tb_int_t xm_lz4_decompress_stream_read(lua_State* lua);
+tb_int_t xm_lz4_decompress_stream_write(lua_State* lua);
 tb_int_t xm_lz4_decompress_stream_close(lua_State* lua);
 
 // the windows functions
@@ -425,8 +429,12 @@ static luaL_Reg const g_lz4_functions[] =
 ,   { "compress_file",          xm_lz4_compress_file           }
 ,   { "decompress_file",        xm_lz4_decompress_file         }
 ,   { "compress_stream_open",   xm_lz4_compress_stream_open    }
+,   { "compress_stream_read",   xm_lz4_compress_stream_read    }
+,   { "compress_stream_write",  xm_lz4_compress_stream_write   }
 ,   { "compress_stream_close",  xm_lz4_compress_stream_close   }
 ,   { "decompress_stream_open", xm_lz4_decompress_stream_open  }
+,   { "decompress_stream_read", xm_lz4_decompress_stream_read  }
+,   { "decompress_stream_write",xm_lz4_decompress_stream_write }
 ,   { "decompress_stream_close",xm_lz4_decompress_stream_close }
 ,   { tb_null,                  tb_null                        }
 };
diff --git a/core/src/xmake/lz4/compress_stream_read.c b/core/src/xmake/lz4/compress_stream_read.c
new file mode 100644
index 000000000..cf5e6b8e3
--- /dev/null
+++ b/core/src/xmake/lz4/compress_stream_read.c
@@ -0,0 +1,79 @@
+/*!A cross-platform build utility based on Lua
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * Copyright (C) 2015-present, TBOOX Open Source Group.
+ *
+ * @author      ruki
+ * @file        compress_stream_read.c
+ *
+ */
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * trace
+ */
+#define TB_TRACE_MODULE_NAME    "compress_stream_read"
+#define TB_TRACE_MODULE_DEBUG   (0)
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * includes
+ */
+#include "prefix.h"
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * implementation
+ */
+
+tb_int_t xm_lz4_compress_stream_read(lua_State* lua)
+{
+    // check
+    tb_assert_and_check_return_val(lua, 0);
+
+    // check handle
+    if (!xm_lua_ispointer(lua, 1))
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushliteral(lua, "invalid handle!");
+        return 2;
+    }
+
+    // get stream
+    xm_lz4_cstream_t* stream = (xm_lz4_cstream_t*)xm_lua_topointer(lua, 1);
+    tb_check_return_val(stream, 0);
+
+    // get data
+    tb_byte_t* data = tb_null;
+    if (lua_isnumber(lua, 2))
+        data = (tb_byte_t*)(tb_size_t)(tb_long_t)lua_tonumber(lua, 2);
+    if (!data)
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushfstring(lua, "invalid data(%p)!", data);
+        return 2;
+    }
+
+    // get size
+    tb_long_t size = 0;
+    if (lua_isnumber(lua, 3)) size = (tb_long_t)lua_tonumber(lua, 3);
+    if (size <= 0)
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushfstring(lua, "invalid size(%d)!", (tb_int_t)size);
+        return 2;
+    }
+
+    // read data
+    tb_long_t real = xm_lz4_cstream_read(stream, data, size);
+    lua_pushinteger(lua, (tb_int_t)real);
+    return 1;
+}
diff --git a/core/src/xmake/lz4/compress_stream_write.c b/core/src/xmake/lz4/compress_stream_write.c
new file mode 100644
index 000000000..066839bab
--- /dev/null
+++ b/core/src/xmake/lz4/compress_stream_write.c
@@ -0,0 +1,74 @@
+/*!A cross-platform build utility based on Lua
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * Copyright (C) 2015-present, TBOOX Open Source Group.
+ *
+ * @author      ruki
+ * @file        compress_stream_write.c
+ *
+ */
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * trace
+ */
+#define TB_TRACE_MODULE_NAME    "compress_stream_write"
+#define TB_TRACE_MODULE_DEBUG   (0)
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * includes
+ */
+#include "prefix.h"
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * implementation
+ */
+
+tb_int_t xm_lz4_compress_stream_write(lua_State* lua)
+{
+    // check
+    tb_assert_and_check_return_val(lua, 0);
+
+    // check handle
+    if (!xm_lua_ispointer(lua, 1))
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushliteral(lua, "invalid handle!");
+        return 2;
+    }
+
+    // get stream
+    xm_lz4_cstream_t* stream = (xm_lz4_cstream_t*)xm_lua_topointer(lua, 1);
+    tb_check_return_val(stream, 0);
+
+    // get data and size
+    tb_size_t        size = 0;
+    tb_byte_t const* data = tb_null;
+    if (lua_isnumber(lua, 2)) data = (tb_byte_t const*)(tb_size_t)(tb_long_t)lua_tonumber(lua, 2);
+    if (lua_isnumber(lua, 3)) size = (tb_size_t)lua_tonumber(lua, 3);
+    if (!data || !size)
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushfstring(lua, "invalid data(%p) and size(%d)!", data, (tb_int_t)size);
+        return 2;
+    }
+
+    // is end?
+    tb_bool_t end = tb_false;
+    if (lua_isboolean(lua, 4)) end = lua_toboolean(lua, 4);
+
+    // write data
+    tb_long_t real = xm_lz4_cstream_write(stream, data, size, end);
+    lua_pushinteger(lua, (tb_int_t)real);
+    return 1;
+}
diff --git a/core/src/xmake/lz4/decompress_stream_read.c b/core/src/xmake/lz4/decompress_stream_read.c
new file mode 100644
index 000000000..61f920354
--- /dev/null
+++ b/core/src/xmake/lz4/decompress_stream_read.c
@@ -0,0 +1,79 @@
+/*!A cross-platform build utility based on Lua
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * Copyright (C) 2015-present, TBOOX Open Source Group.
+ *
+ * @author      ruki
+ * @file        decompress_stream_read.c
+ *
+ */
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * trace
+ */
+#define TB_TRACE_MODULE_NAME    "decompress_stream_read"
+#define TB_TRACE_MODULE_DEBUG   (0)
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * includes
+ */
+#include "prefix.h"
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * implementation
+ */
+
+tb_int_t xm_lz4_decompress_stream_read(lua_State* lua)
+{
+    // check
+    tb_assert_and_check_return_val(lua, 0);
+
+    // check handle
+    if (!xm_lua_ispointer(lua, 1))
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushliteral(lua, "invalid handle!");
+        return 2;
+    }
+
+    // get stream
+    xm_lz4_dstream_t* stream = (xm_lz4_dstream_t*)xm_lua_topointer(lua, 1);
+    tb_check_return_val(stream, 0);
+
+    // get data
+    tb_byte_t* data = tb_null;
+    if (lua_isnumber(lua, 2))
+        data = (tb_byte_t*)(tb_size_t)(tb_long_t)lua_tonumber(lua, 2);
+    if (!data)
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushfstring(lua, "invalid data(%p)!", data);
+        return 2;
+    }
+
+    // get size
+    tb_long_t size = 0;
+    if (lua_isnumber(lua, 3)) size = (tb_long_t)lua_tonumber(lua, 3);
+    if (size <= 0)
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushfstring(lua, "invalid size(%d)!", (tb_int_t)size);
+        return 2;
+    }
+
+    // read data
+    tb_long_t real = xm_lz4_dstream_read(stream, data, size);
+    lua_pushinteger(lua, (tb_int_t)real);
+    return 1;
+}
diff --git a/core/src/xmake/lz4/decompress_stream_write.c b/core/src/xmake/lz4/decompress_stream_write.c
new file mode 100644
index 000000000..00a652349
--- /dev/null
+++ b/core/src/xmake/lz4/decompress_stream_write.c
@@ -0,0 +1,70 @@
+/*!A cross-platform build utility based on Lua
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * Copyright (C) 2015-present, TBOOX Open Source Group.
+ *
+ * @author      ruki
+ * @file        decompress_stream_write.c
+ *
+ */
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * trace
+ */
+#define TB_TRACE_MODULE_NAME    "decompress_stream_write"
+#define TB_TRACE_MODULE_DEBUG   (0)
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * includes
+ */
+#include "prefix.h"
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * implementation
+ */
+
+tb_int_t xm_lz4_decompress_stream_write(lua_State* lua)
+{
+    // check
+    tb_assert_and_check_return_val(lua, 0);
+
+    // check handle
+    if (!xm_lua_ispointer(lua, 1))
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushliteral(lua, "invalid handle!");
+        return 2;
+    }
+
+    // get stream
+    xm_lz4_dstream_t* stream = (xm_lz4_dstream_t*)xm_lua_topointer(lua, 1);
+    tb_check_return_val(stream, 0);
+
+    // get data and size
+    tb_size_t        size = 0;
+    tb_byte_t const* data = tb_null;
+    if (lua_isnumber(lua, 2)) data = (tb_byte_t const*)(tb_size_t)(tb_long_t)lua_tonumber(lua, 2);
+    if (lua_isnumber(lua, 3)) size = (tb_size_t)lua_tonumber(lua, 3);
+    if (!data || !size)
+    {
+        lua_pushinteger(lua, -1);
+        lua_pushfstring(lua, "invalid data(%p) and size(%d)!", data, (tb_int_t)size);
+        return 2;
+    }
+
+    // write data
+    tb_long_t real = xm_lz4_dstream_write(stream, data, size, tb_false);
+    lua_pushinteger(lua, (tb_int_t)real);
+    return 1;
+}
diff --git a/core/src/xmake/makefile b/core/src/xmake/makefile
index 597f3de3b..b9a40c676 100644
--- a/core/src/xmake/makefile
+++ b/core/src/xmake/makefile
@@ -140,9 +140,13 @@ xmake_C_FILES 			+= \
 						lz4/block_decompress \
 						lz4/compress_file \
 						lz4/compress_stream_open \
+						lz4/compress_stream_read \
+						lz4/compress_stream_write \
 						lz4/compress_stream_close \
 						lz4/decompress_file \
 						lz4/decompress_stream_open \
+						lz4/decompress_stream_read \
+						lz4/decompress_stream_write \
 						lz4/decompress_stream_close
 
 iswin =