decompals/wibo
1
0
Fork 0
mirror of https://github.com/decompals/wibo.git synced 2025-10-15 14:45:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix pidfd reuse race condition

Browse Source
This commit is contained in:
Luke Street 2025-10-09 16:52:12 -06:00
parent cadb3cd00e
commit bb4bee83b7
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/processes.cpp
View File

@ -106,9 +106,11 @@ bool ProcessManager::addProcess(Pin<ProcessObject> po) {
if (!po) {
return false;
}
pid_t pid;
int pidfd;
{
std::lock_guard lk(po->m);
pid = po->pid;
pidfd = po->pidfd;
if (pidfd < 0) {
return false;
@ -128,7 +130,7 @@ bool ProcessManager::addProcess(Pin<ProcessObject> po) {
std::lock_guard lk(m);
mReg.emplace(pidfd, std::move(po));
}
DEBUG_LOG("ProcessManager: tracking pidfd %d\n", pidfd);
DEBUG_LOG("ProcessManager: registered pid %d with pidfd %d\n", pid, pidfd);
wake();
return true;
}
@ -181,7 +183,6 @@ void ProcessManager::checkPidfd(int pidfd) {
return;
}
epoll_ctl(mEpollFd, EPOLL_CTL_DEL, pidfd, nullptr);
close(pidfd);
}
DEBUG_LOG("ProcessManager: pidfd %d exited: code=%d status=%d\n", pidfd, si.si_code, si.si_status);
@ -190,10 +191,14 @@ void ProcessManager::checkPidfd(int pidfd) {
{
std::shared_lock lk(m);
auto it = mReg.find(pidfd);
if (it == mReg.end()) {
return;
if (it != mReg.end()) {
po = std::move(it->second);
mReg.erase(it);
}
po = it->second.clone();
}
close(pidfd);
if (!po) {
return;
}
{
std::lock_guard lk(po->m);
@ -205,14 +210,6 @@ void ProcessManager::checkPidfd(int pidfd) {
}
po->cv.notify_all();
po->notifyWaiters(false);
{
std::lock_guard lk(m);
auto it = mReg.find(pidfd);
if (it != mReg.end()) {
mReg.erase(it);
}
}
}
ProcessManager &processes() {