WSAAsyncSelect异步套接字模型Client——》Server

/////////////////////////////////////////////// // WSAAsyncSelect.cpp文件#include <winsock2.h>#pragma comment(lib, "WS2_32")// 链接到WS2_32.lib#include <stdio.h>#define WM_SOCKET WM_USER + 101// 自定义消息class CInitSock{public:CInitSock(BYTE minorVer = 2, BYTE majorVer = 2){// 初始化WS2_32.dllWSADATA wsaData;WORD sockVersion = MAKEWORD(minorVer, majorVer);if(::WSAStartup(sockVersion, &wsaData) != 0){exit(0);}}~CInitSock(){::WSACleanup();}};CInitSock theSock;LRESULT CALLBACK WindowProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);int main1();int main2();void main(){printf("Please select a item:/n");printf("s:Server/n");printf("c:Client/n");char a;while(1){scanf("%c",&a);if(a=='s'){main1();break;}else if(a=='c'){main2();break;}else continue;}}int main1(){char szClassName[] = "MainWClass";WNDCLASSEX wndclass;// 用描述主窗口的参数填充WNDCLASSEX结构wndclass.cbSize = sizeof(wndclass);wndclass.style = CS_HREDRAW|CS_VREDRAW;wndclass.lpfnWndProc = WindowProc;wndclass.cbClsExtra = 0;wndclass.cbWndExtra = 0;wndclass.hInstance = NULL;wndclass.hIcon = ::LoadIcon(NULL, IDI_APPLICATION);wndclass.hCursor = ::LoadCursor(NULL, IDC_ARROW);wndclass.hbrBackground = (HBRUSH)::GetStockObject(WHITE_BRUSH);wndclass.lpszMenuName = NULL;wndclass.lpszClassName = szClassName ;wndclass.hIconSm = NULL;::RegisterClassEx(&wndclass); // 创建主窗口HWND hWnd = ::CreateWindowEx( 0,szClassName,"",WS_OVERLAPPEDWINDOW,CW_USEDEFAULT,CW_USEDEFAULT,CW_USEDEFAULT,CW_USEDEFAULT,NULL,NULL,NULL,NULL);if(hWnd == NULL){::MessageBox(NULL, "创建窗口出错！", "error", MB_OK);return -1;}USHORT nPort = 4567;// 此服务器监听的端口号// 创建监听套节字SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);sockaddr_in sin;sin.sin_family = AF_INET;sin.sin_port = htons(nPort);sin.sin_addr.S_un.S_addr = INADDR_ANY;// 绑定套节字到本地机器if(::bind(sListen, (sockaddr*)&sin, sizeof(sin)) == SOCKET_ERROR){printf(" Failed bind() /n");return -1;}// 将套接字设为窗口通知消息类型。::WSAAsyncSelect(sListen, hWnd, WM_SOCKET, FD_ACCEPT|FD_CLOSE);// 进入监听模式::listen(sListen, 5);// 从消息队列中取出消息MSG msg;while(::GetMessage(&msg, NULL, 0, 0)){// 转化键盘消息::TranslateMessage(&msg);// 将消息发送到相应的窗口函数::DispatchMessage(&msg);}// 当GetMessage返回0时程序结束return msg.wParam;}LRESULT CALLBACK WindowProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam){switch (uMsg){case WM_SOCKET:{// 取得有事件发生的套节字句柄SOCKET s = wParam;// 查看是否出错if(WSAGETSELECTERROR(lParam)){::closesocket(s);return 0;}// 处理发生的事件switch(WSAGETSELECTEVENT(lParam)){case FD_ACCEPT:// 监听中的套接字检测到有连接进入{SOCKET client = ::accept(s, NULL, NULL);::WSAAsyncSelect(client, hWnd, WM_SOCKET, FD_READ|FD_WRITE|FD_CLOSE);}break;case FD_WRITE:{}break;case FD_READ:{char szText[1024] = { 0 };if(::recv(s, szText, 1024, 0) == -1)::closesocket(s);elseprintf("接收数据：%s/n", szText);}break;case FD_CLOSE:{ ::closesocket(s);}break;}}return 0;case WM_DESTROY:::PostQuitMessage(0) ;return 0 ;}// 将我们不处理的消息交给系统做默认处理return ::DefWindowProc(hWnd, uMsg, wParam, lParam);}//Clientint main2(){// 创建套节字SOCKET s = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);if(s == INVALID_SOCKET){printf(" Failed socket() /n");return 0;}// 也可以在这里调用bind函数绑定一个本地地址// 否则系统将会自动安排// 填写远程地址信息sockaddr_in servAddr; servAddr.sin_family = AF_INET;servAddr.sin_port = htons(4567);// 注意，这里要填写服务器程序（TCPServer程序）所在机器的IP地址// 如果你的计算机没有联网，直接使用127.0.0.1即可servAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");if(::connect(s, (sockaddr*)&servAddr, sizeof(servAddr)) == -1){printf(" Failed connect() /n");return 0;} char szText[256];while(TRUE){scanf("%s",szText);//puts(szText);// 向服务器端发送数据::send(s, szText, strlen(szText), 0);}// 关闭套节字::closesocket(s);return 0;}