wcsmbs: optimize wcsncpy
authorAdhemerval Zanella <adhemerval.zanella@linaro.org>
Tue, 5 Feb 2019 20:48:31 +0000 (18:48 -0200)
committerAdhemerval Zanella <adhemerval.zanella@linaro.org>
Wed, 27 Feb 2019 13:00:37 +0000 (10:00 -0300)
This patch rewrites wcsncpy using wcsnlen, wmemset, and wmemcpy.  This is
similar to the optimization done on strncpy by f6482cf29d and 6423d4754c.

Checked on x86_64-linux-gnu.

* wcsmbs/wcsncpy.c (__wcsncpy): Rewrite using wcsnlen, wmemset, and
wmemcpy.

ChangeLog
wcsmbs/wcsncpy.c

index d51188c..97e70aa 100644 (file)
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,5 +1,8 @@
 2019-02-27  Adhemerval Zanella  <adhemerval.zanella@linaro.org>
 
+       * wcsmbs/wcsncpy.c (__wcsncpy): Rewrite using wcsnlen, wmemset, and
+       wmemcpy.
+
        * wcsmbs/wcsncat.c (wcsncat): Rewrite using wcslen, wcsnlen, and
        wmemcpy.
 
index 2fd523c..d5d7f4a 100644 (file)
 wchar_t *
 __wcsncpy (wchar_t *dest, const wchar_t *src, size_t n)
 {
-  wint_t c;
-  wchar_t *const s = dest;
-
-  --dest;
-
-  if (n >= 4)
-    {
-      size_t n4 = n >> 2;
-
-      for (;;)
-       {
-         c = *src++;
-         *++dest = c;
-         if (c == L'\0')
-           break;
-         c = *src++;
-         *++dest = c;
-         if (c == L'\0')
-           break;
-         c = *src++;
-         *++dest = c;
-         if (c == L'\0')
-           break;
-         c = *src++;
-         *++dest = c;
-         if (c == L'\0')
-           break;
-         if (--n4 == 0)
-           goto last_chars;
-       }
-      n = n - (dest - s) - 1;
-      if (n == 0)
-       return s;
-      goto zero_fill;
-    }
-
- last_chars:
-  n &= 3;
-  if (n == 0)
-    return s;
-
-  do
-    {
-      c = *src++;
-      *++dest = c;
-      if (--n == 0)
-       return s;
-    }
-  while (c != L'\0');
-
- zero_fill:
-  do
-    *++dest = L'\0';
-  while (--n > 0);
-
-  return s;
+  size_t size = __wcsnlen (src, n);
+  if (size != n)
+    __wmemset (dest + size, L'\0', n - size);
+  return __wmemcpy (dest, src, size);
 }
 #ifndef WCSNCPY
 weak_alias (__wcsncpy, wcsncpy)