]> git.saurik.com Git - apple/libc.git/blame - ppc/string/strlcat.s
Libc-594.9.5.tar.gz
[apple/libc.git] / ppc / string / strlcat.s
CommitLineData
9385eb3d
A
1/*
2 * Copyright (c) 2002 Apple Computer, Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
9385eb3d
A
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. Please obtain a copy of the License at
10 * http://www.opensource.apple.com/apsl/ and read it before using this
11 * file.
12 *
13 * The Original Code and all software distributed under the License are
14 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
15 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
16 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
18 * Please see the License for the specific language governing rights and
19 * limitations under the License.
20 *
21 * @APPLE_LICENSE_HEADER_END@
22 */
23#define ASSEMBLER
24#include <mach/ppc/asm.h>
25#undef ASSEMBLER
26
59e0d9fe
A
27#define __APPLE_API_PRIVATE
28#include <machine/cpu_capabilities.h>
29#undef __APPLE_API_PRIVATE
30
31/* We use mode-independent "g" opcodes such as "srgi". These expand
32 * into word operations when targeting __ppc__, and into doubleword
33 * operations when targeting __ppc64__.
34 */
35#include <architecture/ppc/mode_independent_asm.h>
36
37
9385eb3d
A
38// *****************
39// * S T R L C A T *
40// *****************
41//
42// size_t strlcat(char *dst, const char *src, size_t count);
43//
44// We optimize the move by doing it word parallel. This introduces
45// a complication: if we blindly did word load/stores until finding
46// a 0, we might get a spurious page fault by touching bytes past it.
47// We are allowed to touch the "count" bytes starting at "dst", but
48// when appending the "src", we must not do a "lwz" that crosses a page
49// boundary, or store past "count".
50//
51// The test for 0s relies on the following inobvious but very efficient
52// word-parallel test:
53// x = dataWord + 0xFEFEFEFF
54// y = ~dataWord & 0x80808080
55// if (x & y) == 0 then no zero found
56// The test maps any non-zero byte to zero, and any zero byte to 0x80,
57// with one exception: 0x01 bytes preceeding the first zero are also
58// mapped to 0x80.
59//
60// Note that "count" is the total buffer length, including the length
61// of the "dst" string. This is different than strncat().
59e0d9fe
A
62//
63// In 64-bit mode, this algorithm is doubleword parallel.
9385eb3d
A
64
65 .text
66 .globl EXT(strlcat)
67
68 .align 5
59e0d9fe
A
69LEXT(strlcat) // size_t strlcat(char *dst, const char *src, size_t count);
70 srgi. r0,r5,LOG2_GPR_BYTES// get #words or doublewords to scan
71#if defined(__ppc__)
72 lis r6,hi16(0xFEFEFEFF) // start to generate 32-bit magic constants
9385eb3d 73 lis r7,hi16(0x80808080)
9385eb3d
A
74 ori r6,r6,lo16(0xFEFEFEFF)
75 ori r7,r7,lo16(0x80808080)
59e0d9fe
A
76#else
77 ld r6,_COMM_PAGE_MAGIC_FE(0) // get 0xFEFEFEFE FEFEFEFF from commpage
78 ld r7,_COMM_PAGE_MAGIC_80(0) // get 0x80808080 80808080 from commpage
79#endif
9385eb3d
A
80 mr r9,r3 // use r9 for dest ptr (r3 remembers dst start)
81 beq-- L0bytes // buffer length <4
82 mtctr r0 // set up loop
83 b L0words // enter word loop
84
85// Loop over words looking for 0.
86// r3 = original start of buffer
87// r4 = source ptr (unaligned)
88// r5 = original buffer size
89// r6 = 0xFEFEFEFF
90// r7 = 0x80808080
91// r9 = dest ptr (unaligned)
59e0d9fe 92// ctr = #words or doublewords remaining in buffer
9385eb3d
A
93
94 .align 5 // align inner loops for speed
95L0words:
59e0d9fe
A
96 lg r8,0(r9) // r8 <- next dest word or doubleword
97 addi r9,r9,GPR_BYTES
9385eb3d
A
98 add r10,r8,r6 // r10 <- word + 0xFEFEFEFF
99 andc r12,r7,r8 // r12 <- ~word & 0x80808080
100 and. r11,r10,r12 // r11 <- nonzero iff word has a 0-byte
101 bdnzt eq,L0words // loop until 0 found or buffer end
102
103 beq-- L0bytes // skip if 0 not found
104
59e0d9fe
A
105 slgi r0,r8,7 // move 0x01 bits (false hits) into 0x80 position
106 subi r9,r9,GPR_BYTES // back up r9 to the start of the word
9385eb3d 107 andc r11,r11,r0 // mask out false hits
59e0d9fe 108 cntlzg r0,r11 // find 0 byte (r0 = 0, 8, 16, or 24)
9385eb3d
A
109 srwi r0,r0,3 // now r0 = 0, 1, 2, or 3
110 add r9,r9,r0 // now r9 points to the 0-byte in dest
111 b L0found // start to append source
112
113// Loop over bytes looking for 0.
114// r3 = original start of buffer
115// r4 = source ptr (unaligned)
116// r5 = original buffer size
117// r6 = 0xFEFEFEFF
118// r7 = 0x80808080
119// r9 = dest ptr (unaligned)
120
121L0bytes:
59e0d9fe 122 andi. r0,r5,GPR_BYTES-1 // get #bytes remaining in buffer
9385eb3d
A
123 mtctr r0 // set up byte loop
124 beq-- L0notfound // skip if 0 not found in buffer (error)
125L0byteloop:
126 lbz r8,0(r9) // r8 <- next dest byte
127 addi r9,r9,1
128 cmpwi r8,0 // 0 ?
129 bdnzf eq,L0byteloop // loop until 0 found or buffer end
130
131 bne-- L0notfound // skip if 0 not found (error)
132 subi r9,r9,1 // back up, so r9 points to the 0
133
134// End of dest found, so we can start appending source. First, align the source,
135// in order to avoid spurious page faults.
136// r3 = original start of buffer
137// r4 = original source ptr (unaligned)
138// r5 = original buffer size
139// r6 = 0xFEFEFEFF
140// r7 = 0x80808080
141// r9 = ptr to 0-byte in dest (unaligned)
142
143L0found:
59e0d9fe 144 andi. r0,r4,GPR_BYTES-1 // is source aligned?
9385eb3d
A
145 add r5,r5,r3 // get ptr to end of buffer
146 sub r5,r5,r9 // get #bytes remaining in buffer, counting the 0 (r5>0)
147 beq Laligned // skip if source already word aligned
59e0d9fe 148 subfic r0,r0,GPR_BYTES // not aligned, get #bytes to align r4
9385eb3d
A
149 b Lbyteloop1 // r5!=0, so skip check
150
151// Copy min(r0,r5) bytes, until 0-byte.
152// r0 = #bytes we propose to copy (NOTE: must be >0)
153// r4 = source ptr (unaligned)
154// r5 = length remaining in buffer (may be 0)
155// r6 = 0xFEFEFEFF
156// r7 = 0x80808080
157// r9 = dest ptr (unaligned)
158
159Lbyteloop:
59e0d9fe 160 cmpgi r5,0 // buffer empty? (note: length is unsigned)
9385eb3d
A
161 beq-- Loverrun // buffer filled before end of source reached
162Lbyteloop1: // entry when we know r5!=0
163 lbz r8,0(r4) // r8 <- next source byte
164 subic. r0,r0,1 // decrement count of bytes to move
165 addi r4,r4,1
166 subi r5,r5,1 // decrement buffer length remaining
167 stb r8,0(r9) // pack into dest
168 cmpwi cr1,r8,0 // 0-byte?
169 addi r9,r9,1
170 beq cr1,L0stored // byte was 0, so done
171 bne Lbyteloop // r0!=0, source not yet aligned
172
59e0d9fe 173// Source is aligned. Loop over words or doublewords until 0-byte found or end
9385eb3d
A
174// of buffer.
175// r3 = original start of buffer
59e0d9fe 176// r4 = source ptr (aligned)
9385eb3d
A
177// r5 = length remaining in buffer
178// r6 = 0xFEFEFEFF
179// r7 = 0x80808080
180// r9 = dest ptr (unaligned)
181
182Laligned:
59e0d9fe 183 srgi. r8,r5,LOG2_GPR_BYTES// get #words or doublewords in buffer
9385eb3d
A
184 addi r0,r5,1 // if no words...
185 beq-- Lbyteloop // ...copy to end of buffer
59e0d9fe
A
186 mtctr r8 // set up loop count
187 rlwinm r5,r5,0,GPR_BYTES-1 // mask buffer length down to leftover bytes
9385eb3d
A
188 b LwordloopEnter
189
59e0d9fe 190// Inner loop: move a word or doubleword at a time, until one of two conditions:
9385eb3d
A
191// - a zero byte is found
192// - end of buffer
193// At this point, registers are as follows:
194// r3 = original start of buffer
59e0d9fe
A
195// r4 = source ptr (aligned)
196// r5 = bytes leftover in buffer (0..GPR_BYTES-1)
9385eb3d
A
197// r6 = 0xFEFEFEFF
198// r7 = 0x80808080
199// r9 = dest ptr (unaligned)
59e0d9fe 200// ctr = loop count
9385eb3d
A
201
202 .align 5 // align inner loop, which is 8 words long
203Lwordloop:
59e0d9fe
A
204 stg r8,0(r9) // pack word into destination
205 addi r9,r9,GPR_BYTES
9385eb3d 206LwordloopEnter:
59e0d9fe
A
207 lg r8,0(r4) // r8 <- next 4 or 8 source bytes
208 addi r4,r4,GPR_BYTES
9385eb3d
A
209 add r10,r8,r6 // r10 <- word + 0xFEFEFEFF
210 andc r12,r7,r8 // r12 <- ~word & 0x80808080
211 and. r11,r10,r12 // r11 <- nonzero iff word has a 0-byte
212 bdnzt eq,Lwordloop // loop if ctr!=0 and cr0_eq
213
214 beq-- Lleftovers // skip if no 0-byte found, copy leftovers
215
216// Found a 0-byte. Store last word up to and including the 0, a byte at a time.
217// r3 = original start of buffer
218// r8 = last word, known to have a 0-byte
219// r9 = dest ptr (one past 0)
220
221Lstorelastbytes:
59e0d9fe
A
222 srgi. r0,r8,GPR_BYTES*8-8 // right justify next byte and test for 0
223 slgi r8,r8,8 // shift next byte into position
9385eb3d
A
224 stb r0,0(r9) // pack into dest
225 addi r9,r9,1
226 bne Lstorelastbytes // loop until 0 stored
227
228// Append op successful, O stored into buffer. Return total length.
229// r3 = original start of buffer
230// r9 = dest ptr (one past 0)
231
232L0stored:
233 sub r3,r9,r3 // get (length+1) of string in buffer
234 subi r3,r3,1 // return length
235 blr
236
59e0d9fe
A
237// 0-byte not found in aligned source words. There are up to GPR_BYTES-1 leftover
238// source bytes, hopefully the 0-byte is among them.
239// r4 = source ptr (aligned)
240// r5 = leftover bytes in buffer (0..GPR_BYTES-1)
9385eb3d
A
241// r6 = 0xFEFEFEFF
242// r7 = 0x80808080
59e0d9fe 243// r8 = last full word or doubleword of source
9385eb3d
A
244// r9 = dest ptr (unaligned)
245
246Lleftovers:
59e0d9fe
A
247 stg r8,0(r9) // store last word
248 addi r9,r9,GPR_BYTES
9385eb3d
A
249 addi r0,r5,1 // make sure r5 terminates byte loop (not r0)
250 b Lbyteloop
251
252// Buffer filled during append without finding the end of source. Overwrite the
253// last byte in buffer with a 0, and compute how long the concatenated string would
254// have been, if the buffer had been large enough.
255// r3 = original start of buffer
256// r4 = source ptr (1st byte not copied into buffer)
257// r9 = dest ptr (one past end of buffer)
258
259Loverrun:
260 sub. r3,r9,r3 // compute #bytes stored in buffer
261 li r0,0 // get a 0
262 beq-- Lskip // buffer was 0-length
263 stb r0,-1(r9) // jam in delimiting 0
264
265// Buffer full, check to see how much longer source is. We don't optimize this,
266// since overruns are an error.
267
268Lskip:
269 lbz r8,0(r4) // get next source byte
270 addi r4,r4,1
271 addi r3,r3,1 // increment length of "ideal" string
272 cmpwi r8,0 // 0?
273 bne Lskip
274
275 subi r3,r3,1 // don't count 0 in length
276 blr // return length of string we "wanted" to create
277
278// 0 not found in buffer (append not yet begun.) We don't store a delimiting 0,
279// but do compute how long the concatenated string would have been, assuming the length
280// of "dst" is the length of the buffer.
281// r3 = original start of buffer
282// r4 = original source ptr
283// r9 = dest ptr (one past end of buffer)
284
285L0notfound:
286 sub r3,r9,r3 // compute #bytes in buffer
287 b Lskip // add strlen(source) to r3
288