Add: sigma rules (#175)
This commit is contained in:
38
tools/sigmac/splitter.py
Normal file
38
tools/sigmac/splitter.py
Normal file
@@ -0,0 +1,38 @@
|
||||
## pip install pyyaml
|
||||
|
||||
import os
|
||||
import ruamel.yaml
|
||||
|
||||
yaml = ruamel.yaml.YAML()
|
||||
|
||||
|
||||
def load_ymls( filepath ):
|
||||
with open(filepath) as f:
|
||||
return list(yaml.load_all(f))
|
||||
|
||||
def dump_yml( filepath, data ):
|
||||
with open(filepath, "w") as stream:
|
||||
yaml.dump(data, stream )
|
||||
|
||||
def main():
|
||||
loaded_ymls = load_ymls("sigma_to_hayabusa.yml")
|
||||
for loaded_yml in loaded_ymls:
|
||||
if loaded_yml == None:
|
||||
continue
|
||||
|
||||
if loaded_yml["yml_path"] == None or len(loaded_yml["yml_path"]) == 0:
|
||||
continue
|
||||
|
||||
out_dir = "hayabusa_rules/" + loaded_yml["yml_path"]
|
||||
out_path = out_dir + "/" + loaded_yml["yml_filename"]
|
||||
|
||||
if not os.path.exists(out_dir):
|
||||
os.makedirs(out_dir)
|
||||
|
||||
loaded_yml.pop("yml_path")
|
||||
loaded_yml.pop("yml_filename")
|
||||
|
||||
dump_yml(out_path,loaded_yml)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user