Fuzz only raw packets for .snoop

seladb · Nov 7, 2023 · ee61e76 · ee61e76
1 parent 4531107
commit ee61e76
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/Tests/Fuzzers/FuzzTarget.cpp b/Tests/Fuzzers/FuzzTarget.cpp
@@ -53,18 +53,22 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
 	pcpp::RawPacket& rawPacket = *packets.front();
 	do
 	{
-		pcpp::Packet parsedPacket(&rawPacket);
-		parsedPacket.toString();
-		auto layer = parsedPacket.getFirstLayer();
-		while (layer != NULL)
+		// go deeper only for .pcap and .pcapng format
+		// for .snoop we are only fuzzing the reader
+		if (0 == strcmp(FILE_EXT, ".pcap") || 0 == strcmp(FILE_EXT, ".pcapng"))
 		{
-			std::cout << layer->toString() << std::endl;
-			layer->getHeaderLen();
-			readParsedPacket(parsedPacket, layer);
-			layer = layer->getNextLayer();
+			pcpp::Packet parsedPacket(&rawPacket);
+			parsedPacket.toString();
+			auto layer = parsedPacket.getFirstLayer();
+			while (layer != NULL)
+			{
+				std::cout << layer->toString() << std::endl;
+				layer->getHeaderLen();
+				readParsedPacket(parsedPacket, layer);
+				layer = layer->getNextLayer();
+			}
+			parsedPacket.computeCalculateFields();
 		}
-		parsedPacket.computeCalculateFields();
-
 	} while (reader->getNextPacket(rawPacket));
 
 	reader->close();