;/* inffas32.asm is a hand tuned assembler version of inffast.c -- fast decoding

; *

; * inffas32.asm is derivated from inffas86.c, with translation of assembly code

; *

; * Copyright (C) 1995-2003 Mark Adler

; * For conditions of distribution and use, see copyright notice in zlib.h

; *

; * Copyright (C) 2003 Chris Anderson <christop@charm.net></christop@charm.net>

; * Please use the copyright conditions above.

; *

; * Mar-13-2003 -- Most of this is derived from inffast.S which is derived from

; * the gcc -S output of zlib-1.2.0/inffast.c.  Zlib-1.2.0 is in beta release at

; * the moment.  I have successfully compiled and tested this code with gcc2.96,

; * gcc3.2, icc5.0, msvc6.0.  It is very close to the speed of inffast.S

; * compiled with gcc -DNO_MMX, but inffast.S is still faster on the P3 with MMX

; * enabled.  I will attempt to merge the MMX code into this version.  Newer

; * versions of this and inffast.S can be found at

; * http://www.eetbeetee.com/zlib/ and http://www.charm.net/~christop/zlib/

; *

; * 2005 : modification by Gilles Vollant

; */

; For Visual C++ 4.x and higher and ML 6.x and higher

;   ml.exe is in directory \MASM611C of Win95 DDK

;   ml.exe is also distributed in http://www.masm32.com/masmdl.htm

;    and in VC++2003 toolkit at http://msdn.microsoft.com/visualc/vctoolkit2003/

;

;

;   compile with command line option

;   ml  /coff /Zi /c /Flinffas32.lst inffas32.asm

;   if you define NO_GZIP (see inflate.h), compile with

;   ml  /coff /Zi /c /Flinffas32.lst /DNO_GUNZIP inffas32.asm

; zlib122sup is 0 fort zlib 1.2.2.1 and lower

; zlib122sup is 8 fort zlib 1.2.2.2 and more (with addition of dmax and head

;        in inflate_state in inflate.h)

zlib1222sup      equ    8

IFDEF GUNZIP

  INFLATE_MODE_TYPE    equ 11

  INFLATE_MODE_BAD     equ 26

ELSE

  IFNDEF NO_GUNZIP

    INFLATE_MODE_TYPE    equ 11

    INFLATE_MODE_BAD     equ 26

  ELSE

    INFLATE_MODE_TYPE    equ 3

    INFLATE_MODE_BAD     equ 17

  ENDIF

ENDIF

; 75 "inffast.S"

;FILE "inffast.S"

;;;GLOBAL _inflate_fast

;;;SECTION .text

	.586p

	.mmx

	name	inflate_fast_x86

	.MODEL	FLAT

_DATA			segment

inflate_fast_use_mmx:

	dd	1

_TEXT			segment

ALIGN 4

	db	'Fast decoding Code from Chris Anderson'

	db	0

ALIGN 4

invalid_literal_length_code_msg:

	db	'invalid literal/length code'

	db	0

ALIGN 4

invalid_distance_code_msg:

	db	'invalid distance code'

	db	0

ALIGN 4

invalid_distance_too_far_msg:

	db	'invalid distance too far back'

	db	0

ALIGN 4

inflate_fast_mask:

dd	0

dd	1

dd	3

dd	7

dd	15

dd	31

dd	63

dd	127

dd	255

dd	511

dd	1023

dd	2047

dd	4095

dd	8191

dd	16383

dd	32767

dd	65535

dd	131071

dd	262143

dd	524287

dd	1048575

dd	2097151

dd	4194303

dd	8388607

dd	16777215

dd	33554431

dd	67108863

dd	134217727

dd	268435455

dd	536870911

dd	1073741823

dd	2147483647

dd	4294967295

mode_state	 equ	0	;/* state->mode	*/

wsize_state	 equ	(32+zlib1222sup)	;/* state->wsize */

write_state	 equ	(36+4+zlib1222sup)	;/* state->write */

window_state	 equ	(40+4+zlib1222sup)	;/* state->window */

hold_state	 equ	(44+4+zlib1222sup)	;/* state->hold	*/

bits_state	 equ	(48+4+zlib1222sup)	;/* state->bits	*/

