Demos/Device/ClassDriver/RNDISEthernet/Lib/Webserver.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2009.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6       www.fourwalledcubicle.com
   7 */
   8
   9 /*
  10   Copyright 2009  Dean Camera (dean [at] fourwalledcubicle [dot] com)
  11
  12   Permission to use, copy, modify, and distribute this software
  13   and its documentation for any purpose and without fee is hereby
  14   granted, provided that the above copyright notice appear in all
  15   copies and that both that the copyright notice and this
  16   permission notice and warranty disclaimer appear in supporting
  17   documentation, and that the name of the author not be used in
  18   advertising or publicity pertaining to distribution of the
  19   software without specific, written prior permission.
  20
  21   The author disclaim all warranties with regard to this
  22   software, including all implied warranties of merchantability
  23   and fitness.  In no event shall the author be liable for any
  24   special, indirect or consequential damages or any damages
  25   whatsoever resulting from loss of use, data or profits, whether
  26   in an action of contract, negligence or other tortious action,
  27   arising out of or in connection with the use or performance of
  28   this software.
  29 */
  30
  31 /** \file
  32  *
  33  *  Simple webserver application for demonstrating the RNDIS demo and TCP/IP stack. This
  34  *  application will serve up a static HTTP webpage when requested by the host.
  35  */
  36
  37 #include "Webserver.h"
  38
  39 /** HTTP server response header, for transmission before the page contents. This indicates to the host that a page exists at the
  40  *  given location, and gives extra connection information.
  41  */
  42 char PROGMEM HTTP200Header[] = "HTTP/1.1 200 OK\r\n"
  43                                "Server: LUFA RNDIS\r\n"
  44                                "Content-type: text/html\r\n"
  45                                "Connection: close\r\n\r\n";
  46
  47 char PROGMEM HTTP404Header[] = "HTTP/1.1 404 Not Found\r\n"
  48                                "Server: LUFA RNDIS\r\n"
  49                                "Connection: close\r\n\r\n";
  50
  51 /** HTTP page to serve to the host when a HTTP request is made. This page is too long for a single response, thus it is automatically
  52  *  broken up into smaller blocks and sent as a series of packets each time the webserver application callback is run.
  53  */
  54 char PROGMEM HTTPPage[]   =
  55                 "<html>"
  56                 "       <head>"
  57                 "               <title>"
  58                 "                       LUFA Webserver Demo"
  59                 "               </title>"
  60                 "       </head>"
  61                 "       <body>"
  62                 "               <h1>Hello from your USB AVR!</h1>"
  63                 "               <p>"
  64                 "                       Hello! Welcome to the LUFA RNDIS Demo Webserver test page, running on your USB AVR via the LUFA library. This demonstrates the HTTP webserver, TCP/IP stack and RNDIS demo all running atop the LUFA USB stack."
  65                 "                       <br /><br />"
  66                 "                       <small>Project Information: <a href=\"http://www.fourwalledcubicle.com/LUFA.php\">http://www.fourwalledcubicle.com/LUFA.php</a>.</small>"
  67                 "                       <hr />"
  68                 "                       <i>LUFA Version: </i>" LUFA_VERSION_STRING
  69                 "               </p>"
  70                 "       </body>"
  71                 "</html>";
  72
  73
  74 /** Initializes the Webserver application, opening the appropriate HTTP port in the TCP handler and registering the application
  75  *  callback routine for packets sent to the HTTP protocol port.
  76  */
  77 void Webserver_Init(void)
  78 {
  79         /* Open the HTTP port in the TCP protocol so that HTTP connections to the device can be established */
  80         TCP_SetPortState(TCP_PORT_HTTP, TCP_Port_Open, Webserver_ApplicationCallback);
  81 }
  82
  83 /** Indicates if a given request equals the given HTTP command.
  84  *
  85  *  \param[in] RequestHeader  HTTP request made by the host
  86  *  \param[in] Command        HTTP command to compare the request to
  87  *
  88  *  \return Boolean true if the command matches the request, false otherwise
  89  */
  90 static bool IsHTTPCommand(uint8_t* RequestHeader, char* Command)
  91 {
  92         /* Returns true if the non null terminated string in RequestHeader matches the null terminated string Command */
  93         return (strncmp((char*)RequestHeader, Command, strlen(Command)) == 0);
  94 }
  95
  96 /** Application callback routine, executed each time the TCP processing task runs. This callback determines what request
  97  *  has been made (if any), and serves up appropriate responses.
  98  *
  99  *  \param[in] ConnectionState  Pointer to a TCP Connection State structure giving connection information
 100  *  \param[in,out] Buffer       Pointer to the application's send/receive packet buffer
 101  */
 102 void Webserver_ApplicationCallback(TCP_ConnectionState_t* ConnectionState, TCP_ConnectionBuffer_t* Buffer)
 103 {
 104         char*          BufferDataStr = (char*)Buffer->Data;
 105         static uint8_t PageBlock     = 0;
 106
 107         /* Check to see if a packet has been received on the HTTP port from a remote host */
 108         if (TCP_APP_HAS_RECEIVED_PACKET(Buffer))
 109         {
 110                 if (IsHTTPCommand(Buffer->Data, "GET"))
 111                 {
 112                         if (IsHTTPCommand(Buffer->Data, "GET / "))
 113                         {
 114                                 PageBlock = 0;
 115
 116                                 /* Copy the HTTP 200 response header into the packet buffer */
 117                                 strcpy_P(BufferDataStr, HTTP200Header);
 118
 119                                 /* Send the buffer contents to the host */
 120                                 TCP_APP_SEND_BUFFER(Buffer, strlen(BufferDataStr));
 121
 122                                 /* Lock the buffer to Device->Host transmissions only while we send the page contents */
 123                                 TCP_APP_CAPTURE_BUFFER(Buffer);
 124                         }
 125                         else
 126                         {
 127                                 /* Copy the HTTP 404 response header into the packet buffer */
 128                                 strcpy_P(BufferDataStr, HTTP404Header);
 129
 130                                 /* Send the buffer contents to the host */
 131                                 TCP_APP_SEND_BUFFER(Buffer, strlen(BufferDataStr));
 132
 133                                 /* All data sent, close the connection */
 134                                 TCP_APP_CLOSECONNECTION(ConnectionState);
 135                         }
 136                 }
 137                 else if (IsHTTPCommand(Buffer->Data, "HEAD"))
 138                 {
 139                         if (IsHTTPCommand(Buffer->Data, "HEAD / "))
 140                         {
 141                                 /* Copy the HTTP response header into the packet buffer */
 142                                 strcpy_P(BufferDataStr, HTTP200Header);
 143
 144                                 /* Send the buffer contents to the host */
 145                                 TCP_APP_SEND_BUFFER(Buffer, strlen(BufferDataStr));
 146                         }
 147                         else
 148                         {
 149                                 /* Copy the HTTP response header into the packet buffer */
 150                                 strcpy_P(BufferDataStr, HTTP404Header);
 151
 152                                 /* Send the buffer contents to the host */
 153                                 TCP_APP_SEND_BUFFER(Buffer, strlen(BufferDataStr));
 154                         }
 155
 156                         /* All data sent, close the connection */
 157                         TCP_APP_CLOSECONNECTION(ConnectionState);
 158                 }
 159                 else if (IsHTTPCommand(Buffer->Data, "TRACE"))
 160                 {
 161                         /* Echo the host's query back to the host */
 162                         TCP_APP_SEND_BUFFER(Buffer, Buffer->Length);
 163
 164                         /* All data sent, close the connection */
 165                         TCP_APP_CLOSECONNECTION(ConnectionState);
 166                 }
 167                 else
 168                 {
 169                         /* Unknown request, just clear the buffer (drop the packet) */
 170                         TCP_APP_CLEAR_BUFFER(Buffer);
 171                 }
 172         }
 173         else if (TCP_APP_HAVE_CAPTURED_BUFFER(Buffer))
 174         {
 175                 uint16_t RemLength = strlen_P(&HTTPPage[PageBlock * HTTP_REPLY_BLOCK_SIZE]);
 176                 uint16_t Length;
 177
 178                 /* Determine the length of the loaded block */
 179                 Length = ((RemLength > HTTP_REPLY_BLOCK_SIZE) ? HTTP_REPLY_BLOCK_SIZE : RemLength);
 180
 181                 /* Copy the next buffer sized block of the page to the packet buffer */
 182                 strncpy_P(BufferDataStr, &HTTPPage[PageBlock * HTTP_REPLY_BLOCK_SIZE], Length);
 183
 184                 /* Send the buffer contents to the host */
 185                 TCP_APP_SEND_BUFFER(Buffer, Length);
 186
 187                 /* Check to see if the entire page has been sent */
 188                 if (PageBlock++ == (sizeof(HTTPPage) / HTTP_REPLY_BLOCK_SIZE))
 189                 {
 190                         /* Unlock the buffer so that the host can fill it with future packets */
 191                         TCP_APP_RELEASE_BUFFER(Buffer);
 192
 193                         /* Close the connection to the host */
 194                         TCP_APP_CLOSECONNECTION(ConnectionState);
 195                 }
 196         }
 197 }