internal/feed: add v3attackvector tests

1 files changed, 46 insertions, 0 deletions

diff --git a/internal/feed/v3attackvector.go b/internal/feed/v3attackvector.go
new file mode 100644
index 0000000..ecc309a
--- /dev/null
+++ b/internal/feed/v3attackvector.go
@@ -0,0 +1,46 @@
+package feed
+
+//go:generate stringer -linecomment -type=V3AttackVector
+
+import (
+  "encoding/json"
+  "fmt"
+)
+
+// CVSS v3 attack vector.
+type V3AttackVector byte
+
+const (
+  V3AVAdjacentNetwork V3AttackVector = iota // ADJACENT_NETWORK
+  V3AVNetwork // NETWORK
+  V3AVLocal // LOCAL
+  V3AVPhysical // PHYSICAL
+)
+
+// Unmarshal CVSS v3 attack vector from JSON.
+func (me *V3AttackVector) UnmarshalJSON(b []byte) error {
+  // decode string, check for error
+  var s string
+  if err := json.Unmarshal(b, &s); err != nil {
+    return err
+  }
+
+  // check value
+  switch s {
+  case "ADJACENT_NETWORK":
+    *me = V3AVAdjacentNetwork
+  case "LOCAL":
+    *me = V3AVLocal
+  case "NETWORK":
+    *me = V3AVNetwork
+  case "PHYSICAL":
+    *me = V3AVPhysical
+  default:
+    // return error
+    return fmt.Errorf("unknown CVSS v3 attack vector: %s", s)
+  }
+
+  // return success
+  return nil
+}
+