dcb22f6eb105ec2a2d1b531c4ec009f5a091d507
[musl] / src / string / strlcpy.c
1 #define _BSD_SOURCE
2 #include <string.h>
3 #include <stdint.h>
4 #include <limits.h>
5
6 #define ALIGN (sizeof(size_t)-1)
7 #define ONES ((size_t)-1/UCHAR_MAX)
8 #define HIGHS (ONES * (UCHAR_MAX/2+1))
9 #define HASZERO(x) ((x)-ONES & ~(x) & HIGHS)
10
11 size_t strlcpy(char *d, const char *s, size_t n)
12 {
13         char *d0 = d;
14         size_t *wd;
15         const size_t *ws;
16
17         if (!n--) goto finish;
18         if (((uintptr_t)s & ALIGN) == ((uintptr_t)d & ALIGN)) {
19                 for (; ((uintptr_t)s & ALIGN) && n && (*d=*s); n--, s++, d++);
20                 if (n && *s) {
21                         wd=(void *)d; ws=(const void *)s;
22                         for (; n>=sizeof(size_t) && !HASZERO(*ws);
23                                n-=sizeof(size_t), ws++, wd++) *wd = *ws;
24                         d=(void *)wd; s=(const void *)ws;
25                 }
26         }
27         for (; n && (*d=*s); n--, s++, d++);
28         *d = 0;
29 finish:
30         return d-d0 + strlen(s);
31 }