Commit 41416f23 authored by Rasmus Villemoes's avatar Rasmus Villemoes Committed by Linus Torvalds

lib/string_helpers.c: change semantics of string_escape_mem

The current semantics of string_escape_mem are inadequate for one of its
current users, vsnprintf().  If that is to honour its contract, it must
know how much space would be needed for the entire escaped buffer, and
string_escape_mem provides no way of obtaining that (short of allocating a
large enough buffer (~4 times input string) to let it play with, and
that's definitely a big no-no inside vsnprintf).

So change the semantics for string_escape_mem to be more snprintf-like:
Return the size of the output that would be generated if the destination
buffer was big enough, but of course still only write to the part of dst
it is allowed to, and (contrary to snprintf) don't do '\0'-termination.
It is then up to the caller to detect whether output was truncated and to
append a '\0' if desired.  Also, we must output partial escape sequences,
otherwise a call such as snprintf(buf, 3, "%1pE", "\123") would cause
printf to write a \0 to buf[2] but leaving buf[0] and buf[1] with whatever
they previously contained.

This also fixes a bug in the escaped_string() helper function, which used
to unconditionally pass a length of "end-buf" to string_escape_mem();
since the latter doesn't check osz for being insanely large, it would
happily write to dst.  For example, kasprintf(GFP_KERNEL, "something and
then %pE", ...); is an easy way to trigger an oops.

In test-string_helpers.c, the -ENOMEM test is replaced with testing for
getting the expected return value even if the buffer is too small.  We
also ensure that nothing is written (by relying on a NULL pointer deref)
if the output size is 0 by passing NULL - this has to work for
kasprintf("%pE") to work.

In net/sunrpc/cache.c, I think qword_add still has the same semantics.
Someone should definitely double-check this.

In fs/proc/array.c, I made the minimum possible change, but longer-term it
should stop poking around in seq_file internals.

[andriy.shevchenko@linux.intel.com: simplify qword_add]
[andriy.shevchenko@linux.intel.com: add missed curly braces]
Signed-off-by: default avatarRasmus Villemoes <linux@rasmusvillemoes.dk>
Acked-by: default avatarAndy Shevchenko <andriy.shevchenko@linux.intel.com>
Signed-off-by: default avatarAndy Shevchenko <andriy.shevchenko@linux.intel.com>
Signed-off-by: default avatarAndrew Morton <akpm@linux-foundation.org>
Signed-off-by: default avatarLinus Torvalds <torvalds@linux-foundation.org>
parent 3aeddc7d
...@@ -99,8 +99,8 @@ static inline void task_name(struct seq_file *m, struct task_struct *p) ...@@ -99,8 +99,8 @@ static inline void task_name(struct seq_file *m, struct task_struct *p)
buf = m->buf + m->count; buf = m->buf + m->count;
/* Ignore error for now */ /* Ignore error for now */
string_escape_str(tcomm, &buf, m->size - m->count, buf += string_escape_str(tcomm, buf, m->size - m->count,
ESCAPE_SPACE | ESCAPE_SPECIAL, "\n\\"); ESCAPE_SPACE | ESCAPE_SPECIAL, "\n\\");
m->count = buf - m->buf; m->count = buf - m->buf;
seq_putc(m, '\n'); seq_putc(m, '\n');
......
...@@ -47,22 +47,22 @@ static inline int string_unescape_any_inplace(char *buf) ...@@ -47,22 +47,22 @@ static inline int string_unescape_any_inplace(char *buf)
#define ESCAPE_ANY_NP (ESCAPE_ANY | ESCAPE_NP) #define ESCAPE_ANY_NP (ESCAPE_ANY | ESCAPE_NP)
#define ESCAPE_HEX 0x20 #define ESCAPE_HEX 0x20
int string_escape_mem(const char *src, size_t isz, char **dst, size_t osz, int string_escape_mem(const char *src, size_t isz, char *dst, size_t osz,
unsigned int flags, const char *esc); unsigned int flags, const char *esc);
static inline int string_escape_mem_any_np(const char *src, size_t isz, static inline int string_escape_mem_any_np(const char *src, size_t isz,
char **dst, size_t osz, const char *esc) char *dst, size_t osz, const char *esc)
{ {
return string_escape_mem(src, isz, dst, osz, ESCAPE_ANY_NP, esc); return string_escape_mem(src, isz, dst, osz, ESCAPE_ANY_NP, esc);
} }
static inline int string_escape_str(const char *src, char **dst, size_t sz, static inline int string_escape_str(const char *src, char *dst, size_t sz,
unsigned int flags, const char *esc) unsigned int flags, const char *esc)
{ {
return string_escape_mem(src, strlen(src), dst, sz, flags, esc); return string_escape_mem(src, strlen(src), dst, sz, flags, esc);
} }
static inline int string_escape_str_any_np(const char *src, char **dst, static inline int string_escape_str_any_np(const char *src, char *dst,
size_t sz, const char *esc) size_t sz, const char *esc)
{ {
return string_escape_str(src, dst, sz, ESCAPE_ANY_NP, esc); return string_escape_str(src, dst, sz, ESCAPE_ANY_NP, esc);
......
...@@ -274,11 +274,6 @@ static bool escape_space(unsigned char c, char **dst, char *end) ...@@ -274,11 +274,6 @@ static bool escape_space(unsigned char c, char **dst, char *end)
return false; return false;
} }
if (out + 2 > end) {
*dst = out + 2;
return true;
}
if (out < end) if (out < end)
*out = '\\'; *out = '\\';
++out; ++out;
...@@ -309,11 +304,6 @@ static bool escape_special(unsigned char c, char **dst, char *end) ...@@ -309,11 +304,6 @@ static bool escape_special(unsigned char c, char **dst, char *end)
return false; return false;
} }
if (out + 2 > end) {
*dst = out + 2;
return true;
}
if (out < end) if (out < end)
*out = '\\'; *out = '\\';
++out; ++out;
...@@ -332,11 +322,6 @@ static bool escape_null(unsigned char c, char **dst, char *end) ...@@ -332,11 +322,6 @@ static bool escape_null(unsigned char c, char **dst, char *end)
if (c) if (c)
return false; return false;
if (out + 2 > end) {
*dst = out + 2;
return true;
}
if (out < end) if (out < end)
*out = '\\'; *out = '\\';
++out; ++out;
...@@ -352,11 +337,6 @@ static bool escape_octal(unsigned char c, char **dst, char *end) ...@@ -352,11 +337,6 @@ static bool escape_octal(unsigned char c, char **dst, char *end)
{ {
char *out = *dst; char *out = *dst;
if (out + 4 > end) {
*dst = out + 4;
return true;
}
if (out < end) if (out < end)
*out = '\\'; *out = '\\';
++out; ++out;
...@@ -378,11 +358,6 @@ static bool escape_hex(unsigned char c, char **dst, char *end) ...@@ -378,11 +358,6 @@ static bool escape_hex(unsigned char c, char **dst, char *end)
{ {
char *out = *dst; char *out = *dst;
if (out + 4 > end) {
*dst = out + 4;
return true;
}
if (out < end) if (out < end)
*out = '\\'; *out = '\\';
++out; ++out;
...@@ -449,20 +424,17 @@ static bool escape_hex(unsigned char c, char **dst, char *end) ...@@ -449,20 +424,17 @@ static bool escape_hex(unsigned char c, char **dst, char *end)
* it if needs. * it if needs.
* *
* Return: * Return:
* The amount of the characters processed to the destination buffer, or * The total size of the escaped output that would be generated for
* %-ENOMEM if the size of buffer is not enough to put an escaped character is * the given input and flags. To check whether the output was
* returned. * truncated, compare the return value to osz. There is room left in
* * dst for a '\0' terminator if and only if ret < osz.
* Even in the case of error @dst pointer will be updated to point to the byte
* after the last processed character.
*/ */
int string_escape_mem(const char *src, size_t isz, char **dst, size_t osz, int string_escape_mem(const char *src, size_t isz, char *dst, size_t osz,
unsigned int flags, const char *esc) unsigned int flags, const char *esc)
{ {
char *p = *dst; char *p = dst;
char *end = p + osz; char *end = p + osz;
bool is_dict = esc && *esc; bool is_dict = esc && *esc;
int ret;
while (isz--) { while (isz--) {
unsigned char c = *src++; unsigned char c = *src++;
...@@ -502,13 +474,6 @@ int string_escape_mem(const char *src, size_t isz, char **dst, size_t osz, ...@@ -502,13 +474,6 @@ int string_escape_mem(const char *src, size_t isz, char **dst, size_t osz,
escape_passthrough(c, &p, end); escape_passthrough(c, &p, end);
} }
if (p > end) { return p - dst;
*dst = end;
return -ENOMEM;
}
ret = p - *dst;
*dst = p;
return ret;
} }
EXPORT_SYMBOL(string_escape_mem); EXPORT_SYMBOL(string_escape_mem);
...@@ -260,16 +260,28 @@ static __init const char *test_string_find_match(const struct test_string_2 *s2, ...@@ -260,16 +260,28 @@ static __init const char *test_string_find_match(const struct test_string_2 *s2,
return NULL; return NULL;
} }
static __init void
test_string_escape_overflow(const char *in, int p, unsigned int flags, const char *esc,
int q_test, const char *name)
{
int q_real;
q_real = string_escape_mem(in, p, NULL, 0, flags, esc);
if (q_real != q_test)
pr_warn("Test '%s' failed: flags = %u, osz = 0, expected %d, got %d\n",
name, flags, q_test, q_real);
}
static __init void test_string_escape(const char *name, static __init void test_string_escape(const char *name,
const struct test_string_2 *s2, const struct test_string_2 *s2,
unsigned int flags, const char *esc) unsigned int flags, const char *esc)
{ {
int q_real = 512; size_t out_size = 512;
char *out_test = kmalloc(q_real, GFP_KERNEL); char *out_test = kmalloc(out_size, GFP_KERNEL);
char *out_real = kmalloc(q_real, GFP_KERNEL); char *out_real = kmalloc(out_size, GFP_KERNEL);
char *in = kmalloc(256, GFP_KERNEL); char *in = kmalloc(256, GFP_KERNEL);
char *buf = out_real;
int p = 0, q_test = 0; int p = 0, q_test = 0;
int q_real;
if (!out_test || !out_real || !in) if (!out_test || !out_real || !in)
goto out; goto out;
...@@ -301,29 +313,19 @@ static __init void test_string_escape(const char *name, ...@@ -301,29 +313,19 @@ static __init void test_string_escape(const char *name,
q_test += len; q_test += len;
} }
q_real = string_escape_mem(in, p, &buf, q_real, flags, esc); q_real = string_escape_mem(in, p, out_real, out_size, flags, esc);
test_string_check_buf(name, flags, in, p, out_real, q_real, out_test, test_string_check_buf(name, flags, in, p, out_real, q_real, out_test,
q_test); q_test);
test_string_escape_overflow(in, p, flags, esc, q_test, name);
out: out:
kfree(in); kfree(in);
kfree(out_real); kfree(out_real);
kfree(out_test); kfree(out_test);
} }
static __init void test_string_escape_nomem(void)
{
char *in = "\eb \\C\007\"\x90\r]";
char out[64], *buf = out;
int rc = -ENOMEM, ret;
ret = string_escape_str_any_np(in, &buf, strlen(in), NULL);
if (ret == rc)
return;
pr_err("Test 'escape nomem' failed: got %d instead of %d\n", ret, rc);
}
static int __init test_string_helpers_init(void) static int __init test_string_helpers_init(void)
{ {
unsigned int i; unsigned int i;
...@@ -342,8 +344,6 @@ static int __init test_string_helpers_init(void) ...@@ -342,8 +344,6 @@ static int __init test_string_helpers_init(void)
for (i = 0; i < (ESCAPE_ANY_NP | ESCAPE_HEX) + 1; i++) for (i = 0; i < (ESCAPE_ANY_NP | ESCAPE_HEX) + 1; i++)
test_string_escape("escape 1", escape1, i, TEST_STRING_2_DICT_1); test_string_escape("escape 1", escape1, i, TEST_STRING_2_DICT_1);
test_string_escape_nomem();
return -EINVAL; return -EINVAL;
} }
module_init(test_string_helpers_init); module_init(test_string_helpers_init);
......
...@@ -1235,8 +1235,12 @@ char *escaped_string(char *buf, char *end, u8 *addr, struct printf_spec spec, ...@@ -1235,8 +1235,12 @@ char *escaped_string(char *buf, char *end, u8 *addr, struct printf_spec spec,
len = spec.field_width < 0 ? 1 : spec.field_width; len = spec.field_width < 0 ? 1 : spec.field_width;
/* Ignore the error. We print as many characters as we can */ /*
string_escape_mem(addr, len, &buf, end - buf, flags, NULL); * string_escape_mem() writes as many characters as it can to
* the given buffer, and returns the total size of the output
* had the buffer been big enough.
*/
buf += string_escape_mem(addr, len, buf, buf < end ? end - buf : 0, flags, NULL);
return buf; return buf;
} }
......
...@@ -1072,10 +1072,12 @@ void qword_add(char **bpp, int *lp, char *str) ...@@ -1072,10 +1072,12 @@ void qword_add(char **bpp, int *lp, char *str)
if (len < 0) return; if (len < 0) return;
ret = string_escape_str(str, &bp, len, ESCAPE_OCTAL, "\\ \n\t"); ret = string_escape_str(str, bp, len, ESCAPE_OCTAL, "\\ \n\t");
if (ret < 0 || ret == len) if (ret >= len) {
bp += len;
len = -1; len = -1;
else { } else {
bp += ret;
len -= ret; len -= ret;
*bp++ = ' '; *bp++ = ' ';
len--; len--;
......
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