From 8bc1e120eae098c759ebdc182a8a2db0ebbe2531 Mon Sep 17 00:00:00 2001 From: Victor Julien Date: Mon, 27 May 2019 09:46:41 +0200 Subject: [PATCH] app-layer: make tests tx aware Make tests minimally TX-aware so we can asume all parsers use the tx API. --- src/app-layer-parser.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/app-layer-parser.c b/src/app-layer-parser.c index 858b56d46a..7ee3f37c69 100644 --- a/src/app-layer-parser.c +++ b/src/app-layer-parser.c @@ -1979,6 +1979,12 @@ static void TestProtocolStateFree(void *s) SCFree(s); } +static uint64_t TestProtocolGetTxCnt(void *state) +{ + /* single tx */ + return 1; +} + void AppLayerParserRegisterProtocolUnittests(uint8_t ipproto, AppProto alproto, void (*RegisterUnittests)(void)) { @@ -2026,6 +2032,7 @@ static int AppLayerParserTest01(void) TestProtocolParser); AppLayerParserRegisterStateFuncs(IPPROTO_TCP, ALPROTO_TEST, TestProtocolStateAlloc, TestProtocolStateFree); + AppLayerParserRegisterGetTxCnt(IPPROTO_TCP, ALPROTO_TEST, TestProtocolGetTxCnt); f = UTHBuildFlow(AF_INET, "1.2.3.4", "4.3.2.1", 20, 40); if (f == NULL) @@ -2080,6 +2087,7 @@ static int AppLayerParserTest02(void) TestProtocolParser); AppLayerParserRegisterStateFuncs(IPPROTO_UDP, ALPROTO_TEST, TestProtocolStateAlloc, TestProtocolStateFree); + AppLayerParserRegisterGetTxCnt(IPPROTO_UDP, ALPROTO_TEST, TestProtocolGetTxCnt); f = UTHBuildFlow(AF_INET, "1.2.3.4", "4.3.2.1", 20, 40); if (f == NULL)