1 /*
2  * Copyright (c) 2008-2015 ARM Ltd
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. The name of the company may not be used to endorse or promote
14  *    products derived from this software without specific prior written
15  *    permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY ARM LTD ``AS IS'' AND ANY EXPRESS OR IMPLIED
18  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20  * IN NO EVENT SHALL ARM LTD BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
22  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
23  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
24  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
25  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
26  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28 
29 #include "arm_asm.h"
30 #include <_ansi.h>
31 #include <string.h>
32 #include <limits.h>
33 
34 #if defined __OPTIMIZE_SIZE__ || defined PREFER_SIZE_OVER_SPEED
35 #if __ARM_ARCH_ISA_THUMB == 2
36 /* Implemented in strlen.S.  */
37 
38 #elif defined (__thumb__)
39 /* Implemented in strlen.S.  */
40 
41 #else
42 #include "../../string/strlen.c"
43 
44 #endif
45 
46 #else /* defined __OPTIMIZE_SIZE__ || defined PREFER_SIZE_OVER_SPEED */
47 #if defined __thumb__ && ! defined __thumb2__
48 #include "../../string/strlen.c"
49 
50 #elif __ARM_ARCH_ISA_THUMB >= 2 && defined __ARM_FEATURE_DSP
51   /* Implemented in strlen.S.  */
52 
53 #else
54 #pragma GCC diagnostic ignored "-Wunused-parameter"
55 size_t __attribute__((naked))
strlen(const char * str)56 strlen (const char* str)
57 {
58   __asm__("len .req r0\n\t"
59        "data .req r3\n\t"
60        "addr .req r1\n\t"
61 
62 #ifdef _ISA_ARM_7
63        "pld [r0]\n\t"
64 #endif
65        /* Word-align address */
66        "bic	addr, r0, #3\n\t"
67        /* Get adjustment for start ... */
68        "ands	len, r0, #3\n\t"
69        "neg	len, len\n\t"
70        /* First word of data */
71        "ldr	data, [addr], #4\n\t"
72        /* Ensure bytes preceeding start ... */
73        "add	ip, len, #4\n\t"
74        "mov	ip, ip, asl #3\n\t"
75        "mvn	r2, #0\n\t"
76        /* ... are masked out */
77 #ifdef __thumb__
78        "itt	ne\n\t"
79 # ifdef __ARMEB__
80        "lslne	r2, ip\n\t"
81 # else
82        "lsrne	r2, ip\n\t"
83 # endif
84        "orrne	data, data, r2\n\t"
85 #else
86        "it	ne\n\t"
87 # ifdef __ARMEB__
88        "orrne	data, data, r2, lsl ip\n\t"
89 # else
90        "orrne	data, data, r2, lsr ip\n\t"
91 # endif
92 #endif
93        /* Magic const 0x01010101 */
94 #ifdef _ISA_ARM_7
95        "movw	ip, #0x101\n\t"
96 #else
97        "mov	ip, #0x1\n\t"
98        "orr	ip, ip, ip, lsl #8\n\t"
99 #endif
100        "orr	ip, ip, ip, lsl #16\n"
101 
102 	/* This is the main loop.  We subtract one from each byte in
103 	   the word: the sign bit changes iff the byte was zero or
104 	   0x80 -- we eliminate the latter case by anding the result
105 	   with the 1-s complement of the data.  */
106        "1:\n\t"
107        /* test (data - 0x01010101)  */
108        "sub	r2, data, ip\n\t"
109        /* ... & ~data */
110        "bic	r2, r2, data\n\t"
111        /* ... & 0x80808080 == 0? */
112        "ands	r2, r2, ip, lsl #7\n\t"
113 #ifdef _ISA_ARM_7
114        /* yes, get more data... */
115        "itt	eq\n\t"
116        "ldreq	data, [addr], #4\n\t"
117        /* and 4 more bytes  */
118        "addeq	len, len, #4\n\t"
119 	/* Unroll the loop a bit.  */
120        "pld	[addr, #8]\n\t"
121        /*  test (data - 0x01010101)  */
122        "ittt	eq\n\t"
123        "subeq	r2, data, ip\n\t"
124        /* ... & ~data */
125        "biceq	r2, r2, data\n\t"
126        /* ... & 0x80808080 == 0? */
127        "andeqs	r2, r2, ip, lsl #7\n\t"
128 #endif
129        "itt	eq\n\t"
130        /* yes, get more data... */
131        "ldreq	data, [addr], #4\n\t"
132        /* and 4 more bytes  */
133        "addeq	len, len, #4\n\t"
134        "beq	1b\n\t"
135 #ifdef __ARMEB__
136        "tst	data, #0xff000000\n\t"
137        "itttt	ne\n\t"
138        "addne	len, len, #1\n\t"
139        "tstne	data, #0xff0000\n\t"
140        "addne	len, len, #1\n\t"
141        "tstne	data, #0xff00\n\t"
142        "it	ne\n\t"
143        "addne	len, len, #1\n\t"
144 #else
145 # ifdef _ISA_ARM_5
146 	/* R2 is the residual sign bits from the above test.  All we
147 	need to do now is establish the position of the first zero
148 	byte... */
149 	/* Little-endian is harder, we need the number of trailing
150 	zeros / 8 */
151 #  ifdef _ISA_ARM_7
152        "rbit	r2, r2\n\t"
153        "clz	r2, r2\n\t"
154 #  else
155        "rsb	r1, r2, #0\n\t"
156        "and	r2, r2, r1\n\t"
157        "clz	r2, r2\n\t"
158        "rsb	r2, r2, #31\n\t"
159 #  endif
160        "add	len, len, r2, lsr #3\n\t"
161 # else  /* No CLZ instruction */
162        "tst	data, #0xff\n\t"
163        "itttt	ne\n\t"
164        "addne	len, len, #1\n\t"
165        "tstne	data, #0xff00\n\t"
166        "addne	len, len, #1\n\t"
167        "tstne	data, #0xff0000\n\t"
168        "it	ne\n\t"
169        "addne	len, len, #1\n\t"
170 # endif
171 #endif
172        "bx	lr\n\t");
173 }
174 #endif
175 #endif
176