mirror of
https://github.com/cea-sec/miasm
synced 2026-06-21 13:48:18 +00:00
Fabrice Desclaux
70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
#!/usr/bin/env python
|
|
#-*- coding:utf-8 -*-
|
|
|
|
import unittest
|
|
|
|
|
|
class TestIrIr2C(unittest.TestCase):
|
|
|
|
def translationTest(self, expr, expected):
|
|
from miasm2.ir.translators.C import TranslatorC
|
|
|
|
self.assertEqual(TranslatorC.from_expr(expr), expected)
|
|
|
|
def test_ExprOp_toC(self):
|
|
from miasm2.expression.expression import ExprInt32, ExprOp
|
|
from miasm2.ir.translators.C import TranslatorC
|
|
|
|
args = [ExprInt32(i) for i in xrange(9)]
|
|
|
|
|
|
# Unary operators
|
|
self.translationTest(
|
|
ExprOp('parity', *args[:1]), r'parity(0x0&0xffffffff)')
|
|
self.translationTest(
|
|
ExprOp('!', *args[:1]), r'(~ 0x0)&0xffffffff')
|
|
self.translationTest(
|
|
ExprOp('hex2bcd', *args[:1]), r'hex2bcd_32(0x0)')
|
|
self.translationTest(ExprOp('fabs', *args[:1]), r'fabs(0x0)')
|
|
self.assertRaises(NotImplementedError, TranslatorC.from_expr,
|
|
ExprOp('X', *args[:1]))
|
|
|
|
# Binary operators
|
|
self.translationTest(
|
|
ExprOp('==', *args[:2]), r'(((0x0&0xffffffff) == (0x1&0xffffffff))?1:0)')
|
|
self.translationTest(
|
|
ExprOp('%', *args[:2]), r'(((0x0&0xffffffff)%(0x1&0xffffffff))&0xffffffff)')
|
|
self.translationTest(
|
|
ExprOp('-', *args[:2]), r'(((0x0&0xffffffff) - (0x1&0xffffffff))&0xffffffff)')
|
|
self.translationTest(
|
|
ExprOp('bsr', *args[:1]), r'x86_bsr(0x0, 0x20)')
|
|
self.translationTest(
|
|
ExprOp('cpuid0', *args[:2]), r'cpuid0(0x0, 0x1)')
|
|
self.translationTest(
|
|
ExprOp('fcom0', *args[:2]), r'fcom0(0x0, 0x1)')
|
|
self.translationTest(
|
|
ExprOp('fadd', *args[:2]), r'fadd(0x0, 0x1)')
|
|
self.translationTest(
|
|
ExprOp('segm', *args[:2]), r'segm2addr(vmcpu, 0x0, 0x1)')
|
|
self.translationTest(
|
|
ExprOp('imod', *args[:2]), r'imod32(vmcpu, 0x0, 0x1)')
|
|
self.translationTest(
|
|
ExprOp('bcdadd', *args[:2]), r'bcdadd_32(0x0, 0x1)')
|
|
self.assertRaises(NotImplementedError, TranslatorC.from_expr,
|
|
ExprOp('X', *args[:2]))
|
|
|
|
# Ternary operators
|
|
self.translationTest(
|
|
ExprOp('div8', *args[:3]), r'(div_op(32, 0x0, 0x1, 0x2) &0xffffffff)')
|
|
|
|
# Other cases
|
|
self.translationTest(
|
|
ExprOp('+', *args[:3]), r'(((0x0&0xffffffff)+(0x1&0xffffffff)+(0x2&0xffffffff))&0xffffffff)')
|
|
self.assertRaises(NotImplementedError, TranslatorC.from_expr,
|
|
ExprOp('X', *args[:3]))
|
|
|
|
if __name__ == '__main__':
|
|
testsuite = unittest.TestLoader().loadTestsFromTestCase(TestIrIr2C)
|
|
report = unittest.TextTestRunner(verbosity=2).run(testsuite)
|
|
exit(len(report.errors + report.failures))
|