lib
/
guard
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: add support for hijack

master v0.1.0
Evan Chen 2021-11-19 15:08:51 +08:00
parent c5073d7a74
commit 6793e8ef1d
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
guard.go
View File

@ -1,7 +1,9 @@
package guard package guard
import ( import (
"bufio"
"encoding/base64" "encoding/base64"
"errors"
"fmt" "fmt"
"net" "net"
"net/http" "net/http"
@ -33,6 +35,14 @@ type responseWriter struct {
err string err string
} }
func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
h, ok := w.ResponseWriter.(http.Hijacker)
if !ok {
return nil, nil, errors.New("hijack not supported")
}
return h.Hijack()
}
func (w *responseWriter) WriteHeader(statusCode int) { func (w *responseWriter) WriteHeader(statusCode int) {
if w.StatueCode != 0 { if w.StatueCode != 0 {
return return