From 4e4e98f328309d1c0a470388d198fa37c27e47d5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gr=C3=A9goire=20Unbekandt?= Date: Fri, 22 Dec 2017 10:48:22 +0100 Subject: [PATCH] vulnsrc_rhel: minor changes Code reorganisation --- ext/vulnsrc/rhel/rhel.go | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/ext/vulnsrc/rhel/rhel.go b/ext/vulnsrc/rhel/rhel.go index 54cb97cb..da54df18 100644 --- a/ext/vulnsrc/rhel/rhel.go +++ b/ext/vulnsrc/rhel/rhel.go @@ -201,6 +201,8 @@ func parseRHSA(ovalReader io.Reader) (vulnerabilities []database.VulnerabilityWi // Init vulnerability vulnerability := database.VulnerabilityWithAffected{ Vulnerability: database.Vulnerability{ + Name: rhsaName(definition), + Link: rhsaLink(definition), Severity: severity(definition), Description: description(definition), }, @@ -211,13 +213,12 @@ func parseRHSA(ovalReader io.Reader) (vulnerabilities []database.VulnerabilityWi // Only RHSA is present if len(definition.References) == 1 { - vulnerability.Name = rhsaName(definition) vulnerability.Link = definition.References[0].URI vulnerabilities = append(vulnerabilities, vulnerability) } else { for _, reference := range definition.References[1:] { - vulnerability.Name = name(reference) - vulnerability.Link = link(reference) + vulnerability.Name = reference.ID + vulnerability.Link = reference.URI vulnerabilities = append(vulnerabilities, vulnerability) } } @@ -387,14 +388,13 @@ func severity(def definition) database.Severity { } } -func name(ref reference) string { - return ref.ID -} - -func link(ref reference) string { - return ref.URI -} - func rhsaName(def definition) string { return strings.TrimSpace(def.Title[:strings.Index(def.Title, ": ")]) } + +func rhsaLink(def definition) (link string) { + if len(def.References) > 0 { + link = def.References[0].URI + } + return +}