Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -89,11 +89,13 @@ def preprocess_yara_rules(repo_path: Path) -> Path:
|
|
| 89 |
if match:
|
| 90 |
original_name = match.group(1)
|
| 91 |
if original_name in seen_rules:
|
|
|
|
|
|
|
| 92 |
seen_rules[original_name] += 1
|
| 93 |
new_name = f"{original_name}_{seen_rules[original_name]}"
|
| 94 |
line = line.replace(original_name, new_name, 1)
|
| 95 |
-
rule_counter += 1
|
| 96 |
else:
|
|
|
|
| 97 |
seen_rules[original_name] = 0 # First occurrence
|
| 98 |
new_content.append(line)
|
| 99 |
|
|
@@ -107,8 +109,8 @@ def preprocess_yara_rules(repo_path: Path) -> Path:
|
|
| 107 |
def compile_yara_rules(repo_path: Path) -> Optional[yara.Rules]:
|
| 108 |
"""Compile YARA rules from repository with error handling"""
|
| 109 |
try:
|
| 110 |
-
|
| 111 |
-
yara_files = list(
|
| 112 |
|
| 113 |
if not yara_files:
|
| 114 |
logger.warning("No YARA files found in repository")
|
|
|
|
| 89 |
if match:
|
| 90 |
original_name = match.group(1)
|
| 91 |
if original_name in seen_rules:
|
| 92 |
+
logger.info(f"{original_name} is unseen")
|
| 93 |
+
rule_counter += 1
|
| 94 |
seen_rules[original_name] += 1
|
| 95 |
new_name = f"{original_name}_{seen_rules[original_name]}"
|
| 96 |
line = line.replace(original_name, new_name, 1)
|
|
|
|
| 97 |
else:
|
| 98 |
+
logger.warn(f"{original_name} is SEEN")
|
| 99 |
seen_rules[original_name] = 0 # First occurrence
|
| 100 |
new_content.append(line)
|
| 101 |
|
|
|
|
| 109 |
def compile_yara_rules(repo_path: Path) -> Optional[yara.Rules]:
|
| 110 |
"""Compile YARA rules from repository with error handling"""
|
| 111 |
try:
|
| 112 |
+
processed_dir = preprocess_yara_rules(repo_path)
|
| 113 |
+
yara_files = list(processed_dir.glob("**/*.yara"))
|
| 114 |
|
| 115 |
if not yara_files:
|
| 116 |
logger.warning("No YARA files found in repository")
|