零基础入门：用C++从零实现TCP Socket网络小工具

个人主页：chian-ocean

文章专栏-Linux

前言：

网络编程中的套接字（Socket）是通信的基本接口，允许不同计算机之间通过网络交换数据。套接字是计算机网络中通信的“端点”，通过它，应用程序可以与网络中的其他计算机进行数据通信。网络套接字接口提供了一种抽象的、平台无关的方式来进行进程间通信（IPC）或网络通信。

网络套接字接口

头文件

• 编写网络的常用的4个头文件，基本常用的函数都在这4个头文件里面。

#include <sys/types.h> // 包含各种系统数据类型
#include <sys/socket.h> // 包含套接字操作相关函数和常量
#include <arpa/inet.h> // 包含与Internet地址转换相关的函数
#include <netinet/in.h> // 定义与网络字节序及IPv4/IPv6地址相关的结构体和常量

接口

socket

socket() 函数是创建网络通信套接字的基础。它用于创建一个套接字（socket）并返回一个套接字描述符（socket descriptor），这个描述符将被用来进行后续的网络通信（例如发送和接收数据）。

int socket(int domain, int type, int protocol);

参数说明

返回值

• 成功时，socket() 返回一个 非负整数，这是一个套接字描述符，代表这个套接字。该描述符将用于后续的套接字操作（如绑定、连接、发送数据等）。
• 失败时，返回 -1，并且设置全局变量 errno 来指示错误类型。

bind()

int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);

参数说明

返回值

• 成功时：返回 0，表示成功将套接字与指定的本地地址绑定。
• 失败时：返回 -1，并将 errno 设置为具体的错误码。

listen()

int listen(int sockfd, int backlog);

参数说明

返回值

• 成功时：返回 0，表示成功将套接字转换为监听状态。
• 失败时：返回 -1，并设置 errno 以指示错误原因。

accept()

int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);

参数说明

返回值

• 成功时：返回 新的套接字描述符，用于与客户端进行通信。这个新的套接字是通过 accept() 函数创建的，它与原始的监听套接字不同，可以用于数据发送和接收。

• 失败时：返回 -1，并设置 errno 以指示错误原因。

connect()

参数说明

返回值

• 成功时：返回 0，表示连接成功。

• 失败时：返回 -1，并且会设置 errno 来指示错误的具体原因。

close()

int close(int fd);

参数说明

返回值

• 成功时：返回 0，表示成功关闭套接字或文件描述符。
• 失败时：返回 -1，并设置 errno 为具体的错误代码

网络套接字封装（TCP）

1. 头文件引用

#include <iostream>
#include <string>
#include <cstring>
#include <strings.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include "log.hpp"

• <iostream>：提供输入输出流的功能，常用于打印日志或错误信息。
• <string>：提供 C++ 标准库的字符串类 std::string，用于字符串处理。
• <cstring> 和 <strings.h>：用于处理字符串相关的操作，如 bzero() 和 strerror()。
• <sys/types.h> 和 <sys/socket.h>：提供套接字编程所需的类型定义和系统调用。
• <arpa/inet.h>：提供与 IP 地址转换相关的函数（如 inet_ntop() 和 inet_addr()）。
• <netinet/in.h>：定义了用于 IPv4 地址和端口的结构体和常量（如 sockaddr_in 和 htons()）。
• "log.hpp"：这是一个自定义的日志头文件，包含了日志记录相关的内容。lg() 函数用于记录日志，lg() 宏应该在 log.hpp 中定义。

2. 全局变量和枚举类型

int backlog = 10;

enum err
{
Socketerr = 1,
Bindeterr,
Listeneterr,
Accepteterr,
};

• backlog：这是传递给 listen() 函数的参数，定义了监听队列的最大长度（即最大客户端连接数）。设置为 10。
• enum err：定义了与套接字相关的错误类型。
  • Socketerr = 1：表示套接字创建失败。
  • Bindeterr：表示套接字绑定失败。
  • Listeneterr：表示监听失败。
  • Accepteterr：表示接受客户端连接失败。

3. Sock 类

3.1 构造函数和析构函数

Sock() {}
~Sock() {}

• 构造函数：默认构造函数，没有进行任何初始化操作。
• 析构函数：默认析构函数，没有执行任何资源清理操作。

3.2 Socket() – 创建套接字

