test_dos.py

#!/usr/bin/env python3
# 
# Cross Platform and Multi Architecture Advanced Binary Emulation Framework
#

import sys, unittest
sys.path.append('..')

from qiling import Qiling
from qiling.const import QL_INTERCEPT, QL_VERBOSE

class Checklist:
    def __init__(self) -> None:
        self.visited_onenter = False
        self.visited_onexit = False

class DOSTest(unittest.TestCase):

    def test_dos_8086_hello(self):
        ql = Qiling(["../examples/rootfs/8086/dos/HI.DOS_COM"], "../examples/rootfs/8086/dos", verbose=QL_VERBOSE.DEBUG)
        ck = Checklist()

        def onenter(ql: Qiling):
            ck.visited_onenter = True

        def onexit(ql: Qiling):
            ck.visited_onexit = True

        ql.set_api((0x21, 0x09), onexit, QL_INTERCEPT.EXIT)
        ql.set_api((0x21, 0x4c), onenter, QL_INTERCEPT.ENTER)

        ql.run()

        self.assertTrue(ck.visited_onenter)
        self.assertTrue(ck.visited_onexit)

if __name__ == "__main__":
    unittest.main()