1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Most of the string-functions are rather heavily hand-optimized,
4 * see especially strsep,strstr,str[c]spn. They should work, but are not
5 * very easy to understand. Everything is done entirely within the register
6 * set, making the functions fast and clean. String instructions have been
7 * used through-out, making for "slightly" unclear code :-)
8 *
9 * AK: On P4 and K7 using non string instruction implementations might be faster
10 * for large memory blocks. But most of them are unlikely to be used on large
11 * strings.
12 */
13
14 #define __NO_FORTIFY
15 #include <linux/string.h>
16 #include <linux/export.h>
17
18 #ifdef __HAVE_ARCH_STRCPY
strcpy(char * dest,const char * src)19 char *strcpy(char *dest, const char *src)
20 {
21 int d0, d1, d2;
22 asm volatile("1:\tlodsb\n\t"
23 "stosb\n\t"
24 "testb %%al,%%al\n\t"
25 "jne 1b"
26 : "=&S" (d0), "=&D" (d1), "=&a" (d2)
27 : "0" (src), "1" (dest) : "memory");
28 return dest;
29 }
30 EXPORT_SYMBOL(strcpy);
31 #endif
32
33 #ifdef __HAVE_ARCH_STRNCPY
strncpy(char * dest,const char * src,size_t count)34 char *strncpy(char *dest, const char *src, size_t count)
35 {
36 int d0, d1, d2, d3;
37 asm volatile("1:\tdecl %2\n\t"
38 "js 2f\n\t"
39 "lodsb\n\t"
40 "stosb\n\t"
41 "testb %%al,%%al\n\t"
42 "jne 1b\n\t"
43 "rep\n\t"
44 "stosb\n"
45 "2:"
46 : "=&S" (d0), "=&D" (d1), "=&c" (d2), "=&a" (d3)
47 : "0" (src), "1" (dest), "2" (count) : "memory");
48 return dest;
49 }
50 EXPORT_SYMBOL(strncpy);
51 #endif
52
53 #ifdef __HAVE_ARCH_STRCAT
strcat(char * dest,const char * src)54 char *strcat(char *dest, const char *src)
55 {
56 int d0, d1, d2, d3;
57 asm volatile("repne\n\t"
58 "scasb\n\t"
59 "decl %1\n"
60 "1:\tlodsb\n\t"
61 "stosb\n\t"
62 "testb %%al,%%al\n\t"
63 "jne 1b"
64 : "=&S" (d0), "=&D" (d1), "=&a" (d2), "=&c" (d3)
65 : "0" (src), "1" (dest), "2" (0), "3" (0xffffffffu) : "memory");
66 return dest;
67 }
68 EXPORT_SYMBOL(strcat);
69 #endif
70
71 #ifdef __HAVE_ARCH_STRNCAT
strncat(char * dest,const char * src,size_t count)72 char *strncat(char *dest, const char *src, size_t count)
73 {
74 int d0, d1, d2, d3;
75 asm volatile("repne\n\t"
76 "scasb\n\t"
77 "decl %1\n\t"
78 "movl %8,%3\n"
79 "1:\tdecl %3\n\t"
80 "js 2f\n\t"
81 "lodsb\n\t"
82 "stosb\n\t"
83 "testb %%al,%%al\n\t"
84 "jne 1b\n"
85 "2:\txorl %2,%2\n\t"
86 "stosb"
87 : "=&S" (d0), "=&D" (d1), "=&a" (d2), "=&c" (d3)
88 : "0" (src), "1" (dest), "2" (0), "3" (0xffffffffu), "g" (count)
89 : "memory");
90 return dest;
91 }
92 EXPORT_SYMBOL(strncat);
93 #endif
94
95 #ifdef __HAVE_ARCH_STRCMP
strcmp(const char * cs,const char * ct)96 int strcmp(const char *cs, const char *ct)
97 {
98 int d0, d1;
99 int res;
100 asm volatile("1:\tlodsb\n\t"
101 "scasb\n\t"
102 "jne 2f\n\t"
103 "testb %%al,%%al\n\t"
104 "jne 1b\n\t"
105 "xorl %%eax,%%eax\n\t"
106 "jmp 3f\n"
107 "2:\tsbbl %%eax,%%eax\n\t"
108 "orb $1,%%al\n"
109 "3:"
110 : "=a" (res), "=&S" (d0), "=&D" (d1)
111 : "1" (cs), "2" (ct)
112 : "memory");
113 return res;
114 }
115 EXPORT_SYMBOL(strcmp);
116 #endif
117
118 #ifdef __HAVE_ARCH_STRNCMP
strncmp(const char * cs,const char * ct,size_t count)119 int strncmp(const char *cs, const char *ct, size_t count)
120 {
121 int res;
122 int d0, d1, d2;
123 asm volatile("1:\tdecl %3\n\t"
124 "js 2f\n\t"
125 "lodsb\n\t"
126 "scasb\n\t"
127 "jne 3f\n\t"
128 "testb %%al,%%al\n\t"
129 "jne 1b\n"
130 "2:\txorl %%eax,%%eax\n\t"
131 "jmp 4f\n"
132 "3:\tsbbl %%eax,%%eax\n\t"
133 "orb $1,%%al\n"
134 "4:"
135 : "=a" (res), "=&S" (d0), "=&D" (d1), "=&c" (d2)
136 : "1" (cs), "2" (ct), "3" (count)
137 : "memory");
138 return res;
139 }
140 EXPORT_SYMBOL(strncmp);
141 #endif
142
143 #ifdef __HAVE_ARCH_STRCHR
strchr(const char * s,int c)144 char *strchr(const char *s, int c)
145 {
146 int d0;
147 char *res;
148 asm volatile("movb %%al,%%ah\n"
149 "1:\tlodsb\n\t"
150 "cmpb %%ah,%%al\n\t"
151 "je 2f\n\t"
152 "testb %%al,%%al\n\t"
153 "jne 1b\n\t"
154 "movl $1,%1\n"
155 "2:\tmovl %1,%0\n\t"
156 "decl %0"
157 : "=a" (res), "=&S" (d0)
158 : "1" (s), "0" (c)
159 : "memory");
160 return res;
161 }
162 EXPORT_SYMBOL(strchr);
163 #endif
164
165 #ifdef __HAVE_ARCH_STRLEN
strlen(const char * s)166 size_t strlen(const char *s)
167 {
168 int d0;
169 size_t res;
170 asm volatile("repne\n\t"
171 "scasb"
172 : "=c" (res), "=&D" (d0)
173 : "1" (s), "a" (0), "0" (0xffffffffu)
174 : "memory");
175 return ~res - 1;
176 }
177 EXPORT_SYMBOL(strlen);
178 #endif
179
180 #ifdef __HAVE_ARCH_MEMCHR
memchr(const void * cs,int c,size_t count)181 void *memchr(const void *cs, int c, size_t count)
182 {
183 int d0;
184 void *res;
185 if (!count)
186 return NULL;
187 asm volatile("repne\n\t"
188 "scasb\n\t"
189 "je 1f\n\t"
190 "movl $1,%0\n"
191 "1:\tdecl %0"
192 : "=D" (res), "=&c" (d0)
193 : "a" (c), "0" (cs), "1" (count)
194 : "memory");
195 return res;
196 }
197 EXPORT_SYMBOL(memchr);
198 #endif
199
200 #ifdef __HAVE_ARCH_MEMSCAN
memscan(void * addr,int c,size_t size)201 void *memscan(void *addr, int c, size_t size)
202 {
203 if (!size)
204 return addr;
205 asm volatile("repnz; scasb\n\t"
206 "jnz 1f\n\t"
207 "dec %%edi\n"
208 "1:"
209 : "=D" (addr), "=c" (size)
210 : "0" (addr), "1" (size), "a" (c)
211 : "memory");
212 return addr;
213 }
214 EXPORT_SYMBOL(memscan);
215 #endif
216
217 #ifdef __HAVE_ARCH_STRNLEN
strnlen(const char * s,size_t count)218 size_t strnlen(const char *s, size_t count)
219 {
220 int d0;
221 int res;
222 asm volatile("movl %2,%0\n\t"
223 "jmp 2f\n"
224 "1:\tcmpb $0,(%0)\n\t"
225 "je 3f\n\t"
226 "incl %0\n"
227 "2:\tdecl %1\n\t"
228 "cmpl $-1,%1\n\t"
229 "jne 1b\n"
230 "3:\tsubl %2,%0"
231 : "=a" (res), "=&d" (d0)
232 : "c" (s), "1" (count)
233 : "memory");
234 return res;
235 }
236 EXPORT_SYMBOL(strnlen);
237 #endif
238