Spaces:
Sleeping
Sleeping
Change yara source
Browse files
app.py
CHANGED
|
@@ -28,7 +28,8 @@ MAL_CLASSES = ['Adialer.C', 'Agent.FYI', 'Allaple.A', 'Allaple.L', 'Alueron.gen!
|
|
| 28 |
|
| 29 |
UPLOAD_DIR = "uploads"
|
| 30 |
os.makedirs(UPLOAD_DIR, exist_ok=True)
|
| 31 |
-
YARA_REPO_URL = "https://github.com/
|
|
|
|
| 32 |
YARA_REPO_DIR = "defenderyara"
|
| 33 |
|
| 34 |
# Environment configuration
|
|
@@ -47,7 +48,7 @@ async def clone_yara_repo():
|
|
| 47 |
|
| 48 |
logger.info(f"Cloning YARA rules from {YARA_REPO_URL}")
|
| 49 |
proc = await asyncio.create_subprocess_exec(
|
| 50 |
-
'git', 'clone', YARA_REPO_URL, str(repo_path),
|
| 51 |
stdout=asyncio.subprocess.PIPE,
|
| 52 |
stderr=asyncio.subprocess.PIPE
|
| 53 |
)
|
|
|
|
| 28 |
|
| 29 |
UPLOAD_DIR = "uploads"
|
| 30 |
os.makedirs(UPLOAD_DIR, exist_ok=True)
|
| 31 |
+
YARA_REPO_URL = "https://github.com/t-tani/defender2yara.git"
|
| 32 |
+
YARA_REPO_BRANCH = "yara-rules"
|
| 33 |
YARA_REPO_DIR = "defenderyara"
|
| 34 |
|
| 35 |
# Environment configuration
|
|
|
|
| 48 |
|
| 49 |
logger.info(f"Cloning YARA rules from {YARA_REPO_URL}")
|
| 50 |
proc = await asyncio.create_subprocess_exec(
|
| 51 |
+
'git', 'clone', '-b', YARA_REPO_BRANCH, '--single-branch', YARA_REPO_URL, str(repo_path),
|
| 52 |
stdout=asyncio.subprocess.PIPE,
|
| 53 |
stderr=asyncio.subprocess.PIPE
|
| 54 |
)
|