filterx: add has-sdata() function

Signed-off-by: Szilard Parrag <[email protected]>
axoflow · Aug 7, 2024 · 0642d73 · 0642d73
1 parent d7eca3a
commit 0642d73
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/lib/filterx/filterx-globals.c b/lib/filterx/filterx-globals.c
@@ -94,6 +94,8 @@ _simple_init(void)
   g_assert(filterx_builtin_simple_function_register("upper", filterx_simple_function_upper));
   g_assert(filterx_builtin_simple_function_register("is_sdata_from_enterprise",
                                                     filterx_simple_function_is_sdata_from_enterprise));
+  g_assert(filterx_builtin_simple_function_register("has_sdata",
+                                                    filterx_simple_function_has_sdata));
 }
 
 static void

diff --git a/lib/filterx/func-sdata.c b/lib/filterx/func-sdata.c
@@ -58,3 +58,18 @@ filterx_simple_function_is_sdata_from_enterprise(FilterXExpr *s, GPtrArray *args
   filterx_object_unref(obj);
   return filterx_boolean_new(contains);
 }
+
+
+FilterXObject *
+filterx_simple_function_has_sdata(FilterXExpr *s, GPtrArray *args)
+{
+  if (args && args->len != 0)
+    {
+      filterx_simple_function_argument_error(s, "Incorrect number of arguments", FALSE);
+      return NULL;
+    }
+
+  FilterXEvalContext *context = filterx_eval_get_context();
+  LogMessage *msg = context->msgs[0];
+  return filterx_boolean_new(msg->num_sdata != 0);
+}
diff --git a/lib/filterx/func-sdata.h b/lib/filterx/func-sdata.h
@@ -27,5 +27,6 @@
 #include "filterx/expr-function.h"
 
 FilterXObject *filterx_simple_function_is_sdata_from_enterprise(FilterXExpr *s, GPtrArray *args);
+FilterXObject *filterx_simple_function_has_sdata(FilterXExpr *s, GPtrArray *args);
 
 #endif