libervia-backend: libervia/backend/plugins/plugin_xep

annotate libervia/backend/plugins/plugin_xep_0384.py @ 4306:94e0968987cd

plugin XEP-0033: code modernisation, improve delivery, data validation: - Code has been rewritten using Pydantic models and `async` coroutines for data validation and cleaner element parsing/generation. - Delivery has been completely rewritten. It now works even if server doesn't support multicast, and send to local multicast service first. Delivering to local multicast service first is due to bad support of XEP-0033 in server (notably Prosody which has an incomplete implementation), and the current impossibility to detect if a sub-domain service handles fully multicast or only for local domains. This is a workaround to have a good balance between backward compatilibity and use of bandwith, and to make it work with the incoming email gateway implementation (the gateway will only deliver to entities of its own domain). - disco feature checking now uses `async` corountines. `host` implementation still use Deferred return values for compatibility with legacy code. rel 450

author	Goffi <goffi@goffi.org>
date	Thu, 26 Sep 2024 16:12:01 +0200
parents	23842a63ea00
children

rev	line source
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	1 #!/usr/bin/env python3
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	3 # Libervia plugin for OMEMO encryption
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	4 # Copyright (C) 2022-2022 Tim Henkes (me@syndace.dev)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	5
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	6 # This program is free software: you can redistribute it and/or modify
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	7 # it under the terms of the GNU Affero General Public License as published by
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	8 # the Free Software Foundation, either version 3 of the License, or
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	9 # (at your option) any later version.
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	10
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	11 # This program is distributed in the hope that it will be useful,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	14 # GNU Affero General Public License for more details.
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	15
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	16 # You should have received a copy of the GNU Affero General Public License
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	18
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	19 import base64
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	20 from datetime import datetime
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	21 import enum
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	22 import logging
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	23 import time
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	24 from typing import (
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	25 Any,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	26 Dict,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	27 FrozenSet,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	28 Iterable,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	29 List,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	30 Literal,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	31 NamedTuple,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	32 Optional,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	33 Set,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	34 Type,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	35 Union,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	36 cast,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	37 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	38 import uuid
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	39 import xml.etree.ElementTree as ET
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	40 from xml.sax.saxutils import quoteattr
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	41
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	42 from typing_extensions import Final, Never, assert_never
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	43 from wokkel import muc, pubsub # type: ignore[import]
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	44 import xmlschema
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	45
4071 4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	46 from libervia.backend.core import exceptions
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	47 from libervia.backend.core.constants import Const as C
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	48 from libervia.backend.core.core_types import MessageData, SatXMPPEntity
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	49 from libervia.backend.core.i18n import _, D_
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	50 from libervia.backend.core.log import getLogger, Logger
4073 7c5654c54fed refactoring: rename `core.sat_main` to `core.main` Goffi <goffi@goffi.org> parents: 4072 diff changeset	51 from libervia.backend.core.main import LiberviaBackend
4071 4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	52 from libervia.backend.core.xmpp import SatXMPPClient
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	53 from libervia.backend.memory import persistent
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	54 from libervia.backend.plugins.plugin_misc_text_commands import TextCommands
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	55 from libervia.backend.plugins.plugin_xep_0045 import XEP_0045
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	56 from libervia.backend.plugins.plugin_xep_0060 import XEP_0060
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	57 from libervia.backend.plugins.plugin_xep_0163 import XEP_0163
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	58 from libervia.backend.plugins.plugin_xep_0334 import XEP_0334
4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	59 from libervia.backend.plugins.plugin_xep_0359 import XEP_0359
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	60 from libervia.backend.plugins.plugin_xep_0420 import (
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	61 XEP_0420,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	62 SCEAffixPolicy,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	63 SCEAffixValues,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	64 SCEProfile,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	65 )
4071 4b842c1fb686 refactoring: renamed `sat` package to `libervia.backend` Goffi <goffi@goffi.org> parents: 4051 diff changeset	66 from libervia.backend.tools import xml_tools
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	67 from twisted.internet import defer
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	68 from twisted.words.protocols.jabber import error, jid
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	69 from twisted.words.xish import domish
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	70
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	71 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	72 import omemo
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	73 import omemo.identity_key_pair
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	74 import twomemo
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	75 import twomemo.etree
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	76 import oldmemo
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	77 import oldmemo.etree
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	78 import oldmemo.migrations
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	79 from xmlschema import XMLSchemaValidationError
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	80
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	81 # An explicit version check of the OMEMO libraries should not be required here, since
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	82 # the stored data is fully versioned and the library will complain if a downgrade is
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	83 # attempted.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	84 except ImportError as import_error:
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	85 raise exceptions.MissingModule(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	86 "You are missing one or more package required by the OMEMO plugin. Please"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	87 " download/install the pip packages 'omemo', 'twomemo', 'oldmemo' and"
3976 db45d49518f6 plugin XEP-0384: log `import_error` when import is failing Goffi <goffi@goffi.org> parents: 3972 diff changeset	88 f" 'xmlschema'.\nexception: {import_error}"
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	89 ) from import_error
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	90
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	91
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	92 __all__ = ["PLUGIN_INFO", "OMEMO"] # pylint: disable=unused-variable
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	93
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	94 log = cast(Logger, getLogger(__name__)) # type: ignore[no-untyped-call]
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	95
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	96
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	97 PLUGIN_INFO = {
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	98 C.PI_NAME: "OMEMO",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	99 C.PI_IMPORT_NAME: "XEP-0384",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	100 C.PI_TYPE: "SEC",
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	101 C.PI_PROTOCOLS: ["XEP-0384"],
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	102 C.PI_DEPENDENCIES: ["XEP-0163", "XEP-0280", "XEP-0334", "XEP-0060", "XEP-0420"],
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	103 C.PI_RECOMMENDATIONS: ["XEP-0045", "XEP-0359", C.TEXT_CMDS],
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	104 C.PI_MAIN: "OMEMO",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	105 C.PI_HANDLER: "no",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	106 C.PI_DESCRIPTION: _("""Implementation of OMEMO"""),
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	107 }
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	108
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	109
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	110 PARAM_CATEGORY = "Security"
b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	111 PARAM_NAME = "omemo_policy"
b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	112
4219 1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	113 NamespaceType = Literal["urn:xmpp:omemo:2", "eu.siacs.conversations.axolotl"]
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	114
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	115
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	116 class LogHandler(logging.Handler):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	117 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	118 Redirect python-omemo's log output to Libervia's log system.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	119 """
2654 e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	120
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	121 def emit(self, record: logging.LogRecord) -> None:
2654 e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	122 log.log(record.levelname, record.getMessage())
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	123
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	124
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	125 sm_logger = logging.getLogger(omemo.SessionManager.LOG_TAG)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	126 sm_logger.setLevel(logging.DEBUG)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	127 sm_logger.propagate = False
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	128 sm_logger.addHandler(LogHandler())
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	129
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	130
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	131 ikp_logger = logging.getLogger(omemo.identity_key_pair.IdentityKeyPair.LOG_TAG)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	132 ikp_logger.setLevel(logging.DEBUG)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	133 ikp_logger.propagate = False
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	134 ikp_logger.addHandler(LogHandler())
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	135
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	136
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	137 # TODO: Add handling for device labels, i.e. show device labels in the trust UI and give
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	138 # the user a way to change their own device label.
2654 e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	139
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	140
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	141 class MUCPlaintextCacheKey(NamedTuple):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	142 # pylint: disable=invalid-name
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	143 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	144 Structure identifying an encrypted message sent to a MUC.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	145 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	146
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	147 client: SatXMPPClient
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	148 room_jid: jid.JID
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	149 message_uid: str
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	150
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	151
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	152 @enum.unique
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	153 class TrustLevel(enum.Enum):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	154 """
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	155 The trust levels required for ATM and BTBV.
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	156 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	157
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	158 TRUSTED: str = "TRUSTED"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	159 BLINDLY_TRUSTED: str = "BLINDLY_TRUSTED"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	160 UNDECIDED: str = "UNDECIDED"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	161 DISTRUSTED: str = "DISTRUSTED"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	162
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	163
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	164 TWOMEMO_DEVICE_LIST_NODE = "urn:xmpp:omemo:2:devices"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	165 OLDMEMO_DEVICE_LIST_NODE = "eu.siacs.conversations.axolotl.devicelist"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	166
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	167
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	168 class StorageImpl(omemo.Storage):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	169 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	170 Storage implementation for OMEMO based on :class:`persistent.LazyPersistentBinaryDict`
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	171 """
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	172
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	173 def __init__(self, profile: str) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	174 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	175 @param profile: The profile this OMEMO data belongs to.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	176 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	177
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	178 # persistent.LazyPersistentBinaryDict does not cache at all, so keep the caching
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	179 # option of omemo.Storage enabled.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	180 super().__init__()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	181
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	182 self.__storage = persistent.LazyPersistentBinaryDict("XEP-0384", profile)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	183
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	184 async def _load(self, key: str) -> omemo.Maybe[omemo.JSONType]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	185 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	186 return omemo.Just(await self.__storage[key])
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	187 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	188 return omemo.Nothing()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	189 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	190 raise omemo.StorageException(f"Error while loading key {key}") from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	191
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	192 async def _store(self, key: str, value: omemo.JSONType) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	193 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	194 await self.__storage.force(key, value)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	195 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	196 raise omemo.StorageException(f"Error while storing key {key}: {value}") from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	197
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	198 async def _delete(self, key: str) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	199 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	200 await self.__storage.remove(key)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	201 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	202 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	203 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	204 raise omemo.StorageException(f"Error while deleting key {key}") from e
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	205
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	206
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	207 class LegacyStorageImpl(oldmemo.migrations.LegacyStorage):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	208 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	209 Legacy storage implementation to migrate data from the old XEP-0384 plugin.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	210 """
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	211
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	212 KEY_DEVICE_ID = "DEVICE_ID"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	213 KEY_STATE = "STATE"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	214 KEY_SESSION = "SESSION"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	215 KEY_ACTIVE_DEVICES = "DEVICES"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	216 KEY_INACTIVE_DEVICES = "INACTIVE_DEVICES"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	217 KEY_TRUST = "TRUST"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	218 KEY_ALL_JIDS = "ALL_JIDS"
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	219
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	220 def __init__(self, profile: str, own_bare_jid: str) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	221 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	222 @param profile: The profile this OMEMO data belongs to.
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	223 @param own_bare_jid: The own bare JID, to return by the :meth:`load_own_data` call.
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	224 """
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	225
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	226 self.__storage = persistent.LazyPersistentBinaryDict("XEP-0384", profile)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	227 self.__own_bare_jid = own_bare_jid
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	228
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	229 async def loadOwnData(self) -> Optional[oldmemo.migrations.OwnData]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	230 own_device_id = await self.__storage.get(LegacyStorageImpl.KEY_DEVICE_ID, None)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	231 if own_device_id is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	232 return None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	233
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	234 return oldmemo.migrations.OwnData(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	235 own_bare_jid=self.__own_bare_jid, own_device_id=own_device_id
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	236 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	237
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	238 async def deleteOwnData(self) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	239 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	240 await self.__storage.remove(LegacyStorageImpl.KEY_DEVICE_ID)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	241 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	242 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	243
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	244 async def loadState(self) -> Optional[oldmemo.migrations.State]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	245 return cast(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	246 Optional[oldmemo.migrations.State],
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	247 await self.__storage.get(LegacyStorageImpl.KEY_STATE, None),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	248 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	249
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	250 async def deleteState(self) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	251 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	252 await self.__storage.remove(LegacyStorageImpl.KEY_STATE)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	253 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	254 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	255
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	256 async def loadSession(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	257 self, bare_jid: str, device_id: int
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	258 ) -> Optional[oldmemo.migrations.Session]:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	259 key = "\n".join([LegacyStorageImpl.KEY_SESSION, bare_jid, str(device_id)])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	260
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	261 return cast(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	262 Optional[oldmemo.migrations.Session], await self.__storage.get(key, None)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	263 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	264
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	265 async def deleteSession(self, bare_jid: str, device_id: int) -> None:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	266 key = "\n".join([LegacyStorageImpl.KEY_SESSION, bare_jid, str(device_id)])
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	267
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	268 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	269 await self.__storage.remove(key)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	270 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	271 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	272
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	273 async def loadActiveDevices(self, bare_jid: str) -> Optional[List[int]]:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	274 key = "\n".join([LegacyStorageImpl.KEY_ACTIVE_DEVICES, bare_jid])
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	275
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	276 return cast(Optional[List[int]], await self.__storage.get(key, None))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	277
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	278 async def loadInactiveDevices(self, bare_jid: str) -> Optional[Dict[int, int]]:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	279 key = "\n".join([LegacyStorageImpl.KEY_INACTIVE_DEVICES, bare_jid])
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	280
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	281 return cast(Optional[Dict[int, int]], await self.__storage.get(key, None))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	282
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	283 async def deleteActiveDevices(self, bare_jid: str) -> None:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	284 key = "\n".join([LegacyStorageImpl.KEY_ACTIVE_DEVICES, bare_jid])
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	285
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	286 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	287 await self.__storage.remove(key)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	288 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	289 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	290
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	291 async def deleteInactiveDevices(self, bare_jid: str) -> None:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	292 key = "\n".join([LegacyStorageImpl.KEY_INACTIVE_DEVICES, bare_jid])
3672 e4054b648111 plugin XEP-0384: fix calls outside of main thread: Goffi <goffi@goffi.org> parents: 3584 diff changeset	293
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	294 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	295 await self.__storage.remove(key)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	296 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	297 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	298
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	299 async def loadTrust(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	300 self, bare_jid: str, device_id: int
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	301 ) -> Optional[oldmemo.migrations.Trust]:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	302 key = "\n".join([LegacyStorageImpl.KEY_TRUST, bare_jid, str(device_id)])
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	303
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	304 return cast(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	305 Optional[oldmemo.migrations.Trust], await self.__storage.get(key, None)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	306 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	307
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	308 async def deleteTrust(self, bare_jid: str, device_id: int) -> None:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	309 key = "\n".join([LegacyStorageImpl.KEY_TRUST, bare_jid, str(device_id)])
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	310
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	311 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	312 await self.__storage.remove(key)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	313 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	314 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	315
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	316 async def listJIDs(self) -> Optional[List[str]]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	317 bare_jids = await self.__storage.get(LegacyStorageImpl.KEY_ALL_JIDS, None)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	318
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	319 return None if bare_jids is None else list(bare_jids)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	320
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	321 async def deleteJIDList(self) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	322 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	323 await self.__storage.remove(LegacyStorageImpl.KEY_ALL_JIDS)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	324 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	325 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	326
3541 888109774673 core: various changes and fixes to work with new storage and D-Bus bridge: Goffi <goffi@goffi.org> parents: 3479 diff changeset	327
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	328 async def download_oldmemo_bundle(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	329 client: SatXMPPClient, xep_0060: XEP_0060, bare_jid: str, device_id: int
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	330 ) -> oldmemo.oldmemo.BundleImpl:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	331 """Download the oldmemo bundle corresponding to a specific device.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	332
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	333 @param client: The client.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	334 @param xep_0060: The XEP-0060 plugin instance to use for pubsub interactions.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	335 @param bare_jid: The bare JID the device belongs to.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	336 @param device_id: The id of the device.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	337 @return: The bundle.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	338 @raise BundleDownloadFailed: if the download failed. Feel free to raise a subclass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	339 instead.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	340 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	341 # Bundle downloads are needed by the session manager and for migrations from legacy,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	342 # thus it is made a separate function.
3541 888109774673 core: various changes and fixes to work with new storage and D-Bus bridge: Goffi <goffi@goffi.org> parents: 3479 diff changeset	343
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	344 namespace = oldmemo.oldmemo.NAMESPACE
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	345 node = f"eu.siacs.conversations.axolotl.bundles:{device_id}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	346
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	347 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	348 items, __ = await xep_0060.get_items(client, jid.JID(bare_jid), node, max_items=1)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	349 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	350 raise omemo.BundleDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	351 f"Bundle download failed for {bare_jid}: {device_id} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	352 f" {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	353 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	354
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	355 if len(items) != 1:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	356 raise omemo.BundleDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	357 f"Bundle download failed for {bare_jid}: {device_id} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	358 f" {namespace}: Unexpected number of items retrieved: {len(items)}."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	359 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	360
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	361 element = next(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	362 iter(xml_tools.domish_elt_2_et_elt(cast(domish.Element, items[0]))), None
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	363 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	364 if element is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	365 raise omemo.BundleDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	366 f"Bundle download failed for {bare_jid}: {device_id} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	367 f" {namespace}: Item download succeeded but parsing failed: {element}."
3541 888109774673 core: various changes and fixes to work with new storage and D-Bus bridge: Goffi <goffi@goffi.org> parents: 3479 diff changeset	368 )
888109774673 core: various changes and fixes to work with new storage and D-Bus bridge: Goffi <goffi@goffi.org> parents: 3479 diff changeset	369
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	370 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	371 return oldmemo.etree.parse_bundle(element, bare_jid, device_id)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	372 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	373 raise omemo.BundleDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	374 f"Bundle parsing failed for {bare_jid}: {device_id} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	375 f" {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	376 ) from e
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	377
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	378
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	379 # ATM only supports protocols based on SCE, which is currently only omemo:2, and relies on
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	380 # so many implementation details of the encryption protocol that it makes more sense to
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	381 # add ATM to the OMEMO plugin directly instead of having it a separate Libervia plugin.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	382 NS_TM: Final = "urn:xmpp:tm:1"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	383 NS_ATM: Final = "urn:xmpp:atm:1"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	384
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	385
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	386 TRUST_MESSAGE_SCHEMA = xmlschema.XMLSchema(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	387 """<?xml version='1.0' encoding='UTF-8'?>
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	388 <xs:schema xmlns:xs='http://www.w3.org/2001/XMLSchema'
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	389 targetNamespace='urn:xmpp:tm:1'
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	390 xmlns='urn:xmpp:tm:1'
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	391 elementFormDefault='qualified'>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	392
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	393 <xs:element name='trust-message'>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	394 <xs:complexType>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	395 <xs:sequence>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	396 <xs:element ref='key-owner' minOccurs='1' maxOccurs='unbounded'/>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	397 </xs:sequence>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	398 <xs:attribute name='usage' type='xs:string' use='required'/>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	399 <xs:attribute name='encryption' type='xs:string' use='required'/>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	400 </xs:complexType>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	401 </xs:element>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	402
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	403 <xs:element name='key-owner'>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	404 <xs:complexType>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	405 <xs:sequence>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	406 <xs:element
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	407 name='trust' type='xs:base64Binary' minOccurs='0' maxOccurs='unbounded'/>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	408 <xs:element
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	409 name='distrust' type='xs:base64Binary' minOccurs='0' maxOccurs='unbounded'/>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	410 </xs:sequence>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	411 <xs:attribute name='jid' type='xs:string' use='required'/>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	412 </xs:complexType>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	413 </xs:element>
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	414 </xs:schema>
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	415 """
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	416 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	417
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	418
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	419 # This is compatible with omemo:2's SCE profile
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	420 TM_SCE_PROFILE = SCEProfile(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	421 rpad_policy=SCEAffixPolicy.REQUIRED,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	422 time_policy=SCEAffixPolicy.REQUIRED,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	423 to_policy=SCEAffixPolicy.OPTIONAL,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	424 from_policy=SCEAffixPolicy.OPTIONAL,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	425 custom_policies={},
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	426 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	427
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	428
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	429 class TrustUpdate(NamedTuple):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	430 # pylint: disable=invalid-name
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	431 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	432 An update to the trust status of an identity key, used by Automatic Trust Management.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	433 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	434
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	435 target_jid: jid.JID
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	436 target_key: bytes
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	437 target_trust: bool
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	438
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	439 def to_dict(self) -> dict[str, Any]:
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	440 """Convert the instance to a serialised dictionary"""
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	441 data = {
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	442 "target_jid": self.target_jid.full(),
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	443 "target_key": self.target_key.hex(),
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	444 "target_trust": self.target_trust,
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	445 }
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	446 return data
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	447
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	448 @staticmethod
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	449 def from_dict(data: dict[str, Any]) -> "TrustUpdate":
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	450 """Load a serialized dictionary"""
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	451 data["target_jid"] = jid.JID(data["target_jid"])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	452 data["target_key"] = bytes.fromhex(data["target_key"])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	453 return TrustUpdate(**data)
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	454
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	455
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	456 class TrustMessageCacheEntry(NamedTuple):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	457 # pylint: disable=invalid-name
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	458 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	459 An entry in the trust message cache used by ATM.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	460 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	461
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	462 sender_jid: jid.JID
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	463 sender_key: bytes
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	464 timestamp: datetime
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	465 trust_update: TrustUpdate
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	466
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	467 def to_dict(self) -> dict[str, Any]:
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	468 """Convert the instance to a serialised dictionary"""
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	469 data = {
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	470 "sender_jid": self.sender_jid.full(),
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	471 "sender_key": self.sender_key.hex(),
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	472 "timestamp": self.timestamp.isoformat(),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	473 "trust_update": self.trust_update.to_dict(),
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	474 }
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	475 return data
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	476
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	477 @staticmethod
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	478 def from_dict(data: dict[str, Any]) -> "TrustMessageCacheEntry":
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	479 """Load a serialized dictionary"""
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	480 data["sender_jid"] = jid.JID(data["sender_jid"])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	481 data["sender_key"] = bytes.fromhex(data["sender_key"])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	482 data["timestamp"] = datetime.fromisoformat(data["timestamp"])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	483 data["trust_update"] = TrustUpdate.from_dict(data["trust_update"])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	484 return TrustMessageCacheEntry(**data)
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	485
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	486
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	487 class PartialTrustMessage(NamedTuple):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	488 # pylint: disable=invalid-name
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	489 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	490 A structure representing a partial trust message, used by :func:`send_trust_messages`
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	491 to build trust messages.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	492 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	493
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	494 recipient_jid: jid.JID
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	495 updated_jid: jid.JID
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	496 trust_updates: FrozenSet[TrustUpdate]
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	497
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	498
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	499 async def manage_trust_message_cache(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	500 client: SatXMPPClient,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	501 session_manager: omemo.SessionManager,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	502 applied_trust_updates: FrozenSet[TrustUpdate],
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	503 ) -> None:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	504 """Manage the ATM trust message cache after trust updates have been applied.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	505
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	506 @param client: The client this operation runs under.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	507 @param session_manager: The session manager to use.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	508 @param applied_trust_updates: The trust updates that have already been applied,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	509 triggering this cache management run.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	510 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	511
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	512 trust_message_cache = persistent.LazyPersistentBinaryDict(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	513 "XEP-0384/TM", client.profile
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	514 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	515
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	516 # Load cache entries
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	517 cache_entries = {
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	518 TrustMessageCacheEntry.from_dict(d)
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	519 for d in await trust_message_cache.get("cache", [])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	520 }
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	521
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	522 # Expire cache entries that were overwritten by the applied trust updates
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	523 cache_entries_by_target = {
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	524 (
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	525 cache_entry.trust_update.target_jid.userhostJID(),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	526 cache_entry.trust_update.target_key,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	527 ): cache_entry
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	528 for cache_entry in cache_entries
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	529 }
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	530
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	531 for trust_update in applied_trust_updates:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	532 cache_entry = cache_entries_by_target.get(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	533 (trust_update.target_jid.userhostJID(), trust_update.target_key), None
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	534 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	535
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	536 if cache_entry is not None:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	537 cache_entries.remove(cache_entry)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	538
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	539 # Apply cached Trust Messages by newly trusted devices
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	540 new_trust_updates: Set[TrustUpdate] = set()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	541
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	542 for trust_update in applied_trust_updates:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	543 if trust_update.target_trust:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	544 # Iterate over a copy such that cache_entries can be modified
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	545 for cache_entry in set(cache_entries):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	546 if (
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	547 cache_entry.sender_jid.userhostJID()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	548 == trust_update.target_jid.userhostJID()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	549 and cache_entry.sender_key == trust_update.target_key
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	550 ):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	551 trust_level = (
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	552 TrustLevel.TRUSTED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	553 if cache_entry.trust_update.target_trust
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	554 else TrustLevel.DISTRUSTED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	555 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	556
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	557 # Apply the trust update
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	558 await session_manager.set_trust(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	559 cache_entry.trust_update.target_jid.userhost(),
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	560 cache_entry.trust_update.target_key,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	561 trust_level.name,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	562 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	563
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	564 # Track the fact that this trust update has been applied
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	565 new_trust_updates.add(cache_entry.trust_update)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	566
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	567 # Remove the corresponding cache entry
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	568 cache_entries.remove(cache_entry)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	569
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	570 # Store the updated cache entries
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	571 await trust_message_cache.force("cache", [tm.to_dict() for tm in cache_entries])
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	572
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	573 # TODO: Notify the user ("feedback") about automatically updated trust?
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	574
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	575 if len(new_trust_updates) > 0:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	576 # If any trust has been updated, recursively perform another run of cache
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	577 # management
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	578 await manage_trust_message_cache(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	579 client, session_manager, frozenset(new_trust_updates)
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	580 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	581
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	582
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	583 async def get_trust_as_trust_updates(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	584 session_manager: omemo.SessionManager, target_jid: jid.JID
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	585 ) -> FrozenSet[TrustUpdate]:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	586 """Get the trust status of all known keys of a JID as trust updates for use with ATM.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	587
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	588 @param session_manager: The session manager to load the trust from.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	589 @param target_jid: The JID to load the trust for.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	590 @return: The trust updates encoding the trust status of all known keys of the JID that
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	591 are either explicitly trusted or distrusted. Undecided keys are not included in
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	592 the trust updates.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	593 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	594
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	595 devices = await session_manager.get_device_information(target_jid.userhost())
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	596
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	597 trust_updates: Set[TrustUpdate] = set()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	598
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	599 for device in devices:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	600 trust_level = TrustLevel(device.trust_level_name)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	601 target_trust: bool
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	602
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	603 if trust_level is TrustLevel.TRUSTED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	604 target_trust = True
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	605 elif trust_level is TrustLevel.DISTRUSTED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	606 target_trust = False
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	607 else:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	608 # Skip devices that are not explicitly trusted or distrusted
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	609 continue
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	610
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	611 trust_updates.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	612 TrustUpdate(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	613 target_jid=target_jid.userhostJID(),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	614 target_key=device.identity_key,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	615 target_trust=target_trust,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	616 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	617 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	618
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	619 return frozenset(trust_updates)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	620
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	621
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	622 async def send_trust_messages(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	623 client: SatXMPPClient,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	624 session_manager: omemo.SessionManager,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	625 applied_trust_updates: FrozenSet[TrustUpdate],
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	626 ) -> None:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	627 """Send information about updated trust to peers via ATM (XEP-0450).
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	628
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	629 @param client: The client.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	630 @param session_manager: The session manager.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	631 @param applied_trust_updates: The trust updates that have already been applied, to
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	632 notify other peers about.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	633 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	634 # NOTE: This currently sends information about oldmemo trust too. This is not
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	635 # specified and experimental, but since twomemo and oldmemo share the same identity
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	636 # keys and trust systems, this could be a cool side effect.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	637
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	638 # Send Trust Messages for newly trusted and distrusted devices
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	639 own_jid = client.jid.userhostJID()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	640 own_trust_updates = await get_trust_as_trust_updates(session_manager, own_jid)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	641
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	642 # JIDs of which at least one device's trust has been updated
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	643 updated_jids = frozenset(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	644 {trust_update.target_jid.userhostJID() for trust_update in applied_trust_updates}
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	645 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	646
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	647 trust_messages: Set[PartialTrustMessage] = set()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	648
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	649 for updated_jid in updated_jids:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	650 # Get the trust updates for that JID
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	651 trust_updates = frozenset(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	652 {
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	653 trust_update
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	654 for trust_update in applied_trust_updates
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	655 if trust_update.target_jid.userhostJID() == updated_jid
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	656 }
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	657 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	658
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	659 if updated_jid == own_jid:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	660 # If the own JID is updated, _all_ peers have to be notified
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	661 # TODO: Using my author's privilege here to shamelessly access private fields
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	662 # and storage keys until I've added public API to get a list of peers to
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	663 # python-omemo.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	664 storage: omemo.Storage = getattr(session_manager, "_SessionManager__storage")
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	665 peer_jids = frozenset(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	666 {
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	667 jid.JID(bare_jid).userhostJID()
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	668 for bare_jid in (
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	669 await storage.load_list(f"/{OMEMO.NS_TWOMEMO}/bare_jids", str)
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	670 ).maybe([])
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	671 }
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	672 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	673
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	674 if len(peer_jids) == 0:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	675 # If there are no peers to notify, notify our other devices about the
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	676 # changes directly
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	677 trust_messages.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	678 PartialTrustMessage(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	679 recipient_jid=own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	680 updated_jid=own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	681 trust_updates=trust_updates,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	682 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	683 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	684 else:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	685 # Otherwise, notify all peers about the changes in trust and let carbons
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	686 # handle the copy to our own JID
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	687 for peer_jid in peer_jids:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	688 trust_messages.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	689 PartialTrustMessage(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	690 recipient_jid=peer_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	691 updated_jid=own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	692 trust_updates=trust_updates,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	693 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	694 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	695
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	696 # Also send full trust information about _every_ peer to our newly
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	697 # trusted devices
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	698 peer_trust_updates = await get_trust_as_trust_updates(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	699 session_manager, peer_jid
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	700 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	701
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	702 trust_messages.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	703 PartialTrustMessage(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	704 recipient_jid=own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	705 updated_jid=peer_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	706 trust_updates=peer_trust_updates,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	707 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	708 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	709
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	710 # Send information about our own devices to our newly trusted devices
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	711 trust_messages.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	712 PartialTrustMessage(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	713 recipient_jid=own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	714 updated_jid=own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	715 trust_updates=own_trust_updates,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	716 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	717 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	718 else:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	719 # Notify our other devices about the changes in trust
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	720 trust_messages.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	721 PartialTrustMessage(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	722 recipient_jid=own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	723 updated_jid=updated_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	724 trust_updates=trust_updates,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	725 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	726 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	727
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	728 # Send a summary of our own trust to newly trusted devices
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	729 trust_messages.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	730 PartialTrustMessage(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	731 recipient_jid=updated_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	732 updated_jid=own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	733 trust_updates=own_trust_updates,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	734 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	735 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	736
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	737 # All trust messages prepared. Merge all trust messages directed at the same
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	738 # recipient.
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	739 recipient_jids = {trust_message.recipient_jid for trust_message in trust_messages}
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	740
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	741 for recipient_jid in recipient_jids:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	742 updated: Dict[jid.JID, Set[TrustUpdate]] = {}
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	743
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	744 for trust_message in trust_messages:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	745 # Merge trust messages directed at that recipient
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	746 if trust_message.recipient_jid == recipient_jid:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	747 # Merge the trust updates
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	748 updated[trust_message.updated_jid] = updated.get(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	749 trust_message.updated_jid, set()
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	750 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	751
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	752 updated[trust_message.updated_jid] \|= trust_message.trust_updates
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	753
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	754 # Build the trust message
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	755 trust_message_elt = domish.Element((NS_TM, "trust-message"))
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	756 trust_message_elt["usage"] = NS_ATM
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	757 trust_message_elt["encryption"] = twomemo.twomemo.NAMESPACE
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	758
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	759 for updated_jid, trust_updates in updated.items():
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	760 key_owner_elt = trust_message_elt.addElement((NS_TM, "key-owner"))
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	761 key_owner_elt["jid"] = updated_jid.userhost()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	762
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	763 for trust_update in trust_updates:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	764 serialized_identity_key = base64.b64encode(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	765 trust_update.target_key
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	766 ).decode("ASCII")
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	767
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	768 if trust_update.target_trust:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	769 key_owner_elt.addElement(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	770 (NS_TM, "trust"), content=serialized_identity_key
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	771 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	772 else:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	773 key_owner_elt.addElement(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	774 (NS_TM, "distrust"), content=serialized_identity_key
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	775 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	776
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	777 # Finally, encrypt and send the trust message!
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	778 message_data = client.generate_message_xml(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	779 MessageData(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	780 {
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	781 "from": own_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	782 "to": recipient_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	783 "uid": str(uuid.uuid4()),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	784 "message": {},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	785 "subject": {},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	786 "type": C.MESS_TYPE_CHAT,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	787 "extra": {},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	788 "timestamp": time.time(),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	789 }
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	790 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	791 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	792
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	793 message_data["xml"].addChild(trust_message_elt)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	794
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	795 plaintext = XEP_0420.pack_stanza(TM_SCE_PROFILE, message_data["xml"])
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	796
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	797 feedback_jid = recipient_jid
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	798
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	799 # TODO: The following is mostly duplicate code
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	800 try:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	801 messages, encryption_errors = await session_manager.encrypt(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	802 frozenset({own_jid.userhost(), recipient_jid.userhost()}),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	803 {OMEMO.NS_TWOMEMO: plaintext},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	804 backend_priority_order=[OMEMO.NS_TWOMEMO],
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	805 identifier=feedback_jid.userhost(),
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	806 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	807 except Exception as e:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	808 msg = _(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	809 # pylint: disable=consider-using-f-string
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	810 "Can't encrypt message for {entities}: {reason}".format(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	811 entities=", ".join({own_jid.userhost(), recipient_jid.userhost()}),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	812 reason=e,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	813 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	814 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	815 log.warning(msg)
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	816 client.feedback(feedback_jid, msg, {C.MESS_EXTRA_INFO: C.EXTRA_INFO_ENCR_ERR})
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	817 raise e
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	818
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	819 if len(encryption_errors) > 0:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	820 log.warning(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	821 f"Ignored the following non-critical encryption errors:"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	822 f" {encryption_errors}"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	823 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	824
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	825 encrypted_errors_stringified = ", ".join(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	826 [
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	827 f"device {err.device_id} of {err.bare_jid} under namespace"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	828 f" {err.namespace}"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	829 for err in encryption_errors
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	830 ]
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	831 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	832
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	833 client.feedback(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	834 feedback_jid,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	835 D_(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	836 "There were non-critical errors during encryption resulting in some"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	837 " of your destinees' devices potentially not receiving the message."
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	838 " This happens when the encryption data/key material of a device is"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	839 " incomplete or broken, which shouldn't happen for actively used"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	840 " devices, and can usually be ignored. The following devices are"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	841 f" affected: {encrypted_errors_stringified}."
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	842 ),
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	843 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	844
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	845 message = next(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	846 message for message in messages if message.namespace == OMEMO.NS_TWOMEMO
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	847 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	848
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	849 # Add the encrypted element
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	850 message_data["xml"].addChild(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	851 xml_tools.et_elt_2_domish_elt(twomemo.etree.serialize_message(message))
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	852 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	853
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	854 await client.a_send(message_data["xml"])
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	855
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	856
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	857 def make_session_manager(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	858 sat: LiberviaBackend, profile: str
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	859 ) -> Type[omemo.SessionManager]:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	860 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	861 @param sat: The SAT instance.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	862 @param profile: The profile.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	863 @return: A non-abstract subclass of :class:`~omemo.session_manager.SessionManager`
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	864 with XMPP interactions and trust handled via the SAT instance.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	865 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	866
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	867 client = sat.get_client(profile)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	868 xep_0060 = cast(XEP_0060, sat.plugins["XEP-0060"])
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	869
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	870 class SessionManagerImpl(omemo.SessionManager):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	871 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	872 Session manager implementation handling XMPP interactions and trust via an
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	873 instance of :class:`~sat.core.sat_main.SAT`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	874 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	875
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	876 @staticmethod
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	877 async def _upload_bundle(bundle: omemo.Bundle) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	878 if isinstance(bundle, twomemo.twomemo.BundleImpl):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	879 element = twomemo.etree.serialize_bundle(bundle)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	880
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	881 node = "urn:xmpp:omemo:2:bundles"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	882 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	883 await xep_0060.send_item(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	884 client,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	885 client.jid.userhostJID(),
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	886 node,
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	887 xml_tools.et_elt_2_domish_elt(element),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	888 item_id=str(bundle.device_id),
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	889 extra={
3933 cecf45416403 plugin XEP-0373 and XEP-0374: Implementation of OX and OXIM: Syndace <me@syndace.dev> parents: 3929 diff changeset	890 XEP_0060.EXTRA_PUBLISH_OPTIONS: {
4218 c0f3f29377f1 plugin XEP-0384: be sure to have `open` access model for created nodes. Goffi <goffi@goffi.org> parents: 4216 diff changeset	891 XEP_0060.OPT_ACCESS_MODEL: "open",
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	892 XEP_0060.OPT_MAX_ITEMS: "max",
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	893 },
4283 23842a63ea00 plugin XEP-0060: add a `force` options for publish options in `send_items`: Goffi <goffi@goffi.org> parents: 4270 diff changeset	894 XEP_0060.EXTRA_ON_PRECOND_NOT_MET: "force",
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	895 },
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	896 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	897 except (error.StanzaError, Exception) as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	898 if (
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	899 isinstance(e, error.StanzaError)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	900 and e.condition == "conflict"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	901 and e.appCondition is not None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	902 # pylint: disable=no-member
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	903 and e.appCondition.name == "precondition-not-met"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	904 ):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	905 # publish options couldn't be set on the fly, manually reconfigure
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	906 # the node and publish again
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	907 raise omemo.BundleUploadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	908 f"precondition-not-met: {bundle}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	909 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	910 # TODO: What can I do here? The correct node configuration is a
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	911 # MUST in the XEP.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	912
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	913 raise omemo.BundleUploadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	914 f"Bundle upload failed: {bundle}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	915 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	916
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	917 return
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	918
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	919 if isinstance(bundle, oldmemo.oldmemo.BundleImpl):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	920 element = oldmemo.etree.serialize_bundle(bundle)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	921
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	922 node = f"eu.siacs.conversations.axolotl.bundles:{bundle.device_id}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	923 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	924 await xep_0060.send_item(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	925 client,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	926 client.jid.userhostJID(),
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	927 node,
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	928 xml_tools.et_elt_2_domish_elt(element),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	929 item_id=xep_0060.ID_SINGLETON,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	930 extra={
4218 c0f3f29377f1 plugin XEP-0384: be sure to have `open` access model for created nodes. Goffi <goffi@goffi.org> parents: 4216 diff changeset	931 XEP_0060.EXTRA_PUBLISH_OPTIONS: {
c0f3f29377f1 plugin XEP-0384: be sure to have `open` access model for created nodes. Goffi <goffi@goffi.org> parents: 4216 diff changeset	932 XEP_0060.OPT_ACCESS_MODEL: "open",
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	933 XEP_0060.OPT_MAX_ITEMS: 1,
4218 c0f3f29377f1 plugin XEP-0384: be sure to have `open` access model for created nodes. Goffi <goffi@goffi.org> parents: 4216 diff changeset	934 },
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	935 XEP_0060.EXTRA_ON_PRECOND_NOT_MET: "publish_without_options",
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	936 },
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	937 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	938 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	939 raise omemo.BundleUploadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	940 f"Bundle upload failed: {bundle}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	941 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	942
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	943 return
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	944
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	945 raise omemo.UnknownNamespace(f"Unknown namespace: {bundle.namespace}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	946
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	947 @staticmethod
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	948 async def _download_bundle(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	949 namespace: str, bare_jid: str, device_id: int
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	950 ) -> omemo.Bundle:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	951 if namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	952 node = "urn:xmpp:omemo:2:bundles"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	953
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	954 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	955 items, __ = await xep_0060.get_items(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	956 client, jid.JID(bare_jid), node, item_ids=[str(device_id)]
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	957 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	958 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	959 raise omemo.BundleDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	960 f"Bundle download failed for {bare_jid}: {device_id} under"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	961 f" namespace {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	962 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	963
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	964 if len(items) != 1:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	965 raise omemo.BundleDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	966 f"Bundle download failed for {bare_jid}: {device_id} under"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	967 f" namespace {namespace}: Unexpected number of items retrieved:"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	968 f" {len(items)}."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	969 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	970
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	971 element = next(
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	972 iter(xml_tools.domish_elt_2_et_elt(cast(domish.Element, items[0]))),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	973 None,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	974 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	975 if element is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	976 raise omemo.BundleDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	977 f"Bundle download failed for {bare_jid}: {device_id} under"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	978 f" namespace {namespace}: Item download succeeded but parsing"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	979 f" failed: {element}."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	980 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	981
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	982 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	983 return twomemo.etree.parse_bundle(element, bare_jid, device_id)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	984 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	985 raise omemo.BundleDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	986 f"Bundle parsing failed for {bare_jid}: {device_id} under"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	987 f" namespace {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	988 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	989
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	990 if namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	991 return await download_oldmemo_bundle(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	992 client, xep_0060, bare_jid, device_id
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	993 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	994
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	995 raise omemo.UnknownNamespace(f"Unknown namespace: {namespace}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	996
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	997 @staticmethod
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	998 async def _delete_bundle(namespace: str, device_id: int) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	999 if namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1000 node = "urn:xmpp:omemo:2:bundles"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1001
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1002 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1003 await xep_0060.retract_items(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1004 client,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1005 client.jid.userhostJID(),
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1006 node,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1007 [str(device_id)],
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1008 notify=False,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1009 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1010 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1011 raise omemo.BundleDeletionFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1012 f"Bundle deletion failed for {device_id} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1013 f" {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1014 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1015
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1016 return
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1017
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1018 if namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1019 node = f"eu.siacs.conversations.axolotl.bundles:{device_id}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1020
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1021 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1022 await xep_0060.deleteNode(client, client.jid.userhostJID(), node)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1023 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1024 raise omemo.BundleDeletionFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1025 f"Bundle deletion failed for {device_id} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1026 f" {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1027 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1028
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1029 return
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1030
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1031 raise omemo.UnknownNamespace(f"Unknown namespace: {namespace}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1032
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1033 @staticmethod
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1034 async def _upload_device_list(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1035 namespace: str, device_list: Dict[int, Optional[str]]
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1036 ) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1037 element: Optional[ET.Element] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1038 node: Optional[str] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1039
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1040 if namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1041 element = twomemo.etree.serialize_device_list(device_list)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1042 node = TWOMEMO_DEVICE_LIST_NODE
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1043 if namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1044 element = oldmemo.etree.serialize_device_list(device_list)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1045 node = OLDMEMO_DEVICE_LIST_NODE
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1046
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1047 if element is None or node is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1048 raise omemo.UnknownNamespace(f"Unknown namespace: {namespace}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1049
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1050 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1051 await xep_0060.send_item(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1052 client,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1053 client.jid.userhostJID(),
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1054 node,
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	1055 xml_tools.et_elt_2_domish_elt(element),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1056 item_id=xep_0060.ID_SINGLETON,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1057 extra={
3933 cecf45416403 plugin XEP-0373 and XEP-0374: Implementation of OX and OXIM: Syndace <me@syndace.dev> parents: 3929 diff changeset	1058 XEP_0060.EXTRA_PUBLISH_OPTIONS: {
cecf45416403 plugin XEP-0373 and XEP-0374: Implementation of OX and OXIM: Syndace <me@syndace.dev> parents: 3929 diff changeset	1059 XEP_0060.OPT_MAX_ITEMS: 1,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1060 XEP_0060.OPT_ACCESS_MODEL: "open",
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1061 },
4283 23842a63ea00 plugin XEP-0060: add a `force` options for publish options in `send_items`: Goffi <goffi@goffi.org> parents: 4270 diff changeset	1062 XEP_0060.EXTRA_ON_PRECOND_NOT_MET: "force",
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1063 },
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1064 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1065 except (error.StanzaError, Exception) as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1066 if (
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1067 isinstance(e, error.StanzaError)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1068 and e.condition == "conflict"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1069 and e.appCondition is not None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1070 # pylint: disable=no-member
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1071 and e.appCondition.name == "precondition-not-met"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1072 ):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1073 # publish options couldn't be set on the fly, manually reconfigure the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1074 # node and publish again
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1075 raise omemo.DeviceListUploadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1076 f"precondition-not-met for namespace {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1077 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1078 # TODO: What can I do here? The correct node configuration is a MUST
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1079 # in the XEP.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1080
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1081 raise omemo.DeviceListUploadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1082 f"Device list upload failed for namespace {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1083 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1084
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1085 @staticmethod
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1086 async def _download_device_list(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1087 namespace: str, bare_jid: str
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1088 ) -> Dict[int, Optional[str]]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1089 node: Optional[str] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1090
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1091 if namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1092 node = TWOMEMO_DEVICE_LIST_NODE
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1093 if namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1094 node = OLDMEMO_DEVICE_LIST_NODE
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1095
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1096 if node is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1097 raise omemo.UnknownNamespace(f"Unknown namespace: {namespace}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1098
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1099 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1100 items, __ = await xep_0060.get_items(client, jid.JID(bare_jid), node)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1101 except exceptions.NotFound:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1102 return {}
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1103 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1104 raise omemo.DeviceListDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1105 f"Device list download failed for {bare_jid} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1106 f" {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1107 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1108
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	1109 if len(items) == 0:
4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	1110 return {}
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1111
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1112 if len(items) != 1:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1113 raise omemo.DeviceListDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1114 f"Device list download failed for {bare_jid} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1115 f" {namespace}: Unexpected number of items retrieved: {len(items)}."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1116 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1117
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1118 element = next(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1119 iter(xml_tools.domish_elt_2_et_elt(cast(domish.Element, items[0]))), None
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1120 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1121
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1122 if element is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1123 raise omemo.DeviceListDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1124 f"Device list download failed for {bare_jid} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1125 f" {namespace}: Item download succeeded but parsing failed:"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1126 f" {element}."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1127 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1128
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1129 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1130 if namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1131 return twomemo.etree.parse_device_list(element)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1132 if namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1133 return oldmemo.etree.parse_device_list(element)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1134 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1135 raise omemo.DeviceListDownloadFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1136 f"Device list download failed for {bare_jid} under namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1137 f" {namespace}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1138 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1139
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1140 raise omemo.UnknownNamespace(f"Unknown namespace: {namespace}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1141
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1142 async def _evaluate_custom_trust_level(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1143 self, device: omemo.DeviceInformation
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1144 ) -> omemo.TrustLevel:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1145 # Get the custom trust level
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1146 try:
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1147 trust_level = TrustLevel(device.trust_level_name)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1148 except ValueError as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1149 raise omemo.UnknownTrustLevel(
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1150 f"Unknown trust level name {device.trust_level_name}"
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1151 ) from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1152
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1153 # The first three cases are a straight-forward mapping
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1154 if trust_level is TrustLevel.TRUSTED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1155 return omemo.TrustLevel.TRUSTED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1156 if trust_level is TrustLevel.UNDECIDED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1157 return omemo.TrustLevel.UNDECIDED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1158 if trust_level is TrustLevel.DISTRUSTED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1159 return omemo.TrustLevel.DISTRUSTED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1160
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1161 # The blindly trusted case is more complicated, since its evaluation depends
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1162 # on the trust system and phase
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1163 if trust_level is TrustLevel.BLINDLY_TRUSTED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1164 # Get the name of the active trust system
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1165 trust_system = cast(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1166 str,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1167 sat.memory.param_get_a(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1168 PARAM_NAME, PARAM_CATEGORY, profile_key=profile
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1169 ),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1170 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1171
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1172 # If the trust model is BTBV, blind trust is always enabled
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1173 if trust_system == "btbv":
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1174 return omemo.TrustLevel.TRUSTED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1175
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1176 # If the trust model is ATM, blind trust is disabled in the second phase
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1177 # and counts as undecided
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1178 if trust_system == "atm":
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1179 # Find out whether we are in phase one or two
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1180 devices = await self.get_device_information(device.bare_jid)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1181
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1182 phase_one = all(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1183 TrustLevel(device.trust_level_name)
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1184 in {TrustLevel.UNDECIDED, TrustLevel.BLINDLY_TRUSTED}
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1185 for device in devices
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1186 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1187
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1188 if phase_one:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1189 return omemo.TrustLevel.TRUSTED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1190
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1191 return omemo.TrustLevel.UNDECIDED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1192
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1193 raise exceptions.InternalError(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1194 f"Unknown trust system active: {trust_system}"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1195 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1196
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1197 assert_never(trust_level)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1198
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1199 async def _make_trust_decision(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1200 self, undecided: FrozenSet[omemo.DeviceInformation], identifier: Optional[str]
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1201 ) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1202 if identifier is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1203 raise omemo.TrustDecisionFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1204 "The identifier must contain the feedback JID."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1205 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1206
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1207 # The feedback JID is transferred via the identifier
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1208 feedback_jid = jid.JID(identifier).userhostJID()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1209
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1210 # Both the ATM and the BTBV trust models work with blind trust before the
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1211 # first manual verification is performed. Thus, we can separate bare JIDs into
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1212 # two pools here, one pool of bare JIDs for which blind trust is active, and
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1213 # one pool of bare JIDs for which manual trust is used instead.
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1214 bare_jids = {device.bare_jid for device in undecided}
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1215
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1216 blind_trust_bare_jids: Set[str] = set()
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1217 manual_trust_bare_jids: Set[str] = set()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1218
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1219 # For each bare JID, decide whether blind trust applies
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1220 for bare_jid in bare_jids:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1221 # Get all known devices belonging to the bare JID
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1222 devices = await self.get_device_information(bare_jid)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1223
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1224 # If the trust levels of all devices correspond to those used by blind
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1225 # trust, blind trust applies. Otherwise, fall back to manual trust.
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1226 if all(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1227 TrustLevel(device.trust_level_name)
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1228 in {TrustLevel.UNDECIDED, TrustLevel.BLINDLY_TRUSTED}
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1229 for device in devices
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1230 ):
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1231 blind_trust_bare_jids.add(bare_jid)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1232 else:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1233 manual_trust_bare_jids.add(bare_jid)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1234
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1235 # With the JIDs sorted into their respective pools, the undecided devices can
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1236 # be categorized too
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1237 blindly_trusted_devices = {
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1238 dev for dev in undecided if dev.bare_jid in blind_trust_bare_jids
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1239 }
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1240 manually_trusted_devices = {
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1241 dev for dev in undecided if dev.bare_jid in manual_trust_bare_jids
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1242 }
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1243
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1244 # Blindly trust devices handled by blind trust
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1245 if len(blindly_trusted_devices) > 0:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1246 for device in blindly_trusted_devices:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1247 await self.set_trust(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1248 device.bare_jid,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1249 device.identity_key,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1250 TrustLevel.BLINDLY_TRUSTED.name,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1251 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1252
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1253 blindly_trusted_devices_stringified = ", ".join(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1254 [
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1255 f"device {device.device_id} of {device.bare_jid} under namespace"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1256 f" {device.namespaces}"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1257 for device in blindly_trusted_devices
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1258 ]
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1259 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1260
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1261 client.feedback(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1262 feedback_jid,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1263 D_(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1264 "Not all destination devices are trusted, unknown devices will be"
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1265 " blindly trusted.\nFollowing devices have been automatically"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1266 f" trusted: {blindly_trusted_devices_stringified}."
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1267 ),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1268 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1269
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1270 # Prompt the user for manual trust decisions on the devices handled by manual
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1271 # trust
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1272 if len(manually_trusted_devices) > 0:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1273 client.feedback(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1274 feedback_jid,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1275 D_(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1276 "Not all destination devices are trusted, we can't encrypt"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1277 " message in such a situation. Please indicate if you trust"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1278 " those devices or not in the trust manager before we can"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1279 " send this message."
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1280 ),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1281 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1282 await self.__prompt_manual_trust(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1283 frozenset(manually_trusted_devices), feedback_jid
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1284 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1285
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1286 @staticmethod
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1287 async def _send_message(message: omemo.Message, bare_jid: str) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1288 element: Optional[ET.Element] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1289
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1290 if message.namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1291 element = twomemo.etree.serialize_message(message)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1292 if message.namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1293 element = oldmemo.etree.serialize_message(message)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1294
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1295 if element is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1296 raise omemo.UnknownNamespace(f"Unknown namespace: {message.namespace}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1297
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1298 message_data = client.generate_message_xml(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1299 MessageData(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1300 {
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1301 "from": client.jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1302 "to": jid.JID(bare_jid),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1303 "uid": str(uuid.uuid4()),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1304 "message": {},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1305 "subject": {},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1306 "type": C.MESS_TYPE_CHAT,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1307 "extra": {},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1308 "timestamp": time.time(),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1309 }
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1310 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1311 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1312
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	1313 message_data["xml"].addChild(xml_tools.et_elt_2_domish_elt(element))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1314
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1315 try:
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	1316 await client.a_send(message_data["xml"])
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1317 except Exception as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1318 raise omemo.MessageSendingFailed() from e
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1319
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1320 async def __prompt_manual_trust(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1321 self, undecided: FrozenSet[omemo.DeviceInformation], feedback_jid: jid.JID
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1322 ) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1323 """Asks the user to decide on the manual trust level of a set of devices.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1324
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1325 Blocks until the user has made a decision and updates the trust levels of all
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1326 devices using :meth:`set_trust`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1327
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1328 @param undecided: The set of devices to prompt manual trust for.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1329 @param feedback_jid: The bare JID to redirect feedback to. In case of a one to
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1330 one message, the recipient JID. In case of a MUC message, the room JID.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1331 @raise TrustDecisionFailed: if the user cancels the prompt.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1332 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1333
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1334 # This session manager handles encryption with both twomemo and oldmemo, but
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1335 # both are currently registered as different plugins and the `defer_xmlui`
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1336 # below requires a single namespace identifying the encryption plugin. Thus,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1337 # get the namespace of the requested encryption method from the encryption
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1338 # session using the feedback JID.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1339 encryption = client.encryption.getSession(feedback_jid)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1340 if encryption is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1341 raise omemo.TrustDecisionFailed(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1342 f"Encryption not requested for {feedback_jid.userhost()}."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1343 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1344
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1345 namespace = encryption["plugin"].namespace
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1346
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1347 # Casting this to Any, otherwise all calls on the variable cause type errors
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1348 # pylint: disable=no-member
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1349 trust_ui = cast(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1350 Any,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1351 xml_tools.XMLUI(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1352 panel_type=C.XMLUI_FORM,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1353 title=D_("OMEMO trust management"),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1354 submit_id="",
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1355 ),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1356 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1357 trust_ui.addText(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1358 D_(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1359 "This is OMEMO trusting system. You'll see below the devices of your "
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1360 "contacts, and a checkbox to trust them or not. A trusted device "
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1361 "can read your messages in plain text, so be sure to only validate "
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1362 "devices that you are sure are belonging to your contact. It's better "
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1363 "to do this when you are next to your contact and their device, so "
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1364 'you can check the "fingerprint" (the number next to the device) '
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1365 "yourself. Do not validate a device if the fingerprint is wrong!"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1366 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1367 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1368
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1369 own_device, __ = await self.get_own_device_information()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1370
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1371 trust_ui.change_container("label")
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1372 trust_ui.addLabel(D_("This device ID"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1373 trust_ui.addText(str(own_device.device_id))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1374 trust_ui.addLabel(D_("This device's fingerprint"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1375 trust_ui.addText(" ".join(self.format_identity_key(own_device.identity_key)))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1376 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1377 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1378
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1379 # At least sort the devices by bare JID such that they aren't listed
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1380 # completely random
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1381 undecided_ordered = sorted(undecided, key=lambda device: device.bare_jid)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1382
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1383 for index, device in enumerate(undecided_ordered):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1384 trust_ui.addLabel(D_("Contact"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1385 trust_ui.addJid(jid.JID(device.bare_jid))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1386 trust_ui.addLabel(D_("Device ID"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1387 trust_ui.addText(str(device.device_id))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1388 trust_ui.addLabel(D_("Fingerprint"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1389 trust_ui.addText(" ".join(self.format_identity_key(device.identity_key)))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1390 trust_ui.addLabel(D_("Trust this device?"))
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1391 trust_ui.addBool(f"trust_{index}", value=C.bool_const(False))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1392 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1393 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1394
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1395 trust_ui_result = await xml_tools.defer_xmlui(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1396 sat,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1397 trust_ui,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1398 action_extra={"meta_encryption_trust": namespace},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1399 profile=profile,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1400 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1401
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1402 if C.bool(trust_ui_result.get("cancelled", "false")):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1403 raise omemo.TrustDecisionFailed("Trust UI cancelled.")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1404
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1405 data_form_result = cast(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1406 Dict[str, str], xml_tools.xmlui_result_2_data_form_result(trust_ui_result)
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1407 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1408
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1409 trust_updates: Set[TrustUpdate] = set()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1410
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1411 for key, value in data_form_result.items():
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1412 if not key.startswith("trust_"):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1413 continue
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1414
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1415 device = undecided_ordered[int(key[len("trust_") :])]
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1416 target_trust = C.bool(value)
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1417 trust_level = (
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1418 TrustLevel.TRUSTED if target_trust else TrustLevel.DISTRUSTED
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1419 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1420
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1421 await self.set_trust(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1422 device.bare_jid, device.identity_key, trust_level.name
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1423 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1424
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1425 trust_updates.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1426 TrustUpdate(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1427 target_jid=jid.JID(device.bare_jid).userhostJID(),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1428 target_key=device.identity_key,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1429 target_trust=target_trust,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1430 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1431 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1432
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1433 # Check whether ATM is enabled and handle everything in case it is
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1434 trust_system = cast(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1435 str,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1436 sat.memory.param_get_a(PARAM_NAME, PARAM_CATEGORY, profile_key=profile),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1437 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1438
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1439 if trust_system == "atm":
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1440 await manage_trust_message_cache(client, self, frozenset(trust_updates))
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1441 await send_trust_messages(client, self, frozenset(trust_updates))
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1442
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1443 return SessionManagerImpl
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1444
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1445
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1446 async def prepare_for_profile(
4072 040095a5dc7f refactoring: rename `SAT` class to `LiberviaBackend` Goffi <goffi@goffi.org> parents: 4071 diff changeset	1447 sat: LiberviaBackend,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1448 profile: str,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1449 initial_own_label: Optional[str],
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1450 signed_pre_key_rotation_period: int = 7 * 24 * 60 * 60,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1451 pre_key_refill_threshold: int = 99,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1452 max_num_per_session_skipped_keys: int = 1000,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1453 max_num_per_message_skipped_keys: Optional[int] = None,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1454 ) -> omemo.SessionManager:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1455 """Prepare the OMEMO library (storage, backends, core) for a specific profile.
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1456
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1457 @param sat: The SAT instance.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1458 @param profile: The profile.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1459 @param initial_own_label: The initial (optional) label to assign to this device if
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1460 supported by any of the backends.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1461 @param signed_pre_key_rotation_period: The rotation period for the signed pre key, in
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1462 seconds. The rotation period is recommended to be between one week (the default)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1463 and one month.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1464 @param pre_key_refill_threshold: The number of pre keys that triggers a refill to 100.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1465 Defaults to 99, which means that each pre key gets replaced with a new one right
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1466 away. The threshold can not be configured to lower than 25.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1467 @param max_num_per_session_skipped_keys: The maximum number of skipped message keys to
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1468 keep around per session. Once the maximum is reached, old message keys are deleted
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1469 to make space for newer ones. Accessible via
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1470 :attr:`max_num_per_session_skipped_keys`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1471 @param max_num_per_message_skipped_keys: The maximum number of skipped message keys to
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1472 accept in a single message. When set to ``None`` (the default), this parameter
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1473 defaults to the per-session maximum (i.e. the value of the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1474 ``max_num_per_session_skipped_keys`` parameter). This parameter may only be 0 if
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1475 the per-session maximum is 0, otherwise it must be a number between 1 and the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1476 per-session maximum. Accessible via :attr:`max_num_per_message_skipped_keys`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1477 @return: A session manager with ``urn:xmpp:omemo:2`` and
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1478 ``eu.siacs.conversations.axolotl`` capabilities, specifically for the given
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1479 profile.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1480 @raise BundleUploadFailed: if a bundle upload failed. Forwarded from
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1481 :meth:`~omemo.session_manager.SessionManager.create`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1482 @raise BundleDownloadFailed: if a bundle download failed. Forwarded from
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1483 :meth:`~omemo.session_manager.SessionManager.create`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1484 @raise BundleDeletionFailed: if a bundle deletion failed. Forwarded from
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1485 :meth:`~omemo.session_manager.SessionManager.create`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1486 @raise DeviceListUploadFailed: if a device list upload failed. Forwarded from
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1487 :meth:`~omemo.session_manager.SessionManager.create`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1488 @raise DeviceListDownloadFailed: if a device list download failed. Forwarded from
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1489 :meth:`~omemo.session_manager.SessionManager.create`.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1490 """
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1491
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1492 client = sat.get_client(profile)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1493 xep_0060 = cast(XEP_0060, sat.plugins["XEP-0060"])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1494
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1495 storage = StorageImpl(profile)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1496
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1497 # TODO: Untested
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1498 await oldmemo.migrations.migrate(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1499 LegacyStorageImpl(profile, client.jid.userhost()),
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1500 storage,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1501 # TODO: Do we want BLINDLY_TRUSTED or TRUSTED here?
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1502 TrustLevel.BLINDLY_TRUSTED.name,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1503 TrustLevel.UNDECIDED.name,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1504 TrustLevel.DISTRUSTED.name,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1505 lambda bare_jid, device_id: download_oldmemo_bundle(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1506 client, xep_0060, bare_jid, device_id
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1507 ),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1508 )
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1509
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1510 session_manager = await make_session_manager(sat, profile).create(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1511 [
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1512 twomemo.Twomemo(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1513 storage,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1514 max_num_per_session_skipped_keys,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1515 max_num_per_message_skipped_keys,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1516 ),
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1517 oldmemo.Oldmemo(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1518 storage,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1519 max_num_per_session_skipped_keys,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1520 max_num_per_message_skipped_keys,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1521 ),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1522 ],
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1523 storage,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1524 client.jid.userhost(),
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1525 initial_own_label,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1526 TrustLevel.UNDECIDED.value,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1527 signed_pre_key_rotation_period,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1528 pre_key_refill_threshold,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1529 omemo.AsyncFramework.TWISTED,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1530 )
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1531
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1532 # This shouldn't hurt here since we're not running on overly constrainted devices.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1533 # TODO: Consider ensuring data consistency regularly/in response to certain events
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1534 await session_manager.ensure_data_consistency()
3240 d85b68e44297 plugin XEP-0384: fixed /omemo_reset + device ID type: Goffi <goffi@goffi.org> parents: 3237 diff changeset	1535
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1536 # TODO: Correct entering/leaving of the history synchronization mode isn't terribly
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1537 # important for now, since it only prevents an extremely unlikely race condition of
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1538 # multiple devices choosing the same pre key for new sessions while the device was
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1539 # offline. I don't believe other clients seriously defend against that race condition
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1540 # either. In the long run, it might still be cool to have triggers for when history
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1541 # sync starts and ends (MAM, MUC catch-up, etc.) and to react to those triggers.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1542 await session_manager.after_history_sync()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1543
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1544 return session_manager
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1545
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1546
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1547 DEFAULT_TRUST_MODEL_PARAM = f"""
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1548 <params>
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1549 <individual>
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1550 <category name="{PARAM_CATEGORY}" label={quoteattr(D_('Security'))}>
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1551 <param name="{PARAM_NAME}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1552 label={quoteattr(D_('OMEMO default trust policy'))}
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1553 type="list" security="3">
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1554 <option value="atm"
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1555 label={quoteattr(D_('Automatic Trust Management (more secure)'))} />
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1556 <option value="btbv"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1557 label={quoteattr(D_('Blind Trust Before Verification (more user friendly)'))}
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1558 selected="true" />
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1559 </param>
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1560 </category>
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1561 </individual>
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1562 </params>
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1563 """
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1564
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1565
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1566 class OMEMO:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1567 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1568 Plugin equipping Libervia with OMEMO capabilities under the (modern)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1569 ``urn:xmpp:omemo:2`` namespace and the (legacy) ``eu.siacs.conversations.axolotl``
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1570 namespace. Both versions of the protocol are handled by this plugin and compatibility
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1571 between the two is maintained. MUC messages are supported next to one to one messages.
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1572 For trust management, the two trust models "ATM" and "BTBV" are supported.
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1573 """
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1574
3918 e63f96e60f7b plugin XEP-0384: expose OLDMEMO and TWOMEMO namespaces: Goffi <goffi@goffi.org> parents: 3914 diff changeset	1575 NS_TWOMEMO = twomemo.twomemo.NAMESPACE
e63f96e60f7b plugin XEP-0384: expose OLDMEMO and TWOMEMO namespaces: Goffi <goffi@goffi.org> parents: 3914 diff changeset	1576 NS_OLDMEMO = oldmemo.oldmemo.NAMESPACE
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1577
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1578 # For MUC/MIX message stanzas, the <to/> affix is a MUST
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1579 SCE_PROFILE_GROUPCHAT = SCEProfile(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1580 rpad_policy=SCEAffixPolicy.REQUIRED,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1581 time_policy=SCEAffixPolicy.OPTIONAL,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1582 to_policy=SCEAffixPolicy.REQUIRED,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1583 from_policy=SCEAffixPolicy.OPTIONAL,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1584 custom_policies={},
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1585 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1586
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1587 # For everything but MUC/MIX message stanzas, the <to/> affix is a MAY
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1588 SCE_PROFILE = SCEProfile(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1589 rpad_policy=SCEAffixPolicy.REQUIRED,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1590 time_policy=SCEAffixPolicy.OPTIONAL,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1591 to_policy=SCEAffixPolicy.OPTIONAL,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1592 from_policy=SCEAffixPolicy.OPTIONAL,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1593 custom_policies={},
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	1594 )
b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	1595
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1596 def __init__(self, host: LiberviaBackend) -> None:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1597 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1598 @param sat: The SAT instance.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1599 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1600
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1601 self.host = host
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1602
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1603 # Add configuration option to choose between manual trust and BTBV as the trust
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1604 # model
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1605 host.memory.update_params(DEFAULT_TRUST_MODEL_PARAM)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1606
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1607 # Plugins
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1608 self._j = cast(XEP_0060, host.plugins["XEP-0060"])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1609 self.__xep_0045 = cast(Optional[XEP_0045], host.plugins.get("XEP-0045"))
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1610 self.__xep_0334 = cast(XEP_0334, host.plugins["XEP-0334"])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1611 self.__xep_0359 = cast(Optional[XEP_0359], host.plugins.get("XEP-0359"))
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1612 self.__xep_0420 = cast(XEP_0420, host.plugins["XEP-0420"])
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1613
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1614 # In contrast to one to one messages, MUC messages are reflected to the sender.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1615 # Thus, the sender does not add messages to their local message log when sending
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1616 # them, but when the reflection is received. This approach does not pair well with
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1617 # OMEMO, since for security reasons it is forbidden to encrypt messages for the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1618 # own device. Thus, when the reflection of an OMEMO message is received, it can't
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1619 # be decrypted and added to the local message log as usual. To counteract this,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1620 # the plaintext of encrypted messages sent to MUCs are cached in this field, such
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1621 # that when the reflection is received, the plaintext can be looked up from the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1622 # cache and added to the local message log.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1623 # TODO: The old plugin expired this cache after some time. I'm not sure that's
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1624 # really necessary.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1625 self.__muc_plaintext_cache: Dict[MUCPlaintextCacheKey, bytes] = {}
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1626
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1627 # Mapping from profile name to corresponding session manager
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1628 self.__session_managers: Dict[str, omemo.SessionManager] = {}
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1629
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1630 # Calls waiting for a specific session manager to be built
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1631 self.__session_manager_waiters: Dict[str, List[defer.Deferred]] = {}
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1632
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1633 # These triggers are used by oldmemo, which doesn't do SCE and only applies to
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1634 # messages. Temporarily, until a more fitting trigger for SCE-based encryption is
4051 c23cad65ae99 core: renamed `messageReceived` trigger to `message_received` Goffi <goffi@goffi.org> parents: 4037 diff changeset	1635 # added, the message_received trigger is also used for twomemo.
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1636 host.trigger.add(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1637 "message_received", self._message_received_trigger, priority=100050
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	1638 )
4164 15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	1639
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1640 host.trigger.add("send", self.__send_trigger, priority=0)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1641 # TODO: Add new triggers here for freshly received and about-to-be-sent stanzas,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1642 # including IQs.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1643
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1644 # Give twomemo a (slightly) higher priority than oldmemo
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1645 host.register_encryption_plugin(self, "OMEMO", twomemo.twomemo.NAMESPACE, 101)
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1646 host.register_encryption_plugin(
4150 26534d959d2d Plugin XEP-0384: rename the pun names "OLDMEMO" and "TWOMEMO" to "OMEMO_legacy" and "OMEMO" for clarity. Goffi <goffi@goffi.org> parents: 4073 diff changeset	1647 self, "OMEMO_legacy", oldmemo.oldmemo.NAMESPACE, 100
26534d959d2d Plugin XEP-0384: rename the pun names "OLDMEMO" and "TWOMEMO" to "OMEMO_legacy" and "OMEMO" for clarity. Goffi <goffi@goffi.org> parents: 4073 diff changeset	1648 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1649
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1650 xep_0163 = cast(XEP_0163, host.plugins["XEP-0163"])
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1651 xep_0163.add_pep_event(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1652 "TWOMEMO_DEVICES",
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1653 TWOMEMO_DEVICE_LIST_NODE,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1654 lambda items_event, profile: defer.ensureDeferred(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1655 self.__on_device_list_update(items_event, profile)
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1656 ),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1657 )
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1658 xep_0163.add_pep_event(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1659 "OLDMEMO_DEVICES",
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1660 OLDMEMO_DEVICE_LIST_NODE,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1661 lambda items_event, profile: defer.ensureDeferred(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1662 self.__on_device_list_update(items_event, profile)
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1663 ),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1664 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1665
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1666 try:
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1667 self.__text_commands = cast(TextCommands, host.plugins[C.TEXT_CMDS])
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1668 except KeyError:
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1669 log.info(_("Text commands not available"))
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1670 else:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1671 self.__text_commands.register_text_commands(self)
524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1672
524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1673 def profile_connected( # pylint: disable=invalid-name
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1674 self, client: SatXMPPClient
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1675 ) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1676 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1677 @param client: The client.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1678 """
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1679
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1680 defer.ensureDeferred(self.get_session_manager(cast(str, client.profile)))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1681
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1682 async def cmd_omemo_reset(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1683 self, client: SatXMPPClient, mess_data: MessageData
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1684 ) -> Literal[False]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1685 """Reset all sessions of devices that belong to the recipient of ``mess_data``.
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1686
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1687 This must only be callable manually by the user. Use this when a session is
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1688 apparently broken, i.e. sending and receiving encrypted messages doesn't work and
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1689 something being wrong has been confirmed manually with the recipient.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1690
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1691 @param client: The client.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1692 @param mess_data: The message data, whose ``to`` attribute will be the bare JID to
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1693 reset all sessions with.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1694 @return: The constant value ``False``, indicating to the text commands plugin that
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1695 the message is not supposed to be sent.
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1696 """
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1697
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1698 twomemo_requested = client.encryption.is_encryption_requested(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1699 mess_data, twomemo.twomemo.NAMESPACE
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1700 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1701 oldmemo_requested = client.encryption.is_encryption_requested(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1702 mess_data, oldmemo.oldmemo.NAMESPACE
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1703 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1704
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1705 if not (twomemo_requested or oldmemo_requested):
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1706 self.__text_commands.feed_back(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1707 client,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1708 _("You need to have OMEMO encryption activated to reset the session"),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1709 mess_data,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1710 )
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1711 return False
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1712
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1713 bare_jid = mess_data["to"].userhost()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1714
3969 8e7d5796fb23 plugin XEP-0391: implement XEP-0391 (Jingle Encrypted Transports) + XEP-0396 (JET-OMEMO): Goffi <goffi@goffi.org> parents: 3967 diff changeset	1715 session_manager = await self.get_session_manager(client.profile)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1716 devices = await session_manager.get_device_information(bare_jid)
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1717
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1718 for device in devices:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1719 log.debug(f"Replacing sessions with device {device}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1720 await session_manager.replace_sessions(device)
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1721
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1722 self.__text_commands.feed_back(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1723 client, _("OMEMO session has been reset"), mess_data
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1724 )
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1725
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	1726 return False
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1727
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1728 async def get_trust_ui( # pylint: disable=invalid-name
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1729 self, client: SatXMPPClient, entity: jid.JID
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1730 ) -> xml_tools.XMLUI:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1731 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1732 @param client: The client.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1733 @param entity: The entity whose device trust levels to manage.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1734 @return: An XMLUI instance which opens a form to manage the trust level of all
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1735 devices belonging to the entity.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1736 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1737
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1738 if entity.resource:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1739 raise ValueError("A bare JID is expected.")
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	1740
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1741 bare_jids: Set[str]
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1742 if self.__xep_0045 is not None and self.__xep_0045.is_joined_room(client, entity):
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1743 bare_jids = self.__get_joined_muc_users(client, self.__xep_0045, entity)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1744 else:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1745 bare_jids = {entity.userhost()}
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1746
3969 8e7d5796fb23 plugin XEP-0391: implement XEP-0391 (Jingle Encrypted Transports) + XEP-0396 (JET-OMEMO): Goffi <goffi@goffi.org> parents: 3967 diff changeset	1747 session_manager = await self.get_session_manager(client.profile)
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	1748
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1749 # At least sort the devices by bare JID such that they aren't listed completely
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1750 # random
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1751 devices = sorted(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1752 cast(Set[omemo.DeviceInformation], set()).union(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1753 *[
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1754 await session_manager.get_device_information(bare_jid)
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1755 for bare_jid in bare_jids
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1756 ]
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1757 ),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1758 key=lambda device: device.bare_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1759 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1760
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1761 async def callback(data: Any, profile: str) -> Dict[Never, Never]:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1762 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1763 @param data: The XMLUI result produces by the trust UI form.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1764 @param profile: The profile.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1765 @return: An empty dictionary. The type of the return value was chosen
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1766 conservatively since the exact options are neither known not needed here.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1767 """
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1768
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1769 if C.bool(data.get("cancelled", "false")):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1770 return {}
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1771
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1772 data_form_result = cast(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1773 Dict[str, str], xml_tools.xmlui_result_2_data_form_result(data)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1774 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1775
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1776 trust_updates: Set[TrustUpdate] = set()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1777
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1778 for key, value in data_form_result.items():
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1779 if not key.startswith("trust_"):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1780 continue
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1781
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1782 device = devices[int(key[len("trust_") :])]
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1783 trust_level_name = value
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1784
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1785 if device.trust_level_name != trust_level_name:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1786 await session_manager.set_trust(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1787 device.bare_jid, device.identity_key, trust_level_name
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1788 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1789
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1790 target_trust: Optional[bool] = None
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1791
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1792 if TrustLevel(trust_level_name) is TrustLevel.TRUSTED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1793 target_trust = True
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1794 if TrustLevel(trust_level_name) is TrustLevel.DISTRUSTED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1795 target_trust = False
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1796
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1797 if target_trust is not None:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1798 trust_updates.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1799 TrustUpdate(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1800 target_jid=jid.JID(device.bare_jid).userhostJID(),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1801 target_key=device.identity_key,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1802 target_trust=target_trust,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1803 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1804 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1805
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1806 # Check whether ATM is enabled and handle everything in case it is
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1807 trust_system = cast(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1808 str,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1809 self.host.memory.param_get_a(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1810 PARAM_NAME, PARAM_CATEGORY, profile_key=profile
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1811 ),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1812 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1813
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1814 if trust_system == "atm":
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1815 if len(trust_updates) > 0:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1816 await manage_trust_message_cache(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1817 client, session_manager, frozenset(trust_updates)
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1818 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1819
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1820 await send_trust_messages(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1821 client, session_manager, frozenset(trust_updates)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1822 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1823
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1824 return {}
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	1825
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	1826 submit_id = self.host.register_callback(callback, with_data=True, one_shot=True)
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1827
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1828 result = xml_tools.XMLUI(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1829 panel_type=C.XMLUI_FORM,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1830 title=D_("OMEMO trust management"),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1831 submit_id=submit_id,
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1832 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1833 # Casting this to Any, otherwise all calls on the variable cause type errors
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1834 # pylint: disable=no-member
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1835 trust_ui = cast(Any, result)
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1836 trust_ui.addText(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1837 D_(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1838 "This is OMEMO trusting system. You'll see below the devices of your"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1839 " contacts, and a list selection to trust them or not. A trusted device"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1840 " can read your messages in plain text, so be sure to only validate"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1841 " devices that you are sure are belonging to your contact. It's better"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1842 " to do this when you are next to your contact and their device, so"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1843 ' you can check the "fingerprint" (the number next to the device)'
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1844 " yourself. Do not validate a device if the fingerprint is wrong!"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1845 " Note that manually validating a fingerprint disables any form of automatic"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1846 " trust."
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1847 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1848 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1849
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1850 own_device, __ = await session_manager.get_own_device_information()
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1851
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1852 trust_ui.change_container("label")
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1853 trust_ui.addLabel(D_("This device ID"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1854 trust_ui.addText(str(own_device.device_id))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1855 trust_ui.addLabel(D_("This device's fingerprint"))
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1856 trust_ui.addText(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1857 " ".join(session_manager.format_identity_key(own_device.identity_key))
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1858 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1859 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1860 trust_ui.addEmpty()
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	1861
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1862 for index, device in enumerate(devices):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1863 trust_ui.addLabel(D_("Contact"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1864 trust_ui.addJid(jid.JID(device.bare_jid))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1865 trust_ui.addLabel(D_("Device ID"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1866 trust_ui.addText(str(device.device_id))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1867 trust_ui.addLabel(D_("Fingerprint"))
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1868 trust_ui.addText(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1869 " ".join(session_manager.format_identity_key(device.identity_key))
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1870 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1871 trust_ui.addLabel(D_("Trust this device?"))
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1872
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1873 current_trust_level = TrustLevel(device.trust_level_name)
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1874 avaiable_trust_levels = {
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1875 TrustLevel.DISTRUSTED,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1876 TrustLevel.TRUSTED,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1877 current_trust_level,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1878 }
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	1879
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1880 trust_ui.addList(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1881 f"trust_{index}",
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1882 options=[trust_level.name for trust_level in avaiable_trust_levels],
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1883 selected=current_trust_level.name,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1884 styles=["inline"],
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1885 )
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1886
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1887 twomemo_active = dict(device.active).get(twomemo.twomemo.NAMESPACE)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1888 if twomemo_active is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1889 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1890 trust_ui.addLabel(D_("(not available for Twomemo)"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1891 if twomemo_active is False:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1892 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1893 trust_ui.addLabel(D_("(inactive for Twomemo)"))
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1894
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1895 oldmemo_active = dict(device.active).get(oldmemo.oldmemo.NAMESPACE)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1896 if oldmemo_active is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1897 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1898 trust_ui.addLabel(D_("(not available for Oldmemo)"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1899 if oldmemo_active is False:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1900 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1901 trust_ui.addLabel(D_("(inactive for Oldmemo)"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1902
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1903 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1904 trust_ui.addEmpty()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1905
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1906 return result
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1907
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1908 @staticmethod
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1909 def __get_joined_muc_users(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1910 client: SatXMPPClient, xep_0045: XEP_0045, room_jid: jid.JID
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1911 ) -> Set[str]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1912 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1913 @param client: The client.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1914 @param xep_0045: A MUC plugin instance.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1915 @param room_jid: The room JID.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1916 @return: A set containing the bare JIDs of the MUC participants.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1917 @raise InternalError: if the MUC is not joined or the entity information of a
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1918 participant isn't available.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1919 """
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1920
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1921 bare_jids: Set[str] = set()
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1922
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1923 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	1924 room = cast(muc.Room, xep_0045.get_room(client, room_jid))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1925 except exceptions.NotFound as e:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1926 raise exceptions.InternalError(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1927 "Participant list of unjoined MUC requested."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1928 ) from e
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1929
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1930 for user in cast(Dict[str, muc.User], room.roster).values():
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1931 entity = cast(Optional[SatXMPPEntity], user.entity)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1932 if entity is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1933 raise exceptions.InternalError(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1934 f"Participant list of MUC requested, but the entity information of"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1935 f" the participant {user} is not available."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1936 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1937
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1938 bare_jids.add(entity.jid.userhost())
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1939
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1940 return bare_jids
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1941
3969 8e7d5796fb23 plugin XEP-0391: implement XEP-0391 (Jingle Encrypted Transports) + XEP-0396 (JET-OMEMO): Goffi <goffi@goffi.org> parents: 3967 diff changeset	1942 async def get_session_manager(self, profile: str) -> omemo.SessionManager:
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1943 """
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1944 @param profile: The profile to prepare for.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1945 @return: A session manager instance for this profile. Creates a new instance if
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1946 none was prepared before.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1947 """
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	1948
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	1949 try:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1950 # Try to return the session manager
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1951 return self.__session_managers[profile]
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1952 except KeyError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1953 # If a session manager for that profile doesn't exist yet, check whether it is
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1954 # currently being built. A session manager being built is signified by the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1955 # profile key existing on __session_manager_waiters.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1956 if profile in self.__session_manager_waiters:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1957 # If the session manager is being built, add ourselves to the waiting
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1958 # queue
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1959 deferred = defer.Deferred()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1960 self.__session_manager_waiters[profile].append(deferred)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1961 return cast(omemo.SessionManager, await deferred)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1962
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1963 # If the session manager is not being built, do so here.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1964 self.__session_manager_waiters[profile] = []
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1965
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1966 # Build and store the session manager
3929 42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1967 try:
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1968 session_manager = await prepare_for_profile(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1969 self.host, profile, initial_own_label="Libervia"
3929 42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1970 )
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1971 except Exception as e:
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1972 # In case of an error during initalization, notify the waiters accordingly
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1973 # and delete them
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1974 for waiter in self.__session_manager_waiters[profile]:
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1975 waiter.errback(e)
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1976 del self.__session_manager_waiters[profile]
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1977
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1978 # Re-raise the exception
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1979 raise
42d3110ac9b1 plugin XEP-0384: proper handling of exceptions in `prepare_for_profile` Syndace <me@syndace.dev> parents: 3919 diff changeset	1980
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1981 self.__session_managers[profile] = session_manager
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1982
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1983 # Notify the waiters and delete them
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1984 for waiter in self.__session_manager_waiters[profile]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1985 waiter.callback(session_manager)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1986 del self.__session_manager_waiters[profile]
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1987
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1988 return session_manager
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	1989
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1990 async def __message_received_trigger_atm(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1991 self,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1992 client: SatXMPPClient,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1993 message_elt: domish.Element,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1994 session_manager: omemo.SessionManager,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1995 sender_device_information: omemo.DeviceInformation,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	1996 timestamp: datetime,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1997 ) -> None:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1998 """Check a newly decrypted message stanza for ATM content and perform ATM in case.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	1999
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2000 @param client: The client which received the message.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2001 @param message_elt: The message element. Can be modified.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2002 @param session_manager: The session manager.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2003 @param sender_device_information: Information about the device that sent/encrypted
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2004 the message.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2005 @param timestamp: Timestamp extracted from the SCE time affix.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2006 """
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2007
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2008 trust_message_cache = persistent.LazyPersistentBinaryDict(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2009 "XEP-0384/TM", client.profile
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2010 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2011
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2012 new_cache_entries: Set[TrustMessageCacheEntry] = set()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2013
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2014 for trust_message_elt in message_elt.elements(NS_TM, "trust-message"):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2015 assert isinstance(trust_message_elt, domish.Element)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2016
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2017 try:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2018 TRUST_MESSAGE_SCHEMA.validate(trust_message_elt.toXml())
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2019 except xmlschema.XMLSchemaValidationError as e:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2020 raise exceptions.ParsingError(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2021 "<trust-message/> element doesn't pass schema validation."
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2022 ) from e
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2023
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2024 if trust_message_elt["usage"] != NS_ATM:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2025 # Skip non-ATM trust message
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2026 continue
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2027
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2028 if trust_message_elt["encryption"] != OMEMO.NS_TWOMEMO:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2029 # Skip non-twomemo trust message
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2030 continue
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2031
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2032 for key_owner_elt in trust_message_elt.elements(NS_TM, "key-owner"):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2033 assert isinstance(key_owner_elt, domish.Element)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2034
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2035 key_owner_jid = jid.JID(key_owner_elt["jid"]).userhostJID()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2036
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2037 for trust_elt in key_owner_elt.elements(NS_TM, "trust"):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2038 assert isinstance(trust_elt, domish.Element)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2039
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2040 new_cache_entries.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2041 TrustMessageCacheEntry(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2042 sender_jid=jid.JID(sender_device_information.bare_jid),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2043 sender_key=sender_device_information.identity_key,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2044 timestamp=timestamp,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2045 trust_update=TrustUpdate(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2046 target_jid=key_owner_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2047 target_key=base64.b64decode(str(trust_elt)),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2048 target_trust=True,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2049 ),
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2050 )
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2051 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2052
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2053 for distrust_elt in key_owner_elt.elements(NS_TM, "distrust"):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2054 assert isinstance(distrust_elt, domish.Element)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2055
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2056 new_cache_entries.add(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2057 TrustMessageCacheEntry(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2058 sender_jid=jid.JID(sender_device_information.bare_jid),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2059 sender_key=sender_device_information.identity_key,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2060 timestamp=timestamp,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2061 trust_update=TrustUpdate(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2062 target_jid=key_owner_jid,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2063 target_key=base64.b64decode(str(distrust_elt)),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2064 target_trust=False,
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2065 ),
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2066 )
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2067 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2068
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2069 # Load existing cache entries
4216 1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	2070 existing_cache_entries = {
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	2071 TrustMessageCacheEntry.from_dict(d)
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	2072 for d in await trust_message_cache.get("cache", [])
1a7a3e4b52a4 core (memory/migration): Goffi <goffi@goffi.org> parents: 4164 diff changeset	2073 }
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2074
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2075 # Discard cache entries by timestamp comparison
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2076 existing_by_target = {
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2077 (
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2078 cache_entry.trust_update.target_jid.userhostJID(),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2079 cache_entry.trust_update.target_key,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2080 ): cache_entry
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2081 for cache_entry in existing_cache_entries
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2082 }
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2083
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2084 # Iterate over a copy here, such that new_cache_entries can be modified
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2085 for new_cache_entry in set(new_cache_entries):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2086 existing_cache_entry = existing_by_target.get(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2087 (
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2088 new_cache_entry.trust_update.target_jid.userhostJID(),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2089 new_cache_entry.trust_update.target_key,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2090 ),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2091 None,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2092 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2093
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2094 if existing_cache_entry is not None:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2095 if existing_cache_entry.timestamp > new_cache_entry.timestamp:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2096 # If the existing cache entry is newer than the new cache entry,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2097 # discard the new one in favor of the existing one
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2098 new_cache_entries.remove(new_cache_entry)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2099 else:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2100 # Otherwise, discard the existing cache entry. This includes the case
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2101 # when both cache entries have matching timestamps.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2102 existing_cache_entries.remove(existing_cache_entry)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2103
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2104 # If the sending device is trusted, apply the new cache entries
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2105 applied_trust_updates: Set[TrustUpdate] = set()
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2106
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2107 if TrustLevel(sender_device_information.trust_level_name) is TrustLevel.TRUSTED:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2108 # Iterate over a copy such that new_cache_entries can be modified
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2109 for cache_entry in set(new_cache_entries):
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2110 trust_update = cache_entry.trust_update
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2111
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2112 trust_level = (
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2113 TrustLevel.TRUSTED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2114 if trust_update.target_trust
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2115 else TrustLevel.DISTRUSTED
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2116 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2117
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2118 await session_manager.set_trust(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2119 trust_update.target_jid.userhost(),
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2120 trust_update.target_key,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2121 trust_level.name,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2122 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2123
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2124 applied_trust_updates.add(trust_update)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2125
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2126 new_cache_entries.remove(cache_entry)
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2127
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2128 # Store the remaining existing and new cache entries
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2129 await trust_message_cache.force(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2130 "cache", [tm.to_dict() for tm in existing_cache_entries \| new_cache_entries]
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2131 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2132
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2133 # If the trust of at least one device was modified, run the ATM cache update logic
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2134 if len(applied_trust_updates) > 0:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2135 await manage_trust_message_cache(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2136 client, session_manager, frozenset(applied_trust_updates)
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2137 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2138
3972 5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2139 async def _message_received_trigger(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2140 self,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2141 client: SatXMPPClient,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2142 message_elt: domish.Element,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2143 post_treat: defer.Deferred,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2144 ) -> bool:
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2145 """
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2146 @param client: The client which received the message.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2147 @param message_elt: The message element. Can be modified.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2148 @param post_treat: A deferred which evaluates to a :class:`MessageData` once the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2149 message has fully progressed through the message receiving flow. Can be used
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2150 to apply treatments to the fully processed message, like marking it as
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2151 encrypted.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2152 @return: Whether to continue the message received flow.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2153 """
4007 1d5a81e3c9e8 plugin XEP-0384: skip MessageReceived trigger when in a component: Goffi <goffi@goffi.org> parents: 4004 diff changeset	2154 if client.is_component:
1d5a81e3c9e8 plugin XEP-0384: skip MessageReceived trigger when in a component: Goffi <goffi@goffi.org> parents: 4004 diff changeset	2155 return True
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2156 muc_plaintext_cache_key: Optional[MUCPlaintextCacheKey] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2157
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2158 sender_jid = jid.JID(message_elt["from"])
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2159 feedback_jid: jid.JID
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2160
3972 5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2161 message_type = message_elt.getAttribute("type", C.MESS_TYPE_NORMAL)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2162 is_muc_message = message_type == C.MESS_TYPE_GROUPCHAT
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2163 if is_muc_message:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2164 if self.__xep_0045 is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2165 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2166 "Ignoring MUC message since plugin XEP-0045 is not available."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2167 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2168 # Can't handle a MUC message without XEP-0045, let the flow continue
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2169 # normally
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2170 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2171
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2172 room_jid = feedback_jid = sender_jid.userhostJID()
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2173
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2174 try:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	2175 room = cast(muc.Room, self.__xep_0045.get_room(client, room_jid))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2176 except exceptions.NotFound:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2177 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2178 f"Ignoring MUC message from a room that has not been joined:"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2179 f" {room_jid}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2180 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2181 # Whatever, let the flow continue
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2182 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2183
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2184 sender_user = cast(Optional[muc.User], room.getUser(sender_jid.resource))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2185 if sender_user is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2186 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2187 f"Ignoring MUC message from room {room_jid} since the sender's user"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2188 f" wasn't found {sender_jid.resource}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2189 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2190 # Whatever, let the flow continue
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2191 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2192
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2193 sender_user_jid = cast(Optional[jid.JID], sender_user.entity)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2194 if sender_user_jid is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2195 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2196 f"Ignoring MUC message from room {room_jid} since the sender's bare"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2197 f" JID couldn't be found from its user information: {sender_user}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2198 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2199 # Whatever, let the flow continue
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2200 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2201
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2202 sender_jid = sender_user_jid
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2203
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2204 message_uid: Optional[str] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2205 if self.__xep_0359 is not None:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	2206 message_uid = self.__xep_0359.get_origin_id(message_elt)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2207 if message_uid is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2208 message_uid = message_elt.getAttribute("id")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2209 if message_uid is not None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2210 muc_plaintext_cache_key = MUCPlaintextCacheKey(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2211 client, room_jid, message_uid
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2212 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2213 else:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2214 # I'm not sure why this check is required, this code is copied from the old
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2215 # plugin.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2216 if sender_jid.userhostJID() == client.jid.userhostJID():
3944 748094d5a74d plugin XEP-0374, XEP-0384: handle cases where "to" is not set in <message> for `feedback_jid` Goffi <goffi@goffi.org> parents: 3943 diff changeset	2217 try:
748094d5a74d plugin XEP-0374, XEP-0384: handle cases where "to" is not set in <message> for `feedback_jid` Goffi <goffi@goffi.org> parents: 3943 diff changeset	2218 feedback_jid = jid.JID(message_elt["to"])
748094d5a74d plugin XEP-0374, XEP-0384: handle cases where "to" is not set in <message> for `feedback_jid` Goffi <goffi@goffi.org> parents: 3943 diff changeset	2219 except KeyError:
748094d5a74d plugin XEP-0374, XEP-0384: handle cases where "to" is not set in <message> for `feedback_jid` Goffi <goffi@goffi.org> parents: 3943 diff changeset	2220 feedback_jid = client.server_jid
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2221 else:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2222 feedback_jid = sender_jid
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2223
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2224 sender_bare_jid = sender_jid.userhost()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2225
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2226 message: Optional[omemo.Message] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2227 encrypted_elt: Optional[domish.Element] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2228
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2229 twomemo_encrypted_elt = cast(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2230 Optional[domish.Element],
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2231 next(message_elt.elements(twomemo.twomemo.NAMESPACE, "encrypted"), None),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2232 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2233
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2234 oldmemo_encrypted_elt = cast(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2235 Optional[domish.Element],
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2236 next(message_elt.elements(oldmemo.oldmemo.NAMESPACE, "encrypted"), None),
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2237 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2238
3943 8dc6a4cfda4b plugin XEP-0384: continue workflow and log error in case of issue in self.__prepare_for_profile: Goffi <goffi@goffi.org> parents: 3933 diff changeset	2239 try:
3969 8e7d5796fb23 plugin XEP-0391: implement XEP-0391 (Jingle Encrypted Transports) + XEP-0396 (JET-OMEMO): Goffi <goffi@goffi.org> parents: 3967 diff changeset	2240 session_manager = await self.get_session_manager(cast(str, client.profile))
3943 8dc6a4cfda4b plugin XEP-0384: continue workflow and log error in case of issue in self.__prepare_for_profile: Goffi <goffi@goffi.org> parents: 3933 diff changeset	2241 except Exception as e:
8dc6a4cfda4b plugin XEP-0384: continue workflow and log error in case of issue in self.__prepare_for_profile: Goffi <goffi@goffi.org> parents: 3933 diff changeset	2242 log.error(f"error while preparing profile for {client.profile}: {e}")
8dc6a4cfda4b plugin XEP-0384: continue workflow and log error in case of issue in self.__prepare_for_profile: Goffi <goffi@goffi.org> parents: 3933 diff changeset	2243 # we don't want to block the workflow
8dc6a4cfda4b plugin XEP-0384: continue workflow and log error in case of issue in self.__prepare_for_profile: Goffi <goffi@goffi.org> parents: 3933 diff changeset	2244 return True
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2245
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2246 if twomemo_encrypted_elt is not None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2247 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2248 message = twomemo.etree.parse_message(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2249 xml_tools.domish_elt_2_et_elt(twomemo_encrypted_elt), sender_bare_jid
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2250 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2251 except (ValueError, XMLSchemaValidationError):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2252 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2253 f"Ingoring malformed encrypted message for namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2254 f" {twomemo.twomemo.NAMESPACE}: {twomemo_encrypted_elt.toXml()}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2255 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2256 else:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2257 encrypted_elt = twomemo_encrypted_elt
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2258
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2259 if oldmemo_encrypted_elt is not None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2260 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2261 message = await oldmemo.etree.parse_message(
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	2262 xml_tools.domish_elt_2_et_elt(oldmemo_encrypted_elt),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2263 sender_bare_jid,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2264 client.jid.userhost(),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2265 session_manager,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2266 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2267 except (ValueError, XMLSchemaValidationError):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2268 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2269 f"Ingoring malformed encrypted message for namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2270 f" {oldmemo.oldmemo.NAMESPACE}: {oldmemo_encrypted_elt.toXml()}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2271 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2272 except omemo.SenderNotFound:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2273 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2274 f"Ingoring encrypted message for namespace"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2275 f" {oldmemo.oldmemo.NAMESPACE} by unknown sender:"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2276 f" {oldmemo_encrypted_elt.toXml()}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2277 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2278 else:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2279 encrypted_elt = oldmemo_encrypted_elt
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2280
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2281 if message is None or encrypted_elt is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2282 # None of our business, let the flow continue
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2283 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2284
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2285 message_elt.children.remove(encrypted_elt)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2286
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2287 log.debug(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2288 f"{message.namespace} message of type {message_type} received from"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2289 f" {sender_bare_jid}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2290 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2291
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2292 plaintext: Optional[bytes]
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2293 device_information: omemo.DeviceInformation
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2294
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2295 if (
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2296 muc_plaintext_cache_key is not None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2297 and muc_plaintext_cache_key in self.__muc_plaintext_cache
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2298 ):
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2299 # Use the cached plaintext
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2300 plaintext = self.__muc_plaintext_cache.pop(muc_plaintext_cache_key)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2301
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2302 # Since this message was sent by us, use the own device information here
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2303 device_information, __ = await session_manager.get_own_device_information()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2304 else:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2305 try:
3969 8e7d5796fb23 plugin XEP-0391: implement XEP-0391 (Jingle Encrypted Transports) + XEP-0396 (JET-OMEMO): Goffi <goffi@goffi.org> parents: 3967 diff changeset	2306 plaintext, device_information, __ = await session_manager.decrypt(message)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2307 except omemo.MessageNotForUs:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2308 # The difference between this being a debug or a warning is whether there
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2309 # is a body included in the message. Without a body, we can assume that
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2310 # it's an empty OMEMO message used for protocol stability reasons, which
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2311 # is not expected to be sent to all devices of all recipients. If a body
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2312 # is included, we can assume that the message carries content and we
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2313 # missed out on something.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2314 if len(list(message_elt.elements(C.NS_CLIENT, "body"))) > 0:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2315 client.feedback(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2316 feedback_jid,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2317 D_(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2318 f"An OMEMO message from {sender_jid.full()} has not been"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2319 f" encrypted for our device, we can't decrypt it."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2320 ),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2321 {C.MESS_EXTRA_INFO: C.EXTRA_INFO_DECR_ERR},
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2322 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2323 log.warning("Message not encrypted for us.")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2324 else:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2325 log.debug("Message not encrypted for us.")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2326
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2327 # No point in further processing this message.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2328 return False
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2329 except Exception as e:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2330 log.warning(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2331 _("Can't decrypt message: {reason}\n{xml}").format(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2332 reason=e, xml=message_elt.toXml()
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2333 )
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2334 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2335 client.feedback(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2336 feedback_jid,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2337 D_(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2338 f"An OMEMO message from {sender_jid.full()} can't be decrypted:"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2339 f" {e}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2340 ),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2341 {C.MESS_EXTRA_INFO: C.EXTRA_INFO_DECR_ERR},
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2342 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2343 # No point in further processing this message
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2344 return False
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2345
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2346 affix_values: Optional[SCEAffixValues] = None
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2347
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2348 if message.namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2349 if plaintext is not None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2350 # XEP_0420.unpack_stanza handles the whole unpacking, including the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2351 # relevant modifications to the element
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2352 sce_profile = (
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2353 OMEMO.SCE_PROFILE_GROUPCHAT if is_muc_message else OMEMO.SCE_PROFILE
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2354 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2355 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2356 affix_values = self.__xep_0420.unpack_stanza(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2357 sce_profile, message_elt, plaintext
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2358 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2359 except Exception as e:
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2360 log.warning(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2361 D_(f"Error unpacking SCE-encrypted message: {e}\n{plaintext}")
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2362 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2363 client.feedback(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2364 feedback_jid,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2365 D_(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2366 f"An OMEMO message from {sender_jid.full()} was rejected:"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2367 f" {e}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2368 ),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2369 {C.MESS_EXTRA_INFO: C.EXTRA_INFO_DECR_ERR},
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2370 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2371 # No point in further processing this message
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2372 return False
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2373 else:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2374 if affix_values.timestamp is not None:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2375 # TODO: affix_values.timestamp contains the timestamp included in
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2376 # the encrypted element here. The XEP says it SHOULD be displayed
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2377 # with the plaintext by clients.
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2378 pass
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2379
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2380 if message.namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2381 # Remove all body elements from the original element, since those act as
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2382 # fallbacks in case the encryption protocol is not supported
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2383 for child in message_elt.elements():
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2384 if child.name == "body":
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2385 message_elt.children.remove(child)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2386
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2387 if plaintext is not None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2388 # Add the decrypted body
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2389 message_elt.addElement("body", content=plaintext.decode("utf-8"))
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2390
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2391 # Mark the message as trusted or untrusted. Undecided counts as untrusted here.
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2392 trust_level = await session_manager._evaluate_custom_trust_level(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2393 device_information
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2394 )
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2395
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2396 if trust_level is omemo.TrustLevel.TRUSTED:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	2397 post_treat.addCallback(client.encryption.mark_as_trusted)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2398 else:
4037 524856bd7b19 massive refactoring to switch from camelCase to snake_case: Goffi <goffi@goffi.org> parents: 4007 diff changeset	2399 post_treat.addCallback(client.encryption.mark_as_untrusted)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2400
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2401 # Mark the message as originally encrypted
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2402 post_treat.addCallback(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2403 client.encryption.mark_as_encrypted, namespace=message.namespace
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2404 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2405
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2406 # Handle potential ATM trust updates
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2407 if affix_values is not None and affix_values.timestamp is not None:
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2408 await self.__message_received_trigger_atm(
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2409 client,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2410 message_elt,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2411 session_manager,
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2412 device_information,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2413 affix_values.timestamp,
3967 f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2414 )
f461f11ea176 plugin XEP-0384: Implementation of Automatic Trust Management: Syndace <me@syndace.dev> parents: 3944 diff changeset	2415
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2416 # Message processed successfully, continue with the flow
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2417 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2418
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2419 async def __send_trigger(self, client: SatXMPPClient, stanza: domish.Element) -> bool:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2420 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2421 @param client: The client sending this message.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2422 @param stanza: The stanza that is about to be sent. Can be modified.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2423 @return: Whether the send message flow should continue or not.
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2424 """
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2425 # SCE is only applicable to message and IQ stanzas
3919 7b2c51bcc8f5 plugin XEP-0384: temporarily disable `<iq>` stanza encryption: Goffi <goffi@goffi.org> parents: 3918 diff changeset	2426 # FIXME: temporary disabling IQ stanza encryption
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2427 if stanza.name not in {"message"}: # , "iq" }:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2428 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2429
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2430 # Get the intended recipient
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2431 recipient = stanza.getAttribute("to", None)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2432 if recipient is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2433 if stanza.name == "message":
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2434 # Message stanzas must have a recipient
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2435 raise exceptions.InternalError(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2436 f"Message without recipient encountered. Blocking further processing"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2437 f" to avoid leaking plaintext data: {stanza.toXml()}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2438 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2439
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2440 # IQs without a recipient are a thing, I believe those simply target the
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2441 # server and are thus not eligible for e2ee anyway.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2442 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2443
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2444 # Parse the JID
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2445 recipient_bare_jid = jid.JID(recipient).userhostJID()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2446
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2447 # Check whether encryption with twomemo is requested
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2448 encryption = client.encryption.getSession(recipient_bare_jid)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2449
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2450 if encryption is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2451 # Encryption is not requested for this recipient
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2452 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2453
4164 15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2454 encryption_ns = encryption["plugin"].namespace
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2455 # All pre-checks done, we can start encrypting!
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2456 if encryption_ns in (twomemo.twomemo.NAMESPACE, oldmemo.oldmemo.NAMESPACE):
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2457 await self.encrypt(
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2458 client,
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2459 encryption_ns,
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2460 stanza,
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2461 recipient_bare_jid,
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2462 stanza.getAttribute("type", C.MESS_TYPE_NORMAL) == C.MESS_TYPE_GROUPCHAT,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2463 stanza.getAttribute("id", None),
4164 15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2464 )
15482dc0b5d1 plugin XEP-0384: fix trigger used for OLDMEMO: Goffi <goffi@goffi.org> parents: 4150 diff changeset	2465 else:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2466 # Encryption is requested for this recipient, but not with twomemo
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2467 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2468
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2469 # Add a store hint if this is a message stanza
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2470 if stanza.name == "message":
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2471 self.__xep_0334.add_hint_elements(stanza, ["store"])
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2472
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2473 # Let the flow continue.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2474 return True
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2475
4219 1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2476 async def download_missing_device_lists(
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2477 self,
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2478 client: SatXMPPClient,
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2479 namespace: NamespaceType,
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2480 recipients: Iterable[jid.JID],
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2481 session_manager: omemo.SessionManager,
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2482 ) -> None:
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2483 """Retrieves missing device lists for recipients outside the profile's roster.
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2484
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2485 @param client: XMPP client.
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2486 @param namespace: The namespace of the OMEMO version to use.
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2487 @param recipients: Recipients to verify device list presence.
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2488 @param session_manager: OMEMO session manager.
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2489 """
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2490 recipients = [j.userhostJID() for j in recipients]
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2491 not_in_roster = [j for j in recipients if not client.roster.is_jid_in_roster(j)]
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2492 for bare_jid in not_in_roster:
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2493 device_information = await session_manager.get_device_information(
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2494 bare_jid.userhost()
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2495 )
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2496 if not device_information or not all(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2497 namespace in di.namespaces for di in device_information
4219 1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2498 ):
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2499 if namespace == self.NS_TWOMEMO:
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2500 algo, node = "OMEMO", TWOMEMO_DEVICE_LIST_NODE
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2501 elif namespace == self.NS_OLDMEMO:
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2502 algo, node = "OMEMO_legacy", OLDMEMO_DEVICE_LIST_NODE
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2503 else:
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2504 raise ValueError(f"Invalid namespace: {namespace!r}")
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2505
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2506 try:
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2507 items, __ = await self._j.get_items(client, bare_jid, node, 1)
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2508
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2509 except Exception:
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2510 log.exception(f"Can't find {algo} devices list for {bare_jid}.")
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2511 else:
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2512 await self._update_device_list(client, bare_jid, items)
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2513 log.warning(f"{algo} devices list updated for {bare_jid}.")
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2514
3972 5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2515 async def encrypt(
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2516 self,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2517 client: SatXMPPClient,
4219 1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2518 namespace: NamespaceType,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2519 stanza: domish.Element,
3972 5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2520 recipient_jids: Union[jid.JID, Set[jid.JID]],
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2521 is_muc_message: bool,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2522 stanza_id: Optional[str],
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2523 ) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2524 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2525 @param client: The client.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2526 @param namespace: The namespace of the OMEMO version to use.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2527 @param stanza: The stanza. Twomemo will encrypt the whole stanza using SCE,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2528 oldmemo will encrypt only the body. The stanza is modified by this call.
3972 5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2529 @param recipient_jid: The JID of the recipients.
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2530 Can be a bare (aka "userhost") JIDs but doesn't have to.
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2531 A single JID can be used.
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2532 @param is_muc_message: Whether the stanza is a message stanza to a MUC room.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2533 @param stanza_id: The id of this stanza. Especially relevant for message stanzas
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2534 to MUC rooms such that the outgoing plaintext can be cached for MUC message
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2535 reflection handling.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2536
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2537 @warning: The calling code MUST take care of adding the store message processing
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2538 hint to the stanza if applicable! This can be done before or after this call,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2539 the order doesn't matter.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2540 """
3972 5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2541 if isinstance(recipient_jids, jid.JID):
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2542 recipient_jids = {recipient_jids}
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2543 if not recipient_jids:
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2544 raise exceptions.InternalError("At least one JID must be specified")
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2545 recipient_jid = next(iter(recipient_jids))
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2546
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2547 muc_plaintext_cache_key: Optional[MUCPlaintextCacheKey] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2548
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2549 recipient_bare_jids: Set[str]
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2550 feedback_jid: jid.JID
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2551
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2552 if is_muc_message:
3972 5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2553 if len(recipient_jids) != 1:
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2554 raise exceptions.InternalError(
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2555 'Only one JID can be set when "is_muc_message" is set'
5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2556 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2557 if self.__xep_0045 is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2558 raise exceptions.InternalError(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2559 "Encryption of MUC message requested, but plugin XEP-0045 is not"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2560 " available."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2561 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2562
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2563 if stanza_id is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2564 raise exceptions.InternalError(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2565 "Encryption of MUC message requested, but stanza id not available."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2566 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2567
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2568 room_jid = feedback_jid = recipient_jid.userhostJID()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2569
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2570 recipient_bare_jids = self.__get_joined_muc_users(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2571 client, self.__xep_0045, room_jid
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2572 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2573
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2574 muc_plaintext_cache_key = MUCPlaintextCacheKey(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2575 client=client, room_jid=room_jid, message_uid=stanza_id
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2576 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2577 else:
3972 5fbdf986670c plugin pte: Pubsub Target Encryption implementation: Goffi <goffi@goffi.org> parents: 3969 diff changeset	2578 recipient_bare_jids = {r.userhost() for r in recipient_jids}
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2579 feedback_jid = recipient_jid.userhostJID()
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2580
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2581 log.debug(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2582 f"Intercepting message that is to be encrypted by {namespace} for"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2583 f" {recipient_bare_jids}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2584 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2585
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2586 def prepare_stanza() -> Optional[bytes]:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2587 """Prepares the stanza for encryption.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2588
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2589 Does so by removing all parts that are not supposed to be sent in plain. Also
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2590 extracts/prepares the plaintext to encrypt.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2591
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2592 @return: The plaintext to encrypt. Returns ``None`` in case body-only
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2593 encryption is requested and no body was found. The function should
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2594 gracefully return in that case, i.e. it's not a critical error that should
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2595 abort the message sending flow.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2596 """
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2597
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2598 if namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2599 return self.__xep_0420.pack_stanza(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2600 OMEMO.SCE_PROFILE_GROUPCHAT if is_muc_message else OMEMO.SCE_PROFILE,
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2601 stanza,
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2602 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2603
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2604 if namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2605 plaintext: Optional[bytes] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2606
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2607 for child in stanza.elements():
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2608 if child.name == "body" and plaintext is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2609 plaintext = str(child).encode("utf-8")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2610
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2611 # Any other sensitive elements to remove here?
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2612 if child.name in {"body", "html"}:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2613 stanza.children.remove(child)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2614
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2615 if plaintext is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2616 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2617 "No body found in intercepted message to be encrypted with"
4231 e11b13418ba6 plugin XEP-0353, XEP-0234, jingle: WebRTC data channel signaling implementation: Goffi <goffi@goffi.org> parents: 4219 diff changeset	2618 f" oldmemo. [{client.profile}]"
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2619 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2620
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2621 return plaintext
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2622
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2623 return assert_never(namespace)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2624
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2625 # The stanza/plaintext preparation was moved into its own little function for type
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2626 # safety reasons.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2627 plaintext = prepare_stanza()
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2628 if plaintext is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2629 return
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2630
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2631 log.debug(f"Plaintext to encrypt: {plaintext}")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2632
3969 8e7d5796fb23 plugin XEP-0391: implement XEP-0391 (Jingle Encrypted Transports) + XEP-0396 (JET-OMEMO): Goffi <goffi@goffi.org> parents: 3967 diff changeset	2633 session_manager = await self.get_session_manager(client.profile)
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2634 await self.download_missing_device_lists(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2635 client, namespace, recipient_jids, session_manager
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2636 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2637
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	2638 try:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2639 messages, encryption_errors = await session_manager.encrypt(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2640 frozenset(recipient_bare_jids),
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2641 {namespace: plaintext},
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2642 backend_priority_order=[namespace],
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2643 identifier=feedback_jid.userhost(),
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	2644 )
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	2645 except Exception as e:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2646 msg = _(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2647 # pylint: disable=consider-using-f-string
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2648 "Can't encrypt message for {entities}: {reason}".format(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2649 entities=", ".join(recipient_bare_jids), reason=e
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2650 )
3237 b0c57c9a4bd8 plugin XEP-0384: OMEMO trust policy: Goffi <goffi@goffi.org> parents: 3236 diff changeset	2651 )
2859 4e875d9eea48 plugin XEP-0384: give feedback to client when encryption failed Goffi <goffi@goffi.org> parents: 2858 diff changeset	2652 log.warning(msg)
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2653 client.feedback(feedback_jid, msg, {C.MESS_EXTRA_INFO: C.EXTRA_INFO_ENCR_ERR})
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	2654 raise e
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	2655
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2656 if len(encryption_errors) > 0:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2657 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2658 f"Ignored the following non-critical encryption errors:"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2659 f" {encryption_errors}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2660 )
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	2661
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2662 encrypted_errors_stringified = ", ".join(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2663 [
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2664 f"device {err.device_id} of {err.bare_jid} under namespace"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2665 f" {err.namespace}"
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2666 for err in encryption_errors
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2667 ]
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2668 )
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	2669
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2670 client.feedback(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2671 feedback_jid,
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2672 D_(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2673 "There were non-critical errors during encryption resulting in some"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2674 " of your destinees' devices potentially not receiving the message."
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2675 " This happens when the encryption data/key material of a device is"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2676 " incomplete or broken, which shouldn't happen for actively used"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2677 " devices, and can usually be ignored. The following devices are"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2678 f" affected: {encrypted_errors_stringified}."
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2679 ),
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2680 )
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	2681
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2682 message = next(message for message in messages if message.namespace == namespace)
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	2683
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2684 if namespace == twomemo.twomemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2685 # Add the encrypted element
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2686 stanza.addChild(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2687 xml_tools.et_elt_2_domish_elt(twomemo.etree.serialize_message(message))
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2688 )
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2689
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2690 if namespace == oldmemo.oldmemo.NAMESPACE:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2691 # Add the encrypted element
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2692 stanza.addChild(
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2693 xml_tools.et_elt_2_domish_elt(oldmemo.etree.serialize_message(message))
0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2694 )
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2695
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2696 if muc_plaintext_cache_key is not None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2697 self.__muc_plaintext_cache[muc_plaintext_cache_key] = plaintext
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	2698
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2699 async def __on_device_list_update(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2700 self, items_event: pubsub.ItemsEvent, profile: str
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2701 ) -> None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2702 """Handle device list updates fired by PEP.
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2703
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2704 @param items_event: The event.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2705 @param profile: The profile this event belongs to.
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2706 """
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	2707
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2708 sender = cast(jid.JID, items_event.sender)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2709 items = cast(List[domish.Element], items_event.items)
4219 1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2710 client = self.host.get_client(profile)
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2711 await self._update_device_list(client, sender, items)
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2712
1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2713 async def _update_device_list(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2714 self, client: SatXMPPEntity, sender: jid.JID, items: list[domish.Element]
4219 1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2715 ) -> None:
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2716
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2717 if len(items) > 1:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2718 log.warning("Ignoring device list update with more than one element.")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2719 return
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2720
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2721 item = next(iter(items), None)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2722 if item is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2723 log.debug("Ignoring empty device list update.")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2724 return
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2725
3914 4cb38c8312a1 plugin XEP-0384, xml_tools: avoid `getItems` timeout + fix empty node crash + parsing: Goffi <goffi@goffi.org> parents: 3911 diff changeset	2726 item_elt = xml_tools.domish_elt_2_et_elt(item)
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	2727
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2728 device_list: Dict[int, Optional[str]] = {}
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2729 namespace: Optional[str] = None
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2730
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2731 list_elt = item_elt.find(f"{{{twomemo.twomemo.NAMESPACE}}}devices")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2732 if list_elt is not None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2733 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2734 device_list = twomemo.etree.parse_device_list(list_elt)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2735 except XMLSchemaValidationError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2736 pass
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	2737 else:
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2738 namespace = twomemo.twomemo.NAMESPACE
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2739
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2740 list_elt = item_elt.find(f"{{{oldmemo.oldmemo.NAMESPACE}}}list")
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2741 if list_elt is not None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2742 try:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2743 device_list = oldmemo.etree.parse_device_list(list_elt)
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2744 except XMLSchemaValidationError:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2745 pass
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2746 else:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2747 namespace = oldmemo.oldmemo.NAMESPACE
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	2748
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2749 if namespace is None:
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2750 log.warning(
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2751 f"Malformed device list update item:"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2752 f" {ET.tostring(item_elt, encoding='unicode')}"
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2753 )
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2754 return
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2755
4219 1b5cf2ee1d86 plugin XEP-0384, XEP-0391: download missing devices list: Goffi <goffi@goffi.org> parents: 4218 diff changeset	2756 session_manager = await self.get_session_manager(client.profile)
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2757
8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2758 await session_manager.update_device_list(
4270 0d7bb4df2343 Reformatted code base using black. Goffi <goffi@goffi.org> parents: 4231 diff changeset	2759 namespace, sender.userhost(), device_list
3911 8289ac1b34f4 plugin XEP-0384: Fully reworked to adjust to the reworked python-omemo: Syndace <me@syndace.dev> parents: 3797 diff changeset	2760 )

Mercurial > libervia-backend

annotate libervia/backend/plugins/plugin_xep_0384.py @ 4306:94e0968987cd