[CWS] add custom CWS product

pkg/config/remote/client.go

@@ -63,8 +63,9 @@ type Client struct {
 	state *state.Repository
 
 	// Listeners
-	apmListeners []func(update map[string]state.APMSamplingConfig)
-	cwsListeners []func(update map[string]state.ConfigCWSDD)
+	apmListeners       []func(update map[string]state.APMSamplingConfig)
+	cwsListeners       []func(update map[string]state.ConfigCWSDD)
+	cwsCustomListeners []func(update map[string]state.ConfigCWSCustom)
 }
 
 // agentGRPCConfigFetcher defines how to retrieve config updates over a
@@ -153,19 +154,20 @@ func newClient(agentName string, updater ConfigUpdater, doTufVerification bool,
 	ctx, close := context.WithCancel(context.Background())
 
 	return &Client{
-		ID:            generateID(),
-		startupSync:   sync.Once{},
-		ctx:           ctx,
-		close:         close,
-		agentName:     agentName,
-		agentVersion:  agentVersion,
-		products:      data.ProductListToString(products),
-		state:         repository,
-		pollInterval:  pollInterval,
-		backoffPolicy: backoffPolicy,
-		apmListeners:  make([]func(update map[string]state.APMSamplingConfig), 0),
-		cwsListeners:  make([]func(update map[string]state.ConfigCWSDD), 0),
-		updater:       updater,
+		ID:                 generateID(),
+		startupSync:        sync.Once{},
+		ctx:                ctx,
+		close:              close,
+		agentName:          agentName,
+		agentVersion:       agentVersion,
+		products:           data.ProductListToString(products),
+		state:              repository,
+		pollInterval:       pollInterval,
+		backoffPolicy:      backoffPolicy,
+		apmListeners:       make([]func(update map[string]state.APMSamplingConfig), 0),
+		cwsListeners:       make([]func(update map[string]state.ConfigCWSDD), 0),
+		cwsCustomListeners: make([]func(update map[string]state.ConfigCWSCustom), 0),
+		updater:            updater,
 	}, nil
 }
 
@@ -251,6 +253,11 @@ func (c *Client) update() error {
 			listener(c.state.CWSDDConfigs())
 		}
 	}
+	if containsProduct(changedProducts, state.ProductCWSCustom) {
+		for _, listener := range c.cwsCustomListeners {
+			listener(c.state.CWSCustomConfigs())
+		}
+	}
 
 	return nil
 }
