KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Allow ignoring SHM close failures

Browse Source 
Useful if we export the mmap beyond the lifetime of the shm object
This commit is contained in:
Kovid Goyal 2022-06-06 20:31:13 +05:30
parent 79f7954048
commit dec62b1929
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
kitty/shm.py
View File

@ -47,11 +47,13 @@ class SharedMemory:
num_bytes_for_size = struct.calcsize(size_fmt) num_bytes_for_size = struct.calcsize(size_fmt)
def __init__( def __init__(
self, name: str = '', size: int = 0, readonly: bool = False, self, name: str = '', size: int = 0, readonly: bool = False,
mode: int = stat.S_IREAD | stat.S_IWRITE, mode: int = stat.S_IREAD | stat.S_IWRITE,
prefix: str = 'kitty-', unlink_on_exit: bool = False prefix: str = 'kitty-',
unlink_on_exit: bool = False, ignore_close_failure: bool = False
): ):
self.unlink_on_exit = unlink_on_exit self.unlink_on_exit = unlink_on_exit
self.ignore_close_failure = ignore_close_failure
if size < 0: if size < 0:
raise TypeError("'size' must be a non-negative integer") raise TypeError("'size' must be a non-negative integer")
if size and name: if size and name:
@ -157,7 +159,11 @@ class SharedMemory:
"""Closes access to the shared memory from this instance but does """Closes access to the shared memory from this instance but does
not destroy the shared memory block.""" not destroy the shared memory block."""
if self._mmap is not None: if self._mmap is not None:
self._mmap.close() try:
self._mmap.close()
except BufferError:
if not self.ignore_close_failure:
raise
self._mmap = None self._mmap = None
if self._fd >= 0: if self._fd >= 0:
os.close(self._fd) os.close(self._fd)