yuezonghe | 824eb0c | 2024-06-27 02:32:26 -0700 | [diff] [blame] | 1 | /* match.S -- x86 assembly version of the zlib longest_match() function. |
| 2 | * Optimized for the Intel 686 chips (PPro and later). |
| 3 | * |
| 4 | * Copyright (C) 1998, 2007 Brian Raiter <breadbox@muppetlabs.com> |
| 5 | * |
| 6 | * This software is provided 'as-is', without any express or implied |
| 7 | * warranty. In no event will the author be held liable for any damages |
| 8 | * arising from the use of this software. |
| 9 | * |
| 10 | * Permission is granted to anyone to use this software for any purpose, |
| 11 | * including commercial applications, and to alter it and redistribute it |
| 12 | * freely, subject to the following restrictions: |
| 13 | * |
| 14 | * 1. The origin of this software must not be misrepresented; you must not |
| 15 | * claim that you wrote the original software. If you use this software |
| 16 | * in a product, an acknowledgment in the product documentation would be |
| 17 | * appreciated but is not required. |
| 18 | * 2. Altered source versions must be plainly marked as such, and must not be |
| 19 | * misrepresented as being the original software. |
| 20 | * 3. This notice may not be removed or altered from any source distribution. |
| 21 | */ |
| 22 | |
| 23 | #ifndef NO_UNDERLINE |
| 24 | #define match_init _match_init |
| 25 | #define longest_match _longest_match |
| 26 | #endif |
| 27 | |
| 28 | #define MAX_MATCH (258) |
| 29 | #define MIN_MATCH (3) |
| 30 | #define MIN_LOOKAHEAD (MAX_MATCH + MIN_MATCH + 1) |
| 31 | #define MAX_MATCH_8 ((MAX_MATCH + 7) & ~7) |
| 32 | |
| 33 | /* stack frame offsets */ |
| 34 | |
| 35 | #define chainlenwmask 0 /* high word: current chain len */ |
| 36 | /* low word: s->wmask */ |
| 37 | #define window 4 /* local copy of s->window */ |
| 38 | #define windowbestlen 8 /* s->window + bestlen */ |
| 39 | #define scanstart 16 /* first two bytes of string */ |
| 40 | #define scanend 12 /* last two bytes of string */ |
| 41 | #define scanalign 20 /* dword-misalignment of string */ |
| 42 | #define nicematch 24 /* a good enough match size */ |
| 43 | #define bestlen 28 /* size of best match so far */ |
| 44 | #define scan 32 /* ptr to string wanting match */ |
| 45 | |
| 46 | #define LocalVarsSize (36) |
| 47 | /* saved ebx 36 */ |
| 48 | /* saved edi 40 */ |
| 49 | /* saved esi 44 */ |
| 50 | /* saved ebp 48 */ |
| 51 | /* return address 52 */ |
| 52 | #define deflatestate 56 /* the function arguments */ |
| 53 | #define curmatch 60 |
| 54 | |
| 55 | /* All the +zlib1222add offsets are due to the addition of fields |
| 56 | * in zlib in the deflate_state structure since the asm code was first written |
| 57 | * (if you compile with zlib 1.0.4 or older, use "zlib1222add equ (-4)"). |
| 58 | * (if you compile with zlib between 1.0.5 and 1.2.2.1, use "zlib1222add equ 0"). |
| 59 | * if you compile with zlib 1.2.2.2 or later , use "zlib1222add equ 8"). |
| 60 | */ |
| 61 | |
| 62 | #define zlib1222add (8) |
| 63 | |
| 64 | #define dsWSize (36+zlib1222add) |
| 65 | #define dsWMask (44+zlib1222add) |
| 66 | #define dsWindow (48+zlib1222add) |
| 67 | #define dsPrev (56+zlib1222add) |
| 68 | #define dsMatchLen (88+zlib1222add) |
| 69 | #define dsPrevMatch (92+zlib1222add) |
| 70 | #define dsStrStart (100+zlib1222add) |
| 71 | #define dsMatchStart (104+zlib1222add) |
| 72 | #define dsLookahead (108+zlib1222add) |
| 73 | #define dsPrevLen (112+zlib1222add) |
| 74 | #define dsMaxChainLen (116+zlib1222add) |
| 75 | #define dsGoodMatch (132+zlib1222add) |
| 76 | #define dsNiceMatch (136+zlib1222add) |
| 77 | |
| 78 | |
| 79 | .file "match.S" |
| 80 | |
| 81 | .globl match_init, longest_match |
| 82 | |
| 83 | .text |
| 84 | |
| 85 | /* uInt longest_match(deflate_state *deflatestate, IPos curmatch) */ |
| 86 | .cfi_sections .debug_frame |
| 87 | |
| 88 | longest_match: |
| 89 | |
| 90 | .cfi_startproc |
| 91 | /* Save registers that the compiler may be using, and adjust %esp to */ |
| 92 | /* make room for our stack frame. */ |
| 93 | |
| 94 | pushl %ebp |
| 95 | .cfi_def_cfa_offset 8 |
| 96 | .cfi_offset ebp, -8 |
| 97 | pushl %edi |
| 98 | .cfi_def_cfa_offset 12 |
| 99 | pushl %esi |
| 100 | .cfi_def_cfa_offset 16 |
| 101 | pushl %ebx |
| 102 | .cfi_def_cfa_offset 20 |
| 103 | subl $LocalVarsSize, %esp |
| 104 | .cfi_def_cfa_offset LocalVarsSize+20 |
| 105 | |
| 106 | /* Retrieve the function arguments. %ecx will hold cur_match */ |
| 107 | /* throughout the entire function. %edx will hold the pointer to the */ |
| 108 | /* deflate_state structure during the function's setup (before */ |
| 109 | /* entering the main loop). */ |
| 110 | |
| 111 | movl deflatestate(%esp), %edx |
| 112 | movl curmatch(%esp), %ecx |
| 113 | |
| 114 | /* uInt wmask = s->w_mask; */ |
| 115 | /* unsigned chain_length = s->max_chain_length; */ |
| 116 | /* if (s->prev_length >= s->good_match) { */ |
| 117 | /* chain_length >>= 2; */ |
| 118 | /* } */ |
| 119 | |
| 120 | movl dsPrevLen(%edx), %eax |
| 121 | movl dsGoodMatch(%edx), %ebx |
| 122 | cmpl %ebx, %eax |
| 123 | movl dsWMask(%edx), %eax |
| 124 | movl dsMaxChainLen(%edx), %ebx |
| 125 | jl LastMatchGood |
| 126 | shrl $2, %ebx |
| 127 | LastMatchGood: |
| 128 | |
| 129 | /* chainlen is decremented once beforehand so that the function can */ |
| 130 | /* use the sign flag instead of the zero flag for the exit test. */ |
| 131 | /* It is then shifted into the high word, to make room for the wmask */ |
| 132 | /* value, which it will always accompany. */ |
| 133 | |
| 134 | decl %ebx |
| 135 | shll $16, %ebx |
| 136 | orl %eax, %ebx |
| 137 | movl %ebx, chainlenwmask(%esp) |
| 138 | |
| 139 | /* if ((uInt)nice_match > s->lookahead) nice_match = s->lookahead; */ |
| 140 | |
| 141 | movl dsNiceMatch(%edx), %eax |
| 142 | movl dsLookahead(%edx), %ebx |
| 143 | cmpl %eax, %ebx |
| 144 | jl LookaheadLess |
| 145 | movl %eax, %ebx |
| 146 | LookaheadLess: movl %ebx, nicematch(%esp) |
| 147 | |
| 148 | /* register Bytef *scan = s->window + s->strstart; */ |
| 149 | |
| 150 | movl dsWindow(%edx), %esi |
| 151 | movl %esi, window(%esp) |
| 152 | movl dsStrStart(%edx), %ebp |
| 153 | lea (%esi,%ebp), %edi |
| 154 | movl %edi, scan(%esp) |
| 155 | |
| 156 | /* Determine how many bytes the scan ptr is off from being */ |
| 157 | /* dword-aligned. */ |
| 158 | |
| 159 | movl %edi, %eax |
| 160 | negl %eax |
| 161 | andl $3, %eax |
| 162 | movl %eax, scanalign(%esp) |
| 163 | |
| 164 | /* IPos limit = s->strstart > (IPos)MAX_DIST(s) ? */ |
| 165 | /* s->strstart - (IPos)MAX_DIST(s) : NIL; */ |
| 166 | |
| 167 | movl dsWSize(%edx), %eax |
| 168 | subl $MIN_LOOKAHEAD, %eax |
| 169 | subl %eax, %ebp |
| 170 | jg LimitPositive |
| 171 | xorl %ebp, %ebp |
| 172 | LimitPositive: |
| 173 | |
| 174 | /* int best_len = s->prev_length; */ |
| 175 | |
| 176 | movl dsPrevLen(%edx), %eax |
| 177 | movl %eax, bestlen(%esp) |
| 178 | |
| 179 | /* Store the sum of s->window + best_len in %esi locally, and in %esi. */ |
| 180 | |
| 181 | addl %eax, %esi |
| 182 | movl %esi, windowbestlen(%esp) |
| 183 | |
| 184 | /* register ush scan_start = *(ushf*)scan; */ |
| 185 | /* register ush scan_end = *(ushf*)(scan+best_len-1); */ |
| 186 | /* Posf *prev = s->prev; */ |
| 187 | |
| 188 | movzwl (%edi), %ebx |
| 189 | movl %ebx, scanstart(%esp) |
| 190 | movzwl -1(%edi,%eax), %ebx |
| 191 | movl %ebx, scanend(%esp) |
| 192 | movl dsPrev(%edx), %edi |
| 193 | |
| 194 | /* Jump into the main loop. */ |
| 195 | |
| 196 | movl chainlenwmask(%esp), %edx |
| 197 | jmp LoopEntry |
| 198 | |
| 199 | .balign 16 |
| 200 | |
| 201 | /* do { |
| 202 | * match = s->window + cur_match; |
| 203 | * if (*(ushf*)(match+best_len-1) != scan_end || |
| 204 | * *(ushf*)match != scan_start) continue; |
| 205 | * [...] |
| 206 | * } while ((cur_match = prev[cur_match & wmask]) > limit |
| 207 | * && --chain_length != 0); |
| 208 | * |
| 209 | * Here is the inner loop of the function. The function will spend the |
| 210 | * majority of its time in this loop, and majority of that time will |
| 211 | * be spent in the first ten instructions. |
| 212 | * |
| 213 | * Within this loop: |
| 214 | * %ebx = scanend |
| 215 | * %ecx = curmatch |
| 216 | * %edx = chainlenwmask - i.e., ((chainlen << 16) | wmask) |
| 217 | * %esi = windowbestlen - i.e., (window + bestlen) |
| 218 | * %edi = prev |
| 219 | * %ebp = limit |
| 220 | */ |
| 221 | LookupLoop: |
| 222 | andl %edx, %ecx |
| 223 | movzwl (%edi,%ecx,2), %ecx |
| 224 | cmpl %ebp, %ecx |
| 225 | jbe LeaveNow |
| 226 | subl $0x00010000, %edx |
| 227 | js LeaveNow |
| 228 | LoopEntry: movzwl -1(%esi,%ecx), %eax |
| 229 | cmpl %ebx, %eax |
| 230 | jnz LookupLoop |
| 231 | movl window(%esp), %eax |
| 232 | movzwl (%eax,%ecx), %eax |
| 233 | cmpl scanstart(%esp), %eax |
| 234 | jnz LookupLoop |
| 235 | |
| 236 | /* Store the current value of chainlen. */ |
| 237 | |
| 238 | movl %edx, chainlenwmask(%esp) |
| 239 | |
| 240 | /* Point %edi to the string under scrutiny, and %esi to the string we */ |
| 241 | /* are hoping to match it up with. In actuality, %esi and %edi are */ |
| 242 | /* both pointed (MAX_MATCH_8 - scanalign) bytes ahead, and %edx is */ |
| 243 | /* initialized to -(MAX_MATCH_8 - scanalign). */ |
| 244 | |
| 245 | movl window(%esp), %esi |
| 246 | movl scan(%esp), %edi |
| 247 | addl %ecx, %esi |
| 248 | movl scanalign(%esp), %eax |
| 249 | movl $(-MAX_MATCH_8), %edx |
| 250 | lea MAX_MATCH_8(%edi,%eax), %edi |
| 251 | lea MAX_MATCH_8(%esi,%eax), %esi |
| 252 | |
| 253 | /* Test the strings for equality, 8 bytes at a time. At the end, |
| 254 | * adjust %edx so that it is offset to the exact byte that mismatched. |
| 255 | * |
| 256 | * We already know at this point that the first three bytes of the |
| 257 | * strings match each other, and they can be safely passed over before |
| 258 | * starting the compare loop. So what this code does is skip over 0-3 |
| 259 | * bytes, as much as necessary in order to dword-align the %edi |
| 260 | * pointer. (%esi will still be misaligned three times out of four.) |
| 261 | * |
| 262 | * It should be confessed that this loop usually does not represent |
| 263 | * much of the total running time. Replacing it with a more |
| 264 | * straightforward "rep cmpsb" would not drastically degrade |
| 265 | * performance. |
| 266 | */ |
| 267 | LoopCmps: |
| 268 | movl (%esi,%edx), %eax |
| 269 | xorl (%edi,%edx), %eax |
| 270 | jnz LeaveLoopCmps |
| 271 | movl 4(%esi,%edx), %eax |
| 272 | xorl 4(%edi,%edx), %eax |
| 273 | jnz LeaveLoopCmps4 |
| 274 | addl $8, %edx |
| 275 | jnz LoopCmps |
| 276 | jmp LenMaximum |
| 277 | LeaveLoopCmps4: addl $4, %edx |
| 278 | LeaveLoopCmps: testl $0x0000FFFF, %eax |
| 279 | jnz LenLower |
| 280 | addl $2, %edx |
| 281 | shrl $16, %eax |
| 282 | LenLower: subb $1, %al |
| 283 | adcl $0, %edx |
| 284 | |
| 285 | /* Calculate the length of the match. If it is longer than MAX_MATCH, */ |
| 286 | /* then automatically accept it as the best possible match and leave. */ |
| 287 | |
| 288 | lea (%edi,%edx), %eax |
| 289 | movl scan(%esp), %edi |
| 290 | subl %edi, %eax |
| 291 | cmpl $MAX_MATCH, %eax |
| 292 | jge LenMaximum |
| 293 | |
| 294 | /* If the length of the match is not longer than the best match we */ |
| 295 | /* have so far, then forget it and return to the lookup loop. */ |
| 296 | |
| 297 | movl deflatestate(%esp), %edx |
| 298 | movl bestlen(%esp), %ebx |
| 299 | cmpl %ebx, %eax |
| 300 | jg LongerMatch |
| 301 | movl windowbestlen(%esp), %esi |
| 302 | movl dsPrev(%edx), %edi |
| 303 | movl scanend(%esp), %ebx |
| 304 | movl chainlenwmask(%esp), %edx |
| 305 | jmp LookupLoop |
| 306 | |
| 307 | /* s->match_start = cur_match; */ |
| 308 | /* best_len = len; */ |
| 309 | /* if (len >= nice_match) break; */ |
| 310 | /* scan_end = *(ushf*)(scan+best_len-1); */ |
| 311 | |
| 312 | LongerMatch: movl nicematch(%esp), %ebx |
| 313 | movl %eax, bestlen(%esp) |
| 314 | movl %ecx, dsMatchStart(%edx) |
| 315 | cmpl %ebx, %eax |
| 316 | jge LeaveNow |
| 317 | movl window(%esp), %esi |
| 318 | addl %eax, %esi |
| 319 | movl %esi, windowbestlen(%esp) |
| 320 | movzwl -1(%edi,%eax), %ebx |
| 321 | movl dsPrev(%edx), %edi |
| 322 | movl %ebx, scanend(%esp) |
| 323 | movl chainlenwmask(%esp), %edx |
| 324 | jmp LookupLoop |
| 325 | |
| 326 | /* Accept the current string, with the maximum possible length. */ |
| 327 | |
| 328 | LenMaximum: movl deflatestate(%esp), %edx |
| 329 | movl $MAX_MATCH, bestlen(%esp) |
| 330 | movl %ecx, dsMatchStart(%edx) |
| 331 | |
| 332 | /* if ((uInt)best_len <= s->lookahead) return (uInt)best_len; */ |
| 333 | /* return s->lookahead; */ |
| 334 | |
| 335 | LeaveNow: |
| 336 | movl deflatestate(%esp), %edx |
| 337 | movl bestlen(%esp), %ebx |
| 338 | movl dsLookahead(%edx), %eax |
| 339 | cmpl %eax, %ebx |
| 340 | jg LookaheadRet |
| 341 | movl %ebx, %eax |
| 342 | LookaheadRet: |
| 343 | |
| 344 | /* Restore the stack and return from whence we came. */ |
| 345 | |
| 346 | addl $LocalVarsSize, %esp |
| 347 | .cfi_def_cfa_offset 20 |
| 348 | popl %ebx |
| 349 | .cfi_def_cfa_offset 16 |
| 350 | popl %esi |
| 351 | .cfi_def_cfa_offset 12 |
| 352 | popl %edi |
| 353 | .cfi_def_cfa_offset 8 |
| 354 | popl %ebp |
| 355 | .cfi_def_cfa_offset 4 |
| 356 | .cfi_endproc |
| 357 | match_init: ret |