diff options
Diffstat (limited to 'newlib/libc/string')
-rw-r--r-- | newlib/libc/string/strndup_r.c | 14 | ||||
-rw-r--r-- | newlib/libc/string/strnlen.c | 2 |
2 files changed, 11 insertions, 5 deletions
diff --git a/newlib/libc/string/strndup_r.c b/newlib/libc/string/strndup_r.c index 86d9eec44..2acf63dec 100644 --- a/newlib/libc/string/strndup_r.c +++ b/newlib/libc/string/strndup_r.c @@ -2,16 +2,22 @@ #include <stdlib.h> #include <string.h> -#define MIN(a,b) ((a) < (b) ? (a) : (b)) - char * _DEFUN (_strndup_r, (reent_ptr, str, n), struct _reent *reent_ptr _AND _CONST char *str _AND size_t n) { - size_t len = MIN(strlen (str), n); - char *copy = _malloc_r (reent_ptr, len + 1); + _CONST char *ptr = str; + size_t len; + char *copy; + + while (n-- > 0 && *ptr) + ptr++; + + len = ptr - str; + + copy = _malloc_r (reent_ptr, len + 1); if (copy) { memcpy (copy, str, len); diff --git a/newlib/libc/string/strnlen.c b/newlib/libc/string/strnlen.c index 92826eeb2..ed60e9371 100644 --- a/newlib/libc/string/strnlen.c +++ b/newlib/libc/string/strnlen.c @@ -42,7 +42,7 @@ _DEFUN (strnlen, (str, n), { _CONST char *start = str; - while (*str && n-- > 0) + while (n-- > 0 && *str) str++; return str - start; |