1. 服务器端代码:
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
#include "stdafx.h"
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
#include <WINSOCK2.H>
#include <stdio.h>
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
#define PORT 5150
#define MSGSIZE 1024
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
#pragma comment(lib, "ws2_32.lib")
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
typedef enum
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedBlockStart.gif) ![](http://www.cppblog.com/Images/OutliningIndicators/ContractedBlock.gif) {
RECV_POSTED
}OPERATION_TYPE;
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
typedef struct
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedBlockStart.gif) ![](http://www.cppblog.com/Images/OutliningIndicators/ContractedBlock.gif) {
WSAOVERLAPPED overlap;
WSABUF Buffer;
char szMessage[MSGSIZE];
DWORD NumberOfBytesRecvd;
DWORD Flags;
OPERATION_TYPE OperationType;
}PER_IO_OPERATION_DATA, *LPPER_IO_OPERATION_DATA;
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
DWORD WINAPI WorkerThread(LPVOID);
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
int main()
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedBlockStart.gif) ![](http://www.cppblog.com/Images/OutliningIndicators/ContractedBlock.gif) {
WSADATA wsaData;
SOCKET sListen, sClient;
SOCKADDR_IN local, client;
DWORD i, dwThreadId;
int iaddrSize = sizeof(SOCKADDR_IN);
HANDLE CompletionPort = INVALID_HANDLE_VALUE;
SYSTEM_INFO systeminfo;
LPPER_IO_OPERATION_DATA lpPerIOData = NULL;
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Initialize Windows Socket library
WSAStartup(0x0202, &wsaData);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Create completion port
CompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Create worker thread
GetSystemInfo(&systeminfo);
for (i = 0; i < systeminfo.dwNumberOfProcessors; i++)
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedSubBlockStart.gif) {
CreateThread(NULL, 0, WorkerThread, CompletionPort, 0, &dwThreadId);
}
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Create listening socket
sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Bind
local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
local.sin_family = AF_INET;
local.sin_port = htons(PORT);
bind(sListen, (struct sockaddr *)&local, sizeof(SOCKADDR_IN));
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Listen
listen(sListen, 3);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
while (TRUE)
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedSubBlockStart.gif) {
// Accept a connection
sClient = accept(sListen, (struct sockaddr *)&client, &iaddrSize);
printf("Accepted client:%s:%d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Associate the newly arrived client socket with completion port
CreateIoCompletionPort((HANDLE)sClient, CompletionPort, (DWORD)sClient, 0);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Launch an asynchronous operation for new arrived connection
lpPerIOData = (LPPER_IO_OPERATION_DATA)HeapAlloc(
GetProcessHeap(),
HEAP_ZERO_MEMORY,
sizeof(PER_IO_OPERATION_DATA));
lpPerIOData->Buffer.len = MSGSIZE;
lpPerIOData->Buffer.buf = lpPerIOData->szMessage;
lpPerIOData->OperationType = RECV_POSTED;
WSARecv(sClient,
&lpPerIOData->Buffer,
1,
&lpPerIOData->NumberOfBytesRecvd,
&lpPerIOData->Flags,
&lpPerIOData->overlap,
NULL);
}
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
PostQueuedCompletionStatus(CompletionPort, 0xFFFFFFFF, 0, NULL);
CloseHandle(CompletionPort);
closesocket(sListen);
WSACleanup();
return 0;
}
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
DWORD WINAPI WorkerThread(LPVOID CompletionPortID)
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedBlockStart.gif) ![](http://www.cppblog.com/Images/OutliningIndicators/ContractedBlock.gif) {
HANDLE CompletionPort=(HANDLE)CompletionPortID;
DWORD dwBytesTransferred;
SOCKET sClient;
LPPER_IO_OPERATION_DATA lpPerIOData = NULL;
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
while (TRUE)
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedSubBlockStart.gif) {
GetQueuedCompletionStatus(
CompletionPort,
&dwBytesTransferred,
(PULONG_PTR)&sClient,
(LPOVERLAPPED *)&lpPerIOData,
INFINITE);
if (dwBytesTransferred == 0xFFFFFFFF)
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedSubBlockStart.gif) {
return 0;
}
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
if (lpPerIOData->OperationType == RECV_POSTED)
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedSubBlockStart.gif) {
if (dwBytesTransferred == 0)
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedSubBlockStart.gif) {
// Connection was closed by client
closesocket(sClient);
HeapFree(GetProcessHeap(), 0, lpPerIOData);
}
else
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedSubBlockStart.gif) {
lpPerIOData->szMessage[dwBytesTransferred] = '\0';
send(sClient, lpPerIOData->szMessage, dwBytesTransferred, 0);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
// Launch another asynchronous operation for sClient
memset(lpPerIOData, 0, sizeof(PER_IO_OPERATION_DATA));
lpPerIOData->Buffer.len = MSGSIZE;
lpPerIOData->Buffer.buf = lpPerIOData->szMessage;
lpPerIOData->OperationType = RECV_POSTED;
WSARecv(sClient,
&lpPerIOData->Buffer,
1,
&lpPerIOData->NumberOfBytesRecvd,
&lpPerIOData->Flags,
&lpPerIOData->overlap,
NULL);
}
}
}
return 0;
}
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
2. 客户端代码片断:
void CTestClientDlg::OnBnClickedOk()
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedBlockStart.gif) ![](http://www.cppblog.com/Images/OutliningIndicators/ContractedBlock.gif) {
// TODO: 在此添加控件通知处理程序代码
//CDialogEx::OnOK();
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
UpdateData(TRUE);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
byte b1,b2,b3,b4;
![](http://www.cppblog.com/Images/OutliningIndicators/ExpandedSubBlockStart.gif) char strIpAddress[20]= {0};
m_IpCtrl.GetAddress(b1,b2,b3,b4);
sprintf(strIpAddress,"%d.%d.%d.%d",b1,b2,b3,b4);
TCHAR buffer[1024];
sprintf(buffer,_T("%d"),m_port);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
SOCKET sockClient=socket(AF_INET,SOCK_STREAM,0);
SOCKADDR_IN addrSrv;
addrSrv.sin_addr.S_un.S_addr=inet_addr(strIpAddress);
addrSrv.sin_family=AF_INET;
addrSrv.sin_port=htons((UINT)m_port);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
//向服务器发出连接请求
connect(sockClient,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
char recvBuf[100];
//接收数据
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
send(sockClient,"This is lisi",strlen("This is lisi")+1,0);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
recv(sockClient,recvBuf,100,0);
printf("%s\n",recvBuf);
![](http://www.cppblog.com/Images/OutliningIndicators/InBlock.gif)
//关闭套接字
closesocket(sockClient);
}
![](http://www.cppblog.com/Images/OutliningIndicators/None.gif)
|