| /* |
| * 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. |
| * |
| * Author: Adam Dunkels <adam@sics.se> |
| * |
| * $Id: httpd.c,v 1.10 2007/06/02 07:32:06 ryohji Exp $ |
| */ |
| |
| #include "contiki.h" |
| #include "httpd.h" |
| #include "httpd-fs.h" |
| #include "httpd-cgi.h" |
| #include "psock.h" |
| #include "http-strings.h" |
| |
| #include <string.h> |
| |
| #define STATE_WAITING 0 |
| #define STATE_OUTPUT 1 |
| |
| #define SEND_STRING(s, str) PSOCK_SEND(s, str, strlen(str)) |
| MEMB(conns, sizeof(struct httpd_state), 8); |
| |
| #define ISO_nl 0x0a |
| #define ISO_space 0x20 |
| #define ISO_bang 0x21 |
| #define ISO_percent 0x25 |
| #define ISO_period 0x2e |
| #define ISO_slash 0x2f |
| #define ISO_colon 0x3a |
| |
| |
| /*---------------------------------------------------------------------------*/ |
| 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; |
| } |
| memcpy(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); |
| } |
| /*---------------------------------------------------------------------------*/ |
| static |
| PT_THREAD(send_part_of_file(struct httpd_state *s)) |
| { |
| PSOCK_BEGIN(&s->sout); |
| |
| PSOCK_SEND(&s->sout, s->file.data, s->len); |
| |
| PSOCK_END(&s->sout); |
| } |
| /*---------------------------------------------------------------------------*/ |
| static void |
| next_scriptstate(struct httpd_state *s) |
| { |
| char *p; |
| p = strchr(s->scriptptr, ISO_nl) + 1; |
| s->scriptlen -= (unsigned short)(p - s->scriptptr); |
| s->scriptptr = p; |
| } |
| /*---------------------------------------------------------------------------*/ |
| static |
| PT_THREAD(handle_script(struct httpd_state *s)) |
| { |
| char *ptr; |
| |
| PT_BEGIN(&s->scriptpt); |
| |
| |
| while(s->file.len > 0) { |
| |
| /* Check if we should start executing a script. */ |
| if(*s->file.data == ISO_percent && |
| *(s->file.data + 1) == ISO_bang) { |
| s->scriptptr = s->file.data + 3; |
| s->scriptlen = s->file.len - 3; |
| if(*(s->scriptptr - 1) == ISO_colon) { |
| httpd_fs_open(s->scriptptr + 1, &s->file); |
| PT_WAIT_THREAD(&s->scriptpt, send_file(s)); |
| } else { |
| PT_WAIT_THREAD(&s->scriptpt, |
| httpd_cgi(s->scriptptr)(s, s->scriptptr)); |
| } |
| next_scriptstate(s); |
| |
| /* The script is over, so we reset the pointers and continue |
| sending the rest of the file. */ |
| s->file.data = s->scriptptr; |
| s->file.len = s->scriptlen; |
| } else { |
| /* See if we find the start of script marker in the block of HTML |
| to be sent. */ |
| |
| if(s->file.len > uip_mss()) { |
| s->len = uip_mss(); |
| } else { |
| s->len = s->file.len; |
| } |
| |
| if(*s->file.data == ISO_percent) { |
| ptr = strchr(s->file.data + 1, ISO_percent); |
| } else { |
| ptr = strchr(s->file.data, ISO_percent); |
| } |
| if(ptr != NULL && |
| ptr != s->file.data) { |
| s->len = (int)(ptr - s->file.data); |
| if(s->len >= uip_mss()) { |
| s->len = uip_mss(); |
| } |
| } |
| PT_WAIT_THREAD(&s->scriptpt, send_part_of_file(s)); |
| s->file.data += s->len; |
| s->file.len -= s->len; |
| |
| } |
| } |
| |
| PT_END(&s->scriptpt); |
| } |
| /*---------------------------------------------------------------------------*/ |
| static |
| PT_THREAD(send_headers(struct httpd_state *s, const char *statushdr)) |
| { |
| static char *ptr; |
| |
| PSOCK_BEGIN(&s->sout); |
| |
| SEND_STRING(&s->sout, statushdr); |
| |
| ptr = strrchr(s->filename, ISO_period); |
| if(ptr == NULL) { |
| SEND_STRING(&s->sout, http_content_type_binary); |
| } else if(strncmp(http_html, ptr, 5) == 0 || |
| strncmp(http_shtml, ptr, 6) == 0) { |
| SEND_STRING(&s->sout, http_content_type_html); |
| } else if(strncmp(http_css, ptr, 4) == 0) { |
| SEND_STRING(&s->sout, http_content_type_css); |
| } else if(strncmp(http_png, ptr, 4) == 0) { |
| SEND_STRING(&s->sout, http_content_type_png); |
| } else if(strncmp(http_gif, ptr, 4) == 0) { |
| SEND_STRING(&s->sout, http_content_type_gif); |
| } else if(strncmp(http_jpg, ptr, 4) == 0) { |
| SEND_STRING(&s->sout, http_content_type_jpg); |
| } else { |
| SEND_STRING(&s->sout, http_content_type_plain); |
| } |
| PSOCK_END(&s->sout); |
| } |
| /*---------------------------------------------------------------------------*/ |
| static |
| PT_THREAD(handle_output(struct httpd_state *s)) |
| { |
| char *ptr; |
| |
| PT_BEGIN(&s->outputpt); |
| |
| if(!httpd_fs_open(s->filename, &s->file)) { |
| httpd_fs_open(http_404_html, &s->file); |
| PT_WAIT_THREAD(&s->outputpt, |
| send_headers(s, |
| http_header_404)); |
| PT_WAIT_THREAD(&s->outputpt, |
| send_file(s)); |
| } else { |
| PT_WAIT_THREAD(&s->outputpt, |
| send_headers(s, |
| http_header_200)); |
| ptr = strchr(s->filename, ISO_period); |
| if(ptr != NULL && strncmp(ptr, http_shtml, 6) == 0) { |
| PT_INIT(&s->scriptpt); |
| PT_WAIT_THREAD(&s->outputpt, handle_script(s)); |
| } else { |
| PT_WAIT_THREAD(&s->outputpt, |
| send_file(s)); |
| } |
| } |
| PSOCK_CLOSE(&s->sout); |
| PT_END(&s->outputpt); |
| } |
| /*---------------------------------------------------------------------------*/ |
| static |
| PT_THREAD(handle_input(struct httpd_state *s)) |
| { |
| PSOCK_BEGIN(&s->sin); |
| |
| PSOCK_READTO(&s->sin, ISO_space); |
| |
| |
| if(strncmp(s->inputbuf, http_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) { |
| strncpy(s->filename, http_index_html, sizeof(s->filename)); |
| } else { |
| s->inputbuf[PSOCK_DATALEN(&s->sin) - 1] = 0; |
| strncpy(s->filename, &s->inputbuf[0], sizeof(s->filename)); |
| } |
| |
| httpd_log_file(uip_conn->ripaddr, s->filename); |
| |
| s->state = STATE_OUTPUT; |
| |
| while(1) { |
| PSOCK_READTO(&s->sin, ISO_nl); |
| |
| if(strncmp(s->inputbuf, http_referer, 8) == 0) { |
| s->inputbuf[PSOCK_DATALEN(&s->sin) - 2] = 0; |
| httpd_log(&s->inputbuf[9]); |
| } |
| } |
| |
| 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, s->inputbuf, sizeof(s->inputbuf) - 1); |
| PSOCK_INIT(&s->sout, s->inputbuf, sizeof(s->inputbuf) - 1); |
| PT_INIT(&s->outputpt); |
| s->state = STATE_WAITING; |
| timer_set(&s->timer, CLOCK_SECOND * 100); |
| handle_connection(s); |
| } else if(s != NULL) { |
| if(uip_poll()) { |
| |
| if(timer_expired(&s->timer)) { |
| uip_abort(); |
| memb_free(&conns, s); |
| } |
| } |
| timer_restart(&s->timer); |
| handle_connection(s); |
| } else { |
| uip_abort(); |
| } |
| } |
| /*---------------------------------------------------------------------------*/ |
| void |
| httpd_init(void) |
| { |
| tcp_listen(HTONS(80)); |
| memb_init(&conns); |
| } |
| /*---------------------------------------------------------------------------*/ |