blob: 3cb70e2aeef67591f2959077f537bace85e962b0 (
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
39
40
41
42
43
44
45
46
47
48
49
50
|
#! /usr/bin/env python2
import sys
from asm_test import Asm_Test_32
class Test_SCAS(Asm_Test_32):
MYSTRING = "test string"
TXT = '''
main:
LEA EDI, DWORD PTR [mystr]
XOR ECX, ECX
DEC ECX
REPNE SCASB
NOT ECX
DEC ECX
RET
mystr:
.string "%s"
''' % MYSTRING
def check(self):
assert(self.myjit.cpu.ECX == len(self.MYSTRING))
assert(self.myjit.cpu.EDI == self.myjit.ir_arch.symbol_pool.getby_name('mystr').offset + len(self.MYSTRING)+1)
class Test_MOVS(Asm_Test_32):
MYSTRING = "test string"
TXT = '''
main:
LEA ESI, DWORD PTR [mystr]
LEA EDI, DWORD PTR [buffer]
MOV ECX, %d
REPE MOVSB
RET
mystr:
.string "%s"
buffer:
.string "%s"
''' % (len(MYSTRING), MYSTRING, " "*len(MYSTRING))
def check(self):
assert(self.myjit.cpu.ECX == 0)
assert(self.myjit.cpu.EDI == self.myjit.ir_arch.symbol_pool.getby_name('buffer').offset + len(self.MYSTRING))
assert(self.myjit.cpu.ESI == self.myjit.ir_arch.symbol_pool.getby_name('mystr').offset + len(self.MYSTRING))
if __name__ == "__main__":
[test(*sys.argv[1:])() for test in [Test_SCAS, Test_MOVS]]
|