Various Win32 API additions/fixes (#616)

Various Win32 API additions/fixes

* add a get_size method to Miasm heap object, which allows the
implementation of mscvrt_realloc
* add the concept of "current directory", with the default value being
arbitrary set to "c:\tmp", which allows the implementation of
{Get,Set}CurrentDirecrtory
* various other methods implemented:

 - advapi32_RegCloseKey
 - advapi32_RegCreateKeyW
 - advapi32_RegSetValueExA
 - advapi32_RegSetValueExW
 - kernel32_GetProcessHeap
 - msvcrt_delete
 - msvcrt_fprintf
 - msvcrt_fwrite
 - msvcrt__mbscpy
 - msvcrt_new
 - msvcrt_realloc
 - msvcrt_sprintf
 - msvcrt_srand
 - msvcrt_strrchr
 - msvcrt_swprintf
 - msvcrt_wcscat
 - msvcrt_wcscmp
 - msvcrt_wcscpy
 - msvcrt__wcsicmp
 - msvcrt_wcslen
 - msvcrt_wcsncpy
 - msvcrt__wcsnicmp
 - msvcrt_wcsrchr

4 files changed, 166 insertions, 1 deletions

diff --git a/test/os_dep/common.py b/test/os_dep/common.py
new file mode 100755
index 00000000..5d525e32
--- /dev/null
+++ b/test/os_dep/common.py
@@ -0,0 +1,35 @@
+#! /usr/bin/env python2
+#-*- coding:utf-8 -*-
+
+import unittest
+import logging
+from miasm2.analysis.machine import Machine
+import miasm2.os_dep.common as commonapi
+from miasm2.core.utils import pck32
+from miasm2.jitter.csts import PAGE_READ, PAGE_WRITE
+
+machine = Machine("x86_32")
+
+jit = machine.jitter()
+jit.init_stack()
+
+class TestCommonAPI(unittest.TestCase):
+
+    def test_get_size(self):
+        heap = commonapi.heap()
+        with self.assertRaises(AssertionError):
+            heap.get_size(jit.vm, 0)
+        heap.alloc(jit, 20)
+        heap.alloc(jit, 40)
+        heap.alloc(jit, 50)
+        heap.alloc(jit, 60)
+        ptr = heap.alloc(jit, 10)
+        heap.alloc(jit, 80)
+        for i in xrange(10):
+            self.assertEqual(heap.get_size(jit.vm, ptr+i), 10)
+
+if __name__ == '__main__':
+    testsuite = unittest.TestLoader().loadTestsFromTestCase(TestCommonAPI)
+    report = unittest.TextTestRunner(verbosity=2).run(testsuite)
+    exit(len(report.errors + report.failures))
+
diff --git a/test/os_dep/linux/stdlib.py b/test/os_dep/linux/stdlib.py
new file mode 100755
index 00000000..ab39a487
--- /dev/null
+++ b/test/os_dep/linux/stdlib.py
@@ -0,0 +1,43 @@
+#! /usr/bin/env python2
+#-*- coding:utf-8 -*-
+
+import unittest
+import logging
+from miasm2.analysis.machine import Machine
+import miasm2.os_dep.linux_stdlib as stdlib
+from miasm2.core.utils import pck32
+from miasm2.jitter.csts import PAGE_READ, PAGE_WRITE
+
+machine = Machine("x86_32")
+
+jit = machine.jitter()
+jit.init_stack()
+
+heap = stdlib.linobjs.heap
+
+class TestLinuxStdlib(unittest.TestCase):
+
+    def test_xxx_sprintf(self):
+        def alloc_str(s):
+            s += "\x00"
+            ptr = heap.alloc(jit, len(s))
+            jit.vm.set_mem(ptr, s)
+            return ptr
+        fmt  = alloc_str("'%s' %d")
+        str_ = alloc_str("coucou")
+        buf = heap.alloc(jit,1024)
+
+        jit.push_uint32_t(1111)
+        jit.push_uint32_t(str_)
+        jit.push_uint32_t(fmt)
+        jit.push_uint32_t(buf)
+        jit.push_uint32_t(0) # ret_ad
+        stdlib.xxx_sprintf(jit)
+        ret = jit.get_str_ansi(buf)
+        self.assertEqual(ret, "'coucou' 1111")
+
+
+if __name__ == '__main__':
+    testsuite = unittest.TestLoader().loadTestsFromTestCase(TestLinuxStdlib)
+    report = unittest.TextTestRunner(verbosity=2).run(testsuite)
+    exit(len(report.errors + report.failures))
diff --git a/test/os_dep/win_api_x86_32.py b/test/os_dep/win_api_x86_32.py
index 2e22ccea..f080ba89 100755
--- a/test/os_dep/win_api_x86_32.py
+++ b/test/os_dep/win_api_x86_32.py
@@ -6,12 +6,14 @@ import logging
 from miasm2.analysis.machine import Machine
 import miasm2.os_dep.win_api_x86_32 as winapi
 from miasm2.core.utils import pck32
+from miasm2.jitter.csts import PAGE_READ, PAGE_WRITE
 
 machine = Machine("x86_32")
 
 jit = machine.jitter()
 jit.init_stack()
 
+heap = winapi.winobjs.heap
 
 class TestWinAPI(unittest.TestCase):
 
@@ -23,6 +25,89 @@ class TestWinAPI(unittest.TestCase):
         vBool = jit.cpu.EAX
         self.assertFalse(vBool)
 
+    def test_msvcrt_sprintf(self):
+        def alloc_str(s):
+            s += "\x00"
+            ptr = heap.alloc(jit, len(s))
+            jit.vm.set_mem(ptr, s)
+            return ptr
+        fmt  = alloc_str("'%s' %d")
+        str_ = alloc_str("coucou")
+        buf = heap.alloc(jit,1024)
+
+        jit.push_uint32_t(1111)
+        jit.push_uint32_t(str_)
+        jit.push_uint32_t(fmt)
+        jit.push_uint32_t(buf)
+        jit.push_uint32_t(0) # ret_ad
+        winapi.msvcrt_sprintf(jit)
+        ret = jit.get_str_ansi(buf)
+        self.assertEqual(ret, "'coucou' 1111")
+
+
+    def test_msvcrt_swprintf(self):
+        def alloc_str(s):
+            s = s.encode("utf-16le")
+            s += "\x00\x00"
+            ptr = heap.alloc(jit, len(s))
+            jit.vm.set_mem(ptr, s)
+            return ptr
+        fmt  = alloc_str("'%s' %d")
+        str_ = alloc_str("coucou")
+        buf = heap.alloc(jit,1024)
+
+        jit.push_uint32_t(1111)
+        jit.push_uint32_t(str_)
+        jit.push_uint32_t(fmt)
+        jit.push_uint32_t(buf)
+        jit.push_uint32_t(0) # ret_ad
+        winapi.msvcrt_swprintf(jit)
+        ret = jit.get_str_unic(buf)
+        self.assertEqual(ret, "'coucou' 1111")
+
+
+    def test_msvcrt_realloc(self):
+        jit.push_uint32_t(10)
+        jit.push_uint32_t(0) # ret_ad
+        winapi.msvcrt_malloc(jit)
+        ptr = jit.cpu.EAX
+
+        jit.push_uint32_t(20)
+        jit.push_uint32_t(ptr)
+        jit.push_uint32_t(0) # ret_ad
+        winapi.msvcrt_realloc(jit)
+        ptr2 = jit.cpu.EAX
+
+        self.assertNotEqual(ptr, ptr2)
+        self.assertEqual(heap.get_size(jit.vm,ptr2), 20)
+
+    def test_GetCurrentDirectory(self):
+
+        # DWORD WINAPI GetCurrentDirectory(size, buf)
+
+        # Test with a buffer long enough
+        addr = 0x80000
+        size = len(winapi.winobjs.cur_dir)+1
+        jit.vm.add_memory_page(addr, PAGE_READ | PAGE_WRITE, "\x00" * (size), "")
+        jit.push_uint32_t(addr)   # buf
+        jit.push_uint32_t(size)   # size
+        jit.push_uint32_t(0)      # @return
+        winapi.kernel32_GetCurrentDirectoryA(jit)
+        dir_ = jit.get_str_ansi(addr)
+        size_ret = jit.cpu.EAX
+        self.assertEqual(len(dir_), size_ret)
+
+        # Test with a buffer too small
+        jit.vm.set_mem(addr, "\xFF"*size)
+        jit.push_uint32_t(addr)   # buf
+        jit.push_uint32_t(5)      # size
+        jit.push_uint32_t(0)      # @return
+        winapi.kernel32_GetCurrentDirectoryA(jit)
+        size_ret = jit.cpu.EAX
+        self.assertEqual(len(dir_)+1, size_ret)
+        dir_short = jit.get_str_ansi(addr)
+        self.assertEqual(dir_short, dir_[:4])
+
     def test_MemoryManagementFunctions(self):
 
         # HGLOBAL WINAPI GlobalAlloc(_In_ UINT uFlags, _In_ SIZE_T dwBytes);
diff --git a/test/test_all.py b/test/test_all.py
index d2ae4fce..23937366 100755
--- a/test/test_all.py
+++ b/test/test_all.py
@@ -265,7 +265,9 @@ testset += RegressionTest(["z3_ir.py"], base_dir="ir/translators",
 testset += RegressionTest(["smt2.py"], base_dir="ir/translators",
                           tags=[TAGS["z3"]])
 ## OS_DEP
-for script in ["win_api_x86_32.py",
+for script in ["common.py",
+               "win_api_x86_32.py",
+               os.path.join("linux", "stdlib.py"),
                ]:
     testset += RegressionTest([script], base_dir="os_dep", tags=[TAGS['gcc']])