add support for shm.h functions

Author: Mathieu Tarral

libvmi/libvmi.py

@@ -844,3 +844,22 @@ def slat_switch(self, slat_idx):
 
     def slat_change_gfn(self, slat_idx, old_gfn, new_gfn):
         status = lib.vmi_slat_change_gfn(self.vmi, slat_idx, old_gfn, new_gfn)
+
+    # shm
+    def shm_snapshot_create(self):
+        status = lib.vmi_shm_snapshot_create(self.vmi)
+        check(status)
+
+    def shm_snapshot_destroy(self):
+        status = lib.vmi_shm_snapshot_destroy(self.vmi)
+        check(status)
+
+    def get_dgpma(self, paddr, count):
+        buf_ptr = ffi.new("void **")
+        size = lib.vmi_get_dgpma(self.vmi, paddr, buf_ptr, count)
+        return buf_ptr, size
+
+    def get_dgvma(self, vaddr, pid, count):
+        buf_ptr = ffi.new("void **")
+        size = lib.vmi_get_dgvma(self.vmi, vaddr, pid, buf_ptr, count)
+        return buf_ptr, size

libvmi/shm_cdef.h

@@ -0,0 +1,18 @@
+status_t vmi_shm_snapshot_create(
+    vmi_instance_t vmi);
+
+status_t vmi_shm_snapshot_destroy(
+    vmi_instance_t vmi);
+
+size_t vmi_get_dgpma(
+    vmi_instance_t vmi,
+    addr_t paddr,
+    void **buf_ptr,
+    size_t count);
+
+size_t vmi_get_dgvma(
+    vmi_instance_t vmi,
+    addr_t vaddr,
+    pid_t pid,
+    void **buf_ptr,
+    size_t count);