From 5c4ff27a0bc70cdefad0c9a68e920e480465c4d9 Mon Sep 17 00:00:00 2001
From: Paul Duncan <pabs@pablotron.org>
Date: Mon, 31 Jan 2022 22:16:23 -0500
Subject: internal/cvss: handle invalid keys and metrics

---
 internal/cvss/v2metric.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'internal/cvss/v2metric.go')

diff --git a/internal/cvss/v2metric.go b/internal/cvss/v2metric.go
index a80cf8a..2dee41a 100644
--- a/internal/cvss/v2metric.go
+++ b/internal/cvss/v2metric.go
@@ -236,6 +236,9 @@ func getV2Metric(s string) (v2Metric, error) {
 
 // Get CVSS 2.0 metric key.
 func (m v2Metric) Key() Key {
-  k, _ := v2KeyLut[m]
-  return k
+  if k, ok := v2KeyLut[m]; ok {
+    return k
+  } else {
+    return v2InvalidKey
+  }
 }
-- 
cgit v1.2.3