gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2025-12-07 05:10:36 -08:00
Code Issues Packages Projects Releases Wiki Activity

scripts: migrate rules: respect nursery status

Browse Source
This commit is contained in:
William Ballenthin
2020-06-22 07:40:55 -06:00
parent 11e456f55e
commit 45dea2b8aa
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
scripts/migrate-rules.py
View File

@@ -47,6 +47,9 @@ def read_rules(rule_directory):
rule = capa.rules.Rule.from_yaml_file(path) rule = capa.rules.Rule.from_yaml_file(path)
rules[rule.name] = rule rules[rule.name] = rule
if "nursery" in path:
rule.meta["nursery"] = True
return rules return rules
@@ -109,12 +112,6 @@ def main(argv=None):
rule.meta["mbc"] = [tag] rule.meta["mbc"] = [tag]
for rule in rules.values(): for rule in rules.values():
namespace = rule.meta.get("namespace")
if not namespace:
logger.info("%s has no proposed namespace, skipping", rule.name)
continue
filename = rule.name filename = rule.name
filename = filename.lower() filename = filename.lower()
filename = filename.replace(" ", "-") filename = filename.replace(" ", "-")
@@ -125,7 +122,12 @@ def main(argv=None):
filename = filename + ".yml" filename = filename + ".yml"
try: try:
directory = os.path.join(args.destination, namespace) if rule.meta.get("nursery"):
directory = os.path.join(args.destination, "nursery")
# this isn't meant to be written into the rule
del rule.meta["nursery"]
else:
directory = os.path.join(args.destination, rule.meta.get("namespace"))
os.makedirs(directory) os.makedirs(directory)
except OSError: except OSError:
pass pass