From 849342adcb7099d23b39192a4369908bb0259eee Mon Sep 17 00:00:00 2001
From: Pablo Neira Ayuso <pablo@netfilter.org>
Date: Wed, 4 Jan 2012 16:45:01 +0100
Subject: expect: extend test_api for the expectation API

Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
---
 qa/test_api.c | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

(limited to 'qa/test_api.c')

diff --git a/qa/test_api.c b/qa/test_api.c
index 60c6520..1efaf2f 100644
--- a/qa/test_api.c
+++ b/qa/test_api.c
@@ -33,6 +33,7 @@ int main(void)
 {
 	int ret, i;
 	struct nf_conntrack *ct, *tmp;
+	struct nf_expect *exp;
 	char data[32];
 	const char *val;
 	int status;
@@ -129,7 +130,58 @@ int main(void)
 		eval_sigterm(status);
 	}
 
+	exp = nfexp_new();
+	if (!exp) {
+		perror("nfexp_new");
+		return 0;
+	}
+
+	printf("== test expect set API ==\n");
+	ret = fork();
+	if (ret == 0) {
+		for (i=0; i<ATTR_EXP_MAX; i++)
+			nfexp_set_attr(exp, i, data);
+		exit(0);
+	} else {
+		wait(&status);
+		eval_sigterm(status);
+	}
+
+	for (i=0; i<ATTR_EXP_MAX; i++)
+		nfexp_set_attr(exp, i, data);
+
+	printf("== test expect get API ==\n");
+	ret = fork();
+	if (ret == 0) {
+		for (i=0; i<ATTR_EXP_MAX; i++)
+			nfexp_get_attr(exp, i);
+		exit(0);
+	} else {
+		wait(&status);
+		eval_sigterm(status);
+	}
+
+	printf("== validate expect set API ==\n");
+	ret = fork();
+	if (ret == 0) {
+		for (i=0; i<ATTR_EXP_MAX; i++) {
+			data[0] = (uint8_t) i;
+			nfexp_set_attr(exp, i, data);
+			val = nfexp_get_attr(exp, i);
+			if (val[0] != data[0]) {
+				printf("ERROR: set/get operations don't match "
+				       "for attribute %d (%x != %x)\n",
+					i, val[0], data[0]);
+			}
+		}
+		exit(0);
+	} else {
+		wait(&status);
+		eval_sigterm(status);
+	}
+
 	nfct_destroy(ct);
 	nfct_destroy(tmp);
+	nfexp_destroy(exp);
 	return EXIT_SUCCESS;
 }
-- 
cgit v1.2.3