Merge pull request #873 from SSWConsulting/handle-nil-header-request

Golang - Handle nil header request
SSWConsulting · Apr 22, 2024 · d99032a · d99032a
2 parents 664f389 + 0b7150a
commit d99032a
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/docker/sswlinkauditor.go b/docker/sswlinkauditor.go
@@ -63,10 +63,12 @@ func getClient() *http.Client {
 }
 
 func addClientHeaders(r *http.Request) {
-    r.Header.Add("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36")
-	r.Header.Set("Cache-Control", "no-cache")
-	r.Header.Set("Connection", "keep-alive")
-	r.Header.Set("Accept-Encoding", "*")
+	if r != nil {
+		r.Header.Add("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36")
+		r.Header.Set("Cache-Control", "no-cache")
+		r.Header.Set("Connection", "keep-alive")
+		r.Header.Set("Accept-Encoding", "*")
+	}
 }
 
 func check(link Link, linkch chan LinkStatus, number int) {
@@ -78,8 +80,10 @@ func check(link Link, linkch chan LinkStatus, number int) {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 	r, e := http.NewRequestWithContext(ctx, "GET", link.url, nil)
-	addClientHeaders(r)
-	r.Header.Add("Accept", "*/*")
+	if r != nil {
+		addClientHeaders(r)
+		r.Header.Add("Accept", "*/*")
+	}
 	dnsErr := new(net.DNSError)
 
 	if e != nil {