summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--slixmpp/plugins/xep_0300/__init__.py16
-rw-r--r--slixmpp/plugins/xep_0300/hash.py80
-rw-r--r--slixmpp/plugins/xep_0300/stanza.py35
3 files changed, 131 insertions, 0 deletions
diff --git a/slixmpp/plugins/xep_0300/__init__.py b/slixmpp/plugins/xep_0300/__init__.py
new file mode 100644
index 00000000..522d40e3
--- /dev/null
+++ b/slixmpp/plugins/xep_0300/__init__.py
@@ -0,0 +1,16 @@
+"""
+ Slixmpp: The Slick XMPP Library
+ Copyright (C) 2017 Emmanuel Gil Peyrot
+ This file is part of Slixmpp.
+
+ See the file LICENSE for copying permission.
+"""
+
+from slixmpp.plugins.base import register_plugin
+
+from slixmpp.plugins.xep_0300 import stanza
+from slixmpp.plugins.xep_0300.stanza import Hash
+from slixmpp.plugins.xep_0300.hash import XEP_0300
+
+
+register_plugin(XEP_0300)
diff --git a/slixmpp/plugins/xep_0300/hash.py b/slixmpp/plugins/xep_0300/hash.py
new file mode 100644
index 00000000..0c71aac1
--- /dev/null
+++ b/slixmpp/plugins/xep_0300/hash.py
@@ -0,0 +1,80 @@
+"""
+ Slixmpp: The Slick XMPP Library
+ Copyright (C) 2017 Emmanuel Gil Peyrot
+ This file is part of Slixmpp.
+
+ See the file LICENSE for copying permission.
+"""
+
+from base64 import b64encode
+import hashlib
+import logging
+
+from slixmpp.plugins import BasePlugin
+from slixmpp.plugins.xep_0300 import stanza, Hash
+
+
+log = logging.getLogger(__name__)
+
+
+class XEP_0300(BasePlugin):
+
+ name = 'xep_0300'
+ description = 'XEP-0300: Use of Cryptographic Hash Functions in XMPP'
+ dependencies = {'xep_0030'}
+ stanza = stanza
+ default_config = {
+ 'block_size': 1024 * 1024, # One MiB
+ 'prefered': 'sha-256',
+ 'enable_sha-1': False,
+ 'enable_sha-256': True,
+ 'enable_sha-512': True,
+ 'enable_sha3-256': True,
+ 'enable_sha3-512': True,
+ 'enable_BLAKE2b256': True,
+ 'enable_BLAKE2b512': True,
+ }
+
+ _hashlib_function = {
+ 'sha-1': hashlib.sha1,
+ 'sha-256': hashlib.sha256,
+ 'sha-512': hashlib.sha512,
+ 'sha3-256': hashlib.sha3_256,
+ 'sha3-512': hashlib.sha3_512,
+ 'BLAKE2b256': lambda: hashlib.blake2b(digest_size=32),
+ 'BLAKE2b512': lambda: hashlib.blake2b(digest_size=64),
+ }
+
+ def plugin_init(self):
+ namespace = 'urn:xmpp:hash-function-text-names:%s'
+ self.enabled_hashes = []
+ for algo in self._hashlib_function:
+ if getattr(self, 'enable_' + algo, False):
+ self.enabled_hashes.append(namespace % algo)
+
+ def session_bind(self, jid):
+ self.xmpp['xep_0030'].add_feature(Hash.namespace)
+
+ for namespace in self.enabled_hashes:
+ self.xmpp['xep_0030'].add_feature(namespace)
+
+ def plugin_end(self):
+ for namespace in self.enabled_hashes:
+ self.xmpp['xep_0030'].del_feature(namespace)
+
+ self.xmpp['xep_0030'].del_feature(feature=Hash.namespace)
+
+ def compute_hash(self, filename, function=None):
+ if function is None:
+ function = self.prefered
+ h = self._hashlib_function[function]()
+ with open(filename, 'rb') as f:
+ while True:
+ block = f.read(self.block_size)
+ if not block:
+ break
+ h.update(block)
+ hash_elem = Hash()
+ hash_elem['algo'] = function
+ hash_elem['value'] = b64encode(h.digest())
+ return hash_elem
diff --git a/slixmpp/plugins/xep_0300/stanza.py b/slixmpp/plugins/xep_0300/stanza.py
new file mode 100644
index 00000000..f5ab483c
--- /dev/null
+++ b/slixmpp/plugins/xep_0300/stanza.py
@@ -0,0 +1,35 @@
+"""
+ Slixmpp: The Slick XMPP Library
+ Copyright (C) 2017 Emmanuel Gil Peyrot
+ This file is part of Slixmpp.
+
+ See the file LICENSE for copying permission.
+"""
+
+from slixmpp.xmlstream import ElementBase
+
+
+class Hash(ElementBase):
+ name = 'hash'
+ namespace = 'urn:xmpp:hashes:2'
+ plugin_attrib = 'hash'
+ interfaces = {'algo', 'value'}
+
+ allowed_algos = ['sha-1', 'sha-256', 'sha-512', 'sha3-256', 'sha3-512', 'BLAKE2b256', 'BLAKE2b512']
+
+ def set_algo(self, value):
+ if value in self.allowed_algos:
+ self._set_attr('algo', value)
+ elif value in [None, '']:
+ self._del_attr('algo')
+ else:
+ raise ValueError('Invalid algo: %s' % value)
+
+ def get_value(self):
+ return self.xml.text
+
+ def set_value(self, value):
+ self.xml.text = value
+
+ def del_value(self):
+ self.xml.text = ''