Generate random CVSS 3.1 vector strings

CVSS31Gen.go

package main

import (
	"flag"
	"fmt"
	"math/rand"
	"strings"
	"time"
)

const (
	cvssBase = "CVSS:3.1/"
)

func main() {
	rand.Seed(time.Now().UnixNano())

	numVectors := flag.Int("n", 1, "Number of CVSS 3.1 vectors to generate")
	flag.Parse()

	for i := 0; i < *numVectors; i++ {
		fmt.Println(generateCVSSVector())
	}
}

func generateCVSSVector() string {
	var sb strings.Builder

	sb.WriteString(cvssBase)
	sb.WriteString("AV:")
	sb.WriteString(getRandomElem([]string{"N", "A", "L", "P"}))
	sb.WriteString("/AC:")
	sb.WriteString(getRandomElem([]string{"L", "H"}))
	sb.WriteString("/PR:")
	sb.WriteString(getRandomElem([]string{"N", "L", "H"}))
	sb.WriteString("/UI:")
	sb.WriteString(getRandomElem([]string{"N", "R"}))
	sb.WriteString("/S:")
	sb.WriteString(getRandomElem([]string{"U", "C"}))
	sb.WriteString("/C:")
	sb.WriteString(getRandomElem([]string{"N", "L", "H"}))
	sb.WriteString("/I:")
	sb.WriteString(getRandomElem([]string{"N", "L", "H"}))
	sb.WriteString("/A:")
	sb.WriteString(getRandomElem([]string{"N", "L", "H"}))

	return sb.String()
}

func getRandomElem(arr []string) string {
	return arr[rand.Intn(len(arr))]
}