uc_mem_write_virtual

need tests and rebase

Author: PhilippTakacs

include/unicorn/unicorn.h

@@ -954,6 +954,10 @@ UNICORN_EXPORT
 uc_err uc_mem_read_virtual(uc_engine *uc, uint64_t address, uint32_t prot,
                            void *bytes, size_t size);
 
+UNICORN_EXPORT
+uc_err uc_mem_write_virtual(uc_engine *uc, uint64_t address, uint32_t prot,
+                           void *bytes, size_t size);
+
 /*
  Translate a virtuall address to a physical address
 

uc.c

@@ -852,6 +852,57 @@ uc_err uc_mem_read_virtual(uc_engine *uc, uint64_t address, uc_prot prot,
     return UC_ERR_OK;
 }
 
+UNICORN_EXPORT
+uc_err uc_mem_write_virtual(uc_engine *uc, uint64_t address, uc_prot prot,
+                           void *_bytes, size_t size)
+{
+    size_t count = 0, len;
+    uint8_t *bytes = _bytes;
+    uint64_t align;
+    uint64_t pagesize;
+    uint64_t paddr = 0;
+
+    UC_INIT(uc);
+
+    // qemu cpu_physical_memory_rw() size is an int
+    if (size > INT_MAX) {
+        restore_jit_state(uc);
+        return UC_ERR_ARG;
+    }
+
+    // The sparc mmu doesn't support probe mode
+    if (uc->arch == UC_ARCH_SPARC && uc->cpu->cc->tlb_fill == uc->cpu->cc->tlb_fill_cpu) {
+        restore_jit_state(uc);
+        return UC_ERR_ARG;
+    }
+
+    if (!(UC_PROT_READ == prot || UC_PROT_WRITE == prot ||
+          UC_PROT_EXEC == prot)) {
+        restore_jit_state(uc);
+        return UC_ERR_ARG;
+    }
+
+    while (count < size) {
+        align = uc->target_page_align;
+        pagesize = uc->target_page_size;
+        len = MIN(size - count, (address & ~align) + pagesize - address);
+	if (!uc_virtual_to_physical(uc, address, prot, &paddr)) {
+            restore_jit_state(uc);
+            return UC_ERR_WRITE_PROT;
+	}
+        if (!uc_mem_write(uc, paddr, bytes, len)) {
+            restore_jit_state(uc);
+            return UC_ERR_WRITE_PROT;
+        }
+        bytes += len;
+        address += len;
+        count += len;
+    }
+    assert(count == size);
+    restore_jit_state(uc);
+    return UC_ERR_OK;
+}
+
 UNICORN_EXPORT
 uc_err uc_mem_read(uc_engine *uc, uint64_t address, void *_bytes, size_t size)
 {