nasm/macros/smartalign.mac

192 lines
7.2 KiB
Plaintext
Raw Normal View History

;; --------------------------------------------------------------------------
;;
;; Copyright 1996-2017 The NASM Authors - All Rights Reserved
;; See the file AUTHORS included with the NASM distribution for
;; the specific copyright holders.
;;
;; Redistribution and use in source and binary forms, with or without
;; modification, are permitted provided that the following
;; conditions are met:
;;
;; * Redistributions of source code must retain the above copyright
;; notice, this list of conditions and the following disclaimer.
;; * Redistributions in binary form must reproduce the above
;; copyright notice, this list of conditions and the following
;; disclaimer in the documentation and/or other materials provided
;; with the distribution.
;;
;; THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
;; CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
;; INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
;; MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
;; DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
;; CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
;; SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
;; NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
;; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
;; HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
;; CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
;; OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
;; EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
;;
;; --------------------------------------------------------------------------
;
; 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_GROUP?__ 1
%define __?ALIGN_32BIT_1B?__ 0x90
%define __?ALIGN_32BIT_GROUP?__ 1
%define __?ALIGN_64BIT_1B?__ 0x90
%define __?ALIGN_64BIT_GROUP?__ 1
%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?__ 0x8d,0xb4,0x00,0x00,0x90
%define __?ALIGN_16BIT_6B?__ 0x8d,0xb4,0x00,0x00,0x89,0xff
%define __?ALIGN_16BIT_7B?__ 0x8d,0xb4,0x00,0x00,0x8d,0x7d,0x00
%define __?ALIGN_16BIT_8B?__ 0x8d,0xb4,0x00,0x00,0x8d,0xbd,0x00,0x00
%define __?ALIGN_16BIT_GROUP?__ 8
%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
%define __?ALIGN_32BIT_GROUP?__ 7
%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_GROUP?__ 4
%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_GROUP?__ 4
%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_GROUP?__ 4
%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_GROUP?__ 4
%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_64BIT_GROUP?__ 4
%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
%define __?ALIGN_32BIT_GROUP?__ 7
%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_GROUP?__ 4
%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_GROUP?__ 4
%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_32BIT_GROUP?__ 8
%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
%define __?ALIGN_64BIT_GROUP?__ 8
%else
%error unknown alignment mode: %1
%endif
%ifnempty %2
%ifidni %2,nojmp
%xdefine __?ALIGN_JMP_THRESHOLD?__ -1
%else
%xdefine __?ALIGN_JMP_THRESHOLD?__ %2
%endif
%endif
%xdefine __?ALIGNMODE?__ %1,__?ALIGN_JMP_THRESHOLD?__
%endmacro
%defalias __ALIGNMODE__ __?ALIGNMODE?__
%unimacro align 1-2+.nolist
%imacro align 1-2+.nolist
sectalign %1 ; align a segment as well
%ifnempty %2
times (((%1) - (($-$$) % (%1))) % (%1)) %2
%elif __?PASS?__ == 0 || __?PASS?__ == 3
times (((%1) - (($-$$) % (%1))) % (%1)) nop
%else
%push
%assign %$pad (((%1) - (($-$$) % (%1))) % (%1))
%if __?ALIGN_JMP_THRESHOLD?__ != -1 && %$pad > __?ALIGN_JMP_THRESHOLD?__
jmp %$end
; We can't re-use %$pad here as $ will have changed!
times (((%1) - (($-$$) % (%1))) % (%1)) nop
%else
times (%$pad / __?ALIGN_%[__?BITS?__]BIT_GROUP?__) \
db __?ALIGN_%[__?BITS?__]BIT_%[__?ALIGN_%[__?BITS?__]BIT_GROUP?__]B?__
%assign %$pad %$pad % __?ALIGN_%[__?BITS?__]BIT_GROUP?__
%if %$pad > 0
db __?ALIGN_%[__?BITS?__]BIT_%[%$pad]B?__
%endif
%endif
%$end:
%pop
%endif
%endmacro
alignmode generic