WinSock TCP 长连接(while循环)

Posted 2021-03-17 zpchcbd

前面的程序，不管服务器端还是客户端，都有一个问题，就是处理完一个请求立即退出了，没有太大的实际意义。能不能像Web服务器那样一直接受客户端的请求呢？能，使用 while 循环即可。

这里唯一需要注意的是，每次 客户端 进行连接完之后，都需要进行socketclose()的操作，原因是 服务端 调用 closesocket() 不仅会关闭服务器端的 socket，还会通知客户端连接已断开，所以客户端也需要直接进行socketclose()的操作

服务端的代码：

#include<winSock2.h>
#include<iostream>
#pragma comment(lib, "ws2_32.lib") //添加动态链接库

using namespace std;

int main(int argc, char * argv[]) {
    const int   BUFSIZE = 1024;
    SOCKET      ListeningSocket; //定义一个套接字变量
    SOCKET      NewConnection; //客户端的请求生成的新的套接字
    SOCKADDR_IN ClientAddr;
    SOCKADDR_IN ServerAddr;
    char        Message[BUFSIZE];
    int         ClientAddrLen;
    ZeroMemory(Message, BUFSIZE);

    if (argc <= 1)
    {
        cout << "USAGE: TcpServer <Listen Port>" << endl;
        return -1;
    }


    int ret; //用来检查初始化成功是否
    WSADATA wsaDATA; // 创建一个WSADATA 用来初始化套接字网络库

    if ((ret = WSAStartup(MAKEWORD(2, 2), &wsaDATA)) != 0) { // 初始化套接字网络库WinSock2.2版本
        cout << "WSAStartup初始化失败 with error " << ret << endl;
        return -1;
    }

    //创建套接字 AF_INET协议 SOCK_STREAM流 TCP协议
    if ((ListeningSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == SOCKET_ERROR) {
        cout << "创建套接字失败 with error " << WSAGetLastError() << endl;
        WSACleanup();
        return -1;
    }

    //绑定套接字
    ServerAddr.sin_family = AF_INET;
    ServerAddr.sin_port = htons(atoi(argv[1]));
    ServerAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);

    if (bind(ListeningSocket, (SOCKADDR *)&ServerAddr, sizeof(SOCKADDR)) == SOCKET_ERROR) {
        cout << "绑定套接字失败 with error " << WSAGetLastError() << endl;
        closesocket(ListeningSocket);
        WSACleanup();
        return -1;
    }

    //监听套接字
    if ((ret = listen(ListeningSocket, 1)) == SOCKET_ERROR) {
        cout << "监听套接字失败 with error " << WSAGetLastError() << endl;
        closesocket(ListeningSocket);
        WSACleanup();
        return -1;
    }
    cout << "正在监听端口" << argv[1] << "中..." << endl;


    //接收客户端数据
    ClientAddrLen = sizeof(SOCKADDR);
    while (true) {
        if ((NewConnection = accept(ListeningSocket, (SOCKADDR*)&ClientAddr, &ClientAddrLen)) == INVALID_SOCKET) {//创建一个当前客户端和服务端的套接字,accept会阻塞，后面的不能继续执行，直到有新的请求
            cout << "创建NewConnection失败 with error " << WSAGetLastError() << endl;
            closesocket(ListeningSocket);
            WSACleanup();
            return -1;
        }

        int StrLen = recv(NewConnection, Message, sizeof(Message), 0);
        cout << "客户端 > " << Message << endl;

        send(NewConnection, Message, StrLen, 0);
        closesocket(NewConnection);

        ZeroMemory(Message, BUFSIZE);
    }

    //关闭套接字
    closesocket(ListeningSocket);
    WSACleanup();


    return 0;
}

客户端的代码：

#include<WinSock2.h>
#include<iostream>
#pragma comment(lib, "ws2_32.lib") //添加动态链接库
#pragma warning(disable:4996) //忽略旧函数使用缺陷的警告

using namespace std;

int main(int argc, char * argv[]) {
    const int   BUFSIZE = 1024;
    SOCKET      ClientSocket;
    SOCKADDR_IN ServerAddr;
    char        SendBuf[BUFSIZE];  //发送存储的数据缓冲区
    char        BufRecv[BUFSIZE]; //接收收到的数据缓冲区

    ZeroMemory(SendBuf, BUFSIZE);
    ZeroMemory(BufRecv, BUFSIZE);
    strcpy(SendBuf, "Hello, My Name is Client");

    if (argc <= 2)
    {
        cout << "USAGE: TcpServer <Server IP> <Server PORT>" << endl;
        return -1;
    }

    WSADATA     WSAData;
    int         ret;
    if ((ret = WSAStartup(MAKEWORD(2, 2), &WSAData)) != 0) {
        cout << "WSAStartup初始化失败 with error " << ret << endl;
        return -1;
    }


    //cout << "连接建立成功!" << endl;;

    while (true) {
        //创建套接字
        if ((ClientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == SOCKET_ERROR) {
            cout << "创建套接字失败 with error " << WSAGetLastError() << endl;
            WSACleanup();
            return -1;
        }

        //连接connet服务端
        ServerAddr.sin_family = AF_INET;
        ServerAddr.sin_addr.s_addr = inet_addr(argv[1]);
        ServerAddr.sin_port = htons(atoi(argv[2]));
        if (connect(ClientSocket, (SOCKADDR *)&ServerAddr, sizeof(ServerAddr)) == SOCKET_ERROR) {
            cout << "连接服务端失败 with error " << WSAGetLastError() << endl;
            closesocket(ClientSocket);
            WSACleanup();
            return -1;
        }

        cout << "客户端 > ";
        cin >> SendBuf;

        //send发送数据
        if ((ret = send(ClientSocket, SendBuf, strlen(SendBuf), 0)) == SOCKET_ERROR) {
            cout << "send failed with error " << WSAGetLastError() << endl;;
            closesocket(ClientSocket);
            WSACleanup();
            return -1;
        }

        recv(ClientSocket, BufRecv, BUFSIZ, 0);
        cout << "服务端 > " << BufRecv << endl;

        ZeroMemory(SendBuf, BUFSIZE);
        ZeroMemory(BufRecv, BUFSIZE);

        //关闭套接字
        closesocket(ClientSocket);
    }

    WSACleanup();
    return 0;


}