From 304bca9898ef72a9e09915d1d65b686fe9c4f187 Mon Sep 17 00:00:00 2001 From: Pablo Neira Ayuso Date: Mon, 7 Oct 2013 15:20:26 +0200 Subject: emulate expectation attachment via myct->exp Signed-off-by: Pablo Neira Ayuso --- main.c | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'main.c') diff --git a/main.c b/main.c index 7f8221d..fdb1abe 100755 --- a/main.c +++ b/main.c @@ -130,8 +130,17 @@ cthelper_process_packet(const uint8_t *pkt, uint32_t pktlen, printf("\e[0m\n"); } - } else + } else { ret = h->cb(pktb, dataoff, ct->myct, ctinfo); + /* Release expectation and expectation counter if added via + * myct->exp. + */ + if (ct->myct->exp) { + cthelper_test_stats.ct_expect_created++; + nfexp_destroy(ct->myct->exp); + ct->myct->exp = NULL; + } + } if (ret == NF_DROP) cthelper_test_stats.pkts_dropped++; -- cgit v1.2.3