fix stream size

1 files changed, 11 insertions, 11 deletions

diff --git a/xmake/modules/private/service/stream.lua b/xmake/modules/private/service/stream.lua
index 5f9aa127b..954660eb4 100644
--- a/xmake/modules/private/service/stream.lua
+++ b/xmake/modules/private/service/stream.lua
@@ -115,8 +115,8 @@ end
 function stream:send_string(str)
     local buff = self._BUFF
     local size = #str
-    buff:u16be_set(1, size)
-    if self:send(buff, 1, 2) then
+    buff:u32be_set(1, size)
+    if self:send(buff, 1, 4) then
         return self:send(bytes(str), 1, size)
     end
 end
@@ -127,8 +127,8 @@ function stream:send_file(filepath)
     -- send size
     local buff = self._BUFF
     local size = os.filesize(filepath)
-    buff:u16be_set(1, size)
-    if not self:send(buff, 1, 2) then
+    buff:u32be_set(1, size)
+    if not self:send(buff, 1, 4) then
         return
     end
 
@@ -153,7 +153,7 @@ end
 
 -- recv the given bytes
 function stream:recv(buff, size)
-    assert(size <= buff:size())
+    assert(size <= buff:size(), "too large size(%d)", size)
 
     -- read data from cache first
     local buffsize = 0
@@ -211,11 +211,11 @@ function stream:recv(buff, size)
     end
 end
 
--- recv u16be
-function stream:recv_u16be()
-    local data = self:recv(self._BUFF, 2)
+-- recv u32be
+function stream:recv_u32be()
+    local data = self:recv(self._BUFF, 4)
     if data then
-        return data:u16be(1)
+        return data:u32be(1)
     end
 end
 
@@ -241,7 +241,7 @@ end
 
 -- recv string
 function stream:recv_string()
-    local size = self:recv_u16be()
+    local size = self:recv_u32be()
     if size then
         local data = self:recv(self._BUFF, size)
         if data then
@@ -252,7 +252,7 @@ end
 
 -- recv file
 function stream:recv_file(filepath)
-    local size = self:recv_u16be()
+    local size = self:recv_u32be()
     if size then
         local buff = self._BUFF
         local recv = 0