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
|
/*
AVR LIBC
strlcpy()
Copyright (c) 2003, Eric B. Weddington, R. Patommel
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
* Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in
the documentation and/or other materials provided with the
distribution.
* Neither the name of the copyright holders nor the names of
contributors may be used to endorse or promote products derived
from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
POSSIBILITY OF SUCH DAMAGE.
*/
/** \ingroup avr_string
\fn size_t strlcpy (char *dst, const char *src, size_t siz)
\brief Copy a string.
Copy src to string dst of size siz. At most siz-1 characters will be
copied. Always NULL terminates (unless siz == 0).
\returns The strlcpy() function returns strlen(src). If retval >= siz,
truncation occurred. */
#if !defined(DOXYGEN)
#include "gasava.inc"
#include "macros.inc"
#define dst_hi r25
#define dst_lo r24
#define src_hi r23
#define src_lo r22
#define siz_hi r21
#define siz_lo r20
#define rWord r24
#define rWord_hi r25
#define rWord_lo r24
#define Z_hi r31
#define Z_lo r30
.text
.global _U(strlcpy)
.type _U(strlcpy), @function
_U(strlcpy):
LOAD_X (dst_lo, dst_hi) ; X = dst
LOAD_Z (src_lo, src_hi) ; Z = src
cp siz_lo, __zero_reg__
cpc siz_hi, __zero_reg__ ; size == 0 ?
breq .strlcpy_truncated
.strlcpy_copy_loop: ; copy src to dst
subi siz_lo, lo8(-(-1))
sbci siz_hi, hi8(-(-1)) ; decrement siz
breq 1f ; --> siz chars copied
ld __tmp_reg__, Z+ ; get next src char
st X+, __tmp_reg__ ; copy char
tst __tmp_reg__ ; end of src string ?
breq .strlcpy_len ; --> all src chars copied
rjmp .strlcpy_copy_loop ; next char
1: st X, __zero_reg__ ; truncate dst string
.strlcpy_truncated: ; find Z = end of src string
ld __tmp_reg__, Z+ ; get next char from src
tst __tmp_reg__ ; end of src string ?
brne .strlcpy_truncated ; next char
.strlcpy_len: ; calculate strlen(src)
sub Z_lo, src_lo
sbc Z_hi, src_hi ; Z points past \0
sbiw Z_lo, 1
#if __AVR_ENHANCED__
movw rWord, Z_lo
#else
mov rWord_lo, Z_lo
mov rWord_hi, Z_hi
#endif
ret
.strlcpy_end:
.size _U(strlcpy), .strlcpy_end - _U(strlcpy)
#endif /* not DOXYGEN */
|