mirror of
https://github.com/trailofbits/buttercup
synced 2026-06-21 14:11:39 +00:00
Evan Downing
32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
"""Fuzzy matching testing"""
|
|
|
|
import pytest
|
|
|
|
from buttercup.common.challenge_task import ChallengeTask
|
|
from buttercup.program_model.codequery import CodeQuery
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"search_string,match_string,threshold",
|
|
[
|
|
("png_read_chunk_header", "png_read_chunk_header", 100),
|
|
("OSS_FUZZ_png_read_chunk_header", "png_read_chunk_header", 80),
|
|
("OSS_FUZZ_png_reciprocal", "png_reciprocal", 75),
|
|
("png.c:png_image_free_function", "png_image_free_function", 80),
|
|
("_Z12default_freeP14png_struct_defPv", "default_free", 50),
|
|
("_Z14limited_mallocP14png_struct_defm", "limited_malloc", 50),
|
|
("_Z14user_read_dataP14png_struct_defPhm", "user_read_data", 50),
|
|
],
|
|
)
|
|
@pytest.mark.integration
|
|
def test_libpng_get_functions_fuzzy(
|
|
libpng_oss_fuzz_task: ChallengeTask, search_string, match_string, threshold
|
|
):
|
|
"""Test that we can get functions in challenge task code"""
|
|
codequery = CodeQuery(libpng_oss_fuzz_task)
|
|
functions = codequery.get_functions(
|
|
search_string, fuzzy=True, fuzzy_threshold=threshold
|
|
)
|
|
assert len(functions) > 0
|
|
assert any(f.name == match_string for f in functions)
|