1 | /********************************************************************\
|
---|
2 |
|
---|
3 | Name: strlcpy.c
|
---|
4 | Created by: Stefan Ritt
|
---|
5 |
|
---|
6 | Contents: Contains strlcpy and strlcat which are versions of
|
---|
7 | strcpy and strcat, but which avoid buffer overflows
|
---|
8 |
|
---|
9 | $Id: strlcpy.c 16 2005-10-07 13:05:38Z ritt $
|
---|
10 |
|
---|
11 | \********************************************************************/
|
---|
12 |
|
---|
13 | #include <stdio.h>
|
---|
14 | #include <string.h>
|
---|
15 | #include "strlcpy.h"
|
---|
16 |
|
---|
17 | /*
|
---|
18 | * Copy src to string dst of size siz. At most siz-1 characters
|
---|
19 | * will be copied. Always NUL terminates (unless size == 0).
|
---|
20 | * Returns strlen(src); if retval >= siz, truncation occurred.
|
---|
21 | */
|
---|
22 | size_t strlcpy(char *dst, const char *src, size_t size)
|
---|
23 | {
|
---|
24 | char *d = dst;
|
---|
25 | const char *s = src;
|
---|
26 | size_t n = size;
|
---|
27 |
|
---|
28 | /* Copy as many bytes as will fit */
|
---|
29 | if (n != 0 && --n != 0) {
|
---|
30 | do {
|
---|
31 | if ((*d++ = *s++) == 0)
|
---|
32 | break;
|
---|
33 | } while (--n != 0);
|
---|
34 | }
|
---|
35 |
|
---|
36 | /* Not enough room in dst, add NUL and traverse rest of src */
|
---|
37 | if (n == 0) {
|
---|
38 | if (size != 0)
|
---|
39 | *d = '\0'; /* NUL-terminate dst */
|
---|
40 | while (*s++);
|
---|
41 | }
|
---|
42 |
|
---|
43 | return (s - src - 1); /* count does not include NUL */
|
---|
44 | }
|
---|
45 |
|
---|
46 | /*-------------------------------------------------------------------*/
|
---|
47 |
|
---|
48 | /*
|
---|
49 | * Appends src to string dst of size siz (unlike strncat, siz is the
|
---|
50 | * full size of dst, not space left). At most siz-1 characters
|
---|
51 | * will be copied. Always NUL terminates (unless size <= strlen(dst)).
|
---|
52 | * Returns strlen(src) + MIN(size, strlen(initial dst)).
|
---|
53 | * If retval >= size, truncation occurred.
|
---|
54 | */
|
---|
55 | size_t strlcat(char *dst, const char *src, size_t size)
|
---|
56 | {
|
---|
57 | char *d = dst;
|
---|
58 | const char *s = src;
|
---|
59 | size_t n = size;
|
---|
60 | size_t dlen;
|
---|
61 |
|
---|
62 | /* Find the end of dst and adjust bytes left but don't go past end */
|
---|
63 | while (n-- != 0 && *d != '\0')
|
---|
64 | d++;
|
---|
65 | dlen = d - dst;
|
---|
66 | n = size - dlen;
|
---|
67 |
|
---|
68 | if (n == 0)
|
---|
69 | return (dlen + strlen(s));
|
---|
70 | while (*s != '\0') {
|
---|
71 | if (n != 1) {
|
---|
72 | *d++ = *s;
|
---|
73 | n--;
|
---|
74 | }
|
---|
75 | s++;
|
---|
76 | }
|
---|
77 | *d = '\0';
|
---|
78 |
|
---|
79 | return (dlen + (s - src)); /* count does not include NUL */
|
---|
80 | }
|
---|
81 |
|
---|
82 | /*-------------------------------------------------------------------*/
|
---|