Merge branch 'stream_features'
This commit is contained in:
commit
cebc798e72
4 changed files with 27 additions and 10 deletions
|
@ -212,6 +212,10 @@ class BaseXMPP(XMLStream):
|
|||
self.stream_version = xml.get('version', '')
|
||||
self.peer_default_lang = xml.get('{%s}lang' % XML_NS, None)
|
||||
|
||||
if not self.is_component and not self.stream_version:
|
||||
log.warning('Legacy XMPP 0.9 protocol detected.')
|
||||
self.event('legacy_protocol')
|
||||
|
||||
def process(self, *args, **kwargs):
|
||||
"""Initialize plugins and begin processing the XML stream.
|
||||
|
||||
|
|
|
@ -273,6 +273,8 @@ class ClientXMPP(BaseXMPP):
|
|||
# Don't continue if the feature requires
|
||||
# restarting the XML stream.
|
||||
return True
|
||||
log.debug('Finished processing stream features.')
|
||||
self.event('stream_negotiated')
|
||||
|
||||
def _handle_roster(self, iq):
|
||||
"""Update the roster after receiving a roster stanza.
|
||||
|
|
|
@ -44,7 +44,7 @@ class FeatureMechanisms(BasePlugin):
|
|||
}
|
||||
|
||||
def plugin_init(self):
|
||||
if not self.use_mech and not self.xmpp.boundjid.user:
|
||||
if not self.use_mech and not self.xmpp.requested_jid.user:
|
||||
self.use_mech = 'ANONYMOUS'
|
||||
|
||||
if self.sasl_callback is None:
|
||||
|
|
|
@ -44,16 +44,27 @@ class XEP_0078(BasePlugin):
|
|||
restart=False,
|
||||
order=self.order)
|
||||
|
||||
self.xmpp.add_event_handler('legacy_protocol',
|
||||
self._handle_legacy_protocol)
|
||||
|
||||
register_stanza_plugin(Iq, stanza.IqAuth)
|
||||
register_stanza_plugin(StreamFeatures, stanza.AuthFeature)
|
||||
|
||||
def plugin_end(self):
|
||||
self.xmpp.del_event_handler('legacy_protocol',
|
||||
self._handle_legacy_protocol)
|
||||
self.xmpp.unregister_feature('auth', self.order)
|
||||
|
||||
def _handle_auth(self, features):
|
||||
# If we can or have already authenticated with SASL, do nothing.
|
||||
if 'mechanisms' in features['features']:
|
||||
return False
|
||||
return self.authenticate()
|
||||
|
||||
def _handle_legacy_protocol(self, event):
|
||||
self.authenticate()
|
||||
|
||||
def authenticate(self):
|
||||
if self.xmpp.authenticated:
|
||||
return False
|
||||
|
||||
|
@ -62,13 +73,13 @@ class XEP_0078(BasePlugin):
|
|||
# Step 1: Request the auth form
|
||||
iq = self.xmpp.Iq()
|
||||
iq['type'] = 'get'
|
||||
iq['to'] = self.xmpp.boundjid.host
|
||||
iq['auth']['username'] = self.xmpp.boundjid.user
|
||||
iq['to'] = self.xmpp.requested_jid.host
|
||||
iq['auth']['username'] = self.xmpp.requested_jid.user
|
||||
|
||||
try:
|
||||
resp = iq.send(now=True)
|
||||
except IqError:
|
||||
log.info("Authentication failed: %s", resp['error']['condition'])
|
||||
except IqError as err:
|
||||
log.info("Authentication failed: %s", err.iq['error']['condition'])
|
||||
self.xmpp.event('failed_auth', direct=True)
|
||||
self.xmpp.disconnect()
|
||||
return True
|
||||
|
@ -81,11 +92,11 @@ class XEP_0078(BasePlugin):
|
|||
# Step 2: Fill out auth form for either password or digest auth
|
||||
iq = self.xmpp.Iq()
|
||||
iq['type'] = 'set'
|
||||
iq['auth']['username'] = self.xmpp.boundjid.user
|
||||
iq['auth']['username'] = self.xmpp.requested_jid.user
|
||||
|
||||
# A resource is required, so create a random one if necessary
|
||||
if self.xmpp.boundjid.resource:
|
||||
iq['auth']['resource'] = self.xmpp.boundjid.resource
|
||||
if self.xmpp.requested_jid.resource:
|
||||
iq['auth']['resource'] = self.xmpp.requested_jid.resource
|
||||
else:
|
||||
iq['auth']['resource'] = '%s' % random.random()
|
||||
|
||||
|
@ -109,12 +120,12 @@ class XEP_0078(BasePlugin):
|
|||
result = iq.send(now=True)
|
||||
except IqError as err:
|
||||
log.info("Authentication failed")
|
||||
self.xmpp.disconnect()
|
||||
self.xmpp.event("failed_auth", direct=True)
|
||||
self.xmpp.disconnect()
|
||||
except IqTimeout:
|
||||
log.info("Authentication failed")
|
||||
self.xmpp.disconnect()
|
||||
self.xmpp.event("failed_auth", direct=True)
|
||||
self.xmpp.disconnect()
|
||||
|
||||
self.xmpp.features.add('auth')
|
||||
|
||||
|
|
Loading…
Reference in a new issue