| ; inffasx64.asm is a hand tuned assembler version of inffast.c - fast decoding |
| ; version for AMD64 on Windows using Microsoft C compiler |
| ; |
| ; inffasx64.asm is automatically convert from AMD64 portion of inffas86.c |
| ; inffasx64.asm is called by inffas8664.c, which contain more info. |
| |
| |
| ; to compile this file, I use option |
| ; ml64.exe /Flinffasx64 /c /Zi inffasx64.asm |
| ; with Microsoft Macro Assembler (x64) for AMD64 |
| ; |
| |
| ; This file compile with Microsoft Macro Assembler (x64) for AMD64 |
| ; |
| ; ml64.exe is given with Visual Studio 2005/2008/2010 and Windows WDK |
| ; |
| ; (you can get Windows WDK with ml64 for AMD64 from |
| ; http://www.microsoft.com/whdc/Devtools/wdk/default.mspx for low price) |
| ; |
| |
| |
| .code |
| inffas8664fnc PROC |
| |
| ; see http://weblogs.asp.net/oldnewthing/archive/2004/01/14/58579.aspx and |
| ; http://msdn.microsoft.com/library/en-us/kmarch/hh/kmarch/64bitAMD_8e951dd2-ee77-4728-8702-55ce4b5dd24a.xml.asp |
| ; |
| ; All registers must be preserved across the call, except for |
| ; rax, rcx, rdx, r8, r-9, r10, and r11, which are scratch. |
| |
| |
| mov [rsp-8],rsi |
| mov [rsp-16],rdi |
| mov [rsp-24],r12 |
| mov [rsp-32],r13 |
| mov [rsp-40],r14 |
| mov [rsp-48],r15 |
| mov [rsp-56],rbx |
| |
| mov rax,rcx |
| |
| mov [rax+8], rbp ; /* save regs rbp and rsp */ |
| mov [rax], rsp |
| |
| mov rsp, rax ; /* make rsp point to &ar */ |
| |
| mov rsi, [rsp+16] ; /* rsi = in */ |
| mov rdi, [rsp+32] ; /* rdi = out */ |
| mov r9, [rsp+24] ; /* r9 = last */ |
| mov r10, [rsp+48] ; /* r10 = end */ |
| mov rbp, [rsp+64] ; /* rbp = lcode */ |
| mov r11, [rsp+72] ; /* r11 = dcode */ |
| mov rdx, [rsp+80] ; /* rdx = hold */ |
| mov ebx, [rsp+88] ; /* ebx = bits */ |
| mov r12d, [rsp+100] ; /* r12d = lmask */ |
| mov r13d, [rsp+104] ; /* r13d = dmask */ |
| ; /* r14d = len */ |
| ; /* r15d = dist */ |
| |
| |
| cld |
| cmp r10, rdi |
| je L_one_time ; /* if only one decode left */ |
| cmp r9, rsi |
| |
| jne L_do_loop |
| |
| |
| L_one_time: |
| mov r8, r12 ; /* r8 = lmask */ |
| cmp bl, 32 |
| ja L_get_length_code_one_time |
| |
| lodsd ; /* eax = *(uint *)in++ */ |
| mov cl, bl ; /* cl = bits, needs it for shifting */ |
| add bl, 32 ; /* bits += 32 */ |
| shl rax, cl |
| or rdx, rax ; /* hold |= *((uint *)in)++ << bits */ |
| jmp L_get_length_code_one_time |
| |
| ALIGN 4 |
| L_while_test: |
| cmp r10, rdi |
| jbe L_break_loop |
| cmp r9, rsi |
| jbe L_break_loop |
| |
| L_do_loop: |
| mov r8, r12 ; /* r8 = lmask */ |
| cmp bl, 32 |
| ja L_get_length_code ; /* if (32 < bits) */ |
| |
| lodsd ; /* eax = *(uint *)in++ */ |
| mov cl, bl ; /* cl = bits, needs it for shifting */ |
| add bl, 32 ; /* bits += 32 */ |
| shl rax, cl |
| or rdx, rax ; /* hold |= *((uint *)in)++ << bits */ |
| |
| L_get_length_code: |
| and r8, rdx ; /* r8 &= hold */ |
| mov eax, [rbp+r8*4] ; /* eax = lcode[hold & lmask] */ |
| |
| mov cl, ah ; /* cl = this.bits */ |
| sub bl, ah ; /* bits -= this.bits */ |
| shr rdx, cl ; /* hold >>= this.bits */ |
| |
| test al, al |
| jnz L_test_for_length_base ; /* if (op != 0) 45.7% */ |
| |
| mov r8, r12 ; /* r8 = lmask */ |
| shr eax, 16 ; /* output this.val char */ |
| stosb |
| |
| L_get_length_code_one_time: |
| and r8, rdx ; /* r8 &= hold */ |
| mov eax, [rbp+r8*4] ; /* eax = lcode[hold & lmask] */ |
| |
| L_dolen: |
| mov cl, ah ; /* cl = this.bits */ |
| sub bl, ah ; /* bits -= this.bits */ |
| shr rdx, cl ; /* hold >>= this.bits */ |
| |
| test al, al |
| jnz L_test_for_length_base ; /* if (op != 0) 45.7% */ |
| |
| shr eax, 16 ; /* output this.val char */ |
| stosb |
| jmp L_while_test |
| |
| ALIGN 4 |
| L_test_for_length_base: |
| mov r14d, eax ; /* len = this */ |
| shr r14d, 16 ; /* len = this.val */ |
| mov cl, al |
| |
| test al, 16 |
| jz L_test_for_second_level_length ; /* if ((op & 16) == 0) 8% */ |
| and cl, 15 ; /* op &= 15 */ |
| jz L_decode_distance ; /* if (!op) */ |
| |
| L_add_bits_to_len: |
| sub bl, cl |
| xor eax, eax |
| inc eax |
| shl eax, cl |
| dec eax |
| and eax, edx ; /* eax &= hold */ |
| shr rdx, cl |
| add r14d, eax ; /* len += hold & mask[op] */ |
| |
| L_decode_distance: |
| mov r8, r13 ; /* r8 = dmask */ |
| cmp bl, 32 |
| ja L_get_distance_code ; /* if (32 < bits) */ |
| |
| lodsd ; /* eax = *(uint *)in++ */ |
| mov cl, bl ; /* cl = bits, needs it for shifting */ |
| add bl, 32 ; /* bits += 32 */ |
| shl rax, cl |
| or rdx, rax ; /* hold |= *((uint *)in)++ << bits */ |
| |
| L_get_distance_code: |
| and r8, rdx ; /* r8 &= hold */ |
| mov eax, [r11+r8*4] ; /* eax = dcode[hold & dmask] */ |
| |
| L_dodist: |
| mov r15d, eax ; /* dist = this */ |
| shr r15d, 16 ; /* dist = this.val */ |
| mov cl, ah |
| sub bl, ah ; /* bits -= this.bits */ |
| shr rdx, cl ; /* hold >>= this.bits */ |
| mov cl, al ; /* cl = this.op */ |
| |
| test al, 16 ; /* if ((op & 16) == 0) */ |
| jz L_test_for_second_level_dist |
| and cl, 15 ; /* op &= 15 */ |
| jz L_check_dist_one |
| |
| L_add_bits_to_dist: |
| sub bl, cl |
| xor eax, eax |
| inc eax |
| shl eax, cl |
| dec eax ; /* (1 << op) - 1 */ |
| and eax, edx ; /* eax &= hold */ |
| shr rdx, cl |
| add r15d, eax ; /* dist += hold & ((1 << op) - 1) */ |
| |
| L_check_window: |
| mov r8, rsi ; /* save in so from can use it's reg */ |
| mov rax, rdi |
| sub rax, [rsp+40] ; /* nbytes = out - beg */ |
| |
| cmp eax, r15d |
| jb L_clip_window ; /* if (dist > nbytes) 4.2% */ |
| |
| mov ecx, r14d ; /* ecx = len */ |
| mov rsi, rdi |
| sub rsi, r15 ; /* from = out - dist */ |
| |
| sar ecx, 1 |
| jnc L_copy_two ; /* if len % 2 == 0 */ |
| |
| rep movsw |
| mov al, [rsi] |
| mov [rdi], al |
| inc rdi |
| |
| mov rsi, r8 ; /* move in back to %rsi, toss from */ |
| jmp L_while_test |
| |
| L_copy_two: |
| rep movsw |
| mov rsi, r8 ; /* move in back to %rsi, toss from */ |
| jmp L_while_test |
| |
| ALIGN 4 |
| L_check_dist_one: |
| cmp r15d, 1 ; /* if dist 1, is a memset */ |
| jne L_check_window |
| cmp [rsp+40], rdi ; /* if out == beg, outside window */ |
| je L_check_window |
| |
| mov ecx, r14d ; /* ecx = len */ |
| mov al, [rdi-1] |
| mov ah, al |
| |
| sar ecx, 1 |
| jnc L_set_two |
| mov [rdi], al |
| inc rdi |
| |
| L_set_two: |
| rep stosw |
| jmp L_while_test |
| |
| ALIGN 4 |
| L_test_for_second_level_length: |
| test al, 64 |
| jnz L_test_for_end_of_block ; /* if ((op & 64) != 0) */ |
| |
| xor eax, eax |
| inc eax |
| shl eax, cl |
| dec eax |
| and eax, edx ; /* eax &= hold */ |
| add eax, r14d ; /* eax += len */ |
| mov eax, [rbp+rax*4] ; /* eax = lcode[val+(hold&mask[op])]*/ |
| jmp L_dolen |
| |
| ALIGN 4 |
| L_test_for_second_level_dist: |
| test al, 64 |
| jnz L_invalid_distance_code ; /* if ((op & 64) != 0) */ |
| |
| xor eax, eax |
| inc eax |
| shl eax, cl |
| dec eax |
| and eax, edx ; /* eax &= hold */ |
| add eax, r15d ; /* eax += dist */ |
| mov eax, [r11+rax*4] ; /* eax = dcode[val+(hold&mask[op])]*/ |
| jmp L_dodist |
| |
| ALIGN 4 |
| L_clip_window: |
| mov ecx, eax ; /* ecx = nbytes */ |
| mov eax, [rsp+92] ; /* eax = wsize, prepare for dist cmp */ |
| neg ecx ; /* nbytes = -nbytes */ |
| |
| cmp eax, r15d |
| jb L_invalid_distance_too_far ; /* if (dist > wsize) */ |
| |
| add ecx, r15d ; /* nbytes = dist - nbytes */ |
| cmp dword ptr [rsp+96], 0 |
| jne L_wrap_around_window ; /* if (write != 0) */ |
| |
| mov rsi, [rsp+56] ; /* from = window */ |
| sub eax, ecx ; /* eax -= nbytes */ |
| add rsi, rax ; /* from += wsize - nbytes */ |
| |
| mov eax, r14d ; /* eax = len */ |
| cmp r14d, ecx |
| jbe L_do_copy ; /* if (nbytes >= len) */ |
| |
| sub eax, ecx ; /* eax -= nbytes */ |
| rep movsb |
| mov rsi, rdi |
| sub rsi, r15 ; /* from = &out[ -dist ] */ |
| jmp L_do_copy |
| |
| ALIGN 4 |
| L_wrap_around_window: |
| mov eax, [rsp+96] ; /* eax = write */ |
| cmp ecx, eax |
| jbe L_contiguous_in_window ; /* if (write >= nbytes) */ |
| |
| mov esi, [rsp+92] ; /* from = wsize */ |
| add rsi, [rsp+56] ; /* from += window */ |
| add rsi, rax ; /* from += write */ |
| sub rsi, rcx ; /* from -= nbytes */ |
| sub ecx, eax ; /* nbytes -= write */ |
| |
| mov eax, r14d ; /* eax = len */ |
| cmp eax, ecx |
| jbe L_do_copy ; /* if (nbytes >= len) */ |
| |
| sub eax, ecx ; /* len -= nbytes */ |
| rep movsb |
| mov rsi, [rsp+56] ; /* from = window */ |
| mov ecx, [rsp+96] ; /* nbytes = write */ |
| cmp eax, ecx |
| jbe L_do_copy ; /* if (nbytes >= len) */ |
| |
| sub eax, ecx ; /* len -= nbytes */ |
| rep movsb |
| mov rsi, rdi |
| sub rsi, r15 ; /* from = out - dist */ |
| jmp L_do_copy |
| |
| ALIGN 4 |
| L_contiguous_in_window: |
| mov rsi, [rsp+56] ; /* rsi = window */ |
| add rsi, rax |
| sub rsi, rcx ; /* from += write - nbytes */ |
| |
| mov eax, r14d ; /* eax = len */ |
| cmp eax, ecx |
| jbe L_do_copy ; /* if (nbytes >= len) */ |
| |
| sub eax, ecx ; /* len -= nbytes */ |
| rep movsb |
| mov rsi, rdi |
| sub rsi, r15 ; /* from = out - dist */ |
| jmp L_do_copy ; /* if (nbytes >= len) */ |
| |
| ALIGN 4 |
| L_do_copy: |
| mov ecx, eax ; /* ecx = len */ |
| rep movsb |
| |
| mov rsi, r8 ; /* move in back to %esi, toss from */ |
| jmp L_while_test |
| |
| L_test_for_end_of_block: |
| test al, 32 |
| jz L_invalid_literal_length_code |
| mov dword ptr [rsp+116], 1 |
| jmp L_break_loop_with_status |
| |
| L_invalid_literal_length_code: |
| mov dword ptr [rsp+116], 2 |
| jmp L_break_loop_with_status |
| |
| L_invalid_distance_code: |
| mov dword ptr [rsp+116], 3 |
| jmp L_break_loop_with_status |
| |
| L_invalid_distance_too_far: |
| mov dword ptr [rsp+116], 4 |
| jmp L_break_loop_with_status |
| |
| L_break_loop: |
| mov dword ptr [rsp+116], 0 |
| |
| L_break_loop_with_status: |
| ; /* put in, out, bits, and hold back into ar and pop esp */ |
| mov [rsp+16], rsi ; /* in */ |
| mov [rsp+32], rdi ; /* out */ |
| mov [rsp+88], ebx ; /* bits */ |
| mov [rsp+80], rdx ; /* hold */ |
| |
| mov rax, [rsp] ; /* restore rbp and rsp */ |
| mov rbp, [rsp+8] |
| mov rsp, rax |
| |
| |
| |
| mov rsi,[rsp-8] |
| mov rdi,[rsp-16] |
| mov r12,[rsp-24] |
| mov r13,[rsp-32] |
| mov r14,[rsp-40] |
| mov r15,[rsp-48] |
| mov rbx,[rsp-56] |
| |
| ret 0 |
| ; : |
| ; : "m" (ar) |
| ; : "memory", "%rax", "%rbx", "%rcx", "%rdx", "%rsi", "%rdi", |
| ; "%r8", "%r9", "%r10", "%r11", "%r12", "%r13", "%r14", "%r15" |
| ; ); |
| |
| inffas8664fnc ENDP |
| ;_TEXT ENDS |
| END |