lencode_state	 equ	(64+4+zlib1222sup)	;/* state->lencode */

distcode_state	 equ	(68+4+zlib1222sup)	;/* state->distcode */

lenbits_state	 equ	(72+4+zlib1222sup)	;/* state->lenbits */

distbits_state	 equ	(76+4+zlib1222sup)	;/* state->distbits */

;;SECTION .text

; 205 "inffast.S"

;GLOBAL	inflate_fast_use_mmx

;SECTION .data

; GLOBAL inflate_fast_use_mmx:object

;.size inflate_fast_use_mmx, 4

; 226 "inffast.S"

;SECTION .text

ALIGN 4

_inflate_fast proc near

.FPO (16, 4, 0, 0, 1, 0)

	push  edi

	push  esi

	push  ebp

	push  ebx

	pushfd

	sub  esp,64

	cld

	mov  esi, [esp+88]

	mov  edi, [esi+28]

	mov  edx, [esi+4]

	mov  eax, [esi+0]

	add  edx,eax

	sub  edx,11

	mov  [esp+44],eax

	mov  [esp+20],edx

	mov  ebp, [esp+92]

	mov  ecx, [esi+16]

	mov  ebx, [esi+12]

	sub  ebp,ecx

	neg  ebp

	add  ebp,ebx

	sub  ecx,257

	add  ecx,ebx

	mov  [esp+60],ebx

	mov  [esp+40],ebp

	mov  [esp+16],ecx

; 285 "inffast.S"

	mov  eax, [edi+lencode_state]

	mov  ecx, [edi+distcode_state]

	mov  [esp+8],eax

	mov  [esp+12],ecx

	mov  eax,1

	mov  ecx, [edi+lenbits_state]

	shl  eax,cl

	dec  eax

	mov  [esp+0],eax

	mov  eax,1

	mov  ecx, [edi+distbits_state]

	shl  eax,cl

	dec  eax

	mov  [esp+4],eax

	mov  eax, [edi+wsize_state]

	mov  ecx, [edi+write_state]

	mov  edx, [edi+window_state]

	mov  [esp+52],eax

	mov  [esp+48],ecx

	mov  [esp+56],edx

	mov  ebp, [edi+hold_state]

	mov  ebx, [edi+bits_state]

; 321 "inffast.S"

	mov  esi, [esp+44]

	mov  ecx, [esp+20]

	cmp  ecx,esi

	ja   L_align_long

	add  ecx,11

	sub  ecx,esi

	mov  eax,12

	sub  eax,ecx

	lea  edi, [esp+28]

	rep movsb

	mov  ecx,eax

	xor  eax,eax

	rep stosb

	lea  esi, [esp+28]

	mov  [esp+20],esi

	jmp  L_is_aligned

L_align_long:

	test  esi,3

	jz   L_is_aligned

	xor  eax,eax

	mov  al, [esi]

	inc  esi

	mov  ecx,ebx

	add  ebx,8

	shl  eax,cl

	or  ebp,eax

	jmp L_align_long

L_is_aligned:

	mov  edi, [esp+60]

; 366 "inffast.S"

L_check_mmx:

	cmp  dword ptr [inflate_fast_use_mmx],2

	je   L_init_mmx

	ja   L_do_loop

	push  eax

	push  ebx

	push  ecx

	push  edx

	pushfd

	mov  eax, [esp]

	xor  dword ptr [esp],0200000h

	popfd

	pushfd

	pop  edx

	xor  edx,eax

	jz   L_dont_use_mmx

	xor  eax,eax

	cpuid

	cmp  ebx,0756e6547h

	jne  L_dont_use_mmx

	cmp  ecx,06c65746eh

	jne  L_dont_use_mmx

	cmp  edx,049656e69h

	jne  L_dont_use_mmx

	mov  eax,1

	cpuid

	shr  eax,8

	and  eax,15

	cmp  eax,6

	jne  L_dont_use_mmx

	test  edx,0800000h

	jnz  L_use_mmx

	jmp  L_dont_use_mmx

L_use_mmx:

	mov  dword ptr [inflate_fast_use_mmx],2

	jmp  L_check_mmx_pop

L_dont_use_mmx:

	mov  dword ptr [inflate_fast_use_mmx],3

