feat: add an option to bypass our encoding

Author: mrhaoxx

http/http.go

@@ -223,6 +223,13 @@ func (w *NgResponseWriter) Header() http.Header {
 	return w.stdrw.Header()
 }
 
+func (w *NgResponseWriter) BypassEncoding() {
+	if w.acceptEncoding != "" {
+		w.Header().Set("Accept-Encoding", w.acceptEncoding)
+		w.acceptEncoding = ""
+	}
+}
+
 func (w *NgResponseWriter) initForWrite() {
 	w.Header().Set("Server", utils.ServerSign)
 	switch w.code {

http/proxy.go

@@ -24,6 +24,7 @@ type HttpHost struct {
 	Backend            *net.URL
 	InsecureSkipVerify bool
 	MaxConnsPerHost    int
+	BypassEncoding     bool
 	Underlying         net.Interface
 
 	proxy   *httputil.ReverseProxy
@@ -171,6 +172,10 @@ func (h *ReverseProxy) HandleHTTP(ctx *HttpCtx) Ret {
 		recover()
 	}()
 
+	if host.BypassEncoding {
+		ctx.Resp.BypassEncoding()
+	}
+
 	if ctx.Req.Header.Get("Upgrade") == "websocket" {
 		host.wsproxy.ServeHTTP(ctx.Resp, ctx.Req)
 	} else {
@@ -187,14 +192,15 @@ func (hpx *ReverseProxy) GetHosts() []*HttpHost {
 	return hpx.hosts
 }
 
-func (hpx *ReverseProxy) Insert(index int, id string, hosts []string, backend *net.URL, MaxConnsPerHost int, InsecureSkipVerify bool) error {
+func (hpx *ReverseProxy) Insert(index int, id string, hosts []string, backend *net.URL, MaxConnsPerHost int, InsecureSkipVerify bool, BypassEncoding bool) error {
 	buf := HttpHost{
 		Id:                 id,
 		ServerName:         utils.MustCompileRegexp(dns.Dnsnames2Regexps(hosts)),
 		Backend:            backend,
 		MaxConnsPerHost:    MaxConnsPerHost,
 		InsecureSkipVerify: InsecureSkipVerify,
 		Underlying:         backend.Underlying,
+		BypassEncoding:     BypassEncoding,
 	}
 	buf.Init()
 

ui/builtin.go

@@ -148,6 +148,11 @@ var _builtin_refs_assertions = map[string]Assert{
 								Default: false,
 								Desc:    "skip TLS certificate verification for backend",
 							},
+							"BypassEncoding": {
+								Type:    "bool",
+								Default: false,
+								Desc:    "bypass encoding for backend",
+							},
 						},
 					},
 				},
@@ -1003,15 +1008,17 @@ var _builtin_refs = map[string]Inst{
 			backend := host.MustGet("backend").ToURL()
 			maxconns := host.MustGet("MaxConnsPerHost").ToInt()
 			tlsskip := host.MustGet("TlsSkipVerify").ToBool()
+			bypassencoding := host.MustGet("BypassEncoding").ToBool()
 
-			proxier.Insert(id, name, hosts, backend, maxconns, tlsskip)
+			proxier.Insert(id, name, hosts, backend, maxconns, tlsskip, bypassencoding)
 
 			zlog.Debug().
 				Str("name", name).
 				Strs("hosts", hosts).
 				Str("backend", backend.String()).
 				Int("maxconns", maxconns).
 				Bool("tlsskip", tlsskip).
+				Bool("bypassencoding", bypassencoding).
 				Msg("new http reverse host")
 		}