From 34cb49f1e9ab27cb4e2455a516a89d62f458d423 Mon Sep 17 00:00:00 2001 From: Victor Julien Date: Tue, 2 Apr 2019 14:53:25 +0200 Subject: [PATCH] flow: add macro to get sp/dp Account for possibility of a reversed flow. --- src/flow.h | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/flow.h b/src/flow.h index a35ae0b9ae..feaa49f1dc 100644 --- a/src/flow.h +++ b/src/flow.h @@ -135,6 +135,11 @@ typedef struct AppLayerParserState_ AppLayerParserState; #define FLOW_IS_IPV6(f) \ (((f)->flags & FLOW_IPV6) == FLOW_IPV6) +#define FLOW_GET_SP(f) \ + ((f)->flags & FLOW_DIR_REVERSED) ? (f)->dp : (f)->sp; +#define FLOW_GET_DP(f) \ + ((f)->flags & FLOW_DIR_REVERSED) ? (f)->sp : (f)->dp; + #define FLOW_COPY_IPV4_ADDR_TO_PACKET(fa, pa) do { \ (pa)->family = AF_INET; \ (pa)->addr_data32[0] = (fa)->addr_data32[0]; \