L_check_mmx_pop:

	pop  edx

	pop  ecx

	pop  ebx

	pop  eax

	jmp  L_check_mmx

; 426 "inffast.S"

ALIGN 4

L_do_loop:

; 437 "inffast.S"

	cmp  bl,15

	ja   L_get_length_code

	xor  eax,eax

	lodsw

	mov  cl,bl

	add  bl,16

	shl  eax,cl

	or  ebp,eax

L_get_length_code:

	mov  edx, [esp+0]

	mov  ecx, [esp+8]

	and  edx,ebp

	mov  eax, [ecx+edx*4]

L_dolen:

	mov  cl,ah

	sub  bl,ah

	shr  ebp,cl

	test  al,al

	jnz   L_test_for_length_base

	shr  eax,16

	stosb

L_while_test:

	cmp  [esp+16],edi

	jbe  L_break_loop

	cmp  [esp+20],esi

	ja   L_do_loop

	jmp  L_break_loop

L_test_for_length_base:

; 502 "inffast.S"

	mov  edx,eax

	shr  edx,16

	mov  cl,al

	test  al,16

	jz   L_test_for_second_level_length

	and  cl,15

	jz   L_save_len

	cmp  bl,cl

	jae  L_add_bits_to_len

	mov  ch,cl

	xor  eax,eax

	lodsw

	mov  cl,bl

	add  bl,16

	shl  eax,cl

	or  ebp,eax

	mov  cl,ch

L_add_bits_to_len:

	mov  eax,1

	shl  eax,cl

	dec  eax

	sub  bl,cl

	and  eax,ebp

	shr  ebp,cl

	add  edx,eax

L_save_len:

	mov  [esp+24],edx

L_decode_distance:

; 549 "inffast.S"

	cmp  bl,15

	ja   L_get_distance_code

	xor  eax,eax

	lodsw

	mov  cl,bl

	add  bl,16

	shl  eax,cl

	or  ebp,eax

L_get_distance_code:

	mov  edx, [esp+4]

	mov  ecx, [esp+12]

	and  edx,ebp

	mov  eax, [ecx+edx*4]

L_dodist:

	mov  edx,eax

	shr  edx,16

	mov  cl,ah

	sub  bl,ah

	shr  ebp,cl

; 584 "inffast.S"

	mov  cl,al

	test  al,16

	jz  L_test_for_second_level_dist

	and  cl,15

	jz  L_check_dist_one

	cmp  bl,cl

	jae  L_add_bits_to_dist

	mov  ch,cl

	xor  eax,eax

	lodsw

	mov  cl,bl

	add  bl,16

	shl  eax,cl

	or  ebp,eax

	mov  cl,ch

L_add_bits_to_dist:

	mov  eax,1

	shl  eax,cl

	dec  eax

	sub  bl,cl

	and  eax,ebp

	shr  ebp,cl

	add  edx,eax

	jmp  L_check_window

L_check_window:

; 625 "inffast.S"

	mov  [esp+44],esi

	mov  eax,edi

	sub  eax, [esp+40]

	cmp  eax,edx

	jb   L_clip_window

	mov  ecx, [esp+24]

	mov  esi,edi

	sub  esi,edx

	sub  ecx,3

	mov  al, [esi]

	mov  [edi],al

	mov  al, [esi+1]

	mov  dl, [esi+2]

	add  esi,3

	mov  [edi+1],al

	mov  [edi+2],dl

	add  edi,3

	rep movsb

	mov  esi, [esp+44]

	jmp  L_while_test

ALIGN 4

L_check_dist_one:

	cmp  edx,1

	jne  L_check_window

	cmp  [esp+40],edi

	je  L_check_window

	dec  edi

	mov  ecx, [esp+24]

	mov  al, [edi]

	sub  ecx,3

	mov  [edi+1],al

	mov  [edi+2],al

	mov  [edi+3],al

	add  edi,4

	rep stosb

	jmp  L_while_test

ALIGN 4

L_test_for_second_level_length:

	test  al,64

	jnz   L_test_for_end_of_block

	mov  eax,1

	shl  eax,cl

	dec  eax

	and  eax,ebp

	add  eax,edx

	mov  edx, [esp+8]

	mov  eax, [edx+eax*4]

	jmp  L_dolen

