好的，下面是一个简单的封装
TcpSocket类的例子：
```c++
#include <iostream>
#include <string>
#include <cstring>
#include <sys/
socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
class
TcpSocket {
public:
TcpSocket() : m_sockfd(-1) {}
~
TcpSocket() {
close();
}
bool create() {
m_sockfd =
socket(AF_INET, SOCK_STREAM, 0);
if (m_sockfd == -1) {
return false;
}
return true;
}
bool bind(const std::string& ip, int port) {
sockaddr_in addr;
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = inet_addr(ip.c_str());
addr.sin_port = htons(port);
int ret = ::bind(m_sockfd, (sockaddr*)&addr, sizeof(addr));
if (ret == -1) {
return false;
}
return true;
}
bool listen(int back
log) {
int ret = ::listen(m_sockfd, back
log);
if (ret == -1) {
return false;
}
return true;
}
bool connect(const std::string& ip, int port) {
sockaddr_in addr;
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = inet_addr(ip.c_str());
addr.sin_port = htons(port);
int ret = ::connect(m_sockfd, (sockaddr*)&addr, sizeof(addr));
if (ret == -1) {
return false;
}
return true;
}
bool accept(
TcpSocket& clientSock) {
sockaddr_in addr;
socklen_t len = sizeof(addr);
int clientfd = ::accept(m_sockfd, (sockaddr*)&addr, &len);
if (clientfd == -1) {
return false;
}
clientSock.setSockfd(clientfd);
clientSock.setAddr(addr);
return true;
}
bool send(const std::string& data) {
int ret = ::send(m_sockfd, data.c_str(), data.size(), 0);
if (ret == -1) {
return false;
}
return true;
}
bool recv(std::string& data) {
char buf[1024] = {0};
int ret = ::recv(m_sockfd, buf, 1024, 0);
if (ret == -1) {
return false;
}
else if (ret == 0) {
return false;
}
data = buf;
return true;
}
void close() {
if (m_sockfd != -1) {
::close(m_sockfd);
m_sockfd = -1;
}
}
int getSockfd() const {
return m_sockfd;
}
void setSockfd(int sockfd) {
m_sockfd = sockfd;
}
void setAddr(const sockaddr_in& addr) {
m_addr = addr;
}
std::string getPeerIp() const {
return inet_ntoa(m_addr.sin_addr);
}
int getPeerPort() const {
return ntohs(m_addr.sin_port);
}
private:
int m_sockfd;
sockaddr_in m_addr;
};
int main() {
TcpSocket serverSock;
if (!serverSock.create()) {
std::cout << "create
socket failed!" << std::endl;
return -1;
}
if (!serverSock.bind("0.0.0.0", 8888)) {
std::cout << "bind
socket failed!" << std::endl;
return -1;
}
if (!serverSock.listen(5)) {
std::cout << "listen
socket failed!" << std::endl;
return -1;
}
while (1) {
TcpSocket clientSock;
if (!serverSock.accept(clientSock)) {
std::cout << "accept
socket failed!" << std::endl;
continue;
}
std::cout << "client ip: " << clientSock.getPeerIp() << ", port: " << clientSock.getPeerPort() << std::endl;
std::string data;
if (!clientSock.recv(data)) {
std::cout << "recv data failed!" << std::endl;
continue;
}
std::cout << "recv data: " << data << std::endl;
if (!clientSock.send("hello client!")) {
std::cout << "send data failed!" << std::endl;
continue;
}
}
return 0;
}
```
这个例子中，
TcpSocket类封装了一些常用的
TCP socket操作，包括创建
socket、绑定端口、监听端口、连接远程主机、接收连接、发送数据和接收数据等。可以根据需求进行扩展。
702
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
