return if hijack failed

darren · Feb 5, 2019 · 4824637 · 4824637
1 parent 2cd8a2a
commit 4824637
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -89,10 +89,11 @@ func (s *Server) handleConnect(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, "Hijacking not supported", http.StatusInternalServerError)
 		return
 	}
-	src, buf, err := hijacker.Hijack()
 
+	src, buf, err := hijacker.Hijack()
 	if err != nil {
 		http.Error(w, err.Error(), http.StatusServiceUnavailable)
+		return
 	}
 
 	src = combine(buf, src)