/src/strlcpy.h

https://github.com/monyxcoin/monyx-coin · C Header · 83 lines · 43 code · 10 blank · 30 comment · 21 complexity · 5bb5bad48e3d6171876a4d7262fd06ae MD5 · raw file

  1. /*
  2. * Copyright (c) 1998 Todd C. Miller <Todd.Miller@courtesan.com>
  3. *
  4. * Permission to use, copy, modify, and distribute this software for any
  5. * purpose with or without fee is hereby granted, provided that the above
  6. * copyright notice and this permission notice appear in all copies.
  7. *
  8. * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9. * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10. * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11. * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12. * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13. * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14. * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15. */
  16. #ifndef BITCOIN_STRLCPY_H
  17. #define BITCOIN_STRLCPY_H
  18. #include <stdlib.h>
  19. #include <string.h>
  20. /*
  21. * Copy src to string dst of size siz. At most siz-1 characters
  22. * will be copied. Always NUL terminates (unless siz == 0).
  23. * Returns strlen(src); if retval >= siz, truncation occurred.
  24. */
  25. inline size_t strlcpy(char *dst, const char *src, size_t siz) {
  26. char *d = dst;
  27. const char *s = src;
  28. size_t n = siz;
  29. /* Copy as many bytes as will fit */
  30. if (n != 0) {
  31. while (--n != 0) {
  32. if ((*d++ = *s++) == '\0')
  33. break;
  34. }
  35. }
  36. /* Not enough room in dst, add NUL and traverse rest of src */
  37. if (n == 0) {
  38. if (siz != 0)
  39. *d = '\0'; /* NUL-terminate dst */
  40. while (*s++);
  41. }
  42. return (s - src - 1); /* count does not include NUL */
  43. }
  44. /*
  45. * Appends src to string dst of size siz (unlike strncat, siz is the
  46. * full size of dst, not space left). At most siz-1 characters
  47. * will be copied. Always NUL terminates (unless siz <= strlen(dst)).
  48. * Returns strlen(src) + MIN(siz, strlen(initial dst)).
  49. * If retval >= siz, truncation occurred.
  50. */
  51. inline size_t strlcat(char *dst, const char *src, size_t siz) {
  52. char *d = dst;
  53. const char *s = src;
  54. size_t n = siz;
  55. size_t dlen;
  56. /* Find the end of dst and adjust bytes left but don't go past end */
  57. while (n-- != 0 && *d != '\0')
  58. d++;
  59. dlen = d - dst;
  60. n = siz - dlen;
  61. if (n == 0)
  62. return (dlen + strlen(s));
  63. while (*s != '\0') {
  64. if (n != 1) {
  65. *d++ = *s;
  66. n--;
  67. }
  68. s++;
  69. }
  70. *d = '\0';
  71. return (dlen + (s - src)); /* count does not include NUL */
  72. }
  73. #endif