SystemSh0cker/gocsaf
1
0
Fork 0
mirror of https://github.com/gocsaf/csaf.git synced 2025-12-22 11:55:40 +01:00
Code Issues Releases Activity Actions

Add extra http header support to downloader and checker.

Browse source
This commit is contained in:
Sascha L. Teichmann 2023-01-25 16:45:05 +01:00
parent 732383561b
commit 51fba46893
7 changed files with 117 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

30
cmd/csaf_checker/processor.go
View file

@ -351,24 +351,30 @@ func (p *processor) httpClient() util.Client {
TLSClientConfig: &tlsConfig,
}
var client util.Client
client := util.Client(&hClient)
// Add extra headers.
if len(p.opts.ExtraHeader) > 0 {
client = &util.HeaderClient{
Client: client,
Header: p.opts.ExtraHeader,
}
}
// Add optional URL logging.
if p.opts.Verbose {
client = &util.LoggingClient{Client: &hClient}
} else {
client = &hClient
client = &util.LoggingClient{Client: client}
}
if p.opts.Rate == nil {
p.client = client
return client
}
p.client = &util.LimitingClient{
Client: client,
Limiter: rate.NewLimiter(rate.Limit(*p.opts.Rate), 1),
// Add optional rate limiting.
if p.opts.Rate != nil {
client = &util.LimitingClient{
Client: client,
Limiter: rate.NewLimiter(rate.Limit(*p.opts.Rate), 1),
}
}
p.client = client
return p.client
}