This commit is contained in:
Jeff Becker 2019-01-19 08:36:22 -05:00
parent eefe4c024f
commit 939a8cee96
No known key found for this signature in database
GPG Key ID: F357B3B42F6F9B05
1 changed files with 5 additions and 0 deletions

View File

@ -42,14 +42,19 @@ func (h *httpProxyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
}
hijacker, ok := w.(http.Hijacker)
if !ok {
outConn.Close()
http.Error(w, "hijack disallowed", http.StatusInternalServerError)
return
}
conn, _, err := hijacker.Hijack()
if err != nil {
outConn.Close()
http.Error(w, err.Error(), http.StatusServiceUnavailable)
return
}
w.WriteHeader(http.StatusOK)
go transfer(conn, outConn)
go transfer(outConn, conn)
} else {
w.WriteHeader(http.StatusMethodNotAllowed)
}