vee1e/flare-floss
1
0
Fork 0
mirror of https://github.com/vee1e/flare-floss.git synced 2026-09-01 17:57:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding tests to ensure >88% string extraction from Unknown Rust binaries #852 (#858)

Browse source 
* Adding tests to ensure >88% string extraction from Unknown go binaries #852
This commit is contained in:
Arnav Kharbanda 2023-08-24 11:12:58 +05:30 • committed by GitHub
parent 2ee4cf5a90
commit 692ce71267
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 58 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/data

@ -1 +1 @@
Subproject commit 8786fff21f11e2006b8e771822eeff630be2f77a
Subproject commit 6aa7328a43e66ff4fa5d6d3b84fde340aec46b62

57
tests/test_language_rust_coverage.py Normal file
View file

@ -0,0 +1,57 @@
import pathlib
import contextlib
import pefile
import pytest
from floss.strings import extract_ascii_unicode_strings
from floss.language.utils import get_extract_stats
from floss.language.rust.extract import extract_rust_strings
@pytest.mark.parametrize(
"binary_file",
[
(
"data/language/rust/rust-unknown-binaries/bin/1.59.0/i386/bf7362a9a5e94d93d5f495ac2535779708f2f09bf0729382aba0f7f64f42f36a"
),
(
"data/language/rust/rust-unknown-binaries/bin/1.64.0/amd64/e37b08d35b237961c2d5a94a5ced3919616037b3e2a73efa77bf992c5335fbf6"
),
(
"data/language/rust/rust-unknown-binaries/bin/1.65.0/amd64/635d89076c3c68520ae7927196c5b9448cb783f4ac0ee0a552d3bb60e899caba"
),
(
"data/language/rust/rust-unknown-binaries/bin/1.68.1/amd64/07e00bbedff9a4aee59056c629a6ac67a34d6f8b8f0082f98d14f0f80ee037a4"
),
(
"data/language/rust/rust-unknown-binaries/bin/1.69.0/amd64/b76d3f6327b9e680c491289ecd38f0a8b2fc7a7ba458e5532d80a78d89af0184"
),
(
"data/language/rust/rust-unknown-binaries/bin/1.69.0/i386/200c308a793630e4f3686dd846f0d55b6368834a859875970b4135f3ca487f46"
),
],
)
def test_language_detection_64(binary_file):
CD = pathlib.Path(__file__).resolve().parent
abs_path = (CD / binary_file).resolve()
assert abs_path.exists(), f"File {binary_file} does not exist"
n = 4
path = pathlib.Path(abs_path)
buf = path.read_bytes()
pe = pefile.PE(data=buf, fast_load=True)
all_ss_strings = list(extract_ascii_unicode_strings(buf, 4))
rust_strings = extract_rust_strings(path, n)
# do not print the output of the function
with contextlib.redirect_stdout(None):
out = get_extract_stats(pe, all_ss_strings, rust_strings, n)
# check that the output percentage is greater than 88%
assert float(out) > 88