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