summaryrefslogtreecommitdiff
path: root/sleekxmpp/componentxmpp.py
diff options
context:
space:
mode:
Diffstat (limited to 'sleekxmpp/componentxmpp.py')
-rwxr-xr-xsleekxmpp/componentxmpp.py98
1 files changed, 98 insertions, 0 deletions
diff --git a/sleekxmpp/componentxmpp.py b/sleekxmpp/componentxmpp.py
new file mode 100755
index 00000000..72111e3c
--- /dev/null
+++ b/sleekxmpp/componentxmpp.py
@@ -0,0 +1,98 @@
+#!/usr/bin/python2.5
+
+"""
+ SleekXMPP: The Sleek XMPP Library
+ Copyright (C) 2007 Nathanael C. Fritz
+ This file is part of SleekXMPP.
+
+ SleekXMPP 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.
+
+ SleekXMPP 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 SleekXMPP; if not, write to the Free Software
+ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+"""
+from __future__ import absolute_import
+from . basexmpp import basexmpp
+from xml.etree import cElementTree as ET
+from . xmlstream.xmlstream import XMLStream
+from . xmlstream.xmlstream import RestartStream
+from . xmlstream.matcher.xmlmask import MatchXMLMask
+from . xmlstream.matcher.xpath import MatchXPath
+from . xmlstream.matcher.many import MatchMany
+from . xmlstream.handler.callback import Callback
+from . xmlstream.stanzabase import StanzaBase
+from . xmlstream import xmlstream as xmlstreammod
+import time
+import logging
+import base64
+import sys
+import random
+import copy
+from . import plugins
+from . import stanza
+import sha
+srvsupport = True
+try:
+ import dns.resolver
+except ImportError:
+ srvsupport = False
+
+
+class ComponentXMPP(basexmpp, XMLStream):
+ """SleekXMPP's client class. Use only for good, not evil."""
+
+ def __init__(self, jid, secret, host, port, plugin_config = {}, plugin_whitelist=[]):
+ XMLStream.__init__(self)
+ self.default_ns = 'jabber:component:accept'
+ basexmpp.__init__(self)
+ self.auto_authorize = None
+ self.stream_header = "<stream:stream xmlns='jabber:component:accept' xmlns:stream='http://etherx.jabber.org/streams' to='%s'>" % jid
+ self.stream_footer = "</stream:stream>"
+ self.server_host = host
+ self.server_port = port
+ self.set_jid(jid)
+ self.secret = secret
+ self.registerHandler(Callback('PresenceProbe', MatchXMLMask("<presence xmlns='%s' type='probe'/>" % self.default_ns), self._handlePresenceProbe))
+ self.registerHandler(Callback('Handshake', MatchXPath('{jabber:component:accept}handshake'), self._handleHandshake))
+ self.registerHandler(Callback('PresenceSubscription', MatchMany(\
+ (MatchXMLMask("<presence xmlns='%s' type='subscribe'/>" % self.default_ns), \
+ MatchXMLMask("<presence xmlns='%s' type='subscribed'/>" % self.default_ns), \
+ MatchXMLMask("<presence xmlns='%s' type='unsubscribe'/>" % self.default_ns), \
+ MatchXMLMask("<presence xmlns='%s' type='unsubscribed'/>" % self.default_ns) \
+ )), self._handlePresenceSubscription))
+
+ def _handlePresenceProbe(self, stanza):
+ xml = stanza.xml
+ self.event("got_presence_probe", ({
+ 'from': xml.attrib['from'],
+ 'to': xml.attrib['to']
+ }))
+
+ def _handlePresenceSubscription(self, presence):
+ xml = presence.xml
+ self.event("changed_subscription", {
+ 'type' : xml.attrib['type'],
+ 'from': xml.attrib['from'],
+ 'to': xml.attrib['to']
+ })
+
+ def start_stream_handler(self, xml):
+ sid = xml.get('id', '')
+ handshake = ET.Element('{jabber:component:accept}handshake')
+ handshake.text = sha.new(u"%s%s" % (sid, self.secret)).hexdigest().lower()
+ self.send(handshake)
+
+ def _handleHandshake(self, xml):
+ self.event("session_start")
+
+ def connect(self):
+ logging.debug("Connecting to %s:%s" % (self.server_host, self.server_port))
+ return xmlstreammod.XMLStream.connect(self, self.server_host, self.server_port)