194 lines
4.2 KiB
Python
194 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Slixmpp OMEMO plugin
|
|
Copyright (C) 2018 Maxime “pep” Buquet <pep@bouah.net>
|
|
This file is part of slixmpp-omemo.
|
|
|
|
See the file LICENSE for copying permission.
|
|
"""
|
|
|
|
from slixmpp import Message
|
|
from slixmpp.plugins.xep_0060.stanza.pubsub import Item
|
|
from slixmpp.plugins.xep_0060.stanza.pubsub_event import EventItem
|
|
from slixmpp.xmlstream import register_stanza_plugin, ElementBase
|
|
|
|
OMEMO_BASE_NS = 'eu.siacs.conversations.axolotl'
|
|
OMEMO_DEVICES_NS = OMEMO_BASE_NS + '.devicelist'
|
|
OMEMO_BUNDLES_NS = OMEMO_BASE_NS + '.bundles'
|
|
|
|
|
|
class Devices(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'list'
|
|
plugin_attrib = 'devices'
|
|
interfaces = set()
|
|
|
|
|
|
class Device(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'device'
|
|
plugin_attrib = name
|
|
plugin_multi_attrib = 'devices'
|
|
interfaces = {'id'}
|
|
|
|
|
|
class Encrypted(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'encrypted'
|
|
plugin_attrib = 'omemo_encrypted'
|
|
interfaces = set()
|
|
|
|
|
|
class Header(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'header'
|
|
plugin_attrib = name
|
|
interfaces = {'sid'}
|
|
|
|
|
|
class Key(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'key'
|
|
plugin_attrib = name
|
|
interfaces = {'rid', 'prekey', 'value'}
|
|
|
|
def get_value(self):
|
|
return self.xml.text
|
|
|
|
def set_value(self, value):
|
|
self.xml.text = str(value)
|
|
|
|
def del_value(self):
|
|
self.xml.text = ''
|
|
|
|
|
|
class IV(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'iv'
|
|
plugin_attrib = name
|
|
interfaces = {'value'}
|
|
|
|
def get_value(self):
|
|
return self.xml.text
|
|
|
|
def set_value(self, value):
|
|
self.xml.text = str(value)
|
|
|
|
def del_value(self):
|
|
self.xml.text = ''
|
|
|
|
|
|
class Payload(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'payload'
|
|
plugin_attrib = name
|
|
interfaces = {'value'}
|
|
|
|
def get_value(self):
|
|
return self.xml.text
|
|
|
|
def set_value(self, value):
|
|
self.xml.text = str(value)
|
|
|
|
def del_value(self):
|
|
self.xml.text = ''
|
|
|
|
|
|
|
|
class Bundle(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'bundle'
|
|
plugin_attrib = name
|
|
interfaces = set()
|
|
|
|
|
|
class SignedPreKeyPublic(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'signedPreKeyPublic'
|
|
plugin_attrib = name
|
|
interfaces = {'signedPreKeyId', 'value'}
|
|
|
|
def get_value(self):
|
|
return self.xml.text
|
|
|
|
def set_value(self, value):
|
|
self.xml.text = str(value)
|
|
|
|
def del_value(self):
|
|
self.xml.text = ''
|
|
|
|
|
|
class SignedPreKeySignature(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'signedPreKeySignature'
|
|
plugin_attrib = name
|
|
interfaces = {'value'}
|
|
|
|
def get_value(self):
|
|
return self.xml.text
|
|
|
|
def set_value(self, value):
|
|
self.xml.text = str(value)
|
|
|
|
def del_value(self):
|
|
self.xml.text = ''
|
|
|
|
|
|
class IdentityKey(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'identityKey'
|
|
plugin_attrib = name
|
|
interfaces = {'value'}
|
|
|
|
def get_value(self):
|
|
return self.xml.text
|
|
|
|
def set_value(self, value):
|
|
self.xml.text = str(value)
|
|
|
|
def del_value(self):
|
|
self.xml.text = ''
|
|
|
|
|
|
class PreKeys(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'prekeys'
|
|
plugin_attrib = name
|
|
interfaces = set()
|
|
|
|
|
|
class PreKeyPublic(ElementBase):
|
|
namespace = OMEMO_BASE_NS
|
|
name = 'preKeyPublic'
|
|
plugin_attrib = name
|
|
plugin_multi_attrib = 'prekeys'
|
|
interfaces = {'preKeyId', 'value'}
|
|
|
|
def get_value(self):
|
|
return self.xml.text
|
|
|
|
def set_value(self, value):
|
|
self.xml.text = str(value)
|
|
|
|
def del_value(self):
|
|
self.xml.text = ''
|
|
|
|
|
|
register_stanza_plugin(Message, Encrypted)
|
|
register_stanza_plugin(Encrypted, Header)
|
|
register_stanza_plugin(Header, Key)
|
|
register_stanza_plugin(Header, IV)
|
|
register_stanza_plugin(Encrypted, Payload)
|
|
|
|
register_stanza_plugin(Item, Devices)
|
|
register_stanza_plugin(EventItem, Devices)
|
|
register_stanza_plugin(Devices, Device, iterable=True)
|
|
|
|
register_stanza_plugin(Item, Bundle)
|
|
register_stanza_plugin(Bundle, SignedPreKeyPublic)
|
|
register_stanza_plugin(Bundle, SignedPreKeySignature)
|
|
register_stanza_plugin(Bundle, IdentityKey)
|
|
register_stanza_plugin(Bundle, PreKeys)
|
|
register_stanza_plugin(PreKeys, PreKeyPublic, iterable=True)
|