ALIGN 4

L_test_for_second_level_dist:

	test  al,64

	jnz   L_invalid_distance_code

	mov  eax,1

	shl  eax,cl

	dec  eax

	and  eax,ebp

	add  eax,edx

	mov  edx, [esp+12]

	mov  eax, [edx+eax*4]

	jmp  L_dodist

ALIGN 4

L_clip_window:

; 721 "inffast.S"

	mov  ecx,eax

	mov  eax, [esp+52]

	neg  ecx

	mov  esi, [esp+56]

	cmp  eax,edx

	jb   L_invalid_distance_too_far

	add  ecx,edx

	cmp  dword ptr [esp+48],0

	jne  L_wrap_around_window

	sub  eax,ecx

	add  esi,eax

; 749 "inffast.S"

	mov  eax, [esp+24]

	cmp  eax,ecx

	jbe  L_do_copy1

	sub  eax,ecx

	rep movsb

	mov  esi,edi

	sub  esi,edx

	jmp  L_do_copy1

	cmp  eax,ecx

	jbe  L_do_copy1

	sub  eax,ecx

	rep movsb

	mov  esi,edi

	sub  esi,edx

	jmp  L_do_copy1

L_wrap_around_window:

; 793 "inffast.S"

	mov  eax, [esp+48]

	cmp  ecx,eax

	jbe  L_contiguous_in_window

	add  esi, [esp+52]

	add  esi,eax

	sub  esi,ecx

	sub  ecx,eax

	mov  eax, [esp+24]

	cmp  eax,ecx

	jbe  L_do_copy1

	sub  eax,ecx

	rep movsb

	mov  esi, [esp+56]

	mov  ecx, [esp+48]

	cmp  eax,ecx

	jbe  L_do_copy1

	sub  eax,ecx

	rep movsb

	mov  esi,edi

	sub  esi,edx

	jmp  L_do_copy1

L_contiguous_in_window:

; 836 "inffast.S"

	add  esi,eax

	sub  esi,ecx

	mov  eax, [esp+24]

	cmp  eax,ecx

	jbe  L_do_copy1

	sub  eax,ecx

	rep movsb

	mov  esi,edi

	sub  esi,edx

L_do_copy1:

; 862 "inffast.S"

	mov  ecx,eax

	rep movsb

	mov  esi, [esp+44]

	jmp  L_while_test

; 878 "inffast.S"

ALIGN 4

L_init_mmx:

	emms

	movd mm0,ebp

	mov  ebp,ebx

; 896 "inffast.S"

	movd mm4,dword ptr [esp+0]

	movq mm3,mm4

	movd mm5,dword ptr [esp+4]

	movq mm2,mm5

	pxor mm1,mm1

	mov  ebx, [esp+8]

	jmp  L_do_loop_mmx

ALIGN 4

L_do_loop_mmx:

	psrlq mm0,mm1

	cmp  ebp,32

	ja  L_get_length_code_mmx

	movd mm6,ebp

	movd mm7,dword ptr [esi]

	add  esi,4

	psllq mm7,mm6

	add  ebp,32

	por mm0,mm7

L_get_length_code_mmx:

	pand mm4,mm0

	movd eax,mm4

	movq mm4,mm3

	mov  eax, [ebx+eax*4]

L_dolen_mmx:

	movzx  ecx,ah

	movd mm1,ecx

	sub  ebp,ecx

	test  al,al

	jnz L_test_for_length_base_mmx

	shr  eax,16

	stosb

L_while_test_mmx:

	cmp  [esp+16],edi

	jbe L_break_loop

	cmp  [esp+20],esi

	ja L_do_loop_mmx

	jmp L_break_loop

L_test_for_length_base_mmx:

	mov  edx,eax

	shr  edx,16

	test  al,16

	jz  L_test_for_second_level_length_mmx

	and  eax,15

	jz L_decode_distance_mmx

	psrlq mm0,mm1

	movd mm1,eax

	movd ecx,mm0

	sub  ebp,eax

	and  ecx, [inflate_fast_mask+eax*4]

	add  edx,ecx

