jshiffer/slixmpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

XEP-0047: prevent tracebacks in stanza reading.

Browse Source
This commit is contained in:
Emmanuel Gil Peyrot 2015-04-14 17:18:34 +02:00
parent d5b1904ebb
commit 45f7cb8bda
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
slixmpp/plugins/xep_0047/stanza.py
View File

@ -24,7 +24,7 @@ class Open(ElementBase):
interfaces = set(('block_size', 'sid', 'stanza')) interfaces = set(('block_size', 'sid', 'stanza'))
def get_block_size(self): def get_block_size(self):
return int(self._get_attr('block-size')) return int(self._get_attr('block-size', '0'))
def set_block_size(self, value): def set_block_size(self, value):
self._set_attr('block-size', str(value)) self._set_attr('block-size', str(value))
@ -47,7 +47,10 @@ class Data(ElementBase):
self._set_attr('seq', str(value)) self._set_attr('seq', str(value))
def get_data(self): def get_data(self):
b64_data = self.xml.text.strip() text = self.xml.text
if not text:
raise XMPPError('not-acceptable', 'IBB data element is empty.')
b64_data = text.strip()
if VALID_B64.match(b64_data).group() == b64_data: if VALID_B64.match(b64_data).group() == b64_data:
return from_b64(b64_data) return from_b64(b64_data)
else: else: