lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-24 20:23:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

check full frame size

Browse Source
This commit is contained in:
zixun
2019-07-26 22:45:52 +08:00
committed by 云风
parent 8aac2114e3
commit b5ea18f44b
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lualib/http/websocket.lua
View File

@@ -250,6 +250,7 @@ local function resolve_accept(self)
local header = err
try_handle(self, "handshake", header)
local recv_count = 0
local recv_buf = {}
while true do
if _isws_closed(self.id) then
@@ -271,10 +272,15 @@ local function resolve_accept(self)
try_handle(self, "message", payload_data)
else
recv_buf[#recv_buf+1] = payload_data
recv_count = recv_count + #payload_data
if recv_count > MAX_FRAME_SIZE then
error("payload_len is too large")
end
if fin then
local s = table.concat(recv_buf)
try_handle(self, "message", s)
recv_buf = {} -- clear recv_buf
recv_count = 0
end
end
end