mirror of
https://github.com/VirusTotal/yara.git
synced 2026-10-11 22:49:58 +00:00
tests: Add new fuzzers to be consumed by OSS-Fuzz (#2197)
Adds new fuzzing harnesses that increases the code coverage relative to a recent oss-fuzz coverage report: https://storage.googleapis.com/oss-fuzz-coverage/yara/reports/20260503/linux/src/report.html These harnesses will be consumed by OSS-Fuzz and start running once landed. Signed-off-by: David Korczynski <[email protected]>
This commit is contained in:
@@ -0,0 +1,86 @@
|
|||||||
|
/*
|
||||||
|
Copyright (c) 2026. The YARA Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
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 <stddef.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <sys/mman.h>
|
||||||
|
#include <fcntl.h>
|
||||||
|
|
||||||
|
extern "C" {
|
||||||
|
#include <yara.h>
|
||||||
|
#include <yara/filemap.h>
|
||||||
|
}
|
||||||
|
|
||||||
|
#include <fuzzer/FuzzedDataProvider.h>
|
||||||
|
|
||||||
|
extern "C" int LLVMFuzzerInitialize(int* argc, char*** argv)
|
||||||
|
{
|
||||||
|
yr_initialize();
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
|
||||||
|
{
|
||||||
|
char temp_file[] = "/tmp/yara-filemap-fuzz-XXXXXX";
|
||||||
|
int fd = mkstemp(temp_file);
|
||||||
|
if (fd < 0) return 0;
|
||||||
|
|
||||||
|
if (write(fd, data, size) != (ssize_t) size)
|
||||||
|
{
|
||||||
|
close(fd);
|
||||||
|
unlink(temp_file);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
YR_MAPPED_FILE mapped_file;
|
||||||
|
if (yr_filemap_map_fd(fd, 0, size, &mapped_file) == ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
yr_filemap_unmap_fd(&mapped_file);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Also try with some offsets and different sizes
|
||||||
|
FuzzedDataProvider fdp(data, size);
|
||||||
|
if (size > 0)
|
||||||
|
{
|
||||||
|
size_t offset = fdp.ConsumeIntegralInRange<size_t>(0, size - 1);
|
||||||
|
size_t map_size = fdp.ConsumeIntegralInRange<size_t>(0, size - offset);
|
||||||
|
if (yr_filemap_map_fd(fd, offset, map_size, &mapped_file) == ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
yr_filemap_unmap_fd(&mapped_file);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
close(fd);
|
||||||
|
unlink(temp_file);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
@@ -0,0 +1,88 @@
|
|||||||
|
/*
|
||||||
|
Copyright (c) 2026. The YARA Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
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 <stddef.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <yara.h>
|
||||||
|
|
||||||
|
const char* RULES = "import \"math\""
|
||||||
|
"rule test {"
|
||||||
|
" condition:"
|
||||||
|
" math.entropy(0, filesize) >= 0 and "
|
||||||
|
" math.mean(0, filesize) >= 0 and "
|
||||||
|
" math.deviation(0, filesize, 0.0) >= 0 and "
|
||||||
|
" math.serial_correlation(0, filesize) >= 0 and "
|
||||||
|
" math.monte_carlo_pi(0, filesize) >= 0 and "
|
||||||
|
" math.min(0, filesize) >= 0 and "
|
||||||
|
" math.max(0, filesize) >= 0 and "
|
||||||
|
" math.count(0x00, 0, filesize) >= 0 and "
|
||||||
|
" math.percentage(0x00, 0, filesize) >= 0"
|
||||||
|
"}";
|
||||||
|
|
||||||
|
YR_RULES* rules = NULL;
|
||||||
|
|
||||||
|
extern "C" int LLVMFuzzerInitialize(int* argc, char*** argv)
|
||||||
|
{
|
||||||
|
YR_COMPILER* compiler;
|
||||||
|
|
||||||
|
if (yr_initialize() != ERROR_SUCCESS)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
if (yr_compiler_create(&compiler) != ERROR_SUCCESS)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
if (yr_compiler_add_string(compiler, RULES, NULL) == 0)
|
||||||
|
yr_compiler_get_rules(compiler, &rules);
|
||||||
|
|
||||||
|
yr_compiler_destroy(compiler);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int callback(
|
||||||
|
YR_SCAN_CONTEXT* context,
|
||||||
|
int message,
|
||||||
|
void* message_data,
|
||||||
|
void* user_data)
|
||||||
|
{
|
||||||
|
return CALLBACK_CONTINUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
|
||||||
|
{
|
||||||
|
if (rules == NULL || size == 0)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
yr_rules_scan_mem(
|
||||||
|
rules, data, size, SCAN_FLAGS_NO_TRYCATCH, callback, NULL, 0);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
@@ -0,0 +1,55 @@
|
|||||||
|
/*
|
||||||
|
Copyright (c) 2026. The YARA Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
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 <stddef.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <tlshc/tlsh.h>
|
||||||
|
|
||||||
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
|
||||||
|
{
|
||||||
|
if (size < 50) // TLSH works better with more data
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
Tlsh* tlsh = tlsh_new();
|
||||||
|
if (!tlsh)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
// Split data into two parts to exercise update
|
||||||
|
size_t first_part = size / 2;
|
||||||
|
tlsh_update(tlsh, data, first_part);
|
||||||
|
tlsh_final(tlsh, data + first_part, size - first_part, 0);
|
||||||
|
|
||||||
|
tlsh_get_hash(tlsh, true);
|
||||||
|
tlsh_get_hash(tlsh, false);
|
||||||
|
|
||||||
|
tlsh_reset(tlsh);
|
||||||
|
tlsh_free(tlsh);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user