基于WSAAsyncSelect模型的两台计算机之间的通信

任务目标

编写Win32程序模拟实现基于WSAAsyncSelect模型的两台计算机之间的通信，要求编程实现服务器端与客户端之间双向数据传递。客户端向服务器端发送“请输出从1到1000内所有的质数”，服务器回应客户端给出结果。

效果图

核心代码

服务器端：
#include <stdio.h>
#include <winsock2.h>
#pragma comment(lib, "WS2_32")  // 链接到WS2_32.lib
#include<math.h>
#define WM_SOCKET WM_USER + 101     // 自定义消息
class CInitSock 
{
    public:
    CInitSock(BYTE minorVer = 2, BYTE majorVer = 2)
    {
        // 初始化WS2_32.dll
        WSADATA wsaData;
        WORD sockVersion = MAKEWORD(minorVer, majorVer);
        if(::WSAStartup(sockVersion, &wsaData) != 0)
           return;
    }
    ~CInitSock()
    {   
        ::WSACleanup(); 
    }
};
CInitSock theSock;       //加载套接字库
LRESULT CALLBACK WindowProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
bool isprime(int p){//判断p是否为质数
   int sq=(int)sqrt(p);
   bool flag=true;
   int i;
   for(i=2;i<=sq;i++){
       if(p%i==0){
           flag=false;
           break;
       }
   }
   if(!flag&&i<=sq)
      return false;
   else 
      return true;

};
char * getallprime(int n){ //将num以内的所有质数放在同一字符串中
    char  szprime[4096] = "质数：" ;
    int len=strlen(szprime)+strlen(",");
    for(int i=2 ; i <= n ; i++ ){
        if(isprime(i)){
            char sznum[10];
            itoa(i,sznum,10);
            char * sztemp=strcat(szprime,sznum);
            len+=strlen(sznum);
            szprime[len-1]=',';
            len+=strlen(",");
        }
    }
    //printf("%s\n",szprime);
  return szprime;   
};
int main()
{
    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);
                    else
                    {
                        printf("接收数据：%s", szText);
                        char * szReply=getallprime(1000); //得到1000以内的所有质数
                        ::send(s, szReply, strlen(szReply), 0); //响应客户端，回以szReply
                    }
                }
                break;
            case FD_CLOSE:
                { 
                    ::closesocket(s);
                }
                break;
            }
        }
        return 0;
    case WM_DESTROY:
        ::PostQuitMessage(0) ;
        return 0 ;
    }

    // 将我们不处理的消息交给系统做默认处理
    return ::DefWindowProc(hWnd, uMsg, wParam, lParam);
}

客户端：
#include <stdio.h>
#include <winsock2.h>
#pragma comment(lib, "WS2_32")  // 链接到WS2_32.lib
class CInitSock 
{
    public:
    CInitSock(BYTE minorVer = 2, BYTE majorVer = 2)
    {
        // 初始化WS2_32.dll
        WSADATA wsaData;
        WORD sockVersion = MAKEWORD(minorVer, majorVer);
        if(::WSAStartup(sockVersion, &wsaData) != 0)
           return;
    }
    ~CInitSock()
    {   
        ::WSACleanup(); 
    }
};
CInitSock theSock;       //加载套接字库

int main()
{
    // 创建套节字
    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);
    //要连接的服务器地址
    servAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");//没有联网，直接使用127.0.0.1即可
    
    if(::connect(s, (sockaddr*)&servAddr, sizeof(servAddr)) == -1)
    {
        printf(" Failed connect() \n");
        return 0;
    }
        //发送数据
    char buf[] = "请输出从1到1000内所有的质数";
    printf("发送数据：%s\n",buf);
    send(s, buf, strlen(buf), 0);
    Sleep(6);
    // 接收数据
    char buff[3000];
    int nRecv = ::recv(s, buff, 3000, 0);
    if(nRecv > 0)
    {
        buff[nRecv] = '\0';
        printf("接收到数据：\n%s\n", buff);
    }
    
    // 关闭套节字
    ::closesocket(s);
    return 0;
}

转载于:https://www.cnblogs.com/boxker/p/10745502.html