# This file is part of Gajim. # # Gajim 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; version 3 only. # # Gajim 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 Gajim. If not, see . # XEP-0030: Service Discovery import weakref import nbxmpp from nbxmpp.structs import StanzaHandler from gajim.common import app from gajim.common import helpers from gajim.common.caps_cache import muc_caps_cache from gajim.common.nec import NetworkIncomingEvent from gajim.common.modules.base import BaseModule from gajim.common.connection_handlers_events import InformationEvent class Discovery(BaseModule): def __init__(self, con): BaseModule.__init__(self, con) self.handlers = [ StanzaHandler(name='iq', callback=self._answer_disco_info, typ='get', ns=nbxmpp.NS_DISCO_INFO), StanzaHandler(name='iq', callback=self._answer_disco_items, typ='get', ns=nbxmpp.NS_DISCO_ITEMS), ] def disco_contact(self, jid, node=None): success_cb = self._con.get_module('Caps').contact_info_received self._disco(nbxmpp.NS_DISCO_INFO, jid, node, success_cb, None) def disco_items(self, jid, node=None, success_cb=None, error_cb=None): self._disco(nbxmpp.NS_DISCO_ITEMS, jid, node, success_cb, error_cb) def disco_info(self, jid, node=None, success_cb=None, error_cb=None): self._disco(nbxmpp.NS_DISCO_INFO, jid, node, success_cb, error_cb) def _disco(self, namespace, jid, node, success_cb, error_cb): if success_cb is None: raise ValueError('success_cb is required') if not app.account_is_connected(self._account): return iq = nbxmpp.Iq(typ='get', to=jid, queryNS=namespace) if node: iq.setQuerynode(node) log_str = 'Request info: %s %s' if namespace == nbxmpp.NS_DISCO_ITEMS: log_str = 'Request items: %s %s' self._log.info(log_str, jid, node or '') # Create weak references so we can pass GUI object methods weak_success_cb = weakref.WeakMethod(success_cb) if error_cb is not None: weak_error_cb = weakref.WeakMethod(error_cb) else: weak_error_cb = None self._con.connection.SendAndCallForResponse( iq, self._disco_response, {'success_cb': weak_success_cb, 'error_cb': weak_error_cb}) def _disco_response(self, _con, stanza, success_cb, error_cb): if not nbxmpp.isResultNode(stanza): if error_cb is not None: error_cb()(stanza.getFrom(), stanza.getError()) else: self._log.info('Error: %s', stanza.getError()) return from_ = stanza.getFrom() node = stanza.getQuerynode() if stanza.getQueryNS() == nbxmpp.NS_DISCO_INFO: identities, features, data, node = self.parse_info_response(stanza) success_cb()(from_, identities, features, data, node) elif stanza.getQueryNS() == nbxmpp.NS_DISCO_ITEMS: items = self.parse_items_response(stanza) success_cb()(from_, node, items) else: self._log.warning('Wrong query namespace: %s', stanza) def parse_items_response(self, stanza): payload = stanza.getQueryPayload() items = [] for item in payload: # CDATA payload is not processed, only nodes if not isinstance(item, nbxmpp.simplexml.Node): continue attr = item.getAttrs() if 'jid' not in attr: self._log.warning('No jid attr in disco items: %s', stanza) continue try: attr['jid'] = helpers.parse_jid(attr['jid']) except helpers.InvalidFormat: self._log.warning('Invalid jid attr in disco items: %s', stanza) continue items.append(attr) return items @classmethod def parse_info_response(cls, stanza): identities, features, data, node = [], [], [], None query = stanza.getTag('query') node = query.getAttr('node') if not node: node = '' childs = stanza.getQueryChildren() if not childs: childs = [] for i in childs: if i.getName() == 'identity': attr = {} for key in i.getAttrs().keys(): attr[key] = i.getAttr(key) identities.append(attr) elif i.getName() == 'feature': var = i.getAttr('var') if var: features.append(var) elif i.getName() == 'x' and i.getNamespace() == nbxmpp.NS_DATA: data.append(nbxmpp.DataForm(node=i)) return identities, features, data, node def discover_server_items(self): server = self._con.get_own_jid().getDomain() self.disco_items(server, success_cb=self._server_items_received) def _server_items_received(self, _from, _node, items): self._log.info('Server items received') for item in items: if 'node' in item: # Only disco components continue self.disco_info(item['jid'], success_cb=self._server_items_info_received) def _server_items_info_received(self, from_, *args): from_ = from_.getStripped() self._log.info('Server item info received: %s', from_) self._parse_transports(from_, *args) try: self._con.get_module('MUC').pass_disco(from_, *args) self._con.get_module('HTTPUpload').pass_disco(from_, *args) self._con.pass_bytestream_disco(from_, *args) except nbxmpp.NodeProcessed: pass app.nec.push_incoming_event( NetworkIncomingEvent('server-disco-received')) def discover_account_info(self): own_jid = self._con.get_own_jid().getStripped() self.disco_info(own_jid, success_cb=self._account_info_received) def _account_info_received(self, from_, *args): from_ = from_.getStripped() self._log.info('Account info received: %s', from_) self._con.get_module('MAM').pass_disco(from_, *args) self._con.get_module('PEP').pass_disco(from_, *args) self._con.get_module('PubSub').pass_disco(from_, *args) self._con.get_module('Bookmarks').pass_disco(from_, *args) features = args[1] if 'urn:xmpp:pep-vcard-conversion:0' in features: self._con.avatar_conversion = True def discover_server_info(self): # Calling this method starts the connect_maschine() server = self._con.get_own_jid().getDomain() self.disco_info(server, success_cb=self._server_info_received) def _server_info_received(self, from_, *args): self._log.info('Server info received: %s', from_) self._con.get_module('SecLabels').pass_disco(from_, *args) self._con.get_module('Blocking').pass_disco(from_, *args) self._con.get_module('VCardTemp').pass_disco(from_, *args) self._con.get_module('Carbons').pass_disco(from_, *args) self._con.get_module('PrivacyLists').pass_disco(from_, *args) self._con.get_module('HTTPUpload').pass_disco(from_, *args) features = args[1] if nbxmpp.NS_REGISTER in features: self._con.register_supported = True if nbxmpp.NS_ADDRESS in features: self._con.addressing_supported = True self._con.connect_machine(restart=True) def _parse_transports(self, from_, identities, _features, _data, _node): for identity in identities: category = identity.get('category') if category not in ('gateway', 'headline'): continue transport_type = identity.get('type') self._log.info('Found transport: %s %s %s', from_, category, transport_type) jid = str(from_) if jid not in app.transport_type: app.transport_type[jid] = transport_type app.logger.save_transport_type(jid, transport_type) if transport_type in self._con.available_transports: self._con.available_transports[transport_type].append(jid) else: self._con.available_transports[transport_type] = [jid] def _answer_disco_items(self, _con, stanza, _properties): from_ = stanza.getFrom() self._log.info('Answer disco items to %s', from_) if self._con.get_module('AdHocCommands').command_items_query(stanza): raise nbxmpp.NodeProcessed node = stanza.getTagAttr('query', 'node') if node is None: result = stanza.buildReply('result') self._con.connection.send(result) raise nbxmpp.NodeProcessed if node == nbxmpp.NS_COMMANDS: self._con.get_module('AdHocCommands').command_list_query(stanza) raise nbxmpp.NodeProcessed def _answer_disco_info(self, _con, stanza, _properties): from_ = stanza.getFrom() self._log.info('Answer disco info %s', from_) if str(from_).startswith('echo.'): # Service that echos all stanzas, ignore it raise nbxmpp.NodeProcessed if self._con.get_module('AdHocCommands').command_info_query(stanza): raise nbxmpp.NodeProcessed node = stanza.getQuerynode() iq = stanza.buildReply('result') query = iq.setQuery() if node: query.setAttr('node', node) query.addChild('identity', attrs=app.gajim_identity) client_version = 'http://gajim.org#' + app.caps_hash[self._account] if node in (None, client_version): for feature in app.gajim_common_features: query.addChild('feature', attrs={'var': feature}) for feature in app.gajim_optional_features[self._account]: query.addChild('feature', attrs={'var': feature}) self._con.connection.send(iq) raise nbxmpp.NodeProcessed def disco_muc(self, jid, callback, update=False): if not app.account_is_connected(self._account): return if muc_caps_cache.is_cached(jid) and not update: callback() return iq = nbxmpp.Iq(typ='get', to=jid, queryNS=nbxmpp.NS_DISCO_INFO) self._log.info('Request MUC info %s', jid) self._con.connection.SendAndCallForResponse( iq, self._muc_info_response, {'callback': callback}) def _muc_info_response(self, _con, stanza, callback): if not nbxmpp.isResultNode(stanza): error = stanza.getError() if error == 'item-not-found': # Groupchat does not exist self._log.info('MUC does not exist: %s', stanza.getFrom()) callback() else: self._log.info('MUC disco error: %s', error) app.nec.push_incoming_event( InformationEvent( None, dialog_name='unable-join-groupchat', args=error)) return self._log.info('MUC info received: %s', stanza.getFrom()) muc_caps_cache.append(stanza) callback() def get_instance(*args, **kwargs): return Discovery(*args, **kwargs), 'Discovery'