libervia-backend: sat/plugins/plugin_xep

annotate sat/plugins/plugin_xep_0384.py @ 2654:e7bfbded652a

plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: - omemo module logs are now integrated in SàT - OmemoStorage adapated to changes - now really delete OTPK each time, behaviour may be modified in future - fixed bad arguments used during decryption - adapted to other changes OMEMO is now working \o/ It still needs some adjustements though: bundles/devices are for the moment requested on each message encryption, and fingerprints management is not implemented yet.

author	Goffi <goffi@goffi.org>
date	Sat, 11 Aug 2018 18:24:55 +0200
parents	0f76813afc57
children	0bef44f8e8ca

rev	line source
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	1 #!/usr/bin/env python2
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	2 # -- coding: utf-8 --
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	3
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	4 # SAT plugin for OMEMO encryption
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	5 # Copyright (C) 2009-2018 Jérôme Poisson (goffi@goffi.org)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	6
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	7 # 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	8 # 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	9 # 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	10 # (at your option) any later version.
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	11
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	12 # 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	13 # 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	14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	15 # GNU Affero General Public License for more details.
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	16
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	17 # 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	18 # 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	19
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	20 from sat.core.i18n import _
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	21 from sat.core.constants import Const as C
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	22 from sat.core.log import getLogger
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	23 from sat.core import exceptions
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	24 from twisted.internet import defer
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	25 from twisted.words.xish import domish
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	26 from twisted.words.protocols.jabber import jid
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	27 from twisted.words.protocols.jabber import error
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	28 from sat.memory import persistent
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	29 from functools import partial
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
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	35 from omemo.extendedpublicbundle import ExtendedPublicBundle
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	36 from omemo import wireformat
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	37 except ImportError:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	38 raise exceptions.MissingModule(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	39 u'Missing module omemo, please download/install it. You can use '
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	40 u'"pip install omemo"'
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	41 )
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	42
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	43 log = getLogger(__name__)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	44
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	45 PLUGIN_INFO = {
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	46 C.PI_NAME: u"OMEMO",
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	47 C.PI_IMPORT_NAME: u"OMEMO",
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	48 C.PI_TYPE: u"SEC",
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	49 C.PI_PROTOCOLS: [u"XEP-0384"],
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	50 C.PI_DEPENDENCIES: [u"XEP-0280", u"XEP-0334", u"XEP-0060"],
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	51 C.PI_MAIN: u"OMEMO",
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	52 C.PI_HANDLER: u"no",
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	53 C.PI_DESCRIPTION: _(u"""Implementation of OMEMO"""),
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	54 }
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	55
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	56 NS_OMEMO = "eu.siacs.conversations.axolotl"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	57 NS_OMEMO_DEVICES = NS_OMEMO + ".devicelist"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	58 NS_OMEMO_BUNDLE = NS_OMEMO + ".bundles:{device_id}"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	59 KEY_STATE = "STATE"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	60 KEY_DEVICE_ID = "DEVICE_ID"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	61 KEY_SESSION = "SESSION"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	62 KEY_ACTIVE_DEVICES = "DEVICES"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	63 KEY_INACTIVE_DEVICES = "DEVICES"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	64
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	65
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	66 # we want to manage log emitted by omemo module ourseves
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	67
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	68 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	69
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	70 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	71 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	72
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	73 @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	74 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	75 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	76 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	77 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	78
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	79
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	80 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	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
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	83 def b64enc(data):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	84 return base64.b64encode(bytes(bytearray(data))).decode("ASCII")
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	85
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	86
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	87 class OmemoStorage(omemo.Storage):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	88
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	89 def __init__(self, client, device_id, persistent_dict):
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	90 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	91 @param persistent_dict(persistent.LazyPersistentBinaryDict): object which will
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	92 store data in SàT database
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	93 """
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	94 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	95 self.device_id = device_id
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	96 self.data = persistent_dict
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	97
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	98 @property
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	99 def is_async(self):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	100 return True
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	101
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	102 def setCb(self, deferred, callback):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	103 """Associate Deferred and callback
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	104
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	105 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	106 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	107 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	108 Deferred is called
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	109 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	110 deferred.addCallback(partial(callback, True))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	111 deferred.addErrback(partial(callback, False))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	112
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	113 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	114 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	115 '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	116
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	117 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	118 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	119 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	120 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	121
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	122 def loadState(self, callback):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	123 d = self.data.get(KEY_STATE)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	124 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	125
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	126 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	127 d = self.data.force(KEY_STATE, state)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	128 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	129
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	130 def loadSession(self, callback, bare_jid, 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	131 key = u'\n'.join([KEY_SESSION, bare_jid, unicode(device_id)])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	132 d = self.data.get(key)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	133 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	134
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	135 def storeSession(self, callback, bare_jid, device_id, session):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	136 key = u'\n'.join([KEY_SESSION, bare_jid, unicode(device_id)])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	137 d = self.data.force(key, session)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	138 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	139
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	140 def loadActiveDevices(self, callback, bare_jid):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	141 key = u'\n'.join([KEY_ACTIVE_DEVICES, bare_jid])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	142 d = self.data.get(key, {})
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	143 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	144
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	145 def loadInactiveDevices(self, callback, bare_jid):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	146 key = u'\n'.join([KEY_INACTIVE_DEVICES, bare_jid])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	147 d = self.data.get(key, {})
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	148 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	149
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	150 def storeActiveDevices(self, callback, bare_jid, devices):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	151 key = u'\n'.join([KEY_ACTIVE_DEVICES, bare_jid])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	152 d = self.data.force(key, devices)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	153 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	154
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	155 def storeInactiveDevices(self, callback, bare_jid, devices):
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 key = u'\n'.join([KEY_INACTIVE_DEVICES, bare_jid])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	157 d = self.data.force(key, devices)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	158 self.setCb(d, callback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	159
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	160 def isTrusted(self, callback, bare_jid, device):
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	161 trusted = True
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	162 callback(True, trusted)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	163
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	164
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	165 class SatOTPKPolicy(omemo.OTPKPolicy):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	166
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	167 @staticmethod
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	168 def decideOTPK(preKeyMessages):
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 # always delete
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	170 return False
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	171
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	172
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	173 class OmemoSession(object):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	174 """Wrapper to use omemo.OmemoSession with Deferred"""
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	175
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	176 def __init__(self, session):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	177 self._session = session
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	178
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	179 @property
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	180 def state(self):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	181 return self._session.state
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	182
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	183 @staticmethod
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	184 def promise2Deferred(promise_):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	185 """Create a Deferred and fire it when promise is resolved
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	186
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	187 @param promise_(promise.Promise): promise to convert
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	188 @return (defer.Deferred): deferred instance linked to the promise
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	189 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	190 d = defer.Deferred()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	191 promise_.then(d.callback, d.errback)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	192 return d
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	193
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	194 @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	195 def create(cls, client, storage, my_device_id = None):
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	196 omemo_session_p = client._xep_0384_session = 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	197 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	198 SatOTPKPolicy,
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	199 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	200 my_device_id)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	201 d = cls.promise2Deferred(omemo_session_p)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	202 d.addCallback(lambda session: cls(session))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	203 return d
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	204
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	205 def newDeviceList(self, devices, jid):
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	206 jid = jid.userhost()
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	207 new_device_p = self._session.newDeviceList(devices, jid)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	208 return self.promise2Deferred(new_device_p)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	209
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	210 def getDevices(self, bare_jid=None):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	211 get_devices_p = self._session.getDevices(bare_jid=bare_jid)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	212 return self.promise2Deferred(get_devices_p)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	213
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	214 def buildSession(self, bare_jid, device, bundle):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	215 bare_jid = bare_jid.userhost()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	216 build_session_p = self._session.buildSession(bare_jid, device, bundle)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	217 return self.promise2Deferred(build_session_p)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	218
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	219 def encryptMessage(self, bare_jids, message, bundles=None, devices=None,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	220 always_trust = False):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	221 """Encrypt a message
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	222
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	223 @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	224 @param message(unicode): message to encode
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	225 @param bundles(dict[jid.JID, dict[int, ExtendedPublicBundle]):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	226 entities => devices => bundles map
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	227 @param devices(iterable[int], None): devices to encode for
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	228 @param always_trust(bool): TODO
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	229 @return D(dict): encryption data
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	230 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	231 if isinstance(bare_jids, jid.JID):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	232 bare_jids = bare_jids.userhost()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	233 else:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	234 bare_jids = [e.userhost() for e in bare_jids]
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	235 if bundles is not None:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	236 bundles = {e.userhost(): v for e, v in bundles.iteritems()}
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	237 encrypt_mess_p = self._session.encryptMessage(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	238 bare_jids=bare_jids,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	239 plaintext=message.encode('utf-8'),
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	240 bundles=bundles,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	241 devices=devices,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	242 always_trust=always_trust)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	243 return self.promise2Deferred(encrypt_mess_p)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	244
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	245 def decryptMessage(self, bare_jid, device, iv, message, is_pre_key_message,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	246 payload=None, from_storage=False):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	247 bare_jid = bare_jid.userhost()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	248 decrypt_mess_p = self._session.decryptMessage(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	249 bare_jid=bare_jid,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	250 device=device,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	251 iv=iv,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	252 message=message,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	253 is_pre_key_message=is_pre_key_message,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	254 payload=payload,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	255 from_storage=from_storage)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	256 return self.promise2Deferred(decrypt_mess_p)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	257
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	258
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	259 class OMEMO(object):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	260 def __init__(self, host):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	261 log.info(_(u"OMEMO plugin initialization"))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	262 self.host = host
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	263 self._p_hints = host.plugins[u"XEP-0334"]
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	264 self._p_carbons = host.plugins[u"XEP-0280"]
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	265 self._p = host.plugins[u"XEP-0060"]
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	266 host.trigger.add("MessageReceived", self._messageReceivedTrigger, priority=100050)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	267 host.trigger.add("sendMessageData", self._sendMessageDataTrigger)
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	268 self.host.registerEncryptionPlugin(self, u"OMEMO", NS_OMEMO, 100)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	269
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	270 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	271 def profileConnected(self, client):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	272 # 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	273 persistent_dict = persistent.LazyPersistentBinaryDict("XEP-0384", client.profile)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	274 # all known devices of profile
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	275 devices = yield self.getDevices(client)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	276 # and our own device id
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	277 device_id = yield persistent_dict.get(KEY_DEVICE_ID)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	278 if device_id is None:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	279 # we have a new device, we create device_id
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	280 device_id = random.randint(1, 2**31-1)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	281 # we check that it's really unique
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	282 while device_id in devices:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	283 device_id = random.randint(1, 2**31-1)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	284 # and we save it
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	285 persistent_dict[KEY_DEVICE_ID] = device_id
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	286
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	287 if device_id not in devices:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	288 devices.add(device_id)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	289 yield self.setDevices(client, devices)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	290
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	291 omemo_storage = OmemoStorage(client, device_id, persistent_dict)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	292 omemo_session = yield OmemoSession.create(client, omemo_storage, device_id)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	293 client._xep_0384_session = omemo_session
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	294 client._xep_0384_device_id = device_id
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	295 yield omemo_session.newDeviceList(devices, client.jid)
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	296 if omemo_session.state.changed:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	297 log.info(_(u"Saving public bundle for this device ({device_id})").format(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	298 device_id=device_id))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	299 bundle = omemo_session.state.getPublicBundle()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	300 yield self.setBundle(client, bundle, device_id)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	301
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	302 ## XMPP PEP nodes manipulation
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	303
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	304 # devices
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	305
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	306 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	307 def getDevices(self, client, entity_jid=None):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	308 """Retrieve list of registered OMEMO devices
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	309
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	310 @param entity_jid(jid.JID, None): get devices from this entity
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	311 None to get our own devices
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	312 @return (set(int)): list of devices
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	313 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	314 if entity_jid is not None:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	315 assert not entity_jid.resource
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	316 devices = set()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	317 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	318 items, metadata = yield self._p.getItems(client, entity_jid, NS_OMEMO_DEVICES)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	319 except error.StanzaError as e:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	320 if e.condition == 'item-not-found':
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	321 log.info(_(u"there is no node to handle OMEMO devices"))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	322 defer.returnValue(devices)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	323
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	324 if len(items) > 1:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	325 log.warning(_(u"OMEMO devices list is stored in more that one items, "
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	326 u"this is not expected"))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	327 if items:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	328 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	329 list_elt = next(items[0].elements(NS_OMEMO, 'list'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	330 except StopIteration:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	331 log.warning(_(u"no list element found in OMEMO devices list"))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	332 return
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	333 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	334 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	335 device_id = int(device_elt['id'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	336 except KeyError:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	337 log.warning(_(u'device element is missing "id" attribute: {elt}')
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	338 .format(elt=device_elt.toXml()))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	339 except ValueError:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	340 log.warning(_(u'invalid device id: {device_id}').format(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	341 device_id=device_elt['id']))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	342 else:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	343 devices.add(device_id)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	344 defer.returnValue(devices)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	345
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	346 def setDevicesEb(self, failure_):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	347 log.warning(_(u"Can't set devices: {reason}").format(reason=failure_))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	348
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	349 def setDevices(self, client, devices):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	350 list_elt = domish.Element((NS_OMEMO, 'list'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	351 for device in devices:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	352 device_elt = list_elt.addElement('device')
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	353 device_elt['id'] = unicode(device)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	354 d = self._p.sendItem(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	355 client, None, NS_OMEMO_DEVICES, list_elt, item_id=self._p.ID_SINGLETON)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	356 d.addErrback(self.setDevicesEb)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	357 return d
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	358
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	359 # bundles
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	360
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	361 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	362 def getBundles(self, client, entity_jid, devices_ids):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	363 """Retrieve public bundles of an entity devices
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	364
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	365 @param entity_jid(jid.JID): bare jid of entity
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	366 @param devices_id(iterable[int]): ids of the devices bundles to retrieve
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	367 @return (dict[int, ExtendedPublicBundle]): bundles collection
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	368 key is device_id
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	369 value is parsed bundle
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	370 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	371 assert not entity_jid.resource
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	372 bundles = {}
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	373 for device_id in devices_ids:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	374 node = NS_OMEMO_BUNDLE.format(device_id=device_id)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	375 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	376 items, metadata = yield self._p.getItems(client, entity_jid, node)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	377 except Exception as e:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	378 log.warning(_(u"Can't get bundle for device {device_id}: {reason}")
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	379 .format(device_id=device_id, reason=e))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	380 continue
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	381 if not items:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	382 log.warning(_(u"no item found in node {node}, can't get public bundle "
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	383 u"for device {device_id}").format(node=node,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	384 device_id=device_id))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	385 continue
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	386 if len(items) > 1:
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	387 log.warning(_(u"more than one item found in {node}, "
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	388 u"this is not expected").format(node=node))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	389 item = items[0]
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	390 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	391 bundle_elt = next(item.elements(NS_OMEMO, 'bundle'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	392 signedPreKeyPublic_elt = next(bundle_elt.elements(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	393 NS_OMEMO, 'signedPreKeyPublic'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	394 signedPreKeySignature_elt = next(bundle_elt.elements(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	395 NS_OMEMO, 'signedPreKeySignature'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	396 identityKey_elt = next(bundle_elt.elements(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	397 NS_OMEMO, 'identityKey'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	398 prekeys_elt = next(bundle_elt.elements(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	399 NS_OMEMO, 'prekeys'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	400 except StopIteration:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	401 log.warning(_(u"invalid bundle for device {device_id}, ignoring").format(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	402 device_id=device_id))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	403 continue
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	404
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	405 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	406 spkPublic = base64.b64decode(unicode(signedPreKeyPublic_elt))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	407 spkSignature = base64.b64decode(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	408 unicode(signedPreKeySignature_elt))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	409
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	410 identityKey = base64.b64decode(unicode(identityKey_elt))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	411 spk = {
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	412 "key": wireformat.decodePublicKey(spkPublic),
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	413 "id": int(signedPreKeyPublic_elt['signedPreKeyId'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	414 }
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	415 ik = wireformat.decodePublicKey(identityKey)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	416 otpks = []
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	417 for preKeyPublic_elt in prekeys_elt.elements(NS_OMEMO, 'preKeyPublic'):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	418 preKeyPublic = base64.b64decode(unicode(preKeyPublic_elt))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	419 otpk = {
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	420 "key": wireformat.decodePublicKey(preKeyPublic),
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	421 "id": int(preKeyPublic_elt['preKeyId'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	422 }
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	423 otpks.append(otpk)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	424
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	425 except Exception as e:
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	426 log.warning(_(u"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	427 .format(device_id=device_id, msg=e))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	428 continue
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	429
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	430 bundles[device_id] = ExtendedPublicBundle(ik, spk, spkSignature, otpks)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	431
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	432 defer.returnValue(bundles)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	433
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	434 def setBundleEb(self, failure_):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	435 log.warning(_(u"Can't set bundle: {reason}").format(reason=failure_))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	436
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	437 def setBundle(self, client, bundle, device_id):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	438 """Set public bundle for this device.
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	439
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	440 @param bundle(ExtendedPublicBundle): bundle to publish
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	441 """
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	442 log.debug(_(u"updating bundle for {device_id}").format(device_id=device_id))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	443 bundle_elt = domish.Element((NS_OMEMO, 'bundle'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	444 signedPreKeyPublic_elt = bundle_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	445 "signedPreKeyPublic",
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	446 content=b64enc(wireformat.encodePublicKey(bundle.spk['key'])))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	447 signedPreKeyPublic_elt['signedPreKeyId'] = unicode(bundle.spk['id'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	448
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	449 bundle_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	450 "signedPreKeySignature",
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	451 content=b64enc(bundle.spk_signature))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	452
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	453 bundle_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	454 "identityKey",
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	455 content=b64enc(wireformat.encodePublicKey(bundle.ik)))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	456
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	457 prekeys_elt = bundle_elt.addElement('prekeys')
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	458 for otpk in bundle.otpks:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	459 preKeyPublic_elt = prekeys_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	460 'preKeyPublic',
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	461 content=b64enc(wireformat.encodePublicKey(otpk["key"])))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	462 preKeyPublic_elt['preKeyId'] = unicode(otpk['id'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	463
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	464 node = NS_OMEMO_BUNDLE.format(device_id=device_id)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	465 d = self._p.sendItem(client, None, node, bundle_elt, item_id=self._p.ID_SINGLETON)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	466 d.addErrback(self.setBundleEb)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	467 return d
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	468
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	469 ## triggers
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	470
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	471 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	472 def encryptMessage(self, client, entity_bare_jid, message):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	473 omemo_session = client._xep_0384_session
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	474 devices = yield self.getDevices(client, entity_bare_jid)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	475 omemo_session.newDeviceList(devices, entity_bare_jid)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	476 bundles = yield self.getBundles(client, entity_bare_jid, devices)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	477 encrypted = yield omemo_session.encryptMessage(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	478 entity_bare_jid,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	479 message,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	480 {entity_bare_jid: bundles})
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	481 defer.returnValue(encrypted)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	482
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	483 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	484 def _messageReceivedTrigger(self, client, message_elt, post_treat):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	485 if message_elt.getAttribute("type") == C.MESS_TYPE_GROUPCHAT:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	486 defer.returnValue(True)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	487 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	488 encrypted_elt = next(message_elt.elements(NS_OMEMO, u"encrypted"))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	489 except StopIteration:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	490 # no OMEMO message here
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	491 defer.returnValue(True)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	492
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	493 # we have an encrypted message let's decrypt it
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	494 from_jid = jid.JID(message_elt['from'])
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	495 omemo_session = client._xep_0384_session
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	496 device_id = client._xep_0384_device_id
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	497 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	498 header_elt = next(encrypted_elt.elements(NS_OMEMO, u'header'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	499 iv_elt = next(header_elt.elements(NS_OMEMO, u'iv'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	500 except StopIteration:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	501 log.warning(_(u"Invalid OMEMO encrypted stanza, ignoring: {xml}")
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	502 .format(xml=message_elt.toXml()))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	503 defer.returnValue(False)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	504 try:
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	505 s_device_id = header_elt['sid']
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	506 except KeyError:
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	507 log.warning(_(u"Invalid OMEMO encrypted stanza, missing sender 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	508 u"ignoring: {xml}")
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	509 .format(xml=message_elt.toXml()))
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	510 defer.returnValue(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	511 try:
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	512 key_elt = next((e for e in header_elt.elements(NS_OMEMO, u'key')
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	513 if int(e[u'rid']) == device_id))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	514 except StopIteration:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	515 log.warning(_(u"This OMEMO encrypted stanza has not been encrypted"
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	516 u"for our device ({device_id}): {xml}").format(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	517 device_id=device_id, xml=encrypted_elt.toXml()))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	518 defer.returnValue(False)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	519 except ValueError as e:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	520 log.warning(_(u"Invalid recipient ID: {msg}".format(msg=e)))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	521 defer.returnValue(False)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	522 is_pre_key = C.bool(key_elt.getAttribute('prekey', 'false'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	523 payload_elt = next(encrypted_elt.elements(NS_OMEMO, u'payload'), None)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	524
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	525 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	526 cipher, plaintext = yield omemo_session.decryptMessage(
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	527 bare_jid=from_jid.userhostJID(),
e7bfbded652a plugin XEP-0384, install: adapted plugin to omemo module changes + added omemo module to dependencies: Goffi <goffi@goffi.org> parents: 2648 diff changeset	528 device=s_device_id,
2648 0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	529 iv=base64.b64decode(bytes(iv_elt)),
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	530 message=base64.b64decode(bytes(key_elt)),
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	531 is_pre_key_message=is_pre_key,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	532 payload=base64.b64decode(bytes(payload_elt))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	533 if payload_elt is not None else None,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	534 from_storage=False
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	535 )
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	536 except Exception as e:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	537 log.error(_(u"Can't decrypt message: {reason}\n{xml}").format(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	538 reason=e, xml=message_elt.toXml()))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	539 defer.returnValue(False)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	540 if omemo_session.state.changed:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	541 bundle = omemo_session.state.getPublicBundle()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	542 # we don't wait for the Deferred (i.e. no yield) on purpose
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	543 # there is no need to block the whole message workflow while
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	544 # updating the bundle
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	545 self.setBundle(client, bundle, device_id)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	546
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	547 message_elt.children.remove(encrypted_elt)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	548 if plaintext:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	549 message_elt.addElement("body", content=plaintext.decode('utf-8'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	550 defer.returnValue(True)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	551
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	552 @defer.inlineCallbacks
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	553 def _sendMessageDataTrigger(self, client, mess_data):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	554 encryption = mess_data.get(C.MESS_KEY_ENCRYPTION)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	555 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	556 return
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	557 message_elt = mess_data["xml"]
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	558 to_jid = mess_data["to"].userhostJID()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	559 log.debug(u"encrypting message")
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	560 body = None
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	561 for child in list(message_elt.children):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	562 if child.name == "body":
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	563 # we remove all unencrypted body,
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	564 # and will only encrypt the first one
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	565 if body is None:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	566 body = child
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	567 message_elt.children.remove(child)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	568 elif child.name == "html":
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	569 # we don't want any XHTML-IM element
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	570 message_elt.children.remove(child)
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	571
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	572 if body is None:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	573 log.warning(u"No message found")
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	574 return
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	575
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	576 encryption_data = yield self.encryptMessage(client, to_jid, unicode(body))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	577
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	578 encrypted_elt = message_elt.addElement((NS_OMEMO, 'encrypted'))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	579 header_elt = encrypted_elt.addElement('header')
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	580 header_elt['sid'] = unicode(encryption_data['sid'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	581 bare_jid_s = to_jid.userhost()
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	582
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	583 for message in (m for m in encryption_data['messages']
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	584 if m['bare_jid'] == bare_jid_s):
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	585 key_elt = header_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	586 'key',
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	587 content=b64enc(message['message']))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	588 key_elt['rid'] = unicode(message['rid'])
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	589 if message['pre_key']:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	590 key_elt['prekey'] = 'true'
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	591
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	592 header_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	593 'iv',
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	594 content=b64enc(encryption_data['iv']))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	595 try:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	596 encrypted_elt.addElement(
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	597 'payload',
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	598 content=b64enc(encryption_data['payload']))
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	599 except KeyError:
0f76813afc57 plugin XEP-0384: OMEMO implementation first draft: Goffi <goffi@goffi.org> parents: diff changeset	600 pass

Mercurial > libervia-backend

annotate sat/plugins/plugin_xep_0384.py @ 2654:e7bfbded652a