mirror of
https://github.com/goharbor/harbor
synced 2025-04-16 17:18:58 +00:00
a890b28e1e
feat: Store vulnerability report from scanner into a relational format Convert vulnerability report JSON obtained from scanner into a relational format describe in:https://github.com/goharbor/community/pull/145 Signed-off-by: prahaladdarkin <prahaladd@vmware.com>
56 lines
1.7 KiB
Go
56 lines
1.7 KiB
Go
// Copyright Project Harbor Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package report
|
|
|
|
import (
|
|
"github.com/goharbor/harbor/src/lib/errors"
|
|
v1 "github.com/goharbor/harbor/src/pkg/scan/rest/v1"
|
|
"github.com/goharbor/harbor/src/pkg/scan/vuln"
|
|
)
|
|
|
|
// Merger is a helper function to merge report together
|
|
type Merger func(r1, r2 interface{}) (interface{}, error)
|
|
|
|
// SupportedMergers declares mappings between mime type and report merger func.
|
|
var SupportedMergers = map[string]Merger{
|
|
v1.MimeTypeNativeReport: MergeNativeReport,
|
|
v1.MimeTypeGenericVulnerabilityReport: MergeNativeReport,
|
|
}
|
|
|
|
// Merge merge report r1 and r2
|
|
func Merge(mimeType string, r1, r2 interface{}) (interface{}, error) {
|
|
m, ok := SupportedMergers[mimeType]
|
|
if !ok {
|
|
return nil, errors.Errorf("no report merger bound with mime type %s", mimeType)
|
|
}
|
|
|
|
return m(r1, r2)
|
|
}
|
|
|
|
// MergeNativeReport merge report r1 and r2
|
|
func MergeNativeReport(r1, r2 interface{}) (interface{}, error) {
|
|
nr1, ok := r1.(*vuln.Report)
|
|
if !ok {
|
|
return nil, errors.New("native report required")
|
|
}
|
|
|
|
nr2, ok := r2.(*vuln.Report)
|
|
if !ok {
|
|
return nil, errors.New("native report required")
|
|
}
|
|
|
|
return nr1.Merge(nr2), nil
|
|
}
|