Invalidate proc-fd cache on close_range

1 files changed, 12 insertions, 0 deletions

diff --git a/internal/eventloop.go b/internal/eventloop.go
index 562a802..7d33f87 100644
--- a/internal/eventloop.go
+++ b/internal/eventloop.go
@@ -566,6 +566,7 @@ func (e *eventLoop) handleFdExit(ep *event.Pair, fdEv *FdEvent) bool {
 		retEv, ok := ep.ExitEv.(*types.RetEvent)
 		if ok && retEv.Ret == 0 {
 			e.fdState().closeRangeFrom(fd)
+			e.deleteProcFdCacheFrom(fd, fdEv.Pid)
 		}
 	}
 	ep.Comm = e.comm(fdEv.GetTid())
@@ -787,6 +788,17 @@ func (e *eventLoop) deleteProcFdCache(fd int32, pid uint32) {
 	delete(e.procFdCacheState(), procFdCacheKey(pid, fd))
 }
 
+func (e *eventLoop) deleteProcFdCacheFrom(first int32, pid uint32) {
+	cache := e.procFdCacheState()
+	for key := range cache {
+		cachePid := uint32(key >> 32)
+		cacheFd := int32(uint32(key))
+		if cachePid == pid && cacheFd >= first {
+			delete(cache, key)
+		}
+	}
+}
+
 func (e *eventLoop) procFdCacheState() map[uint64]file.FdFile {
 	if e.procFdCache == nil {
 		e.procFdCache = make(map[uint64]file.FdFile)