-
Notifications
You must be signed in to change notification settings - Fork 94
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
031f147
commit dde10bf
Showing
13 changed files
with
497 additions
and
486 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,20 @@ | ||
| # Drop | ||
|
|
||
| The `Drop` filter always drops any packet passed through it. This is useful in | ||
| combination with conditional flow filters like [`Matches`](./match.md) | ||
|
|
||
| #### Filter name | ||
| ```text | ||
| quilkin.extensions.filters.drop.v1alpha1.Drop | ||
| ``` | ||
|
|
||
| ### Configuration Examples | ||
| {{#include match.md:example}} | ||
|
|
||
| ### Configuration | ||
|
|
||
| No defined configuration options. | ||
|
|
||
| ### Metrics | ||
|
|
||
| This filter currently exports no metrics. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,20 @@ | ||
| # Pass | ||
|
|
||
| The `Pass` filter that always passes any packet through it. This is useful in | ||
| combination with conditional flow filters like [`Matches`](./match.md) | ||
|
|
||
| #### Filter name | ||
| ```text | ||
| quilkin.extensions.filters.pass.v1alpha1.Pass | ||
| ``` | ||
|
|
||
| ### Configuration Examples | ||
| {{#include match.md:example}} | ||
|
|
||
| ### Configuration | ||
|
|
||
| No defined configuration options. | ||
|
|
||
| ### Metrics | ||
|
|
||
| This filter currently exports no metrics. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,162 @@ | ||
| mod config; | ||
|
|
||
| use crate::{ | ||
| config::ConfigType, | ||
| filters::{prelude::*, registry::FilterRegistry}, | ||
| metadata::Value, | ||
| }; | ||
|
|
||
| pub use config::Config; | ||
|
|
||
| pub const NAME: &str = "quilkin.extensions.filters.matches.v1alpha1.Match"; | ||
|
|
||
| /// Creates a new factory for generating match filters. | ||
| pub fn factory() -> DynFilterFactory { | ||
| Box::from(MatchFactory::new()) | ||
| } | ||
|
|
||
| struct ConfigInstance { | ||
| metadata_key: String, | ||
| branches: Vec<(Value, FilterInstance)>, | ||
| fallthrough: FilterInstance, | ||
| } | ||
|
|
||
| impl ConfigInstance { | ||
| fn new( | ||
| config: config::DirectionalConfig, | ||
| filter_registry: FilterRegistry, | ||
| metrics_registry: prometheus::Registry, | ||
| ) -> Result<Self, Error> { | ||
| let map_to_instance = |filter: &str, config_type: Option<ConfigType>| { | ||
| let args = CreateFilterArgs::new( | ||
| filter_registry.clone(), | ||
| metrics_registry.clone(), | ||
| config_type, | ||
| ) | ||
| .with_metrics_registry(metrics_registry.clone()); | ||
|
|
||
| filter_registry.get(filter, args) | ||
| }; | ||
|
|
||
| let branches = config | ||
| .branches | ||
| .iter() | ||
| .map(|branch| { | ||
| map_to_instance(&branch.filter.id, branch.filter.config.clone()) | ||
| .map(|instance| (branch.value.clone(), instance)) | ||
| }) | ||
| .collect::<Result<_, _>>()?; | ||
|
|
||
| Ok(Self { | ||
| metadata_key: config.metadata_key, | ||
| branches, | ||
| fallthrough: map_to_instance(&config.fallthrough.0.id, config.fallthrough.0.config)?, | ||
| }) | ||
| } | ||
| } | ||
|
|
||
| struct MatchInstance { | ||
| on_read_filters: Option<ConfigInstance>, | ||
| on_write_filters: Option<ConfigInstance>, | ||
| } | ||
|
|
||
| impl MatchInstance { | ||
| fn new( | ||
| config: Config, | ||
| filter_registry: FilterRegistry, | ||
| metrics_registry: prometheus::Registry, | ||
| ) -> Result<Self, Error> { | ||
| let on_read_filters = config | ||
| .on_read | ||
| .map(|config| { | ||
| ConfigInstance::new(config, filter_registry.clone(), metrics_registry.clone()) | ||
| }) | ||
| .transpose()?; | ||
|
|
||
| let on_write_filters = config | ||
| .on_write | ||
| .map(|config| { | ||
| ConfigInstance::new(config, filter_registry.clone(), metrics_registry.clone()) | ||
| }) | ||
| .transpose()?; | ||
|
|
||
| if on_read_filters.is_none() && on_write_filters.is_none() { | ||
| return Err(Error::MissingConfig(NAME)); | ||
| } | ||
|
|
||
| Ok(Self { | ||
| on_read_filters, | ||
| on_write_filters, | ||
| }) | ||
| } | ||
| } | ||
|
|
||
| fn match_filter<'config, Ctx, R>( | ||
| config: &'config Option<ConfigInstance>, | ||
| ctx: Ctx, | ||
| get_metadata: impl for<'ctx> Fn(&'ctx Ctx, &'config String) -> Option<&'ctx Value>, | ||
| and_then: impl Fn(Ctx, &'config FilterInstance) -> Option<R>, | ||
| ) -> Option<R> | ||
| where | ||
| Ctx: Into<R>, | ||
| { | ||
| match config { | ||
| Some(config) => { | ||
| let value = (get_metadata)(&ctx, &config.metadata_key)?; | ||
|
|
||
| match config.branches.iter().find(|(key, _)| key == value) { | ||
| Some((_, instance)) => (and_then)(ctx, instance), | ||
| None => (and_then)(ctx, &config.fallthrough), | ||
| } | ||
| } | ||
| None => Some(ctx.into()), | ||
| } | ||
| } | ||
|
|
||
| impl Filter for MatchInstance { | ||
| #[cfg_attr(feature = "instrument", tracing::instrument(skip(self, ctx)))] | ||
| fn read(&self, ctx: ReadContext) -> Option<ReadResponse> { | ||
| match_filter( | ||
| &self.on_read_filters, | ||
| ctx, | ||
| |ctx, metadata_key| ctx.metadata.get(metadata_key), | ||
| |ctx, instance| instance.filter.read(ctx), | ||
| ) | ||
| } | ||
|
|
||
| #[cfg_attr(feature = "instrument", tracing::instrument(skip(self, ctx)))] | ||
| fn write(&self, ctx: WriteContext) -> Option<WriteResponse> { | ||
| match_filter( | ||
| &self.on_write_filters, | ||
| ctx, | ||
| |ctx, metadata_key| ctx.metadata.get(metadata_key), | ||
| |ctx, instance| instance.filter.write(ctx), | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| struct MatchFactory; | ||
|
|
||
| impl MatchFactory { | ||
| pub fn new() -> Self { | ||
| Self | ||
| } | ||
| } | ||
|
|
||
| impl FilterFactory for MatchFactory { | ||
| fn name(&self) -> &'static str { | ||
| NAME | ||
| } | ||
|
|
||
| fn create_filter(&self, args: CreateFilterArgs) -> Result<FilterInstance, Error> { | ||
| let (config_json, config) = self | ||
| .require_config(args.config)? | ||
| .deserialize::<Config, config::proto::Match>(self.name())?; | ||
|
|
||
| let filter = MatchInstance::new(config, args.filter_registry, args.metrics_registry)?; | ||
| Ok(FilterInstance::new( | ||
| config_json, | ||
| Box::new(filter) as Box<dyn Filter>, | ||
| )) | ||
| } | ||
| } |
Oops, something went wrong.