diff --git a/src/app-layer-dcerpc.c b/src/app-layer-dcerpc.c index dfb550186f..a427848acb 100644 --- a/src/app-layer-dcerpc.c +++ b/src/app-layer-dcerpc.c @@ -5,6 +5,7 @@ * \author Kirby Kuehl */ #include "suricata-common.h" +#include "suricata.h" #include "debug.h" #include "decode.h" @@ -227,7 +228,9 @@ static uint32_t DCERPCParseBINDCTXItem(DCERPC *dcerpc, uint8_t *input, uint32_t TAILQ_INSERT_HEAD(&dcerpc->dcerpcbindbindack.uuid_list, dcerpc->dcerpcbindbindack.uuid_entry, next); #ifdef UNITTESTS - printUUID("BIND", dcerpc->dcerpcbindbindack.uuid_entry); + if (RunmodeIsUnittests()) { + printUUID("BIND", dcerpc->dcerpcbindbindack.uuid_entry); + } #endif dcerpc->dcerpcbindbindack.numctxitemsleft--; dcerpc->bytesprocessed += (44); @@ -431,7 +434,9 @@ static uint32_t DCERPCParseBINDCTXItem(DCERPC *dcerpc, uint8_t *input, uint32_t TAILQ_INSERT_HEAD(&dcerpc->dcerpcbindbindack.uuid_list, dcerpc->dcerpcbindbindack.uuid_entry, next); #ifdef UNITTESTS - printUUID("BIND", dcerpc->dcerpcbindbindack.uuid_entry); + if (RunmodeIsUnittests()) { + printUUID("BIND", dcerpc->dcerpcbindbindack.uuid_entry); + } #endif dcerpc->dcerpcbindbindack.numctxitemsleft--; dcerpc->bytesprocessed += (44); @@ -472,7 +477,9 @@ static uint32_t DCERPCParseBINDACKCTXItem(DCERPC *dcerpc, uint8_t *input, uint32 - dcerpc->dcerpcbindbindack.numctxitemsleft) { uuid_entry->result = dcerpc->dcerpcbindbindack.result; #ifdef UNITTESTS - printUUID("BIND_ACK", uuid_entry); + if (RunmodeIsUnittests()) { + printUUID("BIND_ACK", uuid_entry); + } #endif break; } @@ -582,7 +589,9 @@ static uint32_t DCERPCParseBINDACKCTXItem(DCERPC *dcerpc, uint8_t *input, uint32 - dcerpc->dcerpcbindbindack.numctxitemsleft) { uuid_entry->result = dcerpc->dcerpcbindbindack.result; #ifdef UNITTESTS - printUUID("BIND_ACK", uuid_entry); + if (RunmodeIsUnittests()) { + printUUID("BIND_ACK", uuid_entry); + } #endif break; } diff --git a/src/app-layer-smb.c b/src/app-layer-smb.c index 3a26ca57b6..a0b34c361e 100644 --- a/src/app-layer-smb.c +++ b/src/app-layer-smb.c @@ -928,7 +928,7 @@ static int SMBParse(Flow *f, void *smb_state, AppLayerParserState *pstate, long int parsed = 0; if (pstate == NULL) SCReturnInt(-1); - while (sstate->bytesprocessed < NBSS_HDR_LEN) { + while (input_len && sstate->bytesprocessed < NBSS_HDR_LEN) { retval = NBSSParseHeader(f, smb_state, pstate, input, input_len, output);