Ejemplo n.º 1
0
def discover_nrpe(host_name, command, args):
    # Check if we have already checked this
    if discovery_cache['nrpe'].has_key("%s,%s,%s" % (host_name, command,args)):
        return discovery_cache['nrpe']["%s,%s,%s" % (host_name, command, args)]

    ret = network_scan.runCommand("check_nrpe -H %s -c %s %s" % (host_name, command, args))
    if ret[0] == 0:
        discovery_cache['nrpe']["%s,%s,%s" % (host_name, command, args)] = True
        return True
    discovery_cache['nrpe']["%s,%s,%s" % (host_name, command, args)] = False
    return False
Ejemplo n.º 2
0
def run_discovery(host_name, packs, **kwargs):

    # Reset cache
    discovery_cache['inet'] = {}
    discovery_cache['nrpe'] = {}

    discovered = {}
    # Go through all the packs
    for p in packs:
        discovered[p] = []
        # And loop through the discovery methods
        for d in packs[p]['discovery']:
            # We have a builtin method
            if d[0]:
                method, args = d[0].split(' ', 1)
                # Inet scanner invoked
                if method == 'inet':
                    # Port open
                    if discover_inet(host_name, args):
                        # Append affected services
                        for package in d[2:]:
                            discovered[p].append(package)
                elif method == 'plugin':
                    command = args
                    for old_string,new_string in kwargs.items():
                        command = command.replace(old_string,new_string)
                    ret, stdout, stderr = network_scan.runCommand(command)
                    if ret == 0:
                        for package in d[2:]:
                            discovered[p].append(package)
                elif method == 'nrpe':
                    for old_string,new_string in kwargs.items():
                            args = args.replace(old_string,new_string)
                    try:
                        command, nrpe_args = args.split(None, 1)
                    except ValueError:
                        command = args
                        nrpe_args = ''
                    if discover_nrpe(host_name, command, nrpe_args):
                        for package in d[2:]:
                            discovered[p].append(package)
                else:
                    raise Exception('Unsupported discovery method "%s"' % method)
            elif d[1]:
                raise Exception('shell discovery not implemented')
            else:
                raise Exception('No discovery defined')

    return discovered