fix: handle single-line log files, and reentry into fd_busy

1 files changed, 12 insertions, 7 deletions

diff --git a/poezio/logger.py b/poezio/logger.py
index d82ce9ee..e46be469 100644
--- a/poezio/logger.py
+++ b/poezio/logger.py
@@ -136,7 +136,8 @@ class Logger:
         """
         jidstr = str(jid).replace('/', '\\')
         self._busy_fds[jidstr] = True
-        self._buffered_fds[jidstr] = []
+        if jidstr not in self._buffered_fds:
+            self._buffered_fds[jidstr] = []
 
     def fd_available(self, jid: Union[str, JID]) -> None:
         """Signal to the logger that this logfile is no longer busy.
@@ -371,15 +372,19 @@ def iterate_messages_reverse(filepath: Path) -> Generator[LogDict, None, None]:
             with mmap.mmap(fd.fileno(), 0, prot=mmap.PROT_READ) as m:
                 # start of messages begin with MI or MR, after a \n
                 pos = m.rfind(b"\nM") + 1
-                lines = parse_log_lines(
-                    m[pos:-1].decode(errors='replace').splitlines()
-                )
+                if pos != -1:
+                    lines = parse_log_lines(
+                        m[pos:-1].decode(errors='replace').splitlines()
+                    )
+                elif m[0:1] == b'M':
+                    # Handle the case of a single message present in the log
+                    # file, hence no newline.
+                    lines = parse_log_lines(
+                        m[:].decode(errors='replace').splitlines()
+                    )
                 if lines:
                     yield lines[0]
-                # number of message found so far
-                count = 0
                 while pos > 0:
-                    count += 1
                     old_pos = pos
                     pos = m.rfind(b"\nM", 0, pos) + 1
                     lines = parse_log_lines(