src/third-party/zlib/contrib/masmx64/inffasx64.asm (view raw)
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
23inffas8664fnc 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
69L_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
81ALIGN 4
82L_while_test:
83 cmp r10, rdi
84 jbe L_break_loop
85 cmp r9, rsi
86 jbe L_break_loop
87
88L_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
99L_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
114L_get_length_code_one_time:
115 and r8, rdx ; /* r8 &= hold */
116 mov eax, [rbp+r8*4] ; /* eax = lcode[hold & lmask] */
117
118L_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
130ALIGN 4
131L_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
141L_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
151L_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
162L_get_distance_code:
163 and r8, rdx ; /* r8 &= hold */
164 mov eax, [r11+r8*4] ; /* eax = dcode[hold & dmask] */
165
166L_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
179L_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
189L_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
212L_copy_two:
213 rep movsw
214 mov rsi, r8 ; /* move in back to %rsi, toss from */
215 jmp L_while_test
216
217ALIGN 4
218L_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
233L_set_two:
234 rep stosw
235 jmp L_while_test
236
237ALIGN 4
238L_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
251ALIGN 4
252L_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
265ALIGN 4
266L_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
292ALIGN 4
293L_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
321ALIGN 4
322L_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
337ALIGN 4
338L_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
345L_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
351L_invalid_literal_length_code:
352 mov dword ptr [rsp+116], 2
353 jmp L_break_loop_with_status
354
355L_invalid_distance_code:
356 mov dword ptr [rsp+116], 3
357 jmp L_break_loop_with_status
358
359L_invalid_distance_too_far:
360 mov dword ptr [rsp+116], 4
361 jmp L_break_loop_with_status
362
363L_break_loop:
364 mov dword ptr [rsp+116], 0
365
366L_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
394inffas8664fnc ENDP
395;_TEXT ENDS
396END