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