libervia-backend: sat/plugins/plugin_xep

annotate sat/plugins/plugin_xep_0384.py @ 3231:e756e0eb1be4

core (memory/encryption): automatic start encryption if peer send encrypted message: If peer sends encrypted message and we have no encryption activated, we automatically start encryption to avoid sending plain text message when answering. markAsEncrypted now needs the encryption algorithm namespace as mandatory argument.

author	Goffi <goffi@goffi.org>
date	Mon, 23 Mar 2020 17:52:18 +0100
parents	806a7936a591
children	9477f3197981

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
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	3 # SAT plugin for OMEMO encryption
3136 9d0df638c8b4 dates update Goffi <goffi@goffi.org> parents: 3104 diff changeset	4 # Copyright (C) 2009-2020 Jérôme Poisson (goffi@goffi.org)
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
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	19 from sat.core.i18n import _, D_
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	20 from sat.core.constants import Const as C
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	21 from sat.core.log import getLogger
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	22 from sat.core import exceptions
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	23 from twisted.internet import defer, reactor
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	24 from twisted.words.xish import domish
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	25 from twisted.words.protocols.jabber import jid
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	26 from twisted.words.protocols.jabber import error as jabber_error
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	27 from sat.memory import persistent
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	28 from functools import partial
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	29 from sat.tools import xml_tools
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	30 import logging
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	31 import random
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	32 import base64
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	33 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	34 import omemo
3084 ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	35 from omemo import exceptions as omemo_excpt
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	36 from omemo.extendedpublicbundle import ExtendedPublicBundle
3084 ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	37 except ImportError:
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	38 raise exceptions.MissingModule(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	39 'Missing module omemo, please download/install it. You can use '
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	40 '"pip install omemo"'
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	41 )
3084 ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	42 try:
ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	43 from omemo_backend_signal import BACKEND as omemo_backend
ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	44 except ImportError:
ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	45 raise exceptions.MissingModule(
ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	46 'Missing module omemo-backend-signal, please download/install it. You can use '
ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	47 '"pip install omemo-backend-signal"'
ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	48 )
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	49
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	50 log = getLogger(__name__)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	51
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	52 PLUGIN_INFO = {
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	53 C.PI_NAME: "OMEMO",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	54 C.PI_IMPORT_NAME: "XEP-0384",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	55 C.PI_TYPE: "SEC",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	56 C.PI_PROTOCOLS: ["XEP-0384"],
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	57 C.PI_DEPENDENCIES: ["XEP-0163", "XEP-0280", "XEP-0334", "XEP-0060"],
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	58 C.PI_RECOMMENDATIONS: ["XEP-0045", "XEP-0359", C.TEXT_CMDS],
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	59 C.PI_MAIN: "OMEMO",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	60 C.PI_HANDLER: "no",
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	61 C.PI_DESCRIPTION: _("""Implementation of OMEMO"""),
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	62 }
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	63
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	64 OMEMO_MIN_VER = (0, 11, 0)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	65 NS_OMEMO = "eu.siacs.conversations.axolotl"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	66 NS_OMEMO_DEVICES = NS_OMEMO + ".devicelist"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	67 NS_OMEMO_BUNDLE = NS_OMEMO + ".bundles:{device_id}"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	68 KEY_STATE = "STATE"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	69 KEY_DEVICE_ID = "DEVICE_ID"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	70 KEY_SESSION = "SESSION"
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	71 KEY_TRUST = "TRUST"
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	72 KEY_ACTIVE_DEVICES = "DEVICES"
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	73 KEY_INACTIVE_DEVICES = "INACTIVE_DEVICES"
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	74 KEY_ALL_JIDS = "ALL_JIDS"
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	75 # time before plaintext cache for MUC is expired
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	76 # expressed in seconds, reset on each new MUC message
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	77 MUC_CACHE_TTL = 60 * 5
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	78
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	79
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	80 # we want to manage log emitted by omemo module ourselves
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	81
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	82 class SatHandler(logging.Handler):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	83
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	84 def emit(self, record):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	85 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	86
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	87 @staticmethod
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	88 def install():
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	89 omemo_sm_logger = logging.getLogger("omemo.SessionManager")
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	90 omemo_sm_logger.propagate = False
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	91 omemo_sm_logger.addHandler(SatHandler())
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	92
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	93
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	94 SatHandler.install()
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	95
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	96
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	97 def b64enc(data):
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	98 return base64.b64encode(bytes(bytearray(data))).decode("US-ASCII")
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	99
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	100
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	101 def promise2Deferred(promise_):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	102 """Create a Deferred and fire it when promise is resolved
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	103
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	104 @param promise_(promise.Promise): promise to convert
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	105 @return (defer.Deferred): deferred instance linked to the promise
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	106 """
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	107 d = defer.Deferred()
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	108 promise_.then(d.callback, d.errback)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	109 return d
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	110
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	111
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	112 class OmemoStorage(omemo.Storage):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	113
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	114 def __init__(self, client, device_id, all_jids, persistent_dict):
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	115 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	116 @param persistent_dict(persistent.LazyPersistentBinaryDict): object which will
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	117 store data in SàT database
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	118 """
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	119 self.own_bare_jid_s = client.jid.userhost()
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 self.device_id = device_id
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	121 self.all_jids = all_jids
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	122 self.data = persistent_dict
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	123
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	124 @property
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	125 def is_async(self):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	126 return True
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	127
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	128 def setCb(self, deferred, callback):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	129 """Associate Deferred and callback
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	130
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	131 callback of omemo.Storage expect a boolean with success state then result
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	132 Deferred on the other hand use 2 methods for callback and errback
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	133 This method use partial to call callback with boolean then result when
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	134 Deferred is called
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	135 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	136 deferred.addCallback(partial(callback, True))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	137 deferred.addErrback(partial(callback, False))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	138
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	139 def _checkJid(self, bare_jid):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	140 """Check if jid is know, and store it if not
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	141
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	142 @param bare_jid(unicode): bare jid to check
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	143 @return (D): Deferred fired when jid is stored
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	144 """
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	145 if bare_jid in self.all_jids:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	146 return defer.succeed(None)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	147 else:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	148 self.all_jids.add(bare_jid)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	149 d = self.data.force(KEY_ALL_JIDS, self.all_jids)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	150 return d
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	151
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	152 def loadOwnData(self, callback):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	153 callback(True, {'own_bare_jid': self.own_bare_jid_s,
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	154 'own_device_id': self.device_id})
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	155
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	156 def storeOwnData(self, callback, own_bare_jid, own_device_id):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	157 if own_bare_jid != self.own_bare_jid_s or own_device_id != self.device_id:
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	158 raise exceptions.InternalError('bare jid or device id inconsistency!')
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	159 callback(True, None)
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	160
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	161 def loadState(self, callback):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	162 d = self.data.get(KEY_STATE)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	163 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	164
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	165 def storeState(self, callback, state):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	166 d = self.data.force(KEY_STATE, state)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	167 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	168
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	169 def loadSession(self, callback, bare_jid, device_id):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	170 key = '\n'.join([KEY_SESSION, bare_jid, str(device_id)])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	171 d = self.data.get(key)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	172 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	173
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	174 def storeSession(self, callback, bare_jid, device_id, session):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	175 key = '\n'.join([KEY_SESSION, bare_jid, str(device_id)])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	176 d = self.data.force(key, session)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	177 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	178
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	179 def deleteSession(self, callback, bare_jid, device_id):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	180 key = '\n'.join([KEY_SESSION, bare_jid, str(device_id)])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	181 d = self.data.remove(key)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	182 self.setCb(d, callback)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	183
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	184 def loadActiveDevices(self, callback, bare_jid):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	185 key = '\n'.join([KEY_ACTIVE_DEVICES, bare_jid])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	186 d = self.data.get(key, {})
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	187 if callback is not None:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	188 self.setCb(d, callback)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	189 return d
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	190
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	191 def loadInactiveDevices(self, callback, bare_jid):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	192 key = '\n'.join([KEY_INACTIVE_DEVICES, bare_jid])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	193 d = self.data.get(key, {})
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	194 if callback is not None:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	195 self.setCb(d, callback)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	196 return d
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	197
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	198 def storeActiveDevices(self, callback, bare_jid, devices):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	199 key = '\n'.join([KEY_ACTIVE_DEVICES, bare_jid])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	200 d = self._checkJid(bare_jid)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	201 d.addCallback(lambda _: self.data.force(key, devices))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	202 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	203
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	204 def storeInactiveDevices(self, callback, bare_jid, devices):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	205 key = '\n'.join([KEY_INACTIVE_DEVICES, bare_jid])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	206 d = self._checkJid(bare_jid)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	207 d.addCallback(lambda _: self.data.force(key, devices))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	208 self.setCb(d, callback)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	209
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	210 def storeTrust(self, callback, bare_jid, device_id, trust):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	211 key = '\n'.join([KEY_TRUST, bare_jid, str(device_id)])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	212 d = self.data.force(key, trust)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	213 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	214
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	215 def loadTrust(self, callback, bare_jid, device_id):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	216 key = '\n'.join([KEY_TRUST, bare_jid, str(device_id)])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	217 d = self.data.get(key)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	218 if callback is not None:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	219 self.setCb(d, callback)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	220 return d
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	221
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	222 def listJIDs(self, callback):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	223 d = defer.succeed(self.all_jids)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	224 if callback is not None:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	225 self.setCb(d, callback)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	226 return d
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	227
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	228 def _deleteJID_logResults(self, results):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	229 failed = [success for success, __ in results if not success]
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	230 if failed:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	231 log.warning(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	232 "delete JID failed for {failed_count} on {total_count} operations"
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	233 .format(failed_count=len(failed), total_count=len(results)))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	234 else:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	235 log.info(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	236 "Delete JID operation succeed ({total_count} operations)."
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	237 .format(total_count=len(results)))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	238
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	239 def _deleteJID_gotDevices(self, results, bare_jid):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	240 assert len(results) == 2
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	241 active_success, active_devices = results[0]
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	242 inactive_success, inactive_devices = results[0]
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	243 d_list = []
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	244 for success, devices in results:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	245 if not success:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	246 log.warning("Can't retrieve devices for {bare_jid}: {reason}"
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	247 .format(bare_jid=bare_jid, reason=active_devices))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	248 else:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	249 for device_id in devices:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	250 for key in (KEY_SESSION, KEY_TRUST):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	251 k = '\n'.join([key, bare_jid, str(device_id)])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	252 d_list.append(self.data.remove(k))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	253
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	254 d_list.append(self.data.remove(KEY_ACTIVE_DEVICES, bare_jid))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	255 d_list.append(self.data.remove(KEY_INACTIVE_DEVICES, bare_jid))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	256 d_list.append(lambda __: self.all_jids.discard(bare_jid))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	257 # FIXME: there is a risk of race condition here,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	258 # if self.all_jids is modified between discard and force)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	259 d_list.append(lambda __: self.data.force(KEY_ALL_JIDS, self.all_jids))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	260 d = defer.DeferredList(d_list)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	261 d.addCallback(self._deleteJID_logResults)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	262 return d
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	263
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	264 def deleteJID(self, callback, bare_jid):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	265 """Retrieve all (in)actives of bare_jid, and delete all related keys"""
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	266 d_list = []
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	267
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	268 key = '\n'.join([KEY_ACTIVE_DEVICES, bare_jid])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	269 d_list.append(self.data.get(key, []))
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	270
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	271 key = '\n'.join([KEY_INACTIVE_DEVICES, bare_jid])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	272 d_inactive = self.data.get(key, {})
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	273 # inactive devices are returned as a dict mapping from devices_id to timestamp
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	274 # but we only need devices ids
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	275 d_inactive.addCallback(lambda devices: [k for k, __ in devices])
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	276
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	277 d_list.append(d_inactive)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	278 d = defer.DeferredList(d_list)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	279 d.addCallback(self._deleteJID_gotDevices, bare_jid)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	280 if callback is not None:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	281 self.setCb(d, callback)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	282 return d
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	283
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	284
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	285 class SatOTPKPolicy(omemo.DefaultOTPKPolicy):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	286 pass
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	287
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	288
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	289 class OmemoSession(object):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	290 """Wrapper to use omemo.OmemoSession with Deferred"""
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	291
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	292 def __init__(self, session):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	293 self._session = session
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	294
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	295 @property
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	296 def republish_bundle(self):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	297 return self._session.republish_bundle
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	298
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	299 @property
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	300 def public_bundle(self):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	301 return self._session.public_bundle
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	302
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	303 @classmethod
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	304 def create(cls, client, storage, my_device_id = None):
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	305 omemo_session_p = omemo.SessionManager.create(
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	306 storage,
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	307 SatOTPKPolicy,
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	308 omemo_backend,
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	309 client.jid.userhost(),
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	310 my_device_id)
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	311 d = promise2Deferred(omemo_session_p)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	312 d.addCallback(lambda session: cls(session))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	313 return d
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	314
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	315 def newDeviceList(self, jid, devices):
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	316 jid = jid.userhost()
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	317 new_device_p = self._session.newDeviceList(jid, devices)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	318 return promise2Deferred(new_device_p)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	319
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	320 def getDevices(self, bare_jid=None):
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	321 bare_jid = bare_jid.userhost()
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	322 get_devices_p = self._session.getDevices(bare_jid=bare_jid)
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	323 return promise2Deferred(get_devices_p)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	324
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	325 def buildSession(self, bare_jid, device, bundle):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	326 bare_jid = bare_jid.userhost()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	327 build_session_p = self._session.buildSession(bare_jid, device, bundle)
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	328 return promise2Deferred(build_session_p)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	329
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	330 def deleteSession(self, bare_jid, device):
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	331 bare_jid = bare_jid.userhost()
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	332 delete_session_p = self._session.deleteSession(bare_jid=bare_jid, device=device)
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	333 return promise2Deferred(delete_session_p)
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	334
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	335 def encryptMessage(self, bare_jids, message, bundles=None, expect_problems=None):
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	336 """Encrypt a message
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	337
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	338 @param bare_jids(iterable[jid.JID]): destinees of the message
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	339 @param message(unicode): message to encode
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	340 @param bundles(dict[jid.JID, dict[int, ExtendedPublicBundle]):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	341 entities => devices => bundles map
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	342 @return D(dict): encryption data
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	343 """
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	344 bare_jids = [e.userhost() for e in bare_jids]
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	345 if bundles is not None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	346 bundles = {e.userhost(): v for e, v in bundles.items()}
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	347 encrypt_mess_p = self._session.encryptMessage(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	348 bare_jids=bare_jids,
3084 ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	349 plaintext=message.encode(),
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	350 bundles=bundles,
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	351 expect_problems=expect_problems)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	352 return promise2Deferred(encrypt_mess_p)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	353
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	354 def decryptMessage(self, bare_jid, device, iv, message, is_pre_key_message,
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	355 ciphertext, additional_information=None, allow_untrusted=False):
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	356 bare_jid = bare_jid.userhost()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	357 decrypt_mess_p = self._session.decryptMessage(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	358 bare_jid=bare_jid,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	359 device=device,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	360 iv=iv,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	361 message=message,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	362 is_pre_key_message=is_pre_key_message,
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	363 ciphertext=ciphertext,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	364 additional_information=additional_information,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	365 allow_untrusted=allow_untrusted
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	366 )
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	367 return promise2Deferred(decrypt_mess_p)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	368
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	369 def setTrust(self, bare_jid, device, key, trusted):
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	370 bare_jid = bare_jid.userhost()
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	371 setTrust_p = self._session.setTrust(
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	372 bare_jid=bare_jid,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	373 device=device,
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	374 key=key,
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	375 trusted=trusted,
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	376 )
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	377 return promise2Deferred(setTrust_p)
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	378
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	379 def getTrustForJID(self, bare_jid):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	380 bare_jid = bare_jid.userhost()
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	381 get_trust_p = self._session.getTrustForJID(bare_jid=bare_jid)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	382 return promise2Deferred(get_trust_p)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	383
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	384
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	385 class OMEMO:
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	386
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	387 def __init__(self, host):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	388 log.info(_("OMEMO plugin initialization (omemo module v{version})").format(
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	389 version=omemo.__version__))
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	390 version = tuple(map(int, omemo.__version__.split('.')[:3]))
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	391 if version < OMEMO_MIN_VER:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	392 log.warning(_(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	393 "Your version of omemo module is too old: {v[0]}.{v[1]}.{v[2]} is "
3098 518208085dfb plugin XEP-0384: typo Goffi <goffi@goffi.org> parents: 3085 diff changeset	394 "minimum required, please update.").format(v=OMEMO_MIN_VER))
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	395 raise exceptions.CancelError("module is too old")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	396 self.host = host
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	397 self._p_hints = host.plugins["XEP-0334"]
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	398 self._p_carbons = host.plugins["XEP-0280"]
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	399 self._p = host.plugins["XEP-0060"]
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	400 self._m = host.plugins.get("XEP-0045")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	401 self._sid = host.plugins.get("XEP-0359")
3172 dcebc585c29f core: renamed "MessageReceived" trigger to "messageReceived" for consistency. Goffi <goffi@goffi.org> parents: 3142 diff changeset	402 host.trigger.add("messageReceived", self._messageReceivedTrigger, priority=100050)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	403 host.trigger.add("sendMessageData", self._sendMessageDataTrigger)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	404 self.host.registerEncryptionPlugin(self, "OMEMO", NS_OMEMO, 100)
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	405 pep = host.plugins['XEP-0163']
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	406 pep.addPEPEvent(
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	407 "OMEMO_DEVICES", NS_OMEMO_DEVICES,
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	408 lambda itemsEvent, profile: defer.ensureDeferred(
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	409 self.onNewDevices(itemsEvent, profile))
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	410 )
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	411 try:
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	412 self.text_cmds = self.host.plugins[C.TEXT_CMDS]
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	413 except KeyError:
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	414 log.info(_("Text commands not available"))
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	415 else:
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	416 self.text_cmds.registerTextCommands(self)
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	417
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	418 # Text commands #
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	419
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	420 async def cmd_omemo_reset(self, client, mess_data):
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	421 """reset OMEMO session (use only if encryption is broken)
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	422
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	423 @command(one2one):
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	424 """
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	425 if not client.encryption.isEncryptionRequested(mess_data, NS_OMEMO):
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	426 feedback = _(
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	427 "You need to have OMEMO encryption activated to reset the session")
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	428 self.text_cmds.feedBack(client, feedback, mess_data)
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	429 return False
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	430 to_jid = mess_data["to"].userhostJID()
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	431 session = client._xep_0384_session
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	432 devices = await session.getDevices(to_jid)
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	433
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	434 for device in devices['active']:
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	435 log.debug(f"deleting session for device {device}")
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	436 await session.deleteSession(to_jid, device=device)
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	437
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	438 log.debug("Sending an empty message to trigger key exchange")
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	439 await client.sendMessage(to_jid, {'': ''})
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	440
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	441 feedback = _("OMEMO session has been reset")
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	442 self.text_cmds.feedBack(client, feedback, mess_data)
806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	443 return False
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	444
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	445 @defer.inlineCallbacks
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	446 def trustUICb(self, xmlui_data, trust_data, expect_problems=None,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	447 profile=C.PROF_KEY_NONE):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	448 if C.bool(xmlui_data.get('cancelled', 'false')):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	449 defer.returnValue({})
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	450 client = self.host.getClient(profile)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	451 session = client._xep_0384_session
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	452 answer = xml_tools.XMLUIResult2DataFormResult(xmlui_data)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	453 for key, value in answer.items():
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	454 if key.startswith('trust_'):
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	455 trust_id = key[6:]
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	456 else:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	457 continue
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	458 data = trust_data[trust_id]
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	459 trust = C.bool(value)
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	460 yield session.setTrust(
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	461 data["jid"],
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	462 data["device"],
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	463 data["ik"],
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	464 trusted=trust,
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	465 )
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	466 if not trust and expect_problems is not None:
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	467 expect_problems.setdefault(data['jid'].userhost(), set()).add(
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	468 data['device']
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	469 )
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	470 defer.returnValue({})
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	471
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	472 @defer.inlineCallbacks
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	473 def getTrustUI(self, client, entity_jid=None, trust_data=None, submit_id=None):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	474 """Generate a XMLUI to manage trust
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	475
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	476 @param entity_jid(None, jid.JID): jid of entity to manage
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	477 None to use trust_data
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	478 @param trust_data(None, dict): devices data:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	479 None to use entity_jid
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	480 else a dict mapping from trust ids (unicode) to devices data,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	481 where a device data must have the following keys:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	482 - jid(jid.JID): bare jid of the device owner
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	483 - device(int): device id
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	484 - ik(bytes): identity key
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	485 and may have the following key:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	486 - trusted(bool): True if device is trusted
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	487 @param submit_id(None, unicode): submit_id to use
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	488 if None set UI callback to trustUICb
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	489 @return D(xmlui): trust management form
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	490 """
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	491 # we need entity_jid xor trust_data
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	492 assert entity_jid and not trust_data or not entity_jid and trust_data
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	493 if entity_jid and entity_jid.resource:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	494 raise ValueError("A bare jid is expected")
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	495
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	496 session = client._xep_0384_session
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	497
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	498 if trust_data is None:
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	499 cache = client._xep_0384_cache.setdefault(entity_jid, {})
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	500 trust_data = {}
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	501 if self._m is not None and self._m.isJoinedRoom(client, entity_jid):
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	502 trust_jids = self.getJIDsForRoom(client, entity_jid)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	503 else:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	504 trust_jids = [entity_jid]
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	505 for trust_jid in trust_jids:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	506 trust_session_data = yield session.getTrustForJID(trust_jid)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	507 bare_jid_s = trust_jid.userhost()
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	508 for device_id, trust_info in trust_session_data['active'].items():
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	509 if trust_info is None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	510 # device has never been (un)trusted, we have to retrieve its
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	511 # fingerprint (i.e. identity key or "ik") through public bundle
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	512 if device_id not in cache:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	513 bundles, missing = yield self.getBundles(client,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	514 trust_jid,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	515 [device_id])
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	516 if device_id not in bundles:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	517 log.warning(_(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	518 "Can't find bundle for device {device_id} of user "
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	519 "{bare_jid}, ignoring").format(device_id=device_id,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	520 bare_jid=bare_jid_s))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	521 continue
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	522 cache[device_id] = bundles[device_id]
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	523 # TODO: replace False below by None when undecided
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	524 # trusts are handled
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	525 trust_info = {
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	526 "key": cache[device_id].ik,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	527 "trusted": False
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	528 }
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	529
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	530 ik = trust_info["key"]
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	531 trust_id = str(hash((bare_jid_s, device_id, ik)))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	532 trust_data[trust_id] = {
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	533 "jid": trust_jid,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	534 "device": device_id,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	535 "ik": ik,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	536 "trusted": trust_info["trusted"],
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	537 }
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	538
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	539 if submit_id is None:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	540 submit_id = self.host.registerCallback(partial(self.trustUICb,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	541 trust_data=trust_data),
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	542 with_data=True,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	543 one_shot=True)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	544 xmlui = xml_tools.XMLUI(
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	545 panel_type = C.XMLUI_FORM,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	546 title = D_("OMEMO trust management"),
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	547 submit_id = submit_id
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	548 )
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	549 xmlui.addText(D_(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	550 "This is OMEMO trusting system. You'll see below the devices of your "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	551 "contacts, and a checkbox to trust them or not. A trusted device "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	552 "can read your messages in plain text, so be sure to only validate "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	553 "devices that you are sure are belonging to your contact. It's better "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	554 "to do this when you are next to your contact and her/his device, so "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	555 "you can check the \"fingerprint\" (the number next to the device) "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	556 "yourself. Do not validate a device if the fingerprint is wrong!"))
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	557
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	558 xmlui.changeContainer("label")
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	559 xmlui.addLabel(D_("This device ID"))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	560 xmlui.addText(str(client._xep_0384_device_id))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	561 xmlui.addLabel(D_("This device fingerprint"))
3084 ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	562 ik_hex = session.public_bundle.ik.hex().upper()
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	563 fp_human = ' '.join([ik_hex[i:i+8] for i in range(0, len(ik_hex), 8)])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	564 xmlui.addText(fp_human)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	565 xmlui.addEmpty()
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	566 xmlui.addEmpty()
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	567
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	568
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	569 for trust_id, data in trust_data.items():
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	570 xmlui.addLabel(D_("Contact"))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	571 xmlui.addJid(data['jid'])
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	572 xmlui.addLabel(D_("Device ID"))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	573 xmlui.addText(str(data['device']))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	574 xmlui.addLabel(D_("Fingerprint"))
3084 ffcdd93b61fa plugin XEP-0384: specific warning on missing omemo-backend-signal + fixed encoding following Python 3 port Goffi <goffi@goffi.org> parents: 3028 diff changeset	575 ik_hex = data['ik'].hex().upper()
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	576 fp_human = ' '.join([ik_hex[i:i+8] for i in range(0, len(ik_hex), 8)])
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	577 xmlui.addText(fp_human)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	578 xmlui.addLabel(D_("Trust this device?"))
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	579 xmlui.addBool("trust_{}".format(trust_id),
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	580 value=C.boolConst(data.get('trusted', False)))
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	581
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	582 xmlui.addEmpty()
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	583 xmlui.addEmpty()
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	584
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	585 defer.returnValue(xmlui)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	586
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	587 @defer.inlineCallbacks
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	588 def profileConnected(self, client):
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	589 if self._m is not None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	590 # we keep plain text message for MUC messages we send
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	591 # as we can't encrypt for our own device
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	592 client._xep_0384_muc_cache = {}
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	593 # and we keep them only for some time, in case something goes wrong
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	594 # with the MUC
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	595 client._xep_0384_muc_cache_timer = None
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	596
2925 03a09e16bf28 plugin XEP-0384: wait for client to be ready if session is missing when onNewDevices is called Goffi <goffi@goffi.org> parents: 2860 diff changeset	597 # FIXME: is _xep_0384_ready needed? can we use profileConnecting?
03a09e16bf28 plugin XEP-0384: wait for client to be ready if session is missing when onNewDevices is called Goffi <goffi@goffi.org> parents: 2860 diff changeset	598 # Workflow should be checked
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	599 client._xep_0384_ready = defer.Deferred()
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	600 # we first need to get devices ids (including our own)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	601 persistent_dict = persistent.LazyPersistentBinaryDict("XEP-0384", client.profile)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	602 # all known devices of profile
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	603 devices = yield self.getDevices(client)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	604 # and our own device id
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	605 device_id = yield persistent_dict.get(KEY_DEVICE_ID)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	606 if device_id is None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	607 log.info(_("We have no identity for this device yet, let's generate one"))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	608 # we have a new device, we create device_id
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	609 device_id = random.randint(1, 2**31-1)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	610 # we check that it's really unique
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	611 while device_id in devices:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	612 device_id = random.randint(1, 2**31-1)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	613 # and we save it
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	614 persistent_dict[KEY_DEVICE_ID] = device_id
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	615
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	616 log.debug(f"our OMEMO device id is {device_id}")
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	617
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	618 if device_id not in devices:
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	619 log.debug(f"our device id ({device_id}) is not in the list, adding it")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	620 devices.add(device_id)
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	621 yield defer.ensureDeferred(self.setDevices(client, devices))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	622
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	623 all_jids = yield persistent_dict.get(KEY_ALL_JIDS, set())
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	624
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	625 omemo_storage = OmemoStorage(client, device_id, all_jids, persistent_dict)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	626 omemo_session = yield OmemoSession.create(client, omemo_storage, device_id)
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	627 client._xep_0384_cache = {}
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	628 client._xep_0384_session = omemo_session
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	629 client._xep_0384_device_id = device_id
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	630 yield omemo_session.newDeviceList(client.jid, devices)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	631 if omemo_session.republish_bundle:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	632 log.info(_("Saving public bundle for this device ({device_id})").format(
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	633 device_id=device_id))
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	634 yield defer.ensureDeferred(
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	635 self.setBundle(client, omemo_session.public_bundle, device_id)
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	636 )
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	637 client._xep_0384_ready.callback(None)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	638 del client._xep_0384_ready
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	639
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	640 ## XMPP PEP nodes manipulation
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	641
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	642 # devices
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	643
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	644 def parseDevices(self, items):
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	645 """Parse devices found in items
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	646
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	647 @param items(iterable[domish.Element]): items as retrieved by getItems
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	648 @return set[int]: parsed devices
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	649 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	650 devices = set()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	651 if len(items) > 1:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	652 log.warning(_("OMEMO devices list is stored in more that one items, "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	653 "this is not expected"))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	654 if items:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	655 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	656 list_elt = next(items[0].elements(NS_OMEMO, 'list'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	657 except StopIteration:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	658 log.warning(_("no list element found in OMEMO devices list"))
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	659 return devices
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	660 for device_elt in list_elt.elements(NS_OMEMO, 'device'):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	661 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	662 device_id = int(device_elt['id'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	663 except KeyError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	664 log.warning(_('device element is missing "id" attribute: {elt}')
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	665 .format(elt=device_elt.toXml()))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	666 except ValueError:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	667 log.warning(_('invalid device id: {device_id}').format(
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	668 device_id=device_elt['id']))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	669 else:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	670 devices.add(device_id)
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	671 return devices
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	672
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	673 @defer.inlineCallbacks
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	674 def getDevices(self, client, entity_jid=None):
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	675 """Retrieve list of registered OMEMO devices
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	676
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	677 @param entity_jid(jid.JID, None): get devices from this entity
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	678 None to get our own devices
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	679 @return (set(int)): list of devices
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	680 """
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	681 if entity_jid is not None:
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	682 assert not entity_jid.resource
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	683 try:
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	684 items, metadata = yield self._p.getItems(client, entity_jid, NS_OMEMO_DEVICES)
3142 cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	685 except exceptions.NotFound:
cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	686 log.info(_("there is no node to handle OMEMO devices"))
cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	687 defer.returnValue(set())
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	688
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	689 devices = self.parseDevices(items)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	690 defer.returnValue(devices)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	691
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	692 async def setDevices(self, client, devices):
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	693 log.debug(f"setting devices with {', '.join(str(d) for d in devices)}")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	694 list_elt = domish.Element((NS_OMEMO, 'list'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	695 for device in devices:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	696 device_elt = list_elt.addElement('device')
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	697 device_elt['id'] = str(device)
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	698 try:
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	699 await self._p.sendItem(
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	700 client, None, NS_OMEMO_DEVICES, list_elt,
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	701 item_id=self._p.ID_SINGLETON,
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	702 extra={
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	703 self._p.EXTRA_PUBLISH_OPTIONS: {self._p.OPT_MAX_ITEMS: 1},
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	704 self._p.EXTRA_ON_PRECOND_NOT_MET: "publish_without_options",
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	705 }
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	706 )
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	707 except Exception as e:
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	708 log.warning(_("Can't set devices: {reason}").format(reason=e))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	709
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	710 # bundles
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	711
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	712 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	713 def getBundles(self, client, entity_jid, devices_ids):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	714 """Retrieve public bundles of an entity devices
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	715
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	716 @param entity_jid(jid.JID): bare jid of entity
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	717 @param devices_id(iterable[int]): ids of the devices bundles to retrieve
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	718 @return (tuple(dict[int, ExtendedPublicBundle], list(int))):
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	719 - bundles collection:
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	720 * key is device_id
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	721 * value is parsed bundle
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	722 - set of bundles not found
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	723 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	724 assert not entity_jid.resource
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	725 bundles = {}
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	726 missing = set()
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	727 for device_id in devices_ids:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	728 node = NS_OMEMO_BUNDLE.format(device_id=device_id)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	729 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	730 items, metadata = yield self._p.getItems(client, entity_jid, node)
3142 cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	731 except exceptions.NotFound:
cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	732 log.warning(_("Bundle missing for device {device_id}")
cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	733 .format(device_id=device_id))
cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	734 missing.add(device_id)
cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	735 continue
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	736 except jabber_error.StanzaError as e:
3142 cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	737 log.warning(_("Can't get bundle for device {device_id}: {reason}")
cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	738 .format(device_id=device_id, reason=e))
cd90c24b2836 plugin XEP-0384: fixed exception catching on getItems Goffi <goffi@goffi.org> parents: 3136 diff changeset	739 continue
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	740 if not items:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	741 log.warning(_("no item found in node {node}, can't get public bundle "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	742 "for device {device_id}").format(node=node,
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	743 device_id=device_id))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	744 continue
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	745 if len(items) > 1:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	746 log.warning(_("more than one item found in {node}, "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	747 "this is not expected").format(node=node))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	748 item = items[0]
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	749 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	750 bundle_elt = next(item.elements(NS_OMEMO, 'bundle'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	751 signedPreKeyPublic_elt = next(bundle_elt.elements(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	752 NS_OMEMO, 'signedPreKeyPublic'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	753 signedPreKeySignature_elt = next(bundle_elt.elements(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	754 NS_OMEMO, 'signedPreKeySignature'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	755 identityKey_elt = next(bundle_elt.elements(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	756 NS_OMEMO, 'identityKey'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	757 prekeys_elt = next(bundle_elt.elements(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	758 NS_OMEMO, 'prekeys'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	759 except StopIteration:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	760 log.warning(_("invalid bundle for device {device_id}, ignoring").format(
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	761 device_id=device_id))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	762 continue
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	763
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	764 try:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	765 spkPublic = base64.b64decode(str(signedPreKeyPublic_elt))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	766 spkSignature = base64.b64decode(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	767 str(signedPreKeySignature_elt))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	768
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	769 ik = base64.b64decode(str(identityKey_elt))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	770 spk = {
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	771 "key": spkPublic,
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	772 "id": int(signedPreKeyPublic_elt['signedPreKeyId'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	773 }
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	774 otpks = []
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	775 for preKeyPublic_elt in prekeys_elt.elements(NS_OMEMO, 'preKeyPublic'):
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	776 preKeyPublic = base64.b64decode(str(preKeyPublic_elt))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	777 otpk = {
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	778 "key": preKeyPublic,
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	779 "id": int(preKeyPublic_elt['preKeyId'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	780 }
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	781 otpks.append(otpk)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	782
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	783 except Exception as e:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	784 log.warning(_("error while decoding key for device {device_id}: {msg}")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	785 .format(device_id=device_id, msg=e))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	786 continue
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	787
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	788 bundles[device_id] = ExtendedPublicBundle.parse(omemo_backend, ik, spk,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	789 spkSignature, otpks)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	790
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	791 defer.returnValue((bundles, missing))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	792
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	793 async def setBundle(self, client, bundle, device_id):
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	794 """Set public bundle for this device.
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	795
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	796 @param bundle(ExtendedPublicBundle): bundle to publish
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	797 """
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	798 log.debug(_("updating bundle for {device_id}").format(device_id=device_id))
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	799 bundle = bundle.serialize(omemo_backend)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	800 bundle_elt = domish.Element((NS_OMEMO, 'bundle'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	801 signedPreKeyPublic_elt = bundle_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	802 "signedPreKeyPublic",
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	803 content=b64enc(bundle["spk"]['key']))
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	804 signedPreKeyPublic_elt['signedPreKeyId'] = str(bundle["spk"]['id'])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	805
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	806 bundle_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	807 "signedPreKeySignature",
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	808 content=b64enc(bundle["spk_signature"]))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	809
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	810 bundle_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	811 "identityKey",
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	812 content=b64enc(bundle["ik"]))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	813
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	814 prekeys_elt = bundle_elt.addElement('prekeys')
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	815 for otpk in bundle["otpks"]:
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	816 preKeyPublic_elt = prekeys_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	817 'preKeyPublic',
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	818 content=b64enc(otpk["key"]))
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	819 preKeyPublic_elt['preKeyId'] = str(otpk['id'])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	820
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	821 node = NS_OMEMO_BUNDLE.format(device_id=device_id)
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	822 try:
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	823 await self._p.sendItem(
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	824 client, None, node, bundle_elt, item_id=self._p.ID_SINGLETON,
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	825 extra={
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	826 self._p.EXTRA_PUBLISH_OPTIONS: {self._p.OPT_MAX_ITEMS: 1},
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	827 self._p.EXTRA_ON_PRECOND_NOT_MET: "publish_without_options",
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	828 }
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	829 )
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	830 except Exception as e:
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	831 log.warning(_("Can't set bundle: {reason}").format(reason=e))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	832
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	833 ## PEP node events callbacks
0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	834
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	835 async def onNewDevices(self, itemsEvent, profile):
3218 806a7936a591 plugin XEP-0384: added "/omemo_reset" text command: Goffi <goffi@goffi.org> parents: 3214 diff changeset	836 log.debug("devices list has been updated")
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	837 client = self.host.getClient(profile)
2925 03a09e16bf28 plugin XEP-0384: wait for client to be ready if session is missing when onNewDevices is called Goffi <goffi@goffi.org> parents: 2860 diff changeset	838 try:
03a09e16bf28 plugin XEP-0384: wait for client to be ready if session is missing when onNewDevices is called Goffi <goffi@goffi.org> parents: 2860 diff changeset	839 omemo_session = client._xep_0384_session
03a09e16bf28 plugin XEP-0384: wait for client to be ready if session is missing when onNewDevices is called Goffi <goffi@goffi.org> parents: 2860 diff changeset	840 except AttributeError:
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	841 await client._xep_0384_ready
2925 03a09e16bf28 plugin XEP-0384: wait for client to be ready if session is missing when onNewDevices is called Goffi <goffi@goffi.org> parents: 2860 diff changeset	842 omemo_session = client._xep_0384_session
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	843 entity = itemsEvent.sender
2817 0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	844
2662 0bef44f8e8ca plugin XEP-0384: PEP handling + mark as encrypted: Goffi <goffi@goffi.org> parents: 2654 diff changeset	845 devices = self.parseDevices(itemsEvent.items)
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	846 await omemo_session.newDeviceList(entity, devices)
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	847
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	848 if entity == client.jid.userhostJID():
e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	849 own_device = client._xep_0384_device_id
e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	850 if own_device not in devices:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	851 log.warning(_("Our own device is missing from devices list, fixing it"))
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	852 devices.add(own_device)
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	853 await self.setDevices(client, devices)
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	854
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	855 ## triggers
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	856
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	857 @defer.inlineCallbacks
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	858 def handleProblems(self, client, feedback_jid, bundles, expect_problems, problems):
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	859 """Try to solve problems found by EncryptMessage
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	860
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	861 @param feedback_jid(jid.JID): bare jid where the feedback message must be sent
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	862 @param bundles(dict): bundles data as used in EncryptMessage
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	863 already filled with known bundles, missing bundles
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	864 need to be added to it
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	865 This dict is updated
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	866 @param problems(list): exceptions raised by EncryptMessage
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	867 @param expect_problems(dict): known problems to expect, used in encryptMessage
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	868 This dict will list devices where problems can be ignored
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	869 (those devices won't receive the encrypted data)
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	870 This dict is updated
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	871 """
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	872 # FIXME: not all problems are handled yet
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	873 undecided = {}
2817 0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	874 missing_bundles = {}
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	875 found_bundles = None
2817 0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	876 cache = client._xep_0384_cache
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	877 for problem in problems:
2857 88f10630d5ea plugin XEP-0384: removed version restriction, it is now compatible with (and require) last version of python-omemo (0.10.4) Goffi <goffi@goffi.org> parents: 2823 diff changeset	878 if isinstance(problem, omemo_excpt.TrustException):
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	879 if problem.problem == 'undecided':
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	880 undecided[str(hash(problem))] = problem
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	881 elif problem.problem == 'untrusted':
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	882 expect_problems.setdefault(problem.bare_jid, set()).add(
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	883 problem.device)
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	884 log.info(_(
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	885 "discarding untrusted device {device_id} with key {device_key} "
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	886 "for {entity}").format(
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	887 device_id=problem.device,
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	888 device_key=problem.ik.hex().upper(),
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	889 entity=problem.bare_jid,
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	890 )
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	891 )
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	892 else:
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	893 log.error(
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	894 f"Unexpected trust problem: {problem.problem!r} for device "
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	895 f"{problem.device} for {problem.bare_jid}, ignoring device")
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	896 expect_problems.setdefault(problem.bare_jid, set()).add(
c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	897 problem.device)
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	898 elif isinstance(problem, omemo_excpt.MissingBundleException):
2817 0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	899 pb_entity = jid.JID(problem.bare_jid)
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	900 entity_cache = cache.setdefault(pb_entity, {})
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	901 entity_bundles = bundles.setdefault(pb_entity, {})
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	902 if problem.device in entity_cache:
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	903 entity_bundles[problem.device] = entity_cache[problem.device]
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	904 else:
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	905 found_bundles, missing = yield self.getBundles(
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	906 client, pb_entity, [problem.device])
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	907 entity_cache.update(bundles)
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	908 entity_bundles.update(found_bundles)
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	909 if problem.device in missing:
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	910 missing_bundles.setdefault(pb_entity, set()).add(
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	911 problem.device)
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	912 expect_problems.setdefault(problem.bare_jid, set()).add(
0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	913 problem.device)
2860 851c47cc4ae7 plugin XEP-0384: handle NoEligibleDevicesException Goffi <goffi@goffi.org> parents: 2859 diff changeset	914 elif isinstance(problem, omemo_excpt.NoEligibleDevicesException):
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	915 if undecided or found_bundles:
2860 851c47cc4ae7 plugin XEP-0384: handle NoEligibleDevicesException Goffi <goffi@goffi.org> parents: 2859 diff changeset	916 # we may have new devices after this run, so let's continue for now
851c47cc4ae7 plugin XEP-0384: handle NoEligibleDevicesException Goffi <goffi@goffi.org> parents: 2859 diff changeset	917 continue
851c47cc4ae7 plugin XEP-0384: handle NoEligibleDevicesException Goffi <goffi@goffi.org> parents: 2859 diff changeset	918 else:
851c47cc4ae7 plugin XEP-0384: handle NoEligibleDevicesException Goffi <goffi@goffi.org> parents: 2859 diff changeset	919 raise problem
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	920 else:
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	921 raise problem
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	922
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	923 for peer_jid, devices in missing_bundles.items():
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	924 devices_s = [str(d) for d in devices]
2817 0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	925 log.warning(
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	926 _("Can't retrieve bundle for device(s) {devices} of entity {peer}, "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	927 "the message will not be readable on this/those device(s)").format(
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	928 devices=", ".join(devices_s), peer=peer_jid.full()))
2817 0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	929 client.feedback(
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	930 feedback_jid,
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	931 D_("You're destinee {peer} has missing encryption data on some of "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	932 "his/her device(s) (bundle on device {devices}), the message won't "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	933 "be readable on this/those device.").format(
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	934 peer=peer_jid.full(), devices=", ".join(devices_s)))
2817 0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	935
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	936 if undecided:
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	937 trust_data = {}
3085 c048fc192739 plugin XEP-0384: update to latest python-omemo + better trust management: Goffi <goffi@goffi.org> parents: 3084 diff changeset	938 for trust_id, data in undecided.items():
2744 e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	939 trust_data[trust_id] = {
e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	940 'jid': jid.JID(data.bare_jid),
e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	941 'device': data.device,
e6716d90c2fe plugin XEP-0384: various bug fixes: Goffi <goffi@goffi.org> parents: 2738 diff changeset	942 'ik': data.ik}
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	943
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	944 user_msg = D_("Not all destination devices are trusted, we can't encrypt "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	945 "message in such a situation. Please indicate if you trust "
ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	946 "those devices or not in the trust manager before we can "
2757 1f612547fb2e plugin XEP-0384: give feedback to user when trust must be handled before sending a message Goffi <goffi@goffi.org> parents: 2754 diff changeset	947 "send this message")
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	948 client.feedback(feedback_jid, user_msg)
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	949 xmlui = yield self.getTrustUI(client, trust_data=trust_data, submit_id="")
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	950
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	951 answer = yield xml_tools.deferXMLUI(
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	952 self.host,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	953 xmlui,
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	954 action_extra={
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	955 "meta_encryption_trust": NS_OMEMO,
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	956 },
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	957 profile=client.profile)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	958 yield self.trustUICb(answer, trust_data, expect_problems, client.profile)
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	959
eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	960 @defer.inlineCallbacks
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	961 def encryptMessage(self, client, entity_bare_jids, message, feedback_jid=None):
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	962 if feedback_jid is None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	963 if len(entity_bare_jids) != 1:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	964 log.error(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	965 "feedback_jid must be provided when message is encrypted for more "
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	966 "than one entities")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	967 feedback_jid = entity_bare_jids[0]
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	968 omemo_session = client._xep_0384_session
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	969 expect_problems = {}
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	970 bundles = {}
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	971 loop_idx = 0
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	972 try:
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	973 while True:
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	974 if loop_idx > 10:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	975 msg = _("Too many iterations in encryption loop")
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	976 log.error(msg)
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	977 raise exceptions.InternalError(msg)
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	978 # encryptMessage may fail, in case of e.g. trust issue or missing bundle
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	979 try:
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	980 encrypted = yield omemo_session.encryptMessage(
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	981 entity_bare_jids,
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	982 message,
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	983 bundles,
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	984 expect_problems = expect_problems)
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	985 except omemo_excpt.EncryptionProblemsException as e:
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	986 # we know the problem to solve, we can try to fix them
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	987 yield self.handleProblems(
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	988 client,
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	989 feedback_jid=feedback_jid,
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	990 bundles=bundles,
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	991 expect_problems=expect_problems,
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	992 problems=e.problems)
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	993 loop_idx += 1
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	994 else:
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	995 break
31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	996 except Exception as e:
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	997 msg = _("Can't encrypt message for {entities}: {reason}".format(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	998 entities=', '.join(e.full() for e in entity_bare_jids), reason=e))
2859 4e875d9eea48 plugin XEP-0384: give feedback to client when encryption failed Goffi <goffi@goffi.org> parents: 2858 diff changeset	999 log.warning(msg)
4e875d9eea48 plugin XEP-0384: give feedback to client when encryption failed Goffi <goffi@goffi.org> parents: 2858 diff changeset	1000 extra = {C.MESS_EXTRA_INFO: C.EXTRA_INFO_ENCR_ERR}
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1001 client.feedback(feedback_jid, msg, extra)
2858 31a5038cdf79 plugin XEP-0384: small refactoring to encrypt messages + bugfix: Goffi <goffi@goffi.org> parents: 2857 diff changeset	1002 raise e
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1003
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1004 defer.returnValue(encrypted)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1005
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1006 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1007 def _messageReceivedTrigger(self, client, message_elt, post_treat):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1008 try:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	1009 encrypted_elt = next(message_elt.elements(NS_OMEMO, "encrypted"))
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1010 except StopIteration:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1011 # no OMEMO message here
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1012 defer.returnValue(True)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1013
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1014 # we have an encrypted message let's decrypt it
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1015
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	1016 from_jid = jid.JID(message_elt['from'])
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1017
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1018 if message_elt.getAttribute("type") == C.MESS_TYPE_GROUPCHAT:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1019 # with group chat, we must get the real jid for decryption
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1020 # and use the room as feedback_jid
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1021
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1022 if self._m is None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1023 # plugin XEP-0045 (MUC) is not available
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1024 defer.returnValue(True)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1025
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1026 room_jid = from_jid.userhostJID()
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1027 feedback_jid = room_jid
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1028 if self._sid is not None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1029 mess_id = self._sid.getOriginId(message_elt)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1030 else:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1031 mess_id = None
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1032
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1033 if mess_id is None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1034 mess_id = message_elt.getAttribute('id')
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1035 cache_key = (room_jid, mess_id)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1036
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1037 try:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1038 room = self._m.getRoom(client, room_jid)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1039 except exceptions.NotFound:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1040 log.warning(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1041 f"Received an OMEMO encrypted msg from a room {room_jid} which has "
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1042 f"not been joined, ignoring")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1043 defer.returnValue(True)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1044
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1045 user = room.getUser(from_jid.resource)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1046 if user is None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1047 log.warning(f"Can't find user {user} in room {room_jid}, ignoring")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1048 defer.returnValue(True)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1049 if not user.entity:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1050 log.warning(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1051 f"Real entity of user {user} in room {room_jid} can't be established,"
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1052 f" OMEMO encrypted message can't be decrypted")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1053 defer.returnValue(True)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1054
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1055 # now we have real jid of the entity, we use it instead of from_jid
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1056 from_jid = user.entity.userhostJID()
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1057
2817 0ab62dd3cf05 plugin XEP-0384: better bundle handling + misc improvments Goffi <goffi@goffi.org> parents: 2771 diff changeset	1058 else:
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1059 # we have a one2one message, we can user "from" and "to" normally
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1060
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1061 if from_jid.userhostJID() == client.jid.userhostJID():
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1062 feedback_jid = jid.JID(message_elt['to'])
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1063 else:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1064 feedback_jid = from_jid
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1065
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1066
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1067 if (message_elt.getAttribute("type") == C.MESS_TYPE_GROUPCHAT
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1068 and mess_id is not None
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1069 and cache_key in client._xep_0384_muc_cache):
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1070 plaintext = client._xep_0384_muc_cache.pop(cache_key)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1071 if not client._xep_0384_muc_cache:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1072 client._xep_0384_muc_cache_timer.cancel()
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1073 client._xep_0384_muc_cache_timer = None
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1074 else:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1075 try:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1076 omemo_session = client._xep_0384_session
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1077 except AttributeError:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1078 # on startup, message can ve received before session actually exists
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1079 # so we need to synchronise here
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1080 yield client._xep_0384_ready
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1081 omemo_session = client._xep_0384_session
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1082
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1083 device_id = client._xep_0384_device_id
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1084 try:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1085 header_elt = next(encrypted_elt.elements(NS_OMEMO, 'header'))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1086 iv_elt = next(header_elt.elements(NS_OMEMO, 'iv'))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1087 except StopIteration:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1088 log.warning(_("Invalid OMEMO encrypted stanza, ignoring: {xml}")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1089 .format(xml=message_elt.toXml()))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1090 defer.returnValue(False)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1091 try:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1092 s_device_id = header_elt['sid']
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1093 except KeyError:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1094 log.warning(_("Invalid OMEMO encrypted stanza, missing sender device ID, "
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1095 "ignoring: {xml}")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1096 .format(xml=message_elt.toXml()))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1097 defer.returnValue(False)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1098 try:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1099 key_elt = next((e for e in header_elt.elements(NS_OMEMO, 'key')
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1100 if int(e['rid']) == device_id))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1101 except StopIteration:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1102 log.warning(_("This OMEMO encrypted stanza has not been encrypted "
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1103 "for our device (device_id: {device_id}, fingerprint: "
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1104 "{fingerprint}): {xml}").format(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1105 device_id=device_id,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1106 fingerprint=omemo_session.public_bundle.ik.hex().upper(),
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1107 xml=encrypted_elt.toXml()))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1108 user_msg = (D_("An OMEMO message from {sender} has not been encrypted for "
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1109 "our device, we can't decrypt it").format(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1110 sender=from_jid.full()))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1111 extra = {C.MESS_EXTRA_INFO: C.EXTRA_INFO_DECR_ERR}
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1112 client.feedback(feedback_jid, user_msg, extra)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1113 defer.returnValue(False)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1114 except ValueError as e:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1115 log.warning(_("Invalid recipient ID: {msg}".format(msg=e)))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1116 defer.returnValue(False)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1117 is_pre_key = C.bool(key_elt.getAttribute('prekey', 'false'))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1118 payload_elt = next(encrypted_elt.elements(NS_OMEMO, 'payload'), None)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1119 additional_information = {
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1120 "from_storage": bool(message_elt.delay)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1121 }
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1122
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1123 kwargs = {
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1124 "bare_jid": from_jid.userhostJID(),
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1125 "device": s_device_id,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1126 "iv": base64.b64decode(bytes(iv_elt)),
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1127 "message": base64.b64decode(bytes(key_elt)),
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1128 "is_pre_key_message": is_pre_key,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1129 "ciphertext": base64.b64decode(bytes(payload_elt))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1130 if payload_elt is not None else None,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1131 "additional_information": additional_information,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1132 }
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1133
2738 eb58f26ed236 plugin XEP-0384: update to last python-omemo + trust management: Goffi <goffi@goffi.org> parents: 2662 diff changeset	1134 try:
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1135 try:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1136 plaintext = yield omemo_session.decryptMessage(**kwargs)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1137 except omemo_excpt.TrustException:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1138 post_treat.addCallback(client.encryption.markAsUntrusted)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1139 kwargs['allow_untrusted'] = True
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1140 plaintext = yield omemo_session.decryptMessage(**kwargs)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1141 else:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1142 post_treat.addCallback(client.encryption.markAsTrusted)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1143 plaintext = plaintext.decode()
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1144 except Exception as e:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1145 log.warning(_("Can't decrypt message: {reason}\n{xml}").format(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1146 reason=e, xml=message_elt.toXml()))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1147 user_msg = (D_(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1148 "An OMEMO message from {sender} can't be decrypted: {reason}")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1149 .format(sender=from_jid.full(), reason=e))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1150 extra = {C.MESS_EXTRA_INFO: C.EXTRA_INFO_DECR_ERR}
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1151 client.feedback(feedback_jid, user_msg, extra)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1152 defer.returnValue(False)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1153 finally:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1154 if omemo_session.republish_bundle:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1155 # we don't wait for the Deferred (i.e. no yield) on purpose
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1156 # there is no need to block the whole message workflow while
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1157 # updating the bundle
3214 8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	1158 defer.ensureDeferred(
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	1159 self.setBundle(client, omemo_session.public_bundle, device_id)
8d92d4d829fb plugin XEP-0384: use "max_items=1" for devices and bundles nodes: Goffi <goffi@goffi.org> parents: 3172 diff changeset	1160 )
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1161
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1162 message_elt.children.remove(encrypted_elt)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1163 if plaintext:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	1164 message_elt.addElement("body", content=plaintext)
3231 e756e0eb1be4 core (memory/encryption): automatic start encryption if peer send encrypted message: Goffi <goffi@goffi.org> parents: 3218 diff changeset	1165 post_treat.addCallback(client.encryption.markAsEncrypted, namespace=NS_OMEMO)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1166 defer.returnValue(True)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1167
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1168 def getJIDsForRoom(self, client, room_jid):
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1169 if self._m is None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1170 exceptions.InternalError("XEP-0045 plugin missing, can't encrypt for group chat")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1171 room = self._m.getRoom(client, room_jid)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1172 return [u.entity.userhostJID() for u in room.roster.values()]
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1173
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1174 def _expireMUCCache(self, client):
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1175 client._xep_0384_muc_cache_timer = None
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1176 for (room_jid, uid), msg in client._xep_0384_muc_cache.items():
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1177 client.feedback(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1178 room_jid,
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1179 D_("Our message with UID {uid} has not been received in time, it has "
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1180 "probably been lost. The message was: {msg!r}").format(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1181 uid=uid, msg=str(msg)))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1182
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1183 client._xep_0384_muc_cache.clear()
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1184 log.warning("Cache for OMEMO MUC has expired")
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1185
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1186 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1187 def _sendMessageDataTrigger(self, client, mess_data):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1188 encryption = mess_data.get(C.MESS_KEY_ENCRYPTION)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1189 if encryption is None or encryption['plugin'].namespace != NS_OMEMO:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1190 return
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1191 message_elt = mess_data["xml"]
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1192 if mess_data['type'] == C.MESS_TYPE_GROUPCHAT:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1193 feedback_jid = room_jid = mess_data['to']
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1194 to_jids = self.getJIDsForRoom(client, room_jid)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1195 else:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1196 feedback_jid = to_jid = mess_data["to"].userhostJID()
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1197 to_jids = [to_jid]
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	1198 log.debug("encrypting message")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1199 body = None
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1200 for child in list(message_elt.children):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1201 if child.name == "body":
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1202 # we remove all unencrypted body,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1203 # and will only encrypt the first one
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1204 if body is None:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1205 body = child
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1206 message_elt.children.remove(child)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1207 elif child.name == "html":
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1208 # we don't want any XHTML-IM element
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1209 message_elt.children.remove(child)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1210
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1211 if body is None:
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	1212 log.warning("No message found")
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1213 return
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1214
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1215 body = str(body)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1216
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1217 if mess_data['type'] == C.MESS_TYPE_GROUPCHAT:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1218 key = (room_jid, mess_data['uid'])
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1219 # XXX: we can't encrypt message for our own device for security reason
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1220 # so we keep the plain text version in cache until we receive the
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1221 # message. We don't send it directly to bridge to keep a workflow
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1222 # similar to plain text MUC, so when we see it in frontend we know
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1223 # that it has been sent correctly.
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1224 client._xep_0384_muc_cache[key] = body
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1225 timer = client._xep_0384_muc_cache_timer
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1226 if timer is None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1227 client._xep_0384_muc_cache_timer = reactor.callLater(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1228 MUC_CACHE_TTL, self._expireMUCCache, client)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1229 else:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1230 timer.reset(MUC_CACHE_TTL)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1231 # we use origin-id when possible, to identifiy the message in a stable way
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1232 if self._sid is not None:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1233 self._sid.addOriginId(message_elt, mess_data['uid'])
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1234
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1235 encryption_data = yield self.encryptMessage(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1236 client, to_jids, body, feedback_jid=feedback_jid)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1237
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1238 encrypted_elt = message_elt.addElement((NS_OMEMO, 'encrypted'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1239 header_elt = encrypted_elt.addElement('header')
3028 ab2696e34d29 Python 3 port: Goffi <goffi@goffi.org> parents: 2925 diff changeset	1240 header_elt['sid'] = str(encryption_data['sid'])
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1241
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1242 for to_jid in to_jids:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1243 bare_jid_s = to_jid.userhost()
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1244
3104 118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1245 for rid, data in encryption_data['keys'][bare_jid_s].items():
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1246 key_elt = header_elt.addElement(
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1247 'key',
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1248 content=b64enc(data['data']))
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1249 key_elt['rid'] = str(rid)
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1250 if data['pre_key']:
118d91c932a7 plugin XEP-0384: OMEMO for MUC implementation: Goffi <goffi@goffi.org> parents: 3098 diff changeset	1251 key_elt['prekey'] = 'true'
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1252
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1253 header_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1254 'iv',
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1255 content=b64enc(encryption_data['iv']))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1256 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1257 encrypted_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1258 'payload',
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1259 content=b64enc(encryption_data['payload']))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1260 except KeyError:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1261 pass

Mercurial > libervia-backend

annotate sat/plugins/plugin_xep_0384.py @ 3231:e756e0eb1be4