diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-01 14:24:40 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-01 14:24:40 -0500 |
commit | 6f889af4b0d99ee714bfe33e7f864bc35f567097 (patch) | |
tree | d540e1dd1e80f4fa7d6c2cc886cf8f8ad8afdb88 /internal/feed/v2accesscomplexity.go | |
parent | 5352b6155f7baa772ee676b505cc429976f7d854 (diff) | |
download | cvez-6f889af4b0d99ee714bfe33e7f864bc35f567097.tar.bz2 cvez-6f889af4b0d99ee714bfe33e7f864bc35f567097.zip |
internal/feed: add v2accesscomplexity, tests
Diffstat (limited to 'internal/feed/v2accesscomplexity.go')
-rw-r--r-- | internal/feed/v2accesscomplexity.go | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/internal/feed/v2accesscomplexity.go b/internal/feed/v2accesscomplexity.go new file mode 100644 index 0000000..5885e0d --- /dev/null +++ b/internal/feed/v2accesscomplexity.go @@ -0,0 +1,42 @@ +package feed + +//go:generate stringer -linecomment -type=V2AccessComplexity + +import ( + "encoding/json" + "fmt" +) + +// CVSS v2 access complexity +type V2AccessComplexity byte + +const ( + V2ACLow V2AccessComplexity = iota // LOW + V2ACMedium // MEDIUM + V2ACHigh // HIGH +) + +// Unmarshal CVSS V2 access complexity from JSON. +func (me *V2AccessComplexity) 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 = V2ACLow + case "MEDIUM": + *me = V2ACMedium + case "HIGH": + *me = V2ACHigh + default: + // return error + return fmt.Errorf("unknown CVSS v2 access complexity: %s", s) + } + + // return success + return nil +} |