Projects/Incomplete/Webserver/Webserver.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2010.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6       www.fourwalledcubicle.com
   7 */
   8
   9 /*
  10   Copyright 2010  Dean Camera (dean [at] fourwalledcubicle [dot] com)
  11
  12   Permission to use, copy, modify, distribute, and sell this
  13   software and its documentation for any purpose is hereby granted
  14   without fee, provided that the above copyright notice appear in
  15   all 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  *  Main source file for the Webserver project. This file contains the main tasks of
  34  *  the demo and is responsible for the initial application hardware configuration.
  35  */
  36
  37 #include "Webserver.h"
  38
  39 /** LUFA RNDIS Class driver interface configuration and state information. This structure is
  40  *  passed to all RNDIS Class driver functions, so that multiple instances of the same class
  41  *  within a device can be differentiated from one another.
  42  */
  43 USB_ClassInfo_RNDIS_Host_t Ethernet_RNDIS_Interface =
  44         {
  45                 .Config =
  46                         {
  47                                 .DataINPipeNumber           = 1,
  48                                 .DataINPipeDoubleBank       = false,
  49
  50                                 .DataOUTPipeNumber          = 2,
  51                                 .DataOUTPipeDoubleBank      = false,
  52
  53                                 .NotificationPipeNumber     = 3,
  54                                 .NotificationPipeDoubleBank = false,
  55
  56                                 .HostMaxPacketSize          = UIP_CONF_BUFFER_SIZE,
  57                         },
  58         };
  59
  60 struct timer ConnectionTimer, ARPTimer;
  61 uint16_t MillisecondTickCount;
  62
  63 /** ISR for the management of the connection management timeout counter */
  64 ISR(TIMER0_COMPA_vect, ISR_BLOCK)
  65 {
  66         MillisecondTickCount++;
  67 }
  68
  69 void TCPCallback(void)
  70 {
  71         printf("Callback!\r\n");
  72 }
  73
  74 /** Main program entry point. This routine configures the hardware required by the application, then
  75  *  enters a loop to run the application tasks in sequence.
  76  */
  77 int main(void)
  78 {
  79         SetupHardware();
  80
  81         puts_P(PSTR(ESC_FG_CYAN "RNDIS Host Demo running.\r\n" ESC_FG_WHITE));
  82
  83         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
  84
  85         for (;;)
  86         {
  87                 switch (USB_HostState)
  88                 {
  89                         case HOST_STATE_Addressed:
  90                                 LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
  91
  92                                 uint16_t ConfigDescriptorSize;
  93                                 uint8_t  ConfigDescriptorData[512];
  94
  95                                 if (USB_Host_GetDeviceConfigDescriptor(1, &ConfigDescriptorSize, ConfigDescriptorData,
  96                                                                        sizeof(ConfigDescriptorData)) != HOST_GETCONFIG_Successful)
  97                                 {
  98                                         printf("Error Retrieving Configuration Descriptor.\r\n");
  99                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 100                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 101                                         break;
 102                                 }
 103
 104                                 if (RNDIS_Host_ConfigurePipes(&Ethernet_RNDIS_Interface,
 105                                                               ConfigDescriptorSize, ConfigDescriptorData) != RNDIS_ENUMERROR_NoError)
 106                                 {
 107                                         printf("Attached Device Not a Valid RNDIS Class Device.\r\n");
 108                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 109                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 110                                         break;
 111                                 }
 112
 113                                 if (USB_Host_SetDeviceConfiguration(1) != HOST_SENDCONTROL_Successful)
 114                                 {
 115                                         printf("Error Setting Device Configuration.\r\n");
 116                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 117                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 118                                         break;
 119                                 }
 120
 121                                 if (RNDIS_Host_InitializeDevice(&Ethernet_RNDIS_Interface) != HOST_SENDCONTROL_Successful)
 122                                 {
 123                                         printf("Error Initializing Device.\r\n");
 124
 125                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 126                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 127                                         break;
 128                                 }
 129
 130                                 printf("Device Max Transfer Size: %lu bytes.\r\n", Ethernet_RNDIS_Interface.State.DeviceMaxPacketSize);
 131
 132                                 uint32_t PacketFilter = (REMOTE_NDIS_PACKET_DIRECTED | REMOTE_NDIS_PACKET_BROADCAST | REMOTE_NDIS_PACKET_ALL_MULTICAST);
 133                                 if (RNDIS_Host_SetRNDISProperty(&Ethernet_RNDIS_Interface, OID_GEN_CURRENT_PACKET_FILTER,
 134                                                                 &PacketFilter, sizeof(PacketFilter)) != HOST_SENDCONTROL_Successful)
 135                                 {
 136                                         printf("Error Setting Device Packet Filter.\r\n");
 137
 138                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 139                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 140                                         break;
 141                                 }
 142
 143                                 struct uip_eth_addr MACAddress;
 144                                 if (RNDIS_Host_QueryRNDISProperty(&Ethernet_RNDIS_Interface, OID_802_3_CURRENT_ADDRESS,
 145                                                                   &MACAddress, sizeof(MACAddress)) != HOST_SENDCONTROL_Successful)
 146                                 {
 147                                         printf("Error Getting MAC Address.\r\n");
 148
 149                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 150                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 151                                         break;
 152                                 }
 153
 154                                 printf("MAC Address: 0x%02X 0x%02X 0x%02X 0x%02X 0x%02X 0x%02X\r\n",
 155                                        MACAddress.addr[0], MACAddress.addr[1], MACAddress.addr[2],
 156                                        MACAddress.addr[3], MACAddress.addr[4], MACAddress.addr[5]);
 157
 158                                 uip_setethaddr(MACAddress);
 159
 160                                 printf("RNDIS Device Enumerated.\r\n");
 161                                 USB_HostState = HOST_STATE_Configured;
 162                                 break;
 163                         case HOST_STATE_Configured:
 164                                 ProcessIncommingPacket();
 165                                 ManageConnections();
 166
 167                                 break;
 168                 }
 169
 170                 RNDIS_Host_USBTask(&Ethernet_RNDIS_Interface);
 171                 USB_USBTask();
 172         }
 173 }
 174
 175 void ProcessIncommingPacket(void)
 176 {
 177         if (RNDIS_Host_IsPacketReceived(&Ethernet_RNDIS_Interface))
 178         {
 179                 LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
 180
 181                 /* Read the incomming packet straight into the UIP packet buffer */
 182                 RNDIS_Host_ReadPacket(&Ethernet_RNDIS_Interface, uip_buf, &uip_len);
 183
 184                 printf("RECEIVED PACKET (%d):\r\n", uip_len);
 185                 for (uint16_t i = 0; i < uip_len; i++)
 186                   printf("0x%02X ", uip_buf[i]);
 187                 printf("\r\n\r\n");
 188
 189                 struct uip_eth_hdr* EthernetHeader = (struct uip_eth_hdr*)&uip_buf[0];
 190                 if (EthernetHeader->type == HTONS(UIP_ETHTYPE_IP))
 191                 {
 192                         /* Filter packet by MAC destination */
 193                         uip_arp_ipin();
 194
 195                         /* Process incomming packet */
 196                         uip_input();
 197
 198                         /* Add destination MAC to outgoing packet */
 199                         if (uip_len > 0)
 200                           uip_arp_out();
 201                 }
 202                 else if (EthernetHeader->type == HTONS(UIP_ETHTYPE_ARP))
 203                 {
 204                         /* Process ARP packet */
 205                         uip_arp_arpin();
 206                 }
 207
 208                 /* If a response was generated, send it */
 209                 if (uip_len > 0)
 210                   RNDIS_Host_SendPacket(&Ethernet_RNDIS_Interface, uip_buf, uip_len);
 211
 212                 printf("SENT PACKET (%d):\r\n", uip_len);
 213                 for (uint16_t i = 0; i < uip_len; i++)
 214                   printf("0x%02X ", uip_buf[i]);
 215                 printf("\r\n\r\n");
 216
 217                 LEDs_SetAllLEDs(LEDMASK_USB_READY);
 218         }
 219 }
 220
 221 void ManageConnections(void)
 222 {
 223         /* Manage open connections */
 224         if (timer_expired(&ConnectionTimer))
 225         {
 226                 timer_reset(&ConnectionTimer);
 227
 228                 LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
 229
 230                 for (uint8_t i = 0; i < UIP_CONNS; i++)
 231                 {
 232                         /* Run periodic connection management for each connection */
 233                         uip_periodic(i);
 234
 235                         /* If a response was generated, send it */
 236                         if (uip_len > 0)
 237                           RNDIS_Host_SendPacket(&Ethernet_RNDIS_Interface, uip_buf, uip_len);
 238                 }
 239
 240                 LEDs_SetAllLEDs(LEDMASK_USB_READY);
 241         }
 242
 243         /* Manage ARP cache refreshing */
 244         if (timer_expired(&ARPTimer))
 245         {
 246                 timer_reset(&ARPTimer);
 247                 uip_arp_timer();
 248         }
 249 }
 250
 251 /** Configures the board hardware and chip peripherals for the demo's functionality. */
 252 void SetupHardware(void)
 253 {
 254         /* Disable watchdog if enabled by bootloader/fuses */
 255         MCUSR &= ~(1 << WDRF);
 256         wdt_disable();
 257
 258         /* Disable clock division */
 259         clock_prescale_set(clock_div_1);
 260
 261         /* Hardware Initialization */
 262         SerialStream_Init(9600, false);
 263         LEDs_Init();
 264         USB_Init();
 265
 266         /* uIP Timing Initialization */
 267         clock_init();
 268         timer_set(&ConnectionTimer, CLOCK_SECOND / 2);
 269         timer_set(&ARPTimer, CLOCK_SECOND * 10);
 270
 271         /* uIP Stack Initialization */
 272         uip_init();
 273         uip_ipaddr_t IPAddress, Netmask, GatewayIPAddress;
 274         uip_ipaddr(&IPAddress,        192, 168, 1, 10);
 275         uip_ipaddr(&Netmask,          255, 255, 255, 0);
 276         uip_ipaddr(&GatewayIPAddress, 192, 168, 1, 1);
 277         uip_sethostaddr(&IPAddress);
 278         uip_setnetmask(&Netmask);
 279         uip_setdraddr(&GatewayIPAddress);
 280
 281         /* HTTP Webserver Initialization */
 282         uip_listen(HTONS(80));
 283 }
 284
 285 /** Event handler for the USB_DeviceAttached event. This indicates that a device has been attached to the host, and
 286  *  starts the library USB task to begin the enumeration and USB management process.
 287  */
 288 void EVENT_USB_Host_DeviceAttached(void)
 289 {
 290         puts_P(PSTR("Device Attached.\r\n"));
 291         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 292 }
 293
 294 /** Event handler for the USB_DeviceUnattached event. This indicates that a device has been removed from the host, and
 295  *  stops the library USB task management process.
 296  */
 297 void EVENT_USB_Host_DeviceUnattached(void)
 298 {
 299         puts_P(PSTR("\r\nDevice Unattached.\r\n"));
 300         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 301 }
 302
 303 /** Event handler for the USB_DeviceEnumerationComplete event. This indicates that a device has been successfully
 304  *  enumerated by the host and is now ready to be used by the application.
 305  */
 306 void EVENT_USB_Host_DeviceEnumerationComplete(void)
 307 {
 308         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 309 }
 310
 311 /** Event handler for the USB_HostError event. This indicates that a hardware error occurred while in host mode. */
 312 void EVENT_USB_Host_HostError(const uint8_t ErrorCode)
 313 {
 314         USB_ShutDown();
 315
 316         printf_P(PSTR(ESC_FG_RED "Host Mode Error\r\n"
 317                                  " -- Error Code %d\r\n" ESC_FG_WHITE), ErrorCode);
 318
 319         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 320         for(;;);
 321 }
 322
 323 /** Event handler for the USB_DeviceEnumerationFailed event. This indicates that a problem occurred while
 324  *  enumerating an attached USB device.
 325  */
 326 void EVENT_USB_Host_DeviceEnumerationFailed(const uint8_t ErrorCode, const uint8_t SubErrorCode)
 327 {
 328         printf_P(PSTR(ESC_FG_RED "Dev Enum Error\r\n"
 329                                  " -- Error Code %d\r\n"
 330                                  " -- Sub Error Code %d\r\n"
 331                                  " -- In State %d\r\n" ESC_FG_WHITE), ErrorCode, SubErrorCode, USB_HostState);
 332
 333         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 334 }