1 files changed, 88 insertions, 0 deletions

diff --git a/src/cgi_gateway.cc b/src/cgi_gateway.cc
new file mode 100644
index 0000000..eae7a03
--- a/dev/null
+++ b/src/cgi_gateway.cc
@@ -0,0 +1,88 @@
+#include "kingate/cgi_gateway.h"
+#include "kingate/util.h"
+#include "kingate/exception.h"
+
+namespace kingate {
+
+    cgi_gateway::cgi_gateway(cgi_interface& ci)
+        : iface(ci), b_parsed_content(false) {
+            // Fetch GET content
+            if(iface.has_meta("QUERY_STRING")) {
+                string qs = iface.get_meta("QUERY_STRING");
+                parse_query(qs,get);
+            }
+            // Fetch POST content
+            if(!strcasecmp(get_content_type().c_str(),"application/x-www-form-urlencoded")) {
+                unsigned long cl = get_content_length();
+                if(cl) {
+                    char * tmp = new char[cl];
+                    iface.in().read(tmp,cl);
+                    string qs(tmp,cl);
+                    delete tmp;
+                    parse_query(qs,post);
+                }
+                b_parsed_content = true;
+            }
+        }
+
+    bool cgi_gateway::has_GET(const string& n) const {
+        return get.find(n) != get.end();
+    }
+    string cgi_gateway::get_GET(const string& n) const {
+        params_t::const_iterator i = get.find(n);
+        if(i==get.end())
+            throw exception_notfound(CODEPOINT,"no such parameter");
+        return i->second;
+    }
+    bool cgi_gateway::has_POST(const string& n) const {
+        return post.find(n) != post.end();
+    }
+    string cgi_gateway::get_POST(const string& n) const {
+        params_t::const_iterator i = post.find(n);
+        if(i==post.end())
+            throw exception_notfound(CODEPOINT,"no such parameter");
+        return i->second;
+    }
+    bool cgi_gateway::has_param(const string& n) const {
+        return has_GET(n) || has_POST(n);
+    }
+    string cgi_gateway::get_param(const string& n) const {
+        params_t::const_iterator i = get.find(n);
+        if(i!=get.end())
+            return i->second;
+        i = post.find(n);
+        if(i!=post.end())
+            return i->second;
+        throw exception_notfound(CODEPOINT,"no such parameter");
+    }
+
+    const string& cgi_gateway::get_content_type() const {
+        if(!has_meta("CONTENT_TYPE"))
+            return ""; // XXX:
+        return get_meta("CONTENT_TYPE");
+    }
+    unsigned long cgi_gateway::get_content_length() const {
+        if(!has_meta("CONTENT_LENGTH"))
+            return 0;
+        string cl = get_meta("CONTENT_LENGTH");
+        return strtol(cl.c_str(),NULL,10);
+    }
+
+    void cgi_gateway::parse_query(string& q,params_t& p) {
+        while(!q.empty()) {
+            string::size_type amp = q.find('&');
+            string pp = (amp==string::npos)?q:q.substr(0,amp);
+            if(amp==string::npos)
+                q.clear();
+            else
+                q.erase(0,amp+1);
+            string::size_type eq = pp.find('=');
+            if(eq == string::npos) {
+                p.insert(params_t::value_type("",url_unescape(pp)));
+            }else{
+                p.insert(params_t::value_type(url_unescape(pp.substr(0,eq)),url_unescape(pp.substr(eq+1))));
+            }
+        }
+    }
+
+}