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_MEMCHR	/* inline & arch function */
16 #define __HAVE_ARCH_MEMCMP	/* arch function */
17 #define __HAVE_ARCH_MEMCPY	/* gcc builtin & arch function */
18 #define __HAVE_ARCH_MEMMOVE	/* gcc builtin & arch function */
19 #define __HAVE_ARCH_MEMSCAN	/* inline & arch function */
20 #define __HAVE_ARCH_MEMSET	/* gcc builtin & arch function */
21 #define __HAVE_ARCH_MEMSET16	/* arch function */
22 #define __HAVE_ARCH_MEMSET32	/* arch function */
23 #define __HAVE_ARCH_MEMSET64	/* arch function */
24 #define __HAVE_ARCH_STRCAT	/* inline & arch function */
25 #define __HAVE_ARCH_STRCMP	/* arch function */
26 #define __HAVE_ARCH_STRCPY	/* inline & arch function */
27 #define __HAVE_ARCH_STRLCAT	/* arch function */
28 #define __HAVE_ARCH_STRLCPY	/* arch function */
29 #define __HAVE_ARCH_STRLEN	/* inline & arch function */
30 #define __HAVE_ARCH_STRNCAT	/* arch function */
31 #define __HAVE_ARCH_STRNCPY	/* arch function */
32 #define __HAVE_ARCH_STRNLEN	/* inline & arch function */
33 #define __HAVE_ARCH_STRRCHR	/* arch function */
34 #define __HAVE_ARCH_STRSTR	/* arch function */
35 
36 /* Prototypes for non-inlined arch strings functions. */
37 int memcmp(const void *s1, const void *s2, size_t n);
38 void *memcpy(void *dest, const void *src, size_t n);
39 void *memset(void *s, int c, size_t n);
40 void *memmove(void *dest, const void *src, size_t n);
41 int strcmp(const char *s1, const char *s2);
42 size_t strlcat(char *dest, const char *src, size_t n);
43 size_t strlcpy(char *dest, const char *src, size_t size);
44 char *strncat(char *dest, const char *src, size_t n);
45 char *strncpy(char *dest, const char *src, size_t n);
46 char *strrchr(const char *s, int c);
47 char *strstr(const char *s1, const char *s2);
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 void *__memset16(uint16_t *s, uint16_t v, size_t count);
57 void *__memset32(uint32_t *s, uint32_t v, size_t count);
58 void *__memset64(uint64_t *s, uint64_t v, size_t count);
59 
memset16(uint16_t * s,uint16_t v,size_t count)60 static inline void *memset16(uint16_t *s, uint16_t v, size_t count)
61 {
62 	return __memset16(s, v, count * sizeof(v));
63 }
64 
memset32(uint32_t * s,uint32_t v,size_t count)65 static inline void *memset32(uint32_t *s, uint32_t v, size_t count)
66 {
67 	return __memset32(s, v, count * sizeof(v));
68 }
69 
memset64(uint64_t * s,uint64_t v,size_t count)70 static inline void *memset64(uint64_t *s, uint64_t v, size_t count)
71 {
72 	return __memset64(s, v, count * sizeof(v));
73 }
74 
75 #if !defined(IN_ARCH_STRING_C) && (!defined(CONFIG_FORTIFY_SOURCE) || defined(__NO_FORTIFY))
76 
memchr(const void * s,int c,size_t n)77 static inline void *memchr(const void * s, int c, size_t n)
78 {
79 	register int r0 asm("0") = (char) c;
80 	const void *ret = s + n;
81 
82 	asm volatile(
83 		"0:	srst	%0,%1\n"
84 		"	jo	0b\n"
85 		"	jl	1f\n"
86 		"	la	%0,0\n"
87 		"1:"
88 		: "+a" (ret), "+&a" (s) : "d" (r0) : "cc", "memory");
89 	return (void *) ret;
90 }
91 
memscan(void * s,int c,size_t n)92 static inline void *memscan(void *s, int c, size_t n)
93 {
94 	register int r0 asm("0") = (char) c;
95 	const void *ret = s + n;
96 
97 	asm volatile(
98 		"0:	srst	%0,%1\n"
99 		"	jo	0b\n"
100 		: "+a" (ret), "+&a" (s) : "d" (r0) : "cc", "memory");
101 	return (void *) ret;
102 }
103 
strcat(char * dst,const char * src)104 static inline char *strcat(char *dst, const char *src)
105 {
106 	register int r0 asm("0") = 0;
107 	unsigned long dummy;
108 	char *ret = dst;
109 
110 	asm volatile(
111 		"0:	srst	%0,%1\n"
112 		"	jo	0b\n"
113 		"1:	mvst	%0,%2\n"
114 		"	jo	1b"
115 		: "=&a" (dummy), "+a" (dst), "+a" (src)
116 		: "d" (r0), "0" (0) : "cc", "memory" );
117 	return ret;
118 }
119 
strcpy(char * dst,const char * src)120 static inline char *strcpy(char *dst, const char *src)
121 {
122 	register int r0 asm("0") = 0;
123 	char *ret = dst;
124 
125 	asm volatile(
126 		"0:	mvst	%0,%1\n"
127 		"	jo	0b"
128 		: "+&a" (dst), "+&a" (src) : "d" (r0)
129 		: "cc", "memory");
130 	return ret;
131 }
132 
strlen(const char * s)133 static inline size_t strlen(const char *s)
134 {
135 	register unsigned long r0 asm("0") = 0;
136 	const char *tmp = s;
137 
138 	asm volatile(
139 		"0:	srst	%0,%1\n"
140 		"	jo	0b"
141 		: "+d" (r0), "+a" (tmp) :  : "cc", "memory");
142 	return r0 - (unsigned long) s;
143 }
144 
strnlen(const char * s,size_t n)145 static inline size_t strnlen(const char * s, size_t n)
146 {
147 	register int r0 asm("0") = 0;
148 	const char *tmp = s;
149 	const char *end = s + n;
150 
151 	asm volatile(
152 		"0:	srst	%0,%1\n"
153 		"	jo	0b"
154 		: "+a" (end), "+a" (tmp) : "d" (r0)  : "cc", "memory");
155 	return end - s;
156 }
157 #else /* IN_ARCH_STRING_C */
158 void *memchr(const void * s, int c, size_t n);
159 void *memscan(void *s, int c, size_t n);
160 char *strcat(char *dst, const char *src);
161 char *strcpy(char *dst, const char *src);
162 size_t strlen(const char *s);
163 size_t strnlen(const char * s, size_t n);
164 #endif /* !IN_ARCH_STRING_C */
165 
166 #endif /* __S390_STRING_H_ */
167