nasm/macros/smartalign.mac

264 lines
9.3 KiB
Text
Raw Normal View History

;
; Smart alignment macros
;
USE: smartalign
%imacro alignmode 1-2.nolist
%ifidni %1,nop
%define __ALIGN_JMP_THRESHOLD__ 16
%define __ALIGN_16BIT_1B__ 0x90
%define __ALIGN_16BIT_2B__ 0x90,0x90
%define __ALIGN_16BIT_3B__ 0x90,0x90,0x90
%define __ALIGN_16BIT_4B__ 0x90,0x90,0x90,0x90
%define __ALIGN_16BIT_5B__ 0x90,0x90,0x90,0x90,0x90
%define __ALIGN_16BIT_6B__ 0x90,0x90,0x90,0x90,0x90,0x90
%define __ALIGN_16BIT_7B__ 0x90,0x90,0x90,0x90,0x90,0x90,0x90
%define __ALIGN_16BIT_8B__ 0x90,0x90,0x90,0x90,0x90,0x90,0x90,0x90
%define __ALIGN_32BIT_1B__ 0x90
%define __ALIGN_32BIT_2B__ 0x90,0x90
%define __ALIGN_32BIT_3B__ 0x90,0x90,0x90
%define __ALIGN_32BIT_4B__ 0x90,0x90,0x90,0x90
%define __ALIGN_32BIT_5B__ 0x90,0x90,0x90,0x90,0x90
%define __ALIGN_32BIT_6B__ 0x90,0x90,0x90,0x90,0x90,0x90
%define __ALIGN_32BIT_7B__ 0x90,0x90,0x90,0x90,0x90,0x90,0x90
%define __ALIGN_32BIT_8B__ 0x90,0x90,0x90,0x90,0x90,0x90,0x90,0x90
%define __ALIGN_64BIT_1B__ 0x90
%define __ALIGN_64BIT_2B__ 0x90,0x90
%define __ALIGN_64BIT_3B__ 0x90,0x90,0x90
%define __ALIGN_64BIT_4B__ 0x90,0x90,0x90,0x90
%define __ALIGN_64BIT_5B__ 0x90,0x90,0x90,0x90,0x90
%define __ALIGN_64BIT_6B__ 0x90,0x90,0x90,0x90,0x90,0x90
%define __ALIGN_64BIT_7B__ 0x90,0x90,0x90,0x90,0x90,0x90,0x90
%define __ALIGN_64BIT_8B__ 0x90,0x90,0x90,0x90,0x90,0x90,0x90,0x90
%elifidni %1,generic
%define __ALIGN_JMP_THRESHOLD__ 8
%define __ALIGN_16BIT_1B__ 0x90
%define __ALIGN_16BIT_2B__ 0x89,0xf6
%define __ALIGN_16BIT_3B__ 0x8d,0x74,0x00
%define __ALIGN_16BIT_4B__ 0x8d,0xb4,0x00,0x00
%define __ALIGN_16BIT_5B__ 0x90,0x8d,0xb4,0x00,0x00
%define __ALIGN_16BIT_6B__ 0x89,0xf6,0x8d,0xbd,0x00,0x00
%define __ALIGN_16BIT_7B__ 0x8d,0x74,0x00,0x8d,0xbd,0x00,0x00
%define __ALIGN_16BIT_8B__ 0x8d,0xb4,0x00,0x00,0x8d,0xbd,0x00,0x00
%define __ALIGN_32BIT_1B__ 0x90
%define __ALIGN_32BIT_2B__ 0x89,0xf6
%define __ALIGN_32BIT_3B__ 0x8d,0x76,0x00
%define __ALIGN_32BIT_4B__ 0x8d,0x74,0x26,0x00
%define __ALIGN_32BIT_5B__ 0x90,0x8d,0x74,0x26,0x00
%define __ALIGN_32BIT_6B__ 0x8d,0xb6,0x00,0x00,0x00,0x00
%define __ALIGN_32BIT_7B__ 0x8d,0xb4,0x26,0x00,0x00,0x00,0x00
%undef __ALIGN_32BIT_8B__
%define __ALIGN_64BIT_1B__ 0x90
%define __ALIGN_64BIT_2B__ 0x66,0x90
%define __ALIGN_64BIT_3B__ 0x66,0x66,0x90
%define __ALIGN_64BIT_4B__ 0x66,0x66,0x66,0x90
%define __ALIGN_64BIT_5B__ 0x66,0x66,0x90,0x66,0x90
%define __ALIGN_64BIT_6B__ 0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_64BIT_7B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_64BIT_8B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x66,0x90
%elifidni %1,k8
%define __ALIGN_JMP_THRESHOLD__ 16
%define __ALIGN_16BIT_1B__ 0x90
%define __ALIGN_16BIT_2B__ 0x66,0x90
%define __ALIGN_16BIT_3B__ 0x66,0x66,0x90
%define __ALIGN_16BIT_4B__ 0x66,0x66,0x66,0x90
%define __ALIGN_16BIT_5B__ 0x66,0x66,0x90,0x66,0x90
%define __ALIGN_16BIT_6B__ 0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_16BIT_7B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_16BIT_8B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x66,0x90
%define __ALIGN_32BIT_1B__ 0x90
%define __ALIGN_32BIT_2B__ 0x66,0x90
%define __ALIGN_32BIT_3B__ 0x66,0x66,0x90
%define __ALIGN_32BIT_4B__ 0x66,0x66,0x66,0x90
%define __ALIGN_32BIT_5B__ 0x66,0x66,0x90,0x66,0x90
%define __ALIGN_32BIT_6B__ 0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_32BIT_7B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_32BIT_8B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x66,0x90
%define __ALIGN_64BIT_1B__ 0x90
%define __ALIGN_64BIT_2B__ 0x66,0x90
%define __ALIGN_64BIT_3B__ 0x66,0x66,0x90
%define __ALIGN_64BIT_4B__ 0x66,0x66,0x66,0x90
%define __ALIGN_64BIT_5B__ 0x66,0x66,0x90,0x66,0x90
%define __ALIGN_64BIT_6B__ 0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_64BIT_7B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_64BIT_8B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x66,0x90
%elifidni %1,k7
%define __ALIGN_JMP_THRESHOLD__ 16
%define __ALIGN_16BIT_1B__ 0x90
%define __ALIGN_16BIT_2B__ 0x66,0x90
%define __ALIGN_16BIT_3B__ 0x66,0x66,0x90
%define __ALIGN_16BIT_4B__ 0x66,0x66,0x66,0x90
%define __ALIGN_16BIT_5B__ 0x66,0x66,0x90,0x66,0x90
%define __ALIGN_16BIT_6B__ 0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_16BIT_7B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_16BIT_8B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x66,0x90
%define __ALIGN_32BIT_1B__ 0x90
%define __ALIGN_32BIT_2B__ 0x8b,0xc0
%define __ALIGN_32BIT_3B__ 0x8d,0x04,0x20
%define __ALIGN_32BIT_4B__ 0x8d,0x44,0x20,0x00
%define __ALIGN_32BIT_5B__ 0x8d,0x44,0x20,0x00,0x90
%define __ALIGN_32BIT_6B__ 0x8d,0x80,0x00,0x00,0x00,0x00
%define __ALIGN_32BIT_7B__ 0x8d,0x04,0x05,0x00,0x00,0x00,0x00
%undef __ALIGN_32BIT_8B__
%define __ALIGN_64BIT_1B__ 0x90
%define __ALIGN_64BIT_2B__ 0x66,0x90
%define __ALIGN_64BIT_3B__ 0x66,0x66,0x90
%define __ALIGN_64BIT_4B__ 0x66,0x66,0x66,0x90
%define __ALIGN_64BIT_5B__ 0x66,0x66,0x90,0x66,0x90
%define __ALIGN_64BIT_6B__ 0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_64BIT_7B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x90
%define __ALIGN_64BIT_8B__ 0x66,0x66,0x66,0x90,0x66,0x66,0x66,0x90
%elifidni %1,p6
%define __ALIGN_JMP_THRESHOLD__ 16
%define __ALIGN_16BIT_1B__ 0x90
%define __ALIGN_16BIT_2B__ 0x66,0x90
%define __ALIGN_16BIT_3B__ 0x0f,0x1f,0x00
%define __ALIGN_16BIT_4B__ 0x0f,0x1f,0x40,0x00
%define __ALIGN_16BIT_5B__ 0x90,0x0f,0x1f,0x40,0x00
%define __ALIGN_16BIT_6B__ 0x0f,0x1f,0x00,0x0f,0x1f,0x00
%define __ALIGN_16BIT_7B__ 0x0f,0x1f,0x00,0x0f,0x1f,0x40,0x00
%define __ALIGN_16BIT_8B__ 0x0f,0x1f,0x40,0x00,0x0f,0x1f,0x40,0x00
%define __ALIGN_32BIT_1B__ 0x90
%define __ALIGN_32BIT_2B__ 0x66,0x90
%define __ALIGN_32BIT_3B__ 0x0f,0x1f,0x00
%define __ALIGN_32BIT_4B__ 0x0f,0x1f,0x40,0x00
%define __ALIGN_32BIT_5B__ 0x0f,0x1f,0x44,0x00,0x00
%define __ALIGN_32BIT_6B__ 0x66,0x0f,0x1f,0x44,0x00,0x00
%define __ALIGN_32BIT_7B__ 0x0f,0x1f,0x80,0x00,0x00,0x00,0x00
%define __ALIGN_32BIT_8B__ 0x0f,0x1f,0x84,0x00,0x00,0x00,0x00,0x00
%define __ALIGN_64BIT_1B__ 0x90
%define __ALIGN_64BIT_2B__ 0x66,0x90
%define __ALIGN_64BIT_3B__ 0x0f,0x1f,0x00
%define __ALIGN_64BIT_4B__ 0x0f,0x1f,0x40,0x00
%define __ALIGN_64BIT_5B__ 0x0f,0x1f,0x44,0x00,0x00
%define __ALIGN_64BIT_6B__ 0x66,0x0f,0x1f,0x44,0x00,0x00
%define __ALIGN_64BIT_7B__ 0x0f,0x1f,0x80,0x00,0x00,0x00,0x00
%define __ALIGN_64BIT_8B__ 0x0f,0x1f,0x84,0x00,0x00,0x00,0x00,0x00
%else
%error unknown alignment mode: %1
%endif
%ifnempty %2
%xdefine __ALIGN_JMP_THRESHOLD__ %2
%endif
%xdefine __ALIGNMODE__ %1,__ALIGN_JMP_THRESHOLD__
%endmacro
%unimacro align 1-2+.nolist
%imacro align 1-2+.nolist
%ifnempty %2
times (((%1) - (($-$$) % (%1))) % (%1)) %2
%else
%push
%assign %$pad (((%1) - (($-$$) % (%1))) % (%1))
%if %$pad > __ALIGN_JMP_THRESHOLD__
jmp %$end
; We can't re-use %$pad here as $ will have changed!
times (((%1) - (($-$$) % (%1))) % (%1)) nop
%$end:
%else
%if __BITS__ == 16
%ifdef __ALIGN_16BIT_8B__
%rep %$pad / 8
db __ALIGN_16BIT_8B__
%endrep
%assign %$pad %$pad % 8
%else
%rep %$pad / 7
db __ALIGN_16BIT_7B__
%endrep
%assign %$pad %$pad % 7
%endif
%if %$pad == 1
db __ALIGN_16BIT_1B__
%elif %$pad == 2
db __ALIGN_16BIT_2B__
%elif %$pad == 3
db __ALIGN_16BIT_3B__
%elif %$pad == 4
db __ALIGN_16BIT_4B__
%elif %$pad == 5
db __ALIGN_16BIT_5B__
%elif %$pad == 6
db __ALIGN_16BIT_6B__
%elif %$pad == 7
db __ALIGN_16BIT_7B__
%endif
%elif __BITS__ == 32
%ifdef __ALIGN_32BIT_8B__
%rep %$pad / 8
db __ALIGN_32BIT_8B__
%endrep
%assign %$pad %$pad % 8
%else
%rep %$pad / 7
db __ALIGN_32BIT_7B__
%endrep
%assign %$pad %$pad % 7
%endif
%if %$pad == 1
db __ALIGN_32BIT_1B__
%elif %$pad == 2
db __ALIGN_32BIT_2B__
%elif %$pad == 3
db __ALIGN_32BIT_3B__
%elif %$pad == 4
db __ALIGN_32BIT_4B__
%elif %$pad == 5
db __ALIGN_32BIT_5B__
%elif %$pad == 6
db __ALIGN_32BIT_6B__
%elif %$pad == 7
db __ALIGN_32BIT_7B__
%endif
%elif __BITS__ == 64
%ifdef __ALIGN_64BIT_8B__
%rep %$pad / 8
db __ALIGN_64BIT_8B__
%endrep
%assign %$pad %$pad % 8
%else
%rep %$pad / 7
db __ALIGN_64BIT_7B__
%endrep
%assign %$pad %$pad % 7
%endif
%if %$pad == 1
db __ALIGN_64BIT_1B__
%elif %$pad == 2
db __ALIGN_64BIT_2B__
%elif %$pad == 3
db __ALIGN_64BIT_3B__
%elif %$pad == 4
db __ALIGN_64BIT_4B__
%elif %$pad == 5
db __ALIGN_64BIT_5B__
%elif %$pad == 6
db __ALIGN_64BIT_6B__
%elif %$pad == 7
db __ALIGN_64BIT_7B__
%endif
%else
%error "Invalid __BITS__ value"
%endif
%endif
%pop
%endif
%endmacro
alignmode generic