expect: add class support

This patch allows you to specify the expectation class.

Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>

1 files changed, 6 insertions, 0 deletions

diff --git a/src/expect/getter.c b/src/expect/getter.c
index 2f8df76..06c3bca 100644
--- a/src/expect/getter.c
+++ b/src/expect/getter.c
@@ -39,6 +39,11 @@ static const void *get_exp_attr_flags(const struct nf_expect *exp)
 	return &exp->flags;
 }
 
+static const void *get_exp_attr_class(const struct nf_expect *exp)
+{
+	return &exp->class;
+}
+
 static const void *get_exp_attr_helper_name(const struct nf_expect *exp)
 {
 	return exp->helper_name;
@@ -52,4 +57,5 @@ const get_exp_attr get_exp_attr_array[ATTR_EXP_MAX] = {
 	[ATTR_EXP_ZONE]			= get_exp_attr_zone,
 	[ATTR_EXP_FLAGS]		= get_exp_attr_flags,
 	[ATTR_EXP_HELPER_NAME]		= get_exp_attr_helper_name,
+	[ATTR_EXP_CLASS]		= get_exp_attr_class,
 };