void Socket()
{
sockfd_ = socket(AF_INET, SOCK_STREAM, 0);
if(sockfd_ < 0)
{
lg(FATAL,"Socket error: %d,%s",errno,strerror(errno));
exit(Socketerr);
}
}

• 目的：创建一个 TCP 套接字。
  • AF_INET：表示 IPv4 地址族。
  • SOCK_STREAM：表示 TCP 流套接字（面向连接的套接字）。
  • 0：表示默认协议，通常是 TCP 协议。
• 错误处理：如果 socket() 返回值小于 0，表示套接字创建失败，记录日志并退出程序，退出代码为 Socketerr。

3.3 Bind(uint16_t port) – 绑定套接字

void Bind(uint16_t port)
{
struct sockaddr_in peer;
socklen_t len = sizeof(peer);
bzero(&peer,len);
peer.sin_port = htons(port);
peer.sin_family = AF_INET;
peer.sin_addr.s_addr = INADDR_ANY;

if(bind(sockfd_,(struct sockaddr *)&(peer),len) < 0)
{
lg(FATAL,"Bind error: %d,%s",errno,strerror(errno));
exit(Bindeterr);
}
}

• 目的：将套接字与本地 IP 地址和端口号绑定。通过 INADDR_ANY 将套接字绑定到所有可用的网络接口上，接受来自任何 IP 地址的连接。
  • htons()：将端口号从主机字节序转换为网络字节序。
• 错误处理：如果 bind() 返回值小于 0，表示绑定失败，记录日志并退出程序，退出代码为 Bindeterr。

3.4 Listen() – 开始监听

void Listen()
{
if(listen(sockfd_, backlog) < 0)
{
lg(FATAL,"Listen error: %d,%s",errno,strerror(errno));
exit(Listeneterr);
}
}

• 目的：将套接字设置为监听状态，准备接受客户端的连接。
  • backlog：监听队列的最大长度，定义最多能排队等待的连接数。
• 错误处理：如果 listen() 返回值小于 0，表示监听失败，记录日志并退出程序，退出代码为 Listeneterr。

3.5 Accept(std::string \\* clientip, uint16_t\\* clientport) – 接受连接

int Accept(std::string * clientip, uint16_t* clientport)
{
struct sockaddr_in peer;
socklen_t len = sizeof(peer);
bzero(&peer,len);

int newfd = accept(sockfd_,(struct sockaddr*)&(peer),&len);
if(newfd < 0)
{
lg(FATAL,"Accept error: %d,%s",errno,strerror(errno));
exit(Accepteterr);
}

char ip[64];
inet_ntop(AF_INET,&peer.sin_addr.s_addr,ip,sizeof(ip));
*clientip = ip;
*clientport = ntohs(peer.sin_port);

return newfd;
}

• 目的：接受来自客户端的连接请求，并返回一个新的套接字用于与客户端的通信。
  • accept() 函数返回一个新的套接字 newfd，用于与客户端交换数据。
  • 通过 inet_ntop() 将客户端的 IP 地址从二进制转换为字符串格式，ntohs() 将客户端的端口号转换为主机字节序。
• 错误处理：如果 accept() 返回值小于 0，表示接受连接失败，记录日志并退出程序，退出代码为 Accepteterr。

3.6 Connect(const std::string& ip, const uint16_t& port) – 连接服务器

bool Connect(const std::string& ip,const uint16_t& port)
{
struct sockaddr_in peer;
socklen_t len = sizeof(peer);
bzero(&peer,len);
peer.sin_addr.s_addr = inet_addr(ip.c_str());
peer.sin_port = htons(port);
peer.sin_family = AF_INET;

int n = connect(sockfd_,(struct sockaddr*)&(peer),len);
if(n < 0)
{
lg(WARNING,"Connect error: %d,%s",errno,strerror(errno));
return false;
}

return true;
}

• 目的：客户端通过此函数连接到远程服务器，指定服务器的 IP 地址和端口。
  • inet_addr()：将 IP 地址从字符串转换为网络字节序的二进制格式。
  • htons()：将端口号转换为网络字节序。
• 错误处理：如果 connect() 失败，记录警告日志并返回 false，否则返回 true 表示连接成功。

3.7 GetFd() – 获取套接字描述符

int GetFd()
{
return sockfd_;
}

• 目的：返回套接字描述符，便于外部访问该套接字，用于进一步的操作（如 send()，recv() 等）。

网络小组件链接