user-agent based on version number #74

cloudflare · Nov 14, 2020 · 5246d51 · 5246d51
1 parent 6c1872d
commit 5246d51
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cmd/octorpki/octorpki.go b/cmd/octorpki/octorpki.go
@@ -67,6 +67,7 @@ var (
 	RRDP         = flag.Bool("rrdp", true, "Enable RRDP fetching")
 	RRDPFile     = flag.String("rrdp.file", "cache/rrdp.json", "Save RRDP state")
 	RRDPFailover = flag.Bool("rrdp.failover", true, "Failover to rsync when RRDP fails")
+	UserAgent    = flag.String("useragent", fmt.Sprintf("Cloudflare-RRDP-%v (+https://github.com/cloudflare/cfrpki)", AppVersion), "User-Agent header")
 
 	Mode       = flag.String("mode", "server", "Select output mode (server/oneoff)")
 	WaitStable = flag.Bool("output.wait", true, "Wait until stable state to create the file (returns 503 when unstable on HTTP)")
@@ -1233,7 +1234,7 @@ func main() {
 			},
 			log.StandardLogger()),
 		HTTPFetcher: &syncpki.HTTPFetcher{
-			UserAgent: "Cloudflare-RPKI-RRDP/1.0 (+https://rpki.cloudflare.com)",
+			UserAgent: *UserAgent,
 			Client:    &http.Client{},
 		},
 		ROAList: &prefixfile.ROAList{