Exemplo n.º 1
0
def save_config(module, result):
    result['changed'] = True
    if not module.check_mode:
        run_commands(module, 'write memory')
    else:
        module.warn(
            'Skipping command `copy running-config start-up configuration` '
            'due to check_mode.  Configuration not copied to '
            'non-volatile storage')
def main():
    """ main entry point for module execution
    """
    argument_spec = dict(
        server_type=dict(type='str',
                         choices=['scp', 'https', 'tftp'],
                         required=False),
        server_address=dict(type='str', required=False),
        server_port=dict(type='int'),
        partition=dict(type='str',
                       choices=[
                           'primary', 'secondary', 'fips-ufi-primary-sig',
                           'fips-ufi-secondary-sig'
                       ],
                       required=True),
        filename=dict(type='str', required=False),
        boot_only=dict(type='bool', required=True),
        save_running_config=dict(type='bool'),
        scp_user=dict(type='str', required=False),
        scp_pass=dict(type='str', required=False, no_log=True),
    )

    module = AnsibleModule(argument_spec=argument_spec,
                           supports_check_mode=True)
    checkValidations(module)
    result = {'changed': False}

    commands = map_params_to_obj(module)

    if module.params['save_running_config']:
        if module.params['boot_only']:
            result['commands'] = [commands[0], commands[1]]
        else:
            result['commands'] = [
                commands[0]['command'], commands[1], commands[2]
            ]
    else:
        if module.params['boot_only']:
            result['commands'] = [commands[0]]
        else:
            result['commands'] = [commands[0]['command'], commands[1]]

    if commands:
        if not module.check_mode:
            if module.params['server_type'] == 'scp':
                responses = exec_scp(module, commands[0])
                responses = run_commands(module, commands[1:])
            else:
                responses = run_commands(module, commands)
        result['changed'] = True
        result['responses'] = responses

    module.exit_json(**result)
Exemplo n.º 3
0
def has_lldp(module):
    # run_commands(module, ['skip'])
    output = run_commands(module, ['show lldp'])
    is_lldp_enable = False
    if len(output) > 0 and "LLDP is not running" not in output[0]:
        is_lldp_enable = True

    return is_lldp_enable
Exemplo n.º 4
0
def main():
    """entry point for module execution
    """
    argument_spec = dict(upload=dict(type='str',
                                     required=False,
                                     choices=[
                                         'running-config', 'flash_primary',
                                         'flash_secondary', 'startup-config'
                                     ]),
                         download=dict(type='str',
                                       required=False,
                                       choices=[
                                           'running-config', 'startup-config',
                                           'flash_primary', 'flash_secondary',
                                           'bootrom', 'fips-primary-sig',
                                           'fips-secondary-sig',
                                           'fips-bootrom-sig'
                                       ]),
                         protocol=dict(type='str',
                                       required=True,
                                       choices=['https', 'scp']),
                         remote_server=dict(type='str', required=True),
                         remote_port=dict(type='str', required=False),
                         remote_filename=dict(type='str', required=True),
                         remote_user=dict(type='str', required=False),
                         remote_pass=dict(type='str',
                                          required=False,
                                          no_log=True),
                         public_key=dict(type='str',
                                         required=False,
                                         choices=['rsa', 'dsa']))
    mutually_exclusive = [['upload', 'download']]
    module = AnsibleModule(argument_spec=argument_spec,
                           supports_check_mode=False,
                           mutually_exclusive=mutually_exclusive)

    checkValidations(module)
    warnings = list()
    result = {'changed': False, 'warnings': warnings}

    response = ''
    try:
        command = map_params_to_obj(module)
        result['commands'] = [command["command"]]

        if (module.params['protocol'] == 'scp'):
            response = exec_scp(module, command)
        else:
            response = run_commands(module, command)
        if ('Response Code: 404' in response):
            module.fail_json(msg=response)
        else:
            result['response'] = "in progress..."
        if (module.params["download"] is not None):
            result['changed'] = True
    except ConnectionError as exc:
        module.fail_json(msg=to_text(exc, errors='surrogate_then_replace'))
    module.exit_json(**result)
