Fix header issue for update req

scheiblingco · Feb 4, 2024 · 0b43619 · 0b43619
1 parent 77201d5
commit 0b43619
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -235,8 +235,9 @@ func ListenHealthcheck(cfg *Config) {
 			return
 		}
 
-		header := w.Header().Get("X-Update-Token")
-		if header != updateToken {
+		headerToken := r.Header.Get("X-Update-Token")
+
+		if headerToken == "" || headerToken != updateToken {
 			w.WriteHeader(403)
 			w.Write([]byte("Invalid token"))
 			return