New module: awall: add new module for alpine wall (#24748)
* awall: add new module for alpine wall * awall: Fix is_policy_enabled bug when policy has description
This commit is contained in:
tdtrask
John R Barker
parent
330ad92425
commit
8c50105a6c
1 changed files with 163 additions and 0 deletions
163
lib/ansible/modules/system/awall.py
Normal file
163
lib/ansible/modules/system/awall.py
Normal file
|
|
@ -0,0 +1,163 @@
|
|||
#!/usr/bin/python
|
||||
# -*- coding: utf-8 -*-
|
||||
#
|
||||
# (c) 2017, Ted Trask <ttrask01@yahoo.com>
|
||||
#
|
||||
# This file is part of Ansible
|
||||
#
|
||||
# This module is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# This software is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this software. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
ANSIBLE_METADATA = {'metadata_version': '1.0',
|
||||
'status': ['stableinterface'],
|
||||
'supported_by': 'community'}
|
||||
|
||||
DOCUMENTATION = '''
|
||||
---
|
||||
module: awall
|
||||
short_description: Manage awall policies
|
||||
version_added: "2.4"
|
||||
author: Ted Trask (@tdtrask) <ttrask01@yahoo.com>
|
||||
description:
|
||||
- This modules allows for enable/disable/activate of I(awall) policies.
|
||||
Alpine Wall (I(awall)) generates a firewall configuration from the enabled policy files
|
||||
and activates the configuration on the system.
|
||||
options:
|
||||
name:
|
||||
description:
|
||||
- A policy name, like C(foo), or multiple policies, like C(foo, bar).
|
||||
default: null
|
||||
state:
|
||||
description:
|
||||
- The policy(ies) will be C(enabled)
|
||||
- The policy(ies) will be C(disabled)
|
||||
default: enabled
|
||||
choices: [ "enabled", "disabled" ]
|
||||
activate:
|
||||
description:
|
||||
- Activate the new firewall rules. Can be run with other steps or on it's own.
|
||||
default: False
|
||||
'''
|
||||
|
||||
EXAMPLES = '''
|
||||
- name: Enable "foo" and "bar" policy
|
||||
awall:
|
||||
name: foo,bar
|
||||
state: enabled
|
||||
|
||||
- name: Disable "foo" and "bar" policy and activate new rules
|
||||
awall:
|
||||
name: foo,bar
|
||||
state: disabled
|
||||
activate: False
|
||||
|
||||
- name: Activate currently enabled firewall rules
|
||||
awall:
|
||||
activate: True
|
||||
'''
|
||||
|
||||
RETURN = ''' # '''
|
||||
|
||||
import re
|
||||
from ansible.module_utils.basic import AnsibleModule
|
||||
|
||||
|
||||
def activate(module):
|
||||
cmd = "%s activate --force" % (AWALL_PATH)
|
||||
rc, stdout, stderr = module.run_command(cmd)
|
||||
if rc == 0:
|
||||
return True
|
||||
else:
|
||||
module.fail_json(msg="could not activate new rules", stdout=stdout, stderr=stderr)
|
||||
|
||||
|
||||
def is_policy_enabled(module, name):
|
||||
cmd = "%s list" % (AWALL_PATH)
|
||||
rc, stdout, stderr = module.run_command(cmd)
|
||||
if re.search("^%s\s+enabled" % name, stdout, re.MULTILINE):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def enable_policy(module, names, act):
|
||||
policies = []
|
||||
for name in names:
|
||||
if not is_policy_enabled(module, name):
|
||||
policies.append(name)
|
||||
if not policies:
|
||||
module.exit_json(changed=False, msg="policy(ies) already enabled")
|
||||
names = " ".join(policies)
|
||||
if module.check_mode:
|
||||
cmd = "%s list" % (AWALL_PATH)
|
||||
else:
|
||||
cmd = "%s enable %s" % (AWALL_PATH, names)
|
||||
rc, stdout, stderr = module.run_command(cmd)
|
||||
if rc != 0:
|
||||
module.fail_json(msg="failed to enable %s" % names, stdout=stdout, stderr=stderr)
|
||||
if act and not module.check_mode:
|
||||
activate(module)
|
||||
module.exit_json(changed=True, msg="enabled awall policy(ies): %s" % names)
|
||||
|
||||
|
||||
def disable_policy(module, names, act):
|
||||
policies = []
|
||||
for name in names:
|
||||
if is_policy_enabled(module, name):
|
||||
policies.append(name)
|
||||
if not policies:
|
||||
module.exit_json(changed=False, msg="policy(ies) already disabled")
|
||||
names = " ".join(policies)
|
||||
if module.check_mode:
|
||||
cmd = "%s list" % (AWALL_PATH)
|
||||
else:
|
||||
cmd = "%s disable %s" % (AWALL_PATH, names)
|
||||
rc, stdout, stderr = module.run_command(cmd)
|
||||
if rc != 0:
|
||||
module.fail_json(msg="failed to disable %s" % names, stdout=stdout, stderr=stderr)
|
||||
if act and not module.check_mode:
|
||||
activate(module)
|
||||
module.exit_json(changed=True, msg="disabled awall policy(ies): %s" % names)
|
||||
|
||||
|
||||
def main():
|
||||
module = AnsibleModule(
|
||||
argument_spec=dict(
|
||||
state=dict(default='enabled', choices=['enabled', 'disabled']),
|
||||
name=dict(type='list'),
|
||||
activate=dict(default=False, type='bool'),
|
||||
),
|
||||
required_one_of=[['name', 'activate']],
|
||||
supports_check_mode=True
|
||||
)
|
||||
|
||||
global AWALL_PATH
|
||||
AWALL_PATH = module.get_bin_path('awall', required=True)
|
||||
|
||||
p = module.params
|
||||
|
||||
if p['name']:
|
||||
if p['state'] == 'enabled':
|
||||
enable_policy(module, p['name'], p['activate'])
|
||||
elif p['state'] == 'disabled':
|
||||
disable_policy(module, p['name'], p['activate'])
|
||||
|
||||
if p['activate']:
|
||||
if not module.check_mode:
|
||||
activate(module)
|
||||
module.exit_json(changed=True, msg="activated awall rules")
|
||||
|
||||
module.fail_json(msg="no action defined")
|
||||
|
||||
# import module snippets
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Reference in a new issue