UEFI实战(10) Network - c++编程基础

EAN Urgent;
UINT32 DataLength;
UINT32 FragmentCount;
EFI_TCP4_FRAGMENT_DATA FragmentTable[1];
} EFI_TCP4_TRANSMIT_DATA;//***************************************************************
// EFI_TCP4_FRAGMENT_DATA
//***************************************************************
typedef struct {
UINT32 FragmentLength;
VOID *FragmentBuffer;
} EFI_TCP4_FRAGMENT_DATA;
待发送数据可能在几个不连续的缓冲区内，我们可以将这些缓冲区指针放到 FragmentTable数组内，数组中每个元素表示一个缓冲区。DataLength是数据总长度(各个缓冲区长度之和), FragmentCount是缓冲区个数。
下面的代码中我们只用到了一个缓冲区。同Connect一样，调用Transmit之后要通过WaitForEvent等待发送完成。

SOCKET_STAUS Socket::Send(CHAR8* Data, UINTN Lenth)
{
EFI_STATUS Status = EFI_NOT_FOUND;
if(m_pTcp4Protocol == NULL) return Status;
m_TransData->Push = TRUE;
m_TransData->Urgent = TRUE;
m_TransData->DataLength = (UINT32)Lenth;
m_TransData->FragmentCount = 1;
m_TransData->FragmentTable[0].FragmentLength =m_TransData->DataLength;
m_TransData->FragmentTable[0].FragmentBuffer =Data;
SendToken.Packet.TxData= m_TransData;
Status = m_pTcp4Protocol -> Transmit(m_pTcp4Protocol, &SendToken);
if(EFI_ERROR(Status))
return Status;
SocketWait(SendToken.CompletionToken.Event);
return SendToken.CompletionToken.Status;
}
接收数据

接收与发送相似
typedef
EFI_STATUS
(EFIAPI *EFI_TCP4_RECEIVE) (
IN EFI_TCP4_PROTOCOL *This,
IN EFI_TCP4_IO_TOKEN *Token
);
//***************************************************************
// EFI_TCP4_RECEIVE_DATA
//***************************************************************
typedef struct {
BOOLEAN UrgentFlag;
UINT32 DataLength;
UINT32 FragmentCount;
EFI_TCP4_FRAGMENT_DATA FragmentTable[1];
} EFI_TCP4_RECEIVE_DATA;用户负责分配和释放缓冲区，DataLength是缓冲区总长度，接收完成时，Token中的事件被设置， DataLength也被设置为接收到的数据的长度。

SOCKET Socket::Recv(CHAR8* Buffer, UINTN Lenth)
{
EFI_STATUS Status = EFI_NOT_FOUND;
if(m_pTcp4Protocol == NULL) return Status;

m_RecvData->UrgentFlag = TRUE;
m_RecvData->DataLength = (UINT32)Lenth;
m_RecvData->FragmentCount = 1;
m_RecvData->FragmentTable[0].FragmentLength = m_RecvData->DataLength ;
m_RecvData->FragmentTable[0].FragmentBuffer = (void*)Buffer;
RecvToken.Packet.RxData= m_RecvData;
Status = m_pTcp4Protocol -> Receive(m_pTcp4Protocol, &RecvToken);
if(EFI_ERROR(Status))
return Status;
SocketWait(RecvToken.CompletionToken.Event);
return RecvToken.CompletionToken.Status;
}
回头看一下Token及m_RecvData 等相关辅助数据时如何建立的

SOCKET_STATUS Socket::Initialize()
{
EFI_STATUS Status;
// 建立Configure Data
m_pTcp4ConfigData = new EFI_TCP4_CONFIG_DATA[1];
// 建立 Connect Data
ConnectToken.CompletionToken.Status = EFI_ABORTED;
Status = gBS->CreateEvent(EVT_NOTIFY_SIGNAL, TPL_CALLBACK, (EFI_EVENT_NOTIFY)NopNoify , (VOID*)&ConnectToken, &ConnectToken.CompletionToken.Event );
if(EFI_STATUS(Stauts)) return Status;
// 建立 Transmit Data
Status = gBS->CreateEvent(EVT_NOTIFY_SIGNAL, TPL_CALLBACK, (EFI_EVENT_NOTIFY)NopNoify , (VOID*)&SendToken, &SendToken.CompletionToken.Event);
if(EFI_STATUS(Stauts)) return Status;
SendToken.CompletionToken.Status =EFI_ABORTED;
m_TransData = new EFI_TCP4_TRANSMIT_DATA[1];
// 建立 Recv Data
Status = gBS->CreateEvent(EVT_NOTIFY_SIGNAL, TPL_CALLBACK, (EFI_EVENT_N

UEFI实战(10) Network (四)