github-personal/databunker
1
0
Fork 0
mirror of https://github.com/outbackdingo/databunker.git synced 2026-01-27 18:18:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

make sure to add keep-alive HTTP header

Browse Source
This commit is contained in:
root
2022-01-13 23:53:49 +00:00
parent 3c6a601bb0
commit 4f7b423934
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/bunker.go
View File

@@ -257,11 +257,11 @@ func (e mainEnv) setupRouter() *httprouter.Router {
log.Printf("error: %s\n", err.Error())
w.WriteHeader(404)
} else {
w.WriteHeader(200)
captcha, err := generateCaptcha()
if err != nil {
w.WriteHeader(501)
} else {
w.WriteHeader(200)
data2 := bytes.ReplaceAll(data, []byte("%CAPTCHAURL%"), []byte(captcha))
w.Write(data2)
}
@@ -411,6 +411,7 @@ func reqMiddleware(handler http.Handler) http.Handler {
//log.Printf("Set host %s\n", r.Host)
autocontext.Set(r, "host", r.Host)
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Connection", "keep-alive")
w2 := NewCustomResponseWriter(w)
if strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
w2.Header().Set("Vary", "Accept-Encoding")