【C++】【网络】【Linux系统编程】单例模式，加锁封装TCP/IP协议套接字

个人主页：东洛的克莱斯韦克-CSDN博客

引言

有关套接字编程的细节和更多的系统调用课参考《UNIX环境高级编程》一书，可以在如下网站搜索电子版，该书在第16章详细的介绍了各种接口。

Jiumo Search 鸠摩搜索 - 文档搜索引擎 (jiumodiary.com)

而在实际的编程中，总是用系统调用的接口难免会有些繁琐。我们可以根据自己的需求，用面向对象的思想封装出接口简洁的类。

也可以把类设计成单例，以组件的形式供上层使用。而上层坐拥下三层协议栈，无需关心网络通信的细节。

获取套接字

首先要用socket接口获取套接字文件描述符

 int fd = socket(AF_INET, SOCK_STREAM, 0)

AF_INET表示IPv4因特网域

SOCK_STREAM表示该套接字的类型是全双工，面向字节流和链接的

第三个参数 0 表示根据前两个参数选择默认的协议，此时选择的就是TCP/IPv4协议栈

绑定套接字

bind(_socket_fd, (const struct sockaddr *)&address, (socklen_t)sizeof(address))

绑定套接字之前先用 struct sockaddr_in 结构体填充地址信息。

IPv4因特网域用struct sockaddr_in结构体填充，IPv6因特网域用struct sockaddr_in6结构体填充，为了使不同的地址格式能和套接字绑定，需要把对应的结构体强转成通用地址结构struct sockaddr。

   struct sockaddr_in address;
    memset(&address, 0, sizeof(address)); // 把结构体初始化为0

    address.sin_family = AF_INET;         // IPv4协议家族
    address.sin_addr.s_addr = INADDR_ANY; // 服务器选择任意IP地址接收请求
    address.sin_port = htons(_port);      // 端口号信息,网络序列

填充字段时，需要用htons()函数把端口号改为网络字节序列。在填充之前最好初始化结构体。

INADDR_ANY表示套接字可以接收上层进程所在服务器的任意一个IP地址（公网IP），一台服务器可能会配置多个IP地址。

表明允许监听

listen(_socket_fd, 5)

第一个参数是网络文件描述符。

第二个参数提供了一个提示给系统，表明这个套接字所期望的、还未处理的（即还在等待被接受的连接）连接请求的最大队列长度。换句话说，它告诉系统内核为这个套接字分配多大的空间来存储尚未处理的连接请求。第二个参数不是严格遵守的。

单例模式设计

我们需要把类的构造函数，拷贝函数，赋值重载设为私有。在类似添加一个该类类型的指针，该指针是静态的并且是私有成员。

那么只给外面暴露一个获取该指针的接口，并且只能被获取一次。

获取指针的接口就需要加锁，防止并发问题。【Linux】用5万字满足你对线程的所有♥幻想♥——【线程概念】【线程安全】【多线程并发】【互斥量】【条件变量】【信号量】【锁的原理】【各种锁】【生产者消费者模型】【读者写者问题】-CSDN博客

完整代码示例

套接字代码已经在网络环境中测试过了~

日志代码

//日志打印
#pragma once
#include <iostream>
#include <string>
#include <ctime>
#define INFO    1
#define WARNING 2
#define ERROR   3
#define FATAL   4
#define LOG(level, message) Log(#level, message, __FILE__, __LINE__)

void Log(std::string level, std::string message, std::string file_name, int line)
{
    std::cerr << "[" << level << "]" << "[" << time(nullptr) << "]" << "[" << message << "]" << "[" << file_name << "]" << "[" << line << "]" << std::endl;
}

套接字代码

#pragma once // 防止头文件被重复包含
#include <sys/socket.h>
#include <cstdlib>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <memory>
#include <pthread.h>
#include "log.hpp"

#define BACKLOG 5 // 全连接队列最小值
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;

class tcp_serve
{

public:
  static tcp_serve *git_tcp_object(int port) // 获取单例实例
  {
    if (tcp_ptr == nullptr)
    {
      pthread_mutex_lock(&mutex); // 加锁
      if (tcp_ptr == nullptr)
      {
        tcp_ptr = new tcp_serve(port);
        tcp_ptr->initserve();
      }
      pthread_mutex_unlock(&mutex); // 解锁
    }

    return tcp_ptr;
  }

  int git_socket_fd()
  {
    return _socket_fd;
  }

private:
  void initserve()
  {
    Socket();
    Bind();
    Listen();
  }

  tcp_serve(int port)
      : _port(port), _socket_fd(-1)
  {
  }

  ~tcp_serve()
  {
  }

  tcp_serve(const tcp_serve &x);
  const tcp_serve &operator=(const tcp_serve &x);

  void Socket()
  {
    int fd = socket(AF_INET, SOCK_STREAM, 0); // 用IPv4协议家族，TCP协议
    if (-1 == fd)
    {
      LOG(FATAL, "创建套接字失败");
      exit(1);
    } // 创建套接字失败

    _socket_fd = fd;
    LOG(INFO, "创建套接字成功");
  }

  void Bind()
  {
    struct sockaddr_in address;
    memset(&address, 0, sizeof(address)); // 把结构体初始化为0

    address.sin_family = AF_INET;         // IPv4协议家族
    address.sin_addr.s_addr = INADDR_ANY; // 服务器选择任意IP地址接收请求
    address.sin_port = htons(_port);      // 端口号信息,网络序列

    int opt = 1;
    if (setsockopt(_socket_fd, SOL_SOCKET, SO_REUSEADDR, (const void *)&opt, sizeof(opt)) < 0)
    {
      LOG(WARNING, "端口不能在短时间内连续绑定");
    } // 防止TIME_WAIT状态导致绑定端口失败
    LOG(INFO, "端口复用设置成功");
    if (-1 == bind(_socket_fd, (const struct sockaddr *)&address, (socklen_t)sizeof(address)))
    {
      LOG(FATAL, "绑定套接字失败");
      exit(2); // 绑定套接字失败
    }

    LOG(INFO, "套接字绑定成功");
  }

  void Listen()
  {
    if (-1 == listen(_socket_fd, BACKLOG))
    {

      LOG(FATAL, "声明自己可连接失败");
      exit(3);
    } // 请求链接失败

    LOG(INFO, "套接字允许监听链接");
  }

private:
  int _port;      // 端口
  int _socket_fd; // 套接字文件描述符
  static tcp_serve *tcp_ptr;
};

tcp_serve *tcp_serve::tcp_ptr = nullptr;