]>
Commit | Line | Data |
---|---|---|
1 | /* Copyright (C) 1991,93,96,97,99,2000 Free Software Foundation, Inc. | |
2 | Based on strlen implementation by Torbjorn Granlund (tege@sics.se), | |
3 | with help from Dan Sahlin (dan@sics.se) and | |
4 | commentary by Jim Blandy (jimb@ai.mit.edu); | |
5 | adaptation to memchr suggested by Dick Karpinski (dick@cca.ucsf.edu), | |
6 | and implemented by Roland McGrath (roland@ai.mit.edu). | |
7 | ||
8 | NOTE: The canonical source of this file is maintained with the GNU C Library. | |
9 | Bugs can be reported to bug-glibc@prep.ai.mit.edu. | |
10 | ||
11 | This program is free software; you can redistribute it and/or modify it | |
12 | under the terms of the GNU General Public License as published by the | |
13 | Free Software Foundation; either version 2, or (at your option) any | |
14 | later version. | |
15 | ||
16 | This program is distributed in the hope that it will be useful, | |
17 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
18 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
19 | GNU General Public License for more details. | |
20 | ||
21 | You should have received a copy of the GNU General Public License | |
22 | along with this program; if not, write to the Free Software | |
23 | Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, | |
24 | USA. */ | |
25 | ||
26 | #ifdef HAVE_CONFIG_H | |
27 | # include <config.h> | |
28 | #endif | |
29 | ||
30 | #undef __ptr_t | |
31 | #if defined (__cplusplus) || (defined (__STDC__) && __STDC__) | |
32 | # define __ptr_t void * | |
33 | #else /* Not C++ or ANSI C. */ | |
34 | # define __ptr_t char * | |
35 | #endif /* C++ or ANSI C. */ | |
36 | ||
37 | #if defined _LIBC | |
38 | # include <string.h> | |
39 | # include <memcopy.h> | |
40 | #else | |
41 | # define reg_char char | |
42 | #endif | |
43 | ||
44 | #if HAVE_STDLIB_H || defined _LIBC | |
45 | # include <stdlib.h> | |
46 | #endif | |
47 | ||
48 | #if HAVE_LIMITS_H || defined _LIBC | |
49 | # include <limits.h> | |
50 | #endif | |
51 | ||
52 | #define LONG_MAX_32_BITS 2147483647 | |
53 | ||
54 | #ifndef LONG_MAX | |
55 | # define LONG_MAX LONG_MAX_32_BITS | |
56 | #endif | |
57 | ||
58 | #include <sys/types.h> | |
59 | #if HAVE_BP_SYM_H || defined _LIBC | |
60 | # include <bp-sym.h> | |
61 | #else | |
62 | # define BP_SYM(sym) sym | |
63 | #endif | |
64 | ||
65 | #undef memchr | |
66 | #undef __memchr | |
67 | ||
68 | /* Search no more than N bytes of S for C. */ | |
69 | __ptr_t | |
70 | __memchr (s, c_in, n) | |
71 | const __ptr_t s; | |
72 | int c_in; | |
73 | size_t n; | |
74 | { | |
75 | const unsigned char *char_ptr; | |
76 | const unsigned long int *longword_ptr; | |
77 | unsigned long int longword, magic_bits, charmask; | |
78 | unsigned reg_char c; | |
79 | ||
80 | c = (unsigned char) c_in; | |
81 | ||
82 | /* Handle the first few characters by reading one character at a time. | |
83 | Do this until CHAR_PTR is aligned on a longword boundary. */ | |
84 | for (char_ptr = (const unsigned char *) s; | |
85 | n > 0 && ((unsigned long int) char_ptr | |
86 | & (sizeof (longword) - 1)) != 0; | |
87 | --n, ++char_ptr) | |
88 | if (*char_ptr == c) | |
89 | return (__ptr_t) char_ptr; | |
90 | ||
91 | /* All these elucidatory comments refer to 4-byte longwords, | |
92 | but the theory applies equally well to 8-byte longwords. */ | |
93 | ||
94 | longword_ptr = (unsigned long int *) char_ptr; | |
95 | ||
96 | /* Bits 31, 24, 16, and 8 of this number are zero. Call these bits | |
97 | the "holes." Note that there is a hole just to the left of | |
98 | each byte, with an extra at the end: | |
99 | ||
100 | bits: 01111110 11111110 11111110 11111111 | |
101 | bytes: AAAAAAAA BBBBBBBB CCCCCCCC DDDDDDDD | |
102 | ||
103 | The 1-bits make sure that carries propagate to the next 0-bit. | |
104 | The 0-bits provide holes for carries to fall into. */ | |
105 | ||
106 | if (sizeof (longword) != 4 && sizeof (longword) != 8) | |
107 | abort (); | |
108 | ||
109 | #if LONG_MAX <= LONG_MAX_32_BITS | |
110 | magic_bits = 0x7efefeff; | |
111 | #else | |
112 | magic_bits = ((unsigned long int) 0x7efefefe << 32) | 0xfefefeff; | |
113 | #endif | |
114 | ||
115 | /* Set up a longword, each of whose bytes is C. */ | |
116 | charmask = c | (c << 8); | |
117 | charmask |= charmask << 16; | |
118 | #if LONG_MAX > LONG_MAX_32_BITS | |
119 | charmask |= charmask << 32; | |
120 | #endif | |
121 | ||
122 | /* Instead of the traditional loop which tests each character, | |
123 | we will test a longword at a time. The tricky part is testing | |
124 | if *any of the four* bytes in the longword in question are zero. */ | |
125 | while (n >= sizeof (longword)) | |
126 | { | |
127 | /* We tentatively exit the loop if adding MAGIC_BITS to | |
128 | LONGWORD fails to change any of the hole bits of LONGWORD. | |
129 | ||
130 | 1) Is this safe? Will it catch all the zero bytes? | |
131 | Suppose there is a byte with all zeros. Any carry bits | |
132 | propagating from its left will fall into the hole at its | |
133 | least significant bit and stop. Since there will be no | |
134 | carry from its most significant bit, the LSB of the | |
135 | byte to the left will be unchanged, and the zero will be | |
136 | detected. | |
137 | ||
138 | 2) Is this worthwhile? Will it ignore everything except | |
139 | zero bytes? Suppose every byte of LONGWORD has a bit set | |
140 | somewhere. There will be a carry into bit 8. If bit 8 | |
141 | is set, this will carry into bit 16. If bit 8 is clear, | |
142 | one of bits 9-15 must be set, so there will be a carry | |
143 | into bit 16. Similarly, there will be a carry into bit | |
144 | 24. If one of bits 24-30 is set, there will be a carry | |
145 | into bit 31, so all of the hole bits will be changed. | |
146 | ||
147 | The one misfire occurs when bits 24-30 are clear and bit | |
148 | 31 is set; in this case, the hole at bit 31 is not | |
149 | changed. If we had access to the processor carry flag, | |
150 | we could close this loophole by putting the fourth hole | |
151 | at bit 32! | |
152 | ||
153 | So it ignores everything except 128's, when they're aligned | |
154 | properly. | |
155 | ||
156 | 3) But wait! Aren't we looking for C, not zero? | |
157 | Good point. So what we do is XOR LONGWORD with a longword, | |
158 | each of whose bytes is C. This turns each byte that is C | |
159 | into a zero. */ | |
160 | ||
161 | longword = *longword_ptr++ ^ charmask; | |
162 | ||
163 | /* Add MAGIC_BITS to LONGWORD. */ | |
164 | if ((((longword + magic_bits) | |
165 | ||
166 | /* Set those bits that were unchanged by the addition. */ | |
167 | ^ ~longword) | |
168 | ||
169 | /* Look at only the hole bits. If any of the hole bits | |
170 | are unchanged, most likely one of the bytes was a | |
171 | zero. */ | |
172 | & ~magic_bits) != 0) | |
173 | { | |
174 | /* Which of the bytes was C? If none of them were, it was | |
175 | a misfire; continue the search. */ | |
176 | ||
177 | const unsigned char *cp = (const unsigned char *) (longword_ptr - 1); | |
178 | ||
179 | if (cp[0] == c) | |
180 | return (__ptr_t) cp; | |
181 | if (cp[1] == c) | |
182 | return (__ptr_t) &cp[1]; | |
183 | if (cp[2] == c) | |
184 | return (__ptr_t) &cp[2]; | |
185 | if (cp[3] == c) | |
186 | return (__ptr_t) &cp[3]; | |
187 | #if LONG_MAX > 2147483647 | |
188 | if (cp[4] == c) | |
189 | return (__ptr_t) &cp[4]; | |
190 | if (cp[5] == c) | |
191 | return (__ptr_t) &cp[5]; | |
192 | if (cp[6] == c) | |
193 | return (__ptr_t) &cp[6]; | |
194 | if (cp[7] == c) | |
195 | return (__ptr_t) &cp[7]; | |
196 | #endif | |
197 | } | |
198 | ||
199 | n -= sizeof (longword); | |
200 | } | |
201 | ||
202 | char_ptr = (const unsigned char *) longword_ptr; | |
203 | ||
204 | while (n-- > 0) | |
205 | { | |
206 | if (*char_ptr == c) | |
207 | return (__ptr_t) char_ptr; | |
208 | else | |
209 | ++char_ptr; | |
210 | } | |
211 | ||
212 | return 0; | |
213 | } | |
214 | #ifdef weak_alias | |
215 | weak_alias (__memchr, BP_SYM (memchr)) | |
216 | #endif |