feat(remote): add support for policy.json allow/deny

registry/remote/policy/evaluator.go

@@ -0,0 +1,169 @@
+/*
+Copyright The ORAS Authors.
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package policy
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/oras-project/oras-go/v3/errdef"
+)
+
+// ImageReference represents a reference to an image
+type ImageReference struct {
+	// Transport is the transport type (e.g., "docker")
+	Transport TransportName
+	// Scope is the scope within the transport (e.g., "docker.io/library/nginx")
+	Scope string
+	// Reference is the full reference (e.g., "docker.io/library/nginx:latest")
+	Reference string
+}
+
+// SignedByVerifier verifies GPG/simple signing signatures.
+// Implementations should verify that the image is signed with a valid key
+// as specified in the PRSignedBy requirement.
+type SignedByVerifier interface {
+	Verify(ctx context.Context, req *PRSignedBy, image ImageReference) (bool, error)
+}
+
+// SigstoreVerifier verifies sigstore signatures.
+// Implementations should verify that the image is signed with valid sigstore
+// signatures as specified in the PRSigstoreSigned requirement.
+type SigstoreVerifier interface {
+	Verify(ctx context.Context, req *PRSigstoreSigned, image ImageReference) (bool, error)
+}
+
+// Evaluator evaluates policy requirements against image references
+type Evaluator struct {
+	policy           *Policy
+	signedByVerifier SignedByVerifier
+	sigstoreVerifier SigstoreVerifier
+}
+
+// EvaluatorOption configures an Evaluator
+type EvaluatorOption func(*Evaluator)
+
+// WithSignedByVerifier sets the verifier for PRSignedBy requirements.
+// If not set, evaluating PRSignedBy requirements will return ErrUnsupported.
+func WithSignedByVerifier(v SignedByVerifier) EvaluatorOption {
+	return func(e *Evaluator) {
+		e.signedByVerifier = v
+	}
+}
+
+// WithSigstoreVerifier sets the verifier for PRSigstoreSigned requirements.
+// If not set, evaluating PRSigstoreSigned requirements will return ErrUnsupported.
+func WithSigstoreVerifier(v SigstoreVerifier) EvaluatorOption {
+	return func(e *Evaluator) {
+		e.sigstoreVerifier = v
+	}
+}
+
+// NewEvaluator creates a new policy evaluator
+func NewEvaluator(policy *Policy, opts ...EvaluatorOption) (*Evaluator, error) {
+	if policy == nil {
+		return nil, fmt.Errorf("policy cannot be nil")
+	}
+
+	if err := policy.Validate(); err != nil {
+		return nil, fmt.Errorf("invalid policy: %w", err)
+	}
+
+	e := &Evaluator{
+		policy: policy,
+	}
+
+	for _, opt := range opts {
+		opt(e)
+	}
+
+	return e, nil
+}
+
+// IsImageAllowed determines if an image is allowed by the policy
+func (e *Evaluator) IsImageAllowed(ctx context.Context, image ImageReference) (bool, error) {
+	reqs := e.policy.GetRequirementsForImage(image.Transport, image.Scope)
+
+	if len(reqs) == 0 {
+		// No requirements means reject by default for safety
+		return false, fmt.Errorf("no policy requirements found for %s:%s", image.Transport, image.Scope)
+	}
+
+	// All requirements must be satisfied
+	for _, req := range reqs {
+		allowed, err := e.evaluateRequirement(ctx, req, image)
+		if err != nil {
+			return false, fmt.Errorf("failed to evaluate requirement %s: %w", req.Type(), err)
+		}
+		if !allowed {
+			return false, nil
+		}
+	}
+
+	return true, nil
+}
+
+// evaluateRequirement evaluates a single policy requirement
+func (e *Evaluator) evaluateRequirement(ctx context.Context, req PolicyRequirement, image ImageReference) (bool, error) {
+	switch r := req.(type) {
+	case *InsecureAcceptAnything:
+		return e.evaluateInsecureAcceptAnything(ctx, r, image)
+	case *Reject:
+		return e.evaluateReject(ctx, r, image)
+	case *PRSignedBy:
+		return e.evaluateSignedBy(ctx, r, image)
+	case *PRSigstoreSigned:
+		return e.evaluateSigstoreSigned(ctx, r, image)
+	default:
+		return false, fmt.Errorf("unknown requirement type: %T", req)
+	}
+}
+
+// evaluateInsecureAcceptAnything always accepts the image
+func (e *Evaluator) evaluateInsecureAcceptAnything(ctx context.Context, req *InsecureAcceptAnything, image ImageReference) (bool, error) {
+	return true, nil
+}
+
+// evaluateReject always rejects the image
+func (e *Evaluator) evaluateReject(ctx context.Context, req *Reject, image ImageReference) (bool, error) {
+	return false, nil
+}
+
+// evaluateSignedBy evaluates a signedBy requirement
+func (e *Evaluator) evaluateSignedBy(ctx context.Context, req *PRSignedBy, image ImageReference) (bool, error) {
+	if e.signedByVerifier == nil {
+		return false, fmt.Errorf("signedBy verification requires a SignedByVerifier: %w", errdef.ErrUnsupported)
+	}
+	return e.signedByVerifier.Verify(ctx, req, image)
+}
+
+// evaluateSigstoreSigned evaluates a sigstoreSigned requirement
+func (e *Evaluator) evaluateSigstoreSigned(ctx context.Context, req *PRSigstoreSigned, image ImageReference) (bool, error) {
+	if e.sigstoreVerifier == nil {
+		return false, fmt.Errorf("sigstoreSigned verification requires a SigstoreVerifier: %w", errdef.ErrUnsupported)
+	}
+	return e.sigstoreVerifier.Verify(ctx, req, image)
+}
+
+// ShouldAcceptImage is a convenience function that returns true if the image is allowed
+func ShouldAcceptImage(ctx context.Context, policy *Policy, image ImageReference) (bool, error) {
+	evaluator, err := NewEvaluator(policy)
+	if err != nil {
+		return false, err
+	}
+
+	return evaluator.IsImageAllowed(ctx, image)
+}