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/v3metric.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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

diff --git a/internal/cvss/v3metric.go b/internal/cvss/v3metric.go
index a8f25de..e9a6763 100644
--- a/internal/cvss/v3metric.go
+++ b/internal/cvss/v3metric.go
@@ -314,8 +314,11 @@ var v3MetricStrLut = map[string]v3Metric {
 
 // Get CVSS 3.x metric key.
 func (m v3Metric) Key() Key {
-  k, _ := v3KeyLut[m]
-  return k
+  if k, ok := v3KeyLut[m]; ok {
+    return k
+  } else {
+    return v3InvalidKey
+  }
 }
 
 // Convert string to CVSS 3.1 metric.
-- 
cgit v1.2.3