MAM: Check for first and last element in IQ result

1 files changed, 6 insertions, 0 deletions

diff --git a/nbxmpp/modules/mam.py b/nbxmpp/modules/mam.py
index 0dd7528..b65c82c 100644
--- a/nbxmpp/modules/mam.py
+++ b/nbxmpp/modules/mam.py
@@ -119,6 +119,12 @@ class MAM(BaseModule):
                                'rsm set missing')
 
         complete = fin.getAttr('complete') == 'true'
+        if not complete:
+            if rsm.first is None or rsm.last is None:
+                return raise_error(self._log.warning,
+                                   stanza,
+                                   'stanza-malformed',
+                                   'missing first or last element')
 
         return MAMQueryData(jid=jid,
                             complete=complete,