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, 13 insertions, 0 deletions

diff --git a/src/expect/snprintf_default.c b/src/expect/snprintf_default.c
index 77f9c3b..44b7719 100644
--- a/src/expect/snprintf_default.c
+++ b/src/expect/snprintf_default.c
@@ -19,6 +19,16 @@ __snprintf_expect_timeout(char *buf, unsigned int len,
 	return 0;
 }
 
+static int
+__snprintf_expect_class(char *buf, unsigned int len,
+			  const struct nf_expect *exp)
+{
+	if (test_bit(ATTR_EXP_CLASS, exp->set))
+		return snprintf(buf, len, "class=%u ", exp->class);
+
+	return 0;
+}
+
 static int __snprintf_expect_proto(char *buf, 
 				   unsigned int len,
 				   const struct nf_expect *exp)
@@ -101,6 +111,9 @@ int __snprintf_expect_default(char *buf,
 		BUFFER_SIZE(ret, size, len, offset);
 	}
 
+	ret = __snprintf_expect_class(buf+offset, len, exp);
+	BUFFER_SIZE(ret, size, len, offset);
+
 	if (test_bit(ATTR_EXP_HELPER_NAME, exp->set)) {
 		ret = snprintf(buf+offset, len, "helper=%s", exp->helper_name);
 		BUFFER_SIZE(ret, size, len, offset);