add stream wrap

2 files changed, 52 insertions, 0 deletions

diff --git a/xmake/core/compress/lz4.lua b/xmake/core/compress/lz4.lua
index 88b854760..32a215af0 100644
--- a/xmake/core/compress/lz4.lua
+++ b/xmake/core/compress/lz4.lua
@@ -54,6 +54,14 @@ function _cstream:cdata()
     return self._HANDLE
 end
 
+-- ensure the socket is opened
+function _cstream:_ensure_opened()
+    if not self:cdata() then
+        return false, string.format("%s: has been closed!", self)
+    end
+    return true
+end
+
 -- tostring(stream)
 function _cstream:__tostring()
     return string.format("<lz4/cstream: %s>", self:cdata())
@@ -79,6 +87,14 @@ function _dstream:cdata()
     return self._HANDLE
 end
 
+-- ensure the socket is opened
+function _dstream:_ensure_opened()
+    if not self:cdata() then
+        return false, string.format("%s: has been closed!", self)
+    end
+    return true
+end
+
 -- tostring(stream)
 function _dstream:__tostring()
     return string.format("<lz4/dstream: %s>", self:cdata())
diff --git a/xmake/core/sandbox/modules/import/core/compress/lz4.lua b/xmake/core/sandbox/modules/import/core/compress/lz4.lua
index f450609f8..a93c9aa65 100644
--- a/xmake/core/sandbox/modules/import/core/compress/lz4.lua
+++ b/xmake/core/sandbox/modules/import/core/compress/lz4.lua
@@ -57,6 +57,42 @@ function _dstream_wrap(instance)
     return instance
 end
 
+-- read data from stream
+function sandbox_core_compress_lz4_cstream.read(stream, buff, size, opt)
+    local real, data_or_errors = stream:_read(buff, size, opt)
+    if real < 0 and data_or_errors then
+        raise(data_or_errors)
+    end
+    return real, data_or_errors
+end
+
+-- write data to stream
+function sandbox_core_compress_lz4_cstream.write(stream, data, beof, opt)
+    local real, errors = stream:_write(data, beof, opt)
+    if real < 0 and errors then
+        raise(errors)
+    end
+    return real
+end
+
+-- read data from stream
+function sandbox_core_compress_lz4_dstream.read(stream, buff, size, opt)
+    local real, data_or_errors = stream:_read(buff, size, opt)
+    if real < 0 and data_or_errors then
+        raise(data_or_errors)
+    end
+    return real, data_or_errors
+end
+
+-- write data to stream
+function sandbox_core_compress_lz4_dstream.write(stream, data, beof, opt)
+    local real, errors = stream:_write(data, beof, opt)
+    if real < 0 and errors then
+        raise(errors)
+    end
+    return real
+end
+
 -- compress frame data
 function sandbox_core_compress_lz4.compress(data, opt)
     local result, errors = lz4.compress(data, opt)