FreakU
/
fedproxy
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

more

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
main.go
View File

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