diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-01 12:48:15 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-01 12:48:15 -0500 |
commit | 465f427a04c2f10bc5163f6bceef13693f7e99e6 (patch) | |
tree | ad4c963ee1b918fee01436030ceda2e6c882b225 /internal/feed/v3impact.go | |
parent | f4f3bf7f03e790b5d98004de3498c102bdac50ad (diff) | |
download | cvez-465f427a04c2f10bc5163f6bceef13693f7e99e6.tar.bz2 cvez-465f427a04c2f10bc5163f6bceef13693f7e99e6.zip |
internal/feed: add v2impact, v3impact, and tests
Diffstat (limited to 'internal/feed/v3impact.go')
-rw-r--r-- | internal/feed/v3impact.go | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/internal/feed/v3impact.go b/internal/feed/v3impact.go new file mode 100644 index 0000000..d6c450e --- /dev/null +++ b/internal/feed/v3impact.go @@ -0,0 +1,42 @@ +package feed + +//go:generate stringer -linecomment -type=V3Impact + +import ( + "encoding/json" + "fmt" +) + +// CVSS v3 impact level. +type V3Impact byte + +const ( + V3ImpactNone V3Impact = iota // NONE + V3ImpactLow // LOW + V3ImpactHigh // HIGH +) + +// Unmarshal CVSS v3 impact level from JSON. +func (me *V3Impact) 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 "NONE": + *me = V3ImpactNone + case "LOW": + *me = V3ImpactLow + case "HIGH": + *me = V3ImpactHigh + default: + // return error + return fmt.Errorf("unknown CVSS v3 impact: %s", s) + } + + // return success + return nil +} |