src/callcgi.c

   1 /*
   2     ashd - A Sane HTTP Daemon
   3     Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>
   4
   5     This program is free software: you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation, either version 3 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #include <stdlib.h>
  20 #include <string.h>
  21 #include <stdio.h>
  22 #include <unistd.h>
  23 #include <errno.h>
  24 #include <ctype.h>
  25 #include <signal.h>
  26
  27 #ifdef HAVE_CONFIG_H
  28 #include <config.h>
  29 #endif
  30 #include <utils.h>
  31 #include <log.h>
  32
  33 static char **environ;
  34
  35 static void passdata(FILE *in, FILE *out)
  36 {
  37     int ret;
  38     char *buf;
  39
  40     buf = smalloc(65536);
  41     while(!feof(in)) {
  42         ret = fread(buf, 1, 65536, in);
  43         if(ferror(in)) {
  44             flog(LOG_ERR, "sendfile: could not read input: %s", strerror(errno));
  45             break;
  46         }
  47         if(fwrite(buf, 1, ret, out) != ret) {
  48             flog(LOG_ERR, "sendfile: could not write output: %s", strerror(errno));
  49             break;
  50         }
  51     }
  52     free(buf);
  53 }
  54
  55 static char *absolutify(char *file)
  56 {
  57     char cwd[1024];
  58
  59     if(*file != '/') {
  60         getcwd(cwd, sizeof(cwd));
  61         return(sprintf2("%s/%s", cwd, file));
  62     }
  63     return(sstrdup(file));
  64 }
  65
  66 static void forkchild(int inpath, char *prog, char *file, char *method, char *url, char *rest, int *infd, int *outfd)
  67 {
  68     int i;
  69     char *qp, **env, *name;
  70     int inp[2], outp[2];
  71     pid_t pid;
  72
  73     pipe(inp);
  74     pipe(outp);
  75     if((pid = fork()) < 0) {
  76         flog(LOG_ERR, "callcgi: could not fork");
  77         exit(1);
  78     }
  79     if(pid == 0) {
  80         close(inp[1]);
  81         close(outp[0]);
  82         dup2(inp[0], 0);
  83         dup2(outp[1], 1);
  84         for(i = 3; i < FD_SETSIZE; i++)
  85             close(i);
  86         if((qp = strchr(url, '?')) != NULL)
  87             *(qp++) = 0;
  88         putenv(sprintf2("SERVER_SOFTWARE=ashd/%s", VERSION));
  89         putenv("GATEWAY_INTERFACE=CGI/1.1");
  90         if(getenv("HTTP_VERSION"))
  91             putenv(sprintf2("SERVER_PROTOCOL=%s", getenv("HTTP_VERSION")));
  92         putenv(sprintf2("REQUEST_METHOD=%s", method));
  93         putenv(sprintf2("PATH_INFO=%s", rest));
  94         name = url;
  95         /* XXX: This is an ugly hack (I think), but though I can think
  96          * of several alternatives, none seem to be better. */
  97         if(*rest && (strlen(url) >= strlen(rest)) &&
  98            !strcmp(rest, url + strlen(url) - strlen(rest))) {
  99             name = sprintf2("%.*s", (int)(strlen(url) - strlen(rest)), url);
 100         }
 101         putenv(sprintf2("SCRIPT_NAME=%s", name));
 102         putenv(sprintf2("QUERY_STRING=%s", qp?qp:""));
 103         if(getenv("REQ_HOST"))
 104             putenv(sprintf2("SERVER_NAME=%s", getenv("REQ_HOST")));
 105         if(getenv("REQ_X_ASH_SERVER_PORT"))
 106             putenv(sprintf2("SERVER_PORT=%s", getenv("REQ_X_ASH_SERVER_PORT")));
 107         if(getenv("REQ_X_ASH_PROTOCOL") && !strcmp(getenv("REQ_X_ASH_PROTOCOL"), "https"))
 108             putenv("HTTPS=on");
 109         if(getenv("REQ_X_ASH_ADDRESS"))
 110             putenv(sprintf2("REMOTE_ADDR=%s", getenv("REQ_X_ASH_ADDRESS")));
 111         if(getenv("REQ_CONTENT_TYPE"))
 112             putenv(sprintf2("CONTENT_TYPE=%s", getenv("REQ_CONTENT_TYPE")));
 113         if(getenv("REQ_CONTENT_LENGTH"))
 114             putenv(sprintf2("CONTENT_LENGTH=%s", getenv("REQ_CONTENT_LENGTH")));
 115         for(env = environ; *env; env++) {
 116             if(!strncmp(*env, "REQ_", 4))
 117                 putenv(sprintf2("HTTP_%s", (*env) + 4));
 118         }
 119         /*
 120          * This is (understandably) missing from the CGI
 121          * specification, but PHP seems to require it.
 122          */
 123         putenv(sprintf2("SCRIPT_FILENAME=%s", absolutify(file)));
 124         if(inpath)
 125             execlp(prog, prog, file, NULL);
 126         else
 127             execl(prog, prog, file, NULL);
 128         exit(127);
 129     }
 130     close(inp[0]);
 131     close(outp[1]);
 132     *infd = inp[1];
 133     *outfd = outp[0];
 134 }
 135
 136 static void trim(struct charbuf *buf)
 137 {
 138     char *p;
 139
 140     for(p = buf->b; (p - buf->b < buf->d) && isspace(*p); p++);
 141     memmove(buf->b, p, buf->d -= (p - buf->b));
 142     for(p = buf->b + buf->d - 1; (p > buf->b) && isspace(*p); p--, buf->d--);
 143 }
 144
 145 static char **parseheaders(FILE *s)
 146 {
 147     int c, state;
 148     struct charvbuf hbuf;
 149     struct charbuf buf;
 150
 151     bufinit(hbuf);
 152     bufinit(buf);
 153     state = 0;
 154     while(1) {
 155         c = fgetc(s);
 156     again:
 157         if(state == 0) {
 158             if(c == '\r') {
 159             } else if(c == '\n') {
 160                 break;
 161             } else if(c == EOF) {
 162                 goto fail;
 163             } else {
 164                 state = 1;
 165                 goto again;
 166             }
 167         } else if(state == 1) {
 168             if(c == ':') {
 169                 trim(&buf);
 170                 bufadd(buf, 0);
 171                 bufadd(hbuf, buf.b);
 172                 bufinit(buf);
 173                 state = 2;
 174             } else if(c == '\r') {
 175             } else if(c == '\n') {
 176                 goto fail;
 177             } else if(c == EOF) {
 178                 goto fail;
 179             } else {
 180                 bufadd(buf, c);
 181             }
 182         } else if(state == 2) {
 183             if(c == '\r') {
 184             } else if(c == '\n') {
 185                 trim(&buf);
 186                 bufadd(buf, 0);
 187                 bufadd(hbuf, buf.b);
 188                 bufinit(buf);
 189                 state = 0;
 190             } else if(c == EOF) {
 191                 goto fail;
 192             } else {
 193                 bufadd(buf, c);
 194             }
 195         }
 196     }
 197     bufadd(hbuf, NULL);
 198     return(hbuf.b);
 199
 200 fail:
 201     buffree(hbuf);
 202     buffree(buf);
 203     return(NULL);
 204 }
 205
 206 static char *defstatus(int code)
 207 {
 208     if(code == 200)
 209         return("OK");
 210     else if(code == 201)
 211         return("Created");
 212     else if(code == 202)
 213         return("Accepted");
 214     else if(code == 204)
 215         return("No Content");
 216     else if(code == 300)
 217         return("Multiple Choices");
 218     else if(code == 301)
 219         return("Moved Permanently");
 220     else if(code == 302)
 221         return("Found");
 222     else if(code == 303)
 223         return("See Other");
 224     else if(code == 304)
 225         return("Not Modified");
 226     else if(code == 307)
 227         return("Moved Temporarily");
 228     else if(code == 400)
 229         return("Bad Request");
 230     else if(code == 401)
 231         return("Unauthorized");
 232     else if(code == 403)
 233         return("Forbidden");
 234     else if(code == 404)
 235         return("Not Found");
 236     else if(code == 500)
 237         return("Internal Server Error");
 238     else if(code == 501)
 239         return("Not Implemented");
 240     else if(code == 503)
 241         return("Service Unavailable");
 242     else
 243         return("Unknown status");
 244 }
 245
 246 static void sendstatus(char **headers, FILE *out)
 247 {
 248     char **hp;
 249     char *status, *location;
 250
 251     hp = headers;
 252     status = location = NULL;
 253     while(*hp) {
 254         if(!strcasecmp(hp[0], "status")) {
 255             status = hp[1];
 256             /* Clear this header, so that it is not transmitted by sendheaders. */
 257             **hp = 0;
 258         } else if(!strcasecmp(hp[0], "location")) {
 259             location = hp[1];
 260             hp += 2;
 261         } else {
 262             hp += 2;
 263         }
 264     }
 265     if(status) {
 266         if(strchr(status, ' '))
 267             fprintf(out, "HTTP/1.1 %s\n", status);
 268         else
 269             fprintf(out, "HTTP/1.1 %i %s\n", atoi(status), defstatus(atoi(status)));
 270     } else if(location) {
 271         fprintf(out, "HTTP/1.1 303 See Other\n");
 272     } else {
 273         fprintf(out, "HTTP/1.1 200 OK\n");
 274     }
 275 }
 276
 277 static void sendheaders(char **headers, FILE *out)
 278 {
 279     while(*headers) {
 280         if(**headers)
 281             fprintf(out, "%s: %s\n", headers[0], headers[1]);
 282         headers += 2;
 283     }
 284 }
 285
 286 static void usage(void)
 287 {
 288     flog(LOG_ERR, "usage: callcgi [-c] [-p PROGRAM] METHOD URL REST");
 289 }
 290
 291 int main(int argc, char **argv, char **envp)
 292 {
 293     int c;
 294     char *file, *prog, *sp;
 295     int inpath, cd;
 296     int infd, outfd;
 297     FILE *in, *out;
 298     char **headers;
 299
 300     environ = envp;
 301     signal(SIGPIPE, SIG_IGN);
 302
 303     prog = NULL;
 304     inpath = 0;
 305     cd = 0;
 306     while((c = getopt(argc, argv, "cp:")) >= 0) {
 307         switch(c) {
 308         case 'c':
 309             cd = 1;
 310             break;
 311         case 'p':
 312             prog = optarg;
 313             inpath = 1;
 314             break;
 315         default:
 316             usage();
 317             exit(1);
 318         }
 319     }
 320
 321     if(argc - optind < 3) {
 322         usage();
 323         exit(1);
 324     }
 325     if((file = getenv("REQ_X_ASH_FILE")) == NULL) {
 326         flog(LOG_ERR, "callcgi: needs to be called with the X-Ash-File header");
 327         exit(1);
 328     }
 329
 330     if(cd) {
 331         /* This behavior is encouraged by the CGI specification (RFC 3875, 7.2),
 332          * but not strictly required, and I get the feeling it might break some
 333          * relative paths here or there, so it's not the default for now. */
 334         if((sp = strrchr(file, '/')) != NULL) {
 335             *sp = 0;
 336             if(chdir(file)) {
 337                 *sp = '/';
 338             } else {
 339                 file = sp + 1;
 340             }
 341         }
 342     }
 343
 344     if(prog == NULL)
 345         prog = file;
 346     forkchild(inpath, prog, file, argv[optind], argv[optind + 1], argv[optind + 2], &infd, &outfd);
 347     in = fdopen(infd, "w");
 348     passdata(stdin, in);
 349     fclose(in);
 350     out = fdopen(outfd, "r");
 351     if((headers = parseheaders(out)) == NULL) {
 352         flog(LOG_WARNING, "CGI handler returned invalid headers");
 353         exit(1);
 354     }
 355     sendstatus(headers, stdout);
 356     sendheaders(headers, stdout);
 357     printf("\n");
 358     passdata(out, stdout);
 359     return(0);
 360 }