Stream: fix unittests after ssn pool changes.

diff --git a/src/stream-tcp.c b/src/stream-tcp.c
index 0c433e76acdcba1640c36d63a545c3dcc37c049d..bddf8dce30a9559d2823a998ddd6dc2ac49fa4fb 100644 (file)
--- a/src/stream-tcp.c
+++ b/src/stream-tcp.c
@@ -566,6 +566,22 @@ void StreamTcpInitConfig(char quiet)
      * values. */
     FlowSetProtoFreeFunc(IPPROTO_TCP, StreamTcpSessionClear);
     FlowSetFlowStateFunc(IPPROTO_TCP, StreamTcpGetFlowState);
+
+#ifdef UNITTESTS
+    if (RunmodeIsUnittests()) {
+        SCMutexLock(&ssn_pool_mutex);
+        if (ssn_pool == NULL) {
+            ssn_pool = PoolThreadInit(1, /* thread */
+                    0, /* unlimited */
+                    stream_config.prealloc_sessions,
+                    sizeof(TcpSession),
+                    StreamTcpSessionPoolAlloc,
+                    StreamTcpSessionPoolInit, NULL,
+                    StreamTcpSessionPoolCleanup, NULL);
+        }
+        SCMutexUnlock(&ssn_pool_mutex);
+    }
+#endif
 }
 
 void StreamTcpFreeConfig(char quiet)