1
2
3
4
5
6
7
8#ifndef _S390_STRING_H_
9#define _S390_STRING_H_
10
11#ifndef _LINUX_TYPES_H
12#include <linux/types.h>
13#endif
14
15#define __HAVE_ARCH_MEMCPY
16#define __HAVE_ARCH_MEMMOVE
17#define __HAVE_ARCH_MEMSET
18#define __HAVE_ARCH_MEMSET16
19#define __HAVE_ARCH_MEMSET32
20#define __HAVE_ARCH_MEMSET64
21
22void *memcpy(void *dest, const void *src, size_t n);
23void *memset(void *s, int c, size_t n);
24void *memmove(void *dest, const void *src, size_t n);
25
26#ifndef CONFIG_KASAN
27#define __HAVE_ARCH_MEMCHR
28#define __HAVE_ARCH_MEMCMP
29#define __HAVE_ARCH_MEMSCAN
30#define __HAVE_ARCH_STRCAT
31#define __HAVE_ARCH_STRCMP
32#define __HAVE_ARCH_STRCPY
33#define __HAVE_ARCH_STRLCAT
34#define __HAVE_ARCH_STRLCPY
35#define __HAVE_ARCH_STRLEN
36#define __HAVE_ARCH_STRNCAT
37#define __HAVE_ARCH_STRNCPY
38#define __HAVE_ARCH_STRNLEN
39#define __HAVE_ARCH_STRRCHR
40#define __HAVE_ARCH_STRSTR
41
42
43int memcmp(const void *s1, const void *s2, size_t n);
44int strcmp(const char *s1, const char *s2);
45size_t strlcat(char *dest, const char *src, size_t n);
46size_t strlcpy(char *dest, const char *src, size_t size);
47char *strncat(char *dest, const char *src, size_t n);
48char *strncpy(char *dest, const char *src, size_t n);
49char *strrchr(const char *s, int c);
50char *strstr(const char *s1, const char *s2);
51#endif
52
53#undef __HAVE_ARCH_STRCHR
54#undef __HAVE_ARCH_STRNCHR
55#undef __HAVE_ARCH_STRNCMP
56#undef __HAVE_ARCH_STRPBRK
57#undef __HAVE_ARCH_STRSEP
58#undef __HAVE_ARCH_STRSPN
59
60#if defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__)
61
62extern void *__memcpy(void *dest, const void *src, size_t n);
63extern void *__memset(void *s, int c, size_t n);
64extern void *__memmove(void *dest, const void *src, size_t n);
65
66
67
68
69
70
71#define memcpy(dst, src, len) __memcpy(dst, src, len)
72#define memmove(dst, src, len) __memmove(dst, src, len)
73#define memset(s, c, n) __memset(s, c, n)
74#define strlen(s) __strlen(s)
75
76#define __no_sanitize_prefix_strfunc(x) __##x
77
78#ifndef __NO_FORTIFY
79#define __NO_FORTIFY
80#endif
81
82#else
83#define __no_sanitize_prefix_strfunc(x) x
84#endif
85
86void *__memset16(uint16_t *s, uint16_t v, size_t count);
87void *__memset32(uint32_t *s, uint32_t v, size_t count);
88void *__memset64(uint64_t *s, uint64_t v, size_t count);
89
90static inline void *memset16(uint16_t *s, uint16_t v, size_t count)
91{
92 return __memset16(s, v, count * sizeof(v));
93}
94
95static inline void *memset32(uint32_t *s, uint32_t v, size_t count)
96{
97 return __memset32(s, v, count * sizeof(v));
98}
99
100static inline void *memset64(uint64_t *s, uint64_t v, size_t count)
101{
102 return __memset64(s, v, count * sizeof(v));
103}
104
105#if !defined(IN_ARCH_STRING_C) && (!defined(CONFIG_FORTIFY_SOURCE) || defined(__NO_FORTIFY))
106
107#ifdef __HAVE_ARCH_MEMCHR
108static inline void *memchr(const void * s, int c, size_t n)
109{
110 const void *ret = s + n;
111
112 asm volatile(
113 " lgr 0,%[c]\n"
114 "0: srst %[ret],%[s]\n"
115 " jo 0b\n"
116 " jl 1f\n"
117 " la %[ret],0\n"
118 "1:"
119 : [ret] "+&a" (ret), [s] "+&a" (s)
120 : [c] "d" (c)
121 : "cc", "memory", "0");
122 return (void *) ret;
123}
124#endif
125
126#ifdef __HAVE_ARCH_MEMSCAN
127static inline void *memscan(void *s, int c, size_t n)
128{
129 const void *ret = s + n;
130
131 asm volatile(
132 " lgr 0,%[c]\n"
133 "0: srst %[ret],%[s]\n"
134 " jo 0b\n"
135 : [ret] "+&a" (ret), [s] "+&a" (s)
136 : [c] "d" (c)
137 : "cc", "memory", "0");
138 return (void *) ret;
139}
140#endif
141
142#ifdef __HAVE_ARCH_STRCAT
143static inline char *strcat(char *dst, const char *src)
144{
145 unsigned long dummy = 0;
146 char *ret = dst;
147
148 asm volatile(
149 " lghi 0,0\n"
150 "0: srst %[dummy],%[dst]\n"
151 " jo 0b\n"
152 "1: mvst %[dummy],%[src]\n"
153 " jo 1b"
154 : [dummy] "+&a" (dummy), [dst] "+&a" (dst), [src] "+&a" (src)
155 :
156 : "cc", "memory", "0");
157 return ret;
158}
159#endif
160
161#ifdef __HAVE_ARCH_STRCPY
162static inline char *strcpy(char *dst, const char *src)
163{
164 char *ret = dst;
165
166 asm volatile(
167 " lghi 0,0\n"
168 "0: mvst %[dst],%[src]\n"
169 " jo 0b"
170 : [dst] "+&a" (dst), [src] "+&a" (src)
171 :
172 : "cc", "memory", "0");
173 return ret;
174}
175#endif
176
177#if defined(__HAVE_ARCH_STRLEN) || (defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__))
178static inline size_t __no_sanitize_prefix_strfunc(strlen)(const char *s)
179{
180 unsigned long end = 0;
181 const char *tmp = s;
182
183 asm volatile(
184 " lghi 0,0\n"
185 "0: srst %[end],%[tmp]\n"
186 " jo 0b"
187 : [end] "+&a" (end), [tmp] "+&a" (tmp)
188 :
189 : "cc", "memory", "0");
190 return end - (unsigned long)s;
191}
192#endif
193
194#ifdef __HAVE_ARCH_STRNLEN
195static inline size_t strnlen(const char * s, size_t n)
196{
197 const char *tmp = s;
198 const char *end = s + n;
199
200 asm volatile(
201 " lghi 0,0\n"
202 "0: srst %[end],%[tmp]\n"
203 " jo 0b"
204 : [end] "+&a" (end), [tmp] "+&a" (tmp)
205 :
206 : "cc", "memory", "0");
207 return end - s;
208}
209#endif
210#else
211void *memchr(const void * s, int c, size_t n);
212void *memscan(void *s, int c, size_t n);
213char *strcat(char *dst, const char *src);
214char *strcpy(char *dst, const char *src);
215size_t strlen(const char *s);
216size_t strnlen(const char * s, size_t n);
217#endif
218
219#endif
220