1 files changed, 102 insertions, 0 deletions

diff --git a/src/serv/form.cc b/src/serv/form.cc
new file mode 100644
index 0000000..815f9c4
--- /dev/null
+++ b/src/serv/form.cc
@@ -0,0 +1,102 @@
+/*************************************************************************
+ *
+ * HTCd - Copyright (C) 1998-2006 Henrik Rydberg
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+#include <serv/htcd.h>
+
+const mstring tag="T.";
+const mstring ncookie_tag="ncookie";
+const mstring multipart_tag="multipart";
+const mstring body_tag="T.body";
+
+static mstring Tag(const sref& in)
+{
+  mstring name=in;
+  for(int i=0;i<name.size();i++) if(name[i]=='_') name[i]='.';
+  if(tag.empty()||tag==name.left(tag.size())) return name;
+  name.insert(0,tag);
+  return name;
+}
+
+static void AddConst(tokmap& map,const sref& s,const sref& b) throw(merror_t)
+{
+  tok_t* p=map.get_this(s);
+  if(p) {
+    mstring nb=p->body; nb.append(1,' '); nb+=b;
+    Const(map,s,nb,tok_t::PUBLIC,1);
+  }
+  else Const(map,s,b,tok_t::PUBLIC);
+}
+
+static void PushMultipart(tokmap& map,const Mime& root) throw(merror_t)
+{
+  rpile part;
+  if(root.Multipart(part)) {
+    for(int i=0;i<part.size();i++) {
+      Mime msg(part[i]);
+      if(msg.cont_disp_name.nempty()) {
+        mstring name=Tag(msg.cont_disp_name);
+        if(msg.cont_disp_file.nempty()) {
+          reference_t* tok=Reference(map,name,msg.cont_disp_file,tok_t::PUBLIC);
+          swrite via;
+          if(eqn(msg.cont_trans_enc,trans_base64))
+            decode64(via(tok->data),msg.body);
+          else if(eqn(msg.cont_trans_enc,trans_quote))
+            decodeQP(via(tok->data),msg.body);
+          else
+            via(tok->data)<<msg.body;
+        }
+        else AddConst(map,name,msg.body);
+      }
+    }
+  }
+}
+
+static void PushWWWForm(tokmap& map,const sref& in) throw(merror_t)
+{
+  sspile tmp;
+  Unpack(tmp,in);
+  for(int i=0;i<tmp.size();i++)
+    AddConst(map,Tag(tmp[i].first),tmp[i].second);
+}
+
+void SetCookies(tokmap& map,const httpReq& req)
+{
+  Const(map,ncookie_tag,itoa(req.cookies.size()),tok_t::PUBLIC);
+  int n=0;
+  for(sspile::const_iterator p=req.cookies.begin();p!=req.cookies.end();p++) {
+    mstring name,val;
+    name.convert("cookie%d.name",n);
+    val.convert("cookie%d.val",n);
+    Const(map,name,p->first,tok_t::PUBLIC);
+    Const(map,val,p->second,tok_t::PUBLIC);
+    n++;
+  }
+  if(req.arg.nempty()) PushWWWForm(map,req.arg);
+}
+
+void envAddForm(tokmap& map,const httpReq& req)
+{
+  if(req.method==HTTP_POST) {
+    if(req.cont_type.find(multipart_tag,cset_lcase)>=0) PushMultipart(map,req);
+    else PushWWWForm(map,req.body);
+  }
+  else if(req.method==HTTP_PUT) {
+    Const(map,body_tag,req.body,tok_t::PUBLIC);
+  }
+}