projects / thirdparty / openssl.git / blame
| Commit | Line | Data |
|---|---|---|
| 3c2bdd7d | 1 | # Copyright 2020-2021 The OpenSSL Project Authors. All Rights Reserved. |
| c781eb1c AM |
2 | # Copyright (c) 2020, Intel Corporation. All Rights Reserved. |
| 3 | # | |
| 4 | # Licensed under the Apache License 2.0 (the "License"). You may not use | |
| 5 | # this file except in compliance with the License. You can obtain a copy | |
| 6 | # in the file LICENSE in the source distribution or at | |
| 7 | # https://www.openssl.org/source/license.html | |
| 8 | # | |
| 9 | # | |
| 10 | # Originally written by Ilya Albrekht, Sergey Kirillov and Andrey Matyukov | |
| 11 | # Intel Corporation | |
| 12 | # | |
| 13 | # December 2020 | |
| 14 | # | |
| 15 | # Initial release. | |
| 16 | # | |
| 17 | # Implementation utilizes 256-bit (ymm) registers to avoid frequency scaling issues. | |
| 18 | # | |
| 19 | # IceLake-Client @ 1.3GHz | |
| 20 | # |---------+----------------------+--------------+-------------| | |
| 21 | # | | OpenSSL 3.0.0-alpha9 | this | Unit | | |
| 22 | # |---------+----------------------+--------------+-------------| | |
| 23 | # | rsa2048 | 2 127 659 | 1 015 625 | cycles/sign | | |
| 24 | # | | 611 | 1280 / +109% | sign/s | | |
| 25 | # |---------+----------------------+--------------+-------------| | |
| 26 | # | |
| 27 | ||
| 28 | # $output is the last argument if it looks like a file (it has an extension) | |
| 29 | # $flavour is the first argument if it doesn't look like a file | |
| 30 | $output = $#ARGV >= 0 && $ARGV[$#ARGV] =~ m|\.\w+$| ? pop : undef; | |
| 31 | $flavour = $#ARGV >= 0 && $ARGV[0] !~ m|\.| ? shift : undef; | |
| 32 | ||
| 33 | $win64=0; $win64=1 if ($flavour =~ /[nm]asm|mingw64/ || $output =~ /\.asm$/); | |
| 34 | $avx512ifma=0; | |
| 35 | ||
| 36 | $0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1; | |
| 37 | ( $xlate="${dir}x86_64-xlate.pl" and -f $xlate ) or | |
| 38 | ( $xlate="${dir}../../perlasm/x86_64-xlate.pl" and -f $xlate) or | |
| 39 | die "can't locate x86_64-xlate.pl"; | |
| 40 | ||
| 41 | if (`$ENV{CC} -Wa,-v -c -o /dev/null -x assembler /dev/null 2>&1` | |
| 42 | =~ /GNU assembler version ([2-9]\.[0-9]+)/) { | |
| 43 | $avx512ifma = ($1>=2.26); | |
| 44 | } | |
| 45 | ||
| 46 | if (!$avx512 && $win64 && ($flavour =~ /nasm/ || $ENV{ASM} =~ /nasm/) && | |
| 47 | `nasm -v 2>&1` =~ /NASM version ([2-9]\.[0-9]+)(?:\.([0-9]+))?/) { | |
| 48 | $avx512ifma = ($1==2.11 && $2>=8) + ($1>=2.12); | |
| 49 | } | |
| 50 | ||
| 51 | if (!$avx512 && `$ENV{CC} -v 2>&1` =~ /((?:clang|LLVM) version|.*based on LLVM) ([0-9]+\.[0-9]+)/) { | |
| 788a72e9 | 52 | $avx512ifma = ($2>=7.0); |
| c781eb1c AM |
53 | } |
| 54 | ||
| 55 | open OUT,"| \"$^X\" \"$xlate\" $flavour \"$output\"" | |
| 56 | or die "can't call $xlate: $!"; | |
| 57 | *STDOUT=*OUT; | |
| 58 | ||
| 59 | if ($avx512ifma>0) {{{ | |
| 60 | @_6_args_universal_ABI = ("%rdi","%rsi","%rdx","%rcx","%r8","%r9"); | |
| 61 | ||
| 62 | $code.=<<___; | |
| 63 | .extern OPENSSL_ia32cap_P | |
| 64 | .globl rsaz_avx512ifma_eligible | |
| 65 | .type rsaz_avx512ifma_eligible,\@abi-omnipotent | |
| 66 | .align 32 | |
| 67 | rsaz_avx512ifma_eligible: | |
| 68 | mov OPENSSL_ia32cap_P+8(%rip), %ecx | |
| 69 | xor %eax,%eax | |
| 70 | and \$`1<<31|1<<21|1<<17|1<<16`, %ecx # avx512vl + avx512ifma + avx512dq + avx512f | |
| 71 | cmp \$`1<<31|1<<21|1<<17|1<<16`, %ecx | |
| 72 | cmove %ecx,%eax | |
| 73 | ret | |
| 74 | .size rsaz_avx512ifma_eligible, .-rsaz_avx512ifma_eligible | |
| 75 | ___ | |
| 76 | ||
| 77 | ############################################################################### | |
| 78 | # Almost Montgomery Multiplication (AMM) for 20-digit number in radix 2^52. | |
| 79 | # | |
| 80 | # AMM is defined as presented in the paper | |
| 81 | # "Efficient Software Implementations of Modular Exponentiation" by Shay Gueron. | |
| 82 | # | |
| 83 | # The input and output are presented in 2^52 radix domain, i.e. | |
| 84 | # |res|, |a|, |b|, |m| are arrays of 20 64-bit qwords with 12 high bits zeroed. | |
| 85 | # |k0| is a Montgomery coefficient, which is here k0 = -1/m mod 2^64 | |
| 86 | # (note, the implementation counts only 52 bits from it). | |
| 87 | # | |
| 88 | # NB: the AMM implementation does not perform "conditional" subtraction step as | |
| 89 | # specified in the original algorithm as according to the paper "Enhanced Montgomery | |
| 90 | # Multiplication" by Shay Gueron (see Lemma 1), the result will be always < 2*2^1024 | |
| 91 | # and can be used as a direct input to the next AMM iteration. | |
| 92 | # This post-condition is true, provided the correct parameter |s| is choosen, i.e. | |
| 93 | # s >= n + 2 * k, which matches our case: 1040 > 1024 + 2 * 1. | |
| 94 | # | |
| 95 | # void RSAZ_amm52x20_x1_256(BN_ULONG *res, | |
| 96 | # const BN_ULONG *a, | |
| 97 | # const BN_ULONG *b, | |
| 98 | # const BN_ULONG *m, | |
| 99 | # BN_ULONG k0); | |
| 100 | ############################################################################### | |
| 101 | { | |
| 102 | # input parameters ("%rdi","%rsi","%rdx","%rcx","%r8") | |
| 103 | my ($res,$a,$b,$m,$k0) = @_6_args_universal_ABI; | |
| 104 | ||
| 105 | my $mask52 = "%rax"; | |
| 106 | my $acc0_0 = "%r9"; | |
| 107 | my $acc0_0_low = "%r9d"; | |
| 108 | my $acc0_1 = "%r15"; | |
| 109 | my $acc0_1_low = "%r15d"; | |
| 110 | my $b_ptr = "%r11"; | |
| 111 | ||
| 112 | my $iter = "%ebx"; | |
| 113 | ||
| 114 | my $zero = "%ymm0"; | |
| 115 | my ($R0_0,$R0_0h,$R1_0,$R1_0h,$R2_0) = ("%ymm1", map("%ymm$_",(16..19))); | |
| 116 | my ($R0_1,$R0_1h,$R1_1,$R1_1h,$R2_1) = ("%ymm2", map("%ymm$_",(20..23))); | |
| 117 | my $Bi = "%ymm3"; | |
| 118 | my $Yi = "%ymm4"; | |
| 119 | ||
| 120 | # Registers mapping for normalization. | |
| 121 | # We can reuse Bi, Yi registers here. | |
| 122 | my $TMP = $Bi; | |
| 123 | my $mask52x4 = $Yi; | |
| 124 | my ($T0,$T0h,$T1,$T1h,$T2) = map("%ymm$_", (24..28)); | |
| 125 | ||
| 126 | sub amm52x20_x1() { | |
| 127 | # _data_offset - offset in the |a| or |m| arrays pointing to the beginning | |
| 128 | # of data for corresponding AMM operation; | |
| 129 | # _b_offset - offset in the |b| array pointing to the next qword digit; | |
| 130 | my ($_data_offset,$_b_offset,$_acc,$_R0,$_R0h,$_R1,$_R1h,$_R2,$_k0) = @_; | |
| 131 | my $_R0_xmm = $_R0 =~ s/%y/%x/r; | |
| 132 | $code.=<<___; | |
| 133 | movq $_b_offset($b_ptr), %r13 # b[i] | |
| 134 | ||
| 135 | vpbroadcastq %r13, $Bi # broadcast b[i] | |
| 136 | movq $_data_offset($a), %rdx | |
| 137 | mulx %r13, %r13, %r12 # a[0]*b[i] = (t0,t2) | |
| 138 | addq %r13, $_acc # acc += t0 | |
| 139 | movq %r12, %r10 | |
| 140 | adcq \$0, %r10 # t2 += CF | |
| 141 | ||
| 142 | movq $_k0, %r13 | |
| 143 | imulq $_acc, %r13 # acc * k0 | |
| 144 | andq $mask52, %r13 # yi = (acc * k0) & mask52 | |
| 145 | ||
| 146 | vpbroadcastq %r13, $Yi # broadcast y[i] | |
| 147 | movq $_data_offset($m), %rdx | |
| 148 | mulx %r13, %r13, %r12 # yi * m[0] = (t0,t1) | |
| 149 | addq %r13, $_acc # acc += t0 | |
| 150 | adcq %r12, %r10 # t2 += (t1 + CF) | |
| 151 | ||
| 152 | shrq \$52, $_acc | |
| 153 | salq \$12, %r10 | |
| 154 | or %r10, $_acc # acc = ((acc >> 52) | (t2 << 12)) | |
| 155 | ||
| 156 | vpmadd52luq `$_data_offset+64*0`($a), $Bi, $_R0 | |
| 157 | vpmadd52luq `$_data_offset+64*0+32`($a), $Bi, $_R0h | |
| 158 | vpmadd52luq `$_data_offset+64*1`($a), $Bi, $_R1 | |
| 159 | vpmadd52luq `$_data_offset+64*1+32`($a), $Bi, $_R1h | |
| 160 | vpmadd52luq `$_data_offset+64*2`($a), $Bi, $_R2 | |
| 161 | ||
| 162 | vpmadd52luq `$_data_offset+64*0`($m), $Yi, $_R0 | |
| 163 | vpmadd52luq `$_data_offset+64*0+32`($m), $Yi, $_R0h | |
| 164 | vpmadd52luq `$_data_offset+64*1`($m), $Yi, $_R1 | |
| 165 | vpmadd52luq `$_data_offset+64*1+32`($m), $Yi, $_R1h | |
| 166 | vpmadd52luq `$_data_offset+64*2`($m), $Yi, $_R2 | |
| 167 | ||
| 168 | # Shift accumulators right by 1 qword, zero extending the highest one | |
| 169 | valignq \$1, $_R0, $_R0h, $_R0 | |
| 170 | valignq \$1, $_R0h, $_R1, $_R0h | |
| 171 | valignq \$1, $_R1, $_R1h, $_R1 | |
| 172 | valignq \$1, $_R1h, $_R2, $_R1h | |
| 173 | valignq \$1, $_R2, $zero, $_R2 | |
| 174 | ||
| 175 | vmovq $_R0_xmm, %r13 | |
| 176 | addq %r13, $_acc # acc += R0[0] | |
| 177 | ||
| 178 | vpmadd52huq `$_data_offset+64*0`($a), $Bi, $_R0 | |
| 179 | vpmadd52huq `$_data_offset+64*0+32`($a), $Bi, $_R0h | |
| 180 | vpmadd52huq `$_data_offset+64*1`($a), $Bi, $_R1 | |
| 181 | vpmadd52huq `$_data_offset+64*1+32`($a), $Bi, $_R1h | |
| 182 | vpmadd52huq `$_data_offset+64*2`($a), $Bi, $_R2 | |
| 183 | ||
| 184 | vpmadd52huq `$_data_offset+64*0`($m), $Yi, $_R0 | |
| 185 | vpmadd52huq `$_data_offset+64*0+32`($m), $Yi, $_R0h | |
| 186 | vpmadd52huq `$_data_offset+64*1`($m), $Yi, $_R1 | |
| 187 | vpmadd52huq `$_data_offset+64*1+32`($m), $Yi, $_R1h | |
| 188 | vpmadd52huq `$_data_offset+64*2`($m), $Yi, $_R2 | |
| 189 | ___ | |
| 190 | } | |
| 191 | ||
| 192 | # Normalization routine: handles carry bits in R0..R2 QWs and | |
| 193 | # gets R0..R2 back to normalized 2^52 representation. | |
| 194 | # | |
| 195 | # Uses %r8-14,%e[bcd]x | |
| 196 | sub amm52x20_x1_norm { | |
| 197 | my ($_acc,$_R0,$_R0h,$_R1,$_R1h,$_R2) = @_; | |
| 198 | $code.=<<___; | |
| 199 | # Put accumulator to low qword in R0 | |
| 200 | vpbroadcastq $_acc, $TMP | |
| 201 | vpblendd \$3, $TMP, $_R0, $_R0 | |
| 202 | ||
| 203 | # Extract "carries" (12 high bits) from each QW of R0..R2 | |
| 204 | # Save them to LSB of QWs in T0..T2 | |
| 205 | vpsrlq \$52, $_R0, $T0 | |
| 206 | vpsrlq \$52, $_R0h, $T0h | |
| 207 | vpsrlq \$52, $_R1, $T1 | |
| 208 | vpsrlq \$52, $_R1h, $T1h | |
| 209 | vpsrlq \$52, $_R2, $T2 | |
| 210 | ||
| 211 | # "Shift left" T0..T2 by 1 QW | |
| 212 | valignq \$3, $T1h, $T2, $T2 | |
| 213 | valignq \$3, $T1, $T1h, $T1h | |
| 214 | valignq \$3, $T0h, $T1, $T1 | |
| 215 | valignq \$3, $T0, $T0h, $T0h | |
| 216 | valignq \$3, $zero, $T0, $T0 | |
| 217 | ||
| 218 | # Drop "carries" from R0..R2 QWs | |
| 219 | vpandq $mask52x4, $_R0, $_R0 | |
| 220 | vpandq $mask52x4, $_R0h, $_R0h | |
| 221 | vpandq $mask52x4, $_R1, $_R1 | |
| 222 | vpandq $mask52x4, $_R1h, $_R1h | |
| 223 | vpandq $mask52x4, $_R2, $_R2 | |
| 224 | ||
| 225 | # Sum R0..R2 with corresponding adjusted carries | |
| 226 | vpaddq $T0, $_R0, $_R0 | |
| 227 | vpaddq $T0h, $_R0h, $_R0h | |
| 228 | vpaddq $T1, $_R1, $_R1 | |
| 229 | vpaddq $T1h, $_R1h, $_R1h | |
| 230 | vpaddq $T2, $_R2, $_R2 | |
| 231 | ||
| 232 | # Now handle carry bits from this addition | |
| 233 | # Get mask of QWs which 52-bit parts overflow... | |
| 234 | vpcmpuq \$1, $_R0, $mask52x4, %k1 # OP=lt | |
| 235 | vpcmpuq \$1, $_R0h, $mask52x4, %k2 | |
| 236 | vpcmpuq \$1, $_R1, $mask52x4, %k3 | |
| 237 | vpcmpuq \$1, $_R1h, $mask52x4, %k4 | |
| 238 | vpcmpuq \$1, $_R2, $mask52x4, %k5 | |
| 239 | kmovb %k1, %r14d # k1 | |
| 240 | kmovb %k2, %r13d # k1h | |
| 241 | kmovb %k3, %r12d # k2 | |
| 242 | kmovb %k4, %r11d # k2h | |
| 243 | kmovb %k5, %r10d # k3 | |
| 244 | ||
| 245 | # ...or saturated | |
| 246 | vpcmpuq \$0, $_R0, $mask52x4, %k1 # OP=eq | |
| 247 | vpcmpuq \$0, $_R0h, $mask52x4, %k2 | |
| 248 | vpcmpuq \$0, $_R1, $mask52x4, %k3 | |
| 249 | vpcmpuq \$0, $_R1h, $mask52x4, %k4 | |
| 250 | vpcmpuq \$0, $_R2, $mask52x4, %k5 | |
| 251 | kmovb %k1, %r9d # k4 | |
| 252 | kmovb %k2, %r8d # k4h | |
| 253 | kmovb %k3, %ebx # k5 | |
| 254 | kmovb %k4, %ecx # k5h | |
| 255 | kmovb %k5, %edx # k6 | |
| 256 | ||
| 257 | # Get mask of QWs where carries shall be propagated to. | |
| 258 | # Merge 4-bit masks to 8-bit values to use add with carry. | |
| 259 | shl \$4, %r13b | |
| 260 | or %r13b, %r14b | |
| 261 | shl \$4, %r11b | |
| 262 | or %r11b, %r12b | |
| 263 | ||
| 264 | add %r14b, %r14b | |
| 265 | adc %r12b, %r12b | |
| 266 | adc %r10b, %r10b | |
| 267 | ||
| 268 | shl \$4, %r8b | |
| 269 | or %r8b,%r9b | |
| 270 | shl \$4, %cl | |
| 271 | or %cl, %bl | |
| 272 | ||
| 273 | add %r9b, %r14b | |
| 274 | adc %bl, %r12b | |
| 275 | adc %dl, %r10b | |
| 276 | ||
| 277 | xor %r9b, %r14b | |
| 278 | xor %bl, %r12b | |
| 279 | xor %dl, %r10b | |
| 280 | ||
| 281 | kmovb %r14d, %k1 | |
| 282 | shr \$4, %r14b | |
| 283 | kmovb %r14d, %k2 | |
| 284 | kmovb %r12d, %k3 | |
| 285 | shr \$4, %r12b | |
| 286 | kmovb %r12d, %k4 | |
| 287 | kmovb %r10d, %k5 | |
| 288 | ||
| 289 | # Add carries according to the obtained mask | |
| 290 | vpsubq $mask52x4, $_R0, ${_R0}{%k1} | |
| 291 | vpsubq $mask52x4, $_R0h, ${_R0h}{%k2} | |
| 292 | vpsubq $mask52x4, $_R1, ${_R1}{%k3} | |
| 293 | vpsubq $mask52x4, $_R1h, ${_R1h}{%k4} | |
| 294 | vpsubq $mask52x4, $_R2, ${_R2}{%k5} | |
| 295 | ||
| 296 | vpandq $mask52x4, $_R0, $_R0 | |
| 297 | vpandq $mask52x4, $_R0h, $_R0h | |
| 298 | vpandq $mask52x4, $_R1, $_R1 | |
| 299 | vpandq $mask52x4, $_R1h, $_R1h | |
| 300 | vpandq $mask52x4, $_R2, $_R2 | |
| 301 | ___ | |
| 302 | } | |
| 303 | ||
| 304 | $code.=<<___; | |
| 305 | .text | |
| 306 | ||
| 307 | .globl RSAZ_amm52x20_x1_256 | |
| 308 | .type RSAZ_amm52x20_x1_256,\@function,5 | |
| 309 | .align 32 | |
| 310 | RSAZ_amm52x20_x1_256: | |
| 311 | .cfi_startproc | |
| 312 | endbranch | |
| 313 | push %rbx | |
| 314 | .cfi_push %rbx | |
| 315 | push %rbp | |
| 316 | .cfi_push %rbp | |
| 317 | push %r12 | |
| 318 | .cfi_push %r12 | |
| 319 | push %r13 | |
| 320 | .cfi_push %r13 | |
| 321 | push %r14 | |
| 322 | .cfi_push %r14 | |
| 323 | push %r15 | |
| 324 | .cfi_push %r15 | |
| 325 | .Lrsaz_amm52x20_x1_256_body: | |
| 326 | ||
| 327 | # Zeroing accumulators | |
| 328 | vpxord $zero, $zero, $zero | |
| 329 | vmovdqa64 $zero, $R0_0 | |
| 330 | vmovdqa64 $zero, $R0_0h | |
| 331 | vmovdqa64 $zero, $R1_0 | |
| 332 | vmovdqa64 $zero, $R1_0h | |
| 333 | vmovdqa64 $zero, $R2_0 | |
| 334 | ||
| 335 | xorl $acc0_0_low, $acc0_0_low | |
| 336 | ||
| 337 | movq $b, $b_ptr # backup address of b | |
| 338 | movq \$0xfffffffffffff, $mask52 # 52-bit mask | |
| 339 | ||
| 340 | # Loop over 20 digits unrolled by 4 | |
| 341 | mov \$5, $iter | |
| 342 | ||
| 343 | .align 32 | |
| 344 | .Lloop5: | |
| 345 | ___ | |
| 346 | foreach my $idx (0..3) { | |
| 347 | &amm52x20_x1(0,8*$idx,$acc0_0,$R0_0,$R0_0h,$R1_0,$R1_0h,$R2_0,$k0); | |
| 348 | } | |
| 349 | $code.=<<___; | |
| 350 | lea `4*8`($b_ptr), $b_ptr | |
| 351 | dec $iter | |
| 352 | jne .Lloop5 | |
| 353 | ||
| 354 | vmovdqa64 .Lmask52x4(%rip), $mask52x4 | |
| 355 | ___ | |
| 356 | &amm52x20_x1_norm($acc0_0,$R0_0,$R0_0h,$R1_0,$R1_0h,$R2_0); | |
| 357 | $code.=<<___; | |
| 358 | ||
| 359 | vmovdqu64 $R0_0, ($res) | |
| 360 | vmovdqu64 $R0_0h, 32($res) | |
| 361 | vmovdqu64 $R1_0, 64($res) | |
| 362 | vmovdqu64 $R1_0h, 96($res) | |
| 363 | vmovdqu64 $R2_0, 128($res) | |
| 364 | ||
| 365 | vzeroupper | |
| 366 | mov 0(%rsp),%r15 | |
| 367 | .cfi_restore %r15 | |
| 368 | mov 8(%rsp),%r14 | |
| 369 | .cfi_restore %r14 | |
| 370 | mov 16(%rsp),%r13 | |
| 371 | .cfi_restore %r13 | |
| 372 | mov 24(%rsp),%r12 | |
| 373 | .cfi_restore %r12 | |
| 374 | mov 32(%rsp),%rbp | |
| 375 | .cfi_restore %rbp | |
| 376 | mov 40(%rsp),%rbx | |
| 377 | .cfi_restore %rbx | |
| 378 | lea 48(%rsp),%rsp | |
| 379 | .cfi_adjust_cfa_offset -48 | |
| 380 | .Lrsaz_amm52x20_x1_256_epilogue: | |
| 381 | ret | |
| 382 | .cfi_endproc | |
| 383 | .size RSAZ_amm52x20_x1_256, .-RSAZ_amm52x20_x1_256 | |
| 384 | ___ | |
| 385 | ||
| 386 | $code.=<<___; | |
| 387 | .data | |
| 388 | .align 32 | |
| 389 | .Lmask52x4: | |
| 390 | .quad 0xfffffffffffff | |
| 391 | .quad 0xfffffffffffff | |
| 392 | .quad 0xfffffffffffff | |
| 393 | .quad 0xfffffffffffff | |
| 394 | ___ | |
| 395 | ||
| 396 | ############################################################################### | |
| 397 | # Dual Almost Montgomery Multiplication for 20-digit number in radix 2^52 | |
| 398 | # | |
| 399 | # See description of RSAZ_amm52x20_x1_256() above for details about Almost | |
| 400 | # Montgomery Multiplication algorithm and function input parameters description. | |
| 401 | # | |
| 402 | # This function does two AMMs for two independent inputs, hence dual. | |
| 403 | # | |
| 404 | # void RSAZ_amm52x20_x2_256(BN_ULONG out[2][20], | |
| 405 | # const BN_ULONG a[2][20], | |
| 406 | # const BN_ULONG b[2][20], | |
| 407 | # const BN_ULONG m[2][20], | |
| 408 | # const BN_ULONG k0[2]); | |
| 409 | ############################################################################### | |
| 410 | ||
| 411 | $code.=<<___; | |
| 412 | .text | |
| 413 | ||
| 414 | .globl RSAZ_amm52x20_x2_256 | |
| 415 | .type RSAZ_amm52x20_x2_256,\@function,5 | |
| 416 | .align 32 | |
| 417 | RSAZ_amm52x20_x2_256: | |
| 418 | .cfi_startproc | |
| 419 | endbranch | |
| 420 | push %rbx | |
| 421 | .cfi_push %rbx | |
| 422 | push %rbp | |
| 423 | .cfi_push %rbp | |
| 424 | push %r12 | |
| 425 | .cfi_push %r12 | |
| 426 | push %r13 | |
| 427 | .cfi_push %r13 | |
| 428 | push %r14 | |
| 429 | .cfi_push %r14 | |
| 430 | push %r15 | |
| 431 | .cfi_push %r15 | |
| 432 | .Lrsaz_amm52x20_x2_256_body: | |
| 433 | ||
| 434 | # Zeroing accumulators | |
| 435 | vpxord $zero, $zero, $zero | |
| 436 | vmovdqa64 $zero, $R0_0 | |
| 437 | vmovdqa64 $zero, $R0_0h | |
| 438 | vmovdqa64 $zero, $R1_0 | |
| 439 | vmovdqa64 $zero, $R1_0h | |
| 440 | vmovdqa64 $zero, $R2_0 | |
| 441 | vmovdqa64 $zero, $R0_1 | |
| 442 | vmovdqa64 $zero, $R0_1h | |
| 443 | vmovdqa64 $zero, $R1_1 | |
| 444 | vmovdqa64 $zero, $R1_1h | |
| 445 | vmovdqa64 $zero, $R2_1 | |
| 446 | ||
| 447 | xorl $acc0_0_low, $acc0_0_low | |
| 448 | xorl $acc0_1_low, $acc0_1_low | |
| 449 | ||
| 450 | movq $b, $b_ptr # backup address of b | |
| 451 | movq \$0xfffffffffffff, $mask52 # 52-bit mask | |
| 452 | ||
| 453 | mov \$20, $iter | |
| 454 | ||
| 455 | .align 32 | |
| 456 | .Lloop20: | |
| 457 | ___ | |
| 458 | &amm52x20_x1( 0, 0,$acc0_0,$R0_0,$R0_0h,$R1_0,$R1_0h,$R2_0,"($k0)"); | |
| 459 | # 20*8 = offset of the next dimension in two-dimension array | |
| 460 | &amm52x20_x1(20*8,20*8,$acc0_1,$R0_1,$R0_1h,$R1_1,$R1_1h,$R2_1,"8($k0)"); | |
| 461 | $code.=<<___; | |
| 462 | lea 8($b_ptr), $b_ptr | |
| 463 | dec $iter | |
| 464 | jne .Lloop20 | |
| 465 | ||
| 466 | vmovdqa64 .Lmask52x4(%rip), $mask52x4 | |
| 467 | ___ | |
| 468 | &amm52x20_x1_norm($acc0_0,$R0_0,$R0_0h,$R1_0,$R1_0h,$R2_0); | |
| 469 | &amm52x20_x1_norm($acc0_1,$R0_1,$R0_1h,$R1_1,$R1_1h,$R2_1); | |
| 470 | $code.=<<___; | |
| 471 | ||
| 472 | vmovdqu64 $R0_0, ($res) | |
| 473 | vmovdqu64 $R0_0h, 32($res) | |
| 474 | vmovdqu64 $R1_0, 64($res) | |
| 475 | vmovdqu64 $R1_0h, 96($res) | |
| 476 | vmovdqu64 $R2_0, 128($res) | |
| 477 | ||
| 478 | vmovdqu64 $R0_1, 160($res) | |
| 479 | vmovdqu64 $R0_1h, 192($res) | |
| 480 | vmovdqu64 $R1_1, 224($res) | |
| 481 | vmovdqu64 $R1_1h, 256($res) | |
| 482 | vmovdqu64 $R2_1, 288($res) | |
| 483 | ||
| 484 | vzeroupper | |
| 485 | mov 0(%rsp),%r15 | |
| 486 | .cfi_restore %r15 | |
| 487 | mov 8(%rsp),%r14 | |
| 488 | .cfi_restore %r14 | |
| 489 | mov 16(%rsp),%r13 | |
| 490 | .cfi_restore %r13 | |
| 491 | mov 24(%rsp),%r12 | |
| 492 | .cfi_restore %r12 | |
| 493 | mov 32(%rsp),%rbp | |
| 494 | .cfi_restore %rbp | |
| 495 | mov 40(%rsp),%rbx | |
| 496 | .cfi_restore %rbx | |
| 497 | lea 48(%rsp),%rsp | |
| 498 | .cfi_adjust_cfa_offset -48 | |
| 499 | .Lrsaz_amm52x20_x2_256_epilogue: | |
| 500 | ret | |
| 501 | .cfi_endproc | |
| 502 | .size RSAZ_amm52x20_x2_256, .-RSAZ_amm52x20_x2_256 | |
| 503 | ___ | |
| 504 | } | |
| 505 | ||
| 506 | ############################################################################### | |
| 507 | # Constant time extraction from the precomputed table of powers base^i, where | |
| 508 | # i = 0..2^EXP_WIN_SIZE-1 | |
| 509 | # | |
| 510 | # The input |red_table| contains precomputations for two independent base values, | |
| 511 | # so the |tbl_idx| indicates for which base shall we extract the value. | |
| 512 | # |red_table_idx| is a power index. | |
| 513 | # | |
| 514 | # Extracted value (output) is 20 digit number in 2^52 radix. | |
| 515 | # | |
| 516 | # void extract_multiplier_2x20_win5(BN_ULONG *red_Y, | |
| 517 | # const BN_ULONG red_table[1 << EXP_WIN_SIZE][2][20], | |
| 518 | # int red_table_idx, | |
| 519 | # int tbl_idx); # 0 or 1 | |
| 520 | # | |
| 521 | # EXP_WIN_SIZE = 5 | |
| 522 | ############################################################################### | |
| 523 | { | |
| 524 | # input parameters | |
| 525 | my ($out,$red_tbl,$red_tbl_idx,$tbl_idx) = @_6_args_universal_ABI; | |
| 526 | ||
| 527 | my ($t0,$t1,$t2,$t3,$t4) = map("%ymm$_", (0..4)); | |
| 528 | my $t4xmm = $t4 =~ s/%y/%x/r; | |
| 529 | my ($tmp0,$tmp1,$tmp2,$tmp3,$tmp4) = map("%ymm$_", (16..20)); | |
| 530 | my ($cur_idx,$idx,$ones) = map("%ymm$_", (21..23)); | |
| 531 | ||
| 532 | $code.=<<___; | |
| 533 | .text | |
| 534 | ||
| 535 | .align 32 | |
| 536 | .globl extract_multiplier_2x20_win5 | |
| 537 | .type extract_multiplier_2x20_win5,\@function,4 | |
| 538 | extract_multiplier_2x20_win5: | |
| 539 | .cfi_startproc | |
| 540 | endbranch | |
| 541 | leaq ($tbl_idx,$tbl_idx,4), %rax | |
| 542 | salq \$5, %rax | |
| 543 | addq %rax, $red_tbl | |
| 544 | ||
| 545 | vmovdqa64 .Lones(%rip), $ones # broadcast ones | |
| 546 | vpbroadcastq $red_tbl_idx, $idx | |
| 547 | leaq `(1<<5)*2*20*8`($red_tbl), %rax # holds end of the tbl | |
| 548 | ||
| 549 | vpxor $t4xmm, $t4xmm, $t4xmm | |
| 550 | vmovdqa64 $t4, $t3 # zeroing t0..4, cur_idx | |
| 551 | vmovdqa64 $t4, $t2 | |
| 552 | vmovdqa64 $t4, $t1 | |
| 553 | vmovdqa64 $t4, $t0 | |
| 554 | vmovdqa64 $t4, $cur_idx | |
| 555 | ||
| 556 | .align 32 | |
| 557 | .Lloop: | |
| 558 | vpcmpq \$0, $cur_idx, $idx, %k1 # mask of (idx == cur_idx) | |
| 559 | addq \$320, $red_tbl # 320 = 2 * 20 digits * 8 bytes | |
| 560 | vpaddq $ones, $cur_idx, $cur_idx # increment cur_idx | |
| 561 | vmovdqu64 -320($red_tbl), $tmp0 # load data from red_tbl | |
| 562 | vmovdqu64 -288($red_tbl), $tmp1 | |
| 563 | vmovdqu64 -256($red_tbl), $tmp2 | |
| 564 | vmovdqu64 -224($red_tbl), $tmp3 | |
| 565 | vmovdqu64 -192($red_tbl), $tmp4 | |
| 566 | vpblendmq $tmp0, $t0, ${t0}{%k1} # extract data when mask is not zero | |
| 567 | vpblendmq $tmp1, $t1, ${t1}{%k1} | |
| 568 | vpblendmq $tmp2, $t2, ${t2}{%k1} | |
| 569 | vpblendmq $tmp3, $t3, ${t3}{%k1} | |
| 570 | vpblendmq $tmp4, $t4, ${t4}{%k1} | |
| 571 | cmpq $red_tbl, %rax | |
| 572 | jne .Lloop | |
| 573 | ||
| 574 | vmovdqu64 $t0, ($out) # store t0..4 | |
| 575 | vmovdqu64 $t1, 32($out) | |
| 576 | vmovdqu64 $t2, 64($out) | |
| 577 | vmovdqu64 $t3, 96($out) | |
| 578 | vmovdqu64 $t4, 128($out) | |
| 579 | ||
| 580 | ret | |
| 581 | .cfi_endproc | |
| 582 | .size extract_multiplier_2x20_win5, .-extract_multiplier_2x20_win5 | |
| 583 | ___ | |
| 584 | $code.=<<___; | |
| 585 | .data | |
| 586 | .align 32 | |
| 587 | .Lones: | |
| 588 | .quad 1,1,1,1 | |
| 589 | ___ | |
| 590 | } | |
| 591 | ||
| 592 | if ($win64) { | |
| 593 | $rec="%rcx"; | |
| 594 | $frame="%rdx"; | |
| 595 | $context="%r8"; | |
| 596 | $disp="%r9"; | |
| 597 | ||
| 598 | $code.=<<___ | |
| 599 | .extern __imp_RtlVirtualUnwind | |
| 600 | .type rsaz_def_handler,\@abi-omnipotent | |
| 601 | .align 16 | |
| 602 | rsaz_def_handler: | |
| 603 | push %rsi | |
| 604 | push %rdi | |
| 605 | push %rbx | |
| 606 | push %rbp | |
| 607 | push %r12 | |
| 608 | push %r13 | |
| 609 | push %r14 | |
| 610 | push %r15 | |
| 611 | pushfq | |
| 612 | sub \$64,%rsp | |
| 613 | ||
| 614 | mov 120($context),%rax # pull context->Rax | |
| 615 | mov 248($context),%rbx # pull context->Rip | |
| 616 | ||
| 617 | mov 8($disp),%rsi # disp->ImageBase | |
| 618 | mov 56($disp),%r11 # disp->HandlerData | |
| 619 | ||
| 620 | mov 0(%r11),%r10d # HandlerData[0] | |
| 621 | lea (%rsi,%r10),%r10 # prologue label | |
| 622 | cmp %r10,%rbx # context->Rip<.Lprologue | |
| 623 | jb .Lcommon_seh_tail | |
| 624 | ||
| 625 | mov 152($context),%rax # pull context->Rsp | |
| 626 | ||
| 627 | mov 4(%r11),%r10d # HandlerData[1] | |
| 628 | lea (%rsi,%r10),%r10 # epilogue label | |
| 629 | cmp %r10,%rbx # context->Rip>=.Lepilogue | |
| 630 | jae .Lcommon_seh_tail | |
| 631 | ||
| 632 | lea 48(%rax),%rax | |
| 633 | ||
| 634 | mov -8(%rax),%rbx | |
| 635 | mov -16(%rax),%rbp | |
| 636 | mov -24(%rax),%r12 | |
| 637 | mov -32(%rax),%r13 | |
| 638 | mov -40(%rax),%r14 | |
| 639 | mov -48(%rax),%r15 | |
| 640 | mov %rbx,144($context) # restore context->Rbx | |
| 641 | mov %rbp,160($context) # restore context->Rbp | |
| 642 | mov %r12,216($context) # restore context->R12 | |
| 643 | mov %r13,224($context) # restore context->R13 | |
| 644 | mov %r14,232($context) # restore context->R14 | |
| 645 | mov %r15,240($context) # restore context->R14 | |
| 646 | ||
| 647 | .Lcommon_seh_tail: | |
| 648 | mov 8(%rax),%rdi | |
| 649 | mov 16(%rax),%rsi | |
| 650 | mov %rax,152($context) # restore context->Rsp | |
| 651 | mov %rsi,168($context) # restore context->Rsi | |
| 652 | mov %rdi,176($context) # restore context->Rdi | |
| 653 | ||
| 654 | mov 40($disp),%rdi # disp->ContextRecord | |
| 655 | mov $context,%rsi # context | |
| 656 | mov \$154,%ecx # sizeof(CONTEXT) | |
| 657 | .long 0xa548f3fc # cld; rep movsq | |
| 658 | ||
| 659 | mov $disp,%rsi | |
| 660 | xor %rcx,%rcx # arg1, UNW_FLAG_NHANDLER | |
| 661 | mov 8(%rsi),%rdx # arg2, disp->ImageBase | |
| 662 | mov 0(%rsi),%r8 # arg3, disp->ControlPc | |
| 663 | mov 16(%rsi),%r9 # arg4, disp->FunctionEntry | |
| 664 | mov 40(%rsi),%r10 # disp->ContextRecord | |
| 665 | lea 56(%rsi),%r11 # &disp->HandlerData | |
| 666 | lea 24(%rsi),%r12 # &disp->EstablisherFrame | |
| 667 | mov %r10,32(%rsp) # arg5 | |
| 668 | mov %r11,40(%rsp) # arg6 | |
| 669 | mov %r12,48(%rsp) # arg7 | |
| 670 | mov %rcx,56(%rsp) # arg8, (NULL) | |
| 671 | call *__imp_RtlVirtualUnwind(%rip) | |
| 672 | ||
| 673 | mov \$1,%eax # ExceptionContinueSearch | |
| 674 | add \$64,%rsp | |
| 675 | popfq | |
| 676 | pop %r15 | |
| 677 | pop %r14 | |
| 678 | pop %r13 | |
| 679 | pop %r12 | |
| 680 | pop %rbp | |
| 681 | pop %rbx | |
| 682 | pop %rdi | |
| 683 | pop %rsi | |
| 684 | ret | |
| 685 | .size rsaz_def_handler,.-rsaz_def_handler | |
| 686 | ||
| 687 | .section .pdata | |
| 688 | .align 4 | |
| 689 | .rva .LSEH_begin_RSAZ_amm52x20_x1_256 | |
| 690 | .rva .LSEH_end_RSAZ_amm52x20_x1_256 | |
| 691 | .rva .LSEH_info_RSAZ_amm52x20_x1_256 | |
| 692 | ||
| c781eb1c AM |
693 | .rva .LSEH_begin_RSAZ_amm52x20_x2_256 |
| 694 | .rva .LSEH_end_RSAZ_amm52x20_x2_256 | |
| 695 | .rva .LSEH_info_RSAZ_amm52x20_x2_256 | |
| 696 | ||
| b238e78f AM |
697 | .rva .LSEH_begin_extract_multiplier_2x20_win5 |
| 698 | .rva .LSEH_end_extract_multiplier_2x20_win5 | |
| 699 | .rva .LSEH_info_extract_multiplier_2x20_win5 | |
| 700 | ||
| c781eb1c AM |
701 | .section .xdata |
| 702 | .align 8 | |
| 703 | .LSEH_info_RSAZ_amm52x20_x1_256: | |
| 704 | .byte 9,0,0,0 | |
| 705 | .rva rsaz_def_handler | |
| 706 | .rva .Lrsaz_amm52x20_x1_256_body,.Lrsaz_amm52x20_x1_256_epilogue | |
| c781eb1c AM |
707 | .LSEH_info_RSAZ_amm52x20_x2_256: |
| 708 | .byte 9,0,0,0 | |
| 709 | .rva rsaz_def_handler | |
| 710 | .rva .Lrsaz_amm52x20_x2_256_body,.Lrsaz_amm52x20_x2_256_epilogue | |
| b238e78f AM |
711 | .LSEH_info_extract_multiplier_2x20_win5: |
| 712 | .byte 9,0,0,0 | |
| 713 | .rva rsaz_def_handler | |
| 714 | .rva .LSEH_begin_extract_multiplier_2x20_win5,.LSEH_begin_extract_multiplier_2x20_win5 | |
| c781eb1c AM |
715 | ___ |
| 716 | } | |
| 717 | }}} else {{{ # fallback for old assembler | |
| 718 | $code.=<<___; | |
| 719 | .text | |
| 720 | ||
| 721 | .globl rsaz_avx512ifma_eligible | |
| 722 | .type rsaz_avx512ifma_eligible,\@abi-omnipotent | |
| 723 | rsaz_avx512ifma_eligible: | |
| 724 | xor %eax,%eax | |
| 725 | ret | |
| 726 | .size rsaz_avx512ifma_eligible, .-rsaz_avx512ifma_eligible | |
| 727 | ||
| 728 | .globl RSAZ_amm52x20_x1_256 | |
| 729 | .globl RSAZ_amm52x20_x2_256 | |
| 730 | .globl extract_multiplier_2x20_win5 | |
| 731 | .type RSAZ_amm52x20_x1_256,\@abi-omnipotent | |
| 732 | RSAZ_amm52x20_x1_256: | |
| 733 | RSAZ_amm52x20_x2_256: | |
| 734 | extract_multiplier_2x20_win5: | |
| 735 | .byte 0x0f,0x0b # ud2 | |
| 736 | ret | |
| 737 | .size RSAZ_amm52x20_x1_256, .-RSAZ_amm52x20_x1_256 | |
| 738 | ___ | |
| 739 | }}} | |
| 740 | ||
| 741 | $code =~ s/\`([^\`]*)\`/eval $1/gem; | |
| 742 | print $code; | |
| 743 | close STDOUT or die "error closing STDOUT: $!"; |