arno/clair
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update analyze-local-image with new API

Browse Source 
Signed-off-by: liangchenye <liangchenye@huawei.com>
This commit is contained in:
liangchenye 2016-02-16 15:22:25 +08:00 committed by Jimmy Zelinskie
parent 16a652fa47
commit 9bad9be19b
1 changed files with 35 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
contrib/analyze-local-images/main.go
View File

@ -29,22 +29,16 @@ import (
"strconv" "strconv"
"strings" "strings"
"time" "time"
"github.com/coreos/clair/api/v1"
) )
const ( const (
postLayerURI = "/v1/layers" postLayerURI = "/v1/layers"
getLayerVulnerabilitiesURI = "/v1/layers/%s/vulnerabilities?minimumPriority=%s" getLayerFeaturesURI = "/v1/layers/%s?features=true"
httpPort = 9279 httpPort = 9279
) )
type APIVulnerabilitiesResponse struct {
Vulnerabilities []APIVulnerability
}
type APIVulnerability struct {
ID, Link, Priority, Description, CausedByPackage string
}
func main() { func main() {
endpoint := flag.String("endpoint", "http://127.0.0.1:6060", "Address to Clair API") endpoint := flag.String("endpoint", "http://127.0.0.1:6060", "Address to Clair API")
myAddress := flag.String("my-address", "127.0.0.1", "Address from the point of view of Clair") myAddress := flag.String("my-address", "127.0.0.1", "Address from the point of view of Clair")
@ -118,21 +112,28 @@ func main() {
} }
// Get vulnerabilities // Get vulnerabilities
fmt.Println("Getting image's vulnerabilities") fmt.Println("Getting image's features")
vulnerabilities, err := getVulnerabilities(*endpoint, layerIDs[len(layerIDs)-1], *minimumPriority) features, err := getFeatures(*endpoint, layerIDs[len(layerIDs)-1], *minimumPriority)
if err != nil { if err != nil {
log.Fatalf("- Could not get vulnerabilities: %s\n", err) log.Fatalf("- Could not get features: %s\n", err)
} }
if len(vulnerabilities) == 0 {
isSafe := true
for _, feature := range features {
if len(feature.Vulnerabilities) != 0 {
isSafe = false
fmt.Printf("- # Feature: %s %s %s\n", feature.Name, feature.Namespace, feature.Version)
for _, vulnerability := range feature.Vulnerabilities {
fmt.Printf("\t - Name: %s\n", vulnerability.Name)
fmt.Printf("\t - Priority: %s\n", vulnerability.Severity)
fmt.Printf("\t - Link: %s\n", vulnerability.Link)
fmt.Printf("\t - Description: %s\n", vulnerability.Description)
}
}
}
if isSafe {
fmt.Println("Bravo, your image looks SAFE !") fmt.Println("Bravo, your image looks SAFE !")
} }
for _, vulnerability := range vulnerabilities {
fmt.Printf("- # %s\n", vulnerability.ID)
fmt.Printf(" - Priority: %s\n", vulnerability.Priority)
fmt.Printf(" - Link: %s\n", vulnerability.Link)
fmt.Printf(" - Package: %s\n", vulnerability.CausedByPackage)
fmt.Printf(" - Description: %s\n", vulnerability.Description)
}
} }
func save(imageName string) (string, error) { func save(imageName string) (string, error) {
@ -227,8 +228,8 @@ func historyFromCommand(imageName string) ([]string, error) {
return layers, nil return layers, nil
} }
func analyzeLayer(endpoint, path, layerID, parentLayerID string) error { func analyzeLayer(endpoint, path, layerName, parentLayerName string) error {
payload := struct{ ID, Path, ParentID, ImageFormat string }{ID: layerID, Path: path, ParentID: parentLayerID, ImageFormat: "Docker"} payload := v1.LayerEnvelope{Layer: &v1.Layer{Name: layerName, Path: path, ParentName: parentLayerName, Format: "Docker"}}
jsonPayload, err := json.Marshal(payload) jsonPayload, err := json.Marshal(payload)
if err != nil { if err != nil {
return err return err
@ -255,25 +256,26 @@ func analyzeLayer(endpoint, path, layerID, parentLayerID string) error {
return nil return nil
} }
func getVulnerabilities(endpoint, layerID, minimumPriority string) ([]APIVulnerability, error) { func getFeatures(endpoint, layerID, minimumPriority string) ([]v1.Feature, error) {
response, err := http.Get(endpoint + fmt.Sprintf(getLayerVulnerabilitiesURI, layerID, minimumPriority)) response, err := http.Get(endpoint + fmt.Sprintf(getLayerFeaturesURI, layerID))
if err != nil { if err != nil {
return []APIVulnerability{}, err return []v1.Feature{}, err
} }
defer response.Body.Close() defer response.Body.Close()
if response.StatusCode != 200 { if response.StatusCode != 200 {
body, _ := ioutil.ReadAll(response.Body) body, _ := ioutil.ReadAll(response.Body)
return []APIVulnerability{}, fmt.Errorf("Got response %d with message %s", response.StatusCode, string(body)) return []v1.Feature{}, fmt.Errorf("Got response %d with message %s", response.StatusCode, string(body))
} }
var apiResponse APIVulnerabilitiesResponse var apiResponse v1.LayerEnvelope
err = json.NewDecoder(response.Body).Decode(&apiResponse) if err = json.NewDecoder(response.Body).Decode(&apiResponse); err != nil {
if err != nil { return []v1.Feature{}, err
return []APIVulnerability{}, err } else if apiResponse.Error != nil {
return []v1.Feature{}, errors.New(apiResponse.Error.Message)
} }
return apiResponse.Vulnerabilities, nil return apiResponse.Layer.Features, nil
} }
func restrictedFileServer(path, allowedHost string) http.Handler { func restrictedFileServer(path, allowedHost string) http.Handler {