| 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
|
|---|
