[mirror_ubuntu-bionic-kernel.git] / arch / x86 / include / asm / string_64.h

#ifndef _ASM_X86_STRING_64_H
#define _ASM_X86_STRING_64_H

#ifdef __KERNEL__

/* Written 2002 by Andi Kleen */

/* Only used for special circumstances. Stolen from i386/string.h */
static __always_inline void *__inline_memcpy(void *to, const void *from, size_t n)
{
	unsigned long d0, d1, d2;
	asm volatile("rep ; movsl\n\t"
		     "testb $2,%b4\n\t"
		     "je 1f\n\t"
		     "movsw\n"
		     "1:\ttestb $1,%b4\n\t"
		     "je 2f\n\t"
		     "movsb\n"
		     "2:"
		     : "=&c" (d0), "=&D" (d1), "=&S" (d2)
		     : "0" (n / 4), "q" (n), "1" ((long)to), "2" ((long)from)
		     : "memory");
	return to;
}

/* Even with __builtin_ the compiler may decide to use the out of line
   function. */

#define __HAVE_ARCH_MEMCPY 1
extern void *memcpy(void *to, const void *from, size_t len);
extern void *__memcpy(void *to, const void *from, size_t len);

#ifndef CONFIG_KMEMCHECK
#if (__GNUC__ == 4 && __GNUC_MINOR__ < 3) || __GNUC__ < 4
#define memcpy(dst, src, len)					\
({								\
	size_t __len = (len);					\
	void *__ret;						\
	if (__builtin_constant_p(len) && __len >= 64)		\
		__ret = __memcpy((dst), (src), __len);		\
	else							\
		__ret = __builtin_memcpy((dst), (src), __len);	\
	__ret;							\
})
#endif
#else
/*
 * kmemcheck becomes very happy if we use the REP instructions unconditionally,
 * because it means that we know both memory operands in advance.
 */
#define memcpy(dst, src, len) __inline_memcpy((dst), (src), (len))
#endif

#define __HAVE_ARCH_MEMSET
void *memset(void *s, int c, size_t n);
void *__memset(void *s, int c, size_t n);

#define __HAVE_ARCH_MEMMOVE
void *memmove(void *dest, const void *src, size_t count);
void *__memmove(void *dest, const void *src, size_t count);

int memcmp(const void *cs, const void *ct, size_t count);
size_t strlen(const char *s);
char *strcpy(char *dest, const char *src);
char *strcat(char *dest, const char *src);
int strcmp(const char *cs, const char *ct);

#if defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__)

/*
 * For files that not instrumented (e.g. mm/slub.c) we
 * should use not instrumented version of mem* functions.
 */

#undef memcpy
#define memcpy(dst, src, len) __memcpy(dst, src, len)
#define memmove(dst, src, len) __memmove(dst, src, len)
#define memset(s, c, n) __memset(s, c, n)
#endif

/**
 * memcpy_mcsafe - copy memory with indication if a machine check happened
 *
 * @dst:	destination address
 * @src:	source address
 * @cnt:	number of bytes to copy
 *
 * Low level memory copy function that catches machine checks
 *
 * Return 0 for success, -EFAULT for fail
 */
int memcpy_mcsafe(void *dst, const void *src, size_t cnt);

#endif /* __KERNEL__ */

#endif /* _ASM_X86_STRING_64_H */
Commit	Line	Data
1965aae3 PA	1	#ifndef _ASM_X86_STRING_64_H
1965aae3 PA	2	#define _ASM_X86_STRING_64_H
1da177e4 LT	3
	4	#ifdef __KERNEL__
	5
