osd-contiki/apps/webserver-nano/httpd.c
2011-07-24 12:04:41 -04:00

505 lines
16 KiB
C

/*
* Copyright (c) 2004, Adam Dunkels.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the Institute nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
* This file is part of the Contiki operating system.
*
*/
/* Configurable Contiki Webserver - see httpd.h */
#include <string.h>
#include <stdio.h>
#include "contiki-net.h"
#include "httpd.h"
#include "webserver.h"
#include "httpd-fs.h"
#if WEBSERVER_CONF_CGI
#include "httpd-cgi.h"
#endif
#include "lib/petsciiconv.h"
#if COFFEE_FILES
#include "cfs-coffee-arch.h"
#endif
#define DEBUG 0
#if DEBUG
#define PRINTD PRINTA
#else
#define PRINTD(...)
#endif
#define STATE_WAITING 0
#define STATE_OUTPUT 1
/* Allocate memory for the tcp connections */
MEMB(conns, struct httpd_state, WEBSERVER_CONF_CONNS);
#define ISO_tab 0x09
#define ISO_nl 0x0a
#define ISO_cr 0x0d
#define ISO_space 0x20
#define ISO_bang 0x21
#define ISO_percent 0x25
#define ISO_period 0x2e
#define ISO_slash 0x2f
#define ISO_colon 0x3a
#define ISO_qmark 0x3f
/*---------------------------------------------------------------------------*/
static unsigned short
generate(void *state)
{
struct httpd_state *s = (struct httpd_state *)state;
if(s->file.len > uip_mss()) {
s->len = uip_mss();
} else {
s->len = s->file.len;
}
httpd_fs_cpy(uip_appdata, s->file.data, s->len);
return s->len;
}
/*---------------------------------------------------------------------------*/
static
PT_THREAD(send_file(struct httpd_state *s))
{
PSOCK_BEGIN(&s->sout);
do {
PSOCK_GENERATOR_SEND(&s->sout, generate, s);
s->file.len -= s->len;
s->file.data += s->len;
} while(s->file.len > 0);
PSOCK_END(&s->sout);
}
/*---------------------------------------------------------------------------*/
#if WEBSERVER_CONF_INCLUDE || WEBSERVER_CONF_CGI
static
PT_THREAD(send_part_of_file(struct httpd_state *s))
{
PSOCK_BEGIN(&s->sout);
static int oldfilelen, oldlen;
static char * olddata;
//Store stuff that gets clobbered...
oldfilelen = s->file.len;
oldlen = s->len;
olddata = s->file.data;
s->file.len = s->len;
do {
PSOCK_GENERATOR_SEND(&s->sout, generate, s);
s->file.len -= s->len;
s->file.data += s->len;
} while(s->file.len > 0);
s->len = oldlen;
s->file.len = oldfilelen;
s->file.data = olddata;
PSOCK_END(&s->sout);
}
/*---------------------------------------------------------------------------*/
static void
next_scriptstate(struct httpd_state *s)
{
char *p;
/* Skip over any script parameters to the beginning of the next line */
if((p = (char *)httpd_fs_strchr(s->scriptptr, ISO_nl)) != NULL) {
p += 1;
s->scriptlen -= (unsigned short)(p - s->scriptptr);
s->scriptptr = p;
} else {
s->scriptlen = 0;
}
}
/*---------------------------------------------------------------------------*/
char *
get_scriptname(char *dest, char *fromfile)
{
uint8_t i=0,skip=1;
/* Extract a file or cgi name, trim leading spaces and replace termination with zero */
/* Returns number of characters processed up to the next non-tab or space */
do {
dest[i]=httpd_fs_getchar(fromfile++);
if (dest[i]==ISO_colon) {if (!skip) break;} //allow leading colons
else if (dest[i]==ISO_tab ) {if (skip) continue;else break;}//skip leading tabs
else if (dest[i]==ISO_space) {if (skip) continue;else break;}//skip leading spaces
else if (dest[i]==ISO_nl ) break; //nl is preferred delimiter
else if (dest[i]==ISO_cr ) break; //some editors insert cr
else if (dest[i]==0 ) break; //files are terminated with null
else skip=0;
i++;
} while (i<(MAX_SCRIPT_NAME_LENGTH+1));
fromfile--;
while ((dest[i]==ISO_space) || (dest[i]==ISO_tab)) dest[i]=httpd_fs_getchar(++fromfile);
dest[i]=0;
return (fromfile);
}
/*---------------------------------------------------------------------------*/
static
PT_THREAD(handle_script(struct httpd_state *s))
{
/* Note script includes will attach a leading : to the filename and a trailing zero */
static char scriptname[MAX_SCRIPT_NAME_LENGTH+1],*pptr;
static uint16_t filelength;
PT_BEGIN(&s->scriptpt);
filelength=s->file.len;
while(s->file.len > 0) {
/* Sanity check */
if (s->file.len > filelength) break;
/* Check if we should start executing a script, flagged by %! */
if(httpd_fs_getchar(s->file.data) == ISO_percent &&
httpd_fs_getchar(s->file.data + 1) == ISO_bang) {
/* Extract name, if starts with colon include file else call cgi */
s->scriptptr=get_scriptname(scriptname,s->file.data+2);
s->scriptlen=s->file.len-(s->scriptptr-s->file.data);
PRINTD("httpd: Handle script named %s\n",scriptname);
if(scriptname[0] == ISO_colon) {
#if WEBSERVER_CONF_INCLUDE
if (httpd_fs_open(&scriptname[1], &s->file)) {
PT_WAIT_THREAD(&s->scriptpt, send_file(s));
}
/*TODO dont print anything if file not found */
#endif
} else {
#if WEBSERVER_CONF_CGI
PT_WAIT_THREAD(&s->scriptpt,httpd_cgi(scriptname)(s, s->scriptptr));
#endif
}
next_scriptstate(s);
/* Reset the pointers and continue sending the current file. */
s->file.data = s->scriptptr;
s->file.len = s->scriptlen;
} else {
/* Send file up to the next potential script */
if(s->file.len > uip_mss()) {
s->len = uip_mss();
} else {
s->len = s->file.len;
}
if(httpd_fs_getchar(s->file.data) == ISO_percent) {
pptr = (char *) httpd_fs_strchr(s->file.data + 1, ISO_percent);
} else {
pptr = (char *) httpd_fs_strchr(s->file.data, ISO_percent);
}
if(pptr != NULL && pptr != s->file.data) {
s->len = (int)(pptr - s->file.data);
if(s->len >= uip_mss()) {
s->len = uip_mss();
}
}
PRINTD("httpd: Sending %u bytes from 0x%04x\n",s->file.len,(unsigned int)s->file.data);
PT_WAIT_THREAD(&s->scriptpt, send_part_of_file(s));
s->file.data += s->len;
s->file.len -= s->len;
}
}
PT_END(&s->scriptpt);
}
#endif /* WEBSERVER_CONF_INCLUDE || WEBSERVER_CONF_CGI */
/*---------------------------------------------------------------------------*/
const char httpd_http[] HTTPD_STRING_ATTR = "HTTP/1.0 ";
const char httpd_server[] HTTPD_STRING_ATTR = "\r\nServer: Contiki/2.0 http://www.sics.se/contiki/\r\nConnection: close\r\n";
static unsigned short
generate_status(void *sstr)
{
uint8_t slen=httpd_strlen((char *)sstr);
httpd_memcpy(uip_appdata, httpd_http, sizeof(httpd_http)-1);
httpd_memcpy(uip_appdata+sizeof(httpd_http)-1, (char *)sstr, slen);
slen+=sizeof(httpd_http)-1;
httpd_memcpy(uip_appdata+slen, httpd_server, sizeof(httpd_server)-1);
return slen+sizeof(httpd_server)-1;
}
/*---------------------------------------------------------------------------*/
const char httpd_content[] HTTPD_STRING_ATTR = "Content-type: ";
const char httpd_crlf[] HTTPD_STRING_ATTR = "\r\n\r\n";
static unsigned short
generate_header(void *hstr)
{
uint8_t slen=httpd_strlen((char *)hstr);
httpd_memcpy(uip_appdata,httpd_content,sizeof(httpd_content)-1);
httpd_memcpy(uip_appdata+sizeof(httpd_content)-1, (char *)hstr, slen);
slen+=sizeof(httpd_content)-1;
httpd_memcpy(uip_appdata+slen,httpd_crlf,sizeof(httpd_crlf)-1);
return slen+4;
}
/*---------------------------------------------------------------------------*/
const char httpd_mime_htm[] HTTPD_STRING_ATTR = "text/html";
#if WEBSERVER_CONF_CSS
const char httpd_mime_css[] HTTPD_STRING_ATTR = "text/css";
#endif
#if WEBSERVER_CONF_PNG
const char httpd_mime_png[] HTTPD_STRING_ATTR = "image/png";
#endif
#if WEBSERVER_CONF_GIF
const char httpd_mime_gif[] HTTPD_STRING_ATTR = "image/gif";
#endif
#if WEBSERVER_CONF_JPG
const char httpd_mime_jpg[] HTTPD_STRING_ATTR = "image/jpeg";
const char httpd_jpg [] HTTPD_STRING_ATTR = "jpg";
#endif
#if WEBSERVER_CONF_TXT
const char httpd_mime_txt[] HTTPD_STRING_ATTR = "text/plain";
#endif
#if WEBSERVER_CONF_BIN
const char httpd_mime_bin[] HTTPD_STRING_ATTR = "application/octet-stream";
#endif
#if WEBSERVER_CONF_INCLUDE || WEBSERVER_CONF_CGI
const char httpd_shtml [] HTTPD_STRING_ATTR = ".shtml";
#endif
static
PT_THREAD(send_headers(struct httpd_state *s, const char *statushdr))
{
char *ptr;
PSOCK_BEGIN(&s->sout);
PSOCK_GENERATOR_SEND(&s->sout, generate_status, (char *)statushdr);
ptr = strrchr(s->filename, ISO_period);
if (httpd_strncmp("4", statushdr, 1)==0) { //404
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_htm );
} else if(ptr == NULL) {
#if WEBSERVER_CONF_BIN
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_bin );
#else
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_htm );
#endif
} else {
ptr++;
#if WEBSERVER_CONF_INCLUDE || WEBSERVER_CONF_CGI
if(httpd_strncmp(ptr, &httpd_mime_htm[5],3)== 0 ||httpd_strncmp(ptr, &httpd_shtml[1], 4) == 0) {
#else
if(httpd_strncmp(ptr, &httpd_mime_htm[5],3)== 0) {
#endif
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_htm );
#if WEBSEVER_CONF_CSS
} else if(httpd_strcmp(ptr, &httpd_mime_css[5]) == 0) {
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_css );
#endif
#if WEBSERVER_CONF_PNG
} else if(httpd_strcmp(ptr, &httpd_mime_png[6]) == 0) {
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_png );
#endif
#if WEBSERVER_CONF_GIF
} else if(httpd_strcmp(ptr, &httpd_mime_gif[6])== 0) {
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_gif );
#endif
#if WEBSERVER_CONF_JPG
} else if(httpd_strcmp(ptr, httpd_mime_jpg) == 0) {
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_jpg );
#endif
#if WEBSERVER_CONF_TXT
} else {
PSOCK_GENERATOR_SEND(&s->sout, generate_header, &httpd_mime_txt);
#endif
}
}
PSOCK_END(&s->sout);
}
/*---------------------------------------------------------------------------*/
const char httpd_indexfn [] HTTPD_STRING_ATTR = "/index.html";
#if WEBSERVER_CONF_INCLUDE || WEBSERVER_CONF_CGI
const char httpd_indexsfn [] HTTPD_STRING_ATTR = "/index.shtml";
#endif
const char httpd_404fn [] HTTPD_STRING_ATTR = "/404.html";
const char httpd_404notf [] HTTPD_STRING_ATTR = "404 Not found";
const char httpd_200ok [] HTTPD_STRING_ATTR = "200 OK";
static
PT_THREAD(handle_output(struct httpd_state *s))
{
char *ptr;
PT_BEGIN(&s->outputpt);
if(!httpd_fs_open(s->filename, &s->file)) {
#if WEBSERVER_CONF_INCLUDE || WEBSERVER_CONF_CGI
/* If index.html not found try index.shtml */
if (httpd_strcmp(s->filename,httpd_indexfn)==0) httpd_strcpy(s->filename,httpd_indexsfn);
if (httpd_fs_open(s->filename, &s->file)) goto sendfile;
#endif
httpd_strcpy(s->filename, httpd_404fn);
httpd_fs_open(s->filename, &s->file);
PT_WAIT_THREAD(&s->outputpt, send_headers(s, httpd_404notf));
PT_WAIT_THREAD(&s->outputpt, send_file(s));
} else {
sendfile:
PT_WAIT_THREAD(&s->outputpt, send_headers(s, httpd_200ok));
#if WEBSERVER_CONF_INCLUDE || WEBSERVER_CONF_CGI
ptr = strchr(s->filename, ISO_period);
if((ptr != NULL && httpd_strncmp(ptr, httpd_shtml, 6) == 0) || httpd_strcmp(s->filename,httpd_indexfn)==0) {
PT_INIT(&s->scriptpt);
PT_WAIT_THREAD(&s->outputpt, handle_script(s));
} else {
#else
if (1)
#endif
PT_WAIT_THREAD(&s->outputpt, send_file(s));
}
}
PSOCK_CLOSE(&s->sout);
PT_END(&s->outputpt);
}
/*---------------------------------------------------------------------------*/
#if WEBSERVER_CONF_PASSQUERY
char httpd_query[WEBSERVER_CONF_PASSQUERY];
#endif
const char httpd_get[] HTTPD_STRING_ATTR = "GET ";
const char httpd_ref[] HTTPD_STRING_ATTR = "Referer:";
static
PT_THREAD(handle_input(struct httpd_state *s))
{
PSOCK_BEGIN(&s->sin);
PSOCK_READTO(&s->sin, ISO_space);
if(httpd_strncmp(s->inputbuf, httpd_get, 4) != 0) {
PSOCK_CLOSE_EXIT(&s->sin);
}
PSOCK_READTO(&s->sin, ISO_space);
if(s->inputbuf[0] != ISO_slash) {
PSOCK_CLOSE_EXIT(&s->sin);
}
if(s->inputbuf[1] == ISO_space) {
httpd_strcpy(s->filename, httpd_indexfn);
} else {
s->inputbuf[PSOCK_DATALEN(&s->sin) - 1] = 0;
strncpy(s->filename, &s->inputbuf[0], sizeof(s->filename));
#if WEBSERVER_CONF_PASSQUERY
/* Query string is left in the buffer until zeroed by the application! */
{uint8_t i;
for (i=0;i<sizeof(s->filename)+1;i++) {
if (s->inputbuf[i]==ISO_space) break;
if (s->inputbuf[i]==ISO_qmark) {
s->inputbuf[i]=0;
strncpy(httpd_query,&s->inputbuf[i+1],sizeof(httpd_query));
// raven_lcd_show_text(&s->inputbuf[i]);
}
}
}
#endif
}
#if WEBSERVER_CONF_LOG
webserver_log_file(&uip_conn->ripaddr, s->filename);
// webserver_log(httpd_query);
#endif
s->state = STATE_OUTPUT;
while(1) {
PSOCK_READTO(&s->sin, ISO_nl);
#if WEBSERVER_CONF_LOG && WEBSERVER_CONF_REFERER
if(httpd_strncmp(s->inputbuf, httpd_ref, 8) == 0) {
s->inputbuf[PSOCK_DATALEN(&s->sin) - 2] = 0;
petsciiconv_topetscii(s->inputbuf, PSOCK_DATALEN(&s->sin) - 2);
webserver_log(s->inputbuf);
}
#endif
}
PSOCK_END(&s->sin);
}
/*---------------------------------------------------------------------------*/
static void
handle_connection(struct httpd_state *s)
{
handle_input(s);
if(s->state == STATE_OUTPUT) {
handle_output(s);
}
}
/*---------------------------------------------------------------------------*/
void
httpd_appcall(void *state)
{
struct httpd_state *s = (struct httpd_state *)state;
if(uip_closed() || uip_aborted() || uip_timedout()) {
if(s != NULL) {
memb_free(&conns, s);
}
} else if(uip_connected()) {
s = (struct httpd_state *)memb_alloc(&conns);
if(s == NULL) {
uip_abort();
return;
}
tcp_markconn(uip_conn, s);
PSOCK_INIT(&s->sin, (uint8_t *)s->inputbuf, sizeof(s->inputbuf) - 1);
PSOCK_INIT(&s->sout, (uint8_t *)s->inputbuf, sizeof(s->inputbuf) - 1);
PT_INIT(&s->outputpt);
s->state = STATE_WAITING;
s->timer = 0;
handle_connection(s);
} else if(s != NULL) {
if(uip_poll()) {
++s->timer;
if(s->timer >= 20) {
uip_abort();
memb_free(&conns, s);
}
} else {
s->timer = 0;
}
handle_connection(s);
} else {
uip_abort();
}
}
/*---------------------------------------------------------------------------*/
void
httpd_init(void)
{
tcp_listen(UIP_HTONS(80));
memb_init(&conns);
PRINTD(" sizof(struct httpd_state) = %d\n",sizeof(struct httpd_state));
PRINTA(" %d bytes used for httpd state storage\n",conns.size * conns.num);
#if WEBSERVER_CONF_CGI
httpd_cgi_init();
#endif
}
/*---------------------------------------------------------------------------*/