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 393
