//////////////////////////////////////////////////////////////////////////////////////////
//?? Iocp
實現文件
?
#include
"StdAfx.h"
#include
"iocp.h"
?
static
bool bInit = false;
?
DWORD __stdcall CIOCP::WorkThread(LPVOID Param)
{
???? CIOCP * pthis = (CIOCP *)Param;
?
???? void * re;
???? OVERLAPPED * pOverlap;
???? DWORD berByte;
???? while(pthis->m_workThread)
???? {
???????? int ret;
???????? ret = GetQueuedCompletionStatus(pthis->m_hIocp, &berByte, (LPDWORD)&re, (LPOVERLAPPED *)&pOverlap, INFINITE);
?
???????? if (ret == ERROR_SUCCESS)
???????? {
?
???????? }
?
???????? if (berByte == 0)
???????? {
????????????? //??
客戶端斷開連接
????????????? pthis->OnClose(re);
????????????? OVERLAPPEDPLUS *olp = (OVERLAPPEDPLUS *)pOverlap;
????????????? closesocket(olp->s);
????????????? delete olp;??????? //??
釋放 與socket綁定的結構體變量
????????????? continue;
???????? }
?
???????? if (re == NULL) return 0;
?
???????? OVERLAPPEDPLUS *olp = (OVERLAPPEDPLUS *)pOverlap;
?
???????? switch(olp->OpCode)
???????? {
???????? case OP_READ:
????????????? pthis->OnRead(re, olp->wbuf.buf, berByte);???? //??
調用 OnRead() 通知應用程序，服務器收到來自客戶端的網絡數據
????????????? WSARecv(olp->s, &olp->wbuf, 1, &olp->dwBytes, &olp->dwFlags, &olp->ol, NULL);?//??
繼續調用一個接收的 I/O 異步請求
????????????? break;
???????? default:
????????????? break;
???????? }
???? }
???? return 0;
}
?
DWORD __stdcall CIOCP::AcceptThread(LPVOID Param)
{
???? CIOCP * pthis = (CIOCP *)Param;
???? while(pthis->m_acceptThread)
???? {
???????? SOCKET client;
???????? if ((client= accept(pthis->m_sSocket, NULL, NULL)) == INVALID_SOCKET)
???????? {
????????????? //??
錯誤處理
???????? }
???????? pthis->OnAccept(client);??? //??
調用 OnAccept()通知應用程序有新客戶端連接
????????
???? }
???? return 1;
}
?
CIOCP::CIOCP(void)
{
}
?
CIOCP::~CIOCP(void)
{
}
?
bool
CIOCP::Init(void)
{
???? if (bInit)
???????? returntrue;
?
???? WSADATA wsd;
???? if (WSAStartup(MAKEWORD(2,2), &wsd) != 0)
???????? returnfalse;
?
???? bInit = true;
???? returntrue;
}
?
bool
CIOCP::Listen(int port)
{
???? if (!bInit)
???????? if (!Init())
????????????? returnfalse;
?
???? m_sSocket = socket(AF_INET, SOCK_STREAM, 0);
?
???? if (m_sSocket == INVALID_SOCKET)
???????? returnfalse;
?
???? //SOCKADDR_IN addr;
???? sockaddr_in addr;
???? addr.sin_family = AF_INET;
???? addr.sin_port = htons(port);
???? addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
???? //addr.sin_addr.S_un.S_addr = inet_addr(ip);
?
???? if (bind(m_sSocket, (struct sockaddr *)&addr, sizeof(addr)) == SOCKET_ERROR)
???????? returnfalse;
?
???? if (listen(m_sSocket, 10) == SOCKET_ERROR)
???????? returnfalse;
?
???? if ((m_hIocp = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0)) == NULL)???? //??
創建完成端口的句柄
???????? returnfalse;
?
???? this->m_acceptThread = true;
???? g_haThread = CreateThread(NULL, 0, AcceptThread, (LPVOID)this, 0, &m_athreadID);??? //??
創建連接線程，用來接收客戶端的連接
?
???? this->m_workThread = true;
???? g_hwThread = CreateThread(NULL, 0, WorkThread, (LPVOID)this, 0, &m_wthreadID); //??
創建工作線程，用來處理完成端口消息的
???? returntrue;
}
?
bool
CIOCP::SetIoCompletionPort(SOCKET socket, void *p, char *buf, int len)
{
???? if (CreateIoCompletionPort((HANDLE)socket, m_hIocp, (ULONG_PTR)p, 0) == NULL)
???????? returnfalse;
?
???? OVERLAPPEDPLUS *olp = new OVERLAPPEDPLUS;
???? memset(olp, 0, sizeof(OVERLAPPEDPLUS));
???? olp->s = socket;
???? if (buf)
???? {
???????? //??
這里可以使用用戶自定義的緩沖區地址，如果用戶不想設置，也可以采用默認分配的緩沖區
???????? olp->wbuf.buf = buf;
???????? olp->wbuf.len = len;
???? }
???? else
???? {
???????? olp->wbuf.buf = olp->buf;
???????? olp->wbuf.len = 4096;
???? }
???? olp->OpCode = OP_READ;
???? int ret = WSARecv(olp->s, &olp->wbuf, 1, &olp->dwBytes, &olp->dwFlags, &olp->ol, NULL);
???? if (ret == SOCKET_ERROR)
???????? if (WSAGetLastError() != ERROR_IO_PENDING)
????????????? returnfalse;
???? returntrue;
}
?
void
CIOCP::OnAccept(SOCKET socket)
{
???? this->SetIoCompletionPort(socket, NULL);
}
?
//===================================================================================
?
?
?
?
?
?
bool
CIOCPClient::Connect(char *ip, int port)
{
???????? //??
連接服務器
???? if (!bInit)
???????? if (!Init())
????????????? returnfalse;
?
???? //??
初始化連接socket
???? m_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
???? if (m_socket == SOCKET_ERROR)
???? {
//?????? printf("cocket Create fail");
???????? returnfalse;
???? }
?
???? //
填寫服務器地址信息
???? //
端口為1982
???? // IP
地址為INADDR_ANY，注意使用htonl將IP地址轉換為網絡格式ServerAddr.sin_family = AF_INET;
???? sockaddr_in ClientAddr;
???? ClientAddr.sin_family = AF_INET;
???? ClientAddr.sin_port = htons(port);???
???? ClientAddr.sin_addr.s_addr = inet_addr(ip);
?
???? //
綁定監聽端口
???? bind(m_socket, (SOCKADDR *)&ClientAddr, sizeof(ClientAddr));
?
???? if (connect(m_socket, (SOCKADDR *)&ClientAddr, sizeof(ClientAddr)) == SOCKET_ERROR)
???? {
???????? returnfalse;
???? }
????
???? if ((m_hIocp = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0)) == NULL)???? //??
創建完成端口的句柄
???????? returnfalse;
?
???? this->m_workThread = true;
???? g_hwThread = CreateThread(NULL, 0, WorkThread, (LPVOID)this, 0, &m_wthreadID); //??
創建工作線程，用來處理完成端口消息的
?
???? this->SetIoCompletionPort(m_socket, &m_socket);??? //??
設置完成端口監聽的socket
???? returntrue;
}
?
void
CIOCPClient::Send(char *buf, int len)
{
???? send(m_socket, buf, len, 0);
}
?
?
?
?
?
?
?
?
?
///////////////////////////////////////////////////////////////////////////////////
// IOCPclient
應用代碼
?
#include
"stdafx.h"
#include
"IOCP.h"
#include
"TClientSocket.h"
class
Iocp :public CIOCPClient
{
???? ?void OnRead(void * p, char *buf, int len)
???? ?{
???????? ?printf(buf);
???????? ?Sleep(1000);
???????? ?this->Send(buf, len);
 | | 文件: | IOCP的簡單類封裝_源文件.rar | | 大小: | 3KB | | 下載: | 下載 |
|
|
|
| 文件: |
IOCP的簡單應用實例.rar |
| 大小: |
318KB |
| 下載: |
下載
|
|
???? ?}
};
?
int
_tmain(int argc, _TCHAR* argv[])
{
???? Iocp iocp;
???? iocp.Init();
???? iocp.Connect("127.0.0.1", 4311);
???? iocp.Send("test\0", 5);
????
???? gets(newchar[1000]);
???? return 0;
}