You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
978 B
34 lines
978 B
3 years ago
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
# This is to test NASM directives
|
||
|
|
|
||
|
|
# Github issue: #xxx
|
||
|
|
# Author: Nguyen Anh Quynh
|
||
|
|
|
||
|
|
from keystone import *
|
||
|
|
|
||
|
|
import regress
|
||
|
|
|
||
|
|
class TestX86(regress.RegressTest):
|
||
|
|
def runTest(self):
|
||
|
|
# Initialize Keystone engine
|
||
|
|
ks = Ks(KS_ARCH_X86, KS_MODE_32)
|
||
|
|
# change the syntax to NASM
|
||
|
|
ks.syntax = KS_OPT_SYNTAX_NASM
|
||
|
|
|
||
|
|
# compile an instruction in NASM syntax
|
||
|
|
encoding, count = ks.asm(b"db 12, 13\ndw 14, 15")
|
||
|
|
# Assert the result
|
||
|
|
self.assertEqual(encoding, [ 0x0c, 0x0d, 0x0e, 0x00, 0x0f, 0x00 ])
|
||
|
|
|
||
|
|
encoding, count = ks.asm(b"bits 32\n add eax, ebx\nbits 16\nadd eax, ebx")
|
||
|
|
# Assert the result
|
||
|
|
self.assertEqual(encoding, [ 0x01, 0xd8, 0x66, 0x01, 0xd8 ])
|
||
|
|
|
||
|
|
encoding, count = ks.asm(b"[Bits 32]\n add eax, ebx\n[bits 16]\nadd eax, ebx")
|
||
|
|
# Assert the result
|
||
|
|
self.assertEqual(encoding, [ 0x01, 0xd8, 0x66, 0x01, 0xd8 ])
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
regress.main()
|