pan/decode: Track mmaps with a red-black tree

Rather than emulating page tables, poorly, with a hash table, use a
red-black tree and store the intervals directly. This is deterministic
instead of probabilistic, attaining O(log n) performance for n mapped
intervals which is good enough. Unlike the hash table approach, this
allows us to iterate intervals easily.

Signed-off-by: Alyssa Rosenzweig <alyssa@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/14543>

2 files changed, 40 insertions, 13 deletions

diff --git a/src/panfrost/lib/genxml/decode.h b/src/panfrost/lib/genxml/decode.h
index 5fde024e9d8..998f2688abd 100644
--- a/src/panfrost/lib/genxml/decode.h
+++ b/src/panfrost/lib/genxml/decode.h
@@ -27,6 +27,7 @@
 #define __PAN_DECODE_H__
 
 #include "genxml/gen_macros.h"
+#include "util/rb_tree.h"
 
 #include "wrap.h"
 
@@ -35,6 +36,7 @@ extern FILE *pandecode_dump_stream;
 void pandecode_dump_file_open(void);
 
 struct pandecode_mapped_memory {
+        struct rb_node node;
         size_t length;
         void *addr;
         uint64_t gpu_va;
diff --git a/src/panfrost/lib/genxml/decode_common.c b/src/panfrost/lib/genxml/decode_common.c
index 9fbb892e960..ca5384e4e63 100644
--- a/src/panfrost/lib/genxml/decode_common.c
+++ b/src/panfrost/lib/genxml/decode_common.c
@@ -34,20 +34,47 @@
 #include "util/macros.h"
 #include "util/u_debug.h"
 #include "util/u_dynarray.h"
-#include "util/hash_table.h"
 
 FILE *pandecode_dump_stream;
 
 /* Memory handling */
 
-static struct hash_table_u64 *mmap_table;
+static struct rb_tree mmap_tree;
 
 static struct util_dynarray ro_mappings;
 
+#define to_mapped_memory(x) \
+	rb_node_data(struct pandecode_mapped_memory, x, node)
+
+/*
+ * Compare a GPU VA to a node, considering a GPU VA to be equal to a node if it
+ * is contained in the interval the node represents. This lets us store
+ * intervals in our tree.
+ */
+static int
+pandecode_cmp_key(const struct rb_node *lhs, const void *key)
+{
+        struct pandecode_mapped_memory *mem = to_mapped_memory(lhs);
+        uint64_t *gpu_va = (uint64_t *) key;
+
+        if (mem->gpu_va <= *gpu_va && *gpu_va < (mem->gpu_va + mem->length))
+                return 0;
+        else
+                return mem->gpu_va - *gpu_va;
+}
+
+static int
+pandecode_cmp(const struct rb_node *lhs, const struct rb_node *rhs)
+{
+        return to_mapped_memory(lhs)->gpu_va - to_mapped_memory(rhs)->gpu_va;
+}
+
 static struct pandecode_mapped_memory *
 pandecode_find_mapped_gpu_mem_containing_rw(uint64_t addr)
 {
-        return _mesa_hash_table_u64_search(mmap_table, addr & ~(4096 - 1));
+        struct rb_node *node = rb_tree_search(&mmap_tree, &addr, pandecode_cmp_key);
+
+        return to_mapped_memory(node);
 }
 
 struct pandecode_mapped_memory *
@@ -112,11 +139,8 @@ pandecode_inject_mmap(uint64_t gpu_va, void *cpu, unsigned sz, const char *name)
         mapped_mem->addr = cpu;
         pandecode_add_name(mapped_mem, gpu_va, name);
 
-        /* Add it to the table */
-        assert((gpu_va & 4095) == 0);
-
-        for (unsigned i = 0; i < sz; i += 4096)
-                _mesa_hash_table_u64_insert(mmap_table, gpu_va + i, mapped_mem);
+        /* Add it to the tree */
+        rb_tree_insert(&mmap_tree, &mapped_mem->node, pandecode_cmp);
 }
 
 void
@@ -131,10 +155,8 @@ pandecode_inject_free(uint64_t gpu_va, unsigned sz)
         assert(mem->gpu_va == gpu_va);
         assert(mem->length == sz);
 
+        rb_tree_remove(&mmap_tree, &mem->node);
         free(mem);
-
-        for (unsigned i = 0; i < sz; i += 4096)
-                _mesa_hash_table_u64_remove(mmap_table, gpu_va + i);
 }
 
 char *
@@ -202,7 +224,7 @@ void
 pandecode_initialize(bool to_stderr)
 {
         force_stderr = to_stderr;
-        mmap_table = _mesa_hash_table_u64_create(NULL);
+        rb_tree_init(&mmap_tree);
         util_dynarray_init(&ro_mappings, NULL);
 }
 
@@ -216,7 +238,10 @@ pandecode_next_frame(void)
 void
 pandecode_close(void)
 {
-        _mesa_hash_table_u64_destroy(mmap_table);
+        rb_tree_foreach_safe(struct pandecode_mapped_memory, it, &mmap_tree, node) {
+                free(it);
+        }
+
         util_dynarray_fini(&ro_mappings);
         pandecode_dump_file_close();
 }