Skip to content

Commit 1b9042b

Browse files
authored
reverseproxy: handle buffered data during hijack (#6274)
1 parent 4d6370b commit 1b9042b

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

modules/caddyhttp/responsewriter.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -263,6 +263,8 @@ func (rr *responseRecorder) Hijack() (net.Conn, *bufio.ReadWriter, error) {
263263
conn.(*hijackedConn).updateReadSize(buffered)
264264
data, _ := brw.Peek(buffered)
265265
brw.Reader.Reset(io.MultiReader(bytes.NewReader(data), conn))
266+
// peek to make buffered data appear, as Reset will make it 0
267+
_, _ = brw.Peek(buffered)
266268
} else {
267269
brw.Reader.Reset(conn)
268270
}

modules/caddyhttp/reverseproxy/streaming.go

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,17 @@ func (h *Handler) handleUpgradeResponse(logger *zap.Logger, wg *sync.WaitGroup,
101101
return
102102
}
103103

104+
// There may be buffered data in the *bufio.Reader
105+
// see: https://github.com/caddyserver/caddy/issues/6273
106+
if buffered := brw.Reader.Buffered(); buffered > 0 {
107+
data, _ := brw.Peek(buffered)
108+
_, err := backConn.Write(data)
109+
if err != nil {
110+
logger.Debug("backConn write failed", zap.Error(err))
111+
return
112+
}
113+
}
114+
104115
// Ensure the hijacked client connection, and the new connection established
105116
// with the backend, are both closed in the event of a server shutdown. This
106117
// is done by registering them. We also try to gracefully close connections

0 commit comments

Comments
 (0)