接第一次代码继续分析
uses
JwaWinsock2, Windows, SysUtils;
const
DATA_BUFSIZE = 1024;
IO_TYPE_Accept = 1;
IO_TYPE_Recv = 2;
type
//(1):单IO数据结构
LPVOID = Pointer;
LPPER_IO_OPERATION_DATA = ^PER_IO_OPERATION_DATA ;
PER_IO_OPERATION_DATA = packed record
Overlapped: OVERLAPPED;
IO_TYPE: Cardinal;
DataBuf: TWSABUF;
Buffer: array [0..1024] of CHAR;
end;
刚开始结存iocp的时候可能无法理解为什么要申明这样一个结构。
解释下,这个结构是GetQueuedCompletionStatus,PostQueuedCompletionStatus,WSARecv,WSASend,时需要用到一个POverlapped类型的参数。
也许还会有疑惑,为什么不直接使用系统自带的类型呢?
POverlapped = ^TOverlapped;
_OVERLAPPED = record
Internal: DWORD;
InternalHigh: DWORD;
Offset: DWORD;
OffsetHigh: DWORD;
hEvent: THandle;
end;
///我可以解释下.是为了在PostQueuedCompletionStatus,WSARecv,WSASend尽可能多传递一下信息给GetQueuedCompletionStatus,所以一般都会扩展这一机构体
//再啰嗦下。定义的结构体,Overlapped: OVERLAPPED;必须放在第一个.你懂的。
//PostQueuedCompletionStatus,WSARecv,WSASend会触发工作线程的GetQueuedCompletionStatus返回<上一笔记有提到>
下面片段是Listen过程
//下面循环进行循环获取客户端的请求。
while (TRUE) do
begin
//当客户端有连接请求的时候,WSAAccept函数会新创建一个套接字cSocket。这个套接字就是和客户端通信的时候使用的套接字。
cSocket:= WSAAccept(sSocket, nil, nil, nil, 0);
//判断cSocket套接字创建是否成功,如果不成功则退出。
if (cSocket= SOCKET_ERROR) then
begin
closesocket(sSocket);
exit;
end;
//将套接字、完成端口绑定在一起。
// 最开始的时候没有明白为什么还要调用一次createIoCompletionPort
//
// 后来经过google,和测试
//
// 是将新的套接字(socket)加入到iocp端口<绑定>
// 这样工作线程才能处理这个套接字(socket)的数据包
//如果把下面注释掉,WSARecv这个套接字时,GetQueuedCompletionStatus无法处理到收到的数据包
// 2013年4月19日 09:56:00
// 注意第三个参数也需要进行绑定, 否则在工作线程中GetQueuedCompletionStatus时completionKey会取不到cSocket值
lvPerIOPort := CreateIoCompletionPort(cSocket, lvIOPort, cSocket, 0);
if (lvPerIOPort = 0) then
begin
Exit;
end;
//初始化数据包
PerIoData := LPPER_IO_OPERATION_DATA(GlobalAlloc(GPTR, sizeof(PER_IO_OPERATION_DATA)));
ZeroMemory(@PerIoData.Overlapped, sizeof(OVERLAPPED));
//数据包中的IO类型:有连接请求
PerIoData.IO_TYPE := IO_TYPE_Accept;
//通知工作线程,有新的套接字连接<第三个参数>
PostQueuedCompletionStatus(lvIOPort, 0, cSocket, POverlapped(PerIOData));
end;
下面是IOCP工作线程
function ServerWorkerThread(pData:Pointer): Integer; stdcall;
var
CompletionPort:THANDLE;
BytesTransferred:Cardinal;
PerIoData:LPPER_IO_OPERATION_DATA;
cSocket:TSocket;
Flags:Cardinal;
RecvBytes:Cardinal;
lvResultStatus:BOOL;
lvRet:Integer;
begin
CompletionPort:=THandle(pData);
//得到创建线程是传递过来的IOCP
while(TRUE) do
begin
//工作者线程会停止到GetQueuedCompletionStatus函数处,直到接受到数据为止
lvResultStatus := GetQueuedCompletionStatus(CompletionPort,
BytesTransferred,
cSocket,
POverlapped(PerIoData), INFINITE);
if (lvResultStatus = False) then
begin
//当客户端连接断开或者客户端调用closesocket函数的时候,函数GetQueuedCompletionStatus会返回错误。如果我们加入心跳后,在这里就可以来判断套接字是否依然在连接。
if cSocket<>0 then
begin
closesocket(cSocket);
end;
if PerIoData<>nil then
begin
GlobalFree(DWORD(PerIoData));
end;
continue;
end;
if PerIoData = nil then
begin
closesocket(cSocket);
Break;
end else if (PerIoData<>nil) then
begin
////shutdown(PerHandleData.Socket, 1);
if PerIoData.IO_TYPE = IO_TYPE_Accept then //连接请求
begin
GlobalFree(DWORD(PerIoData));
end else if PerIoData.IO_TYPE = IO_TYPE_Recv then
begin
////可以在这里处理数据……
GlobalFree(DWORD(PerIoData));
end;
/////分配内存<可以加入内存池>
PerIoData := LPPER_IO_OPERATION_DATA(GlobalAlloc(GPTR, sizeof(PER_IO_OPERATION_DATA)));
ZeroMemory(@PerIoData.Overlapped, sizeof(OVERLAPPED));
Flags := 0;
/////进入投递收取动作
PerIoData.IO_TYPE := IO_TYPE_Recv;
PerIoData.DataBuf.len:=DATA_BUFSIZE;
ZeroMemory(@PerIoData.Buffer,sizeof(@PerIoData.Buffer));
PerIoData.DataBuf.buf := @PerIoData.Buffer;
/////异步收取数据
WSARecv(cSocket,
@PerIoData.DataBuf,
1,
RecvBytes,
Flags,
@PerIoData^, nil);
if (WSAGetLastError() <> ERROR_IO_PENDING) then
begin
closesocket(cSocket);
if PerIoData <> nil then
begin
GlobalFree(DWORD(PerIoData));
end;
Continue;
end;
end;
end;
end;