wcslcpy.c 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. /* $OpenBSD: wcslcpy.c,v 1.7 2015/09/12 16:23:14 guenther Exp $ */
  2. /*
  3. * Copyright (c) 1998, 2015 Todd C. Miller <Todd.Miller@courtesan.com>
  4. *
  5. * Permission to use, copy, modify, and distribute this software for any
  6. * purpose with or without fee is hereby granted, provided that the above
  7. * copyright notice and this permission notice appear in all copies.
  8. *
  9. * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  10. * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11. * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  12. * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13. * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14. * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  15. * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16. */
  17. #include <sys/types.h>
  18. #include <wchar.h>
  19. /*
  20. * Copy string src to buffer dst of size dsize. At most dsize-1
  21. * chars will be copied. Always NUL terminates (unless dsize == 0).
  22. * Returns wcslen(src); if retval >= dsize, truncation occurred.
  23. */
  24. size_t
  25. wcslcpy(wchar_t *dst, const wchar_t *src, size_t dsize)
  26. {
  27. const wchar_t *osrc = src;
  28. size_t nleft = dsize;
  29. /* Copy as many bytes as will fit. */
  30. if (nleft != 0) {
  31. while (--nleft != 0) {
  32. if ((*dst++ = *src++) == L'\0')
  33. break;
  34. }
  35. }
  36. /* Not enough room in dst, add NUL and traverse rest of src. */
  37. if (nleft == 0) {
  38. if (dsize != 0)
  39. *dst = L'\0'; /* NUL-terminate dst */
  40. while (*src++)
  41. ;
  42. }
  43. return(src - osrc - 1); /* count does not include NUL */
  44. }