1/* SPDX-License-Identifier: GPL-2.0 */
2/*
3 *  S390 version
4 *    Copyright IBM Corp. 1999
5 *    Author(s): Martin Schwidefsky (schwidefsky@de.ibm.com),
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	/* gcc builtin & arch function */
16#define __HAVE_ARCH_MEMMOVE	/* gcc builtin & arch function */
17#define __HAVE_ARCH_MEMSET	/* gcc builtin & arch function */
18#define __HAVE_ARCH_MEMSET16	/* arch function */
19#define __HAVE_ARCH_MEMSET32	/* arch function */
20#define __HAVE_ARCH_MEMSET64	/* arch function */
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	/* inline & arch function */
28#define __HAVE_ARCH_MEMCMP	/* arch function */
29#define __HAVE_ARCH_MEMSCAN	/* inline & arch function */
30#define __HAVE_ARCH_STRCAT	/* inline & arch function */
31#define __HAVE_ARCH_STRCMP	/* arch function */
32#define __HAVE_ARCH_STRCPY	/* inline & arch function */
33#define __HAVE_ARCH_STRLCAT	/* arch function */
34#define __HAVE_ARCH_STRLEN	/* inline & arch function */
35#define __HAVE_ARCH_STRNCAT	/* arch function */
36#define __HAVE_ARCH_STRNCPY	/* arch function */
37#define __HAVE_ARCH_STRNLEN	/* inline & arch function */
38#define __HAVE_ARCH_STRSTR	/* arch function */
39
40/* Prototypes for non-inlined arch strings functions. */
41int memcmp(const void *s1, const void *s2, size_t n);
42int strcmp(const char *s1, const char *s2);
43size_t strlcat(char *dest, const char *src, size_t n);
44char *strncat(char *dest, const char *src, size_t n);
45char *strncpy(char *dest, const char *src, size_t n);
46char *strstr(const char *s1, const char *s2);
47#endif /* !CONFIG_KASAN */
48
49#undef __HAVE_ARCH_STRCHR
50#undef __HAVE_ARCH_STRNCHR
51#undef __HAVE_ARCH_STRNCMP
52#undef __HAVE_ARCH_STRPBRK
53#undef __HAVE_ARCH_STRSEP
54#undef __HAVE_ARCH_STRSPN
55
56#if defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__)
57
58#define strlen(s) __strlen(s)
59
60#define __no_sanitize_prefix_strfunc(x) __##x
61
62#ifndef __NO_FORTIFY
63#define __NO_FORTIFY /* FORTIFY_SOURCE uses __builtin_memcpy, etc. */
64#endif
65
66#else
67#define __no_sanitize_prefix_strfunc(x) x
68#endif /* defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__) */
69
70void *__memcpy(void *dest, const void *src, size_t n);
71void *__memset(void *s, int c, size_t n);
72void *__memmove(void *dest, const void *src, size_t n);
73void *__memset16(uint16_t *s, uint16_t v, size_t count);
74void *__memset32(uint32_t *s, uint32_t v, size_t count);
75void *__memset64(uint64_t *s, uint64_t v, size_t count);
76
77static inline void *memset16(uint16_t *s, uint16_t v, size_t count)
78{
79	return __memset16(s, v, count * sizeof(v));
80}
81
82static inline void *memset32(uint32_t *s, uint32_t v, size_t count)
83{
84	return __memset32(s, v, count * sizeof(v));
85}
86
87static inline void *memset64(uint64_t *s, uint64_t v, size_t count)
88{
89	return __memset64(s, v, count * sizeof(v));
90}
91
92#if !defined(IN_ARCH_STRING_C) && (!defined(CONFIG_FORTIFY_SOURCE) || defined(__NO_FORTIFY))
93
94#ifdef __HAVE_ARCH_MEMCHR
95static inline void *memchr(const void * s, int c, size_t n)
96{
97	const void *ret = s + n;
98
99	asm volatile(
100		"	lgr	0,%[c]\n"
101		"0:	srst	%[ret],%[s]\n"
102		"	jo	0b\n"
103		"	jl	1f\n"
104		"	la	%[ret],0\n"
105		"1:"
106		: [ret] "+&a" (ret), [s] "+&a" (s)
107		: [c] "d" (c)
108		: "cc", "memory", "0");
109	return (void *) ret;
110}
111#endif
112
113#ifdef __HAVE_ARCH_MEMSCAN
114static inline void *memscan(void *s, int c, size_t n)
115{
116	const void *ret = s + n;
117
118	asm volatile(
119		"	lgr	0,%[c]\n"
120		"0:	srst	%[ret],%[s]\n"
121		"	jo	0b\n"
122		: [ret] "+&a" (ret), [s] "+&a" (s)
123		: [c] "d" (c)
124		: "cc", "memory", "0");
125	return (void *) ret;
126}
127#endif
128
129#ifdef __HAVE_ARCH_STRCAT
130static inline char *strcat(char *dst, const char *src)
131{
132	unsigned long dummy = 0;
133	char *ret = dst;
134
135	asm volatile(
136		"	lghi	0,0\n"
137		"0:	srst	%[dummy],%[dst]\n"
138		"	jo	0b\n"
139		"1:	mvst	%[dummy],%[src]\n"
140		"	jo	1b"
141		: [dummy] "+&a" (dummy), [dst] "+&a" (dst), [src] "+&a" (src)
142		:
143		: "cc", "memory", "0");
144	return ret;
145}
146#endif
147
148#ifdef __HAVE_ARCH_STRCPY
149static inline char *strcpy(char *dst, const char *src)
150{
151	char *ret = dst;
152
153	asm volatile(
154		"	lghi	0,0\n"
155		"0:	mvst	%[dst],%[src]\n"
156		"	jo	0b"
157		: [dst] "+&a" (dst), [src] "+&a" (src)
158		:
159		: "cc", "memory", "0");
160	return ret;
161}
162#endif
163
164#if defined(__HAVE_ARCH_STRLEN) || (defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__))
165static inline size_t __no_sanitize_prefix_strfunc(strlen)(const char *s)
166{
167	unsigned long end = 0;
168	const char *tmp = s;
169
170	asm volatile(
171		"	lghi	0,0\n"
172		"0:	srst	%[end],%[tmp]\n"
173		"	jo	0b"
174		: [end] "+&a" (end), [tmp] "+&a" (tmp)
175		:
176		: "cc", "memory", "0");
177	return end - (unsigned long)s;
178}
179#endif
180
181#ifdef __HAVE_ARCH_STRNLEN
182static inline size_t strnlen(const char * s, size_t n)
183{
184	const char *tmp = s;
185	const char *end = s + n;
186
187	asm volatile(
188		"	lghi	0,0\n"
189		"0:	srst	%[end],%[tmp]\n"
190		"	jo	0b"
191		: [end] "+&a" (end), [tmp] "+&a" (tmp)
192		:
193		: "cc", "memory", "0");
194	return end - s;
195}
196#endif
197#else /* IN_ARCH_STRING_C */
198void *memchr(const void * s, int c, size_t n);
199void *memscan(void *s, int c, size_t n);
200char *strcat(char *dst, const char *src);
201char *strcpy(char *dst, const char *src);
202size_t strlen(const char *s);
203size_t strnlen(const char * s, size_t n);
204#endif /* !IN_ARCH_STRING_C */
205
206#endif /* __S390_STRING_H_ */
207