基于Windows的多人聊天采用多线程

服务端代码

#include <stdio.h>
#include <windows.h>
#include <iostream>
#include <process.h> // 开启线程函数库

#pragma comment(lib, "ws2_32.lib") // 编译器指令，用于链接到需要用到的库文件
#define MAX_CLNT 256
#define MAX_BUF_SIZE 1024

SOCKET clntSocks[MAX_CLNT]; 
HANDLE hMutex;
int Clntcount = 0; // 当前连接数量

// 发送信息函数
void sendMsg(char* message, int len) {
    WaitForSingleObject(hMutex, INFINITE);
    for (int i = 0; i < Clntcount; i++) {
        send(clntSocks[i], message, len, 0);
    }
    ReleaseMutex(hMutex);
}

// 处理客户端连接的函数
unsigned __stdcall HandleCln(void* arg) {
    // 接受传递过来的参数
    SOCKET hclntSock = *((SOCKET*)arg);
    int Len = 0;
    char Message[MAX_BUF_SIZE] = { 0 };
    while (1) {
        Len = recv(hclntSock, Message, sizeof(Message), 0);
        if (Len != -1) {
            // 发送给每一个客户端
            sendMsg(Message, Len);
        }
        else {
            break;  // ==-1代表客户端已经关闭，跳出等待
        }
    }
    WaitForSingleObject(hMutex, INFINITE);
    for (int i = 0; i < Clntcount; i++) {
        if (hclntSock == clntSocks[i]) {
            while (i++ < Clntcount) {
                clntSocks[i] = clntSocks[i+1];
            }
            break;
        }
    }
    Clntcount--;
    ReleaseMutex(hMutex);
    closesocket(hclntSock);
    return 0;
}
int main() {
    // 加载套接字库
    WORD wVersionRequested;
    WSADATA wsaData;
    int err;
    HANDLE hThread;
    wVersionRequested = MAKEWORD(1, 1);
    // 初始化套接字库
    err = WSAStartup(wVersionRequested, &wsaData); // 打开网络库、启动网络库
    if (err != 0)
    {
        return err;
    }

    if (LOBYTE(wsaData.wVersion) != 1 || HIBYTE(wsaData.wVersion) != 1)
    {
        WSACleanup();
        return -1;
    }
    // 新建套接字
    SOCKET sockSrv = socket(AF_INET, SOCK_STREAM, 0);

    SOCKADDR_IN addrSrv;
    addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
    addrSrv.sin_family = AF_INET;
    addrSrv.sin_port = htons(6000);

    // 绑定套接字到本地IP地址，端口号6000
    if (bind(sockSrv, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR)) == SOCKET_ERROR) {
        printf("bind ERRORnum = %d\n", GetLastError());
        return -1;
    }

    // 开始监听
    if (listen(sockSrv, 5) == SOCKET_ERROR) {
        printf("listen ERRORnum = %d\n", GetLastError());
        return -1;
    }

    SOCKADDR_IN addrCli;
    int len = sizeof(SOCKADDR);

    while (true)
    {
        SOCKET sockConn = accept(sockSrv, (sockaddr*)&addrCli, &len);

        WaitForSingleObject(hMutex, INFINITE);
        clntSocks[Clntcount++] = sockConn;
        // 分配线程去处理每个连接的客户端
        hThread = (HANDLE)_beginthreadex(NULL, 0, HandleCln, (void*) & sockConn, 0, NULL);
        printf("Connect client IP: %s \n", inet_ntoa(addrCli.sin_addr));
        printf("Connect client num: %d \n", Clntcount);
        ReleaseMutex(hMutex);
    }
    closesocket(sockSrv);
    WSACleanup();
}

客户端代码

#include <WinSock2.h>
#include <iostream>
#include <windows.h>
#include <process.h>

#pragma comment(lib, "ws2_32.lib")

#define NAME_SIZE 32
#define BUF_SIZE 256

char szName[NAME_SIZE] = "[DEFAULT]";
char szMsg[BUF_SIZE];

//发送消息给服务端
unsigned WINAPI SendMsg(void* arg)
{
	//1 接收传递过来的参数
	SOCKET hClntSock = *((SOCKET*)arg);
	char szNameMsg[NAME_SIZE + BUF_SIZE];  //又有名字，又有消息
	//循环接收来自于控制台的消息
	while (1)
	{
		fgets(szMsg, BUF_SIZE, stdin); //阻塞在这一句
		//退出机制  当收到q或Q  退出
		if (!strcmp(szMsg, "Q\n") || !strcmp(szMsg, "q\n"))
		{
			closesocket(hClntSock);
			exit(0);
		}

		sprintf(szNameMsg, "%s %s", szName, szMsg);//字符串拼接
		send(hClntSock, szNameMsg, strlen(szNameMsg), 0);//发送
	}
	return 0;
}

//接收服务端的消息
unsigned WINAPI RecvMsg(void* arg)
{
	//1 接收传递过来的参数
	SOCKET hClntSock = *((SOCKET*)arg);
	char szNameMsg[NAME_SIZE + BUF_SIZE];  //又有名字，又有消息
	int iLen = 0;
	while (1)
	{
		//recv阻塞
		iLen = recv(hClntSock, szNameMsg, NAME_SIZE + BUF_SIZE - 1, 0);
		//服务端断开
		if (iLen == -1)
		{
			return -1;
		}
		// szNameMsg的0到iLen -1 都是收到的数据 iLen个
		szNameMsg[iLen] = 0;
		//接收到的数据输出到控制台
		fputs(szNameMsg, stdout);
	}
	return 0;
}

// 带参数的main函数，用命令行启动  在当前目录按下shift + 鼠标右键 cmd
int main(int argc, char* argv[])
{
	// 加载套接字库
	WORD wVersionRequested;
	WSADATA wsaData;
	int err;
	SOCKET hSock;
	SOCKADDR_IN servAdr;
	HANDLE hSendThread, hRecvThread;
	wVersionRequested = MAKEWORD(1, 1);
	// 初始化套接字库
	err = WSAStartup(wVersionRequested, &wsaData);
	if (err != 0)
	{
		return err;
	}
	if (LOBYTE(wsaData.wVersion) != 1 || HIBYTE(wsaData.wVersion) != 1)
	{
		WSACleanup();
		return -1;
	}
	sprintf(szName, "[%s]", argv[1]);
	//1 建立socket
	hSock = socket(PF_INET, SOCK_STREAM, 0);

	// 2 配置端口和地址
	memset(&servAdr, 0, sizeof(servAdr));
	servAdr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
	servAdr.sin_family = AF_INET;
	servAdr.sin_port = htons(6000);

	// 3 连接服务器
	if (connect(hSock, (SOCKADDR*)&servAdr, sizeof(servAdr)) == SOCKET_ERROR)
	{
		printf("connect error error code = %d\n", GetLastError());
		return -1;
	}

	// 4  发送服务端的消息   安排一个工人 起一个线程发送消息

	hSendThread = (HANDLE)_beginthreadex(NULL, 0, SendMsg,
		(void*)&hSock, 0, NULL);

	//  5 接收消息给服务端    安排一个工人 起一个线程接收消息

	hRecvThread = (HANDLE)_beginthreadex(NULL, 0, RecvMsg,
		(void*)&hSock, 0, NULL);

	//等待两个线程完成
	WaitForSingleObject(hSendThread, INFINITE);
	WaitForSingleObject(hRecvThread, INFINITE);

	// 6 关闭套接字
	closesocket(hSock);
	WSACleanup();
	return 0;
}