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.
65 lines
2.2 KiB
65 lines
2.2 KiB
3 years ago
|
#!/usr/bin/python
|
||
|
|
# Ingmar Steen, 2016
|
||
|
|
|
||
|
|
# This is to test RIP relative and absolute addressing using all
|
||
|
|
# available syntaxes.
|
||
|
|
|
||
|
|
# Github issue: #32
|
||
|
|
# Author: Ingmar Steen
|
||
|
|
|
||
|
|
from keystone import *
|
||
|
|
|
||
|
|
import regress
|
||
|
|
|
||
|
|
|
||
|
|
class TestX64NasmLabelRel(regress.RegressTest):
|
||
|
|
def runTest(self):
|
||
|
|
# Initialize Keystone engine
|
||
|
|
ks = Ks(KS_ARCH_X86, KS_MODE_64)
|
||
|
|
# change the syntax to NASM
|
||
|
|
ks.syntax = KS_OPT_SYNTAX_NASM
|
||
|
|
|
||
|
|
# nasm uses rel for rip relative addressing
|
||
|
|
encoding, count = ks.asm(b"lea rax, [rel __data]\n__data:")
|
||
|
|
self.assertEqual(encoding, [ 0x48, 0x8d, 0x05, 0x00, 0x00, 0x00, 0x00 ])
|
||
|
|
|
||
|
|
# verify that rip relative addressing is indeed rip relative
|
||
|
|
encoding, count = ks.asm(b"nop\nnop\nlea rax, [rel __data]\n__data:")
|
||
|
|
self.assertEqual(encoding, [ 0x90, 0x90, 0x48, 0x8d, 0x05, 0x00, 0x00, 0x00, 0x00 ])
|
||
|
|
|
||
|
|
|
||
|
|
class TestX64AttLabelRel(regress.RegressTest):
|
||
|
|
def runTest(self):
|
||
|
|
# Initialize Keystone engine
|
||
|
|
ks = Ks(KS_ARCH_X86, KS_MODE_64)
|
||
|
|
# change the syntax to AT&T
|
||
|
|
ks.syntax = KS_OPT_SYNTAX_ATT
|
||
|
|
|
||
|
|
# at&t syntax uses symbol(%rip) for rip relative addressing
|
||
|
|
encoding, count = ks.asm(b"lea __data(%rip), %rax\n__data:")
|
||
|
|
self.assertEqual(encoding, [ 0x48, 0x8d, 0x05, 0x00, 0x00, 0x00, 0x00 ])
|
||
|
|
|
||
|
|
# verify that rip relative addressing is indeed rip relative
|
||
|
|
encoding, count = ks.asm(b"nop\nnop\nlea __data(%rip), %rax\n__data:")
|
||
|
|
self.assertEqual(encoding, [ 0x90, 0x90, 0x48, 0x8d, 0x05, 0x00, 0x00, 0x00, 0x00 ])
|
||
|
|
|
||
|
|
|
||
|
|
class TestX64IntelLabelRel(regress.RegressTest):
|
||
|
|
def runTest(self):
|
||
|
|
# Initialize Keystone engine
|
||
|
|
ks = Ks(KS_ARCH_X86, KS_MODE_64)
|
||
|
|
# change the syntax to intel
|
||
|
|
ks.syntax = KS_OPT_SYNTAX_INTEL
|
||
|
|
|
||
|
|
# intel uses rip + label for rip relative addressing
|
||
|
|
encoding, count = ks.asm(b"lea rax, [rip + __data]\n__data:")
|
||
|
|
self.assertEqual(encoding, [ 0x48, 0x8d, 0x05, 0x00, 0x00, 0x00, 0x00 ])
|
||
|
|
|
||
|
|
# verify that rip relative addressing is indeed rip relative
|
||
|
|
encoding, count = ks.asm(b"nop\nnop\nlea rax, [rip + __data]\n__data:")
|
||
|
|
self.assertEqual(encoding, [ 0x90, 0x90, 0x48, 0x8d, 0x05, 0x00, 0x00, 0x00, 0x00 ])
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
regress.main()
|