diff --git a/internal/waf/requests/request.go b/internal/waf/requests/request.go index a1b982b..28c29b9 100644 --- a/internal/waf/requests/request.go +++ b/internal/waf/requests/request.go @@ -23,13 +23,17 @@ func (this *Request) Raw() *http.Request { } func (this *Request) ReadBody(max int64) (data []byte, err error) { - data, err = ioutil.ReadAll(io.LimitReader(this.Request.Body, max)) + if this.Request.ContentLength > 0 { + data, err = ioutil.ReadAll(io.LimitReader(this.Request.Body, max)) + } return } func (this *Request) RestoreBody(data []byte) { - rawReader := bytes.NewBuffer(data) - buf := make([]byte, 1024) - io.CopyBuffer(rawReader, this.Request.Body, buf) - this.Request.Body = ioutil.NopCloser(rawReader) + if len(data) > 0 { + rawReader := bytes.NewBuffer(data) + buf := make([]byte, 1024) + _, _ = io.CopyBuffer(rawReader, this.Request.Body, buf) + this.Request.Body = ioutil.NopCloser(rawReader) + } }