高性能的socket通讯服务器(完成端口模型--IOCP)
http://blog.csdn.net/GuestCode/archive/2009/07/20/4365243.aspx
关键词:delphi socket通讯服务器例程(完成端口模型--IOCP)、I/O Completion Port、socket通讯登峰造极、IO重叠、 IOCP客户服务端
高性能的socket通讯服务器(完成端口模型--IOCP)
很多人费尽心思,都没有找到一个完美的 I/O CP 例程,甚至跟人于误解,先将本人编写的例程公布出来,希望对那些苦苦寻觅的人带来收获。本例程可以作为初学者的学习之用,亦可以作为大型服务程序的通讯模块。其处理速度可以说,优化到了极点。如果理解了本例程的精髓,加上一个高效的通讯协议,你完全可以用它来构建一个高性能的通讯服务器。
在公布代码前,先谈谈I/O CP。对I/O CP的函数不多做说明了,网上很多,都一样。在此本人仅说一些技术上要注意的问题。
一、如何管理内存
1、IO数据缓冲管理
动态分配内存,是一种灵活的方式。但对于系统资源浪费是巨大的。因此本人采用的是预先分配服务器最大需要的内存,用链表来管理。任何时候分配交还都不需要遍历,仅需要互斥而已。
更巧妙的是,将IO发送信息和内存块有机的结合在一起,减少了链表的管理工作。
//IO操作标志
TIOFlag = (IO_ACCEPT, IO_READ, IO_WRITE);
//IO操作信息
PIOInfo =^ TIOInfo;
TIOInfo = packed record
Overlapped: TOverlapped; //重叠结构
DataBuf: TWSABUF; //IO数据信息
Socket: TSocket;
Flag: TIOFlag;
TickCountSend: DWord;
Next: PIOInfo;
Prior: PIOInfo;
end;
PUNode =^ TUNode;
TUNode = record
Next: Pointer;
end;
PIOMem =^ TIOMem;
TIOMem = packed record
IOInfo: TIOInfo;
Data: array[1..IO_MEM_SIZE] of Byte;
//申请内存的时候,返回的是Data的地址
end;
2、链路数据管理
采用双向链表结构,减少删除节点时遍历消耗的时间
//每个连接的信息
PLink =^ TLink;
TLink = record
Socket: TSocket;
RemoteIP: string[30];
RemotePort: DWord;
//最后收到数据时的系统节拍
TickCountActive: DWord;
//处理该连接的当前线程的信息
Worker: PWorker;
Data: Pointer; //应用层可以设置这个成员,当OnReceive的时候,就不要每次遍历每个连接对应的数据区了
Section: TRTLCriticalSection;
Next: PLink;
Prior: PLink;
end;
二、如何管理线程
每个工作线程创建的时候,调用:OnWorkerThreadCreateEvt,该函数可以返回这个线程对应的信息,比如为该线程创建的数据库连接控件或对应的类等,在OnReceive的可以从Link的Worker访问该成员Worker^.Data。
//工作线程信息
PWorker =^ TWorker;
TWorker = record
ID: THandle;
CompletionPort: THandle;
Data: Pointer; //调用OnWorkerThreadCreateEvt返回的值
//用于反应工作情况的数据
TickCountLong,
TickCountActive: DWord;
ExecCount: Integer;
//线程完成后设置
Finished: THandle;
Next: PWorker;
end;
同理,服务线程也是具有一样的特点。相见源码。
关于线程同步,一直是众多程序头疼的问题。在本例程中,尽量避免了过多的互斥,并有效地防止了死锁现象。用RTLCriticalSection,稍微不注意,就会造成死锁的灾难。哪怕是两行代码的差别,对多线程而言都是灾难的。在本例程中,对数据同步需要操作的是在维护链路链表方面上。服务线程需要计算哪个连接空闲超时了,工作线程需要处理断线情况,应用层主动发送数据时需要对该链路独占,否则一个在发送,一个在处理断线故障,就会发送冲突,导致灾难后果。
在本人的压力测试中,已经有效的解决了这个问题,应用层部分不需要做什么同步工作,可以安心的收发数据了。同时每个线程都支持了数据库连接。
三、到底要创建多少个工作线程合适
很多文章说,有N个CPU就创建N个线程,也有说N*2+2。最不喜欢说话不负责任的人了,本例程可以让刚入门 I/O CP 的人对它有更深入的了解。
例程测试结果:
四、该不该使用类
有人说,抛弃一切类,对于服务器而言,会为类付出很多代价,从我的观点看,为类付出代价的,主要是动态创建的原因。其实,类成员访问和结构成员访问一样,需要相对地址。如果都是预先创建的,两者没有多大的差别。本例程采用裸奔函数的方式,当然在应用层可以采用类来管理,很难想象,如果没有没有类,需要多做多少工作。
五、缺点
不能发大数据包,只能发不超过固定数的数据包。但对于小数据报而言,它将是优秀的。
时间原因,不能做太多的解释和对代码做太多的注释,需要例程源码的可以和本人联系,免费提供。QQ:48092788
例程源码:
http://d.download.csdn.net/down/1546336/guestcode
完成端口通讯服务模块源码:
{******************************************************************************
* UCode 系列组件、控件 *
* 作者:卢益贵 2003~2009 *
* 版权所有 任何未经授权的使用和销售,均保留追究法律责任的权力 *
* *
* UCode 系列由XCtrls-YCtrls-ICtrls-NCode系列演变而来 *
* QQ:48092788 luyigui.blog.gxsky.com *
******************************************************************************}
{******************************************************************************
完成端口模型的socket服务器
******************************************************************************}
unit UTcpServer;
interface
uses
Windows, Classes, UClasses, UWinSock2;
const
//每个IO缓冲区的大小
IO_MEM_SIZE = 2048;
//内存要足够用,可视情况设置
IO_MEM_MAX_COUNT = 1000 * 10;
//最大连接数
SOCK_MAX_COUNT = 3000;
//连接空闲实现,超过这个时间未收到客户端数据则关闭
SOCK_IDLE_OVERTIME = 60;
type
//工作线程信息
PWorker =^ TWorker;
TWorker = record
ID: THandle;
CompletionPort: THandle;
Data: Pointer;
//用于反应工作情况的数据
TickCountLong,
TickCountActive: DWord;
ExecCount: Integer;
//线程完成后设置
Finished: THandle;
Next: PWorker;
end;
//每个连接的信息
PLink =^ TLink;
TLink = record
Socket: TSocket;
RemoteIP: string[30];
RemotePort: DWord;
//最后收到数据时的系统节拍
TickCountActive: DWord;
//处理该连接的当前线程的信息
Worker: PWorker;
Data: Pointer;
Section: TRTLCriticalSection;
Next: PLink;
Prior: PLink;
end;
TOnLinkIdleOvertimeEvt = procedure(Link: PLink);
TOnDisconnectEvt = procedure(Link: PLink);
TOnReceiveEvt = function(Link: PLink; Buf: PByte; Len: Integer): Boolean;
TOnThreadCreateEvt = function(IsWorkerThread: Boolean): Pointer;
//取得链路链表使用情况X%
function GetLinkUse(): real;
//链路链表所占内存
function GetLinkSize(): Integer;
//当前链路数
function GetLinkCount(): Integer;
//空闲链路数
function GetLinkFree(): Integer;
//IO内存使用情况
function GetIOMemUse(): Real;
//IO内存链表占内存数
function GetIOMemSize(): Integer;
//IO内存空闲数
function GetIOMemFree(): Integer;
//交还一个IO内存
procedure FreeIOMem(Mem: Pointer);
//获取一个IO内存区
function GetIOMem(): Pointer;
//获取工作线程的工作情况
function GetWorkerExecInfo(Index: Integer; var TickCount: DWord): Integer;
//获取工作线程的ID
function GetWorkerID(Index: Integer): Integer;
//获取工作线程数量
function GetWorkerCount(): Integer;
//打开一个IP端口,并监听
function StartTcpServer(RemoteIP: String; RemotePort: DWord): Boolean;
//停止并关闭一个IP端口
function StopTcpServer(): Boolean;
//设置响应事件的函数指针,在StartTcpServer之前调用
procedure SetEventProc(OnReceive: TOnReceiveEvt;
OnDisconnect: TOnDisconnectEvt;
OnLinkIdleOvertime: TOnLinkIdleOvertimeEvt;
OnServerThreadCreate: TOnThreadCreateEvt;
OnWorkerThreadCreate: TOnThreadCreateEvt);
//写日志文件
procedure WriteLog(Log: String);
function PostRecv(Link: PLink; IOMem: Pointer): Boolean;
//抛出一个发送事件
function PostSend(Link: PLink; IOMem: Pointer; Len: Integer): Boolean;
//广播数据到所有的链路对方
procedure PostBroadcast(Buf: PByte; Len: Integer);
//当前是否打开
function IsTcpServerActive(): Boolean;
//获取服务线程最后一次工作所占的时间(MS)
function GetServerExecLong(): DWord;
//获取服务线程工作次数
function GetServerExecCount(): Integer;
//获取本地或对外IP地址
function GetLocalIP(IsIntnetIP: Boolean): String;
implementation
uses
IniFiles, SysUtils, ActiveX;
var
ExePath: String = '';
const
HEAP_NO_SERIALIZE = 1; {非互斥, 此标记可允许多个线程同时访问此堆}
HEAP_GENERATE_EXCEPTIONS = 4; {当建立堆出错时, 此标记可激发一个异常并返回异常标识}
HEAP_ZERO_MEMORY = 8; {把分配的内存初始化为 0}
HEAP_REALLOC_IN_PLACE_ONLY = 16; {此标记不允许改变原来的内存位置}
STATUS_ACCESS_VIOLATION = DWORD($C0000005); {参数错误}
STATUS_NO_MEMORY = DWORD($C0000017); {内存不足}
{===============================================================================
IO内存管理
================================================================================}
type
//IO操作标志
TIOFlag = (IO_ACCEPT, IO_READ, IO_WRITE);
//IO操作信息
PIOInfo =^ TIOInfo;
TIOInfo = packed record
Overlapped: TOverlapped; //重叠结构
DataBuf: TWSABUF; //IO数据信息
Socket: TSocket;
Flag: TIOFlag;
TickCountSend: DWord;
Next: PIOInfo;
Prior: PIOInfo;
end;
PUNode =^ TUNode;
TUNode = record
Next: Pointer;
end;
PIOMem =^ TIOMem;
TIOMem = packed record
IOInfo: TIOInfo;
Data: array[1..IO_MEM_SIZE] of Byte;
end;
var
IOMemHead: PIOMem = nil;
IOMemLast: PIOMem = nil;
IOMemUse: Integer = 0;
IOMemSec: TRTLCriticalSection;
IOMemList: array[1..IO_MEM_MAX_COUNT] of Pointer;
function GetIOMem(): Pointer;
begin
//内存要足够用,如果不够,即使是动态分配,神仙也救不了
EnterCriticalSection(IOMemSec);
try
try
Result := @(IOMemHead^.Data);
IOMemHead := PUNode(IOMemHead)^.Next;
IOMemUse := IOMemUse + 1;
except
Result := nil;
WriteLog('GetIOMem: error');
end;
finally
LeaveCriticalSection(IOMemSec);
end;
end;
procedure FreeIOMem(Mem: Pointer);
begin
EnterCriticalSection(IOMemSec);
try
try
Mem := Pointer(Integer(Mem) - sizeof(TIOInfo));
PUNode(Mem).Next := nil;
PUNode(IOMemLast)^.Next := Mem;
IOMemLast := Mem;
IOMemUse := IOMemUse - 1;
except
WriteLog('FreeIOMem: error');
end;
finally
LeaveCriticalSection(IOMemSec);
end;
end;
procedure IniIOMem();
var
i: Integer;
Heap: THandle;
begin
InitializeCriticalSection(IOMemSec);
IOMemHead := HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(TIOMem));
IOMemLast := IOMemHead;
IOMemList[1] := IOMemHead;
Heap := GetProcessHeap();
for i := 2 to IO_MEM_MAX_COUNT do
begin
PUNode(IOMemLast)^.Next := HeapAlloc(Heap, HEAP_ZERO_MEMORY, sizeof(TIOMem));
IOMemList[i] := PUNode(IOMemLast)^.Next;
IOMemLast := PUNode(IOMemLast)^.Next;
end;
PUNode(IOMemLast).Next := nil;
end;
function GetIOMemFree(): Integer;
var
IOMems: PUNode;
begin
EnterCriticalSection(IOMemSec);
Result := 0;
IOMems := PUNode(IOMemHead);
while IOMems <> nil do
begin
Result := Result + 1;
IOMems := IOMems^.Next;
end;
LeaveCriticalSection(IOMemSec);
end;
procedure DeleteIOMem();
var
i: Integer;
Heap: THandle;
begin
Heap := GetProcessHeap();
for i := 1 to IO_MEM_MAX_COUNT do
HeapFree(Heap, HEAP_NO_SERIALIZE, IOMemList[i]);
IOMemUse := 0;
DeleteCriticalSection(IOMemSec);
end;
function GetIOMemSize(): Integer;
begin
Result := IO_MEM_MAX_COUNT * sizeof(TIOMem);
end;
function GetIOMemUse(): Real;
begin
Result := (IOMemUse * 100) / IO_MEM_MAX_COUNT;
end;
{===============================================================================
Socket链路管理
================================================================================}
procedure OnLinkIdleOvertimeDef(Link: PLink);
begin
end;
var
LinkHead: PLink = nil;
LinkLast: PLink = nil;
LinkUse: Integer = 0;
LinkCount: Integer = 0;
LinkSec: TRTLCriticalSection;
LinkList: array[1..SOCK_MAX_COUNT] of PLink;
OnLinkIdleOvertimeEvt: TOnLinkIdleOvertimeEvt = OnLinkIdleOvertimeDef;
LinksHead: PLink = nil;
LinksLast: PLink = nil;
function GetLinkFree(): Integer;
var
Links: PLink;
begin
EnterCriticalSection(LinkSec);
Result := 0;
Links := LinkHead;
while Links <> nil do
begin
Result := Result + 1;
Links := Links^.Next;
end;
LeaveCriticalSection(LinkSec);
end;
function GetLink(): PLink;
begin
try
//内存要足够用,如果不够,即使是动态分配,神仙也救不了
Result := LinkHead;
LinkHead := LinkHead^.Next;
LinkUse := LinkUse + 1;
LinkCount := LinkCount + 1;
if LinksHead = nil then
begin
LinksHead := Result;
LinksHead^.Next := nil;
LinksHead^.Prior := nil;
LinksLast := LinksHead;
end else
begin
Result^.Prior := LinksLast;
LinksLast^.Next := Result;
LinksLast := Result;
LinksLast^.Next := nil;
end;
with Result^ do
begin
Socket := INVALID_SOCKET;
RemoteIP := '';
RemotePort := 0;
TickCountActive := GetTickCount();
Worker := nil;
Data := nil;
end;
except
Result := nil;
WriteLog('GetLink: error');
end;
end;
procedure FreeLink(Link: PLink);
begin
try
with Link^ do
begin
Link^.Worker := nil;
if Link = LinksHead then
begin
LinksHead := Next;
if LinksLast = Link then
LinksLast := LinksHead
else
LinksHead^.Prior := nil;
end else
begin
Prior^.Next := Next;
if Next <> nil then
Next^.Prior := Prior;
if Link = LinksLast then
LinksLast := Prior;
end;
Next := nil;
LinkLast^.Next := Link;
LinkLast := Link;
LinkUse := LinkUse - 1;
LinkCount := LinkCount - 1;
end;
except
WriteLog('FreeLink: error');
end;
end;
procedure CloseLink(Link: PLink);
begin
EnterCriticalSection(LinkSec);
with Link^ do
begin
EnterCriticalSection(Section);
if Socket <> INVALID_SOCKET then
begin
try
CloseSocket(Socket);
except
WriteLog('CloseSocket: error');
end;
Socket := INVALID_SOCKET;
FreeLink(Link);
end;
LeaveCriticalSection(Link^.Section);
end;
LeaveCriticalSection(LinkSec);
end;
procedure CheckLinkLinkIdleOvertime(Data: Pointer);
var
TickCount: DWord;
Long: Integer;
Link: PLink;
begin
EnterCriticalSection(LinkSec);
try
TickCount := GetTickCount();
Link := LinksHead;
while Link <> nil do
with Link^ do
begin
EnterCriticalSection(Section);
if Socket <> INVALID_SOCKET then
begin
if TickCount > TickCountActive then
Long := TickCount - TickCountActive
else
Long := $FFFFFFFF - TickCountActive + TickCount;
if SOCK_IDLE_OVERTIME * 1000 < Long then
begin
try
CloseSocket(Socket);
except
WriteLog('CloseSocket overtime: error');
end;
Socket := INVALID_SOCKET;
Worker := Data;
try
OnLinkIdleOvertimeEvt(Link);
except
WriteLog('OnLinkIdleOvertimeEvt: error');
end;
Worker := nil;
FreeLink(Link);
LeaveCriticalSection(Section);
break;
end;
end else
begin
LeaveCriticalSection(Section);
break;
end;
LeaveCriticalSection(Section);
Link := Link^.Next;
end;
except
WriteLog('CheckLinkLinkIdleOvertime: error');
end;
LeaveCriticalSection(LinkSec);
end;
procedure IniLink();
var
i: Integer;
Heap: THandle;
begin
InitializeCriticalSection(LinkSec);
LinkHead := HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(TLink));
InitializeCriticalSection(LinkHead^.Section);
LinkLast := LinkHead;
LinkList[1] := LinkHead;
Heap := GetProcessHeap();
for i := 2 to SOCK_MAX_COUNT do
begin
LinkLast^.Next := HeapAlloc(Heap, HEAP_ZERO_MEMORY, sizeof(TLink));
LinkLast := LinkLast^.Next;
InitializeCriticalSection(LinkLast^.Section);
LinkList[i] := LinkLast;
end;
LinkLast.Next := nil;
end;
procedure DeleteLink();
var
i: Integer;
Heap: THandle;
begin
Heap := GetProcessHeap();
for i := 1 to SOCK_MAX_COUNT do
begin
DeleteCriticalSection(LinkList[i]^.Section);
HeapFree(Heap, HEAP_NO_SERIALIZE, LinkList[i]);
end;
LinkUse := 0;
LinkCount := 0;
LinksHead := nil;
LinksLast := nil;
DeleteCriticalSection(LinkSec);
end;
function GetLinkSize(): Integer;
begin
Result := SOCK_MAX_COUNT * sizeof(TLink);
end;
function GetLinkUse(): real;
begin
Result := (LinkUse * 100) / SOCK_MAX_COUNT;
end;
function GetLinkCount(): Integer;
begin
Result := LinkCount;
end;
{===============================================================================
工作线程
================================================================================}
procedure OnDisconnectDef(Link: PLink);
begin
end;
function OnReceiveDef(Link: PLink; Buf: PByte; Len: Integer): Boolean;
var
IOMem: Pointer;
i: Integer;
begin
Result := True;
IOMem := GetIOMem();
CopyMemory(IOMem, Buf, Len);
i := 1000000;
while i > 0 do
i := i - 1;
if not PostSend(Link, IOMem, Len) then
FreeIOMem(IOMem);
end;
function OnWorkerThreadCreateDef(IsWorkerThread: Boolean): Pointer;
begin
Result := nil;
end;
var
WorkerHead: PWorker = nil;
WorkerCount: Integer = 0;
OnDisconnectEvt: TOnDisconnectEvt = OnDisconnectDef;
OnReceiveEvt: TOnReceiveEvt = OnReceiveDef;
OnWorkerThreadCreateEvt: TOnThreadCreateEvt = OnWorkerThreadCreateDef;
function GetWorkerCount(): Integer;
begin
Result := WorkerCount;
end;
function WorkerThread(Worker: PWorker): DWORD; stdcall;
var
Link: PLink;
IOInfo: PIOInfo;
Bytes: DWord;
CompletionPort: THandle;
begin
Result := 0;
CompletionPort := Worker^.CompletionPort;
with Worker^ do
begin
TickCountActive := GetTickCount();
TickCountLong := 0;
ExecCount := 0;
end;
WriteLog(Format('Worker thread:%d begin', [Worker^.ID]));
CoInitialize(nil);
try
while True do
begin
try
with Worker^ do
TickCountLong := TickCountLong + GetTickCount() - TickCountActive;
if GetQueuedCompletionStatus(CompletionPort, Bytes, DWORD(Link), POverlapped(IOInfo), INFINITE) = False then
begin
if (Link <> nil) then
with Link^ do
begin
EnterCriticalSection(LinkSec);
EnterCriticalSection(Section);
if Link^.Socket <> INVALID_SOCKET then
begin
try
CloseSocket(Socket);
except
WriteLog(Format('CloseSocket1:%d error', [Worker^.ID]));
end;
Socket := INVALID_SOCKET;
Link^.Worker := Worker;
try
OnDisconnectEvt(Link);
except
WriteLog(Format('OnDisconnectEvt1:%d error', [Worker^.ID]));
end;
Link^.Worker := nil;
FreeLink(Link);
end;
LeaveCriticalSection(Section);
LeaveCriticalSection(LinkSec);
end;
if IOInfo <> nil then
FreeIOMem(IOInfo^.DataBuf.buf);
WriteLog(Format('GetQueuedCompletionStatus:%d error', [Worker^.ID]));
continue;
end;
with Worker^ do
begin
TickCountActive := GetTickCount();
ExecCount := ExecCount + 1;
end;
if (Bytes = 0) then
begin
if (Link <> nil) then
with Link^ do
begin
EnterCriticalSection(LinkSec);
EnterCriticalSection(Section);
if Link^.Socket <> INVALID_SOCKET then
begin
try
CloseSocket(Socket);
except
WriteLog(Format('CloseSocket2:%d error', [Worker^.ID]));
end;
Socket := INVALID_SOCKET;
Link^.Worker := Worker;
try
OnDisconnectEvt(Link);
except
WriteLog(Format('OnDisconnectEvt2:%d error', [Worker^.ID]));
end;
Link^.Worker := nil;
FreeLink(Link);
end;
LeaveCriticalSection(Section);
LeaveCriticalSection(LinkSec);
if IOInfo.Flag = IO_WRITE then
FreeIOMem(IOInfo^.DataBuf.buf)
else
FreeIOMem(IOInfo^.DataBuf.buf);
continue;
end else
begin
if IOInfo <> nil then
FreeIOMem(IOInfo^.DataBuf.buf);
break;
end;
end;
if IOInfo.Flag = IO_WRITE then
begin
FreeIOMem(IOInfo^.DataBuf.buf);
continue;
end;
{if IOInfo.Flag = IO_ACCEPT then
begin
......
continue;
end;}
with Link^, IOInfo^.DataBuf do
begin
Link^.Worker := Worker;
try
OnReceiveEvt(Link, buf, Bytes);
except
WriteLog(Format('OnReceiveEvt:%d error', [Worker^.ID]));
end;
Link^.Worker := nil;
TickCountActive := GetTickCount();
if not PostRecv(Link, buf) then
begin
EnterCriticalSection(LinkSec);
EnterCriticalSection(Section);
if Socket <> INVALID_SOCKET then
begin
try
CloseSocket(Socket);
except
WriteLog(Format('CloseSocket3:%d error', [Worker^.ID]));
end;
Socket := INVALID_SOCKET;
Link^.Worker := Worker;
try
OnDisconnectEvt(Link);
except
WriteLog(Format('OnDisconnectEvt3:%d error', [Worker^.ID]));
end;
Link^.Worker := nil;
FreeLink(Link);
end;
LeaveCriticalSection(Section);
LeaveCriticalSection(LinkSec);
FreeIOMem(buf);
end;
end;
except
WriteLog(Format('Worker thread:%d error', [Worker^.ID]));
end;
end;
finally
CoUninitialize();
WriteLog(Format('Worker thread:%d end', [Worker^.ID]));
SetEvent(Worker^.Finished);
end;
end;
procedure CreateWorkerThread(CompletionPort: THandle);
var
Worker, Workers: PWorker;
i: Integer;
SystemInfo: TSystemInfo;
ThreadHandle: THandle;
begin
GetSystemInfo(SystemInfo);
Workers := nil;
WorkerCount := (SystemInfo.dwNumberOfProcessors * 2 + 2);
for i := 1 to WorkerCount do
begin
Worker := HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(TWorker));
if Workers = nil then
begin
Workers := Worker;
WorkerHead := Workers;
end else
begin
Workers^.Next := Worker;
Workers := Worker;
end;
Worker^.CompletionPort := CompletionPort;
Worker^.Data := OnWorkerThreadCreateEvt(False);
Worker^.Finished := CreateEvent(nil, True, False, nil);
ThreadHandle := CreateThread(nil, 0, @WorkerThread, Worker, 0, Worker^.ID);
if ThreadHandle <> 0 then
CloseHandle(ThreadHandle);
end;
Workers^.Next := nil;
end;
procedure DestroyWorkerThread();
var
Worker, Save: PWorker;
begin
WorkerCount := 0;
Worker := WorkerHead;
while Worker <> nil do
begin
PostQueuedCompletionStatus(Worker^.CompletionPort, 0, 0, nil);
Worker := Worker^.Next;
end;
Worker := WorkerHead;
while Worker <> nil do
begin
with Worker^ do
begin
WaitForSingleObject(Worker^.Finished, INFINITE);
CloseHandle(Worker^.Finished);
Save := Worker^.Next;
end;
HeapFree(GetProcessHeap(), HEAP_NO_SERIALIZE, Worker);
Worker := Save;
end;
end;
function GetWorkerExecInfo(Index: Integer; var TickCount: DWord): Integer;
var
Worker: PWorker;
Count: Integer;
begin
Worker := WorkerHead;
Count := 0;
Result := 0;
while Worker <> nil do
with Worker^ do
begin
Count := Count + 1;
if Count = Index then
begin
TickCount := TickCountLong;
TickCountLong := 0;
Result := Worker^.ExecCount;
break;
end;
Worker := Worker^.Next;
end;
end;
function GetWorkerID(Index: Integer): Integer;
var
Worker: PWorker;
Count: Integer;
begin
Worker := WorkerHead;
Count := 0;
while Worker <> nil do
begin
Count := Count + 1;
if Count = Index then
begin
Count := Worker^.ID;
break;
end;
Worker := Worker^.Next;
end;
Result := Count;
end;
{===============================================================================
服务线程
================================================================================}
function OnServerThreadCreateDef(IsWorkerThread: Boolean): Pointer;
begin
Result := nil;
end;
var
ListenSocket: TSocket = INVALID_SOCKET;
SocketEvent: THandle = WSA_INVALID_EVENT;
CompletionPort: THandle = 0;
Terminated: Boolean = False;
ServerThreadID: DWORD = 0;
ServerExecCount: Integer = 0;
ServerExecLong: DWord = 0;
OnServerThreadCreateEvt: TOnThreadCreateEvt = OnServerThreadCreateDef;
ServerFinished: THandle;
function GetServerExecCount(): Integer;
begin
Result := ServerExecCount;
end;
function GetServerExecLong(): DWord;
begin
Result := ServerExecLong;
ServerExecLong := 0;
end;
function ServerThread(Param: Pointer): DWORD; stdcall;
var
AcceptSocket: TSocket;
Addr: TSockAddrIn;
Len: Integer;
Link: PLink;
IOMem: Pointer;
bNodelay: Boolean;
TickCount: DWord;
WR: DWord;
begin
Result := 0;
CoInitialize(nil);
WriteLog('Server thread begin');
TickCount := GetTickCount();
try
while not Terminated do
begin
try
ServerExecLong := ServerExecLong + (GetTickCount() - TickCount);
WR := WaitForSingleObject(SocketEvent, 10000);
ServerExecCount := ServerExecCount + 1;
TickCount := GetTickCount();
if (WAIT_TIMEOUT = WR) then
begin
CheckLinkLinkIdleOvertime(Param);
continue;
end else
if (WAIT_FAILED = WR) then
begin
continue;
end else
begin
Len := SizeOf(TSockAddrIn);
AcceptSocket := WSAAccept(ListenSocket, @Addr, @Len, nil, 0);
if (AcceptSocket = INVALID_SOCKET) then
continue;
if LinkCount >= SOCK_MAX_COUNT then
begin
try
CloseSocket(AcceptSocket);
except
WriteLog('Link count over');
end;
continue;
end;
bNodelay := True;
if SetSockOpt(AcceptSocket, IPPROTO_TCP, TCP_NODELAY,
PChar(@bNodelay), sizeof(bNodelay)) = SOCKET_ERROR then
begin
try
CloseSocket(AcceptSocket);
except
WriteLog('SetSockOpt: error');
end;
continue;
end;
EnterCriticalSection(LinkSec);
Link := GetLink();
with Link^ do
begin
EnterCriticalSection(Section);
RemoteIP := inet_ntoa(Addr.sin_addr);
RemotePort := Addr.sin_port;
TickCountActive := GetTickCount();
Socket := AcceptSocket;
IOMem := GetIOMem();
if (CreateIoCompletionPort(AcceptSocket, CompletionPort, DWORD(Link), 0) = 0) or
(not PostRecv(Link, IOMem)) then
begin
try
CloseSocket(Socket);
except
WriteLog('CreateIoCompletionPort or PostRecv: error');
end;
Socket := INVALID_SOCKET;
FreeLink(Link);
FreeIOMem(IOMem);
end;
LeaveCriticalSection(Section);
end;
LeaveCriticalSection(LinkSec);
end;
except
WriteLog('Server thread error');
end;
end;
finally
CoUninitialize();
WriteLog('Server thread end');
SetEvent(ServerFinished);
end;
end;
function StartTcpServer(RemoteIP: String; RemotePort: DWord): Boolean;
var
NonBlock: Integer;
bNodelay: Boolean;
Addr: TSockAddrIn;
ThreadHandle: THANDLE;
begin
Result := ListenSocket = INVALID_SOCKET;
if not Result then
exit;
IniIOMem();
IniLink();
ListenSocket := WSASocket(AF_INET, SOCK_STREAM, 0, nil, 0, WSA_FLAG_OVERLAPPED);
Result := ListenSocket <> INVALID_SOCKET;
if not Result then
begin
DeleteLink();
DeleteIOMem();
exit;
end;
bNodelay := True;
NonBlock := 1;
Addr.sin_family := AF_INET;
Addr.sin_addr.s_addr := inet_addr(PChar(RemoteIP));
Addr.sin_port := htons(RemotePort);
Result := (SetSockOpt(ListenSocket, IPPROTO_TCP, TCP_NODELAY, PChar(@bNodelay), sizeof(bNodelay)) <> SOCKET_ERROR) and
(ioctlsocket(ListenSocket, Integer(FIONBIO), NonBlock) <> SOCKET_ERROR) and
(Bind(ListenSocket, @Addr, SizeOf(TSockAddrIn)) <> SOCKET_ERROR) and
(Listen(ListenSocket, SOMAXCONN) <> SOCKET_ERROR);
if not Result then
begin
ListenSocket := INVALID_SOCKET;
DeleteLink();
DeleteIOMem();
exit;
end;
SocketEvent := CreateEvent(nil, FALSE, FALSE, nil);
Result := (SocketEvent <> WSA_INVALID_EVENT);
if (not Result) then
begin
CloseSocket(ListenSocket);
ListenSocket := INVALID_SOCKET;
DeleteLink();
DeleteIOMem();
exit;
end;
Result := (WSAEventSelect(ListenSocket, SocketEvent, FD_ACCEPT) <> SOCKET_ERROR);
if not Result then
begin
CloseSocket(ListenSocket);
ListenSocket := INVALID_SOCKET;
WSACloseEvent(SocketEvent);
SocketEvent := WSA_INVALID_EVENT;
DeleteLink();
DeleteIOMem();
exit;
end;
CompletionPort := CreateIoCompletionPort(INVALID_HANDLE_value, 0, 0, 0);
Result := CompletionPort <> 0;
if not Result then
begin
CloseSocket(ListenSocket);
ListenSocket := INVALID_SOCKET;
WSACloseEvent(SocketEvent);
SocketEvent := WSA_INVALID_EVENT;
DeleteLink();
DeleteIOMem();
exit;
end;
WriteLog('Server Start');
CreateWorkerThread(CompletionPort);
ServerFinished := CreateEvent(nil, True, False, nil);
Result := ServerFinished <> 0;
if not Result then
begin
CloseSocket(ListenSocket);
ListenSocket := INVALID_SOCKET;
WSACloseEvent(SocketEvent);
SocketEvent := WSA_INVALID_EVENT;
DeleteLink();
DeleteIOMem();
exit;
end;
Terminated := False;
ThreadHandle := CreateThread(nil, 0, @ServerThread, OnServerThreadCreateEvt(False), 0, ServerThreadID);
if (ThreadHandle = 0) then
begin
StopTcpServer();
exit;
end;
CloseHandle(ThreadHandle);
end;
function StopTcpServer(): Boolean;
begin
Result := ListenSocket <> INVALID_SOCKET;
if not Result then
exit;
WriteLog('Server Stop');
Terminated := True;
if ServerFinished <> 0 then
begin
WaitForSingleObject(ServerFinished, INFINITE);
CloseHandle(ServerFinished);
ServerFinished := 0;
end;
if SocketEvent <> 0 then
WSACloseEvent(SocketEvent);
SocketEvent := 0;
DestroyWorkerThread();
if ListenSocket <> INVALID_SOCKET then
CloseSocket(ListenSocket);
ListenSocket := INVALID_SOCKET;
if CompletionPort <> 0 then
CloseHandle(CompletionPort);
CompletionPort := 0;
ServerExecCount := 0;
ServerExecLong := 0;
DeleteLink();
DeleteIOMem();
end;
function GetLocalIP(IsIntnetIP: Boolean): String;
type
TaPInAddr = Array[0..10] of PInAddr;
PaPInAddr = ^TaPInAddr;
var
phe: PHostEnt;
pptr: PaPInAddr;
Buffer: Array[0..63] of Char;
I: Integer;
begin
Result := '0.0.0.0';
try
GetHostName(Buffer, SizeOf(Buffer));
phe := GetHostByName(buffer);
if phe = nil then
Exit;
pPtr := PaPInAddr(phe^.h_addr_list);
if IsIntnetIP then
begin
I := 0;
while pPtr^[I] <> nil do
begin
Result := inet_ntoa(pptr^[I]^);
Inc(I);
end;
end else
Result := inet_ntoa(pptr^[0]^);
except
end;
end;
procedure SetEventProc(OnReceive: TOnReceiveEvt;
OnDisconnect: TOnDisconnectEvt;
OnLinkIdleOvertime: TOnLinkIdleOvertimeEvt;
OnServerThreadCreate: TOnThreadCreateEvt;
OnWorkerThreadCreate: TOnThreadCreateEvt);
begin
OnReceiveEvt := OnReceive;
OnDisconnectEvt := OnDisconnect;
OnLinkIdleOvertimeEvt := OnLinkIdleOvertime;
OnServerThreadCreateEvt := OnServerThreadCreate;
OnWorkerThreadCreateEvt := OnWorkerThreadCreate;
end;
function PostRecv(Link: PLink; IOMem: Pointer): Boolean;
var
Flags: DWord;
Bytes: DWord;
IOInfo: PIOInfo;
begin
Result := Link^.Socket <> INVALID_SOCKET;
if Result then
try
Flags := 0;
Bytes := 0;
IOInfo := PIOInfo(Integer(IOMem) - sizeof(TIOInfo));
with IOInfo^ do
begin
ZeroMemory(IOInfo, sizeof(TIOInfo));
DataBuf.buf := IOMem;
DataBuf.len := IO_MEM_SIZE;
Socket := Link^.Socket;
Flag := IO_READ;
Result := (WSARecv(Socket, @DataBuf, 1, @Bytes, @Flags, @Overlapped, nil) <> SOCKET_ERROR) or
(WSAGetLastError() = ERROR_IO_PENDING);
end;
except
Result := False;
WriteLog('PostRecv: error');
end;
end;
function PostSend(Link: PLink; IOMem: Pointer; Len: Integer): Boolean;
var
Bytes: DWord;
IOInfo: PIOInfo;
begin
Result := Link^.Socket <> INVALID_SOCKET;
if Result then
try
Bytes := 0;
IOInfo := PIOInfo(Integer(IOMem) - sizeof(TIOInfo));
with IOInfo^ do
begin
ZeroMemory(IOInfo, sizeof(TIOInfo));
DataBuf.buf := IOMem;
DataBuf.len := Len;
Socket := Link^.Socket;
Flag := IO_WRITE;
Result := (WSASend(Socket, @(DataBuf), 1, @Bytes, 0, @(Overlapped), nil) <> SOCKET_ERROR) or
(WSAGetLastError() = ERROR_IO_PENDING);
end;
except
Result := False;
WriteLog('PostSend: error');
end;
end;
procedure PostBroadcast(Buf: PByte; Len: Integer);
var
IOMem: Pointer;
Link: PLink;
begin
EnterCriticalSection(LinkSec);
Link := LinksHead;
while Link <> nil do
with Link^ do
begin
if Socket <> INVALID_SOCKET then
begin
IOMem := GetIOMem();
CopyMemory(IOMem, Buf, Len);
if not PostSend(Link, IOMem, Len) then
FreeIOMem(IOMem);
end;
Link := Link^.Next;
end;
LeaveCriticalSection(LinkSec);
end;
function IsTcpServerActive(): Boolean;
begin
Result := ListenSocket <> INVALID_SOCKET;
end;
{===============================================================================
日志管理
================================================================================}
var
LogSec: TRTLCriticalSection;
Inifile: TIniFile;
LogCount: Integer = 0;
LogName: String = '';
procedure WriteLog(Log: String);
begin
EnterCriticalSection(LogSec);
try
LogCount := LogCount + 1;
IniFile.WriteString(LogName,
'Index' + IntToStr(LogCount),
DateTimeToStr(Now()) + ':' + Log);
finally
LeaveCriticalSection(LogSec);
end;
end;
{===============================================================================
初始化Window Socket
================================================================================}
var
WSAData: TWSAData;
procedure Startup;
var
ErrorCode: Integer;
begin
ErrorCode := WSAStartup( {$SK_blogItemTitle$}
{$SK_ItemBody$}
{$SK_blogDiary$} {$SK_blogItemLink$} {$SK_blogItemComm$} {$SK_blogItemQuote$} {$SK_blogItemVisit$}
01, WSAData);
if ErrorCode <> 0 then
WriteLog('Window Socket init Error!');
end;
procedure Cleanup;
var
ErrorCode: Integer;
begin
ErrorCode := WSACleanup;
if ErrorCode <> 0 then
WriteLog('Window Socket cleanup error!');
end;
function GetExePath(): String;
var
ModuleName: array[0..1024] of char;
begin
GetModuleFileName(MainInstance, ModuleName, SizeOf(ModuleName));
Result := ExtractFilePath(ModuleName);
end;
initialization
LogName := DateTimeToStr(Now());
InitializeCriticalSection(LogSec);
ExePath := GetExePath();
IniFile := TIniFile.Create(ExePath + 'Logs.Ini');
Startup();
finalization
Cleanup();
DeleteCriticalSection(LogSec);
IniFile.Destroy();
end.
主窗口单元源码:
unit uMainTcpServerIOCP;
interface
uses
Windows, Messages, SysUtils, Variants, Classes, Graphics, Controls, Forms,
Dialogs, ExtCtrls, StdCtrls, ComCtrls, UTcpServer, Sockets, Grids;
type
TfrmMainUTcpServerIOCP = class(TForm)
Label1: TLabel;
Label2: TLabel;
edtIP: TEdit;
edtPort: TEdit;
btn: TButton;
Timer1: TTimer;
Label3: TLabel;
lbIO: TLabel;
Label5: TLabel;
lbIOU: TLabel;
Label7: TLabel;
lbL: TLabel;
Label9: TLabel;
lbLU: TLabel;
Label11: TLabel;
lbLS: TLabel;
Label13: TLabel;
lbW: TLabel;
Info: TStringGrid;
Label4: TLabel;
lbWC: TLabel;
Label8: TLabel;
lbWU: TLabel;
Label12: TLabel;
lbLF: TLabel;
Label15: TLabel;
lbLFL: TLabel;
Label6: TLabel;
lbIOF: TLabel;
lbIOFL: TLabel;
Label16: TLabel;
Timer2: TTimer;
procedure btnClick(Sender: TObject);
procedure FormCreate(Sender: TObject);
procedure Timer1Timer(Sender: TObject);
procedure FormDestroy(Sender: TObject);
procedure Timer2Timer(Sender: TObject);
private
{ Private declarations }
FTickCount: DWord;
public
{ Public declarations }
end;
var
frmMainUTcpServerIOCP: TfrmMainUTcpServerIOCP;
implementation
{$R *.dfm}
{ TfrmMainUTcpServerIOCP }
procedure TfrmMainUTcpServerIOCP.btnClick(Sender: TObject);
var
i: Integer;
C1: Integer;
C2: DWord;
DT: TDateTime;
begin
if btn.Caption = 'Open' then
begin
StartTcpServer(edtIP.Text, StrToInt(edtPort.Text));
if IsTcpServerActive() then
begin
FTickCount := GetTickCount();
Info.RowCount := GetWorkerCount() + 1;
DT := Now();
for i := 1 to Info.RowCount - 1 do
begin
Info.Cells[0, i] := IntToStr(i);
Info.Cells[1, i] := IntToStr(GetWorkerID(i));
C1 := GetWorkerExecInfo(i, C2);
Info.Cells[2, i] := IntToStr(C1);
Info.Cells[3, i] := '0';
Info.Cells[4, i] := IntToStr(C2);
Info.Cells[5, i] := '0';
Info.Cells[6, i] := DateTimeToStr(DT);
end;
Timer1.Enabled := True;
end;
end else
begin
Timer1.Enabled := False;
StopTcpServer();
end;
if IsTcpServerActive() then
btn.Caption := 'Close'
else
btn.Caption := 'Open';
end;
procedure TfrmMainUTcpServerIOCP.FormCreate(Sender: TObject);
begin
edtIP.Text := GetLocalIP(False);
Info.ColCount := 7;
Info.RowCount := 2;
Info.ColWidths[0] := 30;
Info.ColWidths[1] := 30;
Info.ColWidths[2] := 40;
Info.ColWidths[3] := 40;
Info.ColWidths[4] := 30;
Info.ColWidths[5] := 40;
Info.ColWidths[6] := 110;
Info.Cells[0, 0] := '序号';
Info.Cells[1, 0] := 'ID';
Info.Cells[2, 0] := '计数';
Info.Cells[3, 0] := '次/S';
Info.Cells[4, 0] := '时长';
Info.Cells[5, 0] := '使用率';
Info.Cells[6, 0] := '时间';
end;
procedure TfrmMainUTcpServerIOCP.Timer1Timer(Sender: TObject);
var
i: Integer;
Count1, Count2, Count3, TC, TCC: DWord;
begin
if not IsTcpServerActive() then
begin
Timer1.Enabled := False;
exit;
end;
TC := GetTickCount();
TCC := TC - FTickCount;
if TCC = 0 then
TCC := $FFFFFFFF;
lbWC.Caption := IntToStr(GetServerExecCount());
lbWU.Caption := FloatToStrF(GetServerExecLong() / TCC * 100, ffFixed, 10, 3) + '%';
for i := 1 to Info.RowCount - 1 do
begin
Count1 := GetWorkerExecInfo(i, Count2);
TC := GetTickCount();
TCC := TC - FTickCount;
if TCC = 0 then
TCC := $FFFFFFFF;
Count3 := StrToInt(Info.Cells[2, i]);
if Count1 <> Count3 then
begin
Info.Cells[2, i] := IntToStr(Count1);
Info.Cells[3, i] := IntToStr(Count1 - Count3);
Info.Cells[4, i] := IntToStr(Count2);
Info.Cells[5, i] := FloatToStrF(Count2 / TCC * 100, ffFixed, 10, 1) + '%';
Info.Cells[6, i] := DateTimeToStr(Now());
end;
end;
FTickCount := TC;
lbIO.Caption := IntToStr(GetIOMemSize());
lbIOU.Caption := FloatToStrF(GetIOMemUse(), ffFixed, 10, 3) + '%';
Count1 := GetIOMemFree();
lbIOF.Caption := IntToStr(Count1);
lbIOFL.Caption := FloatToStrF(Count1 / IO_MEM_MAX_COUNT * 100, ffFixed, 10, 3) + '%';
lbW.Caption := IntToStr(GetWorkerCount());
lbL.Caption := IntToStr(GetLinkSize());
Count1 := GetLinkFree();
lbLF.Caption := IntToStr(Count1);
lbLFL.Caption := FloatToStrF(Count1 / SOCK_MAX_COUNT * 100, ffFixed, 10, 3) + '%';
lbLU.Caption := FloatToStrF(GetLinkUse(), ffFixed, 10, 3) + '%';
lbLS.Caption := IntToStr(GetLinkCount());
end;
procedure TfrmMainUTcpServerIOCP.FormDestroy(Sender: TObject);
begin
StopTcpServer();
end;
procedure TfrmMainUTcpServerIOCP.Timer2Timer(Sender: TObject);
begin
if not IsTcpServerActive() then
begin
Timer1.Enabled := False;
exit;
end;
PostBroadcast(PByte(PChar('这是来自服务器的数据!')), 21);
end;
end.