teach/adlin
Archived
0
0
Fork 0
Code Issues 1 Pull Requests 4 Releases Activity

token-validator: don't prepend prefix when redirection are outside the site

Browse Source
This commit is contained in:
nemunaire 2020-03-01 18:21:04 +01:00
parent e1f7027109
commit 3837513270
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
token-validator/main.go
View File

@ -27,7 +27,7 @@ func (r ResponseWriterPrefix) Header() http.Header {
} }
func (r ResponseWriterPrefix) WriteHeader(s int) { func (r ResponseWriterPrefix) WriteHeader(s int) {
if v, exists := r.real.Header()["Location"]; exists { if v, exists := r.real.Header()["Location"]; exists && len(v) > 0 && !strings.HasPrefix(v[0], "https://") && !strings.HasPrefix(v[0], "http://") {
r.real.Header().Set("Location", r.prefix+v[0]) r.real.Header().Set("Location", r.prefix+v[0])
} }
r.real.WriteHeader(s) r.real.WriteHeader(s)