wcscpy_chk.c 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. /* Copyright (C) 1995-2019 Free Software Foundation, Inc.
  2. This file is part of the GNU C Library.
  3. Contributed by Ulrich Drepper <drepper@gnu.ai.mit.edu>, 1995.
  4. The GNU C Library is free software; you can redistribute it and/or
  5. modify it under the terms of the GNU Lesser General Public
  6. License as published by the Free Software Foundation; either
  7. version 2.1 of the License, or (at your option) any later version.
  8. The GNU C Library is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  11. Lesser General Public License for more details.
  12. You should have received a copy of the GNU Lesser General Public
  13. License along with the GNU C Library; if not, see
  14. <http://www.gnu.org/licenses/>. */
  15. #include <stddef.h>
  16. #include <wchar.h>
  17. /* Copy SRC to DEST. */
  18. wchar_t *
  19. __wcscpy_chk (wchar_t *dest, const wchar_t *src, size_t n)
  20. {
  21. wint_t c;
  22. wchar_t *wcp;
  23. if (__alignof__ (wchar_t) >= sizeof (wchar_t))
  24. {
  25. const ptrdiff_t off = dest - src - 1;
  26. wcp = (wchar_t *) src;
  27. do
  28. {
  29. if (__glibc_unlikely (n-- == 0))
  30. __chk_fail ();
  31. c = *wcp++;
  32. wcp[off] = c;
  33. }
  34. while (c != L'\0');
  35. }
  36. else
  37. {
  38. wcp = dest;
  39. do
  40. {
  41. if (__glibc_unlikely (n-- == 0))
  42. __chk_fail ();
  43. c = *src++;
  44. *wcp++ = c;
  45. }
  46. while (c != L'\0');
  47. }
  48. return dest;
  49. }