diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-01 13:57:27 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-01 13:57:27 -0500 |
commit | 5352b6155f7baa772ee676b505cc429976f7d854 (patch) | |
tree | 2e34c95bbf8bbe5856de078dc80082cf612ef496 /internal/feed/v2accessvector.go | |
parent | 806a74bdff6cf2f014bb93d01d84611c9ef26e04 (diff) | |
download | cvez-5352b6155f7baa772ee676b505cc429976f7d854.tar.bz2 cvez-5352b6155f7baa772ee676b505cc429976f7d854.zip |
internal/feed: add v2accessvector, v2accessvector tests
Diffstat (limited to 'internal/feed/v2accessvector.go')
-rw-r--r-- | internal/feed/v2accessvector.go | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/internal/feed/v2accessvector.go b/internal/feed/v2accessvector.go new file mode 100644 index 0000000..80490c2 --- /dev/null +++ b/internal/feed/v2accessvector.go @@ -0,0 +1,42 @@ +package feed + +//go:generate stringer -linecomment -type=V2AccessVector + +import ( + "encoding/json" + "fmt" +) + +type V2AccessVector byte + +const ( + V2AVAdjacentNetwork V2AccessVector = iota // ADJACENT_NETWORK + V2AVLocal // LOCAL + V2AVNetwork // NETWORK +) + +// Unmarshal CVSS V2 access vector from JSON. +func (me *V2AccessVector) 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 = V2AVAdjacentNetwork + case "LOCAL": + *me = V2AVLocal + case "NETWORK": + *me = V2AVNetwork + default: + // return error + return fmt.Errorf("unknown CVSS v2 access vector: %s", s) + } + + // return success + return nil +} + |