lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-25 12:43:09 +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 local header = err
try_handle(self, "handshake", header) try_handle(self, "handshake", header)
local recv_count = 0
local recv_buf = {} local recv_buf = {}
while true do while true do
if _isws_closed(self.id) then if _isws_closed(self.id) then
@@ -271,10 +272,15 @@ local function resolve_accept(self)
try_handle(self, "message", payload_data) try_handle(self, "message", payload_data)
else else
recv_buf[#recv_buf+1] = payload_data 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 if fin then
local s = table.concat(recv_buf) local s = table.concat(recv_buf)
try_handle(self, "message", s) try_handle(self, "message", s)
recv_buf = {} -- clear recv_buf recv_buf = {} -- clear recv_buf
recv_count = 0
end end
end end
end end