multiprocessing.shared_memory: failed attach unlinks another process's block
還沒有人認領這個 Issue。
- 主要語言
- Python
- 星號
- 77.2k
- 分支
- 35.9k
- PR 合併指標
- PR 指標待擷取
描述
Bug report
Bug description
SharedMemory.__init__ calls self.unlink() from its OSError cleanup handler:
try:
if create and size:
os.ftruncate(self._fd, size)
stats = os.fstat(self._fd)
size = stats.st_size
self._mmap = mmap.mmap(self._fd, size)
except OSError:
self.unlink()
raise
This runs unconditionally, including on the attach path (create=False).
unlink() calls shm_unlink(), which permanently destroys the named block.
So if mmap() fails while attaching to a block created by another process
(e.g. ENOMEM under memory pressure), the failing attach destroys another
owner's live shared memory block.
Reproduction (simulating an mmap failure during attach):
from unittest import mock
from multiprocessing import shared_memory
owner = shared_memory.SharedMemory(create=True, size=1024)
try:
with mock.patch("multiprocessing.shared_memory.mmap.mmap",
side_effect=OSError("ENOMEM")):
shared_memory.SharedMemory(owner.name) # attach
except OSError:
pass
shared_memory.SharedMemory(owner.name) # FileNotFoundError: block destroyed
Secondary issue: resource_tracker.register() runs after this try/except, so
on the create=True path the block is not yet registered. unlink() still
calls resource_tracker.unregister(), so the resource_tracker does
cache[rtype].remove(name) on an unknown name → KeyError printed as a
traceback, with the tracker's exit code set to 3.
Expected behavior
A failed attach must not unlink a block it does not own. On error the fd should
just be closed; the block should be unlinked only when it was created in this
call, and without a spurious unregister for a never-registered block.
Your environment
- CPython main (3.16.0a0); also affects earlier versions
- Linux / POSIX shared memory (
_USE_POSIX)
Linked PRs
- gh-153280
貢獻指南
從這裡開始
- 先讀完整個 Issue,再讀專案的貢獻指南。
- 在 Issue 下留言說明你要接手 —— 這能避免兩個人做同樣的事。
- Fork 儲存庫,在一個分支上完成修改。
- 送出 Pull Request,並在描述裡引用這個 Issue 編號。
研究方向
從 multiprocessing.shared_memory.SharedMemory.init 開始,使用 issue 中展示的 mock 重現 attach 失敗。檢查 attach 和 create 路徑中的 cleanup 和 resource-tracker 行為。完成的標準是:attach 失敗後,現有 block 仍可使用,且不會產生多餘的 tracker 錯誤;issue 中包含連結的 PR gh-153280。
由索引模型根據 Issue 內容生成。
評估
- 技術堆疊
- python
- 領域
- operating-systems
- Issue 類型
- 缺陷
- 難度
- 3/5
- 預估耗時
- 1-2 天
- 活躍度
- 停滯
- 描述清晰度
- 基本清楚
- 新手友好度
- 25/100