mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-08-24 00:42:22 +02:00
1077 lines
36 KiB
Python
1077 lines
36 KiB
Python
![]() |
#!/usr/bin/python
|
||
|
# -*- coding: utf-8 -*-
|
||
|
#
|
||
|
# Copyright: (c) 2017, F5 Networks Inc.
|
||
|
# GNU General Public License v3.0 (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||
|
|
||
|
from __future__ import absolute_import, division, print_function
|
||
|
__metaclass__ = type
|
||
|
|
||
|
|
||
|
ANSIBLE_METADATA = {'metadata_version': '1.1',
|
||
|
'status': ['preview'],
|
||
|
'supported_by': 'certified'}
|
||
|
|
||
|
DOCUMENTATION = r'''
|
||
|
---
|
||
|
module: bigip_data_group
|
||
|
short_description: Manage data groups on a BIG-IP
|
||
|
description:
|
||
|
- Allows for managing data groups on a BIG-IP. Data groups provide a way to store collections
|
||
|
of values on a BIG-IP for later use in things such as LTM rules, iRules, and ASM policies.
|
||
|
version_added: 2.6
|
||
|
options:
|
||
|
name:
|
||
|
description:
|
||
|
- Specifies the name of the data group.
|
||
|
required: True
|
||
|
type:
|
||
|
description:
|
||
|
- The type of records in this data group.
|
||
|
- This parameter is especially important because it causes BIG-IP to store your data
|
||
|
in different ways so-as to optimize access to it. For example, it would be wrong
|
||
|
to specify a list of records containing IP addresses, but label them as a C(string)
|
||
|
type.
|
||
|
- This value cannot be changed once the data group is created.
|
||
|
choices:
|
||
|
- address
|
||
|
- addr
|
||
|
- ip
|
||
|
- string
|
||
|
- str
|
||
|
- integer
|
||
|
- int
|
||
|
default: string
|
||
|
internal:
|
||
|
description:
|
||
|
- The type of this data group.
|
||
|
- You should only consider setting this value in cases where you know exactly what
|
||
|
you're doing, B(or), you are working with a pre-existing internal data group.
|
||
|
- Be aware that if you deliberately force this parameter to C(yes), and you have a
|
||
|
either a large number of records or a large total records size, this large amount
|
||
|
of data will be reflected in your BIG-IP configuration. This can lead to B(long)
|
||
|
system configuration load times due to needing to parse and verify the large
|
||
|
configuration.
|
||
|
- There is a limit of either 4 megabytes or 65,000 records (whichever is more restrictive)
|
||
|
for uploads when this parameter is C(yes).
|
||
|
- This value cannot be changed once the data group is created.
|
||
|
type: bool
|
||
|
default: no
|
||
|
external_file_name:
|
||
|
description:
|
||
|
- When creating a new data group, this specifies the file name that you want to give an
|
||
|
external data group file on the BIG-IP.
|
||
|
- This parameter is ignored when C(internal) is C(yes).
|
||
|
- This parameter can be used to select an existing data group file to use with an
|
||
|
existing external data group.
|
||
|
- If this value is not provided, it will be given the value specified in C(name) and,
|
||
|
therefore, match the name of the data group.
|
||
|
- This value may only contain letters, numbers, underscores, dashes, or a period.
|
||
|
records:
|
||
|
description:
|
||
|
- Specifies the records that you want to add to a data group.
|
||
|
- If you have a large number of records, it is recommended that you use C(records_content)
|
||
|
instead of typing all those records here.
|
||
|
- The technical limit of either 1. the number of records, or 2. the total size of all
|
||
|
records, varies with the size of the total resources on your system; in particular,
|
||
|
RAM.
|
||
|
- When C(internal) is C(no), at least one record must be specified in either C(records)
|
||
|
or C(records_content).
|
||
|
suboptions:
|
||
|
key:
|
||
|
description:
|
||
|
- The key describing the record in the data group.
|
||
|
- Your key will be used for validation of the C(type) parameter to this module.
|
||
|
required: True
|
||
|
value:
|
||
|
description:
|
||
|
- The value of the key describing the record in the data group.
|
||
|
records_src:
|
||
|
description:
|
||
|
- Path to a file with records in it.
|
||
|
- The file should be well-formed. This means that it includes records, one per line,
|
||
|
that resemble the following format "key separator value". For example, C(foo := bar).
|
||
|
- BIG-IP is strict about this format, but this module is a bit more lax. It will allow
|
||
|
you to include arbitrary amounts (including none) of empty space on either side of
|
||
|
the separator. For an illustration of this, see the Examples section.
|
||
|
- Record keys are limited in length to no more than 65520 characters.
|
||
|
- Values of record keys are limited in length to no more than 65520 characters.
|
||
|
- The total number of records you can have in your BIG-IP is limited by the memory
|
||
|
of the BIG-IP.
|
||
|
- The format of this content is slightly different depending on whether you specify
|
||
|
a C(type) of C(address), C(integer), or C(string). See the examples section for
|
||
|
examples of the different types of payload formats that are expected in your data
|
||
|
group file.
|
||
|
- When C(internal) is C(no), at least one record must be specified in either C(records)
|
||
|
or C(records_content).
|
||
|
separator:
|
||
|
description:
|
||
|
- When specifying C(records_content), this is the string of characters that will
|
||
|
be used to break apart entries in the C(records_content) into key/value pairs.
|
||
|
- By default, this parameter's value is C(:=).
|
||
|
- This value cannot be changed once it is set.
|
||
|
- This parameter is only relevant when C(internal) is C(no). It will be ignored
|
||
|
otherwise.
|
||
|
default: ":="
|
||
|
delete_data_group_file:
|
||
|
description:
|
||
|
- When C(yes), will ensure that the remote data group file is deleted.
|
||
|
- This parameter is only relevant when C(state) is C(absent) and C(internal) is C(no).
|
||
|
default: no
|
||
|
type: bool
|
||
|
partition:
|
||
|
description:
|
||
|
- Device partition to manage resources on.
|
||
|
default: Common
|
||
|
state:
|
||
|
description:
|
||
|
- When C(state) is C(present), ensures the data group exists.
|
||
|
- When C(state) is C(absent), ensures that the data group is removed.
|
||
|
choices:
|
||
|
- present
|
||
|
- absent
|
||
|
default: present
|
||
|
extends_documentation_fragment: f5
|
||
|
author:
|
||
|
- Tim Rupp (@caphrim007)
|
||
|
'''
|
||
|
|
||
|
EXAMPLES = r'''
|
||
|
- name: Create a data group of addresses
|
||
|
bigip_data_group:
|
||
|
name: foo
|
||
|
password: secret
|
||
|
server: lb.mydomain.com
|
||
|
state: present
|
||
|
user: admin
|
||
|
records:
|
||
|
- key: 0.0.0.0/32
|
||
|
value: External_NAT
|
||
|
- key: 10.10.10.10
|
||
|
value: No_NAT
|
||
|
type: address
|
||
|
delegate_to: localhost
|
||
|
|
||
|
- name: Create a data group of strings
|
||
|
bigip_data_group:
|
||
|
name: foo
|
||
|
password: secret
|
||
|
server: lb.mydomain.com
|
||
|
state: present
|
||
|
user: admin
|
||
|
records:
|
||
|
- key: caddy
|
||
|
value: ""
|
||
|
- key: cafeteria
|
||
|
value: ""
|
||
|
- key: cactus
|
||
|
value: ""
|
||
|
type: string
|
||
|
delegate_to: localhost
|
||
|
|
||
|
- name: Create a data group of IP addresses from a file
|
||
|
bigip_data_group:
|
||
|
name: foo
|
||
|
password: secret
|
||
|
server: lb.mydomain.com
|
||
|
state: present
|
||
|
user: admin
|
||
|
records_src: /path/to/dg-file
|
||
|
type: address
|
||
|
delegate_to: localhost
|
||
|
|
||
|
- name: Update an existing internal data group of strings
|
||
|
bigip_data_group:
|
||
|
name: foo
|
||
|
password: secret
|
||
|
server: lb.mydomain.com
|
||
|
state: present
|
||
|
internal: yes
|
||
|
user: admin
|
||
|
records:
|
||
|
- key: caddy
|
||
|
value: ""
|
||
|
- key: cafeteria
|
||
|
value: ""
|
||
|
- key: cactus
|
||
|
value: ""
|
||
|
delegate_to: localhost
|
||
|
|
||
|
- name: Show the data format expected for records_content - address 1
|
||
|
copy:
|
||
|
dest: /path/to/addresses.txt
|
||
|
content: |
|
||
|
network 10.0.0.0 prefixlen 8 := "Network1",
|
||
|
network 172.16.0.0 prefixlen 12 := "Network2",
|
||
|
network 192.168.0.0 prefixlen 16 := "Network3",
|
||
|
network 2402:9400:1000:0:: prefixlen 64 := "Network4",
|
||
|
host 192.168.20.1 := "Host1",
|
||
|
host 172.16.1.1 := "Host2",
|
||
|
host 172.16.1.1/32 := "Host3",
|
||
|
host 2001:0db8:85a3:0000:0000:8a2e:0370:7334 := "Host4",
|
||
|
host 2001:0db8:85a3:0000:0000:8a2e:0370:7334/128 := "Host5"
|
||
|
|
||
|
- name: Show the data format expected for records_content - address 2
|
||
|
copy:
|
||
|
dest: /path/to/addresses.txt
|
||
|
content: |
|
||
|
10.0.0.0/8 := "Network1",
|
||
|
172.16.0.0/12 := "Network2",
|
||
|
192.168.0.0/16 := "Network3",
|
||
|
2402:9400:1000:0::/64 := "Network4",
|
||
|
192.168.20.1 := "Host1",
|
||
|
172.16.1.1 := "Host2",
|
||
|
172.16.1.1/32 := "Host3",
|
||
|
2001:0db8:85a3:0000:0000:8a2e:0370:7334 := "Host4",
|
||
|
2001:0db8:85a3:0000:0000:8a2e:0370:7334/128 := "Host5"
|
||
|
|
||
|
- name: Show the data format expected for records_content - string
|
||
|
copy:
|
||
|
dest: /path/to/strings.txt
|
||
|
content: |
|
||
|
a := alpha,
|
||
|
b := bravo,
|
||
|
c := charlie,
|
||
|
x := x-ray,
|
||
|
y := yankee,
|
||
|
z := zulu,
|
||
|
|
||
|
- name: Show the data format expected for records_content - integer
|
||
|
copy:
|
||
|
dest: /path/to/integers.txt
|
||
|
content: |
|
||
|
1 := bar,
|
||
|
2 := baz,
|
||
|
3,
|
||
|
4,
|
||
|
'''
|
||
|
|
||
|
RETURN = r'''
|
||
|
# only common fields returned
|
||
|
'''
|
||
|
|
||
|
import hashlib
|
||
|
import os
|
||
|
import re
|
||
|
|
||
|
from ansible.module_utils._text import to_text
|
||
|
from ansible.module_utils.basic import AnsibleModule
|
||
|
from ansible.module_utils.basic import env_fallback
|
||
|
from io import StringIO
|
||
|
|
||
|
try:
|
||
|
from library.module_utils.network.f5.bigip import HAS_F5SDK
|
||
|
from library.module_utils.network.f5.bigip import F5Client
|
||
|
from library.module_utils.network.f5.common import F5ModuleError
|
||
|
from library.module_utils.network.f5.common import AnsibleF5Parameters
|
||
|
from library.module_utils.network.f5.common import cleanup_tokens
|
||
|
from library.module_utils.network.f5.common import compare_complex_list
|
||
|
from library.module_utils.network.f5.common import f5_argument_spec
|
||
|
from library.module_utils.network.f5.ipaddress import is_valid_ip
|
||
|
from library.module_utils.network.f5.ipaddress import is_valid_ip_network
|
||
|
from library.module_utils.network.f5.ipaddress import is_valid_ip_interface
|
||
|
from library.module_utils.compat.ipaddress import ip_address
|
||
|
from library.module_utils.compat.ipaddress import ip_network
|
||
|
from library.module_utils.compat.ipaddress import ip_interface
|
||
|
try:
|
||
|
from library.module_utils.network.f5.common import iControlUnexpectedHTTPError
|
||
|
except ImportError:
|
||
|
HAS_F5SDK = False
|
||
|
except ImportError:
|
||
|
from ansible.module_utils.network.f5.bigip import HAS_F5SDK
|
||
|
from ansible.module_utils.network.f5.bigip import F5Client
|
||
|
from ansible.module_utils.network.f5.common import F5ModuleError
|
||
|
from ansible.module_utils.network.f5.common import AnsibleF5Parameters
|
||
|
from ansible.module_utils.network.f5.common import cleanup_tokens
|
||
|
from ansible.module_utils.network.f5.common import compare_complex_list
|
||
|
from ansible.module_utils.network.f5.common import f5_argument_spec
|
||
|
from ansible.module_utils.network.f5.ipaddress import is_valid_ip
|
||
|
from ansible.module_utils.network.f5.ipaddress import is_valid_ip_network
|
||
|
from ansible.module_utils.network.f5.ipaddress import is_valid_ip_interface
|
||
|
from ansible.module_utils.compat.ipaddress import ip_address
|
||
|
from ansible.module_utils.compat.ipaddress import ip_network
|
||
|
from ansible.module_utils.compat.ipaddress import ip_interface
|
||
|
try:
|
||
|
from ansible.module_utils.network.f5.common import iControlUnexpectedHTTPError
|
||
|
except ImportError:
|
||
|
HAS_F5SDK = False
|
||
|
|
||
|
|
||
|
LINE_LIMIT = 65000
|
||
|
SIZE_LIMIT_BYTES = 4000000
|
||
|
|
||
|
|
||
|
def zero_length(content):
|
||
|
content.seek(0, os.SEEK_END)
|
||
|
length = content.tell()
|
||
|
content.seek(0)
|
||
|
if length == 0:
|
||
|
return True
|
||
|
return False
|
||
|
|
||
|
|
||
|
def size_exceeded(content):
|
||
|
records = content
|
||
|
records.seek(0, os.SEEK_END)
|
||
|
size = records.tell()
|
||
|
records.seek(0)
|
||
|
if size > SIZE_LIMIT_BYTES:
|
||
|
return True
|
||
|
return False
|
||
|
|
||
|
|
||
|
def lines_exceeded(content):
|
||
|
result = False
|
||
|
for i, line in enumerate(content):
|
||
|
if i > LINE_LIMIT:
|
||
|
result = True
|
||
|
content.seek(0)
|
||
|
return result
|
||
|
|
||
|
|
||
|
class RecordsEncoder(object):
|
||
|
def __init__(self, record_type=None, separator=None):
|
||
|
self._record_type = record_type
|
||
|
self._separator = separator
|
||
|
self._network_pattern = re.compile(r'^network\s+(?P<addr>[^ ]+)\s+prefixlen\s+(?P<prefix>\d+)\s+.*')
|
||
|
self._host_pattern = re.compile(r'^host\s+(?P<addr>[^ ]+)\s+.*')
|
||
|
|
||
|
def encode(self, record):
|
||
|
if isinstance(record, dict):
|
||
|
return self.encode_dict(record)
|
||
|
else:
|
||
|
return self.encode_string(record)
|
||
|
|
||
|
def encode_dict(self, record):
|
||
|
if self._record_type == 'ip':
|
||
|
return self.encode_address_from_dict(record)
|
||
|
elif self._record_type == 'integer':
|
||
|
return self.encode_integer_from_dict(record)
|
||
|
else:
|
||
|
return self.encode_string_from_dict(record)
|
||
|
|
||
|
def encode_address_from_dict(self, record):
|
||
|
if is_valid_ip_interface(record['key']):
|
||
|
key = ip_interface(u"{0}".format(str(record['key'])))
|
||
|
else:
|
||
|
raise F5ModuleError(
|
||
|
"When specifying an 'address' type, the value to the left of the separator must be an IP."
|
||
|
)
|
||
|
if key and 'value' in record:
|
||
|
if key.network.prefixlen in [32, 128]:
|
||
|
return self.encode_host(str(key.ip), record['value'])
|
||
|
return self.encode_network(
|
||
|
str(key.network.network_address), key.network.prefixlen, record['value']
|
||
|
)
|
||
|
elif key:
|
||
|
if key.network.prefixlen in [32, 128]:
|
||
|
return self.encode_host(str(key.ip), str(key.ip))
|
||
|
return self.encode_network(
|
||
|
str(key.network.network_address), key.network.prefixlen, str(key.network.network_address)
|
||
|
)
|
||
|
|
||
|
def encode_integer_from_dict(self, record):
|
||
|
try:
|
||
|
int(record['key'])
|
||
|
except ValueError:
|
||
|
raise F5ModuleError(
|
||
|
"When specifying an 'integer' type, the value to the left of the separator must be a number."
|
||
|
)
|
||
|
if 'key' in record and 'value' in record:
|
||
|
return '{0} {1} {2}'.format(record['key'], self._separator, record['value'])
|
||
|
elif 'key' in record:
|
||
|
return str(record['key'])
|
||
|
|
||
|
def encode_string_from_dict(self, record):
|
||
|
if 'key' in record and 'value' in record:
|
||
|
return '{0} {1} {2}'.format(record['key'], self._separator, record['value'])
|
||
|
elif 'key' in record:
|
||
|
return '{0} {1} ""'.format(record['key'], self._separator)
|
||
|
|
||
|
def encode_string(self, record):
|
||
|
record = record.strip().strip(',')
|
||
|
if self._record_type == 'ip':
|
||
|
return self.encode_address_from_string(record)
|
||
|
elif self._record_type == 'integer':
|
||
|
return self.encode_integer_from_string(record)
|
||
|
else:
|
||
|
return self.encode_string_from_string(record)
|
||
|
|
||
|
def encode_address_from_string(self, record):
|
||
|
if self._network_pattern.match(record):
|
||
|
# network 192.168.0.0 prefixlen 16 := "Network3",
|
||
|
# network 2402:9400:1000:0:: prefixlen 64 := "Network4",
|
||
|
return record
|
||
|
elif self._host_pattern.match(record):
|
||
|
# host 172.16.1.1/32 := "Host3"
|
||
|
# host 2001:0db8:85a3:0000:0000:8a2e:0370:7334 := "Host4"
|
||
|
return record
|
||
|
else:
|
||
|
# 192.168.0.0/16 := "Network3",
|
||
|
# 2402:9400:1000:0::/64 := "Network4",
|
||
|
parts = record.split(self._separator)
|
||
|
if parts[0] == '':
|
||
|
return
|
||
|
if not is_valid_ip_interface(parts[0]):
|
||
|
raise F5ModuleError(
|
||
|
"When specifying an 'address' type, the value to the left of the separator must be an IP."
|
||
|
)
|
||
|
key = ip_interface(u"{0}".format(str(parts[0])))
|
||
|
|
||
|
if len(parts) == 2:
|
||
|
if key.network.prefixlen in [32, 128]:
|
||
|
return self.encode_host(str(key.ip), parts[1])
|
||
|
return self.encode_network(
|
||
|
str(key.network.network_address), key.network.prefixlen, parts[1]
|
||
|
)
|
||
|
elif len(parts) == 1 and parts[0] != '':
|
||
|
if key.network.prefixlen in [32, 128]:
|
||
|
return self.encode_host(str(key.ip), str(key.ip))
|
||
|
return self.encode_network(
|
||
|
str(key.network.network_address), key.network.prefixlen, str(key.network.network_address)
|
||
|
)
|
||
|
|
||
|
def encode_host(self, key, value):
|
||
|
return 'host {0} {1} {2}'.format(str(key), self._separator, str(value))
|
||
|
|
||
|
def encode_network(self, key, prefixlen, value):
|
||
|
return 'network {0} prefixlen {1} {2} {3}'.format(
|
||
|
str(key), str(prefixlen), self._separator, str(value)
|
||
|
)
|
||
|
|
||
|
def encode_integer_from_string(self, record):
|
||
|
parts = record.split(self._separator)
|
||
|
if len(parts) == 1 and parts[0] == '':
|
||
|
return None
|
||
|
try:
|
||
|
int(parts[0])
|
||
|
except ValueError:
|
||
|
raise F5ModuleError(
|
||
|
"When specifying an 'integer' type, the value to the left of the separator must be a number."
|
||
|
)
|
||
|
if len(parts) == 2:
|
||
|
return '{0} {1} {2}'.format(parts[0], self._separator, parts[1])
|
||
|
elif len(parts) == 1:
|
||
|
return str(parts[0])
|
||
|
|
||
|
def encode_string_from_string(self, record):
|
||
|
parts = record.split(self._separator)
|
||
|
if len(parts) == 2:
|
||
|
return '{0} {1} {2}'.format(parts[0], self._separator, parts[1])
|
||
|
elif len(parts) == 1 and parts[0] != '':
|
||
|
return '{0} {1} ""'.format(parts[0], self._separator)
|
||
|
|
||
|
|
||
|
class RecordsDecoder(object):
|
||
|
def __init__(self, record_type=None, separator=None):
|
||
|
self._record_type = record_type
|
||
|
self._separator = separator
|
||
|
self._network_pattern = re.compile(r'^network\s+(?P<addr>[^ ]+)\s+prefixlen\s+(?P<prefix>\d+)\s+.*')
|
||
|
self._host_pattern = re.compile(r'^host\s+(?P<addr>[^ ]+)\s+.*')
|
||
|
|
||
|
def decode(self, record):
|
||
|
record = record.strip().strip(',')
|
||
|
if self._record_type == 'ip':
|
||
|
return self.decode_address_from_string(record)
|
||
|
else:
|
||
|
return self.decode_from_string(record)
|
||
|
|
||
|
def decode_address_from_string(self, record):
|
||
|
matches = self._network_pattern.match(record)
|
||
|
if matches:
|
||
|
# network 192.168.0.0 prefixlen 16 := "Network3",
|
||
|
# network 2402:9400:1000:0:: prefixlen 64 := "Network4",
|
||
|
key = u"{0}/{1}".format(matches.group('addr'), matches.group('prefix'))
|
||
|
addr = ip_network(key)
|
||
|
value = record.split(self._separator)[1].strip().strip('"')
|
||
|
result = dict(name=str(addr), data=value)
|
||
|
return result
|
||
|
matches = self._host_pattern.match(record)
|
||
|
if matches:
|
||
|
# host 172.16.1.1/32 := "Host3"
|
||
|
# host 2001:0db8:85a3:0000:0000:8a2e:0370:7334 := "Host4"
|
||
|
key = matches.group('addr')
|
||
|
addr = ip_interface(u"{0}".format(str(key)))
|
||
|
value = record.split(self._separator)[1].strip().strip('"')
|
||
|
result = dict(name=str(addr), data=value)
|
||
|
return result
|
||
|
raise F5ModuleError(
|
||
|
'The value "{0}" is not an address'.format(record)
|
||
|
)
|
||
|
|
||
|
def decode_from_string(self, record):
|
||
|
parts = record.split(self._separator)
|
||
|
if len(parts) == 2:
|
||
|
return dict(name=parts[0].strip(), data=parts[1].strip('"').strip())
|
||
|
else:
|
||
|
return dict(name=parts[0].strip(), data="")
|
||
|
|
||
|
|
||
|
class Parameters(AnsibleF5Parameters):
|
||
|
api_map = {
|
||
|
'externalFileName': 'external_file_name'
|
||
|
}
|
||
|
|
||
|
api_attributes = [
|
||
|
'records', 'type'
|
||
|
]
|
||
|
|
||
|
returnables = []
|
||
|
|
||
|
updatables = [
|
||
|
'records', 'checksum'
|
||
|
]
|
||
|
|
||
|
@property
|
||
|
def type(self):
|
||
|
if self._values['type'] in ['address', 'addr', 'ip']:
|
||
|
return 'ip'
|
||
|
elif self._values['type'] in ['integer', 'int']:
|
||
|
return 'integer'
|
||
|
elif self._values['type'] in ['string']:
|
||
|
return 'string'
|
||
|
|
||
|
@property
|
||
|
def records_src(self):
|
||
|
try:
|
||
|
self._values['records_src'].seek(0)
|
||
|
return self._values['records_src']
|
||
|
except AttributeError:
|
||
|
pass
|
||
|
if self._values['records_src']:
|
||
|
records = open(self._values['records_src'])
|
||
|
else:
|
||
|
records = self._values['records']
|
||
|
|
||
|
# There is a 98% chance that the user will supply a data group that is < 1MB.
|
||
|
# 99.917% chance it is less than 10 MB. This is well within the range of typical
|
||
|
# memory available on a system.
|
||
|
#
|
||
|
# If this changes, this may need to be changed to use temporary files instead.
|
||
|
self._values['records_src'] = StringIO()
|
||
|
|
||
|
self._write_records_to_file(records)
|
||
|
return self._values['records_src']
|
||
|
|
||
|
def _write_records_to_file(self, records):
|
||
|
bucket_size = 1000000
|
||
|
bucket = []
|
||
|
encoder = RecordsEncoder(record_type=self.type, separator=self.separator)
|
||
|
for record in records:
|
||
|
result = encoder.encode(record)
|
||
|
if result:
|
||
|
bucket.append(to_text(result + ",\n"))
|
||
|
if len(bucket) == bucket_size:
|
||
|
self._values['records_src'].writelines(bucket)
|
||
|
bucket = []
|
||
|
self._values['records_src'].writelines(bucket)
|
||
|
self._values['records_src'].seek(0)
|
||
|
|
||
|
|
||
|
class ApiParameters(Parameters):
|
||
|
@property
|
||
|
def checksum(self):
|
||
|
if self._values['checksum'] is None:
|
||
|
return None
|
||
|
result = self._values['checksum'].split(':')[2]
|
||
|
return result
|
||
|
|
||
|
@property
|
||
|
def records(self):
|
||
|
if self._values['records'] is None:
|
||
|
return None
|
||
|
return self._values['records']
|
||
|
|
||
|
@property
|
||
|
def records_list(self):
|
||
|
return self.records
|
||
|
|
||
|
|
||
|
class ModuleParameters(Parameters):
|
||
|
@property
|
||
|
def checksum(self):
|
||
|
if self._values['checksum']:
|
||
|
return self._values['checksum']
|
||
|
result = hashlib.sha1()
|
||
|
records = self.records_src
|
||
|
while True:
|
||
|
data = records.read(4096)
|
||
|
if not data:
|
||
|
break
|
||
|
result.update(data.encode('utf-8'))
|
||
|
result = result.hexdigest()
|
||
|
self._values['checksum'] = result
|
||
|
return result
|
||
|
|
||
|
@property
|
||
|
def external_file_name(self):
|
||
|
if self._values['external_file_name'] is None:
|
||
|
name = self.name
|
||
|
else:
|
||
|
name = self._values['external_file_name']
|
||
|
if re.search(r'[^a-zA-Z0-9-_.]', name):
|
||
|
raise F5ModuleError(
|
||
|
"'external_file_name' may only contain letters, numbers, underscores, dashes, or a period."
|
||
|
)
|
||
|
return name
|
||
|
|
||
|
@property
|
||
|
def records(self):
|
||
|
results = []
|
||
|
decoder = RecordsDecoder(record_type=self.type, separator=self.separator)
|
||
|
for record in self.records_src:
|
||
|
result = decoder.decode(record)
|
||
|
if result:
|
||
|
results.append(result)
|
||
|
return results
|
||
|
|
||
|
@property
|
||
|
def records_list(self):
|
||
|
if self._values['records'] is None:
|
||
|
return None
|
||
|
return self.records
|
||
|
|
||
|
|
||
|
class Changes(Parameters):
|
||
|
def to_return(self):
|
||
|
result = {}
|
||
|
try:
|
||
|
for returnable in self.returnables:
|
||
|
result[returnable] = getattr(self, returnable)
|
||
|
result = self._filter_params(result)
|
||
|
except Exception:
|
||
|
pass
|
||
|
return result
|
||
|
|
||
|
|
||
|
class UsableChanges(Changes):
|
||
|
pass
|
||
|
|
||
|
|
||
|
class ReportableChanges(Changes):
|
||
|
pass
|
||
|
|
||
|
|
||
|
class Difference(object):
|
||
|
def __init__(self, want, have=None):
|
||
|
self.want = want
|
||
|
self.have = have
|
||
|
|
||
|
def compare(self, param):
|
||
|
try:
|
||
|
result = getattr(self, param)
|
||
|
return result
|
||
|
except AttributeError:
|
||
|
return self.__default(param)
|
||
|
|
||
|
def __default(self, param):
|
||
|
attr1 = getattr(self.want, param)
|
||
|
try:
|
||
|
attr2 = getattr(self.have, param)
|
||
|
if attr1 != attr2:
|
||
|
return attr1
|
||
|
except AttributeError:
|
||
|
return attr1
|
||
|
|
||
|
@property
|
||
|
def records(self):
|
||
|
# External data groups are compared by their checksum, not their records. This
|
||
|
# is because the BIG-IP does not store the actual records in the API. It instead
|
||
|
# stores the checksum of the file. External DGs have the possibility of being huge
|
||
|
# and we would never want to do a comparison of such huge files.
|
||
|
#
|
||
|
# Therefore, comparison is no-op if the DG being worked with is an external DG.
|
||
|
if self.want.internal is False:
|
||
|
return None
|
||
|
if self.have.records is None and self.want.records == []:
|
||
|
return None
|
||
|
if self.have.records is None:
|
||
|
return self.want.records
|
||
|
result = compare_complex_list(self.want.records, self.have.records)
|
||
|
return result
|
||
|
|
||
|
@property
|
||
|
def type(self):
|
||
|
return None
|
||
|
|
||
|
@property
|
||
|
def checksum(self):
|
||
|
if self.want.internal:
|
||
|
return None
|
||
|
if self.want.checksum != self.have.checksum:
|
||
|
return True
|
||
|
|
||
|
|
||
|
class BaseManager(object):
|
||
|
def __init__(self, *args, **kwargs):
|
||
|
self.module = kwargs.get('module', None)
|
||
|
self.client = kwargs.get('client', None)
|
||
|
self.want = ModuleParameters(params=self.module.params)
|
||
|
self.have = ApiParameters()
|
||
|
self.changes = UsableChanges()
|
||
|
|
||
|
def should_update(self):
|
||
|
result = self._update_changed_options()
|
||
|
if result:
|
||
|
return True
|
||
|
return False
|
||
|
|
||
|
def exec_module(self):
|
||
|
changed = False
|
||
|
result = dict()
|
||
|
state = self.want.state
|
||
|
|
||
|
try:
|
||
|
if state == "present":
|
||
|
changed = self.present()
|
||
|
elif state == "absent":
|
||
|
changed = self.absent()
|
||
|
except iControlUnexpectedHTTPError as e:
|
||
|
raise F5ModuleError(str(e))
|
||
|
|
||
|
reportable = ReportableChanges(params=self.changes.to_return())
|
||
|
changes = reportable.to_return()
|
||
|
result.update(**changes)
|
||
|
result.update(dict(changed=changed))
|
||
|
self._announce_deprecations(result)
|
||
|
return result
|
||
|
|
||
|
def _announce_deprecations(self, result):
|
||
|
warnings = result.pop('__warnings', [])
|
||
|
for warning in warnings:
|
||
|
self.client.module.deprecate(
|
||
|
msg=warning['msg'],
|
||
|
version=warning['version']
|
||
|
)
|
||
|
|
||
|
def _set_changed_options(self):
|
||
|
changed = {}
|
||
|
for key in ApiParameters.returnables:
|
||
|
if getattr(self.want, key) is not None:
|
||
|
changed[key] = getattr(self.want, key)
|
||
|
if changed:
|
||
|
self.changes = UsableChanges(params=changed)
|
||
|
|
||
|
def _update_changed_options(self):
|
||
|
diff = Difference(self.want, self.have)
|
||
|
updatables = ApiParameters.updatables
|
||
|
changed = dict()
|
||
|
for k in updatables:
|
||
|
change = diff.compare(k)
|
||
|
if change is None:
|
||
|
continue
|
||
|
else:
|
||
|
if isinstance(change, dict):
|
||
|
changed.update(change)
|
||
|
else:
|
||
|
changed[k] = change
|
||
|
if changed:
|
||
|
self.changes = UsableChanges(params=changed)
|
||
|
return True
|
||
|
return False
|
||
|
|
||
|
def present(self):
|
||
|
if self.exists():
|
||
|
return self.update()
|
||
|
else:
|
||
|
return self.create()
|
||
|
|
||
|
def absent(self):
|
||
|
if self.exists():
|
||
|
return self.remove()
|
||
|
return False
|
||
|
|
||
|
def remove(self):
|
||
|
if self.module.check_mode:
|
||
|
return True
|
||
|
self.remove_from_device()
|
||
|
if self.exists():
|
||
|
raise F5ModuleError("Failed to delete the resource.")
|
||
|
return True
|
||
|
|
||
|
|
||
|
class InternalManager(BaseManager):
|
||
|
def create(self):
|
||
|
self._set_changed_options()
|
||
|
if size_exceeded(self.want.records_src) or lines_exceeded(self.want.records_src):
|
||
|
raise F5ModuleError(
|
||
|
"The size of the provided data (or file) is too large for an internal data group."
|
||
|
)
|
||
|
if self.module.check_mode:
|
||
|
return True
|
||
|
self.create_on_device()
|
||
|
return True
|
||
|
|
||
|
def update(self):
|
||
|
self.have = self.read_current_from_device()
|
||
|
if not self.should_update():
|
||
|
return False
|
||
|
if self.module.check_mode:
|
||
|
return True
|
||
|
self.update_on_device()
|
||
|
return True
|
||
|
|
||
|
def exists(self):
|
||
|
result = self.client.api.tm.ltm.data_group.internals.internal.exists(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
return result
|
||
|
|
||
|
def create_on_device(self):
|
||
|
params = self.want.api_params()
|
||
|
self.client.api.tm.ltm.data_group.internals.internal.create(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition,
|
||
|
**params
|
||
|
)
|
||
|
|
||
|
def update_on_device(self):
|
||
|
params = self.changes.api_params()
|
||
|
resource = self.client.api.tm.ltm.data_group.internals.internal.load(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
resource.modify(**params)
|
||
|
|
||
|
def remove_from_device(self):
|
||
|
resource = self.client.api.tm.ltm.data_group.internals.internal.load(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
if resource:
|
||
|
resource.delete()
|
||
|
|
||
|
def read_current_from_device(self):
|
||
|
resource = self.client.api.tm.ltm.data_group.internals.internal.load(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
result = resource.attrs
|
||
|
return ApiParameters(params=result)
|
||
|
|
||
|
|
||
|
class ExternalManager(BaseManager):
|
||
|
def absent(self):
|
||
|
result = False
|
||
|
if self.exists():
|
||
|
result = self.remove()
|
||
|
if self.external_file_exists() and self.want.delete_data_group_file:
|
||
|
result = self.remove_data_group_file_from_device()
|
||
|
return result
|
||
|
|
||
|
def create(self):
|
||
|
if zero_length(self.want.records_src):
|
||
|
raise F5ModuleError(
|
||
|
"An external data group cannot be empty."
|
||
|
)
|
||
|
self._set_changed_options()
|
||
|
if self.module.check_mode:
|
||
|
return True
|
||
|
self.create_on_device()
|
||
|
return True
|
||
|
|
||
|
def update(self):
|
||
|
self.have = self.read_current_from_device()
|
||
|
if not self.should_update():
|
||
|
return False
|
||
|
if zero_length(self.want.records_src):
|
||
|
raise F5ModuleError(
|
||
|
"An external data group cannot be empty."
|
||
|
)
|
||
|
if self.module.check_mode:
|
||
|
return True
|
||
|
self.update_on_device()
|
||
|
return True
|
||
|
|
||
|
def exists(self):
|
||
|
result = self.client.api.tm.ltm.data_group.externals.external.exists(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
return result
|
||
|
|
||
|
def external_file_exists(self):
|
||
|
result = self.client.api.tm.sys.file.data_groups.data_group.exists(
|
||
|
name=self.want.external_file_name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
return result
|
||
|
|
||
|
def _upload_to_file(self, name, type, remote_path, update=False):
|
||
|
self.client.api.shared.file_transfer.uploads.upload_stringio(self.want.records_src, name)
|
||
|
resource = self.client.api.tm.sys.file.data_groups
|
||
|
if update:
|
||
|
resource = resource.data_group.load(
|
||
|
name=name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
resource.modify(
|
||
|
sourcePath='file:{0}'.format(remote_path)
|
||
|
)
|
||
|
resource.refresh()
|
||
|
result = resource
|
||
|
else:
|
||
|
result = resource.data_group.create(
|
||
|
name=name,
|
||
|
type=type,
|
||
|
sourcePath='file:{0}'.format(remote_path)
|
||
|
)
|
||
|
return result.name
|
||
|
|
||
|
def create_on_device(self):
|
||
|
name = self.want.external_file_name
|
||
|
remote_path = '/var/config/rest/downloads/{0}'.format(name)
|
||
|
external_file = self._upload_to_file(name, self.want.type, remote_path, update=False)
|
||
|
self.client.api.tm.ltm.data_group.externals.external.create(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition,
|
||
|
externalFileName=external_file
|
||
|
)
|
||
|
self.client.api.tm.util.unix_rm.exec_cmd('run', utilCmdArgs=remote_path)
|
||
|
|
||
|
def update_on_device(self):
|
||
|
name = self.want.external_file_name
|
||
|
remote_path = '/var/config/rest/downloads/{0}'.format(name)
|
||
|
external_file = self._upload_to_file(name, self.have.type, remote_path, update=True)
|
||
|
resource = self.client.api.tm.ltm.data_group.externals.external.load(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
resource.modify(
|
||
|
externalFileName=external_file
|
||
|
)
|
||
|
|
||
|
def remove_from_device(self):
|
||
|
resource = self.client.api.tm.ltm.data_group.externals.external.load(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
if resource:
|
||
|
resource.delete()
|
||
|
|
||
|
# Remove the remote data group file if asked to
|
||
|
if self.want.delete_data_group_file:
|
||
|
self.remove_data_group_file_from_device()
|
||
|
|
||
|
def remove_data_group_file_from_device(self):
|
||
|
resource = self.client.api.tm.sys.file.data_groups.data_group.load(
|
||
|
name=self.want.external_file_name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
if resource:
|
||
|
resource.delete()
|
||
|
return True
|
||
|
return False
|
||
|
|
||
|
def read_current_from_device(self):
|
||
|
"""Reads the current configuration from the device
|
||
|
|
||
|
For an external data group, we are interested in two things from the
|
||
|
current configuration
|
||
|
|
||
|
* ``checksum``
|
||
|
* ``type``
|
||
|
|
||
|
The ``checksum`` will allow us to compare the data group value we have
|
||
|
with the data group value being provided.
|
||
|
|
||
|
The ``type`` will allow us to do validation on the data group value being
|
||
|
provided (if any).
|
||
|
|
||
|
Returns:
|
||
|
ExternalApiParameters: Attributes of the remote resource.
|
||
|
"""
|
||
|
resource = self.client.api.tm.ltm.data_group.externals.external.load(
|
||
|
name=self.want.name,
|
||
|
partition=self.want.partition
|
||
|
)
|
||
|
external_file = os.path.basename(resource.externalFileName)
|
||
|
external_file_partition = os.path.dirname(resource.externalFileName).strip('/')
|
||
|
resource = self.client.api.tm.sys.file.data_groups.data_group.load(
|
||
|
name=external_file,
|
||
|
partition=external_file_partition
|
||
|
)
|
||
|
result = resource.attrs
|
||
|
return ApiParameters(params=result)
|
||
|
|
||
|
|
||
|
class ModuleManager(object):
|
||
|
def __init__(self, *args, **kwargs):
|
||
|
self.kwargs = kwargs
|
||
|
self.module = kwargs.get('module')
|
||
|
self.client = kwargs.get('client', None)
|
||
|
|
||
|
def exec_module(self):
|
||
|
if self.module.params['internal']:
|
||
|
manager = self.get_manager('internal')
|
||
|
else:
|
||
|
manager = self.get_manager('external')
|
||
|
return manager.exec_module()
|
||
|
|
||
|
def get_manager(self, type):
|
||
|
if type == 'internal':
|
||
|
return InternalManager(**self.kwargs)
|
||
|
elif type == 'external':
|
||
|
return ExternalManager(**self.kwargs)
|
||
|
|
||
|
|
||
|
class ArgumentSpec(object):
|
||
|
def __init__(self):
|
||
|
self.supports_check_mode = True
|
||
|
argument_spec = dict(
|
||
|
name=dict(required=True),
|
||
|
type=dict(
|
||
|
choices=['address', 'addr', 'ip', 'string', 'str', 'integer', 'int'],
|
||
|
default='string'
|
||
|
),
|
||
|
delete_data_group_file=dict(type='bool'),
|
||
|
internal=dict(type='bool', default='no'),
|
||
|
records=dict(
|
||
|
type='list',
|
||
|
suboptions=dict(
|
||
|
key=dict(required=True),
|
||
|
value=dict(type='raw')
|
||
|
)
|
||
|
),
|
||
|
records_src=dict(type='path'),
|
||
|
external_file_name=dict(),
|
||
|
separator=dict(default=':='),
|
||
|
state=dict(choices=['absent', 'present'], default='present'),
|
||
|
partition=dict(
|
||
|
default='Common',
|
||
|
fallback=(env_fallback, ['F5_PARTITION'])
|
||
|
)
|
||
|
)
|
||
|
self.argument_spec = {}
|
||
|
self.argument_spec.update(f5_argument_spec)
|
||
|
self.argument_spec.update(argument_spec)
|
||
|
self.mutually_exclusive = [
|
||
|
['records', 'records_content', 'external_file_name']
|
||
|
]
|
||
|
|
||
|
|
||
|
def main():
|
||
|
spec = ArgumentSpec()
|
||
|
|
||
|
module = AnsibleModule(
|
||
|
argument_spec=spec.argument_spec,
|
||
|
supports_check_mode=spec.supports_check_mode
|
||
|
)
|
||
|
if not HAS_F5SDK:
|
||
|
module.fail_json(msg="The python f5-sdk module is required")
|
||
|
|
||
|
try:
|
||
|
client = F5Client(**module.params)
|
||
|
mm = ModuleManager(module=module, client=client)
|
||
|
results = mm.exec_module()
|
||
|
cleanup_tokens(client)
|
||
|
module.exit_json(**results)
|
||
|
except F5ModuleError as ex:
|
||
|
cleanup_tokens(client)
|
||
|
module.fail_json(msg=str(ex))
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
main()
|