wrap: Zero more buffers

Zero more buffers so its easier to figure out what needs to be copied
and modified in the example driver to replay the dumped command buffer.

1 files changed, 5 insertions, 2 deletions

diff --git a/src/wrap.c b/src/wrap.c
index 6b96f36..01ae2a3 100644
--- a/src/wrap.c
+++ b/src/wrap.c
@@ -212,9 +212,12 @@ static void clear_mem(struct mem_entry *mem) {
                       || heap == PVR_HEAP_GENERAL
                       //|| heap == PVR_HEAP_KERNEL_DATA
                       || heap == PVR_HEAP_PERCONTEXT_3DPARAMETERS
-                      || heap == PVR_HEAP_SYNC_INFO;
+                      || heap == PVR_HEAP_SYNC_INFO
+                      || heap == PVR_HEAP_TA_DATA;
     IMG_HANDLE handle = mem->mem_info.hKernelMemInfo;
-    bool is_special_heap = (uintptr_t)handle == 0x14;
+    bool is_special_heap = (uintptr_t)handle == 0x14 ||
+                           (uintptr_t)handle == 0x15 ||
+                           (uintptr_t)handle == 0x10;
     if(mem->type == MEM_TYPE_NORMAL && (valid_heap || is_special_heap)) {
         printf("Clearing heap\n");
         for (int i = 0; i < 2; i++)