(ids) write yaml file for installed rules

src/opnsense/scripts/suricata/installRules.py

@@ -38,6 +38,7 @@ RuleCache = rulecache.RuleCache()
 
 rule_config_fn = ('%s../rules.config'%RuleCache.rule_source_dir)
 rule_target_dir = ('%s../opnsense.rules'%RuleCache.rule_source_dir)
+rule_yaml_list = ('%s../installed_rules.yaml'%RuleCache.rule_source_dir)
 
 # parse OPNsense rule config
 rule_updates = {}
@@ -56,6 +57,7 @@ if not os.path.exists(rule_target_dir):
     os.mkdir(rule_target_dir, 0o755)
 
 # install ruleset
+all_installed_files = []
 for filename in RuleCache.listLocal():
     output_data = []
     for rule_info_record in RuleCache.listRules(filename=filename):
@@ -79,4 +81,13 @@ for filename in RuleCache.listLocal():
         output_data.append(rule)
 
     # write data to file
+    all_installed_files.append(filename.split('/')[-1])
     open('%s/%s'%(rule_target_dir, filename.split('/')[-1]), 'wb').write('\n'.join(output_data))
+
+# flush all written rule filenames into yaml file
+with open(rule_yaml_list,'wb') as f_out:
+    f_out.write('%YAML 1.1\n')
+    f_out.write('---\n')
+    f_out.write('rule-files:\n')
+    for installed_file in all_installed_files:
+        f_out.write(' - %s\n'%installed_file)