Made behavior of mg_get_cookie(..) on errors the same as mg_get_var(..)....

Made behavior of mg_get_cookie(..) on errors the same as mg_get_var(..). Separated the two cases a.) problems with destination buffer and b.) cookie header or cookie parameter not existing.

mongoose.c

@@ -1698,15 +1698,17 @@ int mg_get_cookie(const struct mg_connection *conn, const char *cookie_name,
   const char *s, *p, *end;
   int name_len, len = -1;
 
+  if (dst == NULL || dst_size == 0) {
+      len = -2;
+  } else if (cookie_name == NULL || (s = mg_get_header(conn, "Cookie")) == NULL) {
+      len = -1;
       dst[0] = '\0';
-  if ((s = mg_get_header(conn, "Cookie")) == NULL) {
-    return -1;
-  }
-
+  } else {
     name_len = (int) strlen(cookie_name);
     end = s + strlen(s);
+    dst[0] = '\0';
 
-  for (; (s = strstr(s, cookie_name)) != NULL; s += name_len)
+    for (; (s = strstr(s, cookie_name)) != NULL; s += name_len) {
       if (s[name_len] == '=') {
         s += name_len + 1;
         if ((p = strchr(s, ' ')) == NULL)
@@ -1720,10 +1722,13 @@ int mg_get_cookie(const struct mg_connection *conn, const char *cookie_name,
         if ((size_t) (p - s) < dst_size) {
           len = p - s;
           mg_strlcpy(dst, s, (size_t) len + 1);
+        } else {
+          len = -2;
         }
         break;
       }
-
+    }
+  }
   return len;
 }
 

mongoose.h

@@ -308,9 +308,9 @@ int mg_get_var(const char *data, size_t data_len,
 //
 // Return:
 //   On success, value length.
-//   On error, -1 (either "Cookie:" header is not present at all, or the
-//   requested parameter is not found, or destination buffer is too small
-//   to hold the value).
+//   On error:
+//      -1 (either "Cookie:" header is not present at all or the requested parameter is not found).
+//      -2 (destination buffer is NULL, zero length or too small to hold the value).
 int mg_get_cookie(const struct mg_connection *,
                   const char *cookie_name, char *buf, size_t buf_len);