zlib/contrib/masmx64/inffasx64.asm

   1 ; inffasx64.asm is a hand tuned assembler version of inffast.c - fast decoding
   2 ; version for AMD64 on Windows using Microsoft C compiler
   3 ;
   4 ; inffasx64.asm is automatically convert from AMD64 portion of inffas86.c
   5 ; inffasx64.asm is called by inffas8664.c, which contain more info.
   6
   7
   8 ; to compile this file, I use option
   9 ;   ml64.exe /Flinffasx64 /c /Zi inffasx64.asm
  10 ;   with Microsoft Macro Assembler (x64) for AMD64
  11 ;
  12
  13 ; This file compile with Microsoft Macro Assembler (x64) for AMD64
  14 ;
  15 ;   ml64.exe is given with Visual Studio 2005/2008/2010 and Windows WDK
  16 ;
  17 ;   (you can get Windows WDK with ml64 for AMD64 from
  18 ;      http://www.microsoft.com/whdc/Devtools/wdk/default.mspx for low price)
  19 ;
  20
  21
  22 .code
  23 inffas8664fnc PROC
  24
  25 ; see http://weblogs.asp.net/oldnewthing/archive/2004/01/14/58579.aspx and
  26 ; http://msdn.microsoft.com/library/en-us/kmarch/hh/kmarch/64bitAMD_8e951dd2-ee77-4728-8702-55ce4b5dd24a.xml.asp
  27 ;
  28 ; All registers must be preserved across the call, except for
  29 ;   rax, rcx, rdx, r8, r-9, r10, and r11, which are scratch.
  30
  31
  32         mov [rsp-8],rsi
  33         mov [rsp-16],rdi
  34         mov [rsp-24],r12
  35         mov [rsp-32],r13
  36         mov [rsp-40],r14
  37         mov [rsp-48],r15
  38         mov [rsp-56],rbx
  39
  40         mov rax,rcx
  41
  42         mov     [rax+8], rbp       ; /* save regs rbp and rsp */
  43         mov     [rax], rsp
  44
  45         mov     rsp, rax          ; /* make rsp point to &ar */
  46
  47         mov     rsi, [rsp+16]      ; /* rsi  = in */
  48         mov     rdi, [rsp+32]      ; /* rdi  = out */
  49         mov     r9, [rsp+24]       ; /* r9   = last */
  50         mov     r10, [rsp+48]      ; /* r10  = end */
  51         mov     rbp, [rsp+64]      ; /* rbp  = lcode */
  52         mov     r11, [rsp+72]      ; /* r11  = dcode */
  53         mov     rdx, [rsp+80]      ; /* rdx  = hold */
  54         mov     ebx, [rsp+88]      ; /* ebx  = bits */
  55         mov     r12d, [rsp+100]    ; /* r12d = lmask */
  56         mov     r13d, [rsp+104]    ; /* r13d = dmask */
  57                                           ; /* r14d = len */
  58                                           ; /* r15d = dist */
  59
  60
  61         cld
  62         cmp     r10, rdi
  63         je      L_one_time           ; /* if only one decode left */
  64         cmp     r9, rsi
  65
  66     jne L_do_loop
  67
  68
  69 L_one_time:
  70         mov     r8, r12           ; /* r8 = lmask */
  71         cmp     bl, 32
  72         ja      L_get_length_code_one_time
  73
  74         lodsd                         ; /* eax = *(uint *)in++ */
  75         mov     cl, bl            ; /* cl = bits, needs it for shifting */
  76         add     bl, 32             ; /* bits += 32 */
  77         shl     rax, cl
  78         or      rdx, rax          ; /* hold |= *((uint *)in)++ << bits */
  79         jmp     L_get_length_code_one_time
  80
  81 ALIGN 4
  82 L_while_test:
  83         cmp     r10, rdi
  84         jbe     L_break_loop
  85         cmp     r9, rsi
  86         jbe     L_break_loop
  87
  88 L_do_loop:
  89         mov     r8, r12           ; /* r8 = lmask */
  90         cmp     bl, 32
  91         ja      L_get_length_code    ; /* if (32 < bits) */
  92
  93         lodsd                         ; /* eax = *(uint *)in++ */
  94         mov     cl, bl            ; /* cl = bits, needs it for shifting */
  95         add     bl, 32             ; /* bits += 32 */
  96         shl     rax, cl
  97         or      rdx, rax          ; /* hold |= *((uint *)in)++ << bits */
  98
  99 L_get_length_code:
 100         and     r8, rdx            ; /* r8 &= hold */
 101         mov     eax, [rbp+r8*4]  ; /* eax = lcode[hold & lmask] */
 102
 103         mov     cl, ah            ; /* cl = this.bits */
 104         sub     bl, ah            ; /* bits -= this.bits */
 105         shr     rdx, cl           ; /* hold >>= this.bits */
 106
 107         test    al, al
 108         jnz     L_test_for_length_base ; /* if (op != 0) 45.7% */
 109
 110         mov     r8, r12            ; /* r8 = lmask */
 111         shr     eax, 16            ; /* output this.val char */
 112         stosb
 113
 114 L_get_length_code_one_time:
 115         and     r8, rdx            ; /* r8 &= hold */
 116         mov     eax, [rbp+r8*4] ; /* eax = lcode[hold & lmask] */
 117
 118 L_dolen:
 119         mov     cl, ah            ; /* cl = this.bits */
 120         sub     bl, ah            ; /* bits -= this.bits */
 121         shr     rdx, cl           ; /* hold >>= this.bits */
 122
 123         test    al, al
 124         jnz     L_test_for_length_base ; /* if (op != 0) 45.7% */
 125
 126         shr     eax, 16            ; /* output this.val char */
 127         stosb
 128         jmp     L_while_test
 129
 130 ALIGN 4
 131 L_test_for_length_base:
 132         mov     r14d, eax         ; /* len = this */
 133         shr     r14d, 16           ; /* len = this.val */
 134         mov     cl, al
 135
 136         test    al, 16
 137         jz      L_test_for_second_level_length ; /* if ((op & 16) == 0) 8% */
 138         and     cl, 15             ; /* op &= 15 */
 139         jz      L_decode_distance    ; /* if (!op) */
 140
 141 L_add_bits_to_len:
 142         sub     bl, cl
 143         xor     eax, eax
 144         inc     eax
 145         shl     eax, cl
 146         dec     eax
 147         and     eax, edx          ; /* eax &= hold */
 148         shr     rdx, cl
 149         add     r14d, eax         ; /* len += hold & mask[op] */
 150
 151 L_decode_distance:
 152         mov     r8, r13           ; /* r8 = dmask */
 153         cmp     bl, 32
 154         ja      L_get_distance_code  ; /* if (32 < bits) */
 155
 156         lodsd                         ; /* eax = *(uint *)in++ */
 157         mov     cl, bl            ; /* cl = bits, needs it for shifting */
 158         add     bl, 32             ; /* bits += 32 */
 159         shl     rax, cl
 160         or      rdx, rax          ; /* hold |= *((uint *)in)++ << bits */
 161
 162 L_get_distance_code:
 163         and     r8, rdx           ; /* r8 &= hold */
 164         mov     eax, [r11+r8*4] ; /* eax = dcode[hold & dmask] */
 165
 166 L_dodist:
 167         mov     r15d, eax         ; /* dist = this */
 168         shr     r15d, 16           ; /* dist = this.val */
 169         mov     cl, ah
 170         sub     bl, ah            ; /* bits -= this.bits */
 171         shr     rdx, cl           ; /* hold >>= this.bits */
 172         mov     cl, al            ; /* cl = this.op */
 173
 174         test    al, 16             ; /* if ((op & 16) == 0) */
 175         jz      L_test_for_second_level_dist
 176         and     cl, 15             ; /* op &= 15 */
 177         jz      L_check_dist_one
 178
 179 L_add_bits_to_dist:
 180         sub     bl, cl
 181         xor     eax, eax
 182         inc     eax
 183         shl     eax, cl
 184         dec     eax                 ; /* (1 << op) - 1 */
 185         and     eax, edx          ; /* eax &= hold */
 186         shr     rdx, cl
 187         add     r15d, eax         ; /* dist += hold & ((1 << op) - 1) */
 188
 189 L_check_window:
 190         mov     r8, rsi           ; /* save in so from can use it's reg */
 191         mov     rax, rdi
 192         sub     rax, [rsp+40]      ; /* nbytes = out - beg */
 193
 194         cmp     eax, r15d
 195         jb      L_clip_window        ; /* if (dist > nbytes) 4.2% */
 196
 197         mov     ecx, r14d         ; /* ecx = len */
 198         mov     rsi, rdi
 199         sub     rsi, r15          ; /* from = out - dist */
 200
 201         sar     ecx, 1
 202         jnc     L_copy_two           ; /* if len % 2 == 0 */
 203
 204         rep     movsw
 205         mov     al, [rsi]
 206         mov     [rdi], al
 207         inc     rdi
 208
 209         mov     rsi, r8           ; /* move in back to %rsi, toss from */
 210         jmp     L_while_test
 211
 212 L_copy_two:
 213         rep     movsw
 214         mov     rsi, r8           ; /* move in back to %rsi, toss from */
 215         jmp     L_while_test
 216
 217 ALIGN 4
 218 L_check_dist_one:
 219         cmp     r15d, 1            ; /* if dist 1, is a memset */
 220         jne     L_check_window
 221         cmp     [rsp+40], rdi      ; /* if out == beg, outside window */
 222         je      L_check_window
 223
 224         mov     ecx, r14d         ; /* ecx = len */
 225         mov     al, [rdi-1]
 226         mov     ah, al
 227
 228         sar     ecx, 1
 229         jnc     L_set_two
 230         mov     [rdi], al
 231         inc     rdi
 232
 233 L_set_two:
 234         rep     stosw
 235         jmp     L_while_test
 236
 237 ALIGN 4
 238 L_test_for_second_level_length:
 239         test    al, 64
 240         jnz     L_test_for_end_of_block ; /* if ((op & 64) != 0) */
 241
 242         xor     eax, eax
 243         inc     eax
 244         shl     eax, cl
 245         dec     eax
 246         and     eax, edx         ; /* eax &= hold */
 247         add     eax, r14d        ; /* eax += len */
 248         mov     eax, [rbp+rax*4] ; /* eax = lcode[val+(hold&mask[op])]*/
 249         jmp     L_dolen
 250
 251 ALIGN 4
 252 L_test_for_second_level_dist:
 253         test    al, 64
 254         jnz     L_invalid_distance_code ; /* if ((op & 64) != 0) */
 255
 256         xor     eax, eax
 257         inc     eax
 258         shl     eax, cl
 259         dec     eax
 260         and     eax, edx         ; /* eax &= hold */
 261         add     eax, r15d        ; /* eax += dist */
 262         mov     eax, [r11+rax*4] ; /* eax = dcode[val+(hold&mask[op])]*/
 263         jmp     L_dodist
 264
 265 ALIGN 4
 266 L_clip_window:
 267         mov     ecx, eax         ; /* ecx = nbytes */
 268         mov     eax, [rsp+92]     ; /* eax = wsize, prepare for dist cmp */
 269         neg     ecx                ; /* nbytes = -nbytes */
 270
 271         cmp     eax, r15d
 272         jb      L_invalid_distance_too_far ; /* if (dist > wsize) */
 273
 274         add     ecx, r15d         ; /* nbytes = dist - nbytes */
 275         cmp     dword ptr [rsp+96], 0
 276         jne     L_wrap_around_window ; /* if (write != 0) */
 277
 278         mov     rsi, [rsp+56]     ; /* from  = window */
 279         sub     eax, ecx         ; /* eax  -= nbytes */
 280         add     rsi, rax         ; /* from += wsize - nbytes */
 281
 282         mov     eax, r14d        ; /* eax = len */
 283         cmp     r14d, ecx
 284         jbe     L_do_copy           ; /* if (nbytes >= len) */
 285
 286         sub     eax, ecx         ; /* eax -= nbytes */
 287         rep     movsb
 288         mov     rsi, rdi
 289         sub     rsi, r15         ; /* from = &out[ -dist ] */
 290         jmp     L_do_copy
 291
 292 ALIGN 4
 293 L_wrap_around_window:
 294         mov     eax, [rsp+96]     ; /* eax = write */
 295         cmp     ecx, eax
 296         jbe     L_contiguous_in_window ; /* if (write >= nbytes) */
 297
 298         mov     esi, [rsp+92]     ; /* from  = wsize */
 299         add     rsi, [rsp+56]     ; /* from += window */
 300         add     rsi, rax         ; /* from += write */
 301         sub     rsi, rcx         ; /* from -= nbytes */
 302         sub     ecx, eax         ; /* nbytes -= write */
 303
 304         mov     eax, r14d        ; /* eax = len */
 305         cmp     eax, ecx
 306         jbe     L_do_copy           ; /* if (nbytes >= len) */
 307
 308         sub     eax, ecx         ; /* len -= nbytes */
 309         rep     movsb
 310         mov     rsi, [rsp+56]     ; /* from = window */
 311         mov     ecx, [rsp+96]     ; /* nbytes = write */
 312         cmp     eax, ecx
 313         jbe     L_do_copy           ; /* if (nbytes >= len) */
 314
 315         sub     eax, ecx         ; /* len -= nbytes */
 316         rep     movsb
 317         mov     rsi, rdi
 318         sub     rsi, r15         ; /* from = out - dist */
 319         jmp     L_do_copy
 320
 321 ALIGN 4
 322 L_contiguous_in_window:
 323         mov     rsi, [rsp+56]     ; /* rsi = window */
 324         add     rsi, rax
 325         sub     rsi, rcx         ; /* from += write - nbytes */
 326
 327         mov     eax, r14d        ; /* eax = len */
 328         cmp     eax, ecx
 329         jbe     L_do_copy           ; /* if (nbytes >= len) */
 330
 331         sub     eax, ecx         ; /* len -= nbytes */
 332         rep     movsb
 333         mov     rsi, rdi
 334         sub     rsi, r15         ; /* from = out - dist */
 335         jmp     L_do_copy           ; /* if (nbytes >= len) */
 336
 337 ALIGN 4
 338 L_do_copy:
 339         mov     ecx, eax         ; /* ecx = len */
 340         rep     movsb
 341
 342         mov     rsi, r8          ; /* move in back to %esi, toss from */
 343         jmp     L_while_test
 344
 345 L_test_for_end_of_block:
 346         test    al, 32
 347         jz      L_invalid_literal_length_code
 348         mov     dword ptr [rsp+116], 1
 349         jmp     L_break_loop_with_status
 350
 351 L_invalid_literal_length_code:
 352         mov     dword ptr [rsp+116], 2
 353         jmp     L_break_loop_with_status
 354
 355 L_invalid_distance_code:
 356         mov     dword ptr [rsp+116], 3
 357         jmp     L_break_loop_with_status
 358
 359 L_invalid_distance_too_far:
 360         mov     dword ptr [rsp+116], 4
 361         jmp     L_break_loop_with_status
 362
 363 L_break_loop:
 364         mov     dword ptr [rsp+116], 0
 365
 366 L_break_loop_with_status:
 367 ; /* put in, out, bits, and hold back into ar and pop esp */
 368         mov     [rsp+16], rsi     ; /* in */
 369         mov     [rsp+32], rdi     ; /* out */
 370         mov     [rsp+88], ebx     ; /* bits */
 371         mov     [rsp+80], rdx     ; /* hold */
 372
 373         mov     rax, [rsp]       ; /* restore rbp and rsp */
 374         mov     rbp, [rsp+8]
 375         mov     rsp, rax
 376
 377
 378
 379         mov rsi,[rsp-8]
 380         mov rdi,[rsp-16]
 381         mov r12,[rsp-24]
 382         mov r13,[rsp-32]
 383         mov r14,[rsp-40]
 384         mov r15,[rsp-48]
 385         mov rbx,[rsp-56]
 386
 387     ret 0
 388 ;          :
 389 ;          : "m" (ar)
 390 ;          : "memory", "%rax", "%rbx", "%rcx", "%rdx", "%rsi", "%rdi",
 391 ;            "%r8", "%r9", "%r10", "%r11", "%r12", "%r13", "%r14", "%r15"
 392 ;    );
 393
 394 inffas8664fnc   ENDP
 395 ;_TEXT  ENDS
 396 END