This commit is contained in:
Tristan Weir 2018-12-17 14:44:12 -08:00
Родитель 14b886b7e1
Коммит 0bb216ac37
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 1C5E6CBF4E06FB04
1 изменённых файлов: 12 добавлений и 5 удалений

Просмотреть файл

@ -273,7 +273,11 @@ func GetAuthToken(api ServiceApi) (string, error) {
}`, api.ClientID, api.ClientSecret, api.URL)) }`, api.ClientID, api.ClientSecret, api.URL))
req, _ := http.NewRequest("POST", api.AuthEndpoint, payload) req, err := http.NewRequest("POST", api.AuthEndpoint, payload)
if err != nil {
return "", err
}
req.Header.Add("content-type", "application/json") req.Header.Add("content-type", "application/json")
res, err := http.DefaultClient.Do(req) res, err := http.DefaultClient.Do(req)
@ -282,7 +286,10 @@ func GetAuthToken(api ServiceApi) (string, error) {
} }
defer res.Body.Close() defer res.Body.Close()
bodyJSON, _ := ioutil.ReadAll(res.Body) bodyJSON, err := ioutil.ReadAll(res.Body)
if err != nil {
return "", err
}
// unpack the JSON into an Auth0 token struct // unpack the JSON into an Auth0 token struct
var body Auth0Token var body Auth0Token
@ -299,7 +306,7 @@ func GetAuthToken(api ServiceApi) (string, error) {
// query a ServiceAPI instance for the set of all assets // query a ServiceAPI instance for the set of all assets
// load them into a searchable map, keyed to asset hostname // load them into a searchable map, keyed to asset hostname
// the ServiceAPI object must already be loaded with a Bearer token // the ServiceAPI object must already be loaded with a Bearer token
func GetAssets(m map[string]ServiceApiAsset, api ServiceApi) (err error){ func GetAssets(m map[string]ServiceApiAsset, api ServiceApi) (error){
// get json array of assets from serviceapi // get json array of assets from serviceapi
requestURL := api.URL + "api/v1/assets/" requestURL := api.URL + "api/v1/assets/"
@ -341,7 +348,7 @@ func GetAssets(m map[string]ServiceApiAsset, api ServiceApi) (err error){
m[tempAsset.AssetIdentifier] = tempAsset m[tempAsset.AssetIdentifier] = tempAsset
} }
return return err
} }
// return the operator and team for a given hostname, provided they are in the map of // return the operator and team for a given hostname, provided they are in the map of
@ -351,7 +358,7 @@ func LookupOperatorTeam(hostname string, m map[string]ServiceApiAsset) (operator
operator = m[hostname].Operator operator = m[hostname].Operator
team = m[hostname].Team team = m[hostname].Team
return return operator, team
} }
// cvssFromRisk returns a synthesized CVSS score as a string given a risk label // cvssFromRisk returns a synthesized CVSS score as a string given a risk label