CSEC_PUBLIC/securityonion
1
0
Fork 0
mirror of https://github.com/Security-Onion-Solutions/securityonion.git synced 2025-12-06 09:12:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

[wip] Fixing so-learn script

Browse Source
This commit is contained in:
William Wernert
2021-07-28 10:12:32 -04:00
parent e2abe8840f
commit 91accb0bc6
1 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
salt/common/tools/sbin/so-learn
View File

@@ -88,9 +88,13 @@ def write_pillar(pillar: str, content: dict):
def create_pillar_if_not_exist(pillar:str, content: dict): def create_pillar_if_not_exist(pillar:str, content: dict):
if content.get('learn', {}).get('modules') is None: if content.get('learn', {}).get('modules') is None:
content['learn'] = {}
content['learn']['modules'] = learn_modules content['learn']['modules'] = learn_modules
write_pillar(pillar, content) write_pillar(pillar, content)
content.update()
return content
def apply(module_list: List): def apply(module_list: List):
return_code = 0 return_code = 0
@@ -99,7 +103,7 @@ def apply(module_list: List):
print(f'Applying salt state for {module} module...') print(f'Applying salt state for {module} module...')
cmd = subprocess.run(salt_cmd, stdout=subprocess.DEVNULL) cmd = subprocess.run(salt_cmd, stdout=subprocess.DEVNULL)
if cmd.returncode != 0: if cmd.returncode != 0:
print(f'[ERROR] Failed to apply salt state for {module}') print(f'[ERROR] Failed to apply salt state for {module} module.')
return_code = cmd.returncode return_code = cmd.returncode
return return_code return return_code
@@ -135,13 +139,18 @@ def enable_disable_modules(args, enable: bool):
write_needed = False write_needed = False
for module in args.modules: for module in args.modules:
if module in pillar_modules: if module in pillar_modules:
pillar_modules[module]['enabled'] = enable if pillar_modules[module]['enabled'] == enable:
write_needed = enable action_str = 'enabled' if enable else 'disabled'
print(f'{module} module already {action_str}.', file=sys.stderr)
else:
pillar_modules[module]['enabled'] = enable
write_needed = enable
if write_needed: if write_needed:
args.pillar_dict.update() args.pillar_dict.update()
write_pillar(args.pillarm, args.pillar_dict) write_pillar(args.pillarm, args.pillar_dict)
check_apply(args) salt_proc = check_apply(args)
return salt_proc
def enable_modules(args): def enable_modules(args):
@@ -155,8 +164,8 @@ def disable_modules(args):
def list_modules(*_): def list_modules(*_):
print('Available ML modules:') print('Available ML modules:')
for module in learn_modules: for module in learn_modules:
print(f' - { module["name"] } : {module["description"]}') print(f' - { module["name"] } : {module["description"]}')
return 0
def main(): def main():
@@ -194,25 +203,18 @@ def main():
enable.add_argument('--apply', action='store_const', const=True, required=False, help=enable_apply_help) enable.add_argument('--apply', action='store_const', const=True, required=False, help=enable_apply_help)
enable.add_argument('--yes', '-y', action='store_const', const=True, required=False, help=yes_help) enable.add_argument('--yes', '-y', action='store_const', const=True, required=False, help=yes_help)
disable = subparsers.add_parser('disable') disable = subparsers.add_parser('disable')
disable.set_defaults(func=disable_modules) disable.set_defaults(func=disable_modules)
disable.add_argument('modules', metavar='ML_MODULES', nargs='+', help=module_help_str) disable.add_argument('modules', metavar='ML_MODULES', nargs='+', help=module_help_str)
disable.add_argument('--apply', action='store_const', const=True, required=False, help=disable_apply_help) disable.add_argument('--apply', action='store_const', const=True, required=False, help=disable_apply_help)
disable.add_argument('--yes', '-y', action='store_const', const=True, required=False, help=yes_help) disable.add_argument('--yes', '-y', action='store_const', const=True, required=False, help=yes_help)
list = subparsers.add_parser('list') list = subparsers.add_parser('list')
list.set_defaults(func=list_modules) list.set_defaults(func=list_modules)
args = main_parser.parse_args(sys.argv[1:]) args = main_parser.parse_args(sys.argv[1:])
args.pillar = find_minion_pillar() args.pillar = find_minion_pillar()
args.pillar_dict = create_pillar_if_not_exist(args.pillar, read_pillar(args.pillar))
args.pillar_dict = read_pillar(args.pillar)
create_pillar_if_not_exist(args.pillar, args.pillar_dict)
if hasattr(args, 'func'): if hasattr(args, 'func'):
exit_code = args.func(args) exit_code = args.func(args)