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 ()
     {
         printf("请输入发送的信息:\n");
         char str[] = {  };
         scanf("%s", str);
         //发送
         send(client, str, strlen(str), );
         Sleep();
     }
 }

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

     //对比版本
     if (WSAStartup(MAKEWORD(,),&WSA)!=)
     {
         puts("版本不一致，通信失败");
         system("pause");
         return;
     }
     //创建socket通信
     client = socket(AF_INET, SOCK_STREAM, );
     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, );

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

     //开启一个写的线程
     _beginthread(write, , (void*)client);
     //不断接收信息
     while ()
     {
         char receivebuf[];
         memset(receivebuf, , );//清零
         Ret = recv(client, receivebuf, , );
         if (strlen(receivebuf)>)
         {
             printf("%s\n", 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[] = {  };
 //事件
 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 = ;

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

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

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

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

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

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

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

     //判断版本
     if (WSAStartup(MAKEWORD(, ), &WSA) != )
     {
         puts("版本不一致，通信失败\n");
         system("pause");
         return;
     }
     //创建服务器套接字
     sever = socket(AF_INET, SOCK_STREAM, );
     if (sever == INVALID_SOCKET)
     {
         puts("服务器创建失败\n");
         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, );//清零
     //socket与sockaddr_in绑定
     Ret = bind(sever, (struct sockaddr*)&localeaddr, sizeof(localeaddr));
     if (Ret != )
     {
         puts("绑定失败");
         system("pause");
         return;
     }
     //开始监听
     Ret = listen(sever, );
     if (Ret != )
     {
         puts("监听失败");
         system("pause");
         return;
     }
     puts("服务器启动\n");

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

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

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

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

     system("pause");
 }