C++Socket网络通信多路复用
1.select
可以在windows上面使用！
服务端：
/*
* @Description:SOCKET
* @Author: szq
* @Github: https://github.com/MrQqqq
* @Date: 2020-06-26 00:04:39
* @LastEditors: szq
* @LastEditTime: 2020-06-26 00:04:39
* @FilePath: \cpp\src\BinaryTree\client_TCp.cpp
*/
#include<iostream>
#include<winsock.h>
#include<vector>
#pragma comment(lib,"ws2_32.lib")
using namespace std;
void initialization() {
//初始化套接字库
WORD w_req = MAKEWORD(2, 2);//版本号
WSADATA wsadata;
int err;
err = WSAStartup(w_req, &wsadata);
if (err != 0) {
cout << "初始化套接字库失败！" << endl;
}
else {
cout << "初始化套接字库成功！" << endl;
}
//检测版本号
if (LOBYTE(wsadata.wVersion) != 2 || HIBYTE(wsadata.wHighVersion) != 2) {
cout << "套接字库版本号不符！" << endl;
WSACleanup();
}
else {
cout << "套接字库版本正确！" << endl;
}
}
enum MyCMD {
LOGIN,
LOGOUT,
LOGERROR,
RESULT
};
class DataHeader {
public:
int dataLength;
MyCMD cmd;
};
class Login : public DataHeader{
public:
Login() {
dataLength = sizeof(Login);
cmd = LOGIN;
}
char username[30];
char password[30];
};
class Result : public DataHeader {
public:
int result;
Result(int a) {
dataLength = sizeof(Result);
cmd = RESULT;
result = a;
}
};
vector<SOCKET> g_clients;
int main() {
//定义长度变量
int send_len = 0;
int recv_len = 0;
int len = 0;
//定义发送缓冲区和接收缓冲区
char send_buf[1024];
char recv_buf[1024];
//定义服务器套接字,接受请求套接字
SOCKET s_server;
//服务器地址和客户端地址
SOCKADDR_IN server_addr;
initialization();
//填充服务端信息
server_addr.sin_family = AF_INET;
server_addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
server_addr.sin_port = htons(5010);
//创建套接字
s_server = socket(AF_INET, SOCK_STREAM, 0);
if (bind(s_server, (SOCKADDR*)&server_addr, sizeof(SOCKADDR)) == SOCKET_ERROR) {
cout << "套接字绑定失败！" << endl;
}
else {
cout << "套接字绑定成功！" << endl;
}
if (listen(s_server, SOMAXCONN) < 0) {
cout << "设置监听状态失败！" << endl;
}
else {
cout << "设置监听状态成功！" << endl;
}
cout << "服务器正在监听连接，请稍候..." << endl;
while (1) {
fd_set fdRead;
fd_set fdWrite;
fd_set fdExecpt;
//清空
FD_ZERO(&fdRead);
FD_ZERO(&fdWrite);
FD_ZERO(&fdExecpt);
FD_SET(s_server, &fdRead);
FD_SET(s_server, &fdWrite);
FD_SET(s_server, &fdExecpt);
for (int i = 0; i < g_clients.size(); i++) {
FD_SET(g_clients[i], &fdRead);
}
//nfds是一个整数值，是指fd_set集合中所有描述符的范围，而不是数量，即最大值
timeval t = { 0,0 };//设置成非阻塞
int ret = select(s_server + 1, &fdRead, &fdWrite, &fdExecpt,&t);
if (ret < 0) {
cout << "select任务结束！" << endl;
break;
}
if (FD_ISSET(s_server, &fdRead)) {
FD_CLR(s_server, &fdRead);
SOCKET s_accept = INVALID_SOCKET;
SOCKADDR_IN accept_addr;
len = sizeof(SOCKADDR);
s_accept = accept(s_server, (SOCKADDR*)&accept_addr, &len);
if (s_accept == SOCKET_ERROR) {
cout << "连接失败！" << endl;
return 0;
}
cout << "连接建立，准备接受数据" << endl;
g_clients.push_back(s_accept);
}
for (int i = 0; i < fdRead.fd_count; i++) {
//处理
recv_len = recv(fdRead.fd_array[i], recv_buf, 1024, 0);
if (recv_len < 0) {
cout << "接受失败！" << endl;
auto iter = find(g_clients.begin(), g_clients.end(), fdRead.fd_array[i]);
g_clients.erase(iter);
break;
}
else {
Login *login = (Login*)recv_buf;
if (login->cmd == LOGIN) {
if (strcmp(login->username, "szq") == 0) {
Result result(1);
send_len = send(g_clients[i], (char*)&result, sizeof(Result), 0);
}
cout << "登录信息： 用户名=" << login->username << " 密码=" << login->password << endl;
if (send_len < 0) {
cout << "发送失败！" << endl;
break;
}
}
}
}
}
//关闭套接字
closesocket(s_server);
for (auto skt : g_clients) {
closesocket(skt);
}
return 0;
}
客户端：
#include<iostream>
#include<winsock.h>
#include<cstring>
#pragma comment(lib,"ws2_32.lib")
#pragma warning( disable : 4996)
using namespace std;
void initialization();
enum MyCMD {
LOGIN,
LOGOUT,
LOGERROR,
RESULT
};
class DataHeader {
public:
int dataLength;
MyCMD cmd;
};
class Login : public DataHeader {
public:
Login() {
dataLength = sizeof(Login);
cmd = LOGIN;
}
char username[30];
char password[30];
};
class Result : public DataHeader {
public:
int result;
Result(int a) {
dataLength = sizeof(Result);
cmd = RESULT;
result = a;
}
};
int main() {
//定义长度变量
int send_len = 0;
int recv_len = 0;
//定义发送缓冲区和接受缓冲区
char send_buf[100];
char recv_buf[100];
//定义服务端套接字，接受请求套接字
SOCKET s_server;
//服务端地址客户端地址
SOCKADDR_IN server_addr;
initialization();
//填充服务端信息
server_addr.sin_family = AF_INET;
server_addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
server_addr.sin_port = htons(5010);
//创建套接字
s_server = socket(AF_INET, SOCK_STREAM, 0);
if (connect(s_server, (SOCKADDR *)&server_addr, sizeof(SOCKADDR)) == SOCKET_ERROR) {
cout << "服务器连接失败！" << endl;
WSACleanup();
}
else {
cout << "服务器连接成功！" << endl;
}
//发送,接收数据
while (1) {
char s[30];
cin >> s;
if (strcmp(s,"login") != 0) {
continue;
}
Login login;
strcpy(login.username, "szq");
strcpy(login.password, "123");
send_len = send(s_server, (char*)&login, sizeof(login), 0);
if (send_len < 0) {
cout << "发送失败！" << endl;
break;
}
recv_len = recv(s_server, recv_buf, 100, 0);
if (recv_len < 0) {
cout << "接受失败！" << endl;
break;
}
else {
Result *result = (Result*)recv_buf;
cout << "登录结果为:" << result->result << endl;
}
}
//关闭套接字
closesocket(s_server);
//释放DLL资源
WSACleanup();
return 0;
}
void initialization() {
//初始化套接字库
WORD w_req = MAKEWORD(2, 2);//版本号
WSADATA wsadata;
int err;
err = WSAStartup(w_req, &wsadata);
if (err != 0) {
cout << "初始化套接字库失败！" << endl;
}
else {
cout << "初始化套接字库成功！" << endl;
}
//检测版本号
if (LOBYTE(wsadata.wVersion) != 2 || HIBYTE(wsadata.wHighVersion) != 2) {
cout << "套接字库版本号不符！" << endl;
WSACleanup();
}
else {
cout << "套接字库版本正确！" << endl;
}
//填充服务端地址信息
}
版本2：可以支持跨平台的版本。
服务端
/*
* @Description:SOCKET
* @Author: szq
* @Github: https://github.com/MrQqqq
* @Date: 2020-06-26 00:04:39
* @LastEditors: szq
* @LastEditTime: 2020-06-26 00:04:39
* @FilePath: \cpp\src\BinaryTree\client_TCp.cpp
*/
#include<iostream>
#include<vector>
#ifdef _WIN32
#include<winsock.h>
#pragma comment(lib,"ws2_32.lib")
#else
#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include <netinet/in.h>
#include <cstring>
#define SOCKET int
#define SOCKADDR_IN sockaddr_in
#define SOCKADDR sockaddr
#define SOCKET_ERROR socket_error
#endif
using namespace std;
void initialization() {
#ifdef _WIN32
//初始化套接字库
WORD w_req = MAKEWORD(2, 2);//版本号
WSADATA wsadata;
int err;
err = WSAStartup(w_req, &wsadata);
if (err != 0) {
cout << "初始化套接字库失败！" << endl;
}
else {
cout << "初始化套接字库成功！" << endl;
}
//检测版本号
if (LOBYTE(wsadata.wVersion) != 2 || HIBYTE(wsadata.wHighVersion) != 2) {
cout << "套接字库版本号不符！" << endl;
WSACleanup();
}
else {
cout << "套接字库版本正确！" << endl;
}
#endif
}
enum MyCMD {
LOGIN,
LOGOUT,
LOGERROR,
RESULT
};
class DataHeader {
public:
int dataLength;
MyCMD cmd;
};
class Login : public DataHeader{
public:
Login() {
dataLength = sizeof(Login);
cmd = LOGIN;
}
char username[30];
char password[30];
};
class Result : public DataHeader {
public:
int result;
Result(int a) {
dataLength = sizeof(Result);
cmd = RESULT;
result = a;
}
};
vector<SOCKET> g_clients;
int main() {
//定义长度变量
int send_len = 0;
int recv_len = 0;
unsigned int len = 0;
//定义发送缓冲区和接收缓冲区
char send_buf[1024];
char recv_buf[1024];
//定义服务器套接字,接受请求套接字
SOCKET s_server;
//服务器地址和客户端地址
SOCKADDR_IN server_addr;
initialization();
//填充服务端信息
server_addr.sin_family = AF_INET;
#ifdef _WIN32
server_addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
#else
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
#endif
server_addr.sin_port = htons(5010);
//创建套接字
s_server = socket(AF_INET, SOCK_STREAM, 0);
if (bind(s_server, (SOCKADDR*)&server_addr, sizeof(SOCKADDR)) < 0) {
cout << "套接字绑定失败！" << endl;
}
else {
cout << "套接字绑定成功！" << endl;
}
if (listen(s_server, SOMAXCONN) < 0) {
cout << "设置监听状态失败！" << endl;
}
else {
cout << "设置监听状态成功！" << endl;
}
cout << "服务器正在监听连接，请稍候..." << endl;
while (1) {
fd_set fdRead;
fd_set fdWrite;
fd_set fdExecpt;
//清空
FD_ZERO(&fdRead);
FD_ZERO(&fdWrite);
FD_ZERO(&fdExecpt);
FD_SET(s_server, &fdRead);
FD_SET(s_server, &fdWrite);
FD_SET(s_server, &fdExecpt);
for (int i = 0; i < g_clients.size(); i++) {
FD_SET(g_clients[i], &fdRead);
}
//nfds是一个整数值，是指fd_set集合中所有描述符的范围，而不是数量，即最大值
timeval t = { 0,0 };//设置成非阻塞
int ret = select(s_server + 1, &fdRead, &fdWrite, &fdExecpt,&t);
if (ret < 0) {
cout << "select任务结束！" << endl;
break;
}
if (FD_ISSET(s_server, &fdRead)) {
FD_CLR(s_server, &fdRead);
SOCKET s_accept = 0;
SOCKADDR_IN accept_addr;
len = sizeof(SOCKADDR);
s_accept = accept(s_server, (SOCKADDR*)&accept_addr, &len);
if (s_accept < 1) {
cout << "连接失败！" << endl;
return 0;
}
cout << "连接建立，准备接受数据" << endl;
g_clients.push_back(s_accept);
}
#ifdef _WIN32
for (int i = 0; i < fdRead.fd_count; i++) {
//处理
recv_len = recv(fdRead.fd_array[i], recv_buf, 1024, 0);
if (recv_len < 0) {
cout << "接受失败！" << endl;
auto iter = find(g_clients.begin(), g_clients.end(), fdRead.fd_array[i]);
g_clients.erase(iter);
break;
}
else {
Login *login = (Login*)recv_buf;
if (login->cmd == LOGIN) {
if (strcmp(login->username, "szq") == 0) {
Result result(1);
send_len = send(g_clients[i], (char*)&result, sizeof(Result), 0);
}
cout << "登录信息： 用户名=" << login->username << " 密码=" << login->password << endl;
if (send_len < 0) {
cout << "发送失败！" << endl;
break;
}
}
}
}
#else
/*
* 由于在Linux系统中FD_SET的定义不同，没有相应的fd_count和fd_array,因此这里遍历FD_SET的方式有所不同
*
* */
while(g_clients.size() > 0){
//处理
recv_len = recv(g_clients[0], recv_buf, 1024, 0);
if (recv_len < 0) {
cout << "接受失败！" << endl;
g_clients.erase(g_clients.begin());
}
else {
Login *login = (Login*)recv_buf;
if (login->cmd == LOGIN) {
if (strcmp(login->username, "szq") == 0) {
Result result(1);
send_len = send(g_clients[0], (char*)&result, sizeof(Result), 0);
}
cout << "登录信息： 用户名=" << login->username << " 密码=" << login->password << endl;
if (send_len < 0) {
cout << "发送失败！" << endl;
break;
}
}
}
}
#endif
}
//关闭套接字
#ifdef _WIN32
closesocket(s_server);
#else
close(s_server);
#endif
for (auto skt : g_clients) {
#ifdef _WIN32
closesocket(skt);
#else
close(skt);
#endif
}
return 0;
}
客户端：
#include<iostream>
#include<vector>
#ifdef _WIN32
#include<winsock.h>
#pragma comment(lib,"ws2_32.lib")
#else
#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include <netinet/in.h>
#include<arpa/inet.h>
#include <cstring>
#define SOCKET int
#define SOCKADDR_IN sockaddr_in
#define SOCKADDR sockaddr
#define SOCKET_ERROR 0
#endif
using namespace std;
void initialization() {
#ifdef _WIN32
//初始化套接字库
WORD w_req = MAKEWORD(2, 2);//版本号
WSADATA wsadata;
int err;
err = WSAStartup(w_req, &wsadata);
if (err != 0) {
cout << "初始化套接字库失败！" << endl;
}
else {
cout << "初始化套接字库成功！" << endl;
}
//检测版本号
if (LOBYTE(wsadata.wVersion) != 2 || HIBYTE(wsadata.wHighVersion) != 2) {
cout << "套接字库版本号不符！" << endl;
WSACleanup();
}
else {
cout << "套接字库版本正确！" << endl;
}
#endif
}
enum MyCMD {
LOGIN,
LOGOUT,
LOGERROR,
RESULT
};
class DataHeader {
public:
int dataLength;
MyCMD cmd;
};
class Login : public DataHeader {
public:
Login() {
dataLength = sizeof(Login);
cmd = LOGIN;
}
char username[30];
char password[30];
};
class Result : public DataHeader {
public:
int result;
Result(int a) {
dataLength = sizeof(Result);
cmd = RESULT;
result = a;
}
};
int main() {
//定义长度变量
int send_len = 0;
int recv_len = 0;
//定义发送缓冲区和接受缓冲区
char send_buf[100];
char recv_buf[100];
//定义服务端套接字，接受请求套接字
SOCKET s_server;
//服务端地址客户端地址
SOCKADDR_IN server_addr;
initialization();
//填充服务端信息
server_addr.sin_family = AF_INET;
#ifdef _WIN32
server_addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
#else
server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
#endif
server_addr.sin_port = htons(5010);
//创建套接字
s_server = socket(AF_INET, SOCK_STREAM, 0);
if (connect(s_server, (SOCKADDR *)&server_addr, sizeof(SOCKADDR)) < 0) {
cout << "服务器连接失败！" << endl;
#ifdef _WIN32
WSACleanup();
#endif
}
else {
cout << "服务器连接成功！" << endl;
}
//发送,接收数据
while (1) {
char s[30];
cin >> s;
if (strcmp(s,"login") != 0) {
continue;
}
Login login;
strcpy(login.username, "szq");
strcpy(login.password, "123");
send_len = send(s_server, (char*)&login, sizeof(login), 0);
if (send_len < 0) {
cout << "发送失败！" << endl;
break;
}
recv_len = recv(s_server, recv_buf, 100, 0);
if (recv_len < 0) {
cout << "接受失败！" << endl;
break;
}
else {
Result *result = (Result*)recv_buf;
cout << "登录结果为:" << result->result << endl;
}
}
//关闭套接字
#ifdef _WIN32
closesocket(s_server);
#else
close(s_server);
#endif
//释放DLL资源
#ifdef _WIN32
WSACleanup();
#endif
return 0;
}
239
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
