diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-01 08:25:05 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-01 08:25:05 -0500 |
commit | e66d260451c7be2129b3820069042d0ef4c57665 (patch) | |
tree | 94981d739548424d3ee87188fca9f331f0e2a645 /internal/feed/v3attackcomplexity.go | |
parent | a308b1046b3001950945abe35b20b284865958e7 (diff) | |
download | cvez-e66d260451c7be2129b3820069042d0ef4c57665.tar.bz2 cvez-e66d260451c7be2129b3820069042d0ef4c57665.zip |
internal/feed: add v3attackcomplexity tests
Diffstat (limited to 'internal/feed/v3attackcomplexity.go')
-rw-r--r-- | internal/feed/v3attackcomplexity.go | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/internal/feed/v3attackcomplexity.go b/internal/feed/v3attackcomplexity.go new file mode 100644 index 0000000..6e7481c --- /dev/null +++ b/internal/feed/v3attackcomplexity.go @@ -0,0 +1,42 @@ +package feed + +//go:generate stringer -linecomment -type=V3AttackComplexity + +import ( + "encoding/json" + "fmt" +) + +// CVSS v3 attack complexity +type V3AttackComplexity byte + +const ( + V3ACLow V3AttackComplexity = iota // LOW + V3ACMedium // MEDIUM + V3ACHigh // HIGH +) + +// Unmarshal CVSS v3 attack complexity from JSON. +func (me *V3AttackComplexity) 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 "LOW": + *me = V3ACLow + case "MEDIUM": + *me = V3ACMedium + case "HIGH": + *me = V3ACHigh + default: + // return error + return fmt.Errorf("unknown CVSS v3 attack complexity: %s", s) + } + + // return success + return nil +} |