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