Convert to src-layout ta/nix

1 files changed, 169 insertions, 0 deletions

diff --git a/src/miasm/os_dep/common.py b/src/miasm/os_dep/common.py
new file mode 100644
index 00000000..74100817
--- /dev/null
+++ b/src/miasm/os_dep/common.py
@@ -0,0 +1,169 @@
+import os
+
+from future.utils import viewitems
+
+from miasm.core.utils import force_bytes, force_str
+from miasm.jitter.csts import PAGE_READ, PAGE_WRITE
+from miasm.core.utils import get_caller_name
+from miasm.core.utils import pck64, upck64
+
+BASE_SB_PATH = "file_sb"
+WIN_CODEPAGE = "cp1252"
+
+def get_win_str_a(jitter, ad_str, max_char=None):
+    l = 0
+    tmp = ad_str
+    while ((max_char is None or l < max_char) and
+           jitter.vm.get_mem(tmp, 1) != b"\x00"):
+        tmp += 1
+        l += 1
+    data = jitter.vm.get_mem(ad_str, l)
+    ret = data.decode(WIN_CODEPAGE)
+    return ret
+
+
+def get_win_str_w(jitter, ad_str, max_char=None):
+    l = 0
+    tmp = ad_str
+    while ((max_char is None or l < max_char) and
+           jitter.vm.get_mem(tmp, 2) != b"\x00\x00"):
+        tmp += 2
+        l += 2
+    s = jitter.vm.get_mem(ad_str, l)
+    s = s.decode("utf-16le")
+    return s
+
+def encode_win_str_a(value):
+    value = value.encode(WIN_CODEPAGE) + b"\x00"
+    return value
+
+def encode_win_str_w(value):
+    value = value.encode("utf-16le") + b'\x00' * 2
+    return value
+
+
+def set_win_str_a(jitter, addr, value):
+    value = encode_win_str_a(value)
+    jitter.vm.set_mem(addr, value)
+
+
+def set_win_str_w(jitter, addr, value):
+    value = encode_win_str_w(value)
+    jitter.vm.set_mem(addr, value)
+
+
+class heap(object):
+
+    "Light heap simulation"
+
+    addr = 0x20000000
+    align = 0x1000
+    size = 32
+    mask = (1 << size) - 1
+
+    def next_addr(self, size):
+        """
+        @size: the size to allocate
+        return the future checnk address
+        """
+        ret = self.addr
+        self.addr = (self.addr + size + self.align - 1)
+        self.addr &= self.mask ^ (self.align - 1)
+        return ret
+
+    def alloc(self, jitter, size, perm=PAGE_READ | PAGE_WRITE, cmt=""):
+        """
+        @jitter: a jitter instance
+        @size: the size to allocate
+        @perm: permission flags (see vm_alloc doc)
+        """
+        return self.vm_alloc(jitter.vm, size, perm=perm, cmt=cmt)
+
+    def vm_alloc(self, vm, size, perm=PAGE_READ | PAGE_WRITE, cmt=""):
+        """
+        @vm: a VmMngr instance
+        @size: the size to allocate
+        @perm: permission flags (PAGE_READ, PAGE_WRITE, PAGE_EXEC or any `|`
+            combination of them); default is PAGE_READ|PAGE_WRITE
+        """
+        addr = self.next_addr(size)
+        vm.add_memory_page(
+            addr,
+            perm,
+            b"\x00" * (size),
+            "Heap alloc by %s %s" % (get_caller_name(2), cmt)
+        )
+        return addr
+
+    def get_size(self, vm, ptr):
+        """
+        @vm: a VmMngr instance
+        @size: ptr to get the size of the associated allocation.
+
+        `ptr` can be the base address of a previous allocation, or an address
+        within the allocated range. The size of the whole allocation is always
+        returned, regardless ptr is the base address or not.
+        """
+        assert vm.is_mapped(ptr, 1)
+        data = vm.get_all_memory()
+        ptr_page = data.get(ptr, None)
+        if ptr_page is None:
+            for address, page_info in viewitems(data):
+                if address <= ptr < address + page_info["size"]:
+                    ptr_page = page_info
+                    break
+            else:
+                raise RuntimeError("Must never happen (unmapped but mark as mapped by API)")
+        return ptr_page["size"]
+
+
+def windows_to_sbpath(path):
+    """Convert a Windows path to a valid filename within the sandbox
+    base directory.
+
+    """
+    path = [elt for elt in path.lower().replace('/', '_').split('\\') if elt]
+    return os.path.join(BASE_SB_PATH, *path)
+
+
+def unix_to_sbpath(path):
+    """Convert a POSIX path to a valid filename within the sandbox
+    base directory.
+
+    """
+    path = [elt for elt in path.split('/') if elt]
+    return os.path.join(BASE_SB_PATH, *path)
+
+def get_fmt_args(fmt, cur_arg, get_str, get_arg_n):
+    idx = 0
+    fmt = get_str(fmt)
+    chars_format = '%cdfsuxX'
+    char_percent = '%'
+    char_string = 's'
+    output = ""
+
+    while True:
+        if idx == len(fmt):
+            break
+        char = fmt[idx:idx+1]
+        idx += 1
+        if char == char_percent:
+            token = char_percent
+            while True:
+                char = fmt[idx:idx+1]
+                idx += 1
+                token += char
+                if char in chars_format:
+                    break
+            if char == char_percent:
+                output += char
+                continue
+            if token.endswith(char_string):
+                addr = get_arg_n(cur_arg)
+                arg = get_str(addr)
+            else:
+                arg = get_arg_n(cur_arg)
+            char = token % arg
+            cur_arg += 1
+        output += char
+    return output