libervia-backend: sat/plugins/plugin_xep

annotate sat/plugins/plugin_xep_0260.py @ 3781:e2a1ac1afb38

plugin invitation: use `store` hint to be sure that the invitation is archived

author	Goffi <goffi@goffi.org>
date	Mon, 16 May 2022 14:20:01 +0200
parents	be6d91572633
children	524856bd7b19

rev	line source
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	1 #!/usr/bin/env python3
3137 559a625a236b fixed shebangs Goffi <goffi@goffi.org> parents: 3136 diff changeset	2
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	3
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	4 # SAT plugin for Jingle (XEP-0260)
3479 be6d91572633 date update Goffi <goffi@goffi.org> parents: 3137 diff changeset	5 # Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org)
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	6
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	7 # This program is free software: you can redistribute it and/or modify
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	8 # it under the terms of the GNU Affero General Public License as published by
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	9 # the Free Software Foundation, either version 3 of the License, or
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	10 # (at your option) any later version.
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	11
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	12 # This program is distributed in the hope that it will be useful,
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	15 # GNU Affero General Public License for more details.
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	16
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	17 # You should have received a copy of the GNU Affero General Public License
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	19
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	20 from sat.core.i18n import _
2145 33c8c4973743 core (plugins): added missing contants + use of new constants in PLUGIN_INFO Goffi <goffi@goffi.org> parents: 2144 diff changeset	21 from sat.core.constants import Const as C
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	22 from sat.core.log import getLogger
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	23
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	24 log = getLogger(__name__)
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	25 from sat.core import exceptions
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	26 from wokkel import disco, iwokkel
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	27 from zope.interface import implementer
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	28 from twisted.words.xish import domish
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	29 from twisted.words.protocols.jabber import jid
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	30 from twisted.internet import defer
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	31 import uuid
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	32
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	33 try:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	34 from twisted.words.protocols.xmlstream import XMPPHandler
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	35 except ImportError:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	36 from wokkel.subprotocols import XMPPHandler
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	37
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	38
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	39 NS_JINGLE_S5B = "urn:xmpp:jingle:transports:s5b:1"
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	40
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	41 PLUGIN_INFO = {
2145 33c8c4973743 core (plugins): added missing contants + use of new constants in PLUGIN_INFO Goffi <goffi@goffi.org> parents: 2144 diff changeset	42 C.PI_NAME: "Jingle SOCKS5 Bytestreams",
33c8c4973743 core (plugins): added missing contants + use of new constants in PLUGIN_INFO Goffi <goffi@goffi.org> parents: 2144 diff changeset	43 C.PI_IMPORT_NAME: "XEP-0260",
33c8c4973743 core (plugins): added missing contants + use of new constants in PLUGIN_INFO Goffi <goffi@goffi.org> parents: 2144 diff changeset	44 C.PI_TYPE: "XEP",
2504 67cc54b01a12 plugin file sharing component: first draft: Goffi <goffi@goffi.org> parents: 2489 diff changeset	45 C.PI_MODES: C.PLUG_MODE_BOTH,
2145 33c8c4973743 core (plugins): added missing contants + use of new constants in PLUGIN_INFO Goffi <goffi@goffi.org> parents: 2144 diff changeset	46 C.PI_PROTOCOLS: ["XEP-0260"],
33c8c4973743 core (plugins): added missing contants + use of new constants in PLUGIN_INFO Goffi <goffi@goffi.org> parents: 2144 diff changeset	47 C.PI_DEPENDENCIES: ["XEP-0166", "XEP-0065"],
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	48 C.PI_RECOMMENDATIONS: ["XEP-0261"], # needed for fallback
2145 33c8c4973743 core (plugins): added missing contants + use of new constants in PLUGIN_INFO Goffi <goffi@goffi.org> parents: 2144 diff changeset	49 C.PI_MAIN: "XEP_0260",
33c8c4973743 core (plugins): added missing contants + use of new constants in PLUGIN_INFO Goffi <goffi@goffi.org> parents: 2144 diff changeset	50 C.PI_HANDLER: "yes",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	51 C.PI_DESCRIPTION: _("""Implementation of Jingle SOCKS5 Bytestreams"""),
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	52 }
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	53
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	54
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	55 class ProxyError(Exception):
1756 061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	56 def __str__(self):
061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	57 return "an error happened while trying to use the proxy"
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	58
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	59
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	60 class XEP_0260(object):
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	61 # TODO: udp handling
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	62
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	63 def __init__(self, host):
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	64 log.info(_("plugin Jingle SOCKS5 Bytestreams"))
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	65 self.host = host
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	66 self._j = host.plugins["XEP-0166"] # shortcut to access jingle
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	67 self._s5b = host.plugins["XEP-0065"] # and socks5 bytestream
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	68 try:
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	69 self._jingle_ibb = host.plugins["XEP-0261"]
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	70 except KeyError:
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	71 self._jingle_ibb = None
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	72 self._j.registerTransport(NS_JINGLE_S5B, self._j.TRANSPORT_STREAMING, self, 100)
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	73
2144 1d3f73e065e1 core, jp: component handling + client handling refactoring: Goffi <goffi@goffi.org> parents: 1934 diff changeset	74 def getHandler(self, client):
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	75 return XEP_0260_handler()
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	76
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	77 def _parseCandidates(self, transport_elt):
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	78 """Parse <candidate> elements
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	79
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	80 @param transport_elt(domish.Element): parent <transport> element
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	81 @return (list[plugin_xep_0065.Candidate): list of parsed candidates
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	82 """
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	83 candidates = []
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	84 for candidate_elt in transport_elt.elements(NS_JINGLE_S5B, "candidate"):
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	85 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	86 cid = candidate_elt["cid"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	87 host = candidate_elt["host"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	88 jid_ = jid.JID(candidate_elt["jid"])
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	89 port = int(candidate_elt.getAttribute("port", 1080))
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	90 priority = int(candidate_elt["priority"])
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	91 type_ = candidate_elt.getAttribute("type", self._s5b.TYPE_DIRECT)
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	92 except (KeyError, ValueError):
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	93 raise exceptions.DataError()
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	94 candidate = self._s5b.Candidate(host, port, type_, priority, jid_, cid)
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	95 candidates.append(candidate)
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	96 # self._s5b.registerCandidate(candidate)
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	97 return candidates
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	98
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	99 def _buildCandidates(self, session, candidates, sid, session_hash, client, mode=None):
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	100 """Build <transport> element with candidates
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	101
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	102 @param session(dict): jingle session data
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	103 @param candidates(iterator[plugin_xep_0065.Candidate]): iterator of candidates to add
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	104 @param sid(unicode): transport stream id
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	105 @param client: %(doc_client)s
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	106 @param mode(str, None): 'tcp' or 'udp', or None to have no attribute
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	107 @return (domish.Element): parent <transport> element where <candidate> elements must be added
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	108 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	109 proxy = next(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	110 (
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	111 candidate
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	112 for candidate in candidates
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	113 if candidate.type == self._s5b.TYPE_PROXY
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	114 ),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	115 None,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	116 )
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	117 transport_elt = domish.Element((NS_JINGLE_S5B, "transport"))
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	118 transport_elt["sid"] = sid
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	119 if proxy is not None:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	120 transport_elt["dstaddr"] = session_hash
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	121 if mode is not None:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	122 transport_elt["mode"] = "tcp" # XXX: we only manage tcp for now
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	123
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	124 for candidate in candidates:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	125 log.debug("Adding candidate: {}".format(candidate))
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	126 candidate_elt = transport_elt.addElement("candidate", NS_JINGLE_S5B)
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	127 if candidate.id is None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	128 candidate.id = str(uuid.uuid4())
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	129 candidate_elt["cid"] = candidate.id
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	130 candidate_elt["host"] = candidate.host
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	131 candidate_elt["jid"] = candidate.jid.full()
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	132 candidate_elt["port"] = str(candidate.port)
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	133 candidate_elt["priority"] = str(candidate.priority)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	134 candidate_elt["type"] = candidate.type
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	135 return transport_elt
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	136
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	137 @defer.inlineCallbacks
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	138 def jingleSessionInit(self, client, session, content_name):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	139 content_data = session["contents"][content_name]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	140 transport_data = content_data["transport_data"]
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	141 sid = transport_data["sid"] = str(uuid.uuid4())
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	142 session_hash = transport_data["session_hash"] = self._s5b.getSessionHash(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	143 session["local_jid"], session["peer_jid"], sid
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	144 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	145 transport_data["peer_session_hash"] = self._s5b.getSessionHash(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	146 session["peer_jid"], session["local_jid"], sid
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	147 ) # requester and target are inversed for peer candidates
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	148 transport_data["stream_d"] = self._s5b.registerHash(client, session_hash, None)
2927 69e4716d6268 plugins (jingle) file transfer: use initial "from" attribute as local jid instead of client.jid: Goffi <goffi@goffi.org> parents: 2771 diff changeset	149 candidates = transport_data["candidates"] = yield self._s5b.getCandidates(
69e4716d6268 plugins (jingle) file transfer: use initial "from" attribute as local jid instead of client.jid: Goffi <goffi@goffi.org> parents: 2771 diff changeset	150 client, session["local_jid"])
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	151 mode = "tcp" # XXX: we only manage tcp for now
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	152 transport_elt = self._buildCandidates(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	153 session, candidates, sid, session_hash, client, mode
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	154 )
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	155
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	156 defer.returnValue(transport_elt)
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	157
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	158 def _proxyActivatedCb(self, iq_result_elt, client, candidate, session, content_name):
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	159 """Called when activation confirmation has been received from proxy
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	160
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	161 cf XEP-0260 § 2.4
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	162 """
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	163 # now that the proxy is activated, we have to inform other peer
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	164 iq_elt, transport_elt = self._j.buildAction(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	165 client, self._j.A_TRANSPORT_INFO, session, content_name
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	166 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	167 activated_elt = transport_elt.addElement("activated")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	168 activated_elt["cid"] = candidate.id
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	169 iq_elt.send()
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	170
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	171 def _proxyActivatedEb(self, stanza_error, client, candidate, session, content_name):
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	172 """Called when activation error has been received from proxy
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	173
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	174 cf XEP-0260 § 2.4
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	175 """
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	176 # TODO: fallback to IBB
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	177 # now that the proxy is activated, we have to inform other peer
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	178 iq_elt, transport_elt = self._j.buildAction(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	179 client, self._j.A_TRANSPORT_INFO, session, content_name
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	180 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	181 transport_elt.addElement("proxy-error")
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	182 iq_elt.send()
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	183 log.warning(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	184 "Can't activate proxy, we need to fallback to IBB: {reason}".format(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	185 reason=stanza_error.value.condition
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	186 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	187 )
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	188 self.doFallback(session, content_name, client)
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	189
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	190 def _foundPeerCandidate(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	191 self, candidate, session, transport_data, content_name, client
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	192 ):
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	193 """Called when the best candidate from other peer is found
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	194
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	195 @param candidate(XEP_0065.Candidate, None): selected candidate,
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	196 or None if no candidate is accessible
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	197 @param session(dict): session data
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	198 @param transport_data(dict): transport data
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	199 @param content_name(unicode): name of the current content
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	200 @param client(unicode): %(doc_client)s
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	201 """
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	202
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	203 transport_data["best_candidate"] = candidate
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	204 # we need to disconnect all non selected candidates before removing them
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	205 for c in transport_data["peer_candidates"]:
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	206 if c is None or c is candidate:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	207 continue
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	208 c.discard()
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	209 del transport_data["peer_candidates"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	210 iq_elt, transport_elt = self._j.buildAction(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	211 client, self._j.A_TRANSPORT_INFO, session, content_name
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	212 )
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	213 if candidate is None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	214 log.warning("Can't connect to any peer candidate")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	215 candidate_elt = transport_elt.addElement("candidate-error")
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	216 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	217 log.info("Found best peer candidate: {}".format(str(candidate)))
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	218 candidate_elt = transport_elt.addElement("candidate-used")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	219 candidate_elt["cid"] = candidate.id
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	220 iq_elt.send() # TODO: check result stanza
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	221 self._checkCandidates(session, content_name, transport_data, client)
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	222
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	223 def _checkCandidates(self, session, content_name, transport_data, client):
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	224 """Called when a candidate has been choosed
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	225
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	226 if we have both candidates, we select one, or fallback to an other transport
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	227 @param session(dict): session data
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	228 @param content_name(unicode): name of the current content
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	229 @param transport_data(dict): transport data
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	230 @param client(unicode): %(doc_client)s
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	231 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	232 content_data = session["contents"][content_name]
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	233 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	234 best_candidate = transport_data["best_candidate"]
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	235 except KeyError:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	236 # we have not our best candidate yet
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	237 return
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	238 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	239 peer_best_candidate = transport_data["peer_best_candidate"]
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	240 except KeyError:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	241 # we have not peer best candidate yet
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	242 return
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	243
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	244 # at this point we have both candidates, it's time to choose one
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	245 if best_candidate is None or peer_best_candidate is None:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	246 choosed_candidate = best_candidate or peer_best_candidate
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	247 else:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	248 if best_candidate.priority == peer_best_candidate.priority:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	249 # same priority, we choose initiator one according to XEP-0260 §2.4 #4
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	250 log.debug(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	251 "Candidates have same priority, we select the one choosed by initiator"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	252 )
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	253 if session["initiator"] == session["local_jid"]:
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	254 choosed_candidate = best_candidate
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	255 else:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	256 choosed_candidate = peer_best_candidate
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	257 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	258 choosed_candidate = max(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	259 best_candidate, peer_best_candidate, key=lambda c: c.priority
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	260 )
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	261
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	262 if choosed_candidate is None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	263 log.warning("Socks5 negociation failed, we need to fallback to IBB")
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	264 self.doFallback(session, content_name, client)
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	265 else:
1571 c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	266 if choosed_candidate == peer_best_candidate:
c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	267 # peer_best_candidate was choosed from the candidates we have sent
c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	268 # so our_candidate is true if choosed_candidate is peer_best_candidate
c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	269 our_candidate = True
c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	270 # than also mean that best_candidate must be discarded !
c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	271 try:
c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	272 best_candidate.discard()
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	273 except AttributeError: # but it can be None
1571 c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	274 pass
c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	275 else:
c668081eba1c plugins XEP-0234, XEP-0260, XEP-0261: jingle session termination is managed by application (XEP-0234) instead of transport Goffi <goffi@goffi.org> parents: 1570 diff changeset	276 our_candidate = False
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	277
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	278 log.info(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	279 "Socks5 negociation successful, {who} candidate will be used: {candidate}".format(
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	280 who="our" if our_candidate else "other peer",
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	281 candidate=choosed_candidate,
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	282 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	283 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	284 del transport_data["best_candidate"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	285 del transport_data["peer_best_candidate"]
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	286
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	287 if choosed_candidate.type == self._s5b.TYPE_PROXY:
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	288 # the stream transfer need to wait for proxy activation
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	289 # (see XEP-0260 § 2.4)
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	290 if our_candidate:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	291 d = self._s5b.connectCandidate(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	292 client, choosed_candidate, transport_data["session_hash"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	293 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	294 d.addCallback(
2765 378188abe941 misc: replaced all "dummy" by the more conventional and readable "__" ("_" being used for gettext) Goffi <goffi@goffi.org> parents: 2624 diff changeset	295 lambda __: choosed_candidate.activate(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	296 transport_data["sid"], session["peer_jid"], client
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	297 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	298 )
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	299 args = [client, choosed_candidate, session, content_name]
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	300 d.addCallbacks(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	301 self._proxyActivatedCb, self._proxyActivatedEb, args, None, args
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	302 )
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	303 else:
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	304 # this Deferred will be called when we'll receive activation confirmation from other peer
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	305 d = transport_data["activation_d"] = defer.Deferred()
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	306 else:
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	307 d = defer.succeed(None)
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	308
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	309 if content_data["senders"] == session["role"]:
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	310 # we can now start the stream transfer (or start it after proxy activation)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	311 d.addCallback(
2765 378188abe941 misc: replaced all "dummy" by the more conventional and readable "__" ("_" being used for gettext) Goffi <goffi@goffi.org> parents: 2624 diff changeset	312 lambda __: choosed_candidate.startTransfer(
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	313 transport_data["session_hash"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	314 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	315 )
1756 061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	316 d.addErrback(self._startEb, session, content_name, client)
061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	317
061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	318 def _startEb(self, fail, session, content_name, client):
061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	319 """Called when it's not possible to start the transfer
061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	320
061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	321 Will try to fallback to IBB
061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	322 """
061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	323 try:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	324 reason = str(fail.value)
1756 061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	325 except AttributeError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	326 reason = str(fail)
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	327 log.warning("Cant start transfert, we'll try fallback method: {}".format(reason))
1756 061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	328 self.doFallback(session, content_name, client)
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	329
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	330 def _candidateInfo(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	331 self, candidate_elt, session, content_name, transport_data, client
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	332 ):
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	333 """Called when best candidate has been received from peer (or if none is working)
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	334
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	335 @param candidate_elt(domish.Element): candidate-used or candidate-error element
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	336 (see XEP-0260 §2.3)
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	337 @param session(dict): session data
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	338 @param content_name(unicode): name of the current content
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	339 @param transport_data(dict): transport data
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	340 @param client(unicode): %(doc_client)s
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	341 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	342 if candidate_elt.name == "candidate-error":
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	343 # candidate-error, no candidate worked
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	344 transport_data["peer_best_candidate"] = None
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	345 else:
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	346 # candidate-used, one candidate was choosed
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	347 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	348 cid = candidate_elt.attributes["cid"]
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	349 except KeyError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	350 log.warning("No cid found in <candidate-used>")
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	351 raise exceptions.DataError
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	352 try:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	353 candidate = next((
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	354 c for c in transport_data["candidates"] if c.id == cid
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	355 ))
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	356 except StopIteration:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	357 log.warning("Given cid doesn't correspond to any known candidate !")
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	358 raise exceptions.DataError # TODO: send an error to other peer, and use better exception
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	359 except KeyError:
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	360 # a transport-info can also be intentionaly sent too early by other peer
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	361 # but there is little probability
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	362 log.error(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	363 '"candidates" key doesn\'t exists in transport_data, it should at this point'
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	364 )
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	365 raise exceptions.InternalError
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	366 # at this point we have the candidate choosed by other peer
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	367 transport_data["peer_best_candidate"] = candidate
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	368 log.info("Other peer best candidate: {}".format(candidate))
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	369
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	370 del transport_data["candidates"]
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	371 self._checkCandidates(session, content_name, transport_data, client)
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	372
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	373 def _proxyActivationInfo(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	374 self, proxy_elt, session, content_name, transport_data, client
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	375 ):
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	376 """Called when proxy has been activated (or has sent an error)
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	377
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	378 @param proxy_elt(domish.Element): <activated/> or <proxy-error/> element
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	379 (see XEP-0260 §2.4)
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	380 @param session(dict): session data
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	381 @param content_name(unicode): name of the current content
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	382 @param transport_data(dict): transport data
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	383 @param client(unicode): %(doc_client)s
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	384 """
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	385 try:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	386 activation_d = transport_data.pop("activation_d")
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	387 except KeyError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	388 log.warning("Received unexpected transport-info for proxy activation")
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	389
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	390 if proxy_elt.name == "activated":
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	391 activation_d.callback(None)
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	392 else:
1756 061011fad5b1 plugin XEP-0260: better proxy error handling: Goffi <goffi@goffi.org> parents: 1755 diff changeset	393 activation_d.errback(ProxyError())
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	394
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	395 @defer.inlineCallbacks
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	396 def jingleHandler(self, client, action, session, content_name, transport_elt):
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	397 content_data = session["contents"][content_name]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	398 transport_data = content_data["transport_data"]
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	399
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	400 if action in (self._j.A_ACCEPTED_ACK, self._j.A_PREPARE_RESPONDER):
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	401 pass
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	402
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	403 elif action == self._j.A_SESSION_ACCEPT:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	404 # initiator side, we select a candidate in the ones sent by responder
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	405 assert "peer_candidates" not in transport_data
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	406 transport_data["peer_candidates"] = self._parseCandidates(transport_elt)
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	407
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	408 elif action == self._j.A_START:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	409 session_hash = transport_data["session_hash"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	410 peer_candidates = transport_data["peer_candidates"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	411 stream_object = content_data["stream_object"]
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	412 self._s5b.associateStreamObject(client, session_hash, stream_object)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	413 stream_d = transport_data.pop("stream_d")
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	414 stream_d.chainDeferred(content_data["finished_d"])
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	415 peer_session_hash = transport_data["peer_session_hash"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	416 d = self._s5b.getBestCandidate(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	417 client, peer_candidates, session_hash, peer_session_hash
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	418 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	419 d.addCallback(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	420 self._foundPeerCandidate, session, transport_data, content_name, client
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	421 )
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	422
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	423 elif action == self._j.A_SESSION_INITIATE:
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	424 # responder side, we select a candidate in the ones sent by initiator
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	425 # and we give our candidates
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	426 assert "peer_candidates" not in transport_data
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	427 sid = transport_data["sid"] = transport_elt["sid"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	428 session_hash = transport_data["session_hash"] = self._s5b.getSessionHash(
2927 69e4716d6268 plugins (jingle) file transfer: use initial "from" attribute as local jid instead of client.jid: Goffi <goffi@goffi.org> parents: 2771 diff changeset	429 session["local_jid"], session["peer_jid"], sid
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	430 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	431 peer_session_hash = transport_data[
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	432 "peer_session_hash"
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	433 ] = self._s5b.getSessionHash(
2927 69e4716d6268 plugins (jingle) file transfer: use initial "from" attribute as local jid instead of client.jid: Goffi <goffi@goffi.org> parents: 2771 diff changeset	434 session["peer_jid"], session["local_jid"], sid
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	435 ) # requester and target are inversed for peer candidates
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	436 peer_candidates = transport_data["peer_candidates"] = self._parseCandidates(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	437 transport_elt
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	438 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	439 stream_object = content_data["stream_object"]
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	440 stream_d = self._s5b.registerHash(client, session_hash, stream_object)
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	441 stream_d.chainDeferred(content_data["finished_d"])
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	442 d = self._s5b.getBestCandidate(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	443 client, peer_candidates, session_hash, peer_session_hash
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	444 )
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	445 d.addCallback(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	446 self._foundPeerCandidate, session, transport_data, content_name, client
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	447 )
2927 69e4716d6268 plugins (jingle) file transfer: use initial "from" attribute as local jid instead of client.jid: Goffi <goffi@goffi.org> parents: 2771 diff changeset	448 candidates = yield self._s5b.getCandidates(client, session["local_jid"])
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	449 # we remove duplicate candidates
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	450 candidates = [
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	451 candidate for candidate in candidates if candidate not in peer_candidates
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	452 ]
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	453
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	454 transport_data["candidates"] = candidates
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	455 # we can now build a new <transport> element with our candidates
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	456 transport_elt = self._buildCandidates(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	457 session, candidates, sid, session_hash, client
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	458 )
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	459
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	460 elif action == self._j.A_TRANSPORT_INFO:
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	461 # transport-info can be about candidate or proxy activation
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	462 candidate_elt = None
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	463
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	464 for method, names in (
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	465 (self._candidateInfo, ("candidate-used", "candidate-error")),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	466 (self._proxyActivationInfo, ("activated", "proxy-error")),
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	467 ):
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	468 for name in names:
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	469 try:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	470 candidate_elt = next(transport_elt.elements(NS_JINGLE_S5B, name))
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	471 except StopIteration:
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	472 continue
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	473 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	474 method(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	475 candidate_elt, session, content_name, transport_data, client
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	476 )
1570 37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	477 break
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	478
37d4be4a9fed plugins XEP-0260, XEP-0065: proxy handling: Goffi <goffi@goffi.org> parents: 1567 diff changeset	479 if candidate_elt is None:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	480 log.warning(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	481 "Unexpected transport element: {}".format(transport_elt.toXml())
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	482 )
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	483 elif action == self._j.A_DESTROY:
1758 a66d34353f34 plugin XEP-0260, XEP-0065: fixed session hash handling: Goffi <goffi@goffi.org> parents: 1757 diff changeset	484 # the transport is replaced (fallback ?), We need mainly to kill XEP-0065 session.
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	485 # note that sid argument is not necessary for sessions created by this plugin
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	486 self._s5b.killSession(None, transport_data["session_hash"], None, client)
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	487 else:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	488 log.warning("FIXME: unmanaged action {}".format(action))
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	489
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	490 defer.returnValue(transport_elt)
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	491
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	492 def jingleTerminate(self, client, action, session, content_name, reason_elt):
1755 d2e023da2983 plugin XEP-0260: kill s5b session if session is declined Goffi <goffi@goffi.org> parents: 1631 diff changeset	493 if reason_elt.decline:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	494 log.debug("Session declined, deleting S5B session")
1755 d2e023da2983 plugin XEP-0260: kill s5b session if session is declined Goffi <goffi@goffi.org> parents: 1631 diff changeset	495 # we just need to clean the S5B session if it is declined
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	496 content_data = session["contents"][content_name]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	497 transport_data = content_data["transport_data"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	498 self._s5b.killSession(None, transport_data["session_hash"], None, client)
1755 d2e023da2983 plugin XEP-0260: kill s5b session if session is declined Goffi <goffi@goffi.org> parents: 1631 diff changeset	499
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	500 def _doFallback(self, feature_checked, session, content_name, client):
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	501 """Do the fallback, method called once feature is checked
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	502
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	503 @param feature_checked(bool): True if other peer can do IBB
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	504 """
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	505 if not feature_checked:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	506 log.warning(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	507 "Other peer can't manage jingle IBB, be have to terminate the session"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	508 )
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	509 self._j.terminate(client, self._j.REASON_CONNECTIVITY_ERROR, session)
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	510 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	511 self._j.transportReplace(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	512 client, self._jingle_ibb.NAMESPACE, session, content_name
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	513 )
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	514
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	515 def doFallback(self, session, content_name, client):
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	516 """Fallback to IBB transport, used in last resort
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	517
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	518 @param session(dict): session data
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	519 @param content_name(unicode): name of the current content
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	520 @param client(unicode): %(doc_client)s
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	521 """
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	522 if session["role"] != self._j.ROLE_INITIATOR:
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	523 # only initiator must do the fallback, see XEP-0260 §3
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	524 return
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	525 if self._jingle_ibb is None:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	526 log.warning(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	527 "Jingle IBB (XEP-0261) plugin is not available, we have to close the session"
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	528 )
2489 e2a7bb875957 plugin pipe/stream, file transfert: refactoring and improvments: Goffi <goffi@goffi.org> parents: 2483 diff changeset	529 self._j.terminate(client, self._j.REASON_CONNECTIVITY_ERROR, session)
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	530 else:
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	531 d = self.host.hasFeature(
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	532 client, self._jingle_ibb.NAMESPACE, session["peer_jid"]
56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	533 )
1631 25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	534 d.addCallback(self._doFallback, session, content_name, client)
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	535 return d
25906c0dbc63 plugin XEP-0260, XEP-0261: fallback from S5B to IBB is implemented Goffi <goffi@goffi.org> parents: 1571 diff changeset	536
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	537
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2927 diff changeset	538 @implementer(iwokkel.IDisco)
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	539 class XEP_0260_handler(XMPPHandler):
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	540
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	541 def getDiscoInfo(self, requestor, target, nodeIdentifier=""):
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	542 return [disco.DiscoFeature(NS_JINGLE_S5B)]
dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	543
2624 56f94936df1e code style reformatting using black Goffi <goffi@goffi.org> parents: 2562 diff changeset	544 def getDiscoItems(self, requestor, target, nodeIdentifier=""):
1560 dcce63810733 plugin XEP-0260: first draft Goffi <goffi@goffi.org> parents: diff changeset	545 return []

Mercurial > libervia-backend

annotate sat/plugins/plugin_xep_0260.py @ 3781:e2a1ac1afb38