macros/smartalign.mac

   1 ;; --------------------------------------------------------------------------
   2 ;;
   3 ;;   Copyright 1996-2017 The NASM Authors - All Rights Reserved
   4 ;;   See the file AUTHORS included with the NASM distribution for
   5 ;;   the specific copyright holders.
   6 ;;
   7 ;;   Redistribution and use in source and binary forms, with or without
   8 ;;   modification, are permitted provided that the following
   9 ;;   conditions are met:
  10 ;;
  11 ;;   * Redistributions of source code must retain the above copyright
  12 ;;     notice, this list of conditions and the following disclaimer.
  13 ;;   * Redistributions in binary form must reproduce the above
  14 ;;     copyright notice, this list of conditions and the following
  15 ;;     disclaimer in the documentation and/or other materials provided
  16 ;;     with the distribution.
  17 ;;
  18 ;;     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  19 ;;     CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  20 ;;     INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  21 ;;     MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  22 ;;     DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  23 ;;     CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24 ;;     SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  25 ;;     NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  26 ;;     LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27 ;;     HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  28 ;;     CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  29 ;;     OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  30 ;;     EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31 ;;
  32 ;; --------------------------------------------------------------------------
  33
  34 ;
  35 ; Smart alignment macros
  36 ;
  37 USE: smartalign
  38
  39 %imacro alignmode 1-2.nolist
  40   %ifidni %1,nop
  41     %define __?ALIGN_JMP_THRESHOLD?__ 16
  42
  43     %define __?ALIGN_16BIT_1B?__ 0x90
  44     %define __?ALIGN_16BIT_GROUP?__ 1
  45
  46     %define __?ALIGN_32BIT_1B?__ 0x90
  47     %define __?ALIGN_32BIT_GROUP?__ 1
  48
  49     %define __?ALIGN_64BIT_1B?__ 0x90
  50     %define __?ALIGN_64BIT_GROUP?__ 1
  51   %elifidni %1,generic
  52     %define __?ALIGN_JMP_THRESHOLD?__ 8
  53
  54     %define __?ALIGN_16BIT_1B?__ 0x90
  55     %define __?ALIGN_16BIT_2B?__ 0x89,0xf6
  56     %define __?ALIGN_16BIT_3B?__ 0x8d,0x74,0x00
  57     %define __?ALIGN_16BIT_4B?__ 0x8d,0xb4,0x00,0x00
  58     %define __?ALIGN_16BIT_5B?__ 0x8d,0xb4,0x00,0x00,0x90
  59     %define __?ALIGN_16BIT_6B?__ 0x8d,0xb4,0x00,0x00,0x89,0xff
  60     %define __?ALIGN_16BIT_7B?__ 0x8d,0xb4,0x00,0x00,0x8d,0x7d,0x00
  61     %define __?ALIGN_16BIT_8B?__ 0x8d,0xb4,0x00,0x00,0x8d,0xbd,0x00,0x00
  62     %define __?ALIGN_16BIT_GROUP?__ 8
  63
  64     %define __?ALIGN_32BIT_1B?__ 0x90
  65     %define __?ALIGN_32BIT_2B?__ 0x89,0xf6
  66     %define __?ALIGN_32BIT_3B?__ 0x8d,0x76,0x00
  67     %define __?ALIGN_32BIT_4B?__ 0x8d,0x74,0x26,0x00
  68     %define __?ALIGN_32BIT_5B?__ 0x90,0x8d,0x74,0x26,0x00
  69     %define __?ALIGN_32BIT_6B?__ 0x8d,0xb6,0x00,0x00,0x00,0x00
  70     %define __?ALIGN_32BIT_7B?__ 0x8d,0xb4,0x26,0x00,0x00,0x00,0x00
  71     %define __?ALIGN_32BIT_GROUP?__ 7
  72
  73     %define __?ALIGN_64BIT_1B?__ 0x90
  74     %define __?ALIGN_64BIT_2B?__ 0x66,0x90
  75     %define __?ALIGN_64BIT_3B?__ 0x66,0x66,0x90
  76     %define __?ALIGN_64BIT_4B?__ 0x66,0x66,0x66,0x90
  77     %define __?ALIGN_64BIT_GROUP?__ 4
  78   %elifidni %1,k8
  79     %define __?ALIGN_JMP_THRESHOLD?__ 16
  80
  81     %define __?ALIGN_16BIT_1B?__ 0x90
  82     %define __?ALIGN_16BIT_2B?__ 0x66,0x90
  83     %define __?ALIGN_16BIT_3B?__ 0x66,0x66,0x90
  84     %define __?ALIGN_16BIT_4B?__ 0x66,0x66,0x66,0x90
  85     %define __?ALIGN_16BIT_GROUP?__ 4
  86
  87     %define __?ALIGN_32BIT_1B?__ 0x90
  88     %define __?ALIGN_32BIT_2B?__ 0x66,0x90
  89     %define __?ALIGN_32BIT_3B?__ 0x66,0x66,0x90
  90     %define __?ALIGN_32BIT_4B?__ 0x66,0x66,0x66,0x90
  91     %define __?ALIGN_32BIT_GROUP?__ 4
  92
  93     %define __?ALIGN_64BIT_1B?__ 0x90
  94     %define __?ALIGN_64BIT_2B?__ 0x66,0x90
  95     %define __?ALIGN_64BIT_3B?__ 0x66,0x66,0x90
  96     %define __?ALIGN_64BIT_4B?__ 0x66,0x66,0x66,0x90
  97     %define __?ALIGN_64BIT_GROUP?__ 4
  98   %elifidni %1,k7
  99     %define __?ALIGN_JMP_THRESHOLD?__ 16
 100
 101     %define __?ALIGN_16BIT_1B?__ 0x90
 102     %define __?ALIGN_16BIT_2B?__ 0x66,0x90
 103     %define __?ALIGN_16BIT_3B?__ 0x66,0x66,0x90
 104     %define __?ALIGN_16BIT_4B?__ 0x66,0x66,0x66,0x90
 105     %define __?ALIGN_64BIT_GROUP?__ 4
 106
 107     %define __?ALIGN_32BIT_1B?__ 0x90
 108     %define __?ALIGN_32BIT_2B?__ 0x8b,0xc0
 109     %define __?ALIGN_32BIT_3B?__ 0x8d,0x04,0x20
 110     %define __?ALIGN_32BIT_4B?__ 0x8d,0x44,0x20,0x00
 111     %define __?ALIGN_32BIT_5B?__ 0x8d,0x44,0x20,0x00,0x90
 112     %define __?ALIGN_32BIT_6B?__ 0x8d,0x80,0x00,0x00,0x00,0x00
 113     %define __?ALIGN_32BIT_7B?__ 0x8d,0x04,0x05,0x00,0x00,0x00,0x00
 114     %define __?ALIGN_32BIT_GROUP?__ 7
 115
 116     %define __?ALIGN_64BIT_1B?__ 0x90
 117     %define __?ALIGN_64BIT_2B?__ 0x66,0x90
 118     %define __?ALIGN_64BIT_3B?__ 0x66,0x66,0x90
 119     %define __?ALIGN_64BIT_4B?__ 0x66,0x66,0x66,0x90
 120     %define __?ALIGN_64BIT_GROUP?__ 4
 121   %elifidni %1,p6
 122     %define __?ALIGN_JMP_THRESHOLD?__ 16
 123
 124     %define __?ALIGN_16BIT_1B?__ 0x90
 125     %define __?ALIGN_16BIT_2B?__ 0x66,0x90
 126     %define __?ALIGN_16BIT_3B?__ 0x0f,0x1f,0x00
 127     %define __?ALIGN_16BIT_4B?__ 0x0f,0x1f,0x40,0x00
 128     %define __?ALIGN_16BIT_GROUP?__ 4
 129
 130     %define __?ALIGN_32BIT_1B?__ 0x90
 131     %define __?ALIGN_32BIT_2B?__ 0x66,0x90
 132     %define __?ALIGN_32BIT_3B?__ 0x0f,0x1f,0x00
 133     %define __?ALIGN_32BIT_4B?__ 0x0f,0x1f,0x40,0x00
 134     %define __?ALIGN_32BIT_5B?__ 0x0f,0x1f,0x44,0x00,0x00
 135     %define __?ALIGN_32BIT_6B?__ 0x66,0x0f,0x1f,0x44,0x00,0x00
 136     %define __?ALIGN_32BIT_7B?__ 0x0f,0x1f,0x80,0x00,0x00,0x00,0x00
 137     %define __?ALIGN_32BIT_8B?__ 0x0f,0x1f,0x84,0x00,0x00,0x00,0x00,0x00
 138     %define __?ALIGN_32BIT_GROUP?__ 8
 139
 140     %define __?ALIGN_64BIT_1B?__ 0x90
 141     %define __?ALIGN_64BIT_2B?__ 0x66,0x90
 142     %define __?ALIGN_64BIT_3B?__ 0x0f,0x1f,0x00
 143     %define __?ALIGN_64BIT_4B?__ 0x0f,0x1f,0x40,0x00
 144     %define __?ALIGN_64BIT_5B?__ 0x0f,0x1f,0x44,0x00,0x00
 145     %define __?ALIGN_64BIT_6B?__ 0x66,0x0f,0x1f,0x44,0x00,0x00
 146     %define __?ALIGN_64BIT_7B?__ 0x0f,0x1f,0x80,0x00,0x00,0x00,0x00
 147     %define __?ALIGN_64BIT_8B?__ 0x0f,0x1f,0x84,0x00,0x00,0x00,0x00,0x00
 148     %define __?ALIGN_64BIT_GROUP?__ 8
 149   %else
 150     %error unknown alignment mode: %1
 151   %endif
 152   %ifnempty %2
 153     %ifidni %2,nojmp
 154       %xdefine __?ALIGN_JMP_THRESHOLD?__ -1
 155     %else
 156       %xdefine __?ALIGN_JMP_THRESHOLD?__ %2
 157     %endif
 158   %endif
 159   %xdefine __?ALIGNMODE?__ %1,__?ALIGN_JMP_THRESHOLD?__
 160 %endmacro
 161
 162 %defalias __ALIGNMODE__ __?ALIGNMODE?__
 163
 164 %unimacro align 1-2+.nolist
 165 %imacro align 1-2+.nolist
 166   sectalign %1   ; align a segment as well
 167   %ifnempty %2
 168     times (((%1) - (($-$$) % (%1))) % (%1)) %2
 169   %elif __?PASS?__ == 0 || __?PASS?__ == 3
 170     times (((%1) - (($-$$) % (%1))) % (%1)) nop
 171   %else
 172     %push
 173     %assign %$pad (((%1) - (($-$$) % (%1))) % (%1))
 174     %if __?ALIGN_JMP_THRESHOLD?__ != -1 && %$pad > __?ALIGN_JMP_THRESHOLD?__
 175       jmp %$end
 176       ; We can't re-use %$pad here as $ will have changed!
 177       times (((%1) - (($-$$) % (%1))) % (%1)) nop
 178     %else
 179       times (%$pad / __?ALIGN_%[__?BITS?__]BIT_GROUP?__) \
 180         db __?ALIGN_%[__?BITS?__]BIT_%[__?ALIGN_%[__?BITS?__]BIT_GROUP?__]B?__
 181       %assign %$pad %$pad % __?ALIGN_%[__?BITS?__]BIT_GROUP?__
 182       %if %$pad > 0
 183         db __?ALIGN_%[__?BITS?__]BIT_%[%$pad]B?__
 184       %endif
 185     %endif
 186 %$end:
 187     %pop
 188   %endif
 189 %endmacro
 190
 191         alignmode generic