/** @file\r
+ Socket header file.\r
\r
-Copyright (c) 2005 - 2006, Intel Corporation<BR>\r
-All rights reserved. This program and the accompanying materials\r
+Copyright (c) 2005 - 2012, Intel Corporation. All rights reserved.<BR>\r
+This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution. The full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php<BR>\r
\r
#include <Library/NetLib.h>\r
#include <Library/DebugLib.h>\r
+#include <Library/BaseMemoryLib.h>\r
+#include <Library/MemoryAllocationLib.h>\r
#include <Library/UefiRuntimeServicesTableLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
+#include <Library/UefiDriverEntryPoint.h>\r
#include <Library/UefiLib.h>\r
+#include <Library/DpcLib.h>\r
+#include <Library/PrintLib.h>\r
\r
#define SOCK_SND_BUF 0\r
#define SOCK_RCV_BUF 1\r
///\r
/// Socket state\r
///\r
-typedef enum {\r
- SO_CLOSED = 0,\r
- SO_LISTENING,\r
- SO_CONNECTING,\r
- SO_CONNECTED,\r
- SO_DISCONNECTING\r
-} SOCK_STATE;\r
+#define SO_CLOSED 0\r
+#define SO_LISTENING 1\r
+#define SO_CONNECTING 2\r
+#define SO_CONNECTED 3\r
+#define SO_DISCONNECTING 4\r
\r
///\r
/// Socket configure state\r
///\r
-typedef enum {\r
- SO_UNCONFIGURED = 0,\r
- SO_CONFIGURED_ACTIVE,\r
- SO_CONFIGURED_PASSIVE,\r
- SO_NO_MAPPING\r
-} SOCK_CONFIGURE_STATE;\r
+#define SO_UNCONFIGURED 0\r
+#define SO_CONFIGURED_ACTIVE 1\r
+#define SO_CONFIGURED_PASSIVE 2\r
+#define SO_NO_MAPPING 3\r
\r
+/**\r
+ Set socket SO_NO_MORE_DATA flag.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+**/\r
#define SOCK_NO_MORE_DATA(Sock) ((Sock)->Flag |= SO_NO_MORE_DATA)\r
\r
+/**\r
+ Check whether the socket is unconfigured.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is unconfigued\r
+ @retval False The socket is not unconfigued\r
+ \r
+**/\r
#define SOCK_IS_UNCONFIGURED(Sock) ((Sock)->ConfigureState == SO_UNCONFIGURED)\r
\r
+/**\r
+ Check whether the socket is configured.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is configued\r
+ @retval False The socket is not configued\r
+ \r
+**/\r
#define SOCK_IS_CONFIGURED(Sock) \\r
(((Sock)->ConfigureState == SO_CONFIGURED_ACTIVE) || \\r
((Sock)->ConfigureState == SO_CONFIGURED_PASSIVE))\r
\r
+/**\r
+ Check whether the socket is configured to active mode.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is configued to active mode\r
+ @retval False The socket is not configued to active mode\r
+ \r
+**/\r
#define SOCK_IS_CONFIGURED_ACTIVE(Sock) \\r
((Sock)->ConfigureState == SO_CONFIGURED_ACTIVE)\r
\r
+/**\r
+ Check whether the socket is configured to passive mode.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is configued to passive mode\r
+ @retval False The socket is not configued to passive mode\r
+ \r
+**/\r
#define SOCK_IS_CONNECTED_PASSIVE(Sock) \\r
((Sock)->ConfigureState == SO_CONFIGURED_PASSIVE)\r
\r
+/**\r
+ Check whether the socket is mapped.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is no mapping\r
+ @retval False The socket is mapped\r
+ \r
+**/\r
#define SOCK_IS_NO_MAPPING(Sock) \\r
((Sock)->ConfigureState == SO_NO_MAPPING)\r
\r
+/**\r
+ Check whether the socket is closed.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is closed\r
+ @retval False The socket is not closed\r
+ \r
+**/\r
#define SOCK_IS_CLOSED(Sock) ((Sock)->State == SO_CLOSED)\r
\r
+/**\r
+ Check whether the socket is listening.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is listening\r
+ @retval False The socket is not listening\r
+ \r
+**/\r
#define SOCK_IS_LISTENING(Sock) ((Sock)->State == SO_LISTENING)\r
\r
+/**\r
+ Check whether the socket is connecting.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is connecting\r
+ @retval False The socket is not connecting\r
+ \r
+**/\r
#define SOCK_IS_CONNECTING(Sock) ((Sock)->State == SO_CONNECTING)\r
\r
+/**\r
+ Check whether the socket has connected.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket has connected\r
+ @retval False The socket has not connected\r
+ \r
+**/\r
#define SOCK_IS_CONNECTED(Sock) ((Sock)->State == SO_CONNECTED)\r
\r
+/**\r
+ Check whether the socket is disconnecting.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is disconnecting\r
+ @retval False The socket is not disconnecting\r
+ \r
+**/\r
#define SOCK_IS_DISCONNECTING(Sock) ((Sock)->State == SO_DISCONNECTING)\r
\r
+/**\r
+ Check whether the socket is no more data.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @retval True The socket is no more data\r
+ @retval False The socket still has data\r
+ \r
+**/\r
#define SOCK_IS_NO_MORE_DATA(Sock) (0 != ((Sock)->Flag & SO_NO_MORE_DATA))\r
\r
-#define SOCK_SIGNATURE SIGNATURE_32 ('S', 'O', 'C', 'K')\r
-\r
-#define SOCK_FROM_THIS(a) CR ((a), SOCKET, NetProtocol, SOCK_SIGNATURE)\r
+/**\r
+ Set the size of the receive buffer.\r
+ \r
+ @param Sock Pointer to the socket\r
+ @param Size The size to set\r
\r
+**/\r
#define SET_RCV_BUFFSIZE(Sock, Size) ((Sock)->RcvBuffer.HighWater = (Size))\r
\r
+/**\r
+ Get the size of the receive buffer.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @return The receive buffer size\r
+\r
+**/\r
#define GET_RCV_BUFFSIZE(Sock) ((Sock)->RcvBuffer.HighWater)\r
\r
+/**\r
+ Get the size of the receive data.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @return The received data size\r
+\r
+**/\r
#define GET_RCV_DATASIZE(Sock) (((Sock)->RcvBuffer.DataQueue)->BufSize)\r
\r
+/**\r
+ Set the size of the send buffer.\r
+ \r
+ @param Sock Pointer to the socket\r
+ @param Size The size to set\r
+\r
+**/\r
#define SET_SND_BUFFSIZE(Sock, Size) ((Sock)->SndBuffer.HighWater = (Size))\r
\r
+/**\r
+ Get the size of the send buffer.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @return The send buffer size\r
+\r
+**/\r
#define GET_SND_BUFFSIZE(Sock) ((Sock)->SndBuffer.HighWater)\r
\r
+/**\r
+ Get the size of the send data.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @return The send data size\r
+\r
+**/\r
#define GET_SND_DATASIZE(Sock) (((Sock)->SndBuffer.DataQueue)->BufSize)\r
\r
+/**\r
+ Set the backlog value of the socket.\r
+ \r
+ @param Sock Pointer to the socket\r
+ @param Value The value to set\r
+\r
+**/\r
#define SET_BACKLOG(Sock, Value) ((Sock)->BackLog = (Value))\r
\r
+/**\r
+ Get the backlog value of the socket.\r
+ \r
+ @param Sock Pointer to the socket\r
+ \r
+ @return The backlog value\r
+\r
+**/\r
#define GET_BACKLOG(Sock) ((Sock)->BackLog)\r
\r
+/**\r
+ Set the socket with error state.\r
+ \r
+ @param Sock Pointer to the socket\r
+ @param Error The error state\r
+\r
+**/\r
#define SOCK_ERROR(Sock, Error) ((Sock)->SockError = (Error))\r
\r
#define SND_BUF_HDR_LEN(Sock) \\r
#define RCV_BUF_HDR_LEN(Sock) \\r
((SockBufFirst (&((Sock)->RcvBuffer)))->TotalSize)\r
\r
+#define SOCK_SIGNATURE SIGNATURE_32 ('S', 'O', 'C', 'K')\r
+\r
+#define SOCK_FROM_THIS(a) CR ((a), SOCKET, NetProtocol, SOCK_SIGNATURE)\r
+\r
#define SOCK_FROM_TOKEN(Token) (((SOCK_TOKEN *) (Token))->Sock)\r
\r
#define PROTO_TOKEN_FORM_SOCK(SockToken, Type) \\r
EFI_STATUS Status; ///< The status to be issued\r
} SOCK_COMPLETION_TOKEN;\r
\r
+typedef union {\r
+ VOID *RxData;\r
+ VOID *TxData;\r
+} SOCK_IO_DATA;\r
+\r
///\r
/// The application token with data packet\r
///\r
typedef struct _SOCK_IO_TOKEN {\r
SOCK_COMPLETION_TOKEN Token;\r
- union {\r
- VOID *RxData;\r
- VOID *TxData;\r
- } Packet;\r
+ SOCK_IO_DATA Packet;\r
} SOCK_IO_TOKEN;\r
\r
///\r
/// The request issued from socket layer to protocol layer. \r
///\r
-typedef enum {\r
- SOCK_ATTACH, ///< Attach current socket to a new PCB\r
- SOCK_DETACH, ///< Detach current socket from the PCB\r
- SOCK_CONFIGURE, ///< Configure attached PCB\r
- SOCK_FLUSH, ///< Flush attached PCB\r
- SOCK_SND, ///< Need protocol to send something\r
- SOCK_SNDPUSH, ///< Need protocol to send pushed data\r
- SOCK_SNDURG, ///< Need protocol to send urgent data\r
- SOCK_CONSUMED, ///< Application has retrieved data from socket\r
- SOCK_CONNECT, ///< Need to connect to a peer\r
- SOCK_CLOSE, ///< Need to close the protocol process\r
- SOCK_ABORT, ///< Need to reset the protocol process\r
- SOCK_POLL, ///< Need to poll to the protocol layer\r
- SOCK_ROUTE, ///< Need to add a route information\r
- SOCK_MODE, ///< Need to get the mode data of the protocol\r
- SOCK_GROUP ///< Need to join a mcast group\r
-} SOCK_REQUEST;\r
+#define SOCK_ATTACH 0 ///< Attach current socket to a new PCB\r
+#define SOCK_DETACH 1 ///< Detach current socket from the PCB\r
+#define SOCK_CONFIGURE 2 ///< Configure attached PCB\r
+#define SOCK_FLUSH 3 ///< Flush attached PCB\r
+#define SOCK_SND 4 ///< Need protocol to send something\r
+#define SOCK_SNDPUSH 5 ///< Need protocol to send pushed data\r
+#define SOCK_SNDURG 6 ///< Need protocol to send urgent data\r
+#define SOCK_CONSUMED 7 ///< Application has retrieved data from socket\r
+#define SOCK_CONNECT 8 ///< Need to connect to a peer\r
+#define SOCK_CLOSE 9 ///< Need to close the protocol process\r
+#define SOCK_ABORT 10 ///< Need to reset the protocol process\r
+#define SOCK_POLL 11 ///< Need to poll to the protocol layer\r
+#define SOCK_ROUTE 12 ///< Need to add a route information\r
+#define SOCK_MODE 13 ///< Need to get the mode data of the protocol\r
+#define SOCK_GROUP 14 ///< Need to join a mcast group\r
\r
///\r
/// The socket type.\r
///\r
typedef enum {\r
- SOCK_DGRAM, ///< This socket providing datagram service\r
- SOCK_STREAM ///< This socket providing stream service\r
+ SockDgram, ///< This socket providing datagram service\r
+ SockStream ///< This socket providing stream service\r
} SOCK_TYPE;\r
\r
///\r
EFI_STATUS\r
(*SOCK_PROTO_HANDLER) (\r
IN SOCKET *Socket,\r
- IN SOCK_REQUEST Request,\r
+ IN UINT8 Request,\r
IN VOID *RequestData\r
);\r
\r
Set the state of the socket.\r
\r
@param Sock Pointer to the socket.\r
- @param State The new state to be set.\r
+ @param State The new socket state to be set.\r
\r
**/\r
VOID\r
SockSetState (\r
IN OUT SOCKET *Sock,\r
- IN SOCK_STATE State\r
+ IN UINT8 State\r
);\r
\r
/**\r
///\r
typedef struct _SOCK_INIT_DATA {\r
SOCK_TYPE Type;\r
- SOCK_STATE State;\r
+ UINT8 State;\r
\r
SOCKET *Parent; ///< The parent of this socket\r
UINT32 BackLog; ///< The connection limit for listening socket\r
EFI_DEVICE_PATH_PROTOCOL *ParentDevicePath;\r
EFI_DEVICE_PATH_PROTOCOL *DevicePath;\r
LIST_ENTRY Link; \r
- SOCK_CONFIGURE_STATE ConfigureState;\r
+ UINT8 ConfigureState;\r
SOCK_TYPE Type;\r
- SOCK_STATE State;\r
+ UINT8 State;\r
UINT16 Flag;\r
EFI_LOCK Lock; ///< The lock of socket\r
SOCK_BUFFER SndBuffer; ///< Send buffer of application's data\r
SOCK_BUFFER RcvBuffer; ///< Receive buffer of received data\r
EFI_STATUS SockError; ///< The error returned by low layer protocol\r
- BOOLEAN IsDestroyed;\r
+ BOOLEAN InDestroy;\r
\r
//\r
// Fields used to manage the connection request\r
);\r
\r
/**\r
- Destory the socket Sock and its associated protocol control block.\r
+ Destroy the socket Sock and its associated protocol control block.\r
\r
@param Sock The socket to be destroyed.\r
\r