C++封装TcpSocket类
时间: 2024-01-16 15:05:25 浏览: 72
好的,以下是一个简单的TcpSocket类的封装示例:
```c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#define MAX_BUF_SIZE 1024
class TcpSocket {
private:
int m_sockfd;
public:
TcpSocket() {
m_sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (m_sockfd == -1) {
perror("socket");
exit(EXIT_FAILURE);
}
}
~TcpSocket() {
close(m_sockfd);
}
bool connect(const char* ip, int port) {
struct sockaddr_in server_addr;
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = inet_addr(ip);
server_addr.sin_port = htons(port);
if (::connect(m_sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
perror("connect");
return false;
}
return true;
}
bool send(const char* data, int len) {
int ret = ::send(m_sockfd, data, len, 0);
if (ret == -1) {
perror("send");
return false;
}
return true;
}
bool recv(char* buf, int& len) {
int ret = ::recv(m_sockfd, buf, MAX_BUF_SIZE, 0);
if (ret == -1) {
perror("recv");
return false;
}
len = ret;
return true;
}
};
```
这个类封装了常用的TCP socket操作,包括创建socket、连接服务器、发送数据、接收数据等。为了方便使用,我们在类内部定义了一个私有成员变量 `m_sockfd` 作为socket的文件描述符,在类的构造函数中初始化socket,析构函数中关闭socket。
在类的公有成员函数中,我们实现了 `connect`、`send`、`recv` 等函数。在 `connect` 函数中,我们通过调用 `connect` 系统调用来连接服务器,如果连接失败则返回 false。在 `send` 函数中,我们通过调用 `send` 系统调用来发送数据,如果发送失败则返回 false。在 `recv` 函数中,我们通过调用 `recv` 系统调用来接收数据,如果接收失败则返回 false。
当然,这只是一个简单的示例,实际上还有很多其他的细节需要考虑,例如错误处理、数据分包、粘包处理等。
阅读全文