icinga2/contrib/discover.py

135 lines
4.3 KiB
Python
Raw Normal View History

2012-07-08 11:37:31 +02:00
#!/usr/bin/env python
2013-10-03 20:36:44 +02:00
#/******************************************************************************
# * Icinga 2 *
# * Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org) *
2013-10-03 20:36:44 +02:00
# * *
# * This program 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 2 *
# * of the License, or (at your option) any later version. *
# * *
# * This program 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 program; if not, write to the Free Software Foundation *
# * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
# ******************************************************************************/
2012-07-08 11:37:31 +02:00
import sys
import subprocess
import socket
2013-03-08 16:02:33 +01:00
from xml.dom.minidom import parse
2012-07-08 11:37:31 +02:00
2013-03-08 16:02:33 +01:00
if len(sys.argv) < 2:
print "Syntax: %s <xml-file> [<xml-file> ...]" % (sys.argv[0])
sys.exit(1)
2013-10-07 09:35:35 +02:00
tcp_service_commands = {
'ssh': 'ssh',
'http': 'http_ip',
'https': 'https_ip',
'smtp': 'smtp',
'ssmtp': 'ssmtp'
}
2012-07-08 11:37:31 +02:00
2013-10-07 09:35:35 +02:00
udp_service_commands = {
2013-03-08 16:02:33 +01:00
'ntp': 'ntp_time',
'snmp': 'snmp-uptime'
}
hosts = {}
def process_host(host_element):
global hosts
status = "down"
for status_element in host_element.getElementsByTagName("status"):
status = status_element.getAttribute("state")
2012-07-08 11:37:31 +02:00
2013-03-08 16:02:33 +01:00
if status != "up":
return
2012-07-08 11:37:31 +02:00
2013-03-08 16:02:33 +01:00
for address_element in host_element.getElementsByTagName("address"):
if not address_element.getAttribute("addrtype") in [ "ipv4", "ipv6" ]:
2012-07-08 11:37:31 +02:00
continue
2013-03-08 16:02:33 +01:00
address = address_element.getAttribute("addr")
2012-07-08 11:37:31 +02:00
break
2013-03-08 16:02:33 +01:00
name = address
2012-07-08 11:37:31 +02:00
2013-03-08 16:02:33 +01:00
for hostname_element in host_element.getElementsByTagName("hostname"):
name = hostname_element.getAttribute("name")
2012-07-08 11:37:31 +02:00
2013-03-08 16:02:33 +01:00
try:
services = hosts[name]["services"]
except:
services = {}
2012-07-08 11:37:31 +02:00
2013-03-08 16:02:33 +01:00
for port_element in host_element.getElementsByTagName("port"):
state = "closed"
for state_element in port_element.getElementsByTagName("state"):
state = state_element.getAttribute("state")
if state != "open":
continue
port = int(port_element.getAttribute("portid"))
protocol = port_element.getAttribute("protocol")
2012-07-08 11:37:31 +02:00
try:
2012-07-08 21:24:20 +02:00
serv = socket.getservbyport(port, protocol)
2012-07-08 11:37:31 +02:00
except:
serv = str(port)
try:
2013-03-08 16:02:33 +01:00
if protocol == "tcp":
2013-10-07 09:35:35 +02:00
command = tcp_service_commands[serv]
2013-03-08 16:02:33 +01:00
elif protocol == "udp":
2013-10-07 09:35:35 +02:00
command = udp_service_commands[serv]
2013-03-08 16:02:33 +01:00
else:
raise "Unknown protocol."
except:
2013-10-07 09:35:35 +02:00
command = protocol
2013-10-07 09:35:35 +02:00
if command == "udp":
2013-03-08 16:02:33 +01:00
continue
2013-10-07 09:35:35 +02:00
services[serv] = { "command": command, "port": port }
2013-03-08 16:02:33 +01:00
hosts[name] = { "name": name, "address": address, "services": services }
def print_host(host):
print "object Host \"%s\" {" % (host["name"])
print "\timport \"discovered-host\","
print ""
print "\tmacros.address = \"%s\"," % (host["address"])
print "}"
print ""
2013-03-08 16:02:33 +01:00
for serv, service in host["services"].iteritems():
print "apply Service \"%s\" {" % (serv)
print "\timport \"discovered-service\","
2012-07-08 11:37:31 +02:00
print ""
print "\tcheck_command = \"%s\"," % (service["command"])
2012-07-08 11:37:31 +02:00
print ""
print "\tmacros.port = %s" % (service["port"])
print "}"
2013-10-07 09:35:35 +02:00
print ""
2012-07-08 11:37:31 +02:00
2013-03-08 16:02:33 +01:00
for arg in sys.argv[1:]:
# Expects XML output from 'nmap -oX'
dom = parse(arg)
for host in dom.getElementsByTagName("host"):
process_host(host)
2013-03-08 16:02:33 +01:00
for host in hosts.values():
print_host(host)