C++封装TcpSocket类
时间: 2024-01-16 11:05:25 浏览: 32
好的,以下是一个简单的封装TcpSocket类的C代码示例:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define MAX_BUF_SIZE 1024
class TcpSocket {
private:
int m_sockfd; // socket文件描述符
public:
// 构造函数,初始化socket文件描述符
TcpSocket() {
m_sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (m_sockfd == -1) {
perror("socket failed");
exit(EXIT_FAILURE);
}
}
// 连接服务器
void connectServer(const char* ip, unsigned short 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 failed");
exit(EXIT_FAILURE);
}
}
// 发送数据
void sendData(const char* data, int len) {
int n = send(m_sockfd, data, len, 0);
if (n == -1) {
perror("send failed");
exit(EXIT_FAILURE);
}
}
// 接收数据
int recvData(char* buf, int len) {
int n = recv(m_sockfd, buf, len, 0);
if (n == -1) {
perror("recv failed");
exit(EXIT_FAILURE);
}
return n;
}
// 关闭socket文件描述符
void closeSocket() {
close(m_sockfd);
}
};
```
使用示例:
```c
int main() {
TcpSocket sock;
sock.connectServer("127.0.0.1", 8080);
char buf[MAX_BUF_SIZE];
memset(buf, 0, sizeof(buf));
sock.sendData("hello, server", strlen("hello, server"));
int n = sock.recvData(buf, MAX_BUF_SIZE);
printf("received data: %s\n", buf);
sock.closeSocket();
return 0;
}
```