simple_httpd.c revision 133836
143939Sabial/*- 246493Sabial * Simple_HTTPd v1.1 - a very small, barebones HTTP server 343939Sabial * 443939Sabial * Copyright (c) 1998-1999 Marc Nicholas <marc@netstor.com> 543939Sabial * All rights reserved. 643939Sabial * 746493Sabial * Major rewrite by William Lloyd <wlloyd@slap.net> 846493Sabial * 943939Sabial * Redistribution and use in source and binary forms, with or without 1043939Sabial * modification, are permitted provided that the following conditions 1143939Sabial * are met: 1243939Sabial * 1. Redistributions of source code must retain the above copyright 1343939Sabial * notice, this list of conditions and the following disclaimer. 1443939Sabial * 2. Redistributions in binary form must reproduce the above copyright 1543939Sabial * notice, this list of conditions and the following disclaimer in the 1643939Sabial * documentation and/or other materials provided with the distribution. 1743939Sabial * 1843939Sabial * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 1943939Sabial * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 2043939Sabial * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2143939Sabial * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 2243939Sabial * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2343939Sabial * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2443939Sabial * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2543939Sabial * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2643939Sabial * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 2743939Sabial * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 2843939Sabial * SUCH DAMAGE. 2943939Sabial * 3050479Speter * $FreeBSD: head/release/picobsd/tinyware/simple_httpd/simple_httpd.c 133836 2004-08-16 09:38:34Z dwmalone $ 3138589Sabial */ 3238589Sabial 3338589Sabial#include <sys/stat.h> 3438589Sabial#include <sys/time.h> 3538589Sabial#include <sys/types.h> 3638589Sabial#include <sys/socket.h> 3746493Sabial#include <sys/wait.h> 3838589Sabial#include <netinet/in.h> 3938589Sabial#include <arpa/inet.h> 4046493Sabial 4146493Sabial#include <fcntl.h> 4238589Sabial#include <netdb.h> 4346493Sabial#include <signal.h> 4446493Sabial#include <stdio.h> 4546493Sabial#include <stdlib.h> 4638589Sabial#include <string.h> 4746493Sabial#include <time.h> 4846493Sabial#include <unistd.h> 4938589Sabial 5046493Sabialint http_port = 80; 5146493Sabialint daemonize = 1; 5246493Sabialint verbose = 0; 5338589Sabialint http_sock, con_sock; 5446493Sabial 55113126Sdwmaloneconst char *fetch_mode = NULL; 5646493Sabialchar homedir[100]; 5746493Sabialchar logfile[80]; 58113126Sdwmalonechar *adate(void); 59113126Sdwmalonevoid init_servconnection(void); 60113126Sdwmalonevoid http_date(void); 61113126Sdwmalonevoid http_output(const char *html); 62113126Sdwmalonevoid http_request(void); 63113126Sdwmalonevoid log_line(char *req); 64113126Sdwmalonevoid wait_connection(void); 6546493Sabial 6638589Sabialstruct hostent *hst; 6746493Sabialstruct sockaddr_in source; 6838589Sabial 6946493Sabial/* HTTP basics */ 7046493Sabialstatic char httpd_server_ident[] = "Server: FreeBSD/PicoBSD simple_httpd 1.1\r"; 7146493Sabial 7246493Sabialstatic char http_200[] = "HTTP/1.0 200 OK\r"; 7346493Sabial 74133836Sdwmaloneconst char *default_mime_type = "application/octet-stream"; 75133836Sdwmalone 76133836Sdwmaloneconst char *mime_type[][2] = { 77133836Sdwmalone { "txt", "text/plain" }, 78133836Sdwmalone { "htm", "text/html" }, 79133836Sdwmalone { "html", "text/html" }, 80133836Sdwmalone { "gif", "image/gif" }, 81133836Sdwmalone { "jpg", "image/jpeg" }, 82133836Sdwmalone { "mp3", "audio/mpeg" } 83133836Sdwmalone}; 84133836Sdwmalone 85133836Sdwmaloneconst int mime_type_max = sizeof(mime_type) / sizeof(mime_type[0]) - 1; 86133836Sdwmalone 8746493Sabial/* Two parts, HTTP Header and then HTML */ 88113126Sdwmalonestatic const char *http_404[2] = 8946493Sabial {"HTTP/1.0 404 Not found\r\n", 9046493Sabial"<HTML><HEAD><TITLE>Error</TITLE></HEAD><BODY><H1>Error 404</H1>\ 9146493SabialNot found - file doesn't exist or you do not have permission.\n</BODY></HTML>\r\n" 9246493Sabial}; 9346493Sabial 94113126Sdwmalonestatic const char *http_405[2] = 9546493Sabial {"HTTP/1.0 405 Method Not allowed\r\nAllow: GET,HEAD\r\n", 9646493Sabial"<HTML><HEAD><TITLE>Error</TITLE></HEAD><BODY><H1>Error 405</H1>\ 9746493SabialThis server only supports GET and HEAD requests.\n</BODY></HTML>\r\n" 9846493Sabial}; 9946493Sabial 10046493Sabial/* 10146493Sabial * Only called on initial invocation 10246493Sabial */ 10338589Sabialvoid 10438589Sabialinit_servconnection(void) 10538589Sabial{ 10638589Sabial struct sockaddr_in server; 10738589Sabial 10838589Sabial /* Create a socket */ 10938589Sabial http_sock = socket(AF_INET, SOCK_STREAM, 0); 11038589Sabial if (http_sock < 0) { 11138589Sabial perror("socket"); 11238589Sabial exit(1); 11338589Sabial } 11438589Sabial server.sin_family = AF_INET; 11538589Sabial server.sin_port = htons(http_port); 11638589Sabial server.sin_addr.s_addr = INADDR_ANY; 11738589Sabial if (bind(http_sock, (struct sockaddr *) & server, sizeof(server)) < 0) { 11838589Sabial perror("bind socket"); 11938589Sabial exit(1); 12038589Sabial } 12194130Sasmodai if (verbose) printf("simple_httpd:%d\n",http_port); 12238589Sabial} 12338589Sabial 12446493Sabial/* 12546493Sabial * Wait here until we see an incoming http request 12646493Sabial */ 12794134Sasmodaivoid 12846493Sabialwait_connection(void) 12938589Sabial{ 130113126Sdwmalone socklen_t lg; 13138589Sabial 13238589Sabial lg = sizeof(struct sockaddr_in); 13338589Sabial 13438589Sabial con_sock = accept(http_sock, (struct sockaddr *) & source, &lg); 13538589Sabial if (con_sock <= 0) { 13638589Sabial perror("accept"); 13738589Sabial exit(1); 13838589Sabial } 13938589Sabial} 14038589Sabial 14146493Sabial/* 14246493Sabial * Print timestamp for HTTP HEAD and GET 14346493Sabial */ 14494134Sasmodaivoid 14594134Sasmodaihttp_date(void) 14638589Sabial{ 14738589Sabial time_t tl; 14838589Sabial char buff[50]; 14938589Sabial 15038589Sabial tl = time(NULL); 15138589Sabial strftime(buff, 50, "Date: %a, %d %h %Y %H:%M:%S %Z\r\n", gmtime(&tl)); 15238589Sabial write(con_sock, buff, strlen(buff)); 153113126Sdwmalone /* return(buff); */ 15438589Sabial} 15538589Sabial 15646493Sabial/* 15746493Sabial * Send data to the open socket 15846493Sabial */ 15994134Sasmodaivoid 160113126Sdwmalonehttp_output(const char *html) 16146493Sabial{ 16246493Sabial write(con_sock, html, strlen(html)); 16346493Sabial write(con_sock, "\r\n", 2); 16446493Sabial} 16538589Sabial 16638589Sabial 16746493Sabial/* 16846493Sabial * Create and write the log information to file 16946493Sabial * Log file format is one line per entry 17046493Sabial */ 17194134Sasmodaivoid 17246493Sabiallog_line(char *req) 17338589Sabial{ 17446493Sabial char log_buff[256]; 17546493Sabial char msg[1024]; 17646493Sabial char env_host[80], env_addr[80]; 17746493Sabial long addr; 17846493Sabial FILE *log; 17938589Sabial 18046493Sabial strcpy(log_buff,inet_ntoa(source.sin_addr)); 18146493Sabial sprintf(env_addr, "REMOTE_ADDR=%s",log_buff); 18246493Sabial 18346493Sabial addr=inet_addr(log_buff); 18446493Sabial 18546493Sabial strcpy(msg,adate()); 18646493Sabial strcat(msg," "); 18746493Sabial hst=gethostbyaddr((char*) &addr, 4, AF_INET); 18846493Sabial 18946493Sabial /* If DNS hostname exists */ 19046493Sabial if (hst) { 19146493Sabial strcat(msg,hst->h_name); 19246493Sabial sprintf(env_host, "REMOTE_HOST=%s",hst->h_name); 19338589Sabial } 19446493Sabial strcat(msg," ("); 19546493Sabial strcat(msg,log_buff); 19646493Sabial strcat(msg,") "); 19746493Sabial strcat(msg,req); 19846493Sabial 19946493Sabial if (daemonize) { 20046493Sabial log=fopen(logfile,"a"); 20146493Sabial fprintf(log,"%s\n",msg); 20246493Sabial fclose(log); 20346493Sabial } else 20446493Sabial printf("%s\n",msg); 20546493Sabial 20646493Sabial /* This is for CGI scripts */ 20746493Sabial putenv(env_addr); 20846493Sabial putenv(env_host); 20938589Sabial} 21038589Sabial 21146493Sabial/* 21246493Sabial * We have a connection. Identify what type of request GET, HEAD, CGI, etc 21346493Sabial * and do what needs to be done 21446493Sabial */ 21594134Sasmodaivoid 21694135Sasmodaihttp_request(void) 21738589Sabial{ 21894132Sasmodai int fd, lg, i; 21946493Sabial int cmd = 0; 22046493Sabial char *p, *par; 221133836Sdwmalone const char *filename, *c, *ext, *type; 22246493Sabial struct stat file_status; 22338589Sabial char req[1024]; 22446493Sabial char buff[8192]; 22538589Sabial 22638589Sabial lg = read(con_sock, req, 1024); 22738589Sabial 228113126Sdwmalone if ((p=strstr(req,"\n"))) *p=0; 229113126Sdwmalone if ((p=strstr(req,"\r"))) *p=0; 23038589Sabial 23146493Sabial log_line(req); 23238589Sabial 23346493Sabial c = strtok(req, " "); 23438589Sabial 23546493Sabial /* Error msg if request is nothing */ 23646493Sabial if (c == NULL) { 23746493Sabial http_output(http_404[0]); 23846493Sabial http_output(http_404[1]); 23946493Sabial goto end_request; 24046493Sabial } 24138589Sabial 24246493Sabial if (strncmp(c, "GET", 3) == 0) cmd = 1; 24346493Sabial if (strncmp(c, "HEAD", 4) == 0) cmd = 2; 24438589Sabial 24546493Sabial /* Do error msg for any other type of request */ 24646493Sabial if (cmd == 0) { 24746493Sabial http_output(http_405[0]); 24846493Sabial http_output(http_405[1]); 24946493Sabial goto end_request; 25038589Sabial } 25138589Sabial 25238589Sabial filename = strtok(NULL, " "); 25338589Sabial 25438589Sabial c = strtok(NULL, " "); 255113126Sdwmalone if (fetch_mode != NULL) filename=fetch_mode; 25638589Sabial if (filename == NULL || 25738589Sabial strlen(filename)==1) filename="/index.html"; 25838589Sabial 25946493Sabial while (filename[0]== '/') filename++; 26046493Sabial 26146493Sabial /* CGI handling. Untested */ 26238589Sabial if (!strncmp(filename,"cgi-bin/",8)) 26338589Sabial { 26438589Sabial par=0; 265113126Sdwmalone if ((par=strstr(filename,"?"))) 26638589Sabial { 26738589Sabial *par=0; 26838589Sabial par++; 26938589Sabial } 27038589Sabial if (access(filename,X_OK)) goto conti; 27146493Sabial stat (filename,&file_status); 272112205Sdwmalone if (setuid(file_status.st_uid)) return; 273112205Sdwmalone if (seteuid(file_status.st_uid)) return; 27438589Sabial if (!fork()) 27538589Sabial { 27638589Sabial close(1); 27738589Sabial dup(con_sock); 278113126Sdwmalone /*printf("HTTP/1.0 200 OK\nContent-type: text/html\n\n\n");*/ 27946493Sabial printf("HTTP/1.0 200 OK\r\n"); 28046493Sabial /* Plug in environment variable, others in log_line */ 28146493Sabial putenv("SERVER_SOFTWARE=FreeBSD/PicoBSD"); 28246493Sabial 28379452Sbrian execlp (filename,filename,par,(char *)0); 28438589Sabial } 28538589Sabial wait(&i); 286112205Sdwmalone return; 28738589Sabial } 28838589Sabial conti: 28938589Sabial if (filename == NULL) { 29046493Sabial http_output(http_405[0]); 29146493Sabial http_output(http_405[1]); 29246493Sabial goto end_request; 29338589Sabial } 29446493Sabial /* End of CGI handling */ 29546493Sabial 29646493Sabial /* Reject any request with '..' in it, bad hacker */ 29738589Sabial c = filename; 29838589Sabial while (*c != '\0') 29946493Sabial if (c[0] == '.' && c[1] == '.') { 30046493Sabial http_output(http_404[0]); 30146493Sabial http_output(http_404[1]); 30246493Sabial goto end_request; 30346493Sabial } else 30446493Sabial c++; 30546493Sabial 30646493Sabial /* Open filename */ 30738589Sabial fd = open(filename, O_RDONLY); 30838589Sabial if (fd < 0) { 30946493Sabial http_output(http_404[0]); 31046493Sabial http_output(http_404[1]); 31146493Sabial goto end_request; 31238589Sabial } 31346493Sabial 31446493Sabial /* Get file status information */ 31546493Sabial if (fstat(fd, &file_status) < 0) { 31646493Sabial http_output(http_404[0]); 31746493Sabial http_output(http_404[1]); 318113126Sdwmalone goto end_request2; 31938589Sabial } 32046493Sabial 32146493Sabial /* Is it a regular file? */ 32246493Sabial if (!S_ISREG(file_status.st_mode)) { 32346493Sabial http_output(http_404[0]); 32446493Sabial http_output(http_404[1]); 325113126Sdwmalone goto end_request2; 32646493Sabial } 32738589Sabial 32846493Sabial /* Past this point we are serving either a GET or HEAD */ 32946493Sabial /* Print all the header info */ 33046493Sabial http_output(http_200); 33146493Sabial http_output(httpd_server_ident); 33246493Sabial http_date(); 33338589Sabial 33494133Sasmodai sprintf(buff, "Content-length: %lld\r\n", file_status.st_size); 335113129Sdwmalone write(con_sock, buff, strlen(buff)); 33638589Sabial 337133836Sdwmalone strcpy(buff, "Content-type: "); 338133836Sdwmalone type = default_mime_type; 339133836Sdwmalone if ((ext = strrchr(filename, '.')) != NULL) { 340133836Sdwmalone for (i = mime_type_max; i >= 0; i--) 341133836Sdwmalone if (strcmp(ext + 1, mime_type[i][0]) == 0) { 342133836Sdwmalone type = mime_type[i][1]; 343133836Sdwmalone break; 344133836Sdwmalone } 34546493Sabial } 346133836Sdwmalone strcat(buff, type); 347133836Sdwmalone http_output(buff); 34846493Sabial 34946493Sabial strftime(buff, 50, "Last-Modified: %a, %d %h %Y %H:%M:%S %Z\r\n\r\n", gmtime(&file_status.st_mtime)); 35046493Sabial write(con_sock, buff, strlen(buff)); 35138589Sabial 35246493Sabial /* Send data only if GET request */ 35338589Sabial if (cmd == 1) { 354113126Sdwmalone while ((lg = read(fd, buff, 8192)) > 0) 35546493Sabial write(con_sock, buff, lg); 35638589Sabial } 35738589Sabial 358113126Sdwmaloneend_request2: 359113126Sdwmalone close(fd); 36046493Sabialend_request: 36138589Sabial close(con_sock); 36238589Sabial 36338589Sabial} 36438589Sabial 36546493Sabial/* 36646493Sabial * Simple httpd server for use in PicoBSD or other embedded application. 36746493Sabial * Should satisfy simple httpd needs. For more demanding situations 36846493Sabial * apache is probably a better (but much larger) choice. 36946493Sabial */ 37094134Sasmodaiint 37146493Sabialmain(int argc, char *argv[]) 37238589Sabial{ 37394132Sasmodai int ch, ld; 37446493Sabial int httpd_group = 65534; 37546493Sabial pid_t server_pid; 37646493Sabial 37746493Sabial /* Default for html directory */ 37838589Sabial strcpy (homedir,getenv("HOME")); 37938589Sabial if (!geteuid()) strcpy (homedir,"/httphome"); 38038589Sabial else strcat (homedir,"/httphome"); 38138589Sabial 38246493Sabial /* Defaults for log file */ 38346493Sabial if (geteuid()) { 38446493Sabial strcpy(logfile,getenv("HOME")); 38546493Sabial strcat(logfile,"/"); 38646493Sabial strcat(logfile,"jhttp.log"); 38746493Sabial } else 38846493Sabial strcpy(logfile,"/var/log/jhttpd.log"); 38938589Sabial 39046493Sabial /* Parse command line arguments */ 39146493Sabial while ((ch = getopt(argc, argv, "d:f:g:l:p:vDh")) != -1) 39246493Sabial switch (ch) { 39346493Sabial case 'd': 39446493Sabial strcpy(homedir,optarg); 39546493Sabial break; 39646493Sabial case 'f': 39746493Sabial daemonize = 0; 39846493Sabial verbose = 1; 399113126Sdwmalone fetch_mode = optarg; 40046493Sabial break; 40146493Sabial case 'g': 40246493Sabial httpd_group = atoi(optarg); 40346493Sabial break; 40446493Sabial case 'l': 40546493Sabial strcpy(logfile,optarg); 40646493Sabial break; 40746493Sabial case 'p': 40846493Sabial http_port = atoi(optarg); 40946493Sabial break; 41046493Sabial case 'v': 41146493Sabial verbose = 1; 41246493Sabial break; 41346493Sabial case 'D': 41446493Sabial daemonize = 0; 41546493Sabial break; 41646493Sabial case '?': 41746493Sabial case 'h': 41846493Sabial default: 41946493Sabial printf("usage: simple_httpd [[-d directory][-g grpid][-l logfile][-p port][-vD]]\n"); 42046493Sabial exit(1); 42146493Sabial /* NOTREACHED */ 42246493Sabial } 42346493Sabial 42446493Sabial /* Not running as root and no port supplied, assume 1080 */ 42546493Sabial if ((http_port == 80) && geteuid()) { 42646493Sabial http_port = 1080; 42746493Sabial } 42846493Sabial 42946493Sabial /* Do we really have rights in the html directory? */ 430113126Sdwmalone if (fetch_mode == NULL) { 43146493Sabial if (chdir(homedir)) { 43246493Sabial perror("chdir"); 43346493Sabial puts(homedir); 43446493Sabial exit(1); 43546493Sabial } 43646493Sabial } 43746493Sabial 43846493Sabial /* Create log file if it doesn't exit */ 43946493Sabial if ((access(logfile,W_OK)) && daemonize) { 44046493Sabial ld = open (logfile,O_WRONLY); 44146493Sabial chmod (logfile,00600); 44246493Sabial close(ld); 44346493Sabial } 44446493Sabial 44538589Sabial init_servconnection(); 44638589Sabial 44746493Sabial if (verbose) { 44846493Sabial printf("Server started with options \n"); 44946493Sabial printf("port: %d\n",http_port); 450113126Sdwmalone if (fetch_mode == NULL) printf("html home: %s\n",homedir); 45146493Sabial if (daemonize) printf("logfile: %s\n",logfile); 45246493Sabial } 45338589Sabial 45446493Sabial /* httpd is spawned */ 45546493Sabial if (daemonize) { 456113126Sdwmalone if ((server_pid = fork()) != 0) { 45746493Sabial wait3(0,WNOHANG,0); 45846493Sabial if (verbose) printf("pid: %d\n",server_pid); 45946493Sabial exit(0); 46046493Sabial } 46146493Sabial wait3(0,WNOHANG,0); 46246493Sabial } 46338589Sabial 464113126Sdwmalone if (fetch_mode == NULL) setpgrp(0,httpd_group); 46546493Sabial 46646493Sabial /* How many connections do you want? 46746493Sabial * Keep this lower than the available number of processes 46846493Sabial */ 46946493Sabial if (listen(http_sock,15) < 0) exit(1); 47046493Sabial 47138589Sabial label: 47246493Sabial wait_connection(); 47346493Sabial 47446493Sabial if (fork()) { 47546493Sabial wait3(0,WNOHANG,0); 47646493Sabial close(con_sock); 47746493Sabial goto label; 47846493Sabial } 47946493Sabial 48046493Sabial http_request(); 48146493Sabial 48246493Sabial wait3(0,WNOHANG,0); 48346493Sabial exit(0); 48438589Sabial} 48538589Sabial 48638589Sabial 487113126Sdwmalonechar * 488113126Sdwmaloneadate(void) 48938589Sabial{ 49038589Sabial static char out[50]; 491112205Sdwmalone time_t now; 49238589Sabial struct tm *t; 49338589Sabial time(&now); 49438589Sabial t = localtime(&now); 49543939Sabial sprintf(out, "%02d:%02d:%02d %02d/%02d/%02d", 49643939Sabial t->tm_hour, t->tm_min, t->tm_sec, 49743939Sabial t->tm_mday, t->tm_mon+1, t->tm_year ); 49838589Sabial return out; 49938589Sabial} 500