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