311 lines
9.8 KiB
JavaScript
311 lines
9.8 KiB
JavaScript
|
/**
|
||
|
* An element which displays the features of an image.
|
||
|
*/
|
||
|
angular.module('quay').directive('imageFeatureView', function () {
|
||
|
var directiveDefinitionObject = {
|
||
|
priority: 0,
|
||
|
templateUrl: '/static/directives/image-feature-view.html',
|
||
|
replace: false,
|
||
|
transclude: true,
|
||
|
restrict: 'C',
|
||
|
scope: {
|
||
|
'repository': '=repository',
|
||
|
'image': '=image',
|
||
|
'isEnabled': '=isEnabled'
|
||
|
},
|
||
|
controller: function($scope, $element, Config, ApiService, VulnerabilityService, AngularViewArray, ImageMetadataService) {
|
||
|
var imageMap = null;
|
||
|
|
||
|
$scope.securityFeatures = [];
|
||
|
$scope.featureBreakdown = [];
|
||
|
|
||
|
$scope.options = {
|
||
|
'featureFilter': null,
|
||
|
'predicate': 'fixableScore',
|
||
|
'reverse': false,
|
||
|
};
|
||
|
|
||
|
$scope.tablePredicateClass = function(name, predicate, reverse) {
|
||
|
if (name != predicate) {
|
||
|
return '';
|
||
|
}
|
||
|
|
||
|
return 'current ' + (reverse ? 'reversed' : '');
|
||
|
};
|
||
|
|
||
|
$scope.orderBy = function(predicate) {
|
||
|
if (predicate == $scope.options.predicate) {
|
||
|
$scope.options.reverse = !$scope.options.reverse;
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
$scope.options.reverse = false;
|
||
|
$scope.options.predicate = predicate;
|
||
|
};
|
||
|
|
||
|
var buildOrderedFeatures = function() {
|
||
|
var features = $scope.securityFeatures.slice(0);
|
||
|
$scope.orderedFeatures = AngularViewArray.create();
|
||
|
features.forEach(function(v) {
|
||
|
var featureFilter = $scope.options.featureFilter;
|
||
|
if (featureFilter) {
|
||
|
if ((v['name'].indexOf(featureFilter) < 0) &&
|
||
|
(v['version'].indexOf(featureFilter) < 0) &&
|
||
|
(v['imageId'].indexOf(featureFilter) < 0)) {
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
$scope.orderedFeatures.push(v);
|
||
|
});
|
||
|
|
||
|
$scope.orderedFeatures.entries.sort(function(a, b) {
|
||
|
var left = a[$scope.options['predicate']];
|
||
|
var right = b[$scope.options['predicate']];
|
||
|
|
||
|
if ($scope.options['predicate'] == 'score' ||
|
||
|
$scope.options['predicate'] == 'fixableScore' ||
|
||
|
$scope.options['predicate'] == 'leftoverScore') {
|
||
|
left = left * 1;
|
||
|
right = right * 1;
|
||
|
}
|
||
|
|
||
|
if (left == null) {
|
||
|
left = '0.00';
|
||
|
}
|
||
|
|
||
|
if (right == null) {
|
||
|
right = '0.00';
|
||
|
}
|
||
|
|
||
|
if (left == right) {
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
return left > right ? -1 : 1;
|
||
|
});
|
||
|
|
||
|
if ($scope.options['reverse']) {
|
||
|
$scope.orderedFeatures.entries.reverse();
|
||
|
}
|
||
|
|
||
|
$scope.orderedFeatures.setVisible(true);
|
||
|
};
|
||
|
|
||
|
var buildChart = function() {
|
||
|
var chartData = $scope.featureBreakdown;
|
||
|
var colors = [];
|
||
|
for (var i = 0; i < chartData.length; ++i) {
|
||
|
colors.push(chartData[i].color);
|
||
|
}
|
||
|
|
||
|
nv.addGraph(function() {
|
||
|
var chart = nv.models.pieChart()
|
||
|
.x(function(d) { return d.label })
|
||
|
.y(function(d) { return d.value })
|
||
|
.margin({left: -10, right: -10, top: -10, bottom: -10})
|
||
|
.showLegend(false)
|
||
|
.showLabels(true)
|
||
|
.labelThreshold(.05)
|
||
|
.labelType("percent")
|
||
|
.donut(true)
|
||
|
.color(colors)
|
||
|
.donutRatio(0.5);
|
||
|
|
||
|
d3.select("#featureDonutChart svg")
|
||
|
.datum(chartData)
|
||
|
.transition()
|
||
|
.duration(350)
|
||
|
.call(chart);
|
||
|
|
||
|
return chart;
|
||
|
});
|
||
|
};
|
||
|
|
||
|
var buildFeatures = function(data) {
|
||
|
$scope.securityFeatures = [];
|
||
|
$scope.featureBreakdown = [];
|
||
|
$scope.highestFixableScore = -10000;
|
||
|
|
||
|
var severityMap = {};
|
||
|
var levels = VulnerabilityService.getLevels();
|
||
|
|
||
|
if (data && data.Layer && data.Layer.Features) {
|
||
|
data.Layer.Features.forEach(function(feature) {
|
||
|
var imageId = feature.AddedBy.split('.')[0];
|
||
|
feature_obj = {
|
||
|
'name': feature.Name,
|
||
|
'namespace': feature.Namespace,
|
||
|
'version': feature.Version,
|
||
|
'addedBy': feature.AddedBy,
|
||
|
'imageId': imageId,
|
||
|
'imageCommand': ImageMetadataService.getImageCommand($scope.image, imageId),
|
||
|
'vulnCount': 0,
|
||
|
'severityBreakdown': [],
|
||
|
'fixableBreakdown': [],
|
||
|
'score': 0,
|
||
|
'fixableCount': 0,
|
||
|
'leftoverCount': 0,
|
||
|
'fixableScore': 0,
|
||
|
'leftoverScore': 0,
|
||
|
'unfixableCount': 0
|
||
|
}
|
||
|
|
||
|
if (feature.Vulnerabilities) {
|
||
|
var highestSeverity = null;
|
||
|
var localSeverityMap = {};
|
||
|
var localLeftoverMap = {};
|
||
|
|
||
|
feature.Vulnerabilities.forEach(function(vuln) {
|
||
|
var severity = VulnerabilityService.LEVELS[vuln['Severity']];
|
||
|
var score = severity.score;
|
||
|
if (vuln.Metadata && vuln.Metadata.NVD && vuln.Metadata.NVD.CVSSv2 && vuln.Metadata.NVD.CVSSv2.Score) {
|
||
|
score = vuln.Metadata.NVD.CVSSv2.Score;
|
||
|
severity = VulnerabilityService.getSeverityForCVSS(score);
|
||
|
}
|
||
|
|
||
|
var logScore = (Math.pow(2, score) + 0.1);
|
||
|
feature_obj['score'] += logScore;
|
||
|
|
||
|
if (vuln.FixedBy) {
|
||
|
feature_obj['fixableScore'] += logScore;
|
||
|
feature_obj['fixableCount']++;
|
||
|
} else {
|
||
|
feature_obj['leftoverCount']++;
|
||
|
feature_obj['leftoverScore'] += logScore;
|
||
|
}
|
||
|
|
||
|
if (highestSeverity == null) {
|
||
|
highestSeverity = severity;
|
||
|
} else {
|
||
|
var index = severity['index'];
|
||
|
if (index < highestSeverity) {
|
||
|
highestSeverity = severity;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!localSeverityMap[severity['index']]) {
|
||
|
localSeverityMap[severity['index']] = 0;
|
||
|
}
|
||
|
|
||
|
if (!localLeftoverMap[severity['index']]) {
|
||
|
localLeftoverMap[severity['index']] = 0;
|
||
|
}
|
||
|
|
||
|
localSeverityMap[severity['index']]++;
|
||
|
if (!vuln.FixedBy) {
|
||
|
localLeftoverMap[severity['index']]++;
|
||
|
}
|
||
|
});
|
||
|
|
||
|
if (!severityMap[highestSeverity['index']]) {
|
||
|
severityMap[highestSeverity['index']] = 0;
|
||
|
}
|
||
|
severityMap[highestSeverity['index']]++;
|
||
|
|
||
|
var localSeverityBreakdown = [];
|
||
|
var localLeftoverBreakdown = [];
|
||
|
|
||
|
for (var i = 0; i < levels.length; ++i) {
|
||
|
var level = levels[i];
|
||
|
if (localSeverityMap[level['index']]) {
|
||
|
localSeverityBreakdown.push({
|
||
|
'title': level['title'],
|
||
|
'color': level['color'],
|
||
|
'count': localSeverityMap[level['index']]
|
||
|
})
|
||
|
}
|
||
|
|
||
|
if (localLeftoverMap[level['index']]) {
|
||
|
localLeftoverBreakdown.push({
|
||
|
'title': level['title'],
|
||
|
'color': level['color'],
|
||
|
'count': localLeftoverMap[level['index']]
|
||
|
})
|
||
|
}
|
||
|
}
|
||
|
|
||
|
feature_obj['vulnCount'] = feature.Vulnerabilities.length;
|
||
|
|
||
|
feature_obj['severityBreakdown'] = localSeverityBreakdown;
|
||
|
feature_obj['leftoverBreakdown'] = localLeftoverBreakdown;
|
||
|
|
||
|
if (localSeverityBreakdown) {
|
||
|
feature_obj['primarySeverity'] = localSeverityBreakdown[0];
|
||
|
}
|
||
|
|
||
|
if (localLeftoverBreakdown) {
|
||
|
feature_obj['primaryLeftover'] = localLeftoverBreakdown[0];
|
||
|
}
|
||
|
|
||
|
if (feature.Vulnerabilities.length > 0) {
|
||
|
feature_obj['score'] = feature_obj['score'] / feature.Vulnerabilities.length;
|
||
|
}
|
||
|
|
||
|
if (feature_obj['fixableScore'] > $scope.highestFixableScore) {
|
||
|
$scope.highestFixableScore = feature_obj['fixableScore'];
|
||
|
}
|
||
|
} else {
|
||
|
feature_obj['fixableScore'] = -1;
|
||
|
}
|
||
|
|
||
|
$scope.securityFeatures.push(feature_obj);
|
||
|
});
|
||
|
}
|
||
|
|
||
|
var greenCount = $scope.securityFeatures.length;
|
||
|
for (var i = 0; i < levels.length; ++i) {
|
||
|
var level = levels[i];
|
||
|
if (!severityMap[level['index']]) {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
greenCount -= severityMap[level['index']];
|
||
|
$scope.featureBreakdown.push({
|
||
|
'label': levels[i].title,
|
||
|
'value': severityMap[level['index']],
|
||
|
'color': levels[i].color,
|
||
|
});
|
||
|
}
|
||
|
|
||
|
$scope.featureBreakdown.push({
|
||
|
'label': 'None',
|
||
|
'value': greenCount,
|
||
|
'color': '#2FC98E'
|
||
|
});
|
||
|
|
||
|
buildOrderedFeatures();
|
||
|
};
|
||
|
|
||
|
var loadImageVulnerabilities = function() {
|
||
|
if ($scope.securityResource) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
var params = {
|
||
|
'repository': $scope.repository.namespace + '/' + $scope.repository.name,
|
||
|
'imageid': $scope.image.id,
|
||
|
'vulnerabilities': true,
|
||
|
};
|
||
|
|
||
|
$scope.securityResource = ApiService.getRepoImageSecurityAsResource(params).get(function(resp) {
|
||
|
$scope.securityStatus = resp.status;
|
||
|
buildFeatures(resp.data);
|
||
|
buildChart();
|
||
|
return resp;
|
||
|
});
|
||
|
};
|
||
|
|
||
|
$scope.$watch('options.predicate', buildOrderedFeatures);
|
||
|
$scope.$watch('options.reverse', buildOrderedFeatures);
|
||
|
$scope.$watch('options.featureFilter', buildOrderedFeatures);
|
||
|
|
||
|
$scope.$watch('isEnabled', function(isEnabled) {
|
||
|
if ($scope.isEnabled && $scope.repository && $scope.image) {
|
||
|
loadImageVulnerabilities();
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
};
|
||
|
return directiveDefinitionObject;
|
||
|
});
|