Windows网络编程 I/O复用的多人聊天室
代码：
https://github.com/ChristmasError/TCP-IP-Network-programming/tree/master/WINDOWS%20IO%E5%A4%8D%E7%94%A8%E5%A4%9A%E4%BA%BA%E8%81%8A%E5%A4%A9%E5%AE%A4%E5%AE%9E%E7%8E%B0
请多开几个客户端进行测试。
//server
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS
#include
#include
#include
#include//定义socklen_t
using namespace std;
#pragma comment(lib, "WS2_32") // 链接到WS2_32.lib
#define SERVER_IP "127.0.0.1"// 默认服务器端IP地址
#define SERVER_PORT 8888// 服务器端口号
class server
{
public:
server();
void init();
void process();
private:
int listener;//监听套接字
sockaddr_in serverAddr;//IPV4的地址方式
vector socksArr;//存放创建的套接字
};
int main()
{
server ser;
ser.process();
return 0;
}
server::server()
{
listener = 0;
serverAddr.sin_family = PF_INET;
serverAddr.sin_port = SERVER_PORT;
serverAddr.sin_addr.s_addr = inet_addr(SERVER_IP);//将字符串类型转换uint32_t
}
//初始化函数，创建监听套接字，绑定端口，并进行监听
void server::init()
{
WSADATA wsaData;
WSAStartup(MAKEWORD(2, 2), &wsaData);
listener = socket(AF_INET, SOCK_STREAM, 0);//采用ipv4,TCP传输
if (listener == -1) { printf("Error at socket(): %ld\n", WSAGetLastError()); perror("listener failed"); exit(1); }
else printf("Server is running......\n");
if (bind(listener, (sockaddr *)&serverAddr, sizeof(serverAddr)) < 0)
{
perror("bind error");
exit(1);
}
listen(listener, 5);//listener这个套接字监听申请的链接，最大等待连接队列为5,等待accept()
socksArr.push_back(listener);//将监听套接字加入套接字数组，数组内首个套接字就是服务器的套接字
}
void server::process()
{
int mount = 0;
fd_set fds,fds_copy;
FD_ZERO(&fds);//将fds清零
init();
//下面就是不断的检查申请的连接队列
printf("Server is waiting......\n");
while (1)
{
mount = socksArr.size();
//每次循环更新一次fds数组
for (int i = 0; i //Client
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include
#include
#include
#include
#include
#include//定义socklen_t
#pragma comment(lib, "WS2_32") // 链接到WS2_32.lib
using namespace std;
#define SERVER_IP "127.0.0.1"// 默认服务器端IP地址
#define SERVER_PORT 8888// 服务器端口号
class client
{
public:
client();
void init();
void process();
private:
int user; //客户端套接字（句柄）
int writing;
sockaddr_in serverAddr;
void sendata();
};
int main()
{
client user;
user.process();
return 0;
}
client::client()
{
user = 0;
writing = 0;
serverAddr.sin_family = PF_INET;
serverAddr.sin_port = SERVER_PORT;
serverAddr.sin_addr.s_addr = inet_addr(SERVER_IP);//将字符串类型转换uint32_t
}
void client::init()
{
WSADATA wsaData;
WSAStartup(MAKEWORD(2, 2), &wsaData);
user = socket(AF_INET, SOCK_STREAM, 0);//采用ipv4,TCP传输，成功时返回非负数socket描述符
if (user <= 0)
{
perror("establish client faild");
printf("Error at socket(): %ld\n", WSAGetLastError());
exit(1);
};
printf("establish succesfully\n");//创建成功，阻塞式的等待服务器连接
if (connect(user, (const sockaddr *)&serverAddr, sizeof(serverAddr)) < 0)
{
perror("connect to server faild");
printf("Error at socket(): %ld\n", WSAGetLastError());
exit(1);
}
printf("connect IP:%s Port:%d succesfully\n", SERVER_IP, SERVER_PORT);//创建成功
}
void client::process()
{
char recvbuf[1024];
fd_set fdread, fedwrite;
FD_ZERO(&fdread);//将fdread清零
FD_ZERO(&fedwrite);//将fedwrite清零
init();
while (1)
{
FD_SET(user, &fdread);
if (writing == 0)
FD_SET(user, &fedwrite);
struct timeval timeout = { 1,0 };//每个Select等待三秒
switch (select(0, &fdread, &fedwrite, NULL, &timeout))
{
case -1:
{
printf("Error at socket(): %ld\n", WSAGetLastError());
break;
}
case 0:
break;
default:
{
if (FD_ISSET(user, &fdread))//有待读事件
{
int size = recv(user, recvbuf, sizeof(recvbuf) - 1, 0);
if (size > 0)
{
printf("server : %s\n", recvbuf);
memset(recvbuf, '\0', sizeof(recvbuf));
}
else if (size == 0)
{
printf("server is closed\n");
exit(1);
}
}
if (FD_ISSET(user, &fedwrite))
{
FD_ZERO(&fedwrite);//将fedwrite清零
writing = 1;//表示正在写入
thread sendtask(bind(&client::sendata, this));
sendtask.detach();//将子线程和主进程分离且互相不影响
}
break;
}
}
}
}
void client::sendata()
{
char sendbuf[1024];
char middle[1024];
cin.getline(sendbuf, 1024);//读取一行
send(user, sendbuf, sizeof(sendbuf) - 1, 0);
writing = 0;
}