From 90f12ce74021a3a0b7291c2d77f9d5b214e764a2 Mon Sep 17 00:00:00 2001
From: Jonathan Marek <jonathan@marek.ca>
Date: Fri, 15 Sep 2023 09:18:22 -0400
Subject: [PATCH] tu: add a TU_DEBUG=rd option for cmdstream dumping

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/25246>
---
 src/freedreno/vulkan/tu_knl_drm_msm.cc | 36 ++++++++++++++++++++++++++
 src/freedreno/vulkan/tu_util.cc        |  1 +
 src/freedreno/vulkan/tu_util.h         |  1 +
 3 files changed, 38 insertions(+)

diff --git a/src/freedreno/vulkan/tu_knl_drm_msm.cc b/src/freedreno/vulkan/tu_knl_drm_msm.cc
index 6712d620cc2..720de8295fa 100644
--- a/src/freedreno/vulkan/tu_knl_drm_msm.cc
+++ b/src/freedreno/vulkan/tu_knl_drm_msm.cc
@@ -23,6 +23,7 @@
 #include "tu_device.h"
 #include "tu_dynamic_rendering.h"
 #include "tu_knl_drm.h"
+#include "redump.h"
 
 struct tu_queue_submit
 {
@@ -870,6 +871,41 @@ tu_queue_submit_locked(struct tu_queue *queue, struct tu_queue_submit *submit)
       .syncobj_stride = sizeof(struct drm_msm_gem_submit_syncobj),
    };
 
+   if (TU_DEBUG(RD)) {
+      struct tu_device *device = queue->device;
+      static uint32_t submit_idx;
+      char path[32];
+      sprintf(path, "%.5d.rd", p_atomic_inc_return(&submit_idx));
+      int rd = open(path, O_CLOEXEC | O_WRONLY | O_CREAT | O_TRUNC, 0777);
+      if (rd >= 0) {
+         rd_write_section(rd, RD_CHIP_ID, &device->physical_device->dev_id.chip_id, 4);
+
+         rd_write_section(rd, RD_CMD, "tu-dump", 8);
+
+         for (unsigned i = 0; i < device->bo_count; i++) {
+            struct drm_msm_gem_submit_bo bo = device->bo_list[i];
+            struct tu_bo *tu_bo = tu_device_lookup_bo(device, bo.handle);
+            uint64_t iova = bo.presumed;
+
+            uint32_t buf[3] = { iova, tu_bo->size, iova >> 32 };
+            rd_write_section(rd, RD_GPUADDR, buf, 12);
+            if (bo.flags & MSM_SUBMIT_BO_DUMP) {
+               msm_bo_map(device, tu_bo); /* note: this would need locking to be safe */
+               rd_write_section(rd, RD_BUFFER_CONTENTS, tu_bo->map, tu_bo->size);
+            }
+         }
+
+         for (unsigned i = 0; i < req.nr_cmds; i++) {
+            struct drm_msm_gem_submit_cmd *cmd = &submit->cmds[i];
+            uint64_t iova = device->bo_list[cmd->submit_idx].presumed + cmd->submit_offset;
+            uint32_t size = cmd->size >> 2;
+            uint32_t buf[3] = { iova, size, iova >> 32 };
+            rd_write_section(rd, RD_CMDSTREAM_ADDR, buf, 12);
+         }
+         close(rd);
+      }
+   }
+
    int ret = drmCommandWriteRead(queue->device->fd,
                                  DRM_MSM_GEM_SUBMIT,
                                  &req, sizeof(req));
diff --git a/src/freedreno/vulkan/tu_util.cc b/src/freedreno/vulkan/tu_util.cc
index c7d7f118e4f..08443fa87cf 100644
--- a/src/freedreno/vulkan/tu_util.cc
+++ b/src/freedreno/vulkan/tu_util.cc
@@ -39,6 +39,7 @@ static const struct debug_control tu_debug_options[] = {
    { "3d_load", TU_DEBUG_3D_LOAD },
    { "fdm", TU_DEBUG_FDM },
    { "noconform", TU_DEBUG_NOCONFORM },
+   { "rd", TU_DEBUG_RD },
    { NULL, 0 }
 };
 
diff --git a/src/freedreno/vulkan/tu_util.h b/src/freedreno/vulkan/tu_util.h
index 1af4b618a6b..1a62757ea36 100644
--- a/src/freedreno/vulkan/tu_util.h
+++ b/src/freedreno/vulkan/tu_util.h
@@ -47,6 +47,7 @@ enum tu_debug_flags
    TU_DEBUG_3D_LOAD = 1 << 22,
    TU_DEBUG_FDM = 1 << 23,
    TU_DEBUG_NOCONFORM = 1 << 24,
+   TU_DEBUG_RD = 1 << 25,
 };
 
 struct tu_env {