clamav/fuzz/clamav_dbload_fuzzer.cpp
micasnyd 0037f5825b fuzz: fix cl_load fuzz target
I found two issues with the cl_load fuzz targets, one of which impacts
the scanfile and scanmap fuzz targets:

1. We were defining the preprocessor definitions incorrectly using
  "SCAN_TARGETS" instead of "TARGET" in unit_tests/CMakeLists.txt.
  For the scan fuzz targets this meant it wasn't properly defining
  unique settings for each compiled target.
  For the cl_load fuzz target it's worse, it wasn't setting the
  database file name correctly which means it rejected the filenames
  entirely for not having a legitimate suffix.

2. We were pre-compiling the engine before loading signatures.
  You can't load sigs for an engine that's already compiled, so this
  would also fail right away without trying to load any sigs.
2022-01-18 16:09:44 -07:00

151 lines
4.8 KiB
C++

/*
* Fuzz target for cl_load()
*
* Copyright (C) 2018-2022 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
* Authors: Micah Snyder
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* 3. Neither the name of the copyright holder nor the names of its contributors
* may be used to endorse or promote products derived from this software without
* specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#include <stdint.h>
#include <stdlib.h>
#include <unistd.h>
#include "clamav.h"
void clamav_message_callback(enum cl_msg severity, const char* fullmsg,
const char* msg, void* context)
{
}
class ClamAVState
{
public:
ClamAVState()
{
// Silence all the log messages, none of them are meaningful.
cl_set_clcb_msg(clamav_message_callback);
cl_init(CL_INIT_DEFAULT);
dboptions =
CL_DB_PHISHING | CL_DB_PHISHING_URLS |
CL_DB_BYTECODE | CL_DB_PUA | CL_DB_ENHANCED;
#if defined(CLAMAV_FUZZ_CDB)
tmp_db_name = "dbload_tmp_fuzz.cdb";
#elif defined(CLAMAV_FUZZ_CFG)
tmp_db_name = "dbload_tmp_fuzz.cfg";
#elif defined(CLAMAV_FUZZ_CRB)
tmp_db_name = "dbload_tmp_fuzz.crb";
#elif defined(CLAMAV_FUZZ_FP)
tmp_db_name = "dbload_tmp_fuzz.fp";
#elif defined(CLAMAV_FUZZ_FTM)
tmp_db_name = "dbload_tmp_fuzz.ftm";
#elif defined(CLAMAV_FUZZ_HDB)
tmp_db_name = "dbload_tmp_fuzz.hdb";
#elif defined(CLAMAV_FUZZ_HSB)
tmp_db_name = "dbload_tmp_fuzz.hsb";
#elif defined(CLAMAV_FUZZ_IDB)
tmp_db_name = "dbload_tmp_fuzz.idb";
#elif defined(CLAMAV_FUZZ_IGN)
tmp_db_name = "dbload_tmp_fuzz.ign";
#elif defined(CLAMAV_FUZZ_IGN2)
tmp_db_name = "dbload_tmp_fuzz.ign2";
#elif defined(CLAMAV_FUZZ_LDB)
tmp_db_name = "dbload_tmp_fuzz.ldb";
#elif defined(CLAMAV_FUZZ_MDB)
tmp_db_name = "dbload_tmp_fuzz.mdb";
#elif defined(CLAMAV_FUZZ_MSB)
tmp_db_name = "dbload_tmp_fuzz.msb";
#elif defined(CLAMAV_FUZZ_NDB)
tmp_db_name = "dbload_tmp_fuzz.ndb";
#elif defined(CLAMAV_FUZZ_PDB)
tmp_db_name = "dbload_tmp_fuzz.pdb";
#elif defined(CLAMAV_FUZZ_WDB)
tmp_db_name = "dbload_tmp_fuzz.wdb";
#elif defined(CLAMAV_FUZZ_YARA)
tmp_db_name = "dbload_tmp_fuzz.yara";
#else
tmp_db_name = "dbload_tmp_fuzz";
#endif
}
~ClamAVState()
{
if (NULL != tmp_db_name) {
unlink(tmp_db_name);
}
}
const char* tmp_db_name;
unsigned int dboptions;
};
// Global with static initializer to setup an engine so we don't need to do
// that on each execution.
ClamAVState kClamAVState;
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
{
cl_error_t ret;
unsigned int sigs = 0;
FILE* fuzzdb = NULL;
struct cl_engine* engine = NULL;
fuzzdb = fopen(kClamAVState.tmp_db_name, "w");
fwrite(data, size, 1, fuzzdb);
fclose(fuzzdb);
/* need new engine each time. can't add sigs to compiled engine */
engine = cl_engine_new();
/* load the fuzzer-generated sig db */
if (CL_SUCCESS != (ret = cl_load(
kClamAVState.tmp_db_name,
engine,
&sigs,
kClamAVState.dboptions))) {
printf("cl_load: %s\n", cl_strerror(ret));
goto done;
}
/* build engine */
if (CL_SUCCESS != (ret = cl_engine_compile(engine))) {
printf("cl_engine_compile: %s\n", cl_strerror(ret));
goto done;
}
done:
/* Clean up for the next round */
if (NULL != engine) {
cl_engine_free(engine);
}
return 0;
}