Exemplo n.º 5
0
def main():
    """main entry point for module execution
    """
    argument_spec = dict(commands=dict(type='list', required=True),
                         wait_for=dict(type='list', aliases=['waitfor']),
                         match=dict(default='all', choices=['all', 'any']),
                         retries=dict(default=10, type='int'),
                         interval=dict(default=1, type='int'))

    module = AnsibleModule(argument_spec=argument_spec,
                           supports_check_mode=True)

    result = {'changed': False}

    warnings = list()
    commands = parse_commands(module, warnings)
    result['warnings'] = warnings

    wait_for = module.params['wait_for'] or list()
    conditionals = [Conditional(c) for c in wait_for]
    retries = module.params['retries']
    interval = module.params['interval']
    match = module.params['match']

    while retries > 0:

        responses = run_commands(module, commands)

        for item in list(conditionals):

            if item(responses):
                if match == 'any':
                    conditionals = list()
                    break
                conditionals.remove(item)

        if not conditionals:
            break

        time.sleep(interval)
        retries -= 1

    if conditionals:
        failed_conditions = [item.raw for item in conditionals]
        msg = 'One or more conditional statements have not been satisfied'
        module.fail_json(msg=msg, failed_conditions=failed_conditions)

    result.update({
        'changed': False,
        'stdout': responses,
        'stdout_lines': list(to_lines(responses))
    })

    module.exit_json(**result)
def map_config_to_obj(module):
    obj = []
    # compare = module.params['check_running_config']
    out = run_commands(module, 'sh ipv6 static route')
    for line in out[0].splitlines():
        splitted_line = line.split()
        if len(splitted_line) not in (4, 5, 6):
            continue
        prefix = splitted_line[0]
        next_hop = splitted_line[2]
        if len(splitted_line) >= 4:
            admin_distance = splitted_line[3].split('/')[1]
        else:
            admin_distance = '1'

        obj.append({
            'prefix': prefix,
            'next_hop': next_hop,
            'admin_distance': admin_distance
        })
    return obj
Exemplo n.º 7
0
def main():
    """ main entry point for module execution
    """
    argument_spec = dict(
        src=dict(),
        lines=dict(aliases=['commands'], type='list'),
        parents=dict(type='list'),
        before=dict(type='list'),
        after=dict(type='list'),
        match=dict(default='line', choices=['line', 'strict', 'exact',
                                            'none']),
        replace=dict(default='line', choices=['line', 'block']),
        multiline_delimiter=dict(default='@'),
        running_config=dict(aliases=['config']),
        intended_config=dict(),
        defaults=dict(type='bool', default=False),
        backup=dict(type='bool', default=False),
        save_when=dict(choices=['always', 'never', 'modified', 'changed'],
                       default='never'),
        diff_against=dict(choices=['startup', 'intended', 'running']),
        diff_ignore_lines=dict(type='list'),
    )

    mutually_exclusive = [('lines', 'src'), ('parents', 'src')]

    required_if = [('match', 'strict', ['lines']),
                   ('match', 'exact', ['lines']),
                   ('replace', 'block', ['lines']),
                   ('diff_against', 'intended', ['intended_config'])]

    module = AnsibleModule(argument_spec=argument_spec,
                           mutually_exclusive=mutually_exclusive,
                           required_if=required_if,
                           supports_check_mode=True)

    result = {'changed': False}

    warnings = list()
    check_args(module, warnings)
    result['warnings'] = warnings
    diff_ignore_lines = module.params['diff_ignore_lines']
    config = None
    contents = None
    flags = None if module.params['defaults'] else []
    connection = get_connection(module)

    if module.params['backup'] or (module._diff and
                                   module.params['diff_against'] == 'running'):
        contents = get_config(module, flags=flags)
        config = NetworkConfig(indent=1, contents=contents)
        if module.params['backup']:
            result['__backup__'] = contents

    if any((module.params['lines'], module.params['src'])):
        match = module.params['match']
        replace = module.params['replace']
        path = module.params['parents']

        candidate = get_candidate_config(module)
        running = get_running_config(module, contents, flags=flags)
        try:
            response = connection.get_diff(candidate=candidate,
                                           running=running,
                                           diff_match=match,
                                           diff_ignore_lines=diff_ignore_lines,
                                           path=path,
                                           diff_replace=replace)
        except ConnectionError as exc:
            module.fail_json(msg=to_text(exc, errors='surrogate_then_replace'))

        config_diff = response['config_diff']
        banner_diff = response['banner_diff']

        if config_diff or banner_diff:
            commands = config_diff.split('\n')

            if module.params['before']:
                commands[:0] = module.params['before']

            if module.params['after']:
                commands.extend(module.params['after'])

            result['commands'] = commands
            result['updates'] = commands
            result['banners'] = banner_diff

            # send the configuration commands to the device and merge
            # them with the current running config
            if not module.check_mode:
                if commands:
                    edit_config_or_macro(connection, commands)
                if banner_diff:
                    connection.edit_banner(candidate=json.dumps(banner_diff),
                                           multiline_delimiter=module.
                                           params['multiline_delimiter'])

            result['changed'] = True

    running_config = module.params['running_config']
    startup_config = None

    if module.params['save_when'] == 'always':
        save_config(module, result)
    elif module.params['save_when'] == 'modified':
        output = run_commands(module,
                              ['show running-config', 'show configuration'])

        running_config = NetworkConfig(indent=1,
                                       contents=output[0],
                                       ignore_lines=diff_ignore_lines)
        startup_config = NetworkConfig(indent=1,
                                       contents=output[1],
                                       ignore_lines=diff_ignore_lines)

        if running_config.sha1 != startup_config.sha1:
            save_config(module, result)
    elif module.params['save_when'] == 'changed' and result['changed']:
        save_config(module, result)

    if module._diff:
        if not running_config:
            output = run_commands(module, 'show running-config')
            contents = output[0]
        else:
            contents = running_config

        # recreate the object in order to process diff_ignore_lines
        running_config = NetworkConfig(indent=1,
                                       contents=contents,
                                       ignore_lines=diff_ignore_lines)

        if module.params['diff_against'] == 'running':
            if module.check_mode:
                module.warn(
                    "unable to perform diff against running-config due to check mode"
                )
                contents = None
            else:
                contents = config.config_text

        elif module.params['diff_against'] == 'startup':
            if not startup_config:
                output = run_commands(module, 'show configuration')
                contents = output[0]
            else:
                contents = startup_config.config_text

        elif module.params['diff_against'] == 'intended':
            contents = module.params['intended_config']

        if contents is not None:
            base_config = NetworkConfig(indent=1,
                                        contents=contents,
                                        ignore_lines=diff_ignore_lines)

            if running_config.sha1 != base_config.sha1:
                if module.params['diff_against'] == 'intended':
                    before = running_config
                    after = base_config
                elif module.params['diff_against'] in ('startup', 'running'):
                    before = base_config
                    after = running_config

                result.update({
                    'changed': True,
                    'diff': {
                        'before': str(before),
                        'after': str(after)
                    }
                })

    module.exit_json(**result)
