add some initial XML processing for C2S

This commit is contained in:
Jan Dittberner 2015-02-10 03:20:31 +01:00
parent 48e0593246
commit 9edc6e0c3e

View file

@ -1,9 +1,56 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import absolute_import
from datetime import datetime
import socket
from xml.sax import make_parser
from xml.sax.handler import ContentHandler, feature_namespaces
import nagiosplugin
import socket
from datetime import datetime
NS_XMPP_SASL = 'urn:ietf:params:xml:ns:xmpp-sasl'
NS_XMPP_TLS = 'urn:ietf:params:xml:ns:xmpp-tls'
NS_XMPP_CAPS = 'http://jabber.org/protocol/caps'
class XmppException(Exception):
"""
Custom exception class.
"""
class XmppClientServerResponseHandler(ContentHandler):
seen_elements = set()
mechanisms = []
starttls = False
capabilities = {}
inelem = []
level = 0
def startElementNS(self, name, qname, attrs):
self.inelem.append(name)
self.seen_elements.add(name)
if name == (NS_XMPP_TLS, 'starttls'):
self.starttls = True
elif name == (NS_XMPP_CAPS, 'c'):
for qname in attrs.getQNames():
self.capabilities[qname] = attrs.getValueByQName(qname)
def endElementNS(self, name, qname):
del self.inelem[-1]
def characters(self, content):
if self.inelem[-1] == (NS_XMPP_SASL, 'mechanism'):
self.mechanisms.append(content)
else:
print self.inelem, content
def is_valid_start(self):
return True # TODO: some real implementation
class Xmpp(nagiosplugin.Resource):
@ -18,11 +65,28 @@ class Xmpp(nagiosplugin.Resource):
self.is_server = is_server
self.starttls = starttls
self.servername = servername
self.parser = make_parser()
self.parser.setFeature(feature_namespaces, True)
if self.is_server:
pass # TODO: make server parser
else:
self.contenthandler = XmppClientServerResponseHandler()
self.parser.setContentHandler(self.contenthandler)
def probe(self):
start = datetime.now()
try:
for res in self.get_addrinfo():
def get_addrinfo(self):
if self.ipv6 is None:
addrfamily = 0
elif self.ipv6 is True:
addrfamily = socket.AF_INET6
else:
addrfamily = socket.AF_INET
return socket.getaddrinfo(
self.address, self.port, addrfamily, socket.SOCK_STREAM,
socket.IPPROTO_TCP)
self.result = nagiosplugin.Critical
def open_socket(self, addrinfo):
for res in addrinfo:
af, socktype, proto, canonname, sa = res
try:
s = socket.socket(af, socktype, proto)
@ -37,30 +101,54 @@ class Xmpp(nagiosplugin.Resource):
continue
break
if s is None:
self.state = nagiosplugin.Critical
self.cause = 'could not open socket'
return nagiosplugin.Metric("time", "unknown")
s.close()
raise XmppException("could not open socket")
return s
def handle_server(self, xmppsocket):
pass
def is_valid_client_response(self, xmldata):
self.parser.feed(xmldata)
return self.contenthandler.is_valid_start()
def handle_client(self, xmppsocket):
xmppsocket.sendall((
"<?xml version='1.0' ?><stream:stream to='{servername}' "
"xmlns='jabber:client' "
"xmlns:stream='http://etherx.jabber.org/streams' "
"version='1.0'>"
).format(servername=self.servername))
if not self.is_valid_client_response(xmppsocket.recv(4096)):
raise XmppException("no valid response to XMPP client request")
xmppsocket.sendall("</stream:stream>")
self.parser.feed(xmppsocket.recv(4096))
def probe(self):
start = datetime.now()
try:
addrinfo = self.get_addrinfo()
xmppsocket = self.open_socket(addrinfo)
try:
if self.is_server:
self.handle_server(xmppsocket)
else:
self.handle_client(xmppsocket)
finally:
xmppsocket.close()
self.parser.close()
except socket.gaierror as e:
self.state = nagiosplugin.Critical
self.cause = str(e)
return nagiosplugin.Metric("time", "unknown")
except XmppException as e:
self.state = nagiosplugin.Critical
self.cause = e.message
return nagiosplugin.Metric("time", "unknown")
end = datetime.now()
return nagiosplugin.Metric(
'time', (end - start).total_seconds(), 's', min=0)
def get_addrinfo(self):
if self.ipv6 is None:
addrfamily = 0
elif self.ipv6 is True:
addrfamily = socket.AF_INET6
else:
addrfamily = socket.AF_INET
return socket.getaddrinfo(
self.address, self.port, addrfamily, socket.SOCK_STREAM,
socket.IPPROTO_TCP)
self.result = nagiosplugin.Critical
class XmppContext(nagiosplugin.ScalarContext):
@ -119,7 +207,7 @@ def main():
Xmpp(**kwargs),
XmppContext('time', warning, critical)
)
check.main()
check.main(timeout=0)
if __name__ == "__main__":