在掌握了基于 TCP 的套接字通信流程之后,为了方便使用,提高编码效率,可以对通信操作进行封装,本着有浅入深的原则,先基于 C 语言进行面向过程的函数封装,然后再基于 C++ 进行面向对象的类封装。
基于 TCP 的套接字通信分为两部分:服务器端通信和客户端通信。我们只要掌握了通信流程,封装出对应的功能函数也就不在话下了,先来回顾一下通信流程:
通过通信流程可以看出服务器和客户端有些操作步骤是相同的,因此封装的功能函数是可以共用的,相关的通信函数声明如下:
///////////////////////////////////////////////////
//////////////////// 服务器 ///////////////////////
///////////////////////////////////////////////////
int bindSocket(int lfd, unsigned short port);
int setListen(int lfd);
int acceptConn(int lfd, struct sockaddr_in *addr);
///////////////////////////////////////////////////
//////////////////// 客户端 ///////////////////////
///////////////////////////////////////////////////
int connectToHost(int fd, const char* ip, unsigned short port);
///////////////////////////////////////////////////
///////////////////// 共用 ////////////////////////
///////////////////////////////////////////////////
int createSocket();
int sendMsg(int fd, const char* msg);
int recvMsg(int fd, char* msg, int size);
int closeSocket(int fd);
int readn(int fd, char* buf, int size);
int writen(int fd, const char* msg, int size);
关于函数 readn()
和 writen()
的作用请参考[TCP数据粘包的处理]
// 创建监套接字
int createSocket()
{
int fd = socket(AF_INET, SOCK_STREAM, 0);
if(fd == -1)
{
perror("socket");
return -1;
}
printf("套接字创建成功, fd=%d\n", fd);
return fd;
}
// 绑定本地的IP和端口
int bindSocket(int lfd, unsigned short port)
{
struct sockaddr_in saddr;
saddr.sin_family = AF_INET;
saddr.sin_port = htons(port);
saddr.sin_addr.s_addr = INADDR_ANY; // 0 = 0.0.0.0
int ret = bind(lfd, (struct sockaddr*)&saddr, sizeof(saddr));
if(ret == -1)
{
perror("bind");
return -1;
}
printf("套接字绑定成功, ip: %s, port: %d\n",
inet_ntoa(saddr.sin_addr), port);
return ret;
}
// 设置监听
int setListen(int lfd)
{
int ret = listen(lfd, 128);
if(ret == -1)
{
perror("listen");
return -1;
}
printf("设置监听成功...\n");
return ret;
}
// 阻塞并等待客户端的连接
int acceptConn(int lfd, struct sockaddr_in *addr)
{
int cfd = -1;
if(addr == NULL)
{
cfd = accept(lfd, NULL, NULL);
}
else
{
int addrlen = sizeof(struct sockaddr_in);
cfd = accept(lfd, (struct sockaddr*)addr, &addrlen);
}
if(cfd == -1)
{
perror("accept");
return -1;
}
printf("成功和客户端建立连接...\n");
return cfd;
}
// 接收数据
int recvMsg(int cfd, char** msg)
{
if(msg == NULL || cfd <= 0)
{
return -1;
}
// 接收数据
// 1. 读数据头
int len = 0;
readn(cfd, (char*)&len, 4);
len = ntohl(len);
printf("数据块大小: %d\n", len);
// 根据读出的长度分配内存
char *buf = (char*)malloc(len+1);
int ret = readn(cfd, buf, len);
if(ret != len)
{
return -1;
}
buf[len] = '\0';
*msg = buf;
return ret;
}
// 发送数据
int sendMsg(int cfd, char* msg, int len)
{
if(msg == NULL || len <= 0)
{
return -1;
}
// 申请内存空间: 数据长度 + 包头4字节(存储数据长度)
char* data = (char*)malloc(len+4);
int bigLen = htonl(len);
memcpy(data, &bigLen, 4);
memcpy(data+4, msg, len);
// 发送数据
int ret = writen(cfd, data, len+4);
return ret;
}
// 连接服务器
int connectToHost(int fd, const char* ip, unsigned short port)
{
// 2. 连接服务器IP port
struct sockaddr_in saddr;
saddr.sin_family = AF_INET;
saddr.sin_port = htons(port);
inet_pton(AF_INET, ip, &saddr.sin_addr.s_addr);
int ret = connect(fd, (struct sockaddr*)&saddr, sizeof(saddr));
if(ret == -1)
{
perror("connect");
return -1;
}
printf("成功和服务器建立连接...\n");
return ret;
}
// 关闭套接字
int closeSocket(int fd)
{
int ret = close(fd);
if(ret == -1)
{
perror("close");
}
return ret;
}
// 接收指定的字节数
// 函数调用成功返回 size
int readn(int fd, char* buf, int size)
{
int nread = 0;
int left = size;
char* p = buf;
while(left > 0)
{
if((nread = read(fd, p, left)) > 0)
{
p += nread;
left -= nread;
}
else if(nread == -1)
{
return -1;
}
}
return size;
}
// 发送指定的字节数
// 函数调用成功返回 size
int writen(int fd, const char* msg, int size)
{
int left = size;
int nwrite = 0;
const char* p = msg;
while(left > 0)
{
if((nwrite = write(fd, msg, left)) > 0)
{
p += nwrite;
left -= nwrite;
}
else if(nwrite == -1)
{
return -1;
}
}
return size;
}
编写 C++ 程序应当遵循面向对象三要素:封装、继承、多态。简单地说就是封装之后的类可以隐藏掉某些属性使操作更简单并且类的功能要单一,如果要代码重用可以进行类之间的继承,如果要让函数的使用更加灵活可以使用多态。因此,我们需要封装两个类:客户端类和服务器端的类。
根据面向对象的思想,整个通信过程不管是监听还是通信的套接字都是可以封装到类的内部并且将其隐藏掉,这样相关操作函数的参数也就随之减少了,使用者用起来也更简便。
class TcpClient
{
public:
TcpClient();
~TcpClient();
// int connectToHost(int fd, const char* ip, unsigned short port);
int connectToHost(string ip, unsigned short port);
// int sendMsg(int fd, const char* msg);
int sendMsg(string msg);
// int recvMsg(int fd, char* msg, int size);
string recvMsg();
// int createSocket();
// int closeSocket(int fd);
private:
// int readn(int fd, char* buf, int size);
int readn(char* buf, int size);
// int writen(int fd, const char* msg, int size);
int writen(const char* msg, int size);
private:
int cfd; // 通信的套接字
};
通过对客户端的操作进行封装,我们可以看到有如下的变化:
class TcpServer
{
public:
TcpServer();
~TcpServer();
// int bindSocket(int lfd, unsigned short port) + int setListen(int lfd)
int setListen(unsigned short port);
// int acceptConn(int lfd, struct sockaddr_in *addr);
int acceptConn(struct sockaddr_in *addr);
// int sendMsg(int fd, const char* msg);
int sendMsg(string msg);
// int recvMsg(int fd, char* msg, int size);
string recvMsg();
// int createSocket();
// int closeSocket(int fd);
private:
// int readn(int fd, char* buf, int size);
int readn(char* buf, int size);
// int writen(int fd, const char* msg, int size);
int writen(const char* msg, int size);
private:
int lfd; // 监听的套接字
int cfd; // 通信的套接字
};
通过对服务器端的操作进行封装,我们可以看到这个类和客户端的类结构以及封装思路是差不多的,并且两个类的内部有些操作的重叠的:接收和发送通信数据的函数 recvMsg()、sendMsg()
,以及内部函数 readn()、writen()
。不仅如此服务器端的类设计成这样样子是有缺陷的:服务器端一般需要和多个客户端建立连接,因此通信的套接字就需要有 N 个,但是在上面封装的类里边只有一个。
既然如此,我们如何解决服务器和客户端的代码冗余和服务器不能跟多客户端通信的问题呢?
答:瘦身、减负。可以将服务器的通信功能去掉,只留下监听并建立新连接一个功能。将客户端类变成一个专门用于套接字通信的类即可。服务器端整个流程使用服务器类 + 通信类来处理;客户端整个流程通过通信的类来处理。
根据对第一个版本的分析,可以对以上代码做如下修改:
套接字通信类既可以在客户端使用,也可以在服务器端使用,职责是接收和发送数据包。
类声明
class TcpSocket
{
public:
TcpSocket();
TcpSocket(int socket);
~TcpSocket();
int connectToHost(string ip, unsigned short port);
int sendMsg(string msg);
string recvMsg();
private:
int readn(char* buf, int size);
int writen(const char* msg, int size);
private:
int m_fd; // 通信的套接字
};
类定义
TcpSocket::TcpSocket()
{
m_fd = socket(AF_INET, SOCK_STREAM, 0);
}
TcpSocket::TcpSocket(int socket)
{
m_fd = socket;
}
TcpSocket::~TcpSocket()
{
if (m_fd > 0)
{
close(m_fd);
}
}
int TcpSocket::connectToHost(string ip, unsigned short port)
{
// 连接服务器IP port
struct sockaddr_in saddr;
saddr.sin_family = AF_INET;
saddr.sin_port = htons(port);
inet_pton(AF_INET, ip.data(), &saddr.sin_addr.s_addr);
int ret = connect(m_fd, (struct sockaddr*)&saddr, sizeof(saddr));
if (ret == -1)
{
perror("connect");
return -1;
}
cout << "成功和服务器建立连接..." << endl;
return ret;
}
int TcpSocket::sendMsg(string msg)
{
// 申请内存空间: 数据长度 + 包头4字节(存储数据长度)
char* data = new char[msg.size() + 4];
int bigLen = htonl(msg.size());
memcpy(data, &bigLen, 4);
memcpy(data + 4, msg.data(), msg.size());
// 发送数据
int ret = writen(data, msg.size() + 4);
delete[]data;
return ret;
}
string TcpSocket::recvMsg()
{
// 接收数据
// 1. 读数据头
int len = 0;
readn((char*)&len, 4);
len = ntohl(len);
cout << "数据块大小: " << len << endl;
// 根据读出的长度分配内存
char* buf = new char[len + 1];
int ret = readn(buf, len);
if (ret != len)
{
return string();
}
buf[len] = '\0';
string retStr(buf);
delete[]buf;
return retStr;
}
int TcpSocket::readn(char* buf, int size)
{
int nread = 0;
int left = size;
char* p = buf;
while (left > 0)
{
if ((nread = read(m_fd, p, left)) > 0)
{
p += nread;
left -= nread;
}
else if (nread == -1)
{
return -1;
}
}
return size;
}
int TcpSocket::writen(const char* msg, int size)
{
int left = size;
int nwrite = 0;
const char* p = msg;
while (left > 0)
{
if ((nwrite = write(m_fd, msg, left)) > 0)
{
p += nwrite;
left -= nwrite;
}
else if (nwrite == -1)
{
return -1;
}
}
return size;
}
在第二个版本的套接字通信类中一共有两个构造函数:
TcpSocket::TcpSocket()
{
m_fd = socket(AF_INET, SOCK_STREAM, 0);
}
TcpSocket::TcpSocket(int socket)
{
m_fd = socket;
}
服务器类主要用于套接字通信的服务器端,并且没有通信能力,当服务器和客户端的新连接建立之后,需要通过 TcpSocket 类的带参构造将通信的描述符包装成一个通信对象,这样就可以使用这个对象和客户端通信了。
类声明
class TcpServer
{
public:
TcpServer();
~TcpServer();
int setListen(unsigned short port);
TcpSocket* acceptConn(struct sockaddr_in* addr = nullptr);
private:
int m_fd; // 监听的套接字
};
类定义
TcpServer::TcpServer()
{
m_fd = socket(AF_INET, SOCK_STREAM, 0);
}
TcpServer::~TcpServer()
{
close(m_fd);
}
int TcpServer::setListen(unsigned short port)
{
struct sockaddr_in saddr;
saddr.sin_family = AF_INET;
saddr.sin_port = htons(port);
saddr.sin_addr.s_addr = INADDR_ANY; // 0 = 0.0.0.0
int ret = bind(m_fd, (struct sockaddr*)&saddr, sizeof(saddr));
if (ret == -1)
{
perror("bind");
return -1;
}
cout << "套接字绑定成功, ip: "
<< inet_ntoa(saddr.sin_addr)
<< ", port: " << port << endl;
ret = listen(m_fd, 128);
if (ret == -1)
{
perror("listen");
return -1;
}
cout << "设置监听成功..." << endl;
return ret;
}
TcpSocket* TcpServer::acceptConn(sockaddr_in* addr)
{
if (addr == NULL)
{
return nullptr;
}
socklen_t addrlen = sizeof(struct sockaddr_in);
int cfd = accept(m_fd, (struct sockaddr*)addr, &addrlen);
if (cfd == -1)
{
perror("accept");
return nullptr;
}
printf("成功和客户端建立连接...\n");
return new TcpSocket(cfd);
}
通过调整可以发现,套接字服务器类功能更加单一了,这样设计即解决了代码冗余问题,还能使这两个类更容易维护。
int main()
{
// 1. 创建通信的套接字
TcpSocket tcp;
// 2. 连接服务器IP port
int ret = tcp.connectToHost("192.168.237.131", 10000);
if (ret == -1)
{
return -1;
}
// 3. 通信
int fd1 = open("english.txt", O_RDONLY);
int length = 0;
char tmp[100];
memset(tmp, 0, sizeof(tmp));
while ((length = read(fd1, tmp, sizeof(tmp))) > 0)
{
// 发送数据
tcp.sendMsg(string(tmp, length));
cout << "send Msg: " << endl;
cout << tmp << endl << endl << endl;
memset(tmp, 0, sizeof(tmp));
// 接收数据
usleep(300);
}
sleep(10);
return 0;
}
struct SockInfo
{
TcpServer* s;
TcpSocket* tcp;
struct sockaddr_in addr;
};
void* working(void* arg)
{
struct SockInfo* pinfo = static_cast<struct SockInfo*>(arg);
// 连接建立成功, 打印客户端的IP和端口信息
char ip[32];
printf("客户端的IP: %s, 端口: %d\n",
inet_ntop(AF_INET, &pinfo->addr.sin_addr.s_addr, ip, sizeof(ip)),
ntohs(pinfo->addr.sin_port));
// 5. 通信
while (1)
{
printf("接收数据: .....\n");
string msg = pinfo->tcp->recvMsg();
if (!msg.empty())
{
cout << msg << endl << endl << endl;
}
else
{
break;
}
}
delete pinfo->tcp;
delete pinfo;
return nullptr;
}
int main()
{
// 1. 创建监听的套接字
TcpServer s;
// 2. 绑定本地的IP port并设置监听
s.setListen(10000);
// 3. 阻塞并等待客户端的连接
while (1)
{
SockInfo* info = new SockInfo;
TcpSocket* tcp = s.acceptConn(&info->addr);
if (tcp == nullptr)
{
cout << "重试...." << endl;
continue;
}
// 创建子线程
pthread_t tid;
info->s = &s;
info->tcp = tcp;
pthread_create(&tid, NULL, working, info);
pthread_detach(tid);
}
return 0;
}
本文由哈喽比特于2年以前收录,如有侵权请联系我们。
文章来源:https://mp.weixin.qq.com/s/-9sfkIVIFXe6euIQMfLUFA
京东创始人刘强东和其妻子章泽天最近成为了互联网舆论关注的焦点。有关他们“移民美国”和在美国购买豪宅的传言在互联网上广泛传播。然而,京东官方通过微博发言人发布的消息澄清了这些传言,称这些言论纯属虚假信息和蓄意捏造。
日前,据博主“@超能数码君老周”爆料,国内三大运营商中国移动、中国电信和中国联通预计将集体采购百万台规模的华为Mate60系列手机。
据报道,荷兰半导体设备公司ASML正看到美国对华遏制政策的负面影响。阿斯麦(ASML)CEO彼得·温宁克在一档电视节目中分享了他对中国大陆问题以及该公司面临的出口管制和保护主义的看法。彼得曾在多个场合表达了他对出口管制以及中荷经济关系的担忧。
今年早些时候,抖音悄然上线了一款名为“青桃”的 App,Slogan 为“看见你的热爱”,根据应用介绍可知,“青桃”是一个属于年轻人的兴趣知识视频平台,由抖音官方出品的中长视频关联版本,整体风格有些类似B站。
日前,威马汽车首席数据官梅松林转发了一份“世界各国地区拥车率排行榜”,同时,他发文表示:中国汽车普及率低于非洲国家尼日利亚,每百户家庭仅17户有车。意大利世界排名第一,每十户中九户有车。
近日,一项新的研究发现,维生素 C 和 E 等抗氧化剂会激活一种机制,刺激癌症肿瘤中新血管的生长,帮助它们生长和扩散。
据媒体援引消息人士报道,苹果公司正在测试使用3D打印技术来生产其智能手表的钢质底盘。消息传出后,3D系统一度大涨超10%,不过截至周三收盘,该股涨幅回落至2%以内。
9月2日,坐拥千万粉丝的网红主播“秀才”账号被封禁,在社交媒体平台上引发热议。平台相关负责人表示,“秀才”账号违反平台相关规定,已封禁。据知情人士透露,秀才近期被举报存在违法行为,这可能是他被封禁的部分原因。据悉,“秀才”年龄39岁,是安徽省亳州市蒙城县人,抖音网红,粉丝数量超1200万。他曾被称为“中老年...
9月3日消息,亚马逊的一些股东,包括持有该公司股票的一家养老基金,日前对亚马逊、其创始人贝索斯和其董事会提起诉讼,指控他们在为 Project Kuiper 卫星星座项目购买发射服务时“违反了信义义务”。
据消息,为推广自家应用,苹果现推出了一个名为“Apps by Apple”的网站,展示了苹果为旗下产品(如 iPhone、iPad、Apple Watch、Mac 和 Apple TV)开发的各种应用程序。
特斯拉本周在美国大幅下调Model S和X售价,引发了该公司一些最坚定支持者的不满。知名特斯拉多头、未来基金(Future Fund)管理合伙人加里·布莱克发帖称,降价是一种“短期麻醉剂”,会让潜在客户等待进一步降价。
据外媒9月2日报道,荷兰半导体设备制造商阿斯麦称,尽管荷兰政府颁布的半导体设备出口管制新规9月正式生效,但该公司已获得在2023年底以前向中国运送受限制芯片制造机器的许可。
近日,根据美国证券交易委员会的文件显示,苹果卫星服务提供商 Globalstar 近期向马斯克旗下的 SpaceX 支付 6400 万美元(约 4.65 亿元人民币)。用于在 2023-2025 年期间,发射卫星,进一步扩展苹果 iPhone 系列的 SOS 卫星服务。
据报道,马斯克旗下社交平台𝕏(推特)日前调整了隐私政策,允许 𝕏 使用用户发布的信息来训练其人工智能(AI)模型。新的隐私政策将于 9 月 29 日生效。新政策规定,𝕏可能会使用所收集到的平台信息和公开可用的信息,来帮助训练 𝕏 的机器学习或人工智能模型。
9月2日,荣耀CEO赵明在采访中谈及华为手机回归时表示,替老同事们高兴,觉得手机行业,由于华为的回归,让竞争充满了更多的可能性和更多的魅力,对行业来说也是件好事。
《自然》30日发表的一篇论文报道了一个名为Swift的人工智能(AI)系统,该系统驾驶无人机的能力可在真实世界中一对一冠军赛里战胜人类对手。
近日,非营利组织纽约真菌学会(NYMS)发出警告,表示亚马逊为代表的电商平台上,充斥着各种AI生成的蘑菇觅食科普书籍,其中存在诸多错误。
社交媒体平台𝕏(原推特)新隐私政策提到:“在您同意的情况下,我们可能出于安全、安保和身份识别目的收集和使用您的生物识别信息。”
2023年德国柏林消费电子展上,各大企业都带来了最新的理念和产品,而高端化、本土化的中国产品正在不断吸引欧洲等国际市场的目光。
罗永浩日前在直播中吐槽苹果即将推出的 iPhone 新品,具体内容为:“以我对我‘子公司’的了解,我认为 iPhone 15 跟 iPhone 14 不会有什么区别的,除了序(列)号变了,这个‘不要脸’的东西,这个‘臭厨子’。