Exemplo n.º 8
0
def main():
    """ main entry point for module execution
    """
    argument_spec = dict(count=dict(type="int"),
                         dest=dict(type="str", required=True),
                         timeout=dict(type="int"),
                         ttl=dict(type="int"),
                         size=dict(type="int"),
                         source=dict(type="str"),
                         state=dict(type="str",
                                    choices=["absent", "present"],
                                    default="present"),
                         vrf=dict(type="str"))

    module = AnsibleModule(argument_spec=argument_spec)

    count = module.params["count"]
    dest = module.params["dest"]
    source = module.params["source"]
    timeout = module.params["timeout"]
    ttl = module.params["ttl"]
    size = module.params["size"]
    vrf = module.params["vrf"]
    results = {}
    warnings = list()

    if warnings:
        results["warnings"] = warnings

    response = ''
    try:
        validate_parameters(module, timeout, count)
        results["commands"] = [
            build_ping(dest, count, source, timeout, ttl, size, vrf)
        ]
        ping_results = run_commands(module, commands=results["commands"])
        ping_results_list = ping_results[0].split("\n")

    except ConnectionError as exc:
        module.fail_json(msg=to_text(exc, errors='surrogate_then_replace'))

    validate_fail(module, ping_results[0])

    stats = ""
    statserror = ''
    for line in ping_results_list:
        if line.startswith('Sending'):
            statserror = line
        if line.startswith('Success'):
            stats = line
        elif line.startswith('No reply'):
            stats = statserror
        elif line.startswith('Ping self done'):
            stats = "Ping self done"

    success, rx, tx, rtt = parse_ping(stats)
    loss = abs(100 - int(success))
    results["packet_loss"] = str(loss) + "%"
    results["packets_rx"] = int(rx)
    results["packets_tx"] = int(tx)

    # Convert rtt values to int
    for k, v in rtt.items():
        if rtt[k] is not None:
            rtt[k] = int(v)

    results["rtt"] = rtt

    validate_results(module, loss, results)

    module.exit_json(**results)
Exemplo n.º 9
0
 def run(self, cmd):
     return run_commands(self.module, commands=cmd, check_rc=False)
Exemplo n.º 10
0
 def populate(self):
     self.responses = run_commands(self.module, commands=self.COMMANDS, check_rc=False)