Skip to content

Commit

Permalink
Merge pull request #557 from nikonhub/feature-change_default_http_use…
Browse files Browse the repository at this point in the history
…r_agent

Change default user agent
  • Loading branch information
SuperQ committed Jan 22, 2022
2 parents 70bff79 + 36b2245 commit e34a36b
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions prober/http.go
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ import (
"github.com/go-kit/log/level"
"github.com/prometheus/client_golang/prometheus"
pconfig "github.com/prometheus/common/config"
"github.com/prometheus/common/version"
"golang.org/x/net/publicsuffix"

"github.com/prometheus/blackbox_exporter/config"
Expand Down Expand Up @@ -232,6 +233,8 @@ func (bc *byteCounter) Read(p []byte) (int, error) {
return n, err
}

var userAgentDefaultHeader = fmt.Sprintf("Blackbox Exporter/%s", version.Version)

func ProbeHTTP(ctx context.Context, target string, module config.Module, registry *prometheus.Registry, logger log.Logger) (success bool) {
var redirects int
var (
Expand Down Expand Up @@ -430,6 +433,11 @@ func ProbeHTTP(ctx context.Context, target string, module config.Module, registr
request.Header.Set(key, value)
}

_, hasUserAgent := request.Header["User-Agent"]
if !hasUserAgent {
request.Header.Set("User-Agent", userAgentDefaultHeader)
}

trace := &httptrace.ClientTrace{
DNSStart: tt.DNSStart,
DNSDone: tt.DNSDone,
Expand Down

0 comments on commit e34a36b

Please sign in to comment.