Adding [allow|deny]_addresses settings to yaml config file

stripe · Oct 18, 2024 · 5b0cc7a · 5b0cc7a
1 parent 9556eb9
commit 5b0cc7a
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pkg/smokescreen/config_loader.go b/pkg/smokescreen/config_loader.go
@@ -28,6 +28,8 @@ type yamlConfig struct {
 	Port                 *uint16
 	DenyRanges           []string `yaml:"deny_ranges"`
 	AllowRanges          []string `yaml:"allow_ranges"`
+	DenyAddresses        []string `yaml:"deny_addresses"`
+	AllowAddresses       []string `yaml:"allow_addresses"`
 	Resolvers            []string `yaml:"resolver_addresses"`
 	StatsdAddress        string   `yaml:"statsd_address"`
 	EgressAclFile        string   `yaml:"acl_file"`
@@ -81,6 +83,16 @@ func (c *Config) UnmarshalYAML(unmarshal func(interface{}) error) error {
 		return err
 	}
 
+	err = c.SetDenyAddresses(yc.DenyAddresses)
+	if err != nil {
+		return err
+	}
+
+	err = c.SetAllowAddresses(yc.AllowAddresses)
+	if err != nil {
+		return err
+	}
+
 	err = c.SetResolverAddresses(yc.Resolvers)
 	if err != nil {
 		return err