blob: ca7b8bc14d6cfcfdd14c68aa6ce54e43931220ac (
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
37
38
|
#! /usr/bin/env python2
#-*- coding:utf-8 -*-
from builtins import range
import unittest
import logging
from miasm.analysis.machine import Machine
import miasm.os_dep.common as commonapi
from miasm.core.utils import pck32
from miasm.jitter.csts import PAGE_READ, PAGE_WRITE
from miasm.core.locationdb import LocationDB
machine = Machine("x86_32")
loc_db = LocationDB()
jit = machine.jitter(loc_db)
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))
|