From 3f23f52c5237255c90e889af950e9a62fcbf206a Mon Sep 17 00:00:00 2001
From: Gurvinder Singh <gurvindersinghdahiya@gmail.com>
Date: Fri, 6 Nov 2009 12:22:37 +0200
Subject: [PATCH] Fixed FlowTest01 and StreamReassemblyTest30

---
 src/flow.c                  | 1 +
 src/stream-tcp-reassemble.c | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/flow.c b/src/flow.c
index a1ab966439..6c48816a3c 100644
--- a/src/flow.c
+++ b/src/flow.c
@@ -795,6 +795,7 @@ static int FlowTest01 (void) {
 
     uint8_t proto_map;
 
+    FlowInitFlowProto();
     proto_map = FlowGetProtoMapping(IPPROTO_TCP);
 
     if ((flow_proto[proto_map].new_timeout != FLOW_IPPROTO_TCP_NEW_TIMEOUT) && (flow_proto[proto_map].est_timeout != FLOW_IPPROTO_TCP_EST_TIMEOUT)
diff --git a/src/stream-tcp-reassemble.c b/src/stream-tcp-reassemble.c
index 6cfd9bbf4a..5484fea934 100644
--- a/src/stream-tcp-reassemble.c
+++ b/src/stream-tcp-reassemble.c
@@ -1654,8 +1654,11 @@ static int StreamTcpCheckQueue (uint8_t *stream_contents, StreamMsgQueue *q, uin
                 break;
             /*Gap at end*/
             case 3:
-                if (cnt == 2 && msg->gap.gap_size != 3)
+                if (cnt == 3 && msg->gap.gap_size != 3 &&
+                        msg->flags & STREAM_GAP)
+                {
                     return 0;
+                }
                 break;
         }