python-cffi: returned OwnedMessage objects from Message.replies

If we return regular Message objects, python will try to destroy them,
and the underlying notmuch object, causing e.g. the crash [1].

[1]: id:87sfu6utxg.fsf@tethera.net
This commit is contained in:
David Bremner 2022-01-08 10:03:16 -04:00
parent d9a2b900b6
commit 9e7ea628e6
2 changed files with 2 additions and 4 deletions

View file

@ -357,14 +357,14 @@ class Message(base.NotmuchObject):
This method will only work if the message was created from a This method will only work if the message was created from a
thread. Otherwise it will yield no results. thread. Otherwise it will yield no results.
:returns: An iterator yielding :class:`Message` instances. :returns: An iterator yielding :class:`OwnedMessage` instances.
:rtype: MessageIter :rtype: MessageIter
""" """
# The notmuch_messages_valid call accepts NULL and this will # The notmuch_messages_valid call accepts NULL and this will
# become an empty iterator, raising StopIteration immediately. # become an empty iterator, raising StopIteration immediately.
# Hence no return value checking here. # Hence no return value checking here.
msgs_p = capi.lib.notmuch_message_get_replies(self._msg_p) msgs_p = capi.lib.notmuch_message_get_replies(self._msg_p)
return MessageIter(self, msgs_p, db=self._db) return MessageIter(self, msgs_p, db=self._db, msg_cls=OwnedMessage)
def __hash__(self): def __hash__(self):
return hash(self.messageid) return hash(self.messageid)

View file

@ -24,13 +24,11 @@ show_msgs(thread, 0)
EOF EOF
test_begin_subtest "recursive traversal of replies (no crash)" test_begin_subtest "recursive traversal of replies (no crash)"
test_subtest_known_broken
test_python < recurse.py test_python < recurse.py
error=$? error=$?
test_expect_equal "${error}" 0 test_expect_equal "${error}" 0
test_begin_subtest "recursive traversal of replies (output)" test_begin_subtest "recursive traversal of replies (output)"
test_subtest_known_broken
test_python < recurse.py test_python < recurse.py
tail -n 10 < OUTPUT > OUTPUT.sample tail -n 10 < OUTPUT > OUTPUT.sample
cat <<EOF > EXPECTED cat <<EOF > EXPECTED