_damo_records: Poll target processes if exist

If 'damo record' is started with target command, the target process is
the child of 'damo', so becomes zombie process when it is finished.
Then because zombie process is still a valid process from DAMON's
perspective, kdamond is not terminated, and 'damo' hangs.  Do the pid
polling if any target process exists.

Signed-off-by: SeongJae Park <sj@kernel.org>

1 files changed, 10 insertions, 2 deletions

diff --git a/_damo_records.py b/_damo_records.py
index 3c0faa85..d8641279 100644
--- a/_damo_records.py
+++ b/_damo_records.py
@@ -673,8 +673,16 @@ def __poll_target_pids(handle):
     return True
 
 def poll_target_pids(handle):
-    if (not handle.poll_add_child_tasks and
-        handle.mem_footprint_snapshots is None):
+    has_pid_target = False
+    if handle.kdamonds:
+        for kdamond in handle.kdamonds:
+            for ctx in kdamond.contexts:
+                if _damon.target_has_pid(ctx.ops):
+                    has_pid_target = True
+                    break
+            if has_pid_target:
+                break
+    if has_pid_target is False:
         return False
     rc = __poll_target_pids(handle)
     if rc is True and handle.mem_footprint_snapshots is not None: