arch op: turn into string, implement pattern match

Signed-off-by: Hank Donnay <[email protected]>

Author: hdonnay

archop.go

@@ -0,0 +1,85 @@
+package claircore
+
+import (
+	"bytes"
+	"database/sql/driver"
+	"fmt"
+	"regexp"
+)
+
+type ArchOp uint
+
+const (
+	opInvalid ArchOp = iota // invalid
+
+	OpEquals       // equals
+	OpNotEquals    // not equals
+	OpPatternMatch // pattern match
+)
+
+func (o ArchOp) Cmp(a, b string) bool {
+	switch {
+	case b == "":
+		return true
+	case a == "":
+		return false
+	default:
+	}
+	switch o {
+	case OpEquals:
+		return a == b
+	case OpNotEquals:
+		return a != b
+	case OpPatternMatch:
+		re, err := regexp.Compile(b)
+		if err != nil {
+			return false
+		}
+		return re.MatchString(a)
+	default:
+	}
+	return false
+}
+
+//go:generate stringer -type=ArchOp -linecomment
+
+func (o ArchOp) MarshalText() (text []byte, err error) {
+	return []byte(o.String()), nil
+}
+
+func (o *ArchOp) UnmarshalText(text []byte) error {
+	i := bytes.Index([]byte(_ArchOp_name), text)
+	if i == -1 {
+		*o = ArchOp(0)
+		return nil
+	}
+	idx := uint8(i)
+	for i, off := range _ArchOp_index {
+		if off == idx {
+			*o = ArchOp(i)
+			return nil
+		}
+	}
+	panic("unreachable")
+}
+
+func (o ArchOp) Value() (driver.Value, error) {
+	return o.String(), nil
+}
+
+func (o *ArchOp) Scan(i interface{}) error {
+	switch v := i.(type) {
+	case []byte:
+		return o.UnmarshalText(v)
+	case string:
+		return o.UnmarshalText([]byte(v))
+	case int64:
+		if v >= int64(len(_ArchOp_index)-1) {
+			return fmt.Errorf("unable to scan ArchOp from enum %d", v)
+		}
+		*o = ArchOp(v)
+	default:
+		return fmt.Errorf("unable to scan ArchOp from type %T", i)
+	}
+	return nil
+}

archop_string.go

@@ -292,7 +292,7 @@ func (e *e2e) DeleteUpdateOperations(ctx context.Context) func(*testing.T) {
 }
 
 // checkInsertedVulns confirms vulnerabilitiles are inserted into the database correctly when
-// store.UpdateVulnerabilities is calld.
+// store.UpdateVulnerabilities is called.
 func checkInsertedVulns(ctx context.Context, t *testing.T, db *sqlx.DB, id uuid.UUID, vulns []*claircore.Vulnerability) {
 	const query = `SELECT
 	vuln.hash_kind,

internal/vulnstore/postgres/update_e2e_test.go

@@ -70,7 +70,7 @@ const (
 		repo_key               TEXT,
 		repo_uri               TEXT,
 		fixed_in_version       TEXT,
-		arch_operation         bigint,
+		arch_operation         TEXT,
 		vulnerable_range       VersionRange NOT NULL DEFAULT VersionRange('{}', '{}', '()'),
 		version_kind           TEXT,
 		UNIQUE (hash_kind, hash)

libvuln/migrations/migration1.go

@@ -119,6 +119,8 @@ func mapArchOp(op oval.Operation) claircore.ArchOp {
 		return claircore.OpEquals
 	case oval.OpNotEquals:
 		return claircore.OpNotEquals
+	case oval.OpPatternMatch:
+		return claircore.OpPatternMatch
 	default:
 	}
 	return claircore.ArchOp(0)

pkg/ovalutil/rpm.go

@@ -1,36 +1,9 @@
 package claircore
 
-import "time"
-
-type ArchOp uint
-
-const (
-	_ ArchOp = iota // invalid
-
-	OpEquals    // equals
-	OpNotEquals // not equals
+import (
+	"time"
 )
 
-func (o ArchOp) Cmp(a, b string) bool {
-	switch {
-	case b == "":
-		return true
-	case a == "":
-		return false
-	default:
-	}
-	switch o {
-	case OpEquals:
-		return a == b
-	case OpNotEquals:
-		return a != b
-	default:
-	}
-	return false
-}
-
-//go:generate stringer -type=ArchOp -linecomment
-
 type Vulnerability struct {
 	// unique ID of this vulnerability. this will be created as discovered by the library
 	// and used for persistence and hash map indexes