From 16aebe5add1a16aa97fdbfa8362b0d2f98141d78 Mon Sep 17 00:00:00 2001
From: Victor Julien <victor@inliniac.net>
Date: Mon, 8 Feb 2010 17:49:47 +0100
Subject: [PATCH] Fixup smb tests.

---
 src/app-layer-smb.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/app-layer-smb.c b/src/app-layer-smb.c
index 1f82074486..3a26ca57b6 100644
--- a/src/app-layer-smb.c
+++ b/src/app-layer-smb.c
@@ -1491,9 +1491,11 @@ int SMBParserTest03(void) {
     int r = 0;
     memset(&f, 0, sizeof(f));
     memset(&ssn, 0, sizeof(ssn));
-    StreamL7DataPtrInit(&ssn,StreamL7GetStorageSize());
     f.protoctx = (void *)&ssn;
 
+    StreamTcpInitConfig(TRUE);
+    StreamL7DataPtrInit(&ssn);
+
     r = AppLayerParse(&f, ALPROTO_SMB, STREAM_TOSERVER|STREAM_START, smbbuf1, smblen1);
     if (r != 0) {
         printf("smb header check returned %" PRId32 ", expected 0: ", r);
@@ -1528,6 +1530,8 @@ int SMBParserTest03(void) {
     }
     printUUID("BIND", smb_state->dcerpc.dcerpcbindbindack.uuid_entry);
 end:
+    StreamL7DataPtrFree(&ssn);
+    StreamTcpFreeConfig(TRUE);
     return result;
 }
 
@@ -1596,9 +1600,11 @@ int SMBParserTest04(void) {
     int r = 0;
     memset(&f, 0, sizeof(f));
     memset(&ssn, 0, sizeof(ssn));
-    StreamL7DataPtrInit(&ssn,StreamL7GetStorageSize());
     f.protoctx = (void *)&ssn;
 
+    StreamTcpInitConfig(TRUE);
+    StreamL7DataPtrInit(&ssn);
+
     r = AppLayerParse(&f, ALPROTO_SMB, STREAM_TOSERVER|STREAM_START, smbbuf1, smblen1);
     if (r != 0) {
         printf("smb header check returned %" PRId32 ", expected 0: ", r);
@@ -1638,6 +1644,8 @@ int SMBParserTest04(void) {
         goto end;
     }
 end:
+    StreamL7DataPtrFree(&ssn);
+    StreamTcpFreeConfig(TRUE);
     return result;
 }