实现WAF

internal/waf/requests/request.go

@@ -0,0 +1,35 @@
+package requests
+
+import (
+	"bytes"
+	"io"
+	"io/ioutil"
+	"net/http"
+)
+
+type Request struct {
+	*http.Request
+	BodyData []byte
+}
+
+func NewRequest(raw *http.Request) *Request {
+	return &Request{
+		Request: raw,
+	}
+}
+
+func (this *Request) Raw() *http.Request {
+	return this.Request
+}
+
+func (this *Request) ReadBody(max int64) (data []byte, err error) {
+	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)
+}

internal/waf/requests/response.go

@@ -0,0 +1,15 @@
+package requests
+
+import "net/http"
+
+type Response struct {
+	*http.Response
+
+	BodyData []byte
+}
+
+func NewResponse(resp *http.Response) *Response {
+	return &Response{
+		Response: resp,
+	}
+}