blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2025-10-15 13:02:14 +08:00
Code Issues Projects Releases Wiki Activity

Prevention of Hijack() runtime panics

Browse Source
This commit is contained in:
keitosuwahara 2025-07-13 22:57:28 +09:00
parent 688a429d19
commit b5118f65a2
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
response_writer.go
View File

@ -6,6 +6,7 @@ package gin
import ( import (
"bufio" "bufio"
"errors"
"io" "io"
"net" "net"
"net/http" "net/http"
@ -16,6 +17,8 @@ const (
defaultStatus = http.StatusOK defaultStatus = http.StatusOK
) )
var errHijackAlreadyWritten = errors.New("gin: response already written")
// ResponseWriter ... // ResponseWriter ...
type ResponseWriter interface { type ResponseWriter interface {
http.ResponseWriter http.ResponseWriter
@ -106,6 +109,9 @@ func (w *responseWriter) Written() bool {
// Hijack implements the http.Hijacker interface. // Hijack implements the http.Hijacker interface.
func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) { func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
if w.Written() {
return nil, nil, errHijackAlreadyWritten
}
if w.size < 0 { if w.size < 0 {
w.size = 0 w.size = 0
} }