Added querystring to request attributes

Sebastiano Merlino · Sebastiano Merlino · commit 5f4313e59f81 · 2012-10-10T17:16:26.000+02:00
diff --git a/src/httpserver/http_request.hpp b/src/httpserver/http_request.hpp
@@ -355,6 +355,14 @@ class http_request
         {
             result = this->content;
         }
+        const std::string get_querystring() const
+        {
+            return this->querystring;
+        }
+        void get_querystring(std::string& result) const
+        {
+            result = this->querystring;
+        }
         /**
          * Method used to get the version of the request.
          * @return the version in string representation
@@ -411,6 +419,7 @@ class http_request
             footers(b.footers),
             cookies(b.cookies),
             args(b.args),
+            querystring(b.querystring),
             content(b.content),
             version(b.version),
             requestor(b.requestor),
@@ -427,6 +436,7 @@ class http_request
         std::map<std::string, std::string, header_comparator> footers;
         std::map<std::string, std::string, header_comparator> cookies;
         std::map<std::string, std::string, arg_comparator> args;
+        std::string querystring;
         std::string content;
         std::string version;
         std::string requestor;
diff --git a/src/webserver.cpp b/src/webserver.cpp
@@ -645,6 +645,19 @@ int webserver::build_request_footer (void *cls, enum MHD_ValueKind kind, const c
 int webserver::build_request_args (void *cls, enum MHD_ValueKind kind, const char *key, const char *value)
 {
     modded_request* mr = static_cast<modded_request*>(cls);
+    {
+        char buf[strlen(key) + strlen(value) + 3];
+        if(mr->dhr->querystring == "")
+        {
+            snprintf(buf, sizeof buf, "?%s=%s", key, value);
+            mr->dhr->querystring = buf;
+        }
+        else
+        {
+            snprintf(buf, sizeof buf, "&%s=%s", key, value);
+            mr->dhr->querystring += string(buf);
+        }
+    }
     int size = internal_unescaper((void*) mr->ws, (char*) value);
     mr->dhr->set_arg(key, string(value, size));
     return MHD_YES;
