1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef _ASM_X86_STRING_64_H 3 #define _ASM_X86_STRING_64_H 4 5 #ifdef __KERNEL__ 6 #include <linux/jump_label.h> 7 8 /* Written 2002 by Andi Kleen */ 9 10 /* Even with __builtin_ the compiler may decide to use the out of line 11 function. */ 12 13 #if defined(__SANITIZE_MEMORY__) && defined(__NO_FORTIFY) 14 #include <linux/kmsan_string.h> 15 #endif 16 17 #define __HAVE_ARCH_MEMCPY 1 18 #if defined(__SANITIZE_MEMORY__) && defined(__NO_FORTIFY) 19 #undef memcpy 20 #define memcpy __msan_memcpy 21 #else 22 extern void *memcpy(void *to, const void *from, size_t len); 23 #endif 24 extern void *__memcpy(void *to, const void *from, size_t len); 25 26 #define __HAVE_ARCH_MEMSET 27 #if defined(__SANITIZE_MEMORY__) && defined(__NO_FORTIFY) 28 extern void *__msan_memset(void *s, int c, size_t n); 29 #undef memset 30 #define memset __msan_memset 31 #else 32 void *memset(void *s, int c, size_t n); 33 #endif 34 void *__memset(void *s, int c, size_t n); 35 36 #define __HAVE_ARCH_MEMSET16 37 static inline void *memset16(uint16_t *s, uint16_t v, size_t n) 38 { 39 long d0, d1; 40 asm volatile("rep\n\t" 41 "stosw" 42 : "=&c" (d0), "=&D" (d1) 43 : "a" (v), "1" (s), "0" (n) 44 : "memory"); 45 return s; 46 } 47 48 #define __HAVE_ARCH_MEMSET32 49 static inline void *memset32(uint32_t *s, uint32_t v, size_t n) 50 { 51 long d0, d1; 52 asm volatile("rep\n\t" 53 "stosl" 54 : "=&c" (d0), "=&D" (d1) 55 : "a" (v), "1" (s), "0" (n) 56 : "memory"); 57 return s; 58 } 59 60 #define __HAVE_ARCH_MEMSET64 61 static inline void *memset64(uint64_t *s, uint64_t v, size_t n) 62 { 63 long d0, d1; 64 asm volatile("rep\n\t" 65 "stosq" 66 : "=&c" (d0), "=&D" (d1) 67 : "a" (v), "1" (s), "0" (n) 68 : "memory"); 69 return s; 70 } 71 72 #define __HAVE_ARCH_MEMMOVE 73 #if defined(__SANITIZE_MEMORY__) && defined(__NO_FORTIFY) 74 #undef memmove 75 void *__msan_memmove(void *dest, const void *src, size_t len); 76 #define memmove __msan_memmove 77 #else 78 void *memmove(void *dest, const void *src, size_t count); 79 #endif 80 void *__memmove(void *dest, const void *src, size_t count); 81 82 int memcmp(const void *cs, const void *ct, size_t count); 83 size_t strlen(const char *s); 84 char *strcpy(char *dest, const char *src); 85 char *strcat(char *dest, const char *src); 86 int strcmp(const char *cs, const char *ct); 87 88 #if (defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__)) 89 /* 90 * For files that not instrumented (e.g. mm/slub.c) we 91 * should use not instrumented version of mem* functions. 92 */ 93 94 #undef memcpy 95 #define memcpy(dst, src, len) __memcpy(dst, src, len) 96 #undef memmove 97 #define memmove(dst, src, len) __memmove(dst, src, len) 98 #undef memset 99 #define memset(s, c, n) __memset(s, c, n) 100 101 #ifndef __NO_FORTIFY 102 #define __NO_FORTIFY /* FORTIFY_SOURCE uses __builtin_memcpy, etc. */ 103 #endif 104 105 #endif 106 107 #ifdef CONFIG_ARCH_HAS_UACCESS_FLUSHCACHE 108 #define __HAVE_ARCH_MEMCPY_FLUSHCACHE 1 109 void __memcpy_flushcache(void *dst, const void *src, size_t cnt); 110 static __always_inline void memcpy_flushcache(void *dst, const void *src, size_t cnt) 111 { 112 if (__builtin_constant_p(cnt)) { 113 switch (cnt) { 114 case 4: 115 asm ("movntil %1, %0" : "=m"(*(u32 *)dst) : "r"(*(u32 *)src)); 116 return; 117 case 8: 118 asm ("movntiq %1, %0" : "=m"(*(u64 *)dst) : "r"(*(u64 *)src)); 119 return; 120 case 16: 121 asm ("movntiq %1, %0" : "=m"(*(u64 *)dst) : "r"(*(u64 *)src)); 122 asm ("movntiq %1, %0" : "=m"(*(u64 *)(dst + 8)) : "r"(*(u64 *)(src + 8))); 123 return; 124 } 125 } 126 __memcpy_flushcache(dst, src, cnt); 127 } 128 #endif 129 130 #endif /* __KERNEL__ */ 131 132 #endif /* _ASM_X86_STRING_64_H */ 133