guoswork/frp
1
0
Fork 0
mirror of https://gitee.com/IrisVega/frp.git synced 2024-11-01 22:31:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

models/msg: limit single package length

Browse Source
This commit is contained in:
fatedier 2016-08-07 13:18:54 +08:00
parent 3b06d771ac
commit 1da81ad7d3
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/frp/models/msg/process.go
View File

@ -146,6 +146,11 @@ func pipeDecrypt(r *conn.Conn, w *conn.Conn, conf config.BaseConf, needRecord bo
} }
cnt, res, left = unpkgMsg(newBuf) cnt, res, left = unpkgMsg(newBuf)
if cnt < 0 { if cnt < 0 {
// limit one package length, maximum is 1MB
if len(res) > 1024*1024 {
log.Warn("ProxyName [%s], package length exceeds the limit")
return fmt.Errorf("package length error")
}
continue continue
} }