suricata
runmode-unittests.c
Go to the documentation of this file.
1 /* Copyright (C) 2013 Open Information Security Foundation
2  *
3  * You can copy, redistribute or modify this Program under the terms of
4  * the GNU General Public License version 2 as published by the Free
5  * Software Foundation.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * version 2 along with this program; if not, write to the Free Software
14  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
15  * 02110-1301, USA.
16  */
17 
18 /** \file
19  *
20  * \author Eric Leblond <eric@regit.org>
21  */
22 
23 #include "suricata-common.h"
24 #include "util-unittest.h"
25 #include "runmode-unittests.h"
26 
27 #ifdef UNITTESTS
28 
29 #include "detect-parse.h"
30 #include "detect-engine.h"
31 #include "detect-engine-address.h"
32 #include "detect-engine-proto.h"
33 #include "detect-engine-port.h"
34 #include "detect-engine-mpm.h"
35 #include "detect-engine-sigorder.h"
36 #include "detect-engine-payload.h"
38 #include "detect-engine-state.h"
39 #include "detect-engine-tag.h"
40 #include "detect-engine-modbus.h"
41 #include "detect-fast-pattern.h"
42 #include "flow.h"
43 #include "flow-timeout.h"
44 #include "flow-manager.h"
45 #include "flow-var.h"
46 #include "flow-bit.h"
47 #include "pkt-var.h"
48 
49 #include "host.h"
50 #include "host-bit.h"
51 #include "ippair.h"
52 #include "ippair-bit.h"
53 #include "unix-manager.h"
54 
55 #include "stream-tcp.h"
56 
57 #include "app-layer-detect-proto.h"
58 #include "app-layer-parser.h"
59 #include "app-layer.h"
60 #include "app-layer-dcerpc.h"
61 #include "app-layer-dcerpc-udp.h"
62 #include "app-layer-htp.h"
63 #include "app-layer-ftp.h"
64 #include "app-layer-ssl.h"
65 #include "app-layer-ssh.h"
66 #include "app-layer-smtp.h"
67 
68 #include "util-action.h"
69 #include "util-radix-tree.h"
70 #include "util-host-os-info.h"
71 #include "util-cidr.h"
72 #include "util-unittest-helper.h"
73 #include "util-time.h"
74 #include "util-rule-vars.h"
76 #include "util-threshold-config.h"
77 #include "util-reference-config.h"
78 #include "util-profiling.h"
79 #include "util-magic.h"
80 #include "util-memcmp.h"
81 #include "util-misc.h"
82 #include "util-signal.h"
83 
84 #include "reputation.h"
85 #include "util-atomic.h"
86 #include "util-spm.h"
87 #include "util-hash.h"
88 #include "util-hashlist.h"
89 #include "util-bloomfilter.h"
91 #include "util-pool.h"
92 #include "util-byte.h"
93 #include "util-proto-name.h"
94 #include "util-macset.h"
95 #include "util-memrchr.h"
96 
97 #include "util-mpm-ac.h"
98 #include "util-mpm-hs.h"
99 
100 #include "conf.h"
101 #include "conf-yaml-loader.h"
102 #include "tmqh-flow.h"
103 #include "defrag.h"
104 #include "detect-engine-siggroup.h"
105 
106 #include "util-streaming-buffer.h"
107 #include "util-lua.h"
108 
109 #ifdef OS_WIN32
110 #include "win32-syscall.h"
111 #endif
112 
113 #ifdef WINDIVERT
114 #include "source-windivert.h"
115 #endif
116 
117 #ifdef HAVE_NSS
118 #include <prinit.h>
119 #include <nss.h>
120 #endif
121 
122 #endif /* UNITTESTS */
123 
124 void TmqhSetup (void);
125 
126 #ifdef UNITTESTS
127 static void RegisterUnittests(void)
128 {
184 #ifdef PROFILING
186 #endif
196  DetectPortTests();
203 #ifdef OS_WIN32
204  Win32SyscallRegisterTests();
205 #endif
206 #ifdef WINDIVERT
207  SourceWinDivertRegisterTests();
208 #endif
209 }
210 #endif
211 
212 /**
213  * Run or list unittests
214  *
215  * \param list_unittests If set to 1, list unittests. Run them if set to 0.
216  * \param regex_arg A regular expression to select unittests to run
217  *
218  * This function is terminal and will call exit after being called.
219  */
220 
221 void RunUnittests(int list_unittests, const char *regex_arg)
222 {
223 #ifdef UNITTESTS
224  /* Initializations for global vars, queues, etc (memsets, mutex init..) */
226 
227 #ifdef HAVE_LUAJIT
228  if (LuajitSetupStatesPool() != 0) {
229  exit(EXIT_FAILURE);
230  }
231 #endif
232 
234  /* load the pattern matchers */
235  MpmTableSetup();
236  SpmTableSetup();
237 
238  StorageInit();
239  AppLayerSetup();
240 
241  /* hardcoded initialization code */
242  SigTableSetup(); /* load the rule keywords */
243  TmqhSetup();
244 
245  CIDRInit();
246 
247  SCProtoNameInit();
248 
249  TagInitCtx();
251  SCClassConfInit();
252 
253  UtInitialize();
254 
256 
257  HostBitInitCtx();
258 
259  StorageFinalize();
260  /* test and initialize the unittesting subsystem */
261  if (regex_arg == NULL){
262  regex_arg = ".*";
263  UtRunSelftest(regex_arg); /* inits and cleans up again */
264  }
265 
266 #ifdef HAVE_NSS
267  /* init NSS for hashing */
268  PR_Init(PR_USER_THREAD, PR_PRIORITY_NORMAL, 0);
269  NSS_NoDB_Init(NULL);
270 #endif
271 
272 
275 
276  RegisterUnittests();
277 
278  if (list_unittests) {
280  } else {
281  /* global packet pool */
282  extern intmax_t max_pending_packets;
283  max_pending_packets = 128;
284  PacketPoolInit();
285 
286  uint32_t failed = UtRunTests(regex_arg);
288  UtCleanup();
289 #ifdef BUILD_HYPERSCAN
291 #endif
292  if (failed) {
293  exit(EXIT_FAILURE);
294  }
295  }
296 
297 #ifdef HAVE_LUAJIT
298  LuajitFreeStatesPool();
299 #endif
300 
301  exit(EXIT_SUCCESS);
302 #else
303  FatalError(SC_ERR_FATAL, "Unittests are not build-in");
304 #endif /* UNITTESTS */
305 }
306 
SCProtoNameInit
void SCProtoNameInit()
Function to load the protocol names from the specified protocol file.
Definition: util-proto-name.c:40
util-byte.h
host.h
app-layer-dcerpc.h
TagInitCtx
void TagInitCtx(void)
Definition: detect-engine-tag.c:51
app-layer-dcerpc-udp.h
DecodeUDPV4RegisterTests
void DecodeUDPV4RegisterTests(void)
Definition: decode-udp.c:221
ippair.h
AppLayerHtpNeedFileInspection
void AppLayerHtpNeedFileInspection(void)
Sets a flag that informs the HTP app layer that some module in the engine needs the http request file...
Definition: app-layer-htp.c:515
detect-engine.h
detect-engine-proto.h
SCReferenceConfInit
void SCReferenceConfInit(void)
Definition: util-reference-config.c:56
win32-syscall.h
DetectAddressTests
void DetectAddressTests(void)
app-layer-ssh.h
StorageInit
void StorageInit(void)
Definition: util-storage.c:67
MagicRegisterTests
void MagicRegisterTests(void)
Definition: util-magic.c:585
util-hashlist.h
detect-engine-siggroup.h
stream-tcp.h
detect-engine-modbus.h
BloomFilterCountingRegisterTests
void BloomFilterCountingRegisterTests(void)
Definition: util-bloomfilter-counting.c:393
runmode-unittests.h
TmqhSetup
void TmqhSetup(void)
Definition: tm-queuehandlers.c:39
regex_arg
const char * regex_arg
Definition: runmode-unittests.h:28
FlowRegisterTests
void FlowRegisterTests(void)
Function to register the Flow Unitests.
Definition: flow.c:1390
PoolRegisterTests
void PoolRegisterTests(void)
Definition: util-pool.c:715
SigTableSetup
void SigTableSetup(void)
Definition: detect-engine-register.c:426
DecodePPPRegisterTests
void DecodePPPRegisterTests(void)
Definition: decode-ppp.c:311
util-lua.h
IPPairRegisterUnittests
void IPPairRegisterUnittests(void)
Definition: ippair.c:773
ippair-bit.h
DcePayloadRegisterTests
void DcePayloadRegisterTests(void)
Definition: detect-engine-dcepayload.c:3190
util-macset.h
UtListTests
void UtListTests(const char *regex_arg)
List all registered unit tests.
Definition: util-unittest.c:160
util-hash.h
MpmRegisterTests
void MpmRegisterTests(void)
Definition: util-mpm.c:617
TmqhFlowRegisterTests
void TmqhFlowRegisterTests(void)
Definition: tmqh-flow.c:375
UtRunSelftest
int UtRunSelftest(const char *regex_arg)
Run self tests.
Definition: util-unittest.c:326
RegisterAllModules
void RegisterAllModules(void)
Definition: suricata.c:837
flow-bit.h
TmModuleRegisterTests
void TmModuleRegisterTests(void)
register all unittests for the tm modules
Definition: tm-modules.c:165
util-bloomfilter-counting.h
SCSigRegisterSignatureOrderingTests
void SCSigRegisterSignatureOrderingTests(void)
Definition: detect-engine-sigorder.c:2152
AppLayerUnittestsRegister
void AppLayerUnittestsRegister(void)
Definition: app-layer.c:2649
GlobalsInitPreConfig
void GlobalsInitPreConfig(void)
Definition: suricata.c:316
MemrchrRegisterTests
void MemrchrRegisterTests(void)
Definition: util-memrchr.c:63
app-layer-ftp.h
detect-engine-payload.h
AppLayerParserRegisterUnittests
void AppLayerParserRegisterUnittests(void)
Definition: app-layer-parser.c:1858
HostBitInitCtx
void HostBitInitCtx(void)
Definition: host-bit.c:49
util-unittest.h
util-unittest-helper.h
SigGroupHeadRegisterTests
void SigGroupHeadRegisterTests(void)
Definition: detect-engine-siggroup.c:1205
PacketPoolInit
void PacketPoolInit(void)
Definition: tmqh-packetpool.c:302
SigRegisterTests
void SigRegisterTests(void)
Definition: detect.c:5323
SigTableRegisterTests
void SigTableRegisterTests(void)
Definition: detect-engine-register.c:644
UtilSpmSearchRegistertests
void UtilSpmSearchRegistertests(void)
Definition: util-spm.c:2677
FlowBitRegisterTests
void FlowBitRegisterTests(void)
Definition: flow-bit.c:402
DecodeMPLSRegisterTests
void DecodeMPLSRegisterTests(void)
Definition: decode-mpls.c:377
util-memcmp.h
util-signal.h
DecodeEthernetRegisterTests
void DecodeEthernetRegisterTests(void)
Registers Ethernet unit tests.
Definition: decode-ethernet.c:175
AppLayerSetup
int AppLayerSetup(void)
Setup the app layer.
Definition: app-layer.c:821
app-layer-detect-proto.h
util-cidr.h
app-layer-htp.h
ByteRegisterTests
void ByteRegisterTests(void)
Definition: util-byte.c:1019
SCRadixRegisterTests
void SCRadixRegisterTests(void)
Definition: util-radix-tree.c:4044
DecodeCHDLCRegisterTests
void DecodeCHDLCRegisterTests(void)
Registers Ethernet unit tests.
Definition: decode-chdlc.c:103
SpmTableSetup
void SpmTableSetup(void)
Definition: util-spm.c:123
UtInitialize
void UtInitialize(void)
Initialize unit test list.
Definition: util-unittest.c:260
UTHRegisterTests
void UTHRegisterTests(void)
Definition: util-unittest-helper.c:1198
SCRuleVarsRegisterTests
void SCRuleVarsRegisterTests(void)
Definition: util-rule-vars.c:520
SCLogRegisterTests
void SCLogRegisterTests()
Definition: util-debug.c:1721
AppLayerHtpEnableRequestBodyCallback
void AppLayerHtpEnableRequestBodyCallback(void)
Sets a flag that informs the HTP app layer that some module in the engine needs the http request body...
Definition: app-layer-htp.c:473
HashTableRegisterTests
void HashTableRegisterTests(void)
Definition: util-hash.c:427
util-reference-config.h
detect-engine-mpm.h
DefragRegisterTests
void DefragRegisterTests(void)
Definition: defrag.c:2479
pkt-var.h
SCHInfoRegisterTests
void SCHInfoRegisterTests(void)
Definition: util-host-os-info.c:1602
util-bloomfilter.h
detect-engine-port.h
util-atomic.h
util-time.h
app-layer-parser.h
detect-engine-tag.h
util-profiling.h
util-rule-vars.h
conf-yaml-loader.h
DecodeICMPV6RegisterTests
void DecodeICMPV6RegisterTests(void)
Registers ICMPV6 unit tests.
Definition: decode-icmpv6.c:1585
conf.h
util-magic.h
StorageFinalize
int StorageFinalize(void)
Definition: util-storage.c:139
util-radix-tree.h
source-windivert.h
SCAtomicRegisterTests
void SCAtomicRegisterTests(void)
Definition: util-atomic.c:67
detect-engine-state.h
Data structures and function prototypes for keeping state for the detection engine.
flow-timeout.h
DecodeTCPRegisterTests
void DecodeTCPRegisterTests(void)
Definition: decode-tcp.c:553
reputation.h
util-action.h
UtRunTests
uint32_t UtRunTests(const char *regex_arg)
Run all registered unittests.
Definition: util-unittest.c:190
util-mpm-ac.h
util-proto-name.h
defrag.h
MpmTableSetup
void MpmTableSetup(void)
Definition: util-mpm.c:269
MacSetRegisterTests
void MacSetRegisterTests(void)
Definition: util-macset.c:437
DEFAULT_PACKET_SIZE
#define DEFAULT_PACKET_SIZE
Definition: decode.h:623
BloomFilterRegisterTests
void BloomFilterRegisterTests(void)
Definition: util-bloomfilter.c:275
util-host-os-info.h
HostBitRegisterTests
void HostBitRegisterTests(void)
Definition: host-bit.c:508
default_packet_size
uint32_t default_packet_size
Definition: decode.c:71
DecodeVLANRegisterTests
void DecodeVLANRegisterTests(void)
Definition: decode-vlan.c:262
detect-fast-pattern.h
ConfYamlRegisterTests
void ConfYamlRegisterTests(void)
Definition: conf-yaml-loader.c:968
flow-manager.h
suricata-common.h
ConfRegisterTests
void ConfRegisterTests(void)
Definition: conf.c:1466
util-streaming-buffer.h
MimeDecRegisterTests
void MimeDecRegisterTests(void)
Definition: util-decode-mime.c:3147
util-spm.h
DecodePPPOERegisterTests
void DecodePPPOERegisterTests(void)
Registers PPPOE unit tests.
Definition: decode-pppoe.c:457
util-classification-config.h
FatalError
#define FatalError(x,...)
Definition: util-debug.h:532
SCClassConfInit
void SCClassConfInit(void)
Definition: util-classification-config.c:64
CIDRInit
void CIDRInit(void)
Definition: util-cidr.c:31
SCClassConfRegisterTests
void SCClassConfRegisterTests(void)
This function registers unit tests for Classification Config API.
Definition: util-classification-config.c:819
UtCleanup
void UtCleanup(void)
Cleanup unit test list.
Definition: util-unittest.c:269
detect-engine-sigorder.h
AppLayerProtoDetectUnittestsRegister
void AppLayerProtoDetectUnittestsRegister(void)
Register unittests for app layer proto detection module.
Definition: app-layer-detect-proto.c:3612
DecodeVXLANRegisterTests
void DecodeVXLANRegisterTests(void)
Definition: decode-vxlan.c:276
unix-manager.h
SCRConfRegisterTests
void SCRConfRegisterTests(void)
This function registers unit tests for Reference Config API.
Definition: util-reference-config.c:783
StreamTcpRegisterTests
void StreamTcpRegisterTests(void)
Definition: stream-tcp.c:10657
HashListTableRegisterTests
void HashListTableRegisterTests(void)
Definition: util-hashlist.c:501
max_pending_packets
int max_pending_packets
Definition: suricata.c:212
SC_ERR_FATAL
@ SC_ERR_FATAL
Definition: util-error.h:203
DecodeGRERegisterTests
void DecodeGRERegisterTests(void)
this function registers unit tests for GRE decoder
Definition: decode-gre.c:397
UtilMiscRegisterTests
void UtilMiscRegisterTests(void)
Definition: util-misc.c:1173
detect-parse.h
DecodeICMPV4RegisterTests
void DecodeICMPV4RegisterTests(void)
Registers ICMPV4 unit test.
Definition: decode-icmpv4.c:799
RunUnittests
void RunUnittests(int list_unittests, const char *regex_arg)
Definition: runmode-unittests.c:221
SCReputationRegisterTests
void SCReputationRegisterTests(void)
Definition: reputation.c:147
SCThresholdConfRegisterTests
void SCThresholdConfRegisterTests(void)
This function registers unit tests for Classification Config API.
Definition: util-threshold-config.c:2694
HostRegisterUnittests
void HostRegisterUnittests(void)
Definition: host.c:730
tmqh-flow.h
DecodeIPV4RegisterTests
void DecodeIPV4RegisterTests(void)
Definition: decode-ipv4.c:1643
util-mpm-hs.h
SCProfilingRegisterTests
void SCProfilingRegisterTests(void)
Definition: util-profiling.c:1449
IPPairBitRegisterTests
void IPPairBitRegisterTests(void)
Definition: ippair-bit.c:492
DetectEngineInspectModbusRegisterTests
void DetectEngineInspectModbusRegisterTests(void)
Definition: detect-engine-modbus.c:1444
app-layer-smtp.h
DetectPortTests
void DetectPortTests(void)
UtilActionRegisterTests
void UtilActionRegisterTests(void)
Definition: util-action.c:1596
util-misc.h
PacketPoolDestroy
void PacketPoolDestroy(void)
Definition: tmqh-packetpool.c:335
flow.h
ThreadMacrosRegisterTests
void ThreadMacrosRegisterTests(void)
this function registers unit tests for DetectId
Definition: threads.c:144
MemcmpRegisterTests
void MemcmpRegisterTests(void)
Definition: util-memcmp.c:384
DeStateRegisterTests
void DeStateRegisterTests(void)
Definition: detect-engine-state.c:1358
detect-engine-dcepayload.h
util-pool.h
DecodeGeneveRegisterTests
void DecodeGeneveRegisterTests(void)
Definition: decode-geneve.c:489
flow-var.h
DetectEngineRegisterTests
void DetectEngineRegisterTests()
Definition: detect-engine.c:4533
util-memrchr.h
app-layer-ssl.h
DecodeIPV6RegisterTests
void DecodeIPV6RegisterTests(void)
this function registers unit tests for IPV6 decoder
Definition: decode-ipv6.c:912
detect-engine-address.h
StatsRegisterTests
void StatsRegisterTests(void)
Definition: counters.c:1559
util-threshold-config.h
host-bit.h
app-layer.h
StreamingBufferRegisterTests
void StreamingBufferRegisterTests(void)
Definition: util-streaming-buffer.c:1477
PayloadRegisterTests
void PayloadRegisterTests(void)
Definition: detect-engine-payload.c:1317
MpmHSGlobalCleanup
void MpmHSGlobalCleanup(void)
DecodeRawRegisterTests
void DecodeRawRegisterTests(void)
Registers Raw unit tests.
Definition: decode-raw.c:230
DetectProtoTests
void DetectProtoTests(void)
this function registers unit tests for DetectProto
Definition: detect-engine-proto.c:579