include RSM max in mam query in case server doesn't specify own max

1 files changed, 4 insertions, 2 deletions

diff --git a/src/main/java/de/pixart/messenger/generator/IqGenerator.java b/src/main/java/de/pixart/messenger/generator/IqGenerator.java
index e919b437e..301f4d178 100644
--- a/src/main/java/de/pixart/messenger/generator/IqGenerator.java
+++ b/src/main/java/de/pixart/messenger/generator/IqGenerator.java
@@ -250,11 +250,13 @@ public class IqGenerator extends AbstractGenerator {
         data.put("end", getTimestamp(mam.getEnd()));
         data.submit();
         query.addChild(data);
+        Element set = query.addChild("set", "http://jabber.org/protocol/rsm");
         if (mam.getPagingOrder() == MessageArchiveService.PagingOrder.REVERSE) {
-            query.addChild("set", "http://jabber.org/protocol/rsm").addChild("before").setContent(mam.getReference());
+            set.addChild("before").setContent(mam.getReference());
         } else if (mam.getReference() != null) {
-            query.addChild("set", "http://jabber.org/protocol/rsm").addChild("after").setContent(mam.getReference());
+            set.addChild("after").setContent(mam.getReference());
         }
+        set.addChild("max").setContent(String.valueOf(Config.PAGE_SIZE));
         return packet;
     }