Demos/Device/ClassDriver/RNDISEthernet/Lib/TCP.h

   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  *  Header file for TCP.c.
  34  */
  35
  36 #ifndef _TCP_H_
  37 #define _TCP_H_
  38
  39         /* Includes: */
  40                 #include <avr/io.h>
  41                 #include <stdbool.h>
  42
  43                 #include "EthernetProtocols.h"
  44                 #include "Ethernet.h"
  45                 #include "ProtocolDecoders.h"
  46
  47         /* Macros: */
  48                 /** Maximum number of TCP ports which can be open at the one time. */
  49                 #define MAX_OPEN_TCP_PORTS              1
  50
  51                 /** Maximum number of TCP connections which can be sustained at the one time. */
  52                 #define MAX_TCP_CONNECTIONS             3
  53
  54                 /** TCP window size, giving the maximum number of bytes which can be buffered at the one time. */
  55                 #define TCP_WINDOW_SIZE                 512
  56
  57                 /** Port number for HTTP transmissions. */
  58                 #define TCP_PORT_HTTP                   SwapEndian_16(80)
  59
  60                 /** Data direction indicator for a TCP application buffer, indicating data from host-to-device. */
  61                 #define TCP_PACKETDIR_IN                false
  62
  63                 /** Data direction indicator for a TCP application buffer, indicating data from device-to-host. */
  64                 #define TCP_PACKETDIR_OUT               true
  65
  66                 /** Congestion Window Reduced TCP flag mask. */
  67                 #define TCP_FLAG_CWR                    (1 << 7)
  68
  69                 /** Explicit Congestion Notification TCP flag mask. */
  70                 #define TCP_FLAG_ECE                    (1 << 6)
  71
  72                 /** Urgent TCP flag mask. */
  73                 #define TCP_FLAG_URG                    (1 << 5)
  74
  75                 /** Data Acknowledge TCP flag mask. */
  76                 #define TCP_FLAG_ACK                    (1 << 4)
  77
  78                 /** Data Push TCP flag mask. */
  79                 #define TCP_FLAG_PSH                    (1 << 3)
  80
  81                 /** Reset TCP flag mask. */
  82                 #define TCP_FLAG_RST                    (1 << 2)
  83
  84                 /** Synchronize TCP flag mask. */
  85                 #define TCP_FLAG_SYN                    (1 << 1)
  86
  87                 /** Connection Finalize TCP flag mask. */
  88                 #define TCP_FLAG_FIN                    (1 << 0)
  89
  90                 /** Application macro: Determines if the given application buffer contains a packet received from the host
  91                  *
  92                  *  \param[in] Buffer  Application buffer to check
  93                  *
  94                  *  \return Boolean true if the buffer contains a packet from the host, false otherwise
  95                  */
  96                 #define TCP_APP_HAS_RECEIVED_PACKET(Buffer)  (Buffer->Ready && (Buffer->Direction == TCP_PACKETDIR_IN))
  97
  98                 /** Application macro: Indicates if the application buffer is currently locked by the application for device-to-host transfers.
  99                  *
 100                  *  \param[in] Buffer  Application buffer to check
 101                  *
 102                  *  \return Boolean true if the buffer has been captured by the application for device-to-host transmissions, false otherwise
 103                  */
 104                 #define TCP_APP_HAVE_CAPTURED_BUFFER(Buffer) (!(Buffer->Ready) && Buffer->InUse && \
 105
 106                            (Buffer->Direction == TCP_PACKETDIR_OUT))
 107
 108                 /** Application macro: Indicates if the application can lock the buffer for multiple continued device-to-host transmissions.
 109                  *
 110                  *  \param[in] Buffer  Application buffer to check
 111                  *
 112                  *  \return Boolean true if the buffer may be captured by the application for device-to-host transmissions, false otherwise
 113                  */
 114                 #define TCP_APP_CAN_CAPTURE_BUFFER(Buffer)   Buffer->InUse
 115
 116                 /** Application macro: Captures the application buffer, locking it for device-to-host transmissions only. This should be
 117                  *  performed when the application needs to transmit several packets worth of data in succession with no interruptions from the host.
 118                  *
 119                  *  \pre The application must check that the buffer can be locked first using TCP_APP_CAN_CAPTURE_BUFFER().
 120                  *
 121                  *  \param[in] Buffer  Application buffer to lock
 122                  */
 123                 #define TCP_APP_CAPTURE_BUFFER(Buffer)       MACROS{ Buffer->Direction = TCP_PACKETDIR_OUT; Buffer->InUse = true; }MACROE
 124
 125                 /** Application macro: Releases a captured application buffer, allowing for host-to-device packets to be received.
 126                  *
 127                  *  \param[in] Buffer  Application buffer to release
 128                  */
 129                 #define TCP_APP_RELEASE_BUFFER(Buffer)       MACROS{ Buffer->InUse = false; }MACROE
 130
 131                 /** Application macro: Sends the contents of the given application buffer to the host.
 132                  *
 133                  *  \param[in] Buffer  Application buffer to send
 134                  *  \param[in] Len     Length of data contained in the buffer
 135                  */
 136                 #define TCP_APP_SEND_BUFFER(Buffer, Len)     MACROS{ Buffer->Direction = TCP_PACKETDIR_OUT; Buffer->Length = Len; Buffer->Ready = true; }MACROE
 137
 138                 /** Application macro: Clears the application buffer, ready for a packet to be written to it.
 139                  *
 140                  *  \param[in] Buffer  Application buffer to clear
 141                  */
 142                 #define TCP_APP_CLEAR_BUFFER(Buffer)         MACROS{ Buffer->Ready = false; Buffer->Length = 0; }MACROE
 143
 144                 /** Application macro: Closes an open connection to a host.
 145                  *
 146                  *  \param[in] Connection  Open TCP connection to close
 147                  */
 148                 #define TCP_APP_CLOSECONNECTION(Connection)  MACROS{ Connection->State = TCP_Connection_Closing;  }MACROE
 149
 150         /* Enums: */
 151                 /** Enum for possible TCP port states. */
 152                 enum TCP_PortStates_t
 153                 {
 154                         TCP_Port_Closed            = 0, /**< TCP port closed, no connections to a host may be made on this port. */
 155                         TCP_Port_Open              = 1, /**< TCP port open, connections to a host may be made on this port. */
 156                 };
 157
 158                 /** Enum for possible TCP connection states. */
 159                 enum TCP_ConnectionStates_t
 160                 {
 161                         TCP_Connection_Listen      = 0, /**< Listening for a connection from a host */
 162                         TCP_Connection_SYNSent     = 1, /**< Unused */
 163                         TCP_Connection_SYNReceived = 2, /**< SYN received, waiting for ACK */
 164                         TCP_Connection_Established = 3, /**< Connection established in both directions */
 165                         TCP_Connection_FINWait1    = 4, /**< Closing, waiting for ACK */
 166                         TCP_Connection_FINWait2    = 5, /**< Closing, waiting for FIN ACK */
 167                         TCP_Connection_CloseWait   = 6, /**< Closing, waiting for ACK */
 168                         TCP_Connection_Closing     = 7, /**< Unused */
 169                         TCP_Connection_LastACK     = 8, /**< Unused */
 170                         TCP_Connection_TimeWait    = 9, /**< Unused */
 171                         TCP_Connection_Closed      = 10, /**< Connection closed in both directions */
 172                 };
 173
 174         /* Type Defines: */
 175                 /** Type define for a TCP connection buffer structure, including size, data and direction. */
 176                 typedef struct
 177                 {
 178                         uint16_t               Length; /**< Length of data in the TCP application buffer */
 179                         uint8_t                Data[TCP_WINDOW_SIZE]; /**< TCP application data buffer */
 180                         bool                   Direction; /**< Buffer transmission direction, either TCP_PACKETDIR_IN  or TCP_PACKETDIR_OUT */
 181                         bool                   Ready; /**< If data from host, indicates buffer ready to be read, otherwise indicates
 182                                                        *   buffer ready to be sent to the host
 183                                                        */
 184                         bool                   InUse; /**< Indicates if the buffer is locked to to the current direction, and cannot be changed */
 185                 } TCP_ConnectionBuffer_t;
 186
 187                 /** Type define for a TCP connection information structure. */
 188                 typedef struct
 189                 {
 190                         uint32_t               SequenceNumberIn; /**< Current TCP sequence number for host-to-device */
 191                         uint32_t               SequenceNumberOut; /**< Current TCP sequence number for device-to-host */
 192                         TCP_ConnectionBuffer_t Buffer; /**< Connection application data buffer */
 193                 } TCP_ConnectionInfo_t;
 194
 195                 /** Type define for a complete TCP connection state. */
 196                 typedef struct
 197                 {
 198                         uint16_t               Port; /**< Connection port number on the device */
 199                         uint16_t               RemotePort; /**< Connection port number on the host */
 200                         IP_Address_t           RemoteAddress; /**< Connection protocol IP address of the host */
 201                         TCP_ConnectionInfo_t   Info; /**< Connection information, including application buffer */
 202                         uint8_t                State; /**< Current connection state, a value from the TCP_ConnectionStates_t enum */
 203                 } TCP_ConnectionState_t;
 204
 205                 /** Type define for a TCP port state. */
 206                 typedef struct
 207                 {
 208                         uint16_t               Port; /**< TCP port number on the device */
 209                         uint8_t                State; /**< Current port state, a value from the TCP_PortStates_t enum */
 210                         void                   (*ApplicationHandler) (TCP_ConnectionState_t* ConnectionState,
 211                                                                       TCP_ConnectionBuffer_t* Buffer); /**< Port application handler */
 212                 } TCP_PortState_t;
 213
 214                 /** Type define for a TCP packet header. */
 215                 typedef struct
 216                 {
 217                         uint16_t               SourcePort; /**< Source port of the TCP packet */
 218                         uint16_t               DestinationPort; /**< Destination port of the TCP packet */
 219
 220                         uint32_t               SequenceNumber; /**< Data sequence number of the packet */
 221                         uint32_t               AcknowledgmentNumber; /**< Data acknowledgment number of the packet */
 222
 223                         unsigned char          Reserved : 4; /**< Reserved, must be all 0 */
 224                         unsigned char          DataOffset : 4; /**< Offset of the data from the start of the header, in 4 byte chunks */
 225                         uint8_t                Flags; /**< TCP packet flags */
 226                         uint16_t               WindowSize; /**< Current data window size (bytes remaining in reception buffer) */
 227
 228                         uint16_t               Checksum; /**< TCP checksum */
 229                         uint16_t               UrgentPointer; /**< Urgent data pointer */
 230                 } TCP_Header_t;
 231
 232         /* Function Prototypes: */
 233                 void                  TCP_TCPTask(USB_ClassInfo_RNDIS_Device_t* RNDISInterfaceInfo);
 234                 void                  TCP_Init(void);
 235                 bool                  TCP_SetPortState(uint16_t Port, uint8_t State, void (*Handler)(TCP_ConnectionState_t*, TCP_ConnectionBuffer_t*));
 236                 uint8_t               TCP_GetPortState(uint16_t Port);
 237                 bool                  TCP_SetConnectionState(uint16_t Port, IP_Address_t RemoteAddress, uint16_t RemotePort, uint8_t State);
 238                 uint8_t               TCP_GetConnectionState(uint16_t Port, IP_Address_t RemoteAddress, uint16_t RemotePort);
 239                 TCP_ConnectionInfo_t* TCP_GetConnectionInfo(uint16_t Port, IP_Address_t RemoteAddress, uint16_t RemotePort);
 240                 int16_t               TCP_ProcessTCPPacket(void* IPHeaderInStart, void* TCPHeaderInStart, void* TCPHeaderOutStart);
 241
 242                 #if defined(INCLUDE_FROM_TCP_C)
 243                         static uint16_t TCP_Checksum16(void* TCPHeaderOutStart, IP_Address_t SourceAddress,
 244                                                                                    IP_Address_t DestinationAddress, uint16_t TCPOutSize);
 245                 #endif
 246
 247 #endif