관리-도구

편집 파일: ipa_otptoken.py

# -*- coding: utf-8 -*- # Copyright (c) 2017, Ansible Project # GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt) # SPDX-License-Identifier: GPL-3.0-or-later from __future__ import absolute_import, division, print_function __metaclass__ = type DOCUMENTATION = r''' --- module: ipa_otptoken author: justchris1 (@justchris1) short_description: Manage FreeIPA OTPs version_added: 2.5.0 description: - Add, modify, and delete One Time Passwords in IPA. attributes: check_mode: support: full diff_mode: support: none options: uniqueid: description: Unique ID of the token in IPA. required: true aliases: ["name"] type: str newuniqueid: description: If specified, the unique id specified will be changed to this. type: str otptype: description: - Type of OTP. - "B(Note:) Cannot be modified after OTP is created." type: str choices: [ totp, hotp ] secretkey: description: - Token secret (Base64). - If OTP is created and this is not specified, a random secret will be generated by IPA. - "B(Note:) Cannot be modified after OTP is created." type: str description: description: Description of the token (informational only). type: str owner: description: Assigned user of the token. type: str enabled: description: Mark the token as enabled (default V(true)). default: true type: bool notbefore: description: - First date/time the token can be used. - In the format C(YYYYMMddHHmmss). - For example, C(20180121182022) will allow the token to be used starting on 21 January 2018 at 18:20:22. type: str notafter: description: - Last date/time the token can be used. - In the format C(YYYYMMddHHmmss). - For example, C(20200121182022) will allow the token to be used until 21 January 2020 at 18:20:22. type: str vendor: description: Token vendor name (informational only). type: str model: description: Token model (informational only). type: str serial: description: Token serial (informational only). type: str state: description: State to ensure. choices: ['present', 'absent'] default: 'present' type: str algorithm: description: - Token hash algorithm. - "B(Note:) Cannot be modified after OTP is created." choices: ['sha1', 'sha256', 'sha384', 'sha512'] type: str digits: description: - Number of digits each token code will have. - "B(Note:) Cannot be modified after OTP is created." choices: [ 6, 8 ] type: int offset: description: - TOTP token / IPA server time difference. - "B(Note:) Cannot be modified after OTP is created." type: int interval: description: - Length of TOTP token code validity in seconds. - "B(Note:) Cannot be modified after OTP is created." type: int counter: description: - Initial counter for the HOTP token. - "B(Note:) Cannot be modified after OTP is created." type: int extends_documentation_fragment: - community.general.ipa.documentation - community.general.attributes ''' EXAMPLES = r''' - name: Create a totp for pinky, allowing the IPA server to generate using defaults community.general.ipa_otptoken: uniqueid: Token123 otptype: totp owner: pinky ipa_host: ipa.example.com ipa_user: admin ipa_pass: topsecret - name: Create a 8 digit hotp for pinky with sha256 with specified validity times community.general.ipa_otptoken: uniqueid: Token123 enabled: true otptype: hotp digits: 8 secretkey: UMKSIER00zT2T2tWMUlTRmNlekRCbFQvWFBVZUh2dElHWGR6T3VUR3IzK2xjaFk9 algorithm: sha256 notbefore: 20180121182123 notafter: 20220121182123 owner: pinky ipa_host: ipa.example.com ipa_user: admin ipa_pass: topsecret - name: Update Token123 to indicate a vendor, model, serial number (info only), and description community.general.ipa_otptoken: uniqueid: Token123 vendor: Acme model: acme101 serial: SerialNumber1 description: Acme OTP device ipa_host: ipa.example.com ipa_user: admin ipa_pass: topsecret - name: Disable Token123 community.general.ipa_otptoken: uniqueid: Token123 enabled: false ipa_host: ipa.example.com ipa_user: admin ipa_pass: topsecret - name: Rename Token123 to TokenABC and enable it community.general.ipa_otptoken: uniqueid: Token123 newuniqueid: TokenABC enabled: true ipa_host: ipa.example.com ipa_user: admin ipa_pass: topsecret ''' RETURN = r''' otptoken: description: OTP Token as returned by IPA API returned: always type: dict ''' import base64 import traceback from ansible.module_utils.basic import AnsibleModule, sanitize_keys from ansible_collections.community.general.plugins.module_utils.ipa import IPAClient, ipa_argument_spec from ansible.module_utils.common.text.converters import to_native class OTPTokenIPAClient(IPAClient): def __init__(self, module, host, port, protocol): super(OTPTokenIPAClient, self).__init__(module, host, port, protocol) def otptoken_find(self, name): return self._post_json(method='otptoken_find', name=None, item={'all': True, 'ipatokenuniqueid': name, 'timelimit': '0', 'sizelimit': '0'}) def otptoken_add(self, name, item): return self._post_json(method='otptoken_add', name=name, item=item) def otptoken_mod(self, name, item): return self._post_json(method='otptoken_mod', name=name, item=item) def otptoken_del(self, name): return self._post_json(method='otptoken_del', name=name) def base64_to_base32(base64_string): """Converts base64 string to base32 string""" b32_string = base64.b32encode(base64.b64decode(base64_string)).decode('ascii') return b32_string def base32_to_base64(base32_string): """Converts base32 string to base64 string""" b64_string = base64.b64encode(base64.b32decode(base32_string)).decode('ascii') return b64_string def get_otptoken_dict(ansible_to_ipa, uniqueid=None, newuniqueid=None, otptype=None, secretkey=None, description=None, owner=None, enabled=None, notbefore=None, notafter=None, vendor=None, model=None, serial=None, algorithm=None, digits=None, offset=None, interval=None, counter=None): """Create the dictionary of settings passed in""" otptoken = {} if uniqueid is not None: otptoken[ansible_to_ipa['uniqueid']] = uniqueid if newuniqueid is not None: otptoken[ansible_to_ipa['newuniqueid']] = newuniqueid if otptype is not None: otptoken[ansible_to_ipa['otptype']] = otptype.upper() if secretkey is not None: # For some unknown reason, while IPA returns the secret in base64, # it wants the secret passed in as base32. This makes it more difficult # for comparison (does 'current' equal to 'new'). Moreover, this may # cause some subtle issue in a playbook as the output is encoded # in a different way than if it was passed in as a parameter. For # these reasons, have the module standardize on base64 input (as parameter) # and output (from IPA). otptoken[ansible_to_ipa['secretkey']] = base64_to_base32(secretkey) if description is not None: otptoken[ansible_to_ipa['description']] = description if owner is not None: otptoken[ansible_to_ipa['owner']] = owner if enabled is not None: otptoken[ansible_to_ipa['enabled']] = 'FALSE' if enabled else 'TRUE' if notbefore is not None: otptoken[ansible_to_ipa['notbefore']] = notbefore + 'Z' if notafter is not None: otptoken[ansible_to_ipa['notafter']] = notafter + 'Z' if vendor is not None: otptoken[ansible_to_ipa['vendor']] = vendor if model is not None: otptoken[ansible_to_ipa['model']] = model if serial is not None: otptoken[ansible_to_ipa['serial']] = serial if algorithm is not None: otptoken[ansible_to_ipa['algorithm']] = algorithm if digits is not None: otptoken[ansible_to_ipa['digits']] = str(digits) if offset is not None: otptoken[ansible_to_ipa['offset']] = str(offset) if interval is not None: otptoken[ansible_to_ipa['interval']] = str(interval) if counter is not None: otptoken[ansible_to_ipa['counter']] = str(counter) return otptoken def transform_output(ipa_otptoken, ansible_to_ipa, ipa_to_ansible): """Transform the output received by IPA to a format more friendly before it is returned to the user. IPA returns even simple strings as a list of strings. It also returns bools and int as string. This function cleans that up before return. """ updated_otptoken = ipa_otptoken # Used to hold values that will be sanitized from output as no_log. # For the case where secretkey is not specified at the module, but # is passed back from IPA. sanitize_strings = set() # Rename the IPA parameters to the more friendly ansible module names for them for ipa_parameter in ipa_to_ansible: if ipa_parameter in ipa_otptoken: updated_otptoken[ipa_to_ansible[ipa_parameter]] = ipa_otptoken[ipa_parameter] updated_otptoken.pop(ipa_parameter) # Change the type from IPA's list of string to the appropriate return value type # based on field. By default, assume they should be strings. for ansible_parameter in ansible_to_ipa: if ansible_parameter in updated_otptoken: if isinstance(updated_otptoken[ansible_parameter], list) and len(updated_otptoken[ansible_parameter]) == 1: if ansible_parameter in ['digits', 'offset', 'interval', 'counter']: updated_otptoken[ansible_parameter] = int(updated_otptoken[ansible_parameter][0]) elif ansible_parameter == 'enabled': updated_otptoken[ansible_parameter] = bool(updated_otptoken[ansible_parameter][0]) else: updated_otptoken[ansible_parameter] = updated_otptoken[ansible_parameter][0] if 'secretkey' in updated_otptoken: if isinstance(updated_otptoken['secretkey'], dict): if '__base64__' in updated_otptoken['secretkey']: sanitize_strings.add(updated_otptoken['secretkey']['__base64__']) b64key = updated_otptoken['secretkey']['__base64__'] updated_otptoken.pop('secretkey') updated_otptoken['secretkey'] = b64key sanitize_strings.add(b64key) elif '__base32__' in updated_otptoken['secretkey']: sanitize_strings.add(updated_otptoken['secretkey']['__base32__']) b32key = updated_otptoken['secretkey']['__base32__'] b64key = base32_to_base64(b32key) updated_otptoken.pop('secretkey') updated_otptoken['secretkey'] = b64key sanitize_strings.add(b32key) sanitize_strings.add(b64key) return updated_otptoken, sanitize_strings def validate_modifications(ansible_to_ipa, module, ipa_otptoken, module_otptoken, unmodifiable_after_creation): """Checks to see if the requested modifications are valid. Some elements cannot be modified after initial creation. However, we still want to validate arguments that are specified, but are not different than what is currently set on the server. """ modifications_valid = True for parameter in unmodifiable_after_creation: if ansible_to_ipa[parameter] in module_otptoken and ansible_to_ipa[parameter] in ipa_otptoken: mod_value = module_otptoken[ansible_to_ipa[parameter]] # For someone unknown reason, the returns from IPA put almost all # values in a list, even though passing them in a list (even of # length 1) will be rejected. The module values for all elements # other than type (totp or hotp) have this happen. if parameter == 'otptype': ipa_value = ipa_otptoken[ansible_to_ipa[parameter]] else: if len(ipa_otptoken[ansible_to_ipa[parameter]]) != 1: module.fail_json(msg=("Invariant fail: Return value from IPA is not a list " + "of length 1. Please open a bug report for the module.")) if parameter == 'secretkey': # We stored the secret key in base32 since we had assumed that would need to # be the format if we were contacting IPA to create it. However, we are # now comparing it against what is already set in the IPA server, so convert # back to base64 for comparison. mod_value = base32_to_base64(mod_value) # For the secret key, it is even more specific in that the key is returned # in a dict, in the list, as the __base64__ entry for the IPA response. ipa_value = ipa_otptoken[ansible_to_ipa[parameter]][0]['__base64__'] if '__base64__' in ipa_otptoken[ansible_to_ipa[parameter]][0]: ipa_value = ipa_otptoken[ansible_to_ipa[parameter]][0]['__base64__'] elif '__base32__' in ipa_otptoken[ansible_to_ipa[parameter]][0]: b32key = ipa_otptoken[ansible_to_ipa[parameter]][0]['__base32__'] b64key = base32_to_base64(b32key) ipa_value = b64key else: ipa_value = None else: ipa_value = ipa_otptoken[ansible_to_ipa[parameter]][0] if mod_value != ipa_value: modifications_valid = False fail_message = ("Parameter '" + parameter + "' cannot be changed once " + "the OTP is created and the requested value specified here (" + str(mod_value) + ") differs from what is set in the IPA server (" + str(ipa_value) + ")") module.fail_json(msg=fail_message) return modifications_valid def ensure(module, client): # dict to map from ansible parameter names to attribute names # used by IPA (which are not so friendly). ansible_to_ipa = {'uniqueid': 'ipatokenuniqueid', 'newuniqueid': 'rename', 'otptype': 'type', 'secretkey': 'ipatokenotpkey', 'description': 'description', 'owner': 'ipatokenowner', 'enabled': 'ipatokendisabled', 'notbefore': 'ipatokennotbefore', 'notafter': 'ipatokennotafter', 'vendor': 'ipatokenvendor', 'model': 'ipatokenmodel', 'serial': 'ipatokenserial', 'algorithm': 'ipatokenotpalgorithm', 'digits': 'ipatokenotpdigits', 'offset': 'ipatokentotpclockoffset', 'interval': 'ipatokentotptimestep', 'counter': 'ipatokenhotpcounter'} # Create inverse dictionary for mapping return values ipa_to_ansible = {} for (k, v) in ansible_to_ipa.items(): ipa_to_ansible[v] = k unmodifiable_after_creation = ['otptype', 'secretkey', 'algorithm', 'digits', 'offset', 'interval', 'counter'] state = module.params['state'] uniqueid = module.params['uniqueid'] module_otptoken = get_otptoken_dict(ansible_to_ipa=ansible_to_ipa, uniqueid=module.params.get('uniqueid'), newuniqueid=module.params.get('newuniqueid'), otptype=module.params.get('otptype'), secretkey=module.params.get('secretkey'), description=module.params.get('description'), owner=module.params.get('owner'), enabled=module.params.get('enabled'), notbefore=module.params.get('notbefore'), notafter=module.params.get('notafter'), vendor=module.params.get('vendor'), model=module.params.get('model'), serial=module.params.get('serial'), algorithm=module.params.get('algorithm'), digits=module.params.get('digits'), offset=module.params.get('offset'), interval=module.params.get('interval'), counter=module.params.get('counter')) ipa_otptoken = client.otptoken_find(name=uniqueid) if ansible_to_ipa['newuniqueid'] in module_otptoken: # Check to see if the new unique id is already taken in use ipa_otptoken_new = client.otptoken_find(name=module_otptoken[ansible_to_ipa['newuniqueid']]) if ipa_otptoken_new: module.fail_json(msg=("Requested rename through newuniqueid to " + module_otptoken[ansible_to_ipa['newuniqueid']] + " failed because the new unique id is already in use")) changed = False if state == 'present': if not ipa_otptoken: changed = True if not module.check_mode: # It would not make sense to have a rename after creation, so if the user # specified a newuniqueid, just replace the uniqueid with the updated one # before creation if ansible_to_ipa['newuniqueid'] in module_otptoken: module_otptoken[ansible_to_ipa['uniqueid']] = module_otptoken[ansible_to_ipa['newuniqueid']] uniqueid = module_otptoken[ansible_to_ipa['newuniqueid']] module_otptoken.pop(ansible_to_ipa['newuniqueid']) # IPA wants the unique id in the first position and not as a key/value pair. # Get rid of it from the otptoken dict and just specify it in the name field # for otptoken_add. if ansible_to_ipa['uniqueid'] in module_otptoken: module_otptoken.pop(ansible_to_ipa['uniqueid']) module_otptoken['all'] = True ipa_otptoken = client.otptoken_add(name=uniqueid, item=module_otptoken) else: if not validate_modifications(ansible_to_ipa, module, ipa_otptoken, module_otptoken, unmodifiable_after_creation): module.fail_json(msg="Modifications requested in module are not valid") # IPA will reject 'modifications' that do not actually modify anything # if any of the unmodifiable elements are specified. Explicitly # get rid of them here. They were not different or else the # we would have failed out in validate_modifications. for x in unmodifiable_after_creation: if ansible_to_ipa[x] in module_otptoken: module_otptoken.pop(ansible_to_ipa[x]) diff = client.get_diff(ipa_data=ipa_otptoken, module_data=module_otptoken) if len(diff) > 0: changed = True if not module.check_mode: # IPA wants the unique id in the first position and not as a key/value pair. # Get rid of it from the otptoken dict and just specify it in the name field # for otptoken_mod. if ansible_to_ipa['uniqueid'] in module_otptoken: module_otptoken.pop(ansible_to_ipa['uniqueid']) module_otptoken['all'] = True ipa_otptoken = client.otptoken_mod(name=uniqueid, item=module_otptoken) else: if ipa_otptoken: changed = True if not module.check_mode: client.otptoken_del(name=uniqueid) # Transform the output to use ansible keywords (not the IPA keywords) and # sanitize any key values in the output. ipa_otptoken, sanitize_strings = transform_output(ipa_otptoken, ansible_to_ipa, ipa_to_ansible) module.no_log_values = module.no_log_values.union(sanitize_strings) sanitized_otptoken = sanitize_keys(obj=ipa_otptoken, no_log_strings=module.no_log_values) return changed, sanitized_otptoken def main(): argument_spec = ipa_argument_spec() argument_spec.update(uniqueid=dict(type='str', aliases=['name'], required=True), newuniqueid=dict(type='str'), otptype=dict(type='str', choices=['totp', 'hotp']), secretkey=dict(type='str', no_log=True), description=dict(type='str'), owner=dict(type='str'), enabled=dict(type='bool', default=True), notbefore=dict(type='str'), notafter=dict(type='str'), vendor=dict(type='str'), model=dict(type='str'), serial=dict(type='str'), state=dict(type='str', choices=['present', 'absent'], default='present'), algorithm=dict(type='str', choices=['sha1', 'sha256', 'sha384', 'sha512']), digits=dict(type='int', choices=[6, 8]), offset=dict(type='int'), interval=dict(type='int'), counter=dict(type='int')) module = AnsibleModule(argument_spec=argument_spec, supports_check_mode=True) client = OTPTokenIPAClient(module=module, host=module.params['ipa_host'], port=module.params['ipa_port'], protocol=module.params['ipa_prot']) try: client.login(username=module.params['ipa_user'], password=module.params['ipa_pass']) changed, otptoken = ensure(module, client) except Exception as e: module.fail_json(msg=to_native(e), exception=traceback.format_exc()) module.exit_json(changed=changed, otptoken=otptoken) if __name__ == '__main__': main()