Small constant-sized flushcache copies currently inline only the 4, 8,
and 16-byte cases. Larger constant sizes fall back to
__memcpy_flushcache() even when the destination is naturally aligned.

Factor the movnti sequences into 4/8/16/32/64-byte helpers and extend
the inline fastpath coverage to the additional aligned constant sizes
32, 48, 64, 80, and 96 bytes. Keep the existing 4/8/16-byte cases
handled directly in memcpy_flushcache() so they do not pick up the
extra alignment gating used by the larger fixed-size helpers.

Because memcpy_streaming() maps aligned transfers to
memcpy_flushcache(), these additional fixed-size cases also stay on
the inline movnti path for that helper.

Issue the stores in ascending address order so write-combining sees a
forward stream. Keep all other sizes on __memcpy_flushcache(), and keep
zero-length copies returning immediately.

Signed-off-by: Li Zhe <[email protected]>
---
 arch/x86/include/asm/string_64.h | 122 ++++++++++++++++++++++++++-----
 1 file changed, 104 insertions(+), 18 deletions(-)

diff --git a/arch/x86/include/asm/string_64.h b/arch/x86/include/asm/string_64.h
index aee63108577f..16d1aac2da24 100644
--- a/arch/x86/include/asm/string_64.h
+++ b/arch/x86/include/asm/string_64.h
@@ -83,24 +83,6 @@ int strcmp(const char *cs, const char *ct);
 #ifdef CONFIG_ARCH_HAS_UACCESS_FLUSHCACHE
 #define __HAVE_ARCH_MEMCPY_FLUSHCACHE 1
 void __memcpy_flushcache(void *dst, const void *src, size_t cnt);
-static __always_inline void memcpy_flushcache(void *dst, const void *src, 
size_t cnt)
-{
-       if (__builtin_constant_p(cnt)) {
-               switch (cnt) {
-                       case 4:
-                               asm ("movntil %1, %0" : "=m"(*(u32 *)dst) : 
"r"(*(u32 *)src));
-                               return;
-                       case 8:
-                               asm ("movntiq %1, %0" : "=m"(*(u64 *)dst) : 
"r"(*(u64 *)src));
-                               return;
-                       case 16:
-                               asm ("movntiq %1, %0" : "=m"(*(u64 *)dst) : 
"r"(*(u64 *)src));
-                               asm ("movntiq %1, %0" : "=m"(*(u64 *)(dst + 8)) 
: "r"(*(u64 *)(src + 8)));
-                               return;
-               }
-       }
-       __memcpy_flushcache(dst, src, cnt);
-}
 
 /*
  * Only map memcpy_streaming() to memcpy_flushcache() when the destination
@@ -115,6 +97,110 @@ static __always_inline bool 
memcpy_flushcache_nt_safe(const void *dst,
        return cnt && IS_ALIGNED(d, 8) && IS_ALIGNED(cnt, 4);
 }
 
+static __always_inline void memcpy_flushcache_4(void *dst, const void *src)
+{
+       asm volatile("movntil %1, %0"
+                    : "=m"(*(u32 *)dst)
+                    : "r"(*(const u32 *)src)
+                    : "memory");
+}
+
+static __always_inline void memcpy_flushcache_8(void *dst, const void *src)
+{
+       asm volatile("movntiq %1, %0"
+                    : "=m"(*(u64 *)dst)
+                    : "r"(*(const u64 *)src)
+                    : "memory");
+}
+
+static __always_inline void memcpy_flushcache_16(void *dst,
+                                                const void *src)
+{
+       memcpy_flushcache_8(dst, src);
+       memcpy_flushcache_8(dst + 8, src + 8);
+}
+
+static __always_inline void memcpy_flushcache_32(void *dst,
+                                                const void *src)
+{
+       memcpy_flushcache_16(dst, src);
+       memcpy_flushcache_16(dst + 16, src + 16);
+}
+
+static __always_inline void memcpy_flushcache_64(void *dst,
+                                                const void *src)
+{
+       memcpy_flushcache_32(dst, src);
+       memcpy_flushcache_32(dst + 32, src + 32);
+}
+
+/*
+ * Keep the additional aligned fixed-size cases on the inline movnti path.
+ * Leave the existing 4/8/16-byte cases handled directly in
+ * memcpy_flushcache() so they do not pick up the extra alignment gating
+ * used by the larger fixed-size helpers.
+ */
+static __always_inline int memcpy_flushcache_large(void *dst,
+                                                  const void *src,
+                                                  size_t cnt)
+{
+       unsigned long d = (unsigned long)dst;
+       char *dptr = dst;
+       const char *sptr = src;
+
+       if (!IS_ALIGNED(d, 8))
+               return 0;
+
+       switch (cnt) {
+       case 32:
+               memcpy_flushcache_32(dptr, sptr);
+               return 1;
+       case 48:
+               memcpy_flushcache_32(dptr, sptr);
+               memcpy_flushcache_16(dptr + 32, sptr + 32);
+               return 1;
+       case 64:
+               memcpy_flushcache_64(dptr, sptr);
+               return 1;
+       case 80:
+               memcpy_flushcache_64(dptr, sptr);
+               memcpy_flushcache_16(dptr + 64, sptr + 64);
+               return 1;
+       case 96:
+               memcpy_flushcache_64(dptr, sptr);
+               memcpy_flushcache_32(dptr + 64, sptr + 64);
+               return 1;
+       }
+
+       return 0;
+}
+
+static __always_inline void memcpy_flushcache(void *dst, const void *src,
+                                             size_t cnt)
+{
+       if (!cnt)
+               return;
+
+       if (__builtin_constant_p(cnt)) {
+               switch (cnt) {
+               case 4:
+                       memcpy_flushcache_4(dst, src);
+                       return;
+               case 8:
+                       memcpy_flushcache_8(dst, src);
+                       return;
+               case 16:
+                       memcpy_flushcache_16(dst, src);
+                       return;
+               }
+
+               if (memcpy_flushcache_large(dst, src, cnt))
+                       return;
+       }
+
+       __memcpy_flushcache(dst, src, cnt);
+}
+
 #define __HAVE_ARCH_MEMCPY_STREAMING 1
 static __always_inline void memcpy_streaming(void *dst, const void *src,
                                             size_t cnt)
-- 
2.20.1

Reply via email to