coffee/cpp-ipc
1
0
Fork 0
mirror of https://github.com/mutouyun/cpp-ipc.git synced 2025-12-06 16:56:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: recv may block after disconnect

Browse Source
This commit is contained in:
mutouyun 2021-01-10 14:29:23 +08:00
parent 8e6ae4220b
commit 152b5515d2
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/ipc.cpp
View File

@ -555,7 +555,7 @@ static ipc::buff_t recv(ipc::handle_t h, std::size_t tm) {
recycle_storage(reinterpret_cast<std::size_t>(ptr) - 1, size); recycle_storage(reinterpret_cast<std::size_t>(ptr) - 1, size);
}, reinterpret_cast<void*>(buf_id + 1) }; }, reinterpret_cast<void*>(buf_id + 1) };
} }
else ipc::log("fail: shm::handle for big message. msg_id: %zd, size: %zd\n", msg.id_, remain); else ipc::log("fail: shm::handle for big message. msg_id: %zd, buf_id: %zd, size: %zd\n", msg.id_, buf_id, remain);
} }
// gc // gc
if (rc.size() > 1024) { if (rc.size() > 1024) {

4
src/libipc/waiter_helper.h
View File

@ -55,7 +55,7 @@ struct waiter_helper {
} }
else if (flags.need_dest_.exchange(false, std::memory_order_release)) { else if (flags.need_dest_.exchange(false, std::memory_order_release)) {
ret = false; ret = false;
ctrl.sema_wait(tm); ctrl.sema_wait(default_timeout);
break; break;
} }
else { else {
@ -106,6 +106,7 @@ struct waiter_helper {
template <typename Ctrl> template <typename Ctrl>
static bool quit_waiting(Ctrl & ctrl) { static bool quit_waiting(Ctrl & ctrl) {
auto & flags = ctrl.flags(); auto & flags = ctrl.flags();
flags.need_dest_.store(true, std::memory_order_relaxed);
if (!flags.is_waiting_.exchange(false, std::memory_order_release)) { if (!flags.is_waiting_.exchange(false, std::memory_order_release)) {
return true; return true;
} }
@ -116,7 +117,6 @@ struct waiter_helper {
bool ret = true; bool ret = true;
IPC_UNUSED_ auto guard = ctrl.get_lock(); IPC_UNUSED_ auto guard = ctrl.get_lock();
if (counter.counter_ > 0) { if (counter.counter_ > 0) {
flags.need_dest_.store(true, std::memory_order_relaxed);
ret = ctrl.sema_post(counter.counter_); ret = ctrl.sema_post(counter.counter_);
counter.counter_ -= 1; counter.counter_ -= 1;
ret = ret && ctrl.handshake_wait(default_timeout); ret = ret && ctrl.handshake_wait(default_timeout);