Loading...
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * arch/arm/boot/compressed/string.c
4 *
5 * Small subset of simple string routines
6 */
7
8#define __NO_FORTIFY
9#include <linux/string.h>
10
11/*
12 * The decompressor is built without KASan but uses the same redirects as the
13 * rest of the kernel when CONFIG_KASAN is enabled, defining e.g. memcpy()
14 * to __memcpy() but since we are not linking with the main kernel string
15 * library in the decompressor, that will lead to link failures.
16 *
17 * Undefine KASan's versions, define the wrapped functions and alias them to
18 * the right names so that when e.g. __memcpy() appear in the code, it will
19 * still be linked to this local version of memcpy().
20 */
21#ifdef CONFIG_KASAN
22#undef memcpy
23#undef memmove
24#undef memset
25void *__memcpy(void *__dest, __const void *__src, size_t __n) __alias(memcpy);
26void *__memmove(void *__dest, __const void *__src, size_t count) __alias(memmove);
27void *__memset(void *s, int c, size_t count) __alias(memset);
28#endif
29
30void *memcpy(void *__dest, __const void *__src, size_t __n)
31{
32 int i = 0;
33 unsigned char *d = (unsigned char *)__dest, *s = (unsigned char *)__src;
34
35 for (i = __n >> 3; i > 0; i--) {
36 *d++ = *s++;
37 *d++ = *s++;
38 *d++ = *s++;
39 *d++ = *s++;
40 *d++ = *s++;
41 *d++ = *s++;
42 *d++ = *s++;
43 *d++ = *s++;
44 }
45
46 if (__n & 1 << 2) {
47 *d++ = *s++;
48 *d++ = *s++;
49 *d++ = *s++;
50 *d++ = *s++;
51 }
52
53 if (__n & 1 << 1) {
54 *d++ = *s++;
55 *d++ = *s++;
56 }
57
58 if (__n & 1)
59 *d++ = *s++;
60
61 return __dest;
62}
63
64void *memmove(void *__dest, __const void *__src, size_t count)
65{
66 unsigned char *d = __dest;
67 const unsigned char *s = __src;
68
69 if (__dest == __src)
70 return __dest;
71
72 if (__dest < __src)
73 return memcpy(__dest, __src, count);
74
75 while (count--)
76 d[count] = s[count];
77 return __dest;
78}
79
80size_t strlen(const char *s)
81{
82 const char *sc = s;
83
84 while (*sc != '\0')
85 sc++;
86 return sc - s;
87}
88
89size_t strnlen(const char *s, size_t count)
90{
91 const char *sc;
92
93 for (sc = s; count-- && *sc != '\0'; ++sc)
94 /* nothing */;
95 return sc - s;
96}
97
98int memcmp(const void *cs, const void *ct, size_t count)
99{
100 const unsigned char *su1 = cs, *su2 = ct, *end = su1 + count;
101 int res = 0;
102
103 while (su1 < end) {
104 res = *su1++ - *su2++;
105 if (res)
106 break;
107 }
108 return res;
109}
110
111int strcmp(const char *cs, const char *ct)
112{
113 unsigned char c1, c2;
114 int res = 0;
115
116 do {
117 c1 = *cs++;
118 c2 = *ct++;
119 res = c1 - c2;
120 if (res)
121 break;
122 } while (c1);
123 return res;
124}
125
126void *memchr(const void *s, int c, size_t count)
127{
128 const unsigned char *p = s;
129
130 while (count--)
131 if ((unsigned char)c == *p++)
132 return (void *)(p - 1);
133 return NULL;
134}
135
136char *strchr(const char *s, int c)
137{
138 while (*s != (char)c)
139 if (*s++ == '\0')
140 return NULL;
141 return (char *)s;
142}
143
144char *strrchr(const char *s, int c)
145{
146 const char *last = NULL;
147 do {
148 if (*s == (char)c)
149 last = s;
150 } while (*s++);
151 return (char *)last;
152}
153
154#undef memset
155
156void *memset(void *s, int c, size_t count)
157{
158 char *xs = s;
159 while (count--)
160 *xs++ = c;
161 return s;
162}
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * arch/arm/boot/compressed/string.c
4 *
5 * Small subset of simple string routines
6 */
7
8#define __NO_FORTIFY
9#include <linux/string.h>
10
11/*
12 * The decompressor is built without KASan but uses the same redirects as the
13 * rest of the kernel when CONFIG_KASAN is enabled, defining e.g. memcpy()
14 * to __memcpy() but since we are not linking with the main kernel string
15 * library in the decompressor, that will lead to link failures.
16 *
17 * Undefine KASan's versions, define the wrapped functions and alias them to
18 * the right names so that when e.g. __memcpy() appear in the code, it will
19 * still be linked to this local version of memcpy().
20 */
21#ifdef CONFIG_KASAN
22#undef memcpy
23#undef memmove
24#undef memset
25void *__memcpy(void *__dest, __const void *__src, size_t __n) __alias(memcpy);
26void *__memmove(void *__dest, __const void *__src, size_t count) __alias(memmove);
27void *__memset(void *s, int c, size_t count) __alias(memset);
28#endif
29
30void *memcpy(void *__dest, __const void *__src, size_t __n)
31{
32 int i = 0;
33 unsigned char *d = (unsigned char *)__dest, *s = (unsigned char *)__src;
34
35 for (i = __n >> 3; i > 0; i--) {
36 *d++ = *s++;
37 *d++ = *s++;
38 *d++ = *s++;
39 *d++ = *s++;
40 *d++ = *s++;
41 *d++ = *s++;
42 *d++ = *s++;
43 *d++ = *s++;
44 }
45
46 if (__n & 1 << 2) {
47 *d++ = *s++;
48 *d++ = *s++;
49 *d++ = *s++;
50 *d++ = *s++;
51 }
52
53 if (__n & 1 << 1) {
54 *d++ = *s++;
55 *d++ = *s++;
56 }
57
58 if (__n & 1)
59 *d++ = *s++;
60
61 return __dest;
62}
63
64void *memmove(void *__dest, __const void *__src, size_t count)
65{
66 unsigned char *d = __dest;
67 const unsigned char *s = __src;
68
69 if (__dest == __src)
70 return __dest;
71
72 if (__dest < __src)
73 return memcpy(__dest, __src, count);
74
75 while (count--)
76 d[count] = s[count];
77 return __dest;
78}
79
80size_t strlen(const char *s)
81{
82 const char *sc = s;
83
84 while (*sc != '\0')
85 sc++;
86 return sc - s;
87}
88
89size_t strnlen(const char *s, size_t count)
90{
91 const char *sc;
92
93 for (sc = s; count-- && *sc != '\0'; ++sc)
94 /* nothing */;
95 return sc - s;
96}
97
98int memcmp(const void *cs, const void *ct, size_t count)
99{
100 const unsigned char *su1 = cs, *su2 = ct, *end = su1 + count;
101 int res = 0;
102
103 while (su1 < end) {
104 res = *su1++ - *su2++;
105 if (res)
106 break;
107 }
108 return res;
109}
110
111int strcmp(const char *cs, const char *ct)
112{
113 unsigned char c1, c2;
114 int res = 0;
115
116 do {
117 c1 = *cs++;
118 c2 = *ct++;
119 res = c1 - c2;
120 if (res)
121 break;
122 } while (c1);
123 return res;
124}
125
126void *memchr(const void *s, int c, size_t count)
127{
128 const unsigned char *p = s;
129
130 while (count--)
131 if ((unsigned char)c == *p++)
132 return (void *)(p - 1);
133 return NULL;
134}
135
136char *strchr(const char *s, int c)
137{
138 while (*s != (char)c)
139 if (*s++ == '\0')
140 return NULL;
141 return (char *)s;
142}
143
144char *strrchr(const char *s, int c)
145{
146 const char *last = NULL;
147 do {
148 if (*s == (char)c)
149 last = s;
150 } while (*s++);
151 return (char *)last;
152}
153
154#undef memset
155
156void *memset(void *s, int c, size_t count)
157{
158 char *xs = s;
159 while (count--)
160 *xs++ = c;
161 return s;
162}