From c9183f385129d044f922c64fcc41cb44a8b0d1ae Mon Sep 17 00:00:00 2001 From: "nullable.type" Date: Sun, 16 Dec 2012 12:27:53 +0100 Subject: [PATCH] Made the names of the function arguments of the declaration and the definition of mg_get_var(..) the same. Changed the error codes to have a consistent way of reporting errors. -2 is returned if there is a problem with the size of the destination buffer. -1 is returned if the variable is not found. Updated the documentation of mg_get_var(..) in mongoose.h and removed the redundant comments from mongoose.c. --- mongoose.c | 23 +++++++++++------------ mongoose.h | 12 ++++++------ 2 files changed, 17 insertions(+), 18 deletions(-) diff --git a/mongoose.c b/mongoose.c index ed0e2f92..d341c76b 100644 --- a/mongoose.c +++ b/mongoose.c @@ -1646,13 +1646,7 @@ static int url_decode(const char *src, int src_len, char *dst, return i >= src_len ? j : -1; } -// Scan given buffer and fetch the value of the given variable. -// It can be specified in query string, or in the POST data. -// Return -1 if the variable not found, or length of the URL-decoded value -// stored in dst. The dst buffer is guaranteed to be NUL-terminated if it -// is not NULL or zero-length. If dst is NULL or zero-length, then -// -2 is returned. -int mg_get_var(const char *buf, size_t buf_len, const char *name, +int mg_get_var(const char *data, size_t data_len, const char *name, char *dst, size_t dst_len) { const char *p, *e, *s; size_t name_len; @@ -1660,18 +1654,18 @@ int mg_get_var(const char *buf, size_t buf_len, const char *name, if (dst == NULL || dst_len == 0) { len = -2; - } else if (buf == NULL || name == NULL || buf_len == 0) { + } else if (data == NULL || name == NULL || data_len == 0) { len = -1; dst[0] = '\0'; } else { name_len = strlen(name); - e = buf + buf_len; + e = data + data_len; len = -1; dst[0] = '\0'; - // buf is "var1=val1&var2=val2...". Find variable first - for (p = buf; p + name_len < e; p++) { - if ((p == buf || p[-1] == '&') && p[name_len] == '=' && + // data is "var1=val1&var2=val2...". Find variable first + for (p = data; p + name_len < e; p++) { + if ((p == data || p[-1] == '&') && p[name_len] == '=' && !mg_strncasecmp(name, p, name_len)) { // Point p to variable value @@ -1686,6 +1680,11 @@ int mg_get_var(const char *buf, size_t buf_len, const char *name, // Decode variable into destination buffer len = url_decode(p, (size_t)(s - p), dst, dst_len, 1); + + // Redirect error code from -1 to -2 (destination buffer too small). + if (len == -1) { + len = -2; + } break; } } diff --git a/mongoose.h b/mongoose.h index c51baad4..a9f98c34 100644 --- a/mongoose.h +++ b/mongoose.h @@ -286,19 +286,19 @@ const char *mg_get_header(const struct mg_connection *, const char *name); // or request_info.query_string. // data_len: length of the encoded data. // var_name: variable name to decode from the buffer -// buf: destination buffer for the decoded variable -// buf_len: length of the destination buffer +// dst: destination buffer for the decoded variable +// dst_len: length of the destination buffer // // Return: // On success, length of the decoded variable. // On error: -// -1 (variable not found, or destination buffer is too small). -// -2 (destination buffer is NULL or zero length). +// -1 (variable not found). +// -2 (destination buffer is NULL, zero length or too small to hold the decoded variable). // // Destination buffer is guaranteed to be '\0' - terminated if it is not -// NULL or zero length. In case of failure, dst[0] == '\0'. +// NULL or zero length. int mg_get_var(const char *data, size_t data_len, - const char *var_name, char *buf, size_t buf_len); + const char *var_name, char *dst, size_t dst_len); // Fetch value of certain cookie variable into the destination buffer. //