lib/cgi/cgi-args.cc
changeset 0 6f7a81934006
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/lib/cgi/cgi-args.cc	Wed Jan 16 22:39:43 2008 +0100
@@ -0,0 +1,142 @@
+// Copyright (C) 1999,2000 Bruce Guenter <bruceg@em.ca>
+//
+// 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+#include <config.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include "cgi-args.h"
+#include "fdbuf/fdbuf.h"
+
+// These routines are optimized for the common case:
+// - few variables (less than ten)
+// - short strings (less than 1K each)
+// - infrequent access relative to the remainder of the code
+
+CGIArgs::~CGIArgs()
+{
+  if(head)
+    delete head;
+}
+
+bool CGIArgs::exists(mystring var) const
+{
+  for(arg* ptr = head; ptr; ptr = ptr->next)
+    if(ptr->var == var)
+      return true;
+  return false;
+}
+
+mystring CGIArgs::get(mystring var, mystring dflt) const
+{
+  for(arg* ptr = head; ptr; ptr = ptr->next)
+    if(ptr->var == var)
+      return ptr->val;
+  return dflt;
+}
+
+void CGIArgs::append(mystring var, mystring val)
+{
+  arg* newarg = new arg(var, val);
+  if(tail)
+    tail->next = newarg;
+  else
+    head = newarg;
+  tail = newarg;
+  ++argc;
+}
+
+static inline int fromhex(char ch)
+{
+  if(ch >= '0' && ch <= '9')
+    return ch - '0';
+  else if(ch >= 'a' && ch <= 'f')
+    return ch - 'a' + 0xa;
+  else if(ch >= 'A' && ch <= 'F')
+    return ch - 'A' + 0xa;
+  else
+    return -1;
+}
+
+static bool get_parts(mystring& var, mystring& val)
+{
+  if(!fin) return false;
+#define BUFSIZE 1024
+  char buf[BUFSIZE];
+  int i = 0;
+  char ch;
+  bool havevar = false;
+  while(fin.get(ch)) {
+    switch(ch) {
+    case '+':
+      buf[i++] = ' ';
+      break;
+    case '%':
+      char ch1, ch2;
+      if(fin.get(ch1) && fin.get(ch2))
+	buf[i++] = fromhex(ch1) << 4 | fromhex(ch2);
+      break;
+    case '=':
+      if(havevar)
+	buf[i++] = ch;
+      else {
+	buf[i] = 0;
+	var += buf;
+	i = 0;
+	havevar = true;
+      }
+      break;
+    case '&':
+      buf[i] = 0;
+      if(havevar)
+	val += buf;
+      else
+	var += buf;
+      return true;
+    default:
+      buf[i++] = ch;
+    }
+    if(i >= BUFSIZE-1) {
+      buf[i] = 0;
+      if(!havevar) var += buf;
+      else val += buf;
+      i = 0;
+    }
+  }
+  if(havevar) {
+    buf[i] = 0;
+    val += buf;
+  }
+  return havevar;
+}
+
+void CGIArgs::init()
+{
+  mystring method = getenv("REQUEST_METHOD");
+  mystring content = getenv("CONTENT_TYPE");
+  if(method != "POST" && method != "post")
+    errstr = "Must be accessed via a CGI 'POST' form";
+  else if(content == "multipart/form-data")
+    errstr = "Must be accessed via a CGI 'POST' form";
+  else {
+    for(;;) {
+      mystring var;
+      mystring val;
+      if(!get_parts(var, val))
+	break;
+      append(var, val);
+    }
+  }
+}