@@ -283,6 +290,15 @@ func (c *Client) RegisterCWSDDUpdate(fn func(update map[string]state.ConfigCWSDD
 	fn(c.state.CWSDDConfigs())
 }
 
+// RegisterCWSCustomUpdate registers a callback function to be called after a successful client update that will
+// contain the current state of the CWS_CUSTOM product.
+func (c *Client) RegisterCWSCustomUpdate(fn func(update map[string]state.ConfigCWSCustom)) {
+	c.m.Lock()
+	defer c.m.Unlock()
+	c.cwsCustomListeners = append(c.cwsCustomListeners, fn)
+	fn(c.state.CWSCustomConfigs())
+}
+
 func (c *Client) applyUpdate(pbUpdate *pbgo.ClientGetConfigsResponse) ([]string, error) {
 	fileMap := make(map[string][]byte, len(pbUpdate.TargetFiles))
 	for _, f := range pbUpdate.TargetFiles {

pkg/config/remote/data/product.go

@@ -13,6 +13,8 @@ const (
 	ProductAPMSampling Product = "APM_SAMPLING"
 	// ProductCWSDD is the cloud workload security product
 	ProductCWSDD Product = "CWS_DD"
+	// ProductCWSCustom is the cloud workload security product
+	ProductCWSCustom Product = "CWS_CUSTOM"
 	// ProductTesting1 is a testing product
 	ProductTesting1 Product = "TESTING1"
 )

pkg/remoteconfig/state/configs.go

@@ -22,13 +22,15 @@ import (
 	4. Add a method on the `Repository` to retrieved typed configs for the product.
 */
 
-var allProducts = []string{ProductAPMSampling, ProductCWSDD, ProductASMFeatures, ProductASMDD, ProductASMData}
+var allProducts = []string{ProductAPMSampling, ProductCWSDD, ProductCWSCustom, ProductASMFeatures, ProductASMDD, ProductASMData}
 
 const (
 	// ProductAPMSampling is the apm sampling product
 	ProductAPMSampling = "APM_SAMPLING"
 	// ProductCWSDD is the cloud workload security product managed by datadog employees
 	ProductCWSDD = "CWS_DD"
+	// ProductCWSCustom is the cloud workload security product managed by datadog customers
+	ProductCWSCustom = "CWS_CUSTOM"
 	// ProductASMFeatures is the ASM product used form ASM activation through remote config
 	ProductASMFeatures = "ASM_FEATURES"
 	// ProductASMDD is the application security monitoring product managed by datadog employees
@@ -50,6 +52,8 @@ func parseConfig(product string, raw []byte, metadata Metadata) (interface{}, er
 		c, err = parseASMFeaturesConfig(raw, metadata)
 	case ProductCWSDD:
 		c, err = parseConfigCWSDD(raw, metadata)
+	case ProductCWSCustom:
+		c, err = parseConfigCWSCustom(raw, metadata)
 	case ProductASMDD:
 		c, err = parseConfigASMDD(raw, metadata)
 	case ProductASMData:
@@ -128,6 +132,39 @@ func (r *Repository) CWSDDConfigs() map[string]ConfigCWSDD {
 	return typedConfigs
 }
 
+// ConfigCWSCustom is a deserialized CWS Custom configuration file along with its
+// associated remote config metadata
+type ConfigCWSCustom struct {
+	Config   []byte
+	Metadata Metadata
+}
+
+func parseConfigCWSCustom(data []byte, metadata Metadata) (ConfigCWSCustom, error) {
+	return ConfigCWSCustom{
+		Config:   data,
+		Metadata: metadata,
+	}, nil
+}
+
+// CWSCustomConfigs returns the currently active CWSCustom config files
+func (r *Repository) CWSCustomConfigs() map[string]ConfigCWSCustom {
+	typedConfigs := make(map[string]ConfigCWSCustom)
+
+	configs := r.getConfigs(ProductCWSCustom)
+
+	for path, conf := range configs {
+		// We control this, so if this has gone wrong something has gone horribly wrong
+		typed, ok := conf.(ConfigCWSCustom)
+		if !ok {
+			panic("unexpected config stored as CWSDD Config")
+		}
+
+		typedConfigs[path] = typed
+	}
+
+	return typedConfigs
+}
+
 // ConfigASMDD is a deserialized ASM DD configuration file along with its
 // associated remote config metadata
 type ConfigASMDD struct {

pkg/security/rconfig/rconfig.go

@@ -32,14 +32,15 @@ type RCPolicyProvider struct {
 
 	client               *remote.Client
 	onNewPoliciesReadyCb func()
-	lastConfigs          map[string]state.ConfigCWSDD
+	lastDefaults         map[string]state.ConfigCWSDD
+	lastCustoms          map[string]state.ConfigCWSCustom
 }
 
 var _ rules.PolicyProvider = (*RCPolicyProvider)(nil)
 
 // NewRCPolicyProvider returns a new Remote Config based policy provider
 func NewRCPolicyProvider(name string, agentVersion *semver.Version) (*RCPolicyProvider, error) {
-	c, err := remote.NewGRPCClient(name, agentVersion.String(), []data.Product{data.ProductCWSDD}, securityAgentRCPollInterval)
+	c, err := remote.NewGRPCClient(name, agentVersion.String(), []data.Product{data.ProductCWSDD, data.ProductCWSCustom}, securityAgentRCPollInterval)
 	if err != nil {
 		return nil, err
 	}
@@ -53,14 +54,25 @@ func NewRCPolicyProvider(name string, agentVersion *semver.Version) (*RCPolicyPr
 func (r *RCPolicyProvider) Start() {
 	log.Info("remote-config policies provider started")
 
-	r.client.RegisterCWSDDUpdate(r.rcConfigUpdateCallback)
+	r.client.RegisterCWSDDUpdate(r.rcDefaultsUpdateCallback)
+	r.client.RegisterCWSCustomUpdate(r.rcCustomsUpdateCallback)
 
 	r.client.Start()
 }
 
-func (r *RCPolicyProvider) rcConfigUpdateCallback(configs map[string]state.ConfigCWSDD) {
+func (r *RCPolicyProvider) rcDefaultsUpdateCallback(configs map[string]state.ConfigCWSDD) {
 	r.Lock()
-	r.lastConfigs = configs
+	r.lastDefaults = configs
+	r.Unlock()
+
+	log.Info("new policies from remote-config policy provider")
+
+	r.onNewPoliciesReadyCb()
+}
+
+func (r *RCPolicyProvider) rcCustomsUpdateCallback(configs map[string]state.ConfigCWSCustom) {
+	r.Lock()
+	r.lastCustoms = configs
 	r.Unlock()
 
 	log.Info("new policies from remote-config policy provider")
@@ -84,17 +96,24 @@ func (r *RCPolicyProvider) LoadPolicies(macroFilters []rules.MacroFilter, ruleFi
 	r.RLock()
 	defer r.RUnlock()
 
-	for _, c := range r.lastConfigs {
-		reader := bytes.NewReader(c.Config)
+	load := func(id string, cfg []byte) {
+		reader := bytes.NewReader(cfg)
 
-		policy, err := rules.LoadPolicy(c.Metadata.ID, "remote-config", reader, macroFilters, ruleFilters)
+		policy, err := rules.LoadPolicy(id, "remote-config", reader, macroFilters, ruleFilters)
 		if err != nil {
 			errs = multierror.Append(errs, err)
 		}
 		normalize(policy)
 		policies = append(policies, policy)
 	}
 
+	for _, c := range r.lastDefaults {
+		load(c.Metadata.ID, c.Config)
+	}
+	for _, c := range r.lastCustoms {
+		load(c.Metadata.ID, c.Config)
+	}
+
 	return policies, errs
 }