953b2f1e	6	/* Written 2002 by Andi Kleen */
1da177e4	7
953b2f1e JP	8	/* Only used for special circumstances. Stolen from i386/string.h */
953b2f1e JP	9	static __always_inline void __inline_memcpy(void to, const void *from, size_t n)
1da177e4	10	{
953b2f1e JP	11	unsigned long d0, d1, d2;
	12	asm volatile("rep ; movsl\n\t"
	13	"testb $2,%b4\n\t"
	14	"je 1f\n\t"
	15	"movsw\n"
	16	"1:\ttestb $1,%b4\n\t"
	17	"je 2f\n\t"
	18	"movsb\n"
	19	"2:"
	20	: "=&c" (d0), "=&D" (d1), "=&S" (d2)
	21	: "0" (n / 4), "q" (n), "1" ((long)to), "2" ((long)from)
	22	: "memory");
	23	return to;
1da177e4 LT	24	}
	25
	26	/* Even with __builtin_ the compiler may decide to use the out of line
	27	function. */
	28
	29	#define __HAVE_ARCH_MEMCPY 1
a75ca545	30	extern void memcpy(void to, const void *from, size_t len);
393f203f AR	31	extern void __memcpy(void to, const void *from, size_t len);
393f203f AR	32
f8561296	33	#ifndef CONFIG_KMEMCHECK
a75ca545	34	#if (__GNUC__ == 4 && __GNUC_MINOR__ < 3) \|\| __GNUC__ < 4
953b2f1e JP	35	#define memcpy(dst, src, len) \
	36	({ \
	37	size_t __len = (len); \
	38	void *__ret; \
	39	if (__builtin_constant_p(len) && __len >= 64) \
	40	__ret = __memcpy((dst), (src), __len); \
	41	else \
	42	__ret = __builtin_memcpy((dst), (src), __len); \
	43	__ret; \
	44	})
aac57f81	45	#endif
f8561296 VN	46	#else
	47	/*
	48	* kmemcheck becomes very happy if we use the REP instructions unconditionally,
	49	* because it means that we know both memory operands in advance.
	50	*/
	51	#define memcpy(dst, src, len) __inline_memcpy((dst), (src), (len))
	52	#endif
1da177e4 LT	53
1da177e4 LT	54	#define __HAVE_ARCH_MEMSET
6edfba1b	55	void memset(void s, int c, size_t n);
393f203f	56	void __memset(void s, int c, size_t n);
1da177e4 LT	57
1da177e4 LT	58	#define __HAVE_ARCH_MEMMOVE
953b2f1e	59	void memmove(void dest, const void *src, size_t count);
393f203f	60	void __memmove(void dest, const void *src, size_t count);
1da177e4	61
953b2f1e JP	62	int memcmp(const void cs, const void ct, size_t count);
	63	size_t strlen(const char *s);
	64	char strcpy(char dest, const char *src);
	65	char strcat(char dest, const char *src);
	66	int strcmp(const char cs, const char ct);
1da177e4	67
393f203f AR	68	#if defined(CONFIG_KASAN) && !defined(__SANITIZE_ADDRESS__)
	69
	70	/*
	71	* For files that not instrumented (e.g. mm/slub.c) we
	72	* should use not instrumented version of mem* functions.
	73	*/
	74
	75	#undef memcpy
	76	#define memcpy(dst, src, len) __memcpy(dst, src, len)
	77	#define memmove(dst, src, len) __memmove(dst, src, len)
	78	#define memset(s, c, n) __memset(s, c, n)
	79	#endif
	80
92b0729c TL	81	/**
	82	* memcpy_mcsafe - copy memory with indication if a machine check happened
	83	*
	84	* @dst: destination address
	85	* @src: source address
	86	* @cnt: number of bytes to copy
	87	*
	88	* Low level memory copy function that catches machine checks
	89	*
cbf8b5a2	90	* Return 0 for success, -EFAULT for fail
92b0729c	91	*/
cbf8b5a2	92	int memcpy_mcsafe(void dst, const void src, size_t cnt);
92b0729c	93
1da177e4 LT	94	#endif /* __KERNEL__ */
1da177e4 LT	95
1965aae3	96	#endif /* _ASM_X86_STRING_64_H */