正文
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");
}
