guest-diag: cdt2 payload scan limited to anon mappings (device-mapping reads can block)

1 files changed, 5 insertions, 4 deletions

diff --git a/tools/test/cuda_drv_test2.c b/tools/test/cuda_drv_test2.c
index d4f848a..98a996e 100644
--- a/tools/test/cuda_drv_test2.c
+++ b/tools/test/cuda_drv_test2.c
@@ -88,11 +88,12 @@ static void scan_self_for_payload(const char *tag)
         if (perms[0] != 'r' || b <= a || b - a > (512ULL << 20))
             continue;
         /*
-         * Only scan anonymous and device mappings: reading a truncated
-         * file-backed mapping past EOF raises SIGBUS and killed the client
-         * silently before cuMemcpyDtoH.
+         * Only scan anonymous mappings: reading file-backed or device
+         * mappings can SIGBUS (past EOF) or block in a device fault path,
+         * either of which kills/hangs the client before cuMemcpyDtoH. The
+         * device mapping list above still reveals any BAR window.
          */
-        if (path[0] && path[0] != '[' && strncmp(path, "/dev/", 5))
+        if (path[0] && path[0] != '[')
             continue;
         {
             volatile unsigned long long *p;