diff --git a/marine/marine.py b/marine/marine.py index 33f240c..786be32 100644 --- a/marine/marine.py +++ b/marine/marine.py @@ -90,6 +90,10 @@ def validate_bpf(self, bpf: str) -> bool: bpf = bpf.encode("utf-8") return bool(self._marine.validate_bpf(bpf)) + def validate_display_filter(self, display_filter: str) -> bool: + display_filter = display_filter.encode("utf-8") + return bool(self._marine.validate_display_filter(display_filter)) + @staticmethod def _parse_output(output: str) -> List[str]: # TODO: this is a bottleneck. Find a better way to provide output from the c code diff --git a/tests/marine/test_marine.py b/tests/marine/test_marine.py index 62bcff4..ac35af7 100644 --- a/tests/marine/test_marine.py +++ b/tests/marine/test_marine.py @@ -396,3 +396,11 @@ def test_validate_bpf_success(marine_instance: Marine): def test_validate_bpf_failure(marine_instance: Marine): assert not marine_instance.validate_bpf("what is this bpf?") + + +def test_validate_display_filter_success(marine_instance: Marine): + assert marine_instance.validate_display_filter("tcp") + + +def test_validate_display_filter_failure(marine_instance: Marine): + assert not marine_instance.validate_display_filter("illegal_filter")