changeset 4062:18719058a914

plugin XEP-0294: "Jingle RTP Feedback Negotiation" implementation: rel 438
author Goffi <goffi@goffi.org>
date Tue, 30 May 2023 17:58:43 +0200
parents fddd76dedc97
children e12936318177
files sat/plugins/plugin_xep_0294.py
diffstat 1 files changed, 253 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/sat/plugins/plugin_xep_0294.py	Tue May 30 17:58:43 2023 +0200
@@ -0,0 +1,253 @@
+#!/usr/bin/env python3
+
+# Libervia plugin
+# Copyright (C) 2009-2023 Jérôme Poisson (goffi@goffi.org)
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program 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 Affero General Public License for more details.
+
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from typing import Dict, List, Optional, Union
+
+from twisted.words.protocols.jabber.xmlstream import XMPPHandler
+from twisted.words.xish import domish
+from wokkel import disco, iwokkel
+from zope.interface import implementer
+
+from sat.core import exceptions
+from sat.core.constants import Const as C
+from sat.core.i18n import _
+from sat.core.log import getLogger
+
+log = getLogger(__name__)
+
+NS_JINGLE_RTP_HDREXT = "urn:xmpp:jingle:apps:rtp:rtp-hdrext:0"
+
+PLUGIN_INFO = {
+    C.PI_NAME: "Jingle RTP Header Extensions Negotiation",
+    C.PI_IMPORT_NAME: "XEP-0294",
+    C.PI_TYPE: "XEP",
+    C.PI_MODES: C.PLUG_MODE_BOTH,
+    C.PI_PROTOCOLS: ["XEP-0294"],
+    C.PI_DEPENDENCIES: ["XEP-0167"],
+    C.PI_RECOMMENDATIONS: [],
+    C.PI_MAIN: "XEP_0294",
+    C.PI_HANDLER: "yes",
+    C.PI_DESCRIPTION: _("""Jingle RTP Header Extensions Negotiation"""),
+}
+
+
+class XEP_0294:
+    def __init__(self, host):
+        log.info(f"plugin {PLUGIN_INFO[C.PI_NAME]!r} initialization")
+        host.trigger.add("XEP-0167_parse_sdp_a", self._parse_sdp_a_trigger)
+        host.trigger.add(
+            "XEP-0167_generate_sdp_content", self._generate_sdp_content_trigger
+        )
+        host.trigger.add("XEP-0167_parse_description", self._parse_description_trigger)
+        host.trigger.add("XEP-0167_build_description", self._build_description_trigger)
+
+    def get_handler(self, client):
+        return XEP_0294_handler()
+
+    def _parse_extmap(self, parts: List[str], application_data: dict) -> None:
+        """Parse an individual extmap line and fill application_data accordingly"""
+        if "/" in parts[0]:
+            id_, direction = parts[0].split("/", 1)
+        else:
+            id_ = parts[0]
+            direction = None
+        uri = parts[1]
+        attributes = parts[2:]
+
+        if direction in (None, "sendrecv"):
+            senders = "both"
+        elif direction == "sendonly":
+            senders = "initiator"
+        elif direction == "recvonly":
+            senders = "responder"
+        elif direction == "inactive":
+            senders = "none"
+        else:
+            log.warning(f"invalid direction for extmap: {direction}")
+            senders = "sendrecv"
+
+        rtp_hdr_ext_data: Dict[str, Union[str, dict]] = {
+            "id": id_,
+            "uri": uri,
+            "senders": senders,
+        }
+
+        if attributes:
+            parameters = {}
+            for attribute in attributes:
+                name, *value = attribute.split("=", 1)
+                parameters[name] = value[0] if value else None
+            rtp_hdr_ext_data["parameters"] = parameters
+
+        application_data.setdefault("rtp-hdrext", {})[id_] = rtp_hdr_ext_data
+
+    def _parse_sdp_a_trigger(
+        self,
+        attribute: str,
+        parts: List[str],
+        call_data: dict,
+        metadata: dict,
+        media_type: str,
+        application_data: Optional[dict],
+        transport_data: dict,
+    ) -> None:
+        """Parse "extmap" and "extmap-allow-mixed" attributes"""
+        if attribute == "extmap":
+            if application_data is None:
+                call_data.setdefault("_extmaps", []).append(parts)
+            else:
+                self._parse_extmap(parts, application_data)
+        elif attribute == "extmap-allow-mixed":
+            if application_data is None:
+                call_data["_extmap-allow-mixed"] = True
+            else:
+                application_data["extmap-allow-mixed"] = True
+        elif (
+            application_data is not None
+            and "_extmaps" in call_data
+            and "rtp-hdrext" not in application_data
+        ):
+            extmaps = call_data.pop("_extmaps")
+            for parts in extmaps:
+                self._parse_extmap(parts, application_data)
+        elif (
+            application_data is not None
+            and "_extmap-allow-mixed" in call_data
+            and "extmap-allow-mixed" not in application_data
+        ):
+            value = call_data.pop("_extmap-allow-mixed")
+            application_data["extmap-allow-mixed"] = value
+
+    def _generate_sdp_content_trigger(
+        self,
+        session: dict,
+        local: bool,
+        idx: int,
+        content_data: dict,
+        sdp_lines: List[str],
+        application_data: dict,
+        app_data_key: str,
+        media_data: dict,
+        media: str,
+    ) -> None:
+        """Generate "extmap" and "extmap-allow-mixed" attributes"""
+        rtp_hdrext_dict = media_data.get("rtp-hdrext", {})
+
+        for id_, ext_data in rtp_hdrext_dict.items():
+            senders = ext_data.get("senders")
+            if senders in (None, "both"):
+                direction = "sendrecv"
+            elif senders == "initiator":
+                direction = "sendonly"
+            elif senders == "responder":
+                direction = "recvonly"
+            elif senders == "none":
+                direction = "inactive"
+            else:
+                raise exceptions.InternalError(
+                    f"Invalid senders value for extmap: {ext_data.get('senders')}"
+                )
+
+            parameters_str = ""
+            if "parameters" in ext_data:
+                parameters_str = " " + " ".join(
+                    f"{k}={v}" if v is not None else f"{k}"
+                    for k, v in ext_data["parameters"].items()
+                )
+
+            sdp_lines.append(
+                f"a=extmap:{id_}/{direction} {ext_data['uri']}{parameters_str}"
+            )
+
+        if media_data.get("extmap-allow-mixed", False):
+            sdp_lines.append("a=extmap-allow-mixed")
+
+    def _parse_description_trigger(
+        self, desc_elt: domish.Element, media_data: dict
+    ) -> None:
+        """Parse the <rtp-hdrext> and <extmap-allow-mixed> elements"""
+        for rtp_hdrext_elt in desc_elt.elements(NS_JINGLE_RTP_HDREXT, "rtp-hdrext"):
+            id_ = rtp_hdrext_elt["id"]
+            uri = rtp_hdrext_elt["uri"]
+            senders = rtp_hdrext_elt.getAttribute("senders", "both")
+            # FIXME: workaround for Movim bug https://github.com/movim/movim/issues/1212
+            if senders in ("sendonly", "recvonly", "sendrecv", "inactive"):
+                log.warning("Movim bug workaround for wrong extmap value")
+                if senders == "sendonly":
+                    senders = "initiator"
+                elif senders == "recvonly":
+                    senders = "responder"
+                elif senders == "sendrecv":
+                    senders = "both"
+                else:
+                    senders = "none"
+
+            media_data.setdefault("rtp-hdrext", {})[id_] = {
+                "id": id_,
+                "uri": uri,
+                "senders": senders,
+            }
+
+            parameters = {}
+            for param_elt in rtp_hdrext_elt.elements(NS_JINGLE_RTP_HDREXT, "parameter"):
+                try:
+                    parameters[param_elt["name"]] = param_elt.getAttribute("value")
+                except KeyError:
+                    log.warning(f"invalid parameters (missing name): {param_elt.toXml()}")
+
+            if parameters:
+                media_data["rtp-hdrext"][id_]["parameters"] = parameters
+
+        try:
+            next(desc_elt.elements(NS_JINGLE_RTP_HDREXT, "extmap-allow-mixed"))
+        except StopIteration:
+            pass
+        else:
+            media_data["extmap-allow-mixed"] = True
+
+    def _build_description_trigger(
+        self, desc_elt: domish.Element, media_data: dict, session: dict
+    ) -> None:
+        """Build the <rtp-hdrext> and <extmap-allow-mixed> elements if possible"""
+        for id_, hdrext_data in media_data.get("rtp-hdrext", {}).items():
+            rtp_hdrext_elt = desc_elt.addElement((NS_JINGLE_RTP_HDREXT, "rtp-hdrext"))
+            rtp_hdrext_elt["id"] = id_
+            rtp_hdrext_elt["uri"] = hdrext_data["uri"]
+            senders = hdrext_data.get("senders", "both")
+            if senders != "both":
+                # we must not set "both" senders otherwise calls will fail with Movim due
+                # to https://github.com/movim/movim/issues/1213
+                rtp_hdrext_elt["senders"] = senders
+
+            for name, value in hdrext_data.get("parameters", {}).items():
+                param_elt = rtp_hdrext_elt.addElement((NS_JINGLE_RTP_HDREXT, "parameter"))
+                param_elt["name"] = name
+                if value is not None:
+                    param_elt["value"] = value
+
+        if media_data.get("extmap-allow-mixed", False):
+            desc_elt.addElement((NS_JINGLE_RTP_HDREXT, "extmap-allow-mixed"))
+
+
+@implementer(iwokkel.IDisco)
+class XEP_0294_handler(XMPPHandler):
+    def getDiscoInfo(self, requestor, target, nodeIdentifier=""):
+        return [disco.DiscoFeature(NS_JINGLE_RTP_HDREXT)]
+
+    def getDiscoItems(self, requestor, target, nodeIdentifier=""):
+        return []