@@ -91,22 +91,25 @@ static int
cifs_strlcpy_to_host(char **dst, const char *src, const int maxlen,
const bool is_unicode, const struct nls_table *nls_codepage)
{
- int plen;
+ int src_len, dst_len;
if (is_unicode) {
- plen = UniStrnlen((wchar_t *)src, maxlen);
- *dst = kmalloc((4 * plen) + 2, GFP_KERNEL);
+ src_len = UniStrnlen((wchar_t *)src, maxlen);
+ *dst = kmalloc((4 * src_len) + 2, GFP_KERNEL);
if (!*dst)
goto cifs_strlcpy_to_host_ErrExit;
- cifs_strfromUCS_le(*dst, (__le16 *)src, plen, nls_codepage);
- (*dst)[plen] = 0;
- (*dst)[plen+1] = 0; /* needed for Unicode */
+ dst_len = cifs_strfromUCS_le(*dst, (__le16 *)src, src_len,
+ nls_codepage);
+ /*
+ * cifs_strfromUCS_le() ensures single byte NULL termination
+ */
+ (*dst)[dst_len + 1] = 0; /* needed for Unicode, to be safe */
} else {
- plen = strnlen(src, maxlen);
- *dst = kmalloc(plen + 2, GFP_KERNEL);
+ src_len = strnlen(src, maxlen);
+ *dst = kmalloc(src_len + 1, GFP_KERNEL);
if (!*dst)
goto cifs_strlcpy_to_host_ErrExit;
- strlcpy(*dst, src, plen);
+ strlcpy(*dst, src, src_len);
}
return 0;