Files
unicorn/tests/regress/potential_memory_leak.py
Eli 9f578946d5 Revamp Python regression tests suite (#2022)
* Fix Python regression test suite (partial)

* Fix Python regression test suite

* Add a test for mapping at high addresses

* Add ctl tests
2024-10-13 13:14:10 +08:00

37 lines
1022 B
Python
Executable File

#!/usr/bin/env python
import platform
import resource
import regress
from unicorn import *
ITERATIONS = 10000
class MemoryLeak(regress.RegressTest):
def test(self):
if platform.system() == "Darwin":
rusage_multiplier = 1
elif platform.system() == "Linux":
rusage_multiplier = 1024
else:
# resource.getrusage(...) is platform dependent. Only tested under OS X and Linux.
self.skipTest('not OSx neither Linux')
max_rss_before = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss * rusage_multiplier
for _ in range(ITERATIONS):
mu = Uc(UC_ARCH_X86, UC_MODE_64)
mu.mem_map(0, 0x1000)
mu.emu_start(0, 0)
max_rss_after = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss * rusage_multiplier
rss_increase_per_iteration = (max_rss_after - max_rss_before) / ITERATIONS
self.assertLess(rss_increase_per_iteration, 8000.0)
if __name__ == '__main__':
regress.main()