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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
import pychrysalide
from chrysacase import ChrysalideTestCase
from pychrysalide import MemoryDataSize
from pychrysalide.arch.operands import ImmediateOperand
from pychrysalide.glibext import StringBuilder
class TestImmediate(ChrysalideTestCase):
"""TestCase for arch.ImmediateOperand."""
def testBasicImmediate(self):
"""Check basic properties of immediate values."""
imm = ImmediateOperand(MemoryDataSize._32_BITS_UNSIGNED, 0x123)
self.assertEqual(imm.size, MemoryDataSize._32_BITS_UNSIGNED)
self.assertEqual(imm.value, 0x123)
self.assertFalse(imm.is_negative)
def testStringBuilderForImmediatesOverriding(self):
"""Override the StringBuilder interface for immediate values."""
class MyImmOperand(ImmediateOperand, StringBuilder):
def __init__(self):
super().__init__(MemoryDataSize._32_BITS_UNSIGNED, 0x123)
def _to_string(self, flags=0):
return 'NaN'
op = MyImmOperand()
self.assertEqual(op.to_string(), 'NaN')
self.assertEqual(str(op), 'NaN')
self.assertEqual(f'{op}', 'NaN')
def validateValue(self, value, size, padding, strings):
"""Check all kinds of things with a given immediate operand."""
for d in strings.keys():
op = ImmediateOperand(size, value)
self.assertTrue(op.size == size)
self.assertTrue(op.value == value)
if padding:
op.set_flag(ImmediateOperand.ImmOperandFlag.ZERO_PADDING)
else:
op.unset_flag(ImmediateOperand.ImmOperandFlag.ZERO_PADDING_BY_DEFAULT)
op.display = d
string = op.to_string()
self.assertEqual(string, strings[d])
def testByteOne(self):
"""Run sanity checks on immediate operand with value 1."""
strings = {
ImmediateOperand.ImmOperandDisplay.BIN: 'b1',
ImmediateOperand.ImmOperandDisplay.OCT: '01',
ImmediateOperand.ImmOperandDisplay.DEC: '1',
ImmediateOperand.ImmOperandDisplay.HEX: '0x1'
}
self.validateValue(1, pychrysalide.MemoryDataSize._8_BITS_UNSIGNED, False, strings)
def testByteOnePadded(self):
"""Run sanity checks on immediate operand with padded value 1."""
strings = {
ImmediateOperand.ImmOperandDisplay.BIN: 'b00000001',
ImmediateOperand.ImmOperandDisplay.OCT: '01',
ImmediateOperand.ImmOperandDisplay.DEC: '1',
ImmediateOperand.ImmOperandDisplay.HEX: '0x01'
}
self.validateValue(1, pychrysalide.MemoryDataSize._8_BITS_UNSIGNED, True, strings)
|
