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 19 void *memcpy(void *dest, const void *src, size_t n); 20 void *memset(void *s, int c, size_t n); 21 void *memmove(void *dest, const void *src, size_t n); 22 23 #if !defined(CONFIG_KASAN) && !defined(CONFIG_KMSAN) 24 #define __HAVE_ARCH_MEMCHR /* inline & arch function */ 25 #define __HAVE_ARCH_MEMCMP /* arch function */ 26 #define __HAVE_ARCH_MEMSCAN /* inline & arch function */ 27 #define __HAVE_ARCH_STRCAT /* inline & arch function */ 28 #define __HAVE_ARCH_STRCMP /* arch function */ 29 #define __HAVE_ARCH_STRCPY /* inline & arch function */ 30 #define __HAVE_ARCH_STRLCAT /* arch function */ 31 #define __HAVE_ARCH_STRLEN /* inline & arch function */ 32 #define __HAVE_ARCH_STRNCAT /* arch function */ 33 #define __HAVE_ARCH_STRNCPY /* arch function */ 34 #define __HAVE_ARCH_STRNLEN /* inline & arch function */ 35 #define __HAVE_ARCH_STRSTR /* arch function */ 36 #define __HAVE_ARCH_MEMSET16 /* arch function */ 37 #define __HAVE_ARCH_MEMSET32 /* arch function */ 38 #define __HAVE_ARCH_MEMSET64 /* arch function */ 39 40 /* Prototypes for non-inlined arch strings functions. */ 41 int memcmp(const void *s1, const void *s2, size_t n); 42 int strcmp(const char *s1, const char *s2); 43 size_t strlcat(char *dest, const char *src, size_t n); 44 char *strncat(char *dest, const char *src, size_t n); 45 char *strncpy(char *dest, const char *src, size_t n); 46 char *strstr(const char *s1, const char *s2); 47 #endif /* !defined(CONFIG_KASAN) && !defined(CONFIG_KMSAN) */ 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 70 void *__memcpy(void *dest, const void *src, size_t n); 71 void *__memset(void *s, int c, size_t n); 72 void *__memmove(void *dest, const void *src, size_t n); 73 void *__memset16(uint16_t *s, uint16_t v, size_t count); 74 void *__memset32(uint32_t *s, uint32_t v, size_t count); 75 void *__memset64(uint64_t *s, uint64_t v, size_t count); 76 77 #ifdef __HAVE_ARCH_MEMSET16 78 static inline void *memset16(uint16_t *s, uint16_t v, size_t count) 79 { 80 return __memset16(s, v, count * sizeof(v)); 81 } 82 #endif 83 84 #ifdef __HAVE_ARCH_MEMSET32 85 static inline void *memset32(uint32_t *s, uint32_t v, size_t count) 86 { 87 return __memset32(s, v, count * sizeof(v)); 88 } 89 #endif 90 91 #ifdef __HAVE_ARCH_MEMSET64 92 #ifdef IN_BOOT_STRING_C 93 void *memset64(uint64_t *s, uint64_t v, size_t count); 94 #else 95 static inline void *memset64(uint64_t *s, uint64_t v, size_t count) 96 { 97 return __memset64(s, v, count * sizeof(v)); 98 } 99 #endif 100 #endif 101 102 #if !defined(IN_ARCH_STRING_C) && (!defined(CONFIG_FORTIFY_SOURCE) || defined(__NO_FORTIFY)) 103 104 #ifdef __HAVE_ARCH_MEMCHR 105 static inline void *memchr(const void * s, int c, size_t n) 106 { 107 const void *ret = s + n; 108 109 asm volatile( 110 " lgr 0,%[c]\n" 111 "0: srst %[ret],%[s]\n" 112 " jo 0b\n" 113 " jl 1f\n" 114 " la %[ret],0\n" 115 "1:" 116 : [ret] "+&a" (ret), [s] "+&a" (s) 117 : [c] "d" (c) 118 : "cc", "memory", "0"); 119 return (void *) ret; 120 } 121 #endif 122 123 #ifdef __HAVE_ARCH_MEMSCAN 124 static inline void *memscan(void *s, int c, size_t n) 125 { 126 const void *ret = s + n; 127 128 asm volatile( 129 " lgr 0,%[c]\n" 130 "0: srst %[ret],%[s]\n" 131 " jo 0b\n" 132 : [ret] "+&a" (ret), [s] "+&a" (s) 133 : [c] "d" (c) 134 : "cc", "memory", "0"); 135 return (void *) ret; 136 } 137 #endif 138 139 #ifdef __HAVE_ARCH_STRCAT 140 static inline char *strcat(char *dst, const char *src) 141 { 142 unsigned long dummy = 0; 143 char *ret = dst; 144 145 asm volatile( 146 " lghi 0,0\n" 147 "0: srst %[dummy],%[dst]\n" 148 " jo 0b\n" 149 "1: mvst %[dummy],%[src]\n" 150 " jo 1b" 151 : [dummy] "+&a" (dummy), [dst] "+&a" (dst), [src] "+&a" (src) 152 : 153 : "cc", "memory", "0"); 154 return ret; 155 } 156 #endif 157 158 #ifdef __HAVE_ARCH_STRCPY 159 static inline char *strcpy(char *dst, const char *src) 160 { 161 char *ret = dst; 162 163 asm volatile( 164 " lghi 0,0\n" 165 "0: mvst %[dst],%[src]\n" 166 " jo 0b" 167 : [dst] "+&a" (dst), [src] "+&a" (src) 168 : 169 : "cc", "memory", "0"); 170 return ret; 171 } 172 #endif 173 174 #if defined(__HAVE_ARCH_STRLEN) || (defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__)) 175 static inline size_t __no_sanitize_prefix_strfunc(strlen)(const char *s) 176 { 177 unsigned long end = 0; 178 const char *tmp = s; 179 180 asm volatile( 181 " lghi 0,0\n" 182 "0: srst %[end],%[tmp]\n" 183 " jo 0b" 184 : [end] "+&a" (end), [tmp] "+&a" (tmp) 185 : 186 : "cc", "memory", "0"); 187 return end - (unsigned long)s; 188 } 189 #endif 190 191 #ifdef __HAVE_ARCH_STRNLEN 192 static inline size_t strnlen(const char * s, size_t n) 193 { 194 const char *tmp = s; 195 const char *end = s + n; 196 197 asm volatile( 198 " lghi 0,0\n" 199 "0: srst %[end],%[tmp]\n" 200 " jo 0b" 201 : [end] "+&a" (end), [tmp] "+&a" (tmp) 202 : 203 : "cc", "memory", "0"); 204 return end - s; 205 } 206 #endif 207 #else /* IN_ARCH_STRING_C */ 208 void *memchr(const void * s, int c, size_t n); 209 void *memscan(void *s, int c, size_t n); 210 char *strcat(char *dst, const char *src); 211 char *strcpy(char *dst, const char *src); 212 size_t strlen(const char *s); 213 size_t strnlen(const char * s, size_t n); 214 #endif /* !IN_ARCH_STRING_C */ 215 216 #endif /* __S390_STRING_H_ */ 217