From 1b8a0c3b93dba1720e91e88fbb50eb5f0de5ddb2 Mon Sep 17 00:00:00 2001
From: Rob Best <robertbest89@gmail.com>
Date: Sun, 28 Apr 2024 18:54:55 +0100
Subject: [PATCH] Use custom User-Agent header (#178)

---
 prober/http_file.go | 1 +
 prober/https.go     | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/prober/http_file.go b/prober/http_file.go
index 1baccc4..d2194f5 100644
--- a/prober/http_file.go
+++ b/prober/http_file.go
@@ -35,6 +35,7 @@ func ProbeHTTPFile(ctx context.Context, logger log.Logger, target string, module
 	if err != nil {
 		return fmt.Errorf("creating http request: %w", err)
 	}
+	req.Header.Set("User-Agent", userAgent)
 	resp, err := client.Do(req)
 	if err != nil {
 		return fmt.Errorf("making http request: %w", err)
diff --git a/prober/https.go b/prober/https.go
index f5e0bac..82e4412 100644
--- a/prober/https.go
+++ b/prober/https.go
@@ -12,9 +12,12 @@ import (
 	"github.com/go-kit/log"
 	"github.com/go-kit/log/level"
 	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/common/version"
 	"github.com/ribbybibby/ssl_exporter/v2/config"
 )
 
+var userAgent = fmt.Sprintf("SSLExporter/%s", version.Version)
+
 // ProbeHTTPS performs a https probe
 func ProbeHTTPS(ctx context.Context, logger log.Logger, target string, module config.Module, registry *prometheus.Registry) error {
 	tlsConfig, err := newTLSConfig("", registry, &module.TLSConfig)
@@ -57,6 +60,7 @@ func ProbeHTTPS(ctx context.Context, logger log.Logger, target string, module co
 		return err
 	}
 	request = request.WithContext(ctx)
+	request.Header.Set("User-Agent", userAgent)
 	resp, err := client.Do(request)
 	if err != nil {
 		return err