TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-07 18:50:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
GoEdgeLab
2022-08-04 11:34:06 +08:00
parent 9c00b07d9e
commit bcd6784a8f
31 changed files with 66 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/waf/requests/test_request.go
View File

@@ -5,7 +5,6 @@ package requests
import (
"bytes"
"io"
"io/ioutil"
"net"
"net/http"
)
@@ -48,7 +47,7 @@ func (this *TestRequest) WAFRemoteIP() string {
func (this *TestRequest) WAFReadBody(max int64) (data []byte, err error) {
if this.req.ContentLength > 0 {
data, err = ioutil.ReadAll(io.LimitReader(this.req.Body, max))
data, err = io.ReadAll(io.LimitReader(this.req.Body, max))
}
return
}
@@ -58,7 +57,7 @@ func (this *TestRequest) WAFRestoreBody(data []byte) {
rawReader := bytes.NewBuffer(data)
buf := make([]byte, 1024)
_, _ = io.CopyBuffer(rawReader, this.req.Body, buf)
this.req.Body = ioutil.NopCloser(rawReader)
this.req.Body = io.NopCloser(rawReader)
}
}