suricata
fuzz_sigpcap.c
Go to the documentation of this file.
1 /**
2  * @file
3  * @author Philippe Antoine <contact@catenacyber.fr>
4  * fuzz target for signature file and pcap file
5  */
6 
7 #include "suricata-common.h"
8 #include "source-pcap-file.h"
9 #include "detect-engine.h"
11 #include "util-reference-config.h"
12 #include "app-layer.h"
13 #include "tm-queuehandlers.h"
14 #include "util-cidr.h"
15 #include "util-profiling.h"
16 #include "util-proto-name.h"
17 #include "detect-engine-tag.h"
19 #include "host-bit.h"
20 #include "ippair-bit.h"
21 #include "app-layer-htp.h"
22 #include "detect-fast-pattern.h"
23 #include "util-unittest-helper.h"
24 #include "conf-yaml-loader.h"
25 #include "pkt-var.h"
26 #include "flow-util.h"
27 #include "flow-worker.h"
28 #include "tm-modules.h"
29 #include "tmqh-packetpool.h"
30 #include "util-file.h"
31 #include "util-conf.h"
32 #include "packet.h"
33 
34 int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size);
35 
36 
37 static int initialized = 0;
40 //FlowWorkerThreadData
41 void *fwd;
43 SC_ATOMIC_EXTERN(unsigned int, engine_stage);
44 
45 #include "confyaml.c"
46 
47 int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
48 {
49  pcap_t * pkts;
50  char errbuf[PCAP_ERRBUF_SIZE];
51  const u_char *pkt;
52  struct pcap_pkthdr *header;
53  int r;
54  Packet *p;
55  size_t pos;
56  size_t pcap_cnt = 0;
57 
58  if (initialized == 0) {
59  //Redirects logs to /dev/null
60  setenv("SC_LOG_OP_IFACE", "file", 0);
61  setenv("SC_LOG_FILE", "/dev/null", 0);
62 
63  InitGlobal();
64 
67  //redirect logs to /tmp
68  ConfigSetLogDirectory("/tmp/");
69  //disables checksums validation for fuzzing
71  abort();
72  }
73  // do not load rules before reproducible DetectEngineReload
74  remove("/tmp/fuzz.rules");
75  surifuzz.sig_file = strdup("/tmp/fuzz.rules");
77  //loads rules after init
79 
83 
84  memset(&tv, 0, sizeof(tv));
86  if (tv.flow_queue == NULL)
87  abort();
92 
93  extern uint16_t max_pending_packets;
94  max_pending_packets = 128;
96  SC_ATOMIC_SET(engine_stage, SURICATA_RUNTIME);
97  initialized = 1;
98  }
99 
100  /* TODO add yaml config
101  for (pos = 0; pos < size; pos++) {
102  if (data[pos] == 0) {
103  break;
104  }
105  }
106  if (ConfYamlLoadString(data, pos) != 0) {
107  return 0;
108  }
109  if (pos < size) {
110  //skip zero
111  pos++;
112  }
113  data += pos;
114  size -= pos;*/
115 
116  for (pos=0; pos < size; pos++) {
117  if (data[pos] == 0) {
118  break;
119  }
120  }
121  if (pos > 0 && pos < size) {
122  // dump signatures to a file so as to reuse SigLoadSignatures
123  if (TestHelperBufferToFile(surifuzz.sig_file, data, pos-1) < 0) {
124  return 0;
125  }
126  } else {
127  if (TestHelperBufferToFile(surifuzz.sig_file, data, pos) < 0) {
128  return 0;
129  }
130  }
131 
132  if (DetectEngineReload(&surifuzz) < 0) {
133  return 0;
134  }
136 
138  de_ctx->ref_cnt--;
140  FlowWorkerReplaceDetectCtx(fwd, new_det_ctx);
141 
142  DetectEngineThreadCtxDeinit(NULL, old_det_ctx);
143 
144  if (pos < size) {
145  //skip zero
146  pos++;
147  }
148  data += pos;
149  size -= pos;
150 
151  //rewrite buffer to a file as libpcap does not have buffer inputs
152  if (TestHelperBufferToFile("/tmp/fuzz.pcap", data, size) < 0) {
153  return 0;
154  }
155 
156  //initialize structure
157  pkts = pcap_open_offline("/tmp/fuzz.pcap", errbuf);
158  if (pkts == NULL) {
159  return 0;
160  }
161 
162  //loop over packets
163  r = pcap_next_ex(pkts, &header, &pkt);
164  p = PacketGetFromAlloc();
165  if (r <= 0 || header->ts.tv_sec >= INT_MAX - 3600 || header->ts.tv_usec < 0) {
166  goto bail;
167  }
168  p->ts = SCTIME_FROM_TIMEVAL(&header->ts);
169  p->datalink = pcap_datalink(pkts);
170  p->pkt_src = PKT_SRC_WIRE;
171  while (r > 0) {
172  if (PacketCopyData(p, pkt, header->caplen) == 0) {
173  // DecodePcapFile
175  if (ecode == TM_ECODE_FAILED) {
176  break;
177  }
178  Packet *extra_p = PacketDequeueNoLock(&tv.decode_pq);
179  while (extra_p != NULL) {
180  PacketFreeOrRelease(extra_p);
181  extra_p = PacketDequeueNoLock(&tv.decode_pq);
182  }
184  extra_p = PacketDequeueNoLock(&tv.decode_pq);
185  while (extra_p != NULL) {
186  PacketFreeOrRelease(extra_p);
187  extra_p = PacketDequeueNoLock(&tv.decode_pq);
188  }
189  }
190  r = pcap_next_ex(pkts, &header, &pkt);
191  if (r <= 0 || header->ts.tv_sec >= INT_MAX - 3600 || header->ts.tv_usec < 0) {
192  goto bail;
193  }
194  PacketRecycle(p);
195  p->ts = SCTIME_FROM_TIMEVAL(&header->ts);
196  p->datalink = pcap_datalink(pkts);
197  p->pkt_src = PKT_SRC_WIRE;
198  pcap_cnt++;
199  p->pcap_cnt = pcap_cnt;
200  }
201 bail:
202  //close structure
203  pcap_close(pkts);
204  PacketFree(p);
205  FlowReset();
206 
207  return 0;
208 }
ThreadVars_::flow_queue
struct FlowQueue_ * flow_queue
Definition: threadvars.h:134
ts
uint64_t ts
Definition: source-erf-file.c:55
detect-engine.h
DetectEngineThreadCtxInitForReload
DetectEngineThreadCtx * DetectEngineThreadCtxInitForReload(ThreadVars *tv, DetectEngineCtx *new_de_ctx, int mt)
Definition: detect-engine.c:3355
PacketFreeOrRelease
void PacketFreeOrRelease(Packet *p)
Return a packet to where it was allocated.
Definition: decode.c:191
flow-util.h
PacketCopyData
int PacketCopyData(Packet *p, const uint8_t *pktdata, uint32_t pktlen)
Copy data to Packet payload and set packet length.
Definition: decode.c:292
source-pcap-file.h
tv
ThreadVars tv
Definition: fuzz_sigpcap.c:38
fwd
void * fwd
Definition: fuzz_sigpcap.c:41
SC_ATOMIC_SET
#define SC_ATOMIC_SET(name, val)
Set the value for the atomic variable.
Definition: util-atomic.h:386
DetectEngineCtx_::ref_cnt
uint32_t ref_cnt
Definition: detect.h:964
Packet_::pcap_cnt
uint64_t pcap_cnt
Definition: decode.h:607
ippair-bit.h
PacketRecycle
void PacketRecycle(Packet *p)
Definition: packet.c:170
LLVMFuzzerTestOneInput
int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
Definition: fuzz_sigpcap.c:47
DetectEngineCtx_
main detection engine ctx
Definition: detect.h:836
DetectEngineGetCurrent
DetectEngineCtx * DetectEngineGetCurrent(void)
Definition: detect-engine.c:3743
FlowReset
void FlowReset(void)
Definition: flow.c:679
StatsSetupPrivate
int StatsSetupPrivate(ThreadVars *tv)
Definition: counters.c:1224
SC_ATOMIC_EXTERN
SC_ATOMIC_EXTERN(unsigned int, engine_stage)
tm-modules.h
GlobalsInitPreConfig
void GlobalsInitPreConfig(void)
Definition: suricata.c:357
SURICATA_RUNTIME
@ SURICATA_RUNTIME
Definition: suricata.h:96
TM_ECODE_FAILED
@ TM_ECODE_FAILED
Definition: tm-threads-common.h:85
tmqh-packetpool.h
util-unittest-helper.h
dtv
DecodeThreadVars * dtv
Definition: fuzz_sigpcap.c:39
PacketPoolInit
void PacketPoolInit(void)
Definition: tmqh-packetpool.c:262
Packet_::datalink
int datalink
Definition: decode.h:620
DecodeRegisterPerfCounters
void DecodeRegisterPerfCounters(DecodeThreadVars *dtv, ThreadVars *tv)
Definition: decode.c:528
util-cidr.h
app-layer-htp.h
PreRunPostPrivsDropInit
void PreRunPostPrivsDropInit(const int runmode)
Definition: suricata.c:2248
PKT_SRC_WIRE
@ PKT_SRC_WIRE
Definition: decode.h:54
de_ctx
DetectEngineCtx * de_ctx
Definition: fuzz_siginit.c:17
DetectEngineThreadCtx_
Definition: detect.h:1092
Packet_::ts
SCTime_t ts
Definition: decode.h:484
PacketDequeueNoLock
Packet * PacketDequeueNoLock(PacketQueueNoLock *qnl)
Definition: packet-queue.c:208
flow-worker.h
util-reference-config.h
SCInstance_::delayed_detect
int delayed_detect
Definition: suricata.h:149
ThreadVars_
Per thread variable structure.
Definition: threadvars.h:57
SCTIME_FROM_TIMEVAL
#define SCTIME_FROM_TIMEVAL(tv)
Definition: util-time.h:79
pkt-var.h
TmModule_::Func
TmEcode(* Func)(ThreadVars *, Packet *, void *)
Definition: tm-modules.h:53
PacketFree
void PacketFree(Packet *p)
Return a malloced packet.
Definition: decode.c:134
FlowQueueNew
FlowQueue * FlowQueueNew(void)
Definition: flow-queue.c:35
TestHelperBufferToFile
int TestHelperBufferToFile(const char *name, const uint8_t *data, size_t size)
writes the contents of a buffer into a file
Definition: util-unittest-helper.c:103
ConfYamlLoadString
int ConfYamlLoadString(const char *string, size_t len)
Load configuration from a YAML string.
Definition: conf-yaml-loader.c:511
PostConfLoadedSetup
int PostConfLoadedSetup(SCInstance *suri)
Definition: suricata.c:2633
detect-engine-tag.h
util-profiling.h
Packet_
Definition: decode.h:436
PostConfLoadedDetectSetup
void PostConfLoadedDetectSetup(SCInstance *suri)
Definition: suricata.c:2542
tmm_modules
TmModule tmm_modules[TMM_SIZE]
Definition: tm-modules.c:33
conf-yaml-loader.h
TMM_DECODEPCAPFILE
@ TMM_DECODEPCAPFILE
Definition: tm-threads-common.h:41
confyaml.c
TmEcode
TmEcode
Definition: tm-threads-common.h:83
max_pending_packets
uint16_t max_pending_packets
Definition: suricata.c:187
util-proto-name.h
setenv
void setenv(const char *name, const char *value, int overwrite)
TMM_FLOWWORKER
@ TMM_FLOWWORKER
Definition: tm-threads-common.h:34
tm-queuehandlers.h
util-file.h
FlowWorkerGetDetectCtxPtr
void * FlowWorkerGetDetectCtxPtr(void *flow_worker)
Definition: flow-worker.c:691
detect-fast-pattern.h
util-conf.h
suricata-common.h
DetectEngineThreadCtxDeinit
TmEcode DetectEngineThreadCtxDeinit(ThreadVars *, void *)
Definition: detect-engine.c:3501
configNoChecksum
const char configNoChecksum[]
Definition: confyaml.c:1
run_mode
int run_mode
Definition: suricata.c:176
packet.h
TmModule_::ThreadInit
TmEcode(* ThreadInit)(ThreadVars *, const void *, void **)
Definition: tm-modules.h:48
util-classification-config.h
SCInstance_::sig_file
char * sig_file
Definition: suricata.h:128
ConfigSetLogDirectory
TmEcode ConfigSetLogDirectory(const char *name)
Definition: util-conf.c:33
PacketGetFromAlloc
Packet * PacketGetFromAlloc(void)
Get a malloced packet.
Definition: decode.c:173
Packet_::pkt_src
uint8_t pkt_src
Definition: decode.h:592
DecodeThreadVars_
Structure to hold thread specific data for all decode modules.
Definition: decode.h:685
DecodeThreadVarsAlloc
DecodeThreadVars * DecodeThreadVarsAlloc(ThreadVars *tv)
Alloc and setup DecodeThreadVars.
Definition: decode.c:691
ThreadVars_::decode_pq
PacketQueueNoLock decode_pq
Definition: threadvars.h:111
DetectEngineReload
int DetectEngineReload(const SCInstance *suri)
Reload the detection engine.
Definition: detect-engine.c:4682
SCInstance_
Definition: suricata.h:123
InitGlobal
int InitGlobal(void)
Global initialization common to all runmodes.
Definition: suricata.c:2858
surifuzz
SCInstance surifuzz
Definition: fuzz_sigpcap.c:42
SCInstance_::sig_file_exclusive
bool sig_file_exclusive
Definition: suricata.h:129
RUNMODE_PCAP_FILE
@ RUNMODE_PCAP_FILE
Definition: runmodes.h:30
host-bit.h
detect-engine-threshold.h
FlowWorkerReplaceDetectCtx
void FlowWorkerReplaceDetectCtx(void *flow_worker, void *detect_ctx)
Definition: flow-worker.c:684
app-layer.h