strlcpy.c (1904B)
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 17 #include "pkg.h" 18 19 /* 20 * Copy src to string dst of size siz. At most siz-1 characters 21 * will be copied. Always NUL terminates (unless siz == 0). 22 * Returns strlen(src); if retval >= siz, truncation occurred. 23 */ 24 size_t 25 strlcpy(char *dst, const char *src, size_t siz) 26 { 27 char *d = dst; 28 const char *s = src; 29 size_t n = siz; 30 /* Copy as many bytes as will fit */ 31 if (n != 0) { 32 while (--n != 0) { 33 if ((*d++ = *s++) == '\0') 34 break; 35 } 36 } 37 /* Not enough room in dst, add NUL and traverse rest of src */ 38 if (n == 0) { 39 if (siz != 0) 40 *d = '\0'; /* NUL-terminate dst */ 41 while (*s++) 42 ; 43 } 44 return(s - src - 1); /* count does not include NUL */ 45 } 46 47 size_t 48 estrlcpy(char *dst, const char *src, size_t siz) 49 { 50 size_t r; 51 52 r = strlcpy(dst, src, siz); 53 if (r >= siz) { 54 fprintf(stderr, "destination buffer too small\n"); 55 exit(EXIT_FAILURE); 56 } 57 return r; 58 }