104.tcp多线程读写实现群聊

客户端:

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<stdlib.h>
#include <winsock.h>
#include <process.h>
#pragma comment(lib,"ws2_32.lib")

#define port 5529
#define ip_addr "192.168.1.108"

//客户端写的线程
void write(void *p)
{
    //指针类型转换
    SOCKET client = (SOCKET)p;
    while (1)
    {
        printf("请输入发送的信息:
");
        char str[256] = { 0 };
        scanf("%s", str);
        //发送
        send(client, str, strlen(str), 0);
        Sleep(100);
    }
}

void main()
{
    //对比版本
    WSADATA WSA;
    //客户端套接字
    SOCKET client;
    //服务器地址
    struct sockaddr_in severaddr;
    int addrlength = 0;
    HANDLE hthread = NULL;//线程句柄
    int Ret = 0;;
    char senbuf[256] = { 0 };

    //对比版本
    if (WSAStartup(MAKEWORD(2,2),&WSA)!=0)
    {
        puts("版本不一致，通信失败");
        system("pause");
        return;
    }
    //创建socket通信
    client = socket(AF_INET, SOCK_STREAM, 0);
    if (client == INVALID_SOCKET)
    {
        puts("客户端创建失败");
        system("pause");
    }
    //设置服务器信息
    //协议族
    severaddr.sin_family = AF_INET;
    //设置地址
    severaddr.sin_addr.s_addr = inet_addr(ip_addr);
    //设置端口
    severaddr.sin_port = htons(port);
    //清空
    memset(severaddr.sin_zero, 0x00, 8);

    //连接到服务器
    Ret = connect(client, (struct sockaddr*) &severaddr, sizeof(severaddr));
    //判断是否连接成功
    if (Ret!=0)
    {
        puts("客户端连接失败");
        system("pause");
    }
    else
    {
        puts("客户端连接成功");
    }

    //开启一个写的线程
    _beginthread(write, 0, (void*)client);
    //不断接收信息
    while (1)
    {
        char receivebuf[256];
        memset(receivebuf, 0, 256);//清零
        Ret = recv(client, receivebuf, 256, 0);
        if (strlen(receivebuf)>0)
        {
            printf("%s
", receivebuf);
        }
    }

    //关闭客户端
    closesocket(client);
    WSACleanup();
}

服务器端

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<stdlib.h>
#include <winsock.h>
#include <process.h>
#pragma comment(lib,"ws2_32.lib")

//定义端口
#define port 5529
//设置本地ip
#define ip_addr "169.254.29.232"
//发送消息的缓存区
char sendbuf[256] = { 0 };
//事件
HANDLE event=NULL;
//创建互斥量
HANDLE mutex = NULL;

//保存连接的客户端信息
struct ipinfo
{
    SOCKET client;//客户端
    struct sockaddr_in clientaddr;
};

//读取线程是实时的,写入线程是有事件响应的,服务器输入或者收到消息则向客户端发送消息

//创建线程,用于向客户端发送信息
DWORD WINAPI clientwrite(void *p)
{
    //数指针类型转换
    SOCKET client = ((struct ipinfo*)p)->client;
    //获取连接的客户端信息
    struct sockaddr_in clientaddr = ((struct ipinfo*)p)->clientaddr;
    int Ret = 0;

    //不断读取
    while (1)
    {
        //等待事件
        WaitForSingleObject(event, INFINITE);
        //创建互斥量
        WaitForSingleObject(mutex, INFINITE);
        //判断
        if (strlen(sendbuf)!=0)
        {
            Ret = send(client, sendbuf, strlen(sendbuf), 0);
            if (Ret == 0 || Ret == SOCKET_ERROR)
            {
                //判断连接的客户端是否退出
                printf("%s,%d退出
",inet_ntoa(clientaddr.sin_addr), clientaddr.sin_port);
                return;
            }
        }
        ReleaseMutex(mutex);
        ResetEvent(event);//手动复位
    }
    return 0;
}

//创建线程,用于读取客户端写入的信息
DWORD WINAPI clientthreadread(void *p)
{
    //数指针类型转换
    SOCKET client = ((struct ipinfo*)p)->client;
    //获取连接的客户端信息
    struct sockaddr_in clientaddr = ((struct ipinfo*)p)->clientaddr;

    int Ret = 0;
    //接收到的信息
    char receivebufall[256] = {0};
    while (1)
    {

        char receivebuf[256] = { 0 };
        //接收
        Ret = recv(client, receivebuf, 256, 0);
        if (Ret == 0 ||Ret ==SOCKET_ERROR)
        {
            printf("%s,%d退出
", inet_ntoa(clientaddr.sin_addr), clientaddr.sin_port);
            return;
        }

        //收到
        if (strlen(receivebuf) > 0)
        {
            memset(receivebufall, 0, 256);
            sprintf(receivebufall, "收到%s:来自%s %d
", receivebuf, inet_ntoa(clientaddr.sin_addr), clientaddr.sin_port);
            printf("%s", receivebufall);
        }

        //互斥量修改全局数据
        WaitForSingleObject(mutex, INFINITE);
        memset(sendbuf, 0, 256);
        strcpy(sendbuf, receivebufall);
        ReleaseMutex(mutex);
        //收到消息后设置事件,向所有客户端发送消息
        SetEvent(event);
    }
    return 0;
}

//创建线程
//服务器
void mains(void *p)
{
    //对比版本
    WSADATA WSA;
    //客户端
    SOCKET client, sever;
    //服务器地址,和连接的客户端信息
    struct sockaddr_in localeaddr, clientaddr;
    //sockaddr_in的结构体大小
    int addrlength = 0;
    //线程句柄
    HANDLE hthread = NULL;
    //发送和接收消息的返回值
    int Ret = 0;
    char senbuf[256] = { 0 };

    //判断版本
    if (WSAStartup(MAKEWORD(2, 2), &WSA) != 0)
    {
        puts("版本不一致，通信失败
");
        system("pause");
        return;
    }
    //创建服务器套接字
    sever = socket(AF_INET, SOCK_STREAM, 0);
    if (sever == INVALID_SOCKET)
    {
        puts("服务器创建失败
");
        system("pause");
        return;
    }
    //设置本地的sockaddr_in
    localeaddr.sin_family = AF_INET;
    localeaddr.sin_addr.s_addr = inet_addr(ip_addr);
    localeaddr.sin_port = htons(port);
    memset(localeaddr.sin_zero, 0x00, 8);//清零
    //socket与sockaddr_in绑定
    Ret = bind(sever, (struct sockaddr*)&localeaddr, sizeof(localeaddr));
    if (Ret != 0)
    {
        puts("绑定失败");
        system("pause");
        return;
    }
    //开始监听
    Ret = listen(sever, 5);
    if (Ret != 0)
    {
        puts("监听失败");
        system("pause");
        return;
    }
    puts("服务器启动
");

    while (1)
    {
        //获取长度
        addrlength = sizeof(clientaddr);
        //通过套接字接受客户端连接
        client = accept(sever, (struct sockaddr*) &clientaddr, &addrlength);
        if (client == INVALID_SOCKET)
        {
            puts("接收失败");
            system("pause");
            return;
        }
        printf("客户端连接%s  %d
", inet_ntoa(clientaddr.sin_addr), clientaddr.sin_port);

        //全局变量pinfo只是起到一个中转的作用,最后结果存放在线程里面
        //创建连接的客户端信息
        struct ipinfo pinfo;
        pinfo.client = client;
        pinfo.clientaddr = clientaddr;
        //开启一个写的线程
        hthread = CreateThread(NULL, 0, clientwrite, (void*)&pinfo, 0, NULL);
        //开启一个读的线程
        hthread = CreateThread(NULL, 0, clientthreadread, (void*)&pinfo, 0, NULL);
    }

    closesocket(sever);
    closesocket(client);
    WSACleanup();
}

            

void main()
{
    event = CreateEvent(NULL, TRUE, FALSE, NULL);
    mutex = CreateMutex(NULL, FALSE, NULL);//排斥
    _beginthread(mains, 0, NULL);
    while (1)
    {    
        printf("请输入向客户端发送的信息:");
        scanf("%s", sendbuf);
        SetEvent(event);
    }

    system("pause");
}