Files
releases/plugins/logs/plugin.go
T
Lander Visterin d0a7d11246 plugins: accept any HTTP 2xx for status and logs (#4897)
Implementations of the Status and Decision Log API may return
response codes that indicate success but are not 200,
for example 204 No Content.

OPA should treat those statuses as successful instead of errors.

Signed-off-by: Lander Visterin <lander.visterin@klarrio.com>
2022-07-20 15:13:40 +02:00

933 lines
25 KiB
Go

// Copyright 2018 The OPA Authors. All rights reserved.
// Use of this source code is governed by an Apache2
// license that can be found in the LICENSE file.
// Package logs implements decision log buffering and uploading.
package logs
import (
"context"
"encoding/json"
"fmt"
"math"
"math/rand"
"net/http"
"net/url"
"reflect"
"strings"
"sync"
"time"
"golang.org/x/time/rate"
"github.com/open-policy-agent/opa/ast"
"github.com/open-policy-agent/opa/internal/ref"
"github.com/open-policy-agent/opa/logging"
"github.com/open-policy-agent/opa/metrics"
"github.com/open-policy-agent/opa/plugins"
"github.com/open-policy-agent/opa/plugins/rest"
"github.com/open-policy-agent/opa/rego"
"github.com/open-policy-agent/opa/server"
"github.com/open-policy-agent/opa/storage"
"github.com/open-policy-agent/opa/util"
)
// Logger defines the interface for decision logging plugins.
type Logger interface {
plugins.Plugin
Log(context.Context, EventV1) error
}
// EventV1 represents a decision log event.
// WARNING: The AST() function for EventV1 must be kept in sync with
// the struct. Any changes here MUST be reflected in the AST()
// implementation below.
type EventV1 struct {
Labels map[string]string `json:"labels"`
DecisionID string `json:"decision_id"`
Revision string `json:"revision,omitempty"` // Deprecated: Use Bundles instead
Bundles map[string]BundleInfoV1 `json:"bundles,omitempty"`
Path string `json:"path,omitempty"`
Query string `json:"query,omitempty"`
Input *interface{} `json:"input,omitempty"`
Result *interface{} `json:"result,omitempty"`
MappedResult *interface{} `json:"mapped_result,omitempty"`
Erased []string `json:"erased,omitempty"`
Masked []string `json:"masked,omitempty"`
Error error `json:"error,omitempty"`
RequestedBy string `json:"requested_by,omitempty"`
Timestamp time.Time `json:"timestamp"`
Metrics map[string]interface{} `json:"metrics,omitempty"`
inputAST ast.Value
}
// BundleInfoV1 describes a bundle associated with a decision log event.
type BundleInfoV1 struct {
Revision string `json:"revision,omitempty"`
}
// AST returns the BundleInfoV1 as an AST value
func (b *BundleInfoV1) AST() ast.Value {
result := ast.NewObject()
if len(b.Revision) > 0 {
result.Insert(ast.StringTerm("revision"), ast.StringTerm(b.Revision))
}
return result
}
// Key ast.Term values for the Rego AST representation of the EventV1
var labelsKey = ast.StringTerm("labels")
var decisionIDKey = ast.StringTerm("decision_id")
var revisionKey = ast.StringTerm("revision")
var bundlesKey = ast.StringTerm("bundles")
var pathKey = ast.StringTerm("path")
var queryKey = ast.StringTerm("query")
var inputKey = ast.StringTerm("input")
var resultKey = ast.StringTerm("result")
var mappedResultKey = ast.StringTerm("mapped_result")
var erasedKey = ast.StringTerm("erased")
var maskedKey = ast.StringTerm("masked")
var errorKey = ast.StringTerm("error")
var requestedByKey = ast.StringTerm("requested_by")
var timestampKey = ast.StringTerm("timestamp")
var metricsKey = ast.StringTerm("metrics")
// AST returns the Rego AST representation for a given EventV1 object.
// This avoids having to round trip through JSON while applying a decision log
// mask policy to the event.
func (e *EventV1) AST() (ast.Value, error) {
var err error
event := ast.NewObject()
if e.Labels != nil {
labelsObj := ast.NewObject()
for k, v := range e.Labels {
labelsObj.Insert(ast.StringTerm(k), ast.StringTerm(v))
}
event.Insert(labelsKey, ast.NewTerm(labelsObj))
} else {
event.Insert(labelsKey, ast.NullTerm())
}
event.Insert(decisionIDKey, ast.StringTerm(e.DecisionID))
if len(e.Revision) > 0 {
event.Insert(revisionKey, ast.StringTerm(e.Revision))
}
if len(e.Bundles) > 0 {
bundlesObj := ast.NewObject()
for k, v := range e.Bundles {
bundlesObj.Insert(ast.StringTerm(k), ast.NewTerm(v.AST()))
}
event.Insert(bundlesKey, ast.NewTerm(bundlesObj))
}
if len(e.Path) > 0 {
event.Insert(pathKey, ast.StringTerm(e.Path))
}
if len(e.Query) > 0 {
event.Insert(queryKey, ast.StringTerm(e.Query))
}
if e.Input != nil {
if e.inputAST == nil {
e.inputAST, err = roundtripJSONToAST(e.Input)
if err != nil {
return nil, err
}
}
event.Insert(inputKey, ast.NewTerm(e.inputAST))
}
if e.Result != nil {
results, err := roundtripJSONToAST(e.Result)
if err != nil {
return nil, err
}
event.Insert(resultKey, ast.NewTerm(results))
}
if e.MappedResult != nil {
mResults, err := roundtripJSONToAST(e.MappedResult)
if err != nil {
return nil, err
}
event.Insert(mappedResultKey, ast.NewTerm(mResults))
}
if len(e.Erased) > 0 {
erased := make([]*ast.Term, len(e.Erased))
for i, v := range e.Erased {
erased[i] = ast.StringTerm(v)
}
event.Insert(erasedKey, ast.NewTerm(ast.NewArray(erased...)))
}
if len(e.Masked) > 0 {
masked := make([]*ast.Term, len(e.Masked))
for i, v := range e.Masked {
masked[i] = ast.StringTerm(v)
}
event.Insert(maskedKey, ast.NewTerm(ast.NewArray(masked...)))
}
if e.Error != nil {
evalErr, err := roundtripJSONToAST(e.Error)
if err != nil {
return nil, err
}
event.Insert(errorKey, ast.NewTerm(evalErr))
}
if len(e.RequestedBy) > 0 {
event.Insert(requestedByKey, ast.StringTerm(e.RequestedBy))
}
// Use the timestamp JSON marshaller to ensure the format is the same as
// round tripping through JSON.
timeBytes, err := e.Timestamp.MarshalJSON()
if err != nil {
return nil, err
}
event.Insert(timestampKey, ast.StringTerm(strings.Trim(string(timeBytes), "\"")))
if e.Metrics != nil {
m, err := ast.InterfaceToValue(e.Metrics)
if err != nil {
return nil, err
}
event.Insert(metricsKey, ast.NewTerm(m))
}
return event, nil
}
func roundtripJSONToAST(x interface{}) (ast.Value, error) {
rawPtr := util.Reference(x)
// roundtrip through json: this turns slices (e.g. []string, []bool) into
// []interface{}, the only array type ast.InterfaceToValue can work with
if err := util.RoundTrip(rawPtr); err != nil {
return nil, err
}
return ast.InterfaceToValue(*rawPtr)
}
const (
// min amount of time to wait following a failure
minRetryDelay = time.Millisecond * 100
defaultMinDelaySeconds = int64(300)
defaultMaxDelaySeconds = int64(600)
defaultUploadSizeLimitBytes = int64(32768) // 32KB limit
defaultBufferSizeLimitBytes = int64(0) // unlimited
defaultMaskDecisionPath = "/system/log/mask"
logDropCounterName = "decision_logs_dropped"
defaultResourcePath = "/logs"
)
// ReportingConfig represents configuration for the plugin's reporting behaviour.
type ReportingConfig struct {
BufferSizeLimitBytes *int64 `json:"buffer_size_limit_bytes,omitempty"` // max size of in-memory buffer
UploadSizeLimitBytes *int64 `json:"upload_size_limit_bytes,omitempty"` // max size of upload payload
MinDelaySeconds *int64 `json:"min_delay_seconds,omitempty"` // min amount of time to wait between successful poll attempts
MaxDelaySeconds *int64 `json:"max_delay_seconds,omitempty"` // max amount of time to wait between poll attempts
MaxDecisionsPerSecond *float64 `json:"max_decisions_per_second,omitempty"` // max number of decision logs to buffer per second
Trigger *plugins.TriggerMode `json:"trigger,omitempty"` // trigger mode
}
// Config represents the plugin configuration.
type Config struct {
Plugin *string `json:"plugin"`
Service string `json:"service"`
PartitionName string `json:"partition_name,omitempty"`
Reporting ReportingConfig `json:"reporting"`
MaskDecision *string `json:"mask_decision"`
ConsoleLogs bool `json:"console"`
Resource *string `json:"resource"`
maskDecisionRef ast.Ref
}
func (c *Config) validateAndInjectDefaults(services []string, pluginsList []string, trigger *plugins.TriggerMode) error {
if c.Plugin != nil {
var found bool
for _, other := range pluginsList {
if other == *c.Plugin {
found = true
break
}
}
if !found {
return fmt.Errorf("invalid plugin name %q in decision_logs", *c.Plugin)
}
} else if c.Service == "" && len(services) != 0 && !c.ConsoleLogs {
// For backwards compatibility allow defaulting to the first
// service listed, but only if console logging is disabled. If enabled
// we can't tell if the deployer wanted to use only console logs or
// both console logs and the default service option.
c.Service = services[0]
} else if c.Service != "" {
found := false
for _, svc := range services {
if svc == c.Service {
found = true
break
}
}
if !found {
return fmt.Errorf("invalid service name %q in decision_logs", c.Service)
}
}
t, err := plugins.ValidateAndInjectDefaultsForTriggerMode(trigger, c.Reporting.Trigger)
if err != nil {
return fmt.Errorf("invalid decision_log config: %w", err)
}
c.Reporting.Trigger = t
min := defaultMinDelaySeconds
max := defaultMaxDelaySeconds
// reject bad min/max values
if c.Reporting.MaxDelaySeconds != nil && c.Reporting.MinDelaySeconds != nil {
if *c.Reporting.MaxDelaySeconds < *c.Reporting.MinDelaySeconds {
return fmt.Errorf("max reporting delay must be >= min reporting delay in decision_logs")
}
min = *c.Reporting.MinDelaySeconds
max = *c.Reporting.MaxDelaySeconds
} else if c.Reporting.MaxDelaySeconds == nil && c.Reporting.MinDelaySeconds != nil {
return fmt.Errorf("reporting configuration missing 'max_delay_seconds' in decision_logs")
} else if c.Reporting.MinDelaySeconds == nil && c.Reporting.MaxDelaySeconds != nil {
return fmt.Errorf("reporting configuration missing 'min_delay_seconds' in decision_logs")
}
// scale to seconds
minSeconds := int64(time.Duration(min) * time.Second)
c.Reporting.MinDelaySeconds = &minSeconds
maxSeconds := int64(time.Duration(max) * time.Second)
c.Reporting.MaxDelaySeconds = &maxSeconds
// default the upload size limit
uploadLimit := defaultUploadSizeLimitBytes
if c.Reporting.UploadSizeLimitBytes != nil {
uploadLimit = *c.Reporting.UploadSizeLimitBytes
}
c.Reporting.UploadSizeLimitBytes = &uploadLimit
if c.Reporting.BufferSizeLimitBytes != nil && c.Reporting.MaxDecisionsPerSecond != nil {
return fmt.Errorf("invalid decision_log config, specify either 'buffer_size_limit_bytes' or 'max_decisions_per_second'")
}
// default the buffer size limit
bufferLimit := defaultBufferSizeLimitBytes
if c.Reporting.BufferSizeLimitBytes != nil {
bufferLimit = *c.Reporting.BufferSizeLimitBytes
}
c.Reporting.BufferSizeLimitBytes = &bufferLimit
if c.MaskDecision == nil {
maskDecision := defaultMaskDecisionPath
c.MaskDecision = &maskDecision
}
c.maskDecisionRef, err = ref.ParseDataPath(*c.MaskDecision)
if err != nil {
return fmt.Errorf("invalid mask_decision in decision_logs: %w", err)
}
if c.PartitionName != "" {
resourcePath := fmt.Sprintf("/logs/%v", c.PartitionName)
c.Resource = &resourcePath
} else if c.Resource == nil {
resourcePath := defaultResourcePath
c.Resource = &resourcePath
} else {
if _, err := url.Parse(*c.Resource); err != nil {
return fmt.Errorf("invalid resource path %q: %w", *c.Resource, err)
}
}
return nil
}
// Plugin implements decision log buffering and uploading.
type Plugin struct {
manager *plugins.Manager
config Config
buffer *logBuffer
enc *chunkEncoder
mtx sync.Mutex
stop chan chan struct{}
reconfig chan reconfigure
mask *rego.PreparedEvalQuery
maskMutex sync.Mutex
limiter *rate.Limiter
metrics metrics.Metrics
logger logging.Logger
}
type reconfigure struct {
config interface{}
done chan struct{}
}
// ParseConfig validates the config and injects default values.
func ParseConfig(config []byte, services []string, pluginList []string) (*Config, error) {
t := plugins.DefaultTriggerMode
return NewConfigBuilder().
WithBytes(config).
WithServices(services).
WithPlugins(pluginList).
WithTriggerMode(&t).
Parse()
}
// ConfigBuilder assists in the construction of the plugin configuration.
type ConfigBuilder struct {
raw []byte
services []string
plugins []string
trigger *plugins.TriggerMode
}
// NewConfigBuilder returns a new ConfigBuilder to build and parse the plugin config.
func NewConfigBuilder() *ConfigBuilder {
return &ConfigBuilder{}
}
// WithBytes sets the raw plugin config.
func (b *ConfigBuilder) WithBytes(config []byte) *ConfigBuilder {
b.raw = config
return b
}
// WithServices sets the services that implement control plane APIs.
func (b *ConfigBuilder) WithServices(services []string) *ConfigBuilder {
b.services = services
return b
}
// WithPlugins sets the list of named plugins for decision logging.
func (b *ConfigBuilder) WithPlugins(plugins []string) *ConfigBuilder {
b.plugins = plugins
return b
}
// WithTriggerMode sets the plugin trigger mode.
func (b *ConfigBuilder) WithTriggerMode(trigger *plugins.TriggerMode) *ConfigBuilder {
b.trigger = trigger
return b
}
// Parse validates the config and injects default values.
func (b *ConfigBuilder) Parse() (*Config, error) {
if b.raw == nil {
return nil, nil
}
var parsedConfig Config
if err := util.Unmarshal(b.raw, &parsedConfig); err != nil {
return nil, err
}
if parsedConfig.Plugin == nil && parsedConfig.Service == "" && len(b.services) == 0 && !parsedConfig.ConsoleLogs {
// Nothing to validate or inject
return nil, nil
}
if err := parsedConfig.validateAndInjectDefaults(b.services, b.plugins, b.trigger); err != nil {
return nil, err
}
return &parsedConfig, nil
}
// New returns a new Plugin with the given config.
func New(parsedConfig *Config, manager *plugins.Manager) *Plugin {
plugin := &Plugin{
manager: manager,
config: *parsedConfig,
stop: make(chan chan struct{}),
buffer: newLogBuffer(*parsedConfig.Reporting.BufferSizeLimitBytes),
enc: newChunkEncoder(*parsedConfig.Reporting.UploadSizeLimitBytes),
reconfig: make(chan reconfigure),
logger: manager.Logger().WithFields(map[string]interface{}{"plugin": Name}),
}
if parsedConfig.Reporting.MaxDecisionsPerSecond != nil {
limit := *parsedConfig.Reporting.MaxDecisionsPerSecond
plugin.limiter = rate.NewLimiter(rate.Limit(limit), int(math.Max(1, limit)))
}
manager.RegisterCompilerTrigger(plugin.compilerUpdated)
manager.UpdatePluginStatus(Name, &plugins.Status{State: plugins.StateNotReady})
return plugin
}
// WithMetrics sets the global metrics provider to be used by the plugin.
func (p *Plugin) WithMetrics(m metrics.Metrics) *Plugin {
p.metrics = m
p.enc.WithMetrics(m)
return p
}
// Name identifies the plugin on manager.
const Name = "decision_logs"
// Lookup returns the decision logs plugin registered with the manager.
func Lookup(manager *plugins.Manager) *Plugin {
if p := manager.Plugin(Name); p != nil {
return p.(*Plugin)
}
return nil
}
// Start starts the plugin.
func (p *Plugin) Start(ctx context.Context) error {
p.logger.Info("Starting decision logger.")
go p.loop()
p.manager.UpdatePluginStatus(Name, &plugins.Status{State: plugins.StateOK})
return nil
}
// Stop stops the plugin.
func (p *Plugin) Stop(ctx context.Context) {
p.logger.Info("Stopping decision logger.")
if *p.config.Reporting.Trigger == plugins.TriggerPeriodic {
if _, ok := ctx.Deadline(); ok && p.config.Service != "" {
p.flushDecisions(ctx)
}
}
done := make(chan struct{})
p.stop <- done
<-done
p.manager.UpdatePluginStatus(Name, &plugins.Status{State: plugins.StateNotReady})
}
func (p *Plugin) flushDecisions(ctx context.Context) {
p.logger.Info("Flushing decision logs.")
done := make(chan bool)
go func(ctx context.Context, done chan bool) {
for ctx.Err() == nil {
if _, err := p.oneShot(ctx); err != nil {
p.logger.Error("Error flushing decisions: %s", err)
// Wait some before retrying, but skip incrementing interval since we are shutting down
time.Sleep(1 * time.Second)
} else {
done <- true
return
}
}
}(ctx, done)
select {
case <-done:
p.logger.Info("All decisions in buffer uploaded.")
case <-ctx.Done():
switch ctx.Err() {
case context.DeadlineExceeded, context.Canceled:
p.logger.Error("Plugin stopped with decisions possibly still in buffer.")
}
}
}
// Log appends a decision log event to the buffer for uploading.
func (p *Plugin) Log(ctx context.Context, decision *server.Info) error {
bundles := map[string]BundleInfoV1{}
for name, info := range decision.Bundles {
bundles[name] = BundleInfoV1{Revision: info.Revision}
}
event := EventV1{
Labels: p.manager.Labels(),
DecisionID: decision.DecisionID,
Revision: decision.Revision,
Bundles: bundles,
Path: decision.Path,
Query: decision.Query,
Input: decision.Input,
Result: decision.Results,
MappedResult: decision.MappedResults,
RequestedBy: decision.RemoteAddr,
Timestamp: decision.Timestamp,
inputAST: decision.InputAST,
}
if decision.Metrics != nil {
event.Metrics = decision.Metrics.All()
}
if decision.Error != nil {
event.Error = decision.Error
}
err := p.maskEvent(ctx, decision.Txn, &event)
if err != nil {
// TODO(tsandall): see note below about error handling.
p.logger.Error("Log event masking failed: %v.", err)
return nil
}
if p.config.ConsoleLogs {
err := p.logEvent(event)
if err != nil {
p.logger.Error("Failed to log to console: %v.", err)
}
}
if p.config.Service != "" {
p.mtx.Lock()
p.encodeAndBufferEvent(event)
p.mtx.Unlock()
}
if p.config.Plugin != nil {
proxy, ok := p.manager.Plugin(*p.config.Plugin).(Logger)
if !ok {
return fmt.Errorf("plugin does not implement Logger interface")
}
return proxy.Log(ctx, event)
}
return nil
}
// Reconfigure notifies the plugin with a new configuration.
func (p *Plugin) Reconfigure(_ context.Context, config interface{}) {
done := make(chan struct{})
p.reconfig <- reconfigure{config: config, done: done}
p.maskMutex.Lock()
defer p.maskMutex.Unlock()
p.mask = nil
<-done
}
// Trigger can be used to control when the plugin attempts to upload
// a new decision log in manual triggering mode.
func (p *Plugin) Trigger(ctx context.Context) error {
done := make(chan error)
go func() {
if p.config.Service != "" {
err := p.doOneShot(ctx)
if err != nil {
if ctx.Err() == nil {
done <- err
}
}
}
close(done)
}()
select {
case err := <-done:
return err
case <-ctx.Done():
return ctx.Err()
}
}
// compilerUpdated is called when a compiler trigger on the plugin manager
// fires. This indicates a new compiler instance is available. The decision
// logger needs to prepare a new masking query.
func (p *Plugin) compilerUpdated(storage.Transaction) {
p.maskMutex.Lock()
defer p.maskMutex.Unlock()
p.mask = nil
}
func (p *Plugin) loop() {
ctx, cancel := context.WithCancel(context.Background())
var retry int
for {
var waitC chan struct{}
if *p.config.Reporting.Trigger == plugins.TriggerPeriodic && p.config.Service != "" {
err := p.doOneShot(ctx)
var delay time.Duration
if err == nil {
min := float64(*p.config.Reporting.MinDelaySeconds)
max := float64(*p.config.Reporting.MaxDelaySeconds)
delay = time.Duration(((max - min) * rand.Float64()) + min)
} else {
delay = util.DefaultBackoff(float64(minRetryDelay), float64(*p.config.Reporting.MaxDelaySeconds), retry)
}
p.logger.Debug("Waiting %v before next upload/retry.", delay)
waitC = make(chan struct{})
go func() {
select {
case <-time.After(delay):
if err != nil {
retry++
} else {
retry = 0
}
close(waitC)
case <-ctx.Done():
}
}()
}
select {
case <-waitC:
case update := <-p.reconfig:
p.reconfigure(update.config)
update.done <- struct{}{}
case done := <-p.stop:
cancel()
done <- struct{}{}
return
}
}
}
func (p *Plugin) doOneShot(ctx context.Context) error {
uploaded, err := p.oneShot(ctx)
if err != nil {
p.logger.Error("%v.", err)
} else if uploaded {
p.logger.Info("Logs uploaded successfully.")
} else {
p.logger.Debug("Log upload queue was empty.")
}
return err
}
func (p *Plugin) oneShot(ctx context.Context) (ok bool, err error) {
// Make a local copy of the plugins's encoder and buffer and create
// a new encoder and buffer. This is needed as locking the buffer for
// the upload duration will block policy evaluation and result in
// increased latency for OPA clients
p.mtx.Lock()
oldChunkEnc := p.enc
oldBuffer := p.buffer
p.buffer = newLogBuffer(*p.config.Reporting.BufferSizeLimitBytes)
p.enc = newChunkEncoder(*p.config.Reporting.UploadSizeLimitBytes).WithMetrics(p.metrics)
p.mtx.Unlock()
// Along with uploading the compressed events in the buffer
// to the remote server, flush any pending compressed data to the
// underlying writer and add to the buffer.
chunk, err := oldChunkEnc.Flush()
if err != nil {
return false, err
}
for _, ch := range chunk {
p.bufferChunk(oldBuffer, ch)
}
if oldBuffer.Len() == 0 {
return false, nil
}
for bs := oldBuffer.Pop(); bs != nil; bs = oldBuffer.Pop() {
if err == nil {
err = uploadChunk(ctx, p.manager.Client(p.config.Service), *p.config.Resource, bs)
}
if err != nil {
if p.limiter != nil {
events, decErr := newChunkDecoder(bs).decode()
if decErr != nil {
continue
}
p.mtx.Lock()
for _, event := range events {
p.encodeAndBufferEvent(event)
}
p.mtx.Unlock()
} else {
// requeue the chunk
p.mtx.Lock()
p.bufferChunk(p.buffer, bs)
p.mtx.Unlock()
}
}
}
return err == nil, err
}
func (p *Plugin) reconfigure(config interface{}) {
newConfig := config.(*Config)
if reflect.DeepEqual(p.config, *newConfig) {
p.logger.Debug("Decision log uploader configuration unchanged.")
return
}
p.logger.Info("Decision log uploader configuration changed.")
p.config = *newConfig
}
func (p *Plugin) encodeAndBufferEvent(event EventV1) {
if p.limiter != nil {
if !p.limiter.Allow() {
if p.metrics != nil {
p.metrics.Counter(logDropCounterName).Incr()
}
p.logger.Error("Decision log dropped as rate limit exceeded. Reduce reporting interval or increase rate limit.")
return
}
}
result, err := p.enc.Write(event)
if err != nil {
// TODO(tsandall): revisit this now that we have an API that
// can return an error. Should the default behaviour be to
// fail-closed as we do for plugins?
p.logger.Error("Log encoding failed: %v.", err)
return
}
for _, chunk := range result {
p.bufferChunk(p.buffer, chunk)
}
}
func (p *Plugin) bufferChunk(buffer *logBuffer, bs []byte) {
dropped := buffer.Push(bs)
if dropped > 0 {
p.logger.Error("Dropped %v chunks from buffer. Reduce reporting interval or increase buffer size.", dropped)
}
}
func (p *Plugin) maskEvent(ctx context.Context, txn storage.Transaction, event *EventV1) error {
mask, err := func() (rego.PreparedEvalQuery, error) {
p.maskMutex.Lock()
defer p.maskMutex.Unlock()
if p.mask == nil {
query := ast.NewBody(ast.NewExpr(ast.NewTerm(p.config.maskDecisionRef)))
r := rego.New(
rego.ParsedQuery(query),
rego.Compiler(p.manager.GetCompiler()),
rego.Store(p.manager.Store),
rego.Transaction(txn),
rego.Runtime(p.manager.Info),
rego.EnablePrintStatements(p.manager.EnablePrintStatements()),
rego.PrintHook(p.manager.PrintHook()),
)
pq, err := r.PrepareForEval(context.Background())
if err != nil {
return rego.PreparedEvalQuery{}, err
}
p.mask = &pq
}
return *p.mask, nil
}()
if err != nil {
return err
}
input, err := event.AST()
if err != nil {
return err
}
rs, err := mask.Eval(
ctx,
rego.EvalParsedInput(input),
rego.EvalTransaction(txn),
)
if err != nil {
return err
} else if len(rs) == 0 {
return nil
}
mRuleSet, err := newMaskRuleSet(
rs[0].Expressions[0].Value,
func(mRule *maskRule, err error) {
p.logger.Error("mask rule skipped: %s: %s", mRule.String(), err.Error())
},
)
if err != nil {
return err
}
mRuleSet.Mask(event)
return nil
}
func uploadChunk(ctx context.Context, client rest.Client, uploadPath string, data []byte) error {
resp, err := client.
WithHeader("Content-Type", "application/json").
WithHeader("Content-Encoding", "gzip").
WithBytes(data).
Do(ctx, "POST", uploadPath)
if err != nil {
return fmt.Errorf("log upload failed: %w", err)
}
defer util.Close(resp)
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
return fmt.Errorf("log upload failed, server replied with HTTP %v %v", resp.StatusCode, http.StatusText(resp.StatusCode))
}
return nil
}
func (p *Plugin) logEvent(event EventV1) error {
eventBuf, err := json.Marshal(&event)
if err != nil {
return err
}
fields := map[string]interface{}{}
err = util.UnmarshalJSON(eventBuf, &fields)
if err != nil {
return err
}
p.manager.ConsoleLogger().WithFields(fields).WithFields(map[string]interface{}{
"type": "openpolicyagent.org/decision_logs",
}).Info("Decision Log")
return nil
}