L_decode_distance_mmx:

	psrlq mm0,mm1

	cmp  ebp,32

	ja L_get_dist_code_mmx

	movd mm6,ebp

	movd mm7,dword ptr [esi]

	add  esi,4

	psllq mm7,mm6

	add  ebp,32

	por mm0,mm7

L_get_dist_code_mmx:

	mov  ebx, [esp+12]

	pand mm5,mm0

	movd eax,mm5

	movq mm5,mm2

	mov  eax, [ebx+eax*4]

L_dodist_mmx:

	movzx  ecx,ah

	mov  ebx,eax

	shr  ebx,16

	sub  ebp,ecx

	movd mm1,ecx

	test  al,16

	jz L_test_for_second_level_dist_mmx

	and  eax,15

	jz L_check_dist_one_mmx

L_add_bits_to_dist_mmx:

	psrlq mm0,mm1

	movd mm1,eax

	movd ecx,mm0

	sub  ebp,eax

	and  ecx, [inflate_fast_mask+eax*4]

	add  ebx,ecx

L_check_window_mmx:

	mov  [esp+44],esi

	mov  eax,edi

	sub  eax, [esp+40]

	cmp  eax,ebx

	jb L_clip_window_mmx

	mov  ecx,edx

	mov  esi,edi

	sub  esi,ebx

	sub  ecx,3

	mov  al, [esi]

	mov  [edi],al

	mov  al, [esi+1]

	mov  dl, [esi+2]

	add  esi,3

	mov  [edi+1],al

	mov  [edi+2],dl

	add  edi,3

	rep movsb

	mov  esi, [esp+44]

	mov  ebx, [esp+8]

	jmp  L_while_test_mmx

ALIGN 4

L_check_dist_one_mmx:

	cmp  ebx,1

	jne  L_check_window_mmx

	cmp  [esp+40],edi

	je   L_check_window_mmx

	dec  edi

	mov  ecx,edx

	mov  al, [edi]

	sub  ecx,3

	mov  [edi+1],al

	mov  [edi+2],al

	mov  [edi+3],al

	add  edi,4

	rep stosb

	mov  ebx, [esp+8]

	jmp  L_while_test_mmx

ALIGN 4

L_test_for_second_level_length_mmx:

	test  al,64

	jnz L_test_for_end_of_block

	and  eax,15

	psrlq mm0,mm1

	movd ecx,mm0

	and  ecx, [inflate_fast_mask+eax*4]

	add  ecx,edx

	mov  eax, [ebx+ecx*4]

	jmp L_dolen_mmx

ALIGN 4

L_test_for_second_level_dist_mmx:

	test  al,64

	jnz L_invalid_distance_code

	and  eax,15

	psrlq mm0,mm1

	movd ecx,mm0

	and  ecx, [inflate_fast_mask+eax*4]

	mov  eax, [esp+12]

	add  ecx,ebx

	mov  eax, [eax+ecx*4]

	jmp  L_dodist_mmx

ALIGN 4

L_clip_window_mmx:

	mov  ecx,eax

	mov  eax, [esp+52]

	neg  ecx

	mov  esi, [esp+56]

	cmp  eax,ebx

	jb  L_invalid_distance_too_far

	add  ecx,ebx

	cmp  dword ptr [esp+48],0

	jne  L_wrap_around_window_mmx

	sub  eax,ecx

	add  esi,eax

	cmp  edx,ecx

	jbe  L_do_copy1_mmx

	sub  edx,ecx

	rep movsb

	mov  esi,edi

	sub  esi,ebx

	jmp  L_do_copy1_mmx

	cmp  edx,ecx

	jbe  L_do_copy1_mmx

	sub  edx,ecx

	rep movsb

	mov  esi,edi

	sub  esi,ebx

	jmp  L_do_copy1_mmx

L_wrap_around_window_mmx:

	mov  eax, [esp+48]

	cmp  ecx,eax

	jbe  L_contiguous_in_window_mmx

	add  esi, [esp+52]

	add  esi,eax

	sub  esi,ecx

	sub  ecx,eax

	cmp  edx,ecx

	jbe  L_do_copy1_mmx

	sub  edx,ecx

	rep movsb

	mov  esi, [esp+56]