完成端口iocp——在螺丝壳里做道场

WINDOWS 2000以后的操作系统才支持IOCP。WINSOCK2.0才支持IOCP。

首先要有一个WINSOCK2.PAS的WINSOCK2.0接口调用声明单元。

WINSOCK的版本号： WINSOCK_VERSION = $0202;

动态库：ws2_32 = 'ws2_32.dll';

1）服务端首先要创建一个监听SOCKET，用于监听客户端连接。

1.1)加载WINSOCK2协议

if WSAStartup($0202, WsaData) <> 0 then
raise ESocketError.Create(GetLastWsaErrorStr);

1.2）创建服务端监听SOCKET

VAR FSocket: TSocket;

FSocket := WSASocket(PF_INET, SOCK_STREAM, 0, nil, 0, WSA_FLAG_OVERLAPPED);
if FSocket = INVALID_SOCKET then
raise ESocketError.Create(GetLastWsaErrorStr);

1.3)为创建的服务端监听SOCKET准备IP地址，端口号。。。

var Addr: TSockAddr;

FillChar(Addr, SizeOf(Addr), 0);
Addr.sin_family := AF_INET;
Addr.sin_port := htons(FPort);
Addr.sin_addr.S_addr := htonl(INADDR_ANY); //在任何地址上监听，如果有多块网卡，会每块都监听

1.4)为服务端监听SOCKET绑定

if bind(FSocket, @Addr, SizeOf(Addr)) <> 0 then
raise ESocketError.Create(GetLastWsaErrorStr);

1.5)在服务端监听SOCKET上开启监听

if listen(FSocket, MaxInt) <> 0 then
raise ESocketError.Create(GetLastWsaErrorStr);

1.6)服务端监听线程。当然也可以在主线程执行一个WHILE循环不停地接收客户端的连接，但使用监听线程显然更好。

var
ClientSocket: TSocket;

ClientSocket := WSAAccept(FSocket, nil, nil, nil, 0);  // FSocket就是前面已经创建好的服务端监听SOCKET

if ClientSocket <> INVALID_SOCKET then begin

PostQueuedCompletionStatus(FIocpHandle, 0, ClientSocket, nil); // 客户端连接被提交到完成端口队列中

end;

2）IOCP

2.1)创建完成端口。

var FIocpHandle: THandle;

FIocpHandle := CreateIoCompletionPort(INVALID_HANDLE_VALUE, 0, 0, 0);
if FIocpHandle = 0 then
raise ESocketError.Create(GetLastErrorStr);

2.1)将客户端连接和IOCP绑定

var cSocket: PClientSocket;

CreateIoCompletionPort(ClientSocket, FIOCPHandle, DWORD(cSocket), 0);

 2.2)接收客户端数据

type
{* 完成端口操作定义 *}
TIocpOperate = (ioNone, ioRead, ioWrite, ioStream, ioExit);
PIocpRecord = ^TIocpRecord;
TIocpRecord = record
Overlapped: TOverlapped; //完成端口重叠结构
WsaBuf: TWsaBuf; //完成端口的缓冲区定义
IocpOperate: TIOCPOperate; //当前操作类型
end;

var FIocpRecv: PIocpRecord;

iFlags, iTransfer: Cardinal;

FIocpRecv.Overlapped.Internal := 0;
FIocpRecv.Overlapped.InternalHigh := 0;
FIocpRecv.Overlapped.Offset := 0;
FIocpRecv.Overlapped.OffsetHigh := 0;
FIocpRecv.Overlapped.hEvent := 0;
FIocpRecv.IocpOperate := ioRead;
iFlags := 0;

WSARecv(FSocket, @FIocpRecv.WsaBuf, 1, iTransfer, iFlags, @FIocpRecv.Overlapped, nil);

3)工作线程处理接收到的客户端数据

3.1)工作线程是最繁忙，服务器上CPU的数量决定了工作线程的数量，CPU数量越多可以创建的工作线程数量也就越多。

TClientSocket = record
Lock: TCriticalSection;
SocketHandle: TSocketHandle;
IocpRecv: TIocpRecord; //投递请求结构体
IdleDT: TDateTime;
end;
PClientSocket = ^TClientSocket;

var
ClientSocket: PClientSocket;
IocpRecord: PIocpRecord;
iWorkCount: Cardinal;

GetQueuedCompletionStatus(FIocpHandle, iWorkCount, DWORD(ClientSocket), POverlapped(IocpRecord), INFINITE);

IOCP大致的流程就是这样，当然服务端发送数据给客户端的代码此处省略，相信读者根据接收代码已经知道怎么弄了。

 

IOCP为我们提供了一个系统级的消息队列（称之为完成队列），事件循环就是围绕着这个完成队列展开的。在发起IO操作后系统会进行异步处理（如果能立刻处理的话也会直接处理掉），当操作完成后自动向这个队列投递一条消息，不管是直接处理还是异步处理，最后总会投递完成消息。

IOCP完成队列返回的消息是一个OVERLAPPED结构体和一个ULONG_PTR complete_key。complete_key是在用户将Socket handle关联到IOCP的时候绑定的，其实用性不是很大，而OVERLAPPED结构体则是在用户发起IO操作的时候设置的，并且OVERLAPPED结构可以由用户通过继承的方式来扩展，因此如何用好OVERLAPPED结构在螺丝壳里做道场，就成了封装好IOCP的关键。