Skip to content

Commit c5e3118

Browse files
author
ashariyar
committed
remove unnecessary _custom_yara_rules_file_basenames(), issue with default rules ending up in filename was resolved by not doing 'rules_paths +=' in yaralyzer_helper.py a while ago
1 parent f71e8dd commit c5e3118

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

pdfalyzer/config.py

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,8 @@ def get_export_basepath(cls, export_method: Callable) -> str:
5959
export_basename += '_noBOMs'
6060

6161
# YARA rules suffixes
62-
if cls._custom_yara_rules_file_basenames():
63-
export_basename += f"__scannedby_" + ','.join(cls._custom_yara_rules_file_basenames())
62+
if cls.args.yara_rules_files:
63+
export_basename += f"__scannedby_" + ','.join(sorted([f.name for f in cls.args.yara_rules_files]))
6464

6565
if cls.args.no_default_yara_rules:
6666
export_basename += '_customrulesonly'
@@ -101,15 +101,6 @@ def prefixed_env_var(cls, var: str) -> str:
101101
prefix = super().ENV_VAR_PREFIX if var in YARALYZER_SPECIFIC_OPTIONS else cls.ENV_VAR_PREFIX
102102
return (var if var.startswith(prefix) else f"{prefix}_{var}").upper()
103103

104-
@classmethod
105-
def _custom_yara_rules_file_basenames(cls) -> list[str]:
106-
"""Returns yara rules files requested by -Y option only (excludes included `YARA_RULES_FILES`)."""
107-
# TODO: YaralyzerConfig is updating the same ._args this class uses when _build-yaralyzer() is called (i think)
108-
# so this class's ._args.yara_rules_files ends up with all the defaults PDF yara rules.
109-
yara_rules_files = cls.args.yara_rules_files or []
110-
yara_rules_basenames = [Path(f).name for f in yara_rules_files if not Path(f).name in YARA_RULES_FILES]
111-
return sorted(yara_rules_basenames)
112-
113104
@classmethod
114105
def _set_class_vars_from_env(cls) -> None:
115106
"""Set log related class vars and find path to pdf-parser.py (if any)."""

0 commit comments

Comments
 (0)