go-gost/x
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update handler recorder object

Browse Source
This commit is contained in:
ginuerzh
2024-09-15 10:51:23 +08:00
parent 69455ace9d
commit f681f7aa03
8 changed files with 108 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files
handler/forward/local/handler.go
+7 -3
View File
@@ -247,12 +247,15 @@ func (h *forwardHandler) handleHTTP(ctx context.Context, rw io.ReadWriteCloser,
Scheme: req.URL.Scheme, Scheme: req.URL.Scheme,
Method: req.Method, Method: req.Method,
URI: req.RequestURI, URI: req.RequestURI,
RequestHeader: req.Header.Clone(), Request: xrecorder.HTTPRequestRecorderObject{
ContentLength: req.ContentLength,
Header: req.Header.Clone(),
},
} }
defer func() { defer func() {
if err != nil { if err != nil {
ro.HTTP.StatusCode = resp.StatusCode ro.HTTP.StatusCode = resp.StatusCode
ro.HTTP.ResponseHeader = resp.Header ro.HTTP.Response.Header = resp.Header
ro.Duration = time.Since(start) ro.Duration = time.Since(start)
ro.Err = err.Error() ro.Err = err.Error()
@@ -393,7 +396,8 @@ func (h *forwardHandler) handleHTTP(ctx context.Context, rw io.ReadWriteCloser,
ro.Err = err.Error() ro.Err = err.Error()
} }
if res != nil && ro.HTTP != nil { if res != nil && ro.HTTP != nil {
ro.HTTP.ResponseHeader = res.Header ro.HTTP.Response.ContentLength = res.ContentLength
ro.HTTP.Response.Header = res.Header
ro.HTTP.StatusCode = res.StatusCode ro.HTTP.StatusCode = res.StatusCode
} }
ro.Record(ctx, h.recorder.Recorder) ro.Record(ctx, h.recorder.Recorder)
handler/forward/remote/handler.go
+7 -3
View File
@@ -247,12 +247,15 @@ func (h *forwardHandler) handleHTTP(ctx context.Context, rw io.ReadWriteCloser,
Scheme: req.URL.Scheme, Scheme: req.URL.Scheme,
Method: req.Method, Method: req.Method,
URI: req.RequestURI, URI: req.RequestURI,
RequestHeader: req.Header.Clone(), Request: xrecorder.HTTPRequestRecorderObject{
ContentLength: req.ContentLength,
Header: req.Header.Clone(),
},
} }
defer func() { defer func() {
if err != nil { if err != nil {
ro.HTTP.StatusCode = resp.StatusCode ro.HTTP.StatusCode = resp.StatusCode
ro.HTTP.ResponseHeader = resp.Header ro.HTTP.Response.Header = resp.Header
ro.Duration = time.Since(start) ro.Duration = time.Since(start)
ro.Err = err.Error() ro.Err = err.Error()
@@ -394,7 +397,8 @@ func (h *forwardHandler) handleHTTP(ctx context.Context, rw io.ReadWriteCloser,
ro.Err = err.Error() ro.Err = err.Error()
} }
if res != nil && ro.HTTP != nil { if res != nil && ro.HTTP != nil {
ro.HTTP.ResponseHeader = res.Header ro.HTTP.Response.ContentLength = res.ContentLength
ro.HTTP.Response.Header = res.Header
ro.HTTP.StatusCode = res.StatusCode ro.HTTP.StatusCode = res.StatusCode
} }
ro.Record(ctx, h.recorder.Recorder) ro.Record(ctx, h.recorder.Recorder)
handler/http/handler.go
+11 -4
View File
@@ -208,11 +208,14 @@ func (h *httpHandler) handleRequest(ctx context.Context, conn net.Conn, req *htt
Scheme: req.URL.Scheme, Scheme: req.URL.Scheme,
Method: req.Method, Method: req.Method,
URI: req.RequestURI, URI: req.RequestURI,
RequestHeader: req.Header.Clone(), Request: xrecorder.HTTPRequestRecorderObject{
ContentLength: req.ContentLength,
Header: req.Header.Clone(),
},
} }
defer func() { defer func() {
ro.HTTP.StatusCode = resp.StatusCode ro.HTTP.StatusCode = resp.StatusCode
ro.HTTP.ResponseHeader = resp.Header ro.HTTP.Response.Header = resp.Header
}() }()
clientID, ok := h.authenticate(ctx, conn, req, resp, log) clientID, ok := h.authenticate(ctx, conn, req, resp, log)
@@ -339,8 +342,11 @@ func (h *httpHandler) handleProxy(ctx context.Context, rw io.ReadWriteCloser, cc
Scheme: req.URL.Scheme, Scheme: req.URL.Scheme,
Method: req.Method, Method: req.Method,
URI: req.RequestURI, URI: req.RequestURI,
RequestHeader: req.Header.Clone(),
StatusCode: resp.StatusCode, StatusCode: resp.StatusCode,
Request: xrecorder.HTTPRequestRecorderObject{
ContentLength: req.ContentLength,
Header: req.Header.Clone(),
},
} }
} }
@@ -378,7 +384,8 @@ func (h *httpHandler) handleProxy(ctx context.Context, rw io.ReadWriteCloser, cc
ro.Err = err.Error() ro.Err = err.Error()
} }
if res != nil && ro.HTTP != nil { if res != nil && ro.HTTP != nil {
ro.HTTP.ResponseHeader = res.Header ro.HTTP.Response.ContentLength = res.ContentLength
ro.HTTP.Response.Header = res.Header
ro.HTTP.StatusCode = res.StatusCode ro.HTTP.StatusCode = res.StatusCode
} }
ro.Record(ctx, h.recorder.Recorder) ro.Record(ctx, h.recorder.Recorder)
handler/http2/handler.go
+5 -2
View File
@@ -203,11 +203,14 @@ func (h *http2Handler) roundTrip(ctx context.Context, w http.ResponseWriter, req
Scheme: req.URL.Scheme, Scheme: req.URL.Scheme,
Method: req.Method, Method: req.Method,
URI: req.RequestURI, URI: req.RequestURI,
RequestHeader: req.Header.Clone(), Request: xrecorder.HTTPRequestRecorderObject{
ContentLength: req.ContentLength,
Header: req.Header.Clone(),
},
} }
defer func() { defer func() {
ro.HTTP.StatusCode = resp.StatusCode ro.HTTP.StatusCode = resp.StatusCode
ro.HTTP.ResponseHeader = resp.Header ro.HTTP.Response.Header = resp.Header
}() }()
clientID, ok := h.authenticate(ctx, w, req, resp, log) clientID, ok := h.authenticate(ctx, w, req, resp, log)
handler/redirect/tcp/handler.go
+6 -2
View File
@@ -185,7 +185,10 @@ func (h *redirectHandler) handleHTTP(ctx context.Context, rw io.ReadWriter, radd
Scheme: req.URL.Scheme, Scheme: req.URL.Scheme,
Method: req.Method, Method: req.Method,
URI: req.RequestURI, URI: req.RequestURI,
RequestHeader: req.Header, Request: xrecorder.HTTPRequestRecorderObject{
ContentLength: req.ContentLength,
Header: req.Header,
},
} }
if log.IsLevelEnabled(logger.TraceLevel) { if log.IsLevelEnabled(logger.TraceLevel) {
@@ -246,7 +249,8 @@ func (h *redirectHandler) handleHTTP(ctx context.Context, rw io.ReadWriter, radd
defer resp.Body.Close() defer resp.Body.Close()
ro.HTTP.StatusCode = resp.StatusCode ro.HTTP.StatusCode = resp.StatusCode
ro.HTTP.ResponseHeader = resp.Header ro.HTTP.Response.ContentLength = resp.ContentLength
ro.HTTP.Response.Header = resp.Header
if log.IsLevelEnabled(logger.TraceLevel) { if log.IsLevelEnabled(logger.TraceLevel) {
dump, _ := httputil.DumpResponse(resp, false) dump, _ := httputil.DumpResponse(resp, false)
handler/sni/handler.go
+6 -2
View File
@@ -134,7 +134,10 @@ func (h *sniHandler) handleHTTP(ctx context.Context, rw io.ReadWriter, raddr net
Scheme: req.URL.Scheme, Scheme: req.URL.Scheme,
Method: req.Method, Method: req.Method,
URI: req.RequestURI, URI: req.RequestURI,
RequestHeader: req.Header, Request: xrecorder.HTTPRequestRecorderObject{
ContentLength: req.ContentLength,
Header: req.Header,
},
} }
if log.IsLevelEnabled(logger.TraceLevel) { if log.IsLevelEnabled(logger.TraceLevel) {
@@ -192,7 +195,8 @@ func (h *sniHandler) handleHTTP(ctx context.Context, rw io.ReadWriter, raddr net
defer resp.Body.Close() defer resp.Body.Close()
ro.HTTP.StatusCode = resp.StatusCode ro.HTTP.StatusCode = resp.StatusCode
ro.HTTP.ResponseHeader = resp.Header ro.HTTP.Response.ContentLength = resp.ContentLength
ro.HTTP.Response.Header = resp.Header
if log.IsLevelEnabled(logger.TraceLevel) { if log.IsLevelEnabled(logger.TraceLevel) {
dump, _ := httputil.DumpResponse(resp, false) dump, _ := httputil.DumpResponse(resp, false)
handler/tunnel/entrypoint.go
+7 -3
View File
@@ -96,7 +96,10 @@ func (ep *entrypoint) handle(ctx context.Context, conn net.Conn) error {
Proto: req.Proto, Proto: req.Proto,
Scheme: req.URL.Scheme, Scheme: req.URL.Scheme,
URI: req.RequestURI, URI: req.RequestURI,
RequestHeader: req.Header, Request: xrecorder.HTTPRequestRecorderObject{
ContentLength: req.ContentLength,
Header: req.Header,
},
}, },
} }
defer func() { defer func() {
@@ -107,7 +110,7 @@ func (ep *entrypoint) handle(ctx context.Context, conn net.Conn) error {
}).Debugf("%s >-< %s", conn.RemoteAddr(), req.Host) }).Debugf("%s >-< %s", conn.RemoteAddr(), req.Host)
ro.HTTP.StatusCode = resp.StatusCode ro.HTTP.StatusCode = resp.StatusCode
ro.HTTP.ResponseHeader = resp.Header ro.HTTP.Response.Header = resp.Header
ro.Duration = d ro.Duration = d
ro.Err = err.Error() ro.Err = err.Error()
@@ -243,8 +246,9 @@ func (ep *entrypoint) handle(ctx context.Context, conn net.Conn) error {
ro.Err = err.Error() ro.Err = err.Error()
} }
if res != nil && ro.HTTP != nil { if res != nil && ro.HTTP != nil {
ro.HTTP.ResponseHeader = res.Header
ro.HTTP.StatusCode = res.StatusCode ro.HTTP.StatusCode = res.StatusCode
ro.HTTP.Response.ContentLength = res.ContentLength
ro.HTTP.Response.Header = res.Header
} }
ro.Record(ctx, ep.recorder.Recorder) ro.Record(ctx, ep.recorder.Recorder)
}() }()
recorder/recorder.go
+12 -4
View File
@@ -15,6 +15,16 @@ const (
RecorderServiceHandlerTunnel = "recorder.service.handler.tunnel" RecorderServiceHandlerTunnel = "recorder.service.handler.tunnel"
) )
type HTTPRequestRecorderObject struct {
ContentLength int64 `json:"contentLength"`
Header http.Header `json:"header"`
}
type HTTPResponseRecorderObject struct {
ContentLength int64 `json:"contentLength"`
Header http.Header `json:"header"`
}
type HTTPRecorderObject struct { type HTTPRecorderObject struct {
Host string `json:"host"` Host string `json:"host"`
Method string `json:"method"` Method string `json:"method"`
@@ -22,10 +32,8 @@ type HTTPRecorderObject struct {
Scheme string `json:"scheme"` Scheme string `json:"scheme"`
URI string `json:"uri"` URI string `json:"uri"`
StatusCode int `json:"statusCode"` StatusCode int `json:"statusCode"`
RequestHeader http.Header `json:"requestHeader"` Request HTTPRequestRecorderObject `json:"request"`
ResponseHeader http.Header `json:"responseHeader"` Response HTTPResponseRecorderObject `json:"response"`
// RequestBody string
// ResponseBody string
} }
type DNSRecorderObject struct { type DNSRecorderObject struct {