chiark / gitweb /
ucgi/: Return useful status codes when things go wrong.
[userv-utils.git] / www-cgi / ucgicommon.c
index 0facfbb7bb20f39eb0201e7f2974a8869587b5ac..2fa6260af02d4373a6d430df489e7b113eccdfd7 100644 (file)
@@ -18,6 +18,9 @@
  * $Id$
  */
 
+#include <assert.h>
+#include <ctype.h>
+#include <stdarg.h>
 #include <stdio.h>
 #include <string.h>
 #include <errno.h>
 
 #include "ucgi.h"
 
-const char *const envok[]= {
-  "AUTH_TYPE",
-  "CONTENT_LENGTH",
-  "CONTENT_TYPE",
-  "DOCUMENT_ROOT",
-  "GATEWAY_INTERFACE",
-  "HTTP_ACCEPT",
-  "HTTP_ACCEPT_CHARSET",
-  "HTTP_ACCEPT_ENCODING",
-  "HTTP_ACCEPT_LANGUAGE",
-  "HTTP_CACHE_CONTROL",
-  "HTTP_CONNECTION",
-  "HTTP_CONTENT_ENCODING",
-  "HTTP_COOKIE",
-  "HTTP_DNT",
-  "HTTP_HOST",
-  "HTTP_KEEP_ALIVE",
-  "HTTP_NEGOTIATE",
-  "HTTP_PRAGMA",
-  "HTTP_REFERER",
-  "HTTP_USER_AGENT",
-  "HTTP_VIA",
-  "HTTP_X_FORWARDED_FOR",
-  "HTTPS",
-  "PATH_INFO",
-  "PATH_TRANSLATED",
-  "QUERY_STRING",
-  "REMOTE_ADDR",
-  "REMOTE_HOST",
-  "REMOTE_USER",
-  "REMOTE_IDENT",
-  "REQUEST_METHOD",
-  "REQUEST_URI",
-  "SCRIPT_FILENAME",
-  "SCRIPT_NAME",
-  "SCRIPT_URI",
-  "SCRIPT_URL",
-  "SERVER_ADDR",
-  "SERVER_ADMIN",
-  "SERVER_NAME",
-  "SERVER_PORT",
-  "SERVER_PROTOCOL",
-  "SERVER_SIGNATURE",
-  "SERVER_SOFTWARE",
-  0
-};
-const int nenvok= sizeof(envok)/sizeof(envok[0]);
-
 int debugmode= 0;
 
 static void outerror(void) {
@@ -82,18 +37,20 @@ static void outerror(void) {
 }
 
 void syserror(const char *m) {
-  if (printf("Content-Type: text/plain\n\n"
+  if (printf("Content-Type: text/plain\n"
+            "Status: 500\n\n"
             "ucgi: system call error:\n"
             "%s: %s\n",
             m,strerror(errno))==EOF || fflush(stdout)) outerror();
   exit(0);
 }
 
-void error(const char *m) {
-  if (printf("Content-Type: text/plain\n\n"
+void error(const char *m, int st) {
+  if (printf("Content-Type: text/plain\n"
+            "Status: %d\n\n"
             "ucgi: error:\n"
             "%s\n",
-            m)==EOF || fflush(stdout)) outerror();
+            st, m)==EOF || fflush(stdout)) outerror();
   exit(0);
 }
 
@@ -117,75 +74,149 @@ void xsetenv(const char *en, const char *ev, int overwrite) {
   if (setenv(en,ev,overwrite)) syserror("setenv");
 }
 
+const char **load_filters(unsigned flags, const char *first, ...)
+{
+  va_list ap;
+  const char *name, *p, *q, **v;
+  char *pp;
+  size_t l, n, sz;
+  FILE *fp;
+  char buf[MAX_ENVVAR_NAME];
+
+  D( if (debugmode) printf(";; load_filters...\n"); )
+  va_start(ap, first);
+  for (name= first; name; name= va_arg(ap, const char *)) {
+    fp= fopen(name, "r"); if (fp) goto opened;
+    D( if (debugmode)
+        printf(";;   file `%s': %s\n", name, strerror(errno)); )
+    if (errno != ENOENT) syserror("failed to open environment filters");
+  }
+  va_end(ap);
+  if (flags & LOADF_MUST) syserror("failed to open environment filters");
+  D( if (debugmode) printf(";;   using default filters\n"); )
+  return 0;
+
+opened:
+  va_end(ap);
+  D( if (debugmode) printf(";;   file `%s': OK\n", name); )
+
+  n= 0; sz= 128; v= xmalloc(sz * sizeof(*v));
+  for (;;) {
+    if (!fgets(buf, sizeof(buf), fp)) break;
+    l= strlen(buf);
+    if (buf[l - 1] == '\n') buf[--l]= 0;
+    if (l + 1 == sizeof(buf))
+      error("line too long in environment filter file", 500);
+    p= buf; q= p + l;
+    while (isspace((unsigned char)*p)) p++;
+    while (q > p && isspace((unsigned char)q[-1])) q--;
+    if (*p == '#' || p == q) continue;
+    l= q - p;
+    pp= xmalloc(l + 1);
+    memcpy(pp, p, l);
+    pp[l]= 0;
+    v[n++]= pp;
+    D( if (debugmode) printf(";;   filter: `%s'\n", pp); )
+    if (n >= sz) {
+      sz *= 2;
+      v= xrealloc(v, sz * sizeof(*v));
+    }
+  }
+  if (ferror(fp)) syserror("failed to read environment filters");
+  fclose(fp);
+  return v;
+}
+
+static int envvar_match(unsigned flags, const char *en,
+                       const char *const *patv,
+                       const char *const *defaults,
+                       const char **ev)
+{
+  const char *const *patp;
+  const char *q, *pat;
+  int acceptp;
+  int rc;
+
+  if (!patv) { patv= defaults; defaults= 0; }
+  for (patp= patv; (pat= *patp); patp++) {
+    q= en;
+    acceptp= 1;
+    if (*pat == '!' && (flags & FILTF_WILDCARD)) { acceptp= 0; pat++; }
+    else if (*pat == '?') {
+      if (strcmp(pat + 1, "DEFAULTS") == 0) {
+       assert(defaults);
+       rc= envvar_match(flags, en, defaults, 0, ev);
+       if (rc) return rc;
+      } else
+       error("unknown pattern directive", 500);
+      continue;
+    }
+
+    for (;;) {
+      if (!*pat) {
+       if (*q != '=') {
+         D( if (debugmode)
+              printf(";;     mismatch `%s' (prefix)\n", *patp); )
+         break;
+       }
+       D( if (debugmode) printf(";;     matched pattern `%s'\n", *patp); )
+       goto match;
+      } else if (*pat == '*' && (flags & FILTF_WILDCARD)) {
+       q = strchr(q, '=');
+       if (!q) {
+         D( if (debugmode)
+              printf(";;     mismatch `%s' (discard: no `=')\n", *patp); )
+         return -1;
+       }
+       D( if (debugmode)
+            printf(";;     wildcard match for `%s'\n", *patp); )
+       goto match;
+      } else {
+       if (*pat++ != *q++) {
+         D( if (debugmode) printf(";;     mismatch `%s'\n", *patp); )
+         break;
+       }
+      }
+    }
+  }
+  return 0;
+
+match:
+  if (!acceptp) return -1;
+  *ev= q + 1;
+  return +1;
+}
+
 void filter_environment(unsigned flags, const char *prefix_in,
                        const char *const *patv,
+                       const char *const *defaults,
                        void (*foundone)(const char *fulln,
                                         const char *en, const char *ev,
                                         void *p),
                        void *p)
 {
   char *const *ep;
-  const char *const *patp;
-  const char *en, *ev, *pat, *q;
+  const char *en, *ev;
   char enbuf[MAX_ENVVAR_NAME];
   size_t n, pn = strlen(prefix_in);
-  int acceptp;
 
   D( if (debugmode) printf(";; filter_environment...\n"); )
   for (ep= environ; (en= *ep); ep++) {
     D( if (debugmode) printf(";;   consider env-var `%s'\n", en); )
     if (strncmp(en, prefix_in, pn) != 0 || !en[pn]) {
       D( if (debugmode) printf(";;     doesn't match prefix\n"); )
-      goto next_ev;
+      continue;
     }
-    for (patp= patv; (pat= *patp); patp++) {
-      q= en + pn;
-      acceptp= 1;
-      if (*pat == '!' && (flags & FILTF_WILDCARD)) {
-       acceptp= 0; pat++;
-      }
-      for (;;) {
-       if (!*pat) {
-         if (*q != '=') {
-           D( if (debugmode)
-                printf(";;     mismatch `%s' (prefix)\n", *patp); )
-           goto next_pat;
-         }
-         D( if (debugmode) printf(";;     matched `%s'\n", *patp); )
-         ev = q + 1;
-         break;
-       } else if (*pat == '*' && (flags & FILTF_WILDCARD)) {
-         q = strchr(q, '=');
-         if (!q) {
-           D( if (debugmode)
-                printf(";;     mismatch `%s' (discard: no `=')\n", *patp); )
-           goto next_ev;
-         }
-         D( if (debugmode)
-              printf(";;     wildcard match for `%s'\n", *patp); )
-         ev = q + 1;
-         break;
-       } else
-         if (*pat++ != *q++) {
-           D( if (debugmode) printf(";;     mismatch `%s'\n", *patp); )
-           goto next_pat;
-         }
-      }
-      if (acceptp) {
-       n= q - en;
-       if (n >= sizeof(enbuf))
-         error("environment variable name too long");
-       memcpy(enbuf, en, n);
-       enbuf[n]= 0;
-       D( if (debugmode)
-            printf(";;     full = `%s'; tail = `%s'; value = `%s'\n",
-                   enbuf, enbuf + pn, ev); )
-       foundone(enbuf, enbuf + pn, ev, p);
-      } D( else if (debugmode)
-            printf(";;     matched negated pattern\n"); )
-      goto next_ev;
-    next_pat:;
+    if (envvar_match(flags, en + pn, patv, defaults, &ev) > 0) {
+      n= strcspn(en, "=");
+      if (n >= sizeof(enbuf))
+       error("environment variable name too long", 500);
+      memcpy(enbuf, en, n);
+      enbuf[n]= 0;
+      D( if (debugmode)
+          printf(";;     full = `%s'; tail = `%s'; value = `%s'\n",
+                 enbuf, enbuf + pn, ev); )
+      foundone(enbuf, enbuf + pn, ev, p);
     }
-  next_ev:;
   }
 }