fix websocket for tunnel

2023-10-17 23:04:30 +08:00
parent 0a565120df
commit a9f0dda805
13 changed files with 83 additions and 186 deletions
--- a/handler/forward/remote/handler.go
+++ b/handler/forward/remote/handler.go
@ -276,6 +276,14 @@ func (h *forwardHandler) handleHTTP(ctx context.Context, rw io.ReadWriter, remot
 				return resp.Write(rw)
 			}

+			if req.Header.Get("Upgrade") == "websocket" {
+				err := xnet.CopyBuffer(cc, br, 8192)
+				if err == nil {
+					err = io.EOF
+				}
+				return err
+			}
+
 			res, err := http.ReadResponse(bufio.NewReader(cc), req)
 			if err != nil {
 				log.Warnf("read response from node %s(%s): %v", target.Name, target.Addr, err)