add http body for handler recorder

handler/http/handler.go

@@ -39,6 +39,10 @@ import (
 	"github.com/go-gost/x/registry"
 )
 
+const (
+	defaultBodySize = 1024 * 1024 * 10 // 10MB
+)
+
 func init() {
 	registry.HandlerRegistry().Register("http", NewHandler)
 }
@@ -341,19 +345,17 @@ func (h *httpHandler) handleProxy(ctx context.Context, rw io.ReadWriteCloser, cc
 		start := time.Now()
 		ro.Time = start
 
-		if ro.HTTP != nil {
-			ro.HTTP = &xrecorder.HTTPRecorderObject{
-				Host:       req.Host,
-				Proto:      req.Proto,
-				Scheme:     req.URL.Scheme,
-				Method:     req.Method,
-				URI:        req.RequestURI,
-				StatusCode: resp.StatusCode,
-				Request: xrecorder.HTTPRequestRecorderObject{
-					ContentLength: req.ContentLength,
-					Header:        req.Header.Clone(),
-				},
-			}
+		ro.HTTP = &xrecorder.HTTPRecorderObject{
+			Host:       req.Host,
+			Proto:      req.Proto,
+			Scheme:     req.URL.Scheme,
+			Method:     req.Method,
+			URI:        req.RequestURI,
+			StatusCode: resp.StatusCode,
+			Request: xrecorder.HTTPRequestRecorderObject{
+				ContentLength: req.ContentLength,
+				Header:        req.Header.Clone(),
+			},
 		}
 
 		// HTTP/1.0
@@ -370,6 +372,18 @@ func (h *httpHandler) handleProxy(ctx context.Context, rw io.ReadWriteCloser, cc
 		req.Header.Del("Gost-Target")
 		req.Header.Del("X-Gost-Target")
 
+		var reqBody *xhttp.Body
+		if opts := h.recorder.Options; opts != nil && opts.HTTPBody {
+			if req.Body != nil {
+				maxSize := opts.MaxBodySize
+				if maxSize <= 0 {
+					maxSize = defaultBodySize
+				}
+				reqBody = xhttp.NewBody(req.Body, maxSize)
+				req.Body = reqBody
+			}
+		}
+
 		if err = req.Write(cc); err != nil {
 			resp.Write(rw)
 
@@ -380,19 +394,29 @@ func (h *httpHandler) handleProxy(ctx context.Context, rw io.ReadWriteCloser, cc
 			return err
 		}
 
+		if reqBody != nil {
+			ro.HTTP.Request.Body = reqBody.Content()
+			ro.HTTP.Request.ContentLength = reqBody.Length()
+		}
+
 		go func() {
 			var err error
 			var res *http.Response
+			var respBody *xhttp.Body
 
 			defer func() {
 				ro.Duration = time.Since(start)
 				if err != nil {
 					ro.Err = err.Error()
 				}
-				if res != nil && ro.HTTP != nil {
-					ro.HTTP.Response.ContentLength = res.ContentLength
-					ro.HTTP.Response.Header = res.Header
+				if res != nil {
 					ro.HTTP.StatusCode = res.StatusCode
+					ro.HTTP.Response.Header = res.Header
+					ro.HTTP.Response.ContentLength = res.ContentLength
+					if respBody != nil {
+						ro.HTTP.Response.Body = respBody.Content()
+						ro.HTTP.Response.ContentLength = respBody.Length()
+					}
 				}
 				ro.Record(ctx, h.recorder.Recorder)
 			}()
@@ -423,6 +447,15 @@ func (h *httpHandler) handleProxy(ctx context.Context, rw io.ReadWriteCloser, cc
 				res.ProtoMinor = req.ProtoMinor
 			}
 
+			if opts := h.recorder.Options; opts != nil && opts.HTTPBody {
+				maxSize := opts.MaxBodySize
+				if maxSize <= 0 {
+					maxSize = defaultBodySize
+				}
+				respBody = xhttp.NewBody(res.Body, maxSize)
+				res.Body = respBody
+			}
+
 			if err = res.Write(rw); err != nil {
 				rw.Close()
 				log.Errorf("write response: %v", err)