plugin: allow selecting wasm target for evaluation

This turns the simple grpc tests scenario into a matrix test for both
evaluation variants.

Should wait for the next opa release so the inter query builtin cache
is wired into the wasm eval.

Signed-off-by: Stephan Renatus <[email protected]>

Author: srenatus

Diff for: .github/workflows/checks.yaml

@@ -92,6 +92,11 @@ jobs:
   e2e-envoy-grpc:
     name: Envoy gRPC End-to-End Test
     runs-on: ubuntu-18.04
+    strategy:
+      matrix:
+        include:
+          - eval: wasm
+          - eval: rego
     steps:
       - name: Check out code
         uses: actions/checkout@v2
@@ -104,7 +109,7 @@ jobs:
         working-directory: examples/grpc
 
       - name: Run test
-        run: make test-setup test
+        run: make test-setup test EVAL_TARGET=${{ matrix.target }}
         working-directory: examples/grpc
 
       - name: Run test log dump and cleanup

Diff for: envoyauth/evaluation.go

@@ -21,6 +21,7 @@ type EvalContext interface {
 	Store() storage.Store
 	Compiler() *ast.Compiler
 	Runtime() *ast.Term
+	Target() string
 	PreparedQueryDoOnce() *sync.Once
 	InterQueryBuiltinCache() iCache.InterQueryCache
 	PreparedQuery() *rego.PreparedEvalQuery
@@ -83,7 +84,8 @@ func constructPreparedQuery(evalContext EvalContext, txn storage.Transaction, m
 			rego.Compiler(evalContext.Compiler()),
 			rego.Store(evalContext.Store()),
 			rego.Transaction(txn),
-			rego.Runtime(evalContext.Runtime()))
+			rego.Runtime(evalContext.Runtime()),
+			rego.Target(evalContext.Target()))
 
 		r := rego.New(opts...)
 

Diff for: examples/grpc/Makefile

@@ -2,6 +2,7 @@ SHELL:=bash
 GRPCURL_IMAGE:=fullstorydev/grpcurl:v1.7.0
 GRPCURL=docker run --network=host -i --rm -v $$(pwd)/testsrv.pb:/testsrv.pb $(GRPCURL_IMAGE) \
   -d @ -plaintext -protoset /testsrv.pb 127.0.0.1:51051
+EVAL_TARGET?=rego
 
 all: testsrv.pb testsrv-image test-setup test test-teardown
 
@@ -14,7 +15,7 @@ testsrv.pb: testsrv/test.proto
 
 .PHONY: test-setup
 test-setup:
-	docker-compose up -d
+	EVAL_TARGET=$(EVAL_TARGET) docker-compose up -d
 
 .PHONY: test-teardown
 test-teardown:

Diff for: examples/grpc/docker-compose.yaml

@@ -14,12 +14,13 @@ services:
     command:
       - run
       - --server
-      - --config-file=/opa.yaml
+      - --config-file=/opa-${EVAL_TARGET}.yaml
       - /policy.rego
     volumes:
       - ./testsrv.pb:/testsrv.pb
       - ./policy.rego:/policy.rego
-      - ./opa.yaml:/opa.yaml
+      - ./opa-wasm.yaml:/opa-wasm.yaml
+      - ./opa-rego.yaml:/opa-rego.yaml
   testsrv:
     image: testsrv:latest
     ports:

Diff for: examples/grpc/opa.yaml renamed to examples/grpc/opa-rego.yaml

@@ -0,0 +1,10 @@
+plugins:
+  envoy_ext_authz_grpc:
+    addr: ":9191"
+    path: envoy/authz/allow
+    dry-run: false
+    enable-reflection: true
+    proto-descriptor: /testsrv.pb
+    eval-target: wasm
+decision_logs:
+  console: true

Diff for: examples/grpc/opa-wasm.yaml

@@ -41,10 +41,13 @@ import (
 	"github.com/open-policy-agent/opa/util"
 )
 
-const defaultAddr = ":9191"
-const defaultPath = "envoy/authz/allow"
-const defaultDryRun = false
-const defaultEnableReflection = false
+const (
+	defaultAddr             = ":9191"
+	defaultPath             = "envoy/authz/allow"
+	defaultDryRun           = false
+	defaultEnableReflection = false
+	defaultEvalTarget       = "rego"
+)
 
 // PluginName is the name to register with the OPA plugin manager
 const PluginName = "envoy_ext_authz_grpc"
@@ -58,6 +61,7 @@ func Validate(m *plugins.Manager, bs []byte) (*Config, error) {
 		Addr:             defaultAddr,
 		DryRun:           defaultDryRun,
 		EnableReflection: defaultEnableReflection,
+		EvalTarget:       defaultEvalTarget,
 	}
 
 	if err := util.Unmarshal(bs, &cfg); err != nil {
@@ -133,6 +137,7 @@ type Config struct {
 	Path             string `json:"path"`
 	DryRun           bool   `json:"dry-run"`
 	EnableReflection bool   `json:"enable-reflection"`
+	EvalTarget       string `json:"eval-target"`
 	parsedQuery      ast.Body
 	ProtoDescriptor  string `json:"proto-descriptor"`
 	protoSet         *protoregistry.Files
@@ -167,6 +172,10 @@ func (p *envoyExtAuthzGrpcServer) Runtime() *ast.Term {
 	return p.manager.Info
 }
 
+func (p *envoyExtAuthzGrpcServer) Target() string {
+	return p.cfg.EvalTarget
+}
+
 func (p *envoyExtAuthzGrpcServer) PreparedQueryDoOnce() *sync.Once {
 	return p.preparedQueryDoOnce
 }
@@ -243,6 +252,7 @@ func (p *envoyExtAuthzGrpcServer) listen() {
 		"path":              p.cfg.Path,
 		"dry-run":           p.cfg.DryRun,
 		"enable-reflection": p.cfg.EnableReflection,
+		"eval-target":       p.cfg.EvalTarget,
 	}).Info("Starting gRPC server.")
 
 	p.manager.UpdatePluginStatus(PluginName, &plugins.Status{State: plugins.StateOK})