tool_utils: strncat => strlcat

parent 9c8b2b13
...@@ -52,6 +52,10 @@ ...@@ -52,6 +52,10 @@
#include "include/strlcpy.h" #include "include/strlcpy.h"
#endif #endif
#ifndef HAVE_STRLCAT
#include "include/strlcat.h"
#endif
int lxc_fill_elevated_privileges(char *flaglist, int *flags) int lxc_fill_elevated_privileges(char *flaglist, int *flags)
{ {
char *token, *saveptr = NULL; char *token, *saveptr = NULL;
...@@ -503,22 +507,24 @@ char *lxc_string_join(const char *sep, const char **parts, bool use_as_prefix) ...@@ -503,22 +507,24 @@ char *lxc_string_join(const char *sep, const char **parts, bool use_as_prefix)
char **p; char **p;
size_t sep_len = strlen(sep); size_t sep_len = strlen(sep);
size_t result_len = use_as_prefix * sep_len; size_t result_len = use_as_prefix * sep_len;
size_t buf_len;
/* calculate new string length */ /* calculate new string length */
for (p = (char **)parts; *p; p++) for (p = (char **)parts; *p; p++)
result_len += (p > (char **)parts) * sep_len + strlen(*p); result_len += (p > (char **)parts) * sep_len + strlen(*p);
result = calloc(result_len + 1, 1); buf_len = result_len + 1;
result = calloc(buf_len, 1);
if (!result) if (!result)
return NULL; return NULL;
if (use_as_prefix) if (use_as_prefix)
(void)strlcpy(result, sep, result_len + 1); (void)strlcpy(result, sep, buf_len);
for (p = (char **)parts; *p; p++) { for (p = (char **)parts; *p; p++) {
if (p > (char **)parts) if (p > (char **)parts)
strncat(result, sep, sep_len); (void)strlcat(result, sep, buf_len);
strncat(result, *p, strlen(*p)); (void)strlcat(result, *p, buf_len);
} }
return result; return result;
...@@ -1071,6 +1077,7 @@ char *must_make_path(const char *first, ...) ...@@ -1071,6 +1077,7 @@ char *must_make_path(const char *first, ...)
va_list args; va_list args;
char *cur, *dest; char *cur, *dest;
size_t full_len = strlen(first); size_t full_len = strlen(first);
size_t buf_len;
dest = must_copy_string(first); dest = must_copy_string(first);
...@@ -1080,11 +1087,12 @@ char *must_make_path(const char *first, ...) ...@@ -1080,11 +1087,12 @@ char *must_make_path(const char *first, ...)
if (cur[0] != '/') if (cur[0] != '/')
full_len++; full_len++;
dest = must_realloc(dest, full_len + 1); buf_len = full_len + 1;
dest = must_realloc(dest, buf_len);
if (cur[0] != '/') if (cur[0] != '/')
strncat(dest, "/", 1); (void)strlcat(dest, "/", buf_len);
strncat(dest, cur, strlen(cur)); (void)strlcat(dest, cur, buf_len);
} }
va_end(args); va_end(args);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment