blob: 525120753cc0d7a521c4a2efae8f1ea70bb04954 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
#! /usr/bin/env python2
#-*- coding:utf-8 -*-
from builtins import range
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 range(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))
|