1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
/* Copyright (C) 1992-2023 Free Software Foundation, Inc.
This file is part of the GNU C Library.
The GNU C Library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
The GNU C Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with the GNU C Library; if not, see
<https://www.gnu.org/licenses/>. */
#define NO_MEMPCPY_STPCPY_REDIRECT
#include <string.h>
#include <stdint.h>
#include <memcopy.h>
#include <string-fzb.h>
#include <string-misc.h>
#undef __stpcpy
#undef stpcpy
#ifndef STPCPY
# define STPCPY __stpcpy
#endif
static __always_inline char *
write_byte_from_word (op_t *dest, op_t word)
{
char *d = (char *) dest;
for (size_t i = 0; i < OPSIZ; i++, ++d)
{
char c = extractbyte (word, i);
*d = c;
if (c == '\0')
break;
}
return d;
}
static __always_inline char *
stpcpy_aligned_loop (op_t *restrict dst, const op_t *restrict src)
{
op_t word;
while (1)
{
word = *src++;
if (has_zero (word))
break;
*dst++ = word;
}
return write_byte_from_word (dst, word);
}
static __always_inline char *
stpcpy_unaligned_loop (op_t *restrict dst, const op_t *restrict src,
uintptr_t ofs)
{
op_t w2a = *src++;
uintptr_t sh_1 = ofs * CHAR_BIT;
uintptr_t sh_2 = OPSIZ * CHAR_BIT - sh_1;
op_t w2 = MERGE (w2a, sh_1, (op_t)-1, sh_2);
if (!has_zero (w2))
{
op_t w2b;
/* Unaligned loop. The invariant is that W2B, which is "ahead" of W1,
does not contain end-of-string. Therefore it is safe (and necessary)
to read another word from each while we do not have a difference. */
while (1)
{
w2b = *src++;
w2 = MERGE (w2a, sh_1, w2b, sh_2);
/* Check if there is zero on w2a. */
if (has_zero (w2))
goto out;
*dst++ = w2;
if (has_zero (w2b))
break;
w2a = w2b;
}
/* Align the final partial of P2. */
w2 = MERGE (w2b, sh_1, 0, sh_2);
}
out:
return write_byte_from_word (dst, w2);
}
/* Copy SRC to DEST, returning the address of the terminating '\0' in DEST. */
char *
STPCPY (char *dest, const char *src)
{
/* Copy just a few bytes to make DEST aligned. */
size_t len = (-(uintptr_t) dest) % OPSIZ;
for (; len != 0; len--, ++dest)
{
char c = *src++;
*dest = c;
if (c == '\0')
return dest;
}
/* DEST is now aligned to op_t, SRC may or may not be. */
uintptr_t ofs = (uintptr_t) src % OPSIZ;
return ofs == 0 ? stpcpy_aligned_loop ((op_t*) dest, (const op_t *) src)
: stpcpy_unaligned_loop ((op_t*) dest,
(const op_t *) (src - ofs) , ofs);
}
weak_alias (__stpcpy, stpcpy)
libc_hidden_def (__stpcpy)
libc_hidden_builtin_def (stpcpy)
|
