【Linux】Socket 编程 TCP

打印 上一主题 下一主题

主题 949|帖子 949|积分 2847

一.TCP socket API 详解

  1. int socket(int domain, int type, int protocol);
复制代码


  • socket() 打开一个网络通讯端口,如果成功的话,就像 open()一样返回一个文件描述符。
  • 应用程序可以像读写文件一样用 read/write 在网络上收发数据。
  • 如果 socket() 调用出错则返回-1。
  • 对于 IPv4 网络通信,domain 参数指定为 AF_INET。
  • 对于 TCP 协议,type 参数指定为 SOCK_STREAM,表示面向字节流的传输协议。
  • protocol 参数的先容从略,指定为 0 即可。
  1. int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
复制代码


  • 服务器程序所监听的网络地址和端口号通常是固定稳固的,客户端程序得知服务器程序的地址和端口号后就可以向服务器发起连接。服务器需要调用 bind 绑定网络地址(INADDR_ANY)和固定的端口号。
  • bind()成功返回 0,失败返回-1。
  • bind()的作用是将参数 sockfd 和 myaddr 绑定在一起,使 sockfd 这个用于网络通讯的文件描述符监听 myaddr 所描述的地址和端口号。
  • struct sockaddr* 是一个通用指针类型,myaddr 参数现实上可以接受多种协议的 sockaddr 结构体(例如网络通信用 struct sockaddr_in*),而它们的长度各不雷同,以是需要第三个参数 addrlen指定结构体的长度。
我们的程序中对 myaddr 参数是如许初始化的:


  • 将整个结构体清零。
  • 设置地址类型为 AF_INET。
  • 网络地址为 INADDR_ANY,这个宏表示本地的任意 IP 地址,由于服务器可能有多个网卡,每个网卡也可能绑定多个 IP 地址,如许设置可以在所有的 IP 地址上监听,直到与某个客户端建立了连接时才确定下来到底用哪个 IP 地址。
  • 端口号为 SERV_PORT,我们界说为 8080。
  1. int listen(int sockfd, int backlog);
复制代码


  • listen()声明 sockfd 处于监听状态,而且最多答应有 backlog 个客户端处于连接等待状态(它们位于监听队列中),如果接收到更多的连接哀求就忽略,这里设置不会太大,一样平常是 5。
  • listen()成功返回 0,失败返回-1。
  1. int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);
复制代码


  • 三次握手完成后,服务器调用 accept()接受连接。
  • 如果服务器调用 accept()时还没有客户端的连接哀求,就阻塞等待直到有客户端连接上来。
  • addr 是一个输出型参数,accept()返回时传出客户端的地址和端口号。
  • 如果给 addr 参数传 nullptr,表示不关心客户端的地址。
  • addrlen 参数是一个输入输出参数,输入的是调用者提供的,缓冲区 addr 的长度以避免缓冲区溢出题目,输出的是客户端地址结构体的现实长度。
  • 留意:输入的 sockfd 是监听套接字,accept 成功时:返回的是现实进行网络通信的套接字。

  1. int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
复制代码


  • 客户端需要调用 connect()连接服务器。
  • connect 和 bind 的参数形式一致,区别在于 bind 的参数是本身的地址,而connect 的参数是对方的地址。
  • connect()成功返回 0,出错返回-1。
二.EchoServer(单历程/多历程/多线程/线程池)

1.Makefile

  1. .PHONY:all
  2. all:server_tcp client_tcp
  3. client_tcp:TcpClient.cc
  4.         g++ -o $@ $^ -std=c++17
  5. server_tcp:TcpServer.cc
  6.         g++ -o $@ $^ -std=c++17 -lpthread
  7. .PHONY:clean
  8. clean:
  9.         rm -f client_tcp server_tcp
复制代码
2.Mutex.hpp

  1. #pragma once
  2. #include <pthread.h>
  3. namespace MutexModule
  4. {
  5.     class Mutex
  6.     {
  7.         Mutex(const Mutex &m) = delete;
  8.         const Mutex &operator=(const Mutex &m) = delete;
  9.     public:
  10.         Mutex()
  11.         {
  12.             ::pthread_mutex_init(&_mutex, nullptr);
  13.         }
  14.         ~Mutex()
  15.         {
  16.             ::pthread_mutex_destroy(&_mutex);
  17.         }
  18.         void Lock()
  19.         {
  20.             ::pthread_mutex_lock(&_mutex);
  21.         }
  22.         void Unlock()
  23.         {
  24.             ::pthread_mutex_unlock(&_mutex);
  25.         }
  26.         pthread_mutex_t *LockAddr() { return &_mutex; }
  27.     private:
  28.         pthread_mutex_t _mutex;
  29.     };
  30.     class LockGuard
  31.     {
  32.     public:
  33.         LockGuard(Mutex &mutex)
  34.             : _mutex(mutex)
  35.         {
  36.             _mutex.Lock();
  37.         }
  38.         ~LockGuard()
  39.         {
  40.             _mutex.Unlock();
  41.         }
  42.     private:
  43.         Mutex &_mutex; // 使用引用: 互斥锁不支持拷贝
  44.     };
  45. }
复制代码
3.Cond.hpp

  1. #pragma
  2. #include <pthread.h>
  3. #include "Mutex.hpp"
  4. namespace CondModule
  5. {
  6.     using namespace MutexModule;
  7.     class Cond
  8.     {
  9.     public:
  10.         Cond()
  11.         {
  12.             ::pthread_cond_init(&_cond, nullptr);
  13.         }
  14.         ~Cond()
  15.         {
  16.             ::pthread_cond_destroy(&_cond);
  17.         }
  18.         void Wait(Mutex &mutex) // 线程释放曾经持有的锁, 不能拷贝
  19.         {
  20.             ::pthread_cond_wait(&_cond, mutex.LockAddr());
  21.         }
  22.         void Signal()
  23.         {
  24.             ::pthread_cond_signal(&_cond);
  25.         }
  26.         void Broadcast()
  27.         {
  28.             ::pthread_cond_broadcast(&_cond);
  29.         }
  30.     private:
  31.         pthread_cond_t _cond;
  32.     };
  33. }
复制代码
4.Log.hpp

  1. #pragma once
  2. #include <iostream>
  3. #include <cstdio>
  4. #include <string>
  5. #include <filesystem>
  6. #include <fstream>
  7. #include <sstream>
  8. #include <memory>
  9. #include <unistd.h>
  10. #include <time.h>
  11. #include "Mutex.hpp"
  12. namespace LogModule
  13. {
  14.     using namespace MutexModule;
  15.     // 获取系统时间
  16.     std::string CurrentTime()
  17.     {
  18.         time_t time_stamp = ::time(nullptr); // 获取时间戳
  19.         struct tm curr;
  20.         localtime_r(&time_stamp, &curr); // 将时间戳转化为可读性强的信息
  21.         char buffer[1024];
  22.         snprintf(buffer, sizeof(buffer), "%4d-%02d-%02d %02d:%02d:%02d",
  23.                  curr.tm_year + 1900,
  24.                  curr.tm_mon + 1,
  25.                  curr.tm_mday,
  26.                  curr.tm_hour,
  27.                  curr.tm_min,
  28.                  curr.tm_sec);
  29.         return buffer;
  30.     }
  31.     // 日志文件: 默认路径和默认文件名
  32.     const std::string defaultlogpath = "./log/";
  33.     const std::string defaultlogname = "log.txt";
  34.     // 日志等级
  35.     enum class LogLevel
  36.     {
  37.         DEBUG = 1,
  38.         INFO,
  39.         WARNING,
  40.         ERROR,
  41.         FATAL
  42.     };
  43.     std::string Level2String(LogLevel level)
  44.     {
  45.         switch (level)
  46.         {
  47.         case LogLevel::DEBUG:
  48.             return "DEBUG";
  49.         case LogLevel::INFO:
  50.             return "INFO";
  51.         case LogLevel::WARNING:
  52.             return "WARNING";
  53.         case LogLevel::ERROR:
  54.             return "ERROR";
  55.         case LogLevel::FATAL:
  56.             return "FATAL";
  57.         default:
  58.             return "NONE";
  59.         }
  60.     }
  61.     // 3. 策略模式: 刷新策略
  62.     class LogStrategy
  63.     {
  64.     public:
  65.         virtual ~LogStrategy() = default;
  66.         // 纯虚函数: 无法实例化对象, 派生类可以重载该函数, 实现不同的刷新方式
  67.         virtual void SyncLog(const std::string &message) = 0;
  68.     };
  69.     // 3.1 控制台策略
  70.     class ConsoleLogStrategy : public LogStrategy
  71.     {
  72.     public:
  73.         ConsoleLogStrategy() {}
  74.         ~ConsoleLogStrategy() {}
  75.         void SyncLog(const std::string &message) override
  76.         {
  77.             LockGuard lockguard(_mutex);
  78.             std::cout << message << std::endl;
  79.         }
  80.     private:
  81.         Mutex _mutex;
  82.     };
  83.     // 3.2 文件级(磁盘)策略
  84.     class FileLogStrategy : public LogStrategy
  85.     {
  86.     public:
  87.         FileLogStrategy(const std::string &logpath = defaultlogpath, const std::string &logname = defaultlogname)
  88.             : _logpath(logpath), _logname(logname)
  89.         {
  90.             // 判断_logpath目录是否存在
  91.             if (std::filesystem::exists(_logpath))
  92.             {
  93.                 return;
  94.             }
  95.             try
  96.             {
  97.                 std::filesystem::create_directories(_logpath);
  98.             }
  99.             catch (std::filesystem::filesystem_error &e)
  100.             {
  101.                 std::cerr << e.what() << std::endl;
  102.             }
  103.         }
  104.         ~FileLogStrategy() {}
  105.         void SyncLog(const std::string &message) override
  106.         {
  107.             LockGuard lockguard(_mutex);
  108.             std::string log = _logpath + _logname;
  109.             std::ofstream out(log, std::ios::app); // 以追加的方式打开文件
  110.             if (!out.is_open())
  111.             {
  112.                 return;
  113.             }
  114.             out << message << "\n"; // 将信息刷新到out流中
  115.             out.close();
  116.         }
  117.     private:
  118.         std::string _logpath;
  119.         std::string _logname;
  120.         Mutex _mutex;
  121.     };
  122.     // 4. 日志类: 构建日志字符串, 根据策略进行刷新
  123.     class Logger
  124.     {
  125.     public:
  126.         Logger()
  127.         {
  128.             // 默认往控制台上刷新
  129.             _strategy = std::make_shared<ConsoleLogStrategy>();
  130.         }
  131.         ~Logger() {}
  132.         void EnableConsoleLog()
  133.         {
  134.             _strategy = std::make_shared<ConsoleLogStrategy>();
  135.         }
  136.         void EnableFileLog()
  137.         {
  138.             _strategy = std::make_shared<FileLogStrategy>();
  139.         }
  140.         // 内部类: 记录完整的日志信息
  141.         class LogMessage
  142.         {
  143.         public:
  144.             LogMessage(LogLevel level, const std::string &filename, int line, Logger &logger)
  145.                 : _currtime(CurrentTime()), _level(level), _pid(::getpid())
  146.                 , _filename(filename), _line(line), _logger(logger)
  147.             {
  148.                 std::stringstream ssbuffer;
  149.                 ssbuffer << "[" << _currtime << "] "
  150.                          << "[" << Level2String(_level) << "] "
  151.                          << "[" << _pid << "] "
  152.                          << "[" << _filename << "] "
  153.                          << "[" << _line << "] - ";
  154.                 _loginfo = ssbuffer.str();
  155.             }
  156.             ~LogMessage()
  157.             {
  158.                 if(_logger._strategy)
  159.                 {
  160.                     _logger._strategy->SyncLog(_loginfo);
  161.                 }
  162.             }
  163.             template <class T>
  164.             LogMessage &operator<<(const T &info)
  165.             {
  166.                 std::stringstream ssbuffer;
  167.                 ssbuffer << info;
  168.                 _loginfo += ssbuffer.str();
  169.                 return *this;
  170.             }
  171.         private:
  172.             std::string _currtime;  // 当前日志时间
  173.             LogLevel _level;       // 日志水平
  174.             pid_t _pid;            // 进程pid
  175.             std::string _filename; // 文件名
  176.             uint32_t _line;        // 日志行号
  177.             Logger &_logger;       // 负责根据不同的策略进行刷新
  178.             std::string _loginfo;  // 日志信息
  179.         };
  180.         // 故意拷贝, 形成LogMessage临时对象, 后续在被<<时,会被持续引用,
  181.         // 直到完成输入,才会自动析构临时LogMessage, 至此完成了日志的刷新,
  182.         // 同时形成的临时对象内包含独立日志数据, 未来采用宏替换, 获取文件名和代码行数
  183.         LogMessage operator()(LogLevel level, const std::string &filename, int line)
  184.         {
  185.             return LogMessage(level, filename, line, *this);
  186.         }
  187.     private:
  188.         // 纯虚类不能实例化对象, 但是可以定义指针
  189.         std::shared_ptr<LogStrategy> _strategy; // 日志刷新策略方案
  190.     };
  191.     // 定义全局logger对象
  192.     Logger logger;
  193. // 编译时进行宏替换: 方便随时获取行号和文件名
  194. #define LOG(level) logger(level, __FILE__, __LINE__)
  195. // 提供选择使用何种日志策略的方法
  196. #define ENABLE_CONSOLE_LOG() logger.EnableConsoleLog()
  197. #define ENABLE_FILE_LOG() logger.EnableFileLog()
  198. }
复制代码
5.Thread.hpp

  1. #pragma once
  2. #include <iostream>
  3. #include <string>
  4. #include <functional>
  5. #include <pthread.h>
  6. #include <sys/types.h>
  7. #include <unistd.h>
  8. namespace ThreadModule
  9. {
  10.     using func_t = std::function<void(std::string)>;
  11.     static int number = 1;
  12.     // 强类型枚举: 枚举的成员名称被限定在枚举类型的作用域内
  13.     enum class TSTATUS
  14.     {
  15.         NEW,
  16.         RUNNING,
  17.         STOP
  18.     };
  19.     class Thread
  20.     {
  21.     private:
  22.         // 成员方法: 需要加上static表示不需要this指针, 否则回调函数报错
  23.         // 而要执行_func()函数又需要由this指针, 所以Routine函数传this指针
  24.         static void *Routine(void *args)
  25.         {
  26.             Thread *t = static_cast<Thread *>(args);
  27.             t->_func(t->Name());
  28.             return nullptr;
  29.         }
  30.         void EnableDetach() { _joinable = false; }
  31.     public:
  32.         Thread(func_t func)
  33.             : _func(func), _status(TSTATUS::NEW), _joinable(true)
  34.         {
  35.             _name = "Thread-" + std::to_string(number++);
  36.             _pid = getpid();
  37.         }
  38.         ~Thread() {}
  39.         // 线程创建
  40.         bool Start()
  41.         {
  42.             if (_status != TSTATUS::RUNNING)
  43.             {
  44.                 int n = pthread_create(&_tid, nullptr, Routine, this);
  45.                 if (n != 0)
  46.                     return false;
  47.                 _status = TSTATUS::RUNNING;
  48.                 return true;
  49.             }
  50.             return false;
  51.         }
  52.         // 线程退出
  53.         bool Stop()
  54.         {
  55.             if (_status == TSTATUS::RUNNING)
  56.             {
  57.                 int n = ::pthread_cancel(_tid);
  58.                 if (n != 0)
  59.                     return false;
  60.                 _status = TSTATUS::STOP;
  61.                 return true;
  62.             }
  63.             return false;
  64.         }
  65.         // 线程等待
  66.         bool Join()
  67.         {
  68.             if (_joinable)
  69.             {
  70.                 int n = ::pthread_join(_tid, nullptr);
  71.                 if (n != 0)
  72.                     return false;
  73.                 _status = TSTATUS::STOP;
  74.                 return true;
  75.             }
  76.             return false;
  77.         }
  78.         // 线程分离
  79.         bool Detach()
  80.         {
  81.             EnableDetach();
  82.             int n = ::pthread_detach(_tid);
  83.             if (n != 0)
  84.                 return false;
  85.             return true;
  86.         }
  87.         // 线程是否分离
  88.         bool IsJoinable() {  return _joinable; }
  89.         std::string Name() { return _name; }
  90.     private:
  91.         std::string _name;
  92.         pthread_t _tid;
  93.         pid_t _pid;
  94.         bool _joinable; // 线程是否是分离的, 默认不是
  95.         func_t _func;
  96.         TSTATUS _status;
  97.     };
  98. }
复制代码
6.ThreadPool.hpp

  1. #pragma once
  2. #include <iostream>
  3. #include <string>
  4. #include <vector>
  5. #include <queue>
  6. #include <memory>
  7. #include "Mutex.hpp"
  8. #include "Cond.hpp"
  9. #include "Thread.hpp"
  10. #include "Log.hpp"
  11. namespace ThreadPoolModule
  12. {
  13.     using namespace MutexModule;
  14.     using namespace CondModule;
  15.     using namespace ThreadModule;
  16.     using namespace LogModule;
  17.     using thread_t = std::shared_ptr<Thread>;
  18.     const static int defaultnum = 15;
  19.     template <class T>
  20.     class ThreadPool
  21.     {
  22.     private:
  23.         bool IsEmpty() { return _taskq.empty(); }
  24.         void HandlerTask(std::string name)
  25.         {
  26.             LOG(LogLevel::INFO) << "线程: " << name << ", 进入HandlerTask执行逻辑";
  27.             while (true)
  28.             {
  29.                 // 1. 拿任务: 访问共享资源, 需要加锁
  30.                 T task;
  31.                 {
  32.                     LockGuard lockguard(_mutex);
  33.                     while (IsEmpty() && _isrunning) // while替代if: 防止伪唤醒
  34.                     {
  35.                         _wait_num++;
  36.                         _cond.Wait(_mutex); // 没任务时: 线程在条件变量上阻塞等待
  37.                         _wait_num--;
  38.                     }
  39.                     // 2. 任务队列不为空 && 线程池退出
  40.                     if (IsEmpty() && !_isrunning)
  41.                         break;
  42.                     task = _taskq.front();
  43.                     _taskq.pop();
  44.                 }
  45.                 // 3. 处理任务: 并发处理, 不需要持有锁
  46.                 task();
  47.             }
  48.             LOG(LogLevel::INFO) << "线程: " << name << ", 退出";
  49.         }
  50.         ThreadPool(int num = defaultnum)
  51.             : _num(num), _wait_num(0), _isrunning(false)
  52.         {
  53.             for (int i = 0; i < _num; i++)
  54.             {
  55.                 // 在类中: bind类的公有方法, 需要取地址 + 传入this指针
  56.                 // 在类外: bind类的公有方法, 需要取地址 + 传入类的匿名对象
  57.                 _threads.push_back(std::make_shared<Thread>(std::bind(&ThreadPool::HandlerTask, this, std::placeholders::_1))); // push_back()会调用移动构造
  58.                 LOG(LogLevel::INFO) << "构建线程" << _threads.back()->Name() << "对象...成功";
  59.             }
  60.         }
  61.         ThreadPool<T>(const ThreadPool<T> &) = delete;
  62.         ThreadPool<T> &operator=(const ThreadPool<T> &) = delete;
  63.     public:
  64.         ~ThreadPool() {}
  65.         // 获取单例对象
  66.         static ThreadPool<T> *GetInstance()
  67.         {
  68.             // 若单例为空: 需要加锁创建单例对象
  69.             if(instance == nullptr)
  70.             {
  71.                 LockGuard lockguard(_lock);
  72.                 if(instance == nullptr)
  73.                 {
  74.                     LOG(LogLevel::INFO) << "单例首次被执行, 需要加载对象...";
  75.                     instance = new ThreadPool<T>();
  76.                     instance->Start();
  77.                 }
  78.             }
  79.             // 若单例不为空: 直接返回单例对象
  80.             return instance;
  81.         }
  82.         void Equeue(T in)
  83.         {
  84.             LockGuard lockguard(_mutex);
  85.             if (!_isrunning) return;
  86.             _taskq.push(in);
  87.             if (_wait_num > 0)
  88.             {
  89.                 _cond.Signal(); // 唤醒线程
  90.             }
  91.         }
  92.         void Start()
  93.         {
  94.             if (_isrunning) return;
  95.             _isrunning = true;
  96.             for (auto &thread_ptr : _threads)
  97.             {
  98.                 thread_ptr->Start();
  99.                 LOG(LogLevel::INFO) << "启动线程" << thread_ptr->Name() << "...成功";
  100.             }
  101.         }
  102.         void Stop()
  103.         {
  104.             LockGuard lockguard(_mutex);
  105.             if (_isrunning)
  106.             {
  107.                 // 1. 不能再新增任务了
  108.                 _isrunning = false;
  109.                 // 2. 让线程自己退出(唤醒所有的线程) && 历史任务被执行完
  110.                 if (_wait_num > 0)
  111.                 {
  112.                     _cond.Broadcast();
  113.                 }
  114.             }
  115.         }
  116.         void Wait()
  117.         {
  118.             for (auto &thread_ptr : _threads)
  119.             {
  120.                 thread_ptr->Join();
  121.                 LOG(LogLevel::INFO) << "回收线程" << thread_ptr->Name() << "...成功";
  122.             }
  123.         }
  124.     private:
  125.         int _num;                       // 线程的个数
  126.         std::vector<thread_t> _threads; // 线程池
  127.         std::queue<T> _taskq;           // 共享资源: 任务队列
  128.         int _wait_num;                  // 等待的线程数目
  129.         bool _isrunning;                // 线程池是否运行
  130.         Mutex _mutex; // 锁
  131.         Cond _cond;   // 条件变量
  132.         static ThreadPool<T> *instance; // 单例对象
  133.         static Mutex _lock;             // 用来保护单例
  134.     };
  135.     // 静态成员: 类内声明, 类外定义
  136.     template<class T>
  137.     ThreadPool<T> *ThreadPool<T>::instance = nullptr;
  138.    
  139.     template<class T>
  140.     Mutex ThreadPool<T>::_lock;
  141. }
复制代码
7.InetAddr.hpp

  1. #pragma once
  2. #include <iostream>
  3. #include <string>
  4. #include <sys/types.h>
  5. #include <sys/socket.h>
  6. #include <netinet/in.h>
  7. #include <arpa/inet.h>
  8. #include "Common.hpp"
  9. class InetAddr
  10. {
  11. private:
  12.     // 端口号: 网络序列->主机序列
  13.     void PortNetToHost()
  14.     {
  15.         _port = ::ntohs(_net_addr.sin_port);
  16.     }
  17.     // IP: 网络序列->主机序列
  18.     void IpNetToHost()
  19.     {
  20.         char ipbuffer[64];
  21.         ::inet_ntop(AF_INET, &_net_addr.sin_addr, ipbuffer, sizeof(ipbuffer));
  22.         _ip = ipbuffer;
  23.     }
  24. public:
  25.     InetAddr() {}
  26.     InetAddr(const struct sockaddr_in &addr)
  27.         : _net_addr(addr)
  28.     {
  29.         PortNetToHost();
  30.         IpNetToHost();
  31.     }
  32.     InetAddr(uint16_t port)
  33.         : _port(port), _ip("")
  34.     {
  35.         _net_addr.sin_family = AF_INET;
  36.         _net_addr.sin_port = ::htons(_port);
  37.         _net_addr.sin_addr.s_addr = INADDR_ANY;
  38.     }
  39.     ~InetAddr() {}
  40.     bool operator==(const InetAddr& addr) { return _ip == addr._ip && _port == addr._port; }
  41.     struct sockaddr *NetAddr() { return CONV(&_net_addr); }
  42.     socklen_t NetAddrLen() { return sizeof(_net_addr); }
  43.     std::string Ip() { return _ip; }
  44.     uint16_t Port() { return _port; }
  45.     std::string Addr() { return Ip() + ":" + std::to_string(Port()); }
  46. private:
  47.     struct sockaddr_in _net_addr;
  48.     std::string _ip; // 主机序列: IP
  49.     uint16_t _port;  // 主机序列: 端口号
  50. };
复制代码
7.CommandExecute.hpp

  1. #pragma once
  2. #include <iostream>
  3. #include <string>
  4. #include <set>
  5. #include <cstdlib>
  6. const int line_size = 1024;
  7. class Command
  8. {
  9. public:
  10.     Command()
  11.     {
  12.         _white_list.insert("ls");
  13.         _white_list.insert("pwd");
  14.         _white_list.insert("ls -l");
  15.         _white_list.insert("touch file.txt");
  16.         _white_list.insert("who");
  17.         _white_list.insert("whoami");
  18.     }
  19.     bool SafeCheck(const std::string &cmdstr)
  20.     {
  21.         auto iter = _white_list.find(cmdstr);
  22.         return iter == _white_list.end() ? false : true;
  23.     }
  24.     // 给出命令字符串, 返回执行后的结果
  25.     std::string Execute(std::string cmdstr)
  26.     {
  27.         // 1. pipe
  28.         // 2. fork + dup2(pipe[1], 1) + exec*, 执行结果给父进程, pipe[0]
  29.         // 3. return
  30.         // FILE *popen(const char *command, const char *type);
  31.         // int pclose(FILE * stream);
  32.         if(!SafeCheck(cmdstr))
  33.         {
  34.             return std::string(cmdstr + "不支持");
  35.         }
  36.         // 以读模式打开管道,执行命令
  37.         FILE *fp = ::popen(cmdstr.c_str(), "r");
  38.         if (fp == nullptr)
  39.         {
  40.             return std::string("fail");
  41.         }
  42.         char buffer[line_size];
  43.         std::string result;
  44.         while (true)
  45.         {
  46.             // 从管道中读取命令的输出
  47.             char *ret = ::fgets(buffer, sizeof(buffer), fp); // 自动添加'\0'
  48.             if (ret == nullptr) break;
  49.             result += ret;
  50.         }
  51.         ::pclose(fp);
  52.         return result.empty() ? std::string("Done") : result;
  53.     }
  54. private:
  55.     std::set<std::string> _white_list; // 命令白名单: 允许执行白名单中的命令
  56. };
复制代码
9.TcpServer.hpp

  1. #pragma once
  2. #include <iostream>
  3. #include <string>
  4. #include <cstring>
  5. #include <cerrno>
  6. #include <functional>
  7. #include <pthread.h>
  8. #include <signal.h>
  9. #include <sys/wait.h>
  10. #include <sys/types.h>
  11. #include <sys/socket.h>
  12. #include <netinet/in.h>
  13. #include <arpa/inet.h>
  14. #include "Log.hpp"
  15. #include "Common.hpp"
  16. #include "InetAddr.hpp"
  17. #include "ThreadPool.hpp"
  18. using namespace LogModule;
  19. using namespace ThreadPoolModule;
  20. using task_t = std::function<void()>;
  21. using handler_t = std::function<std::string(std::string)>;
  22. #define BACKLOG 8
  23. uint16_t gport = 8080;
  24. class TcpServer
  25. {
  26. public:
  27.     TcpServer(handler_t handler, int port = gport)
  28.         : _handler(handler), _port(port), _isrunning(false)
  29.     {}
  30.     ~TcpServer() {}
  31.     void InitServer()
  32.     {
  33.         // 1. 创建TCP套接字
  34.         _listensockfd = ::socket(AF_INET, SOCK_STREAM, 0);
  35.         if (_listensockfd < 0)
  36.         {
  37.             LOG(LogLevel::FATAL) << "socket error";
  38.             Die(SOCKET_ERR);
  39.         }
  40.         LOG(LogLevel::INFO) << "socket create success, listensockfd is: " << _listensockfd;
  41.         // 2. 绑定
  42.         struct sockaddr_in local;
  43.         memset(&local, 0, sizeof(local));
  44.         local.sin_family = AF_INET;
  45.         local.sin_port = ::htons(_port);
  46.         local.sin_addr.s_addr = INADDR_ANY;
  47.         int n = ::bind(_listensockfd, CONV(&local), sizeof(local));
  48.         if (n < 0)
  49.         {
  50.             LOG(LogLevel::FATAL) << "bind error";
  51.             Die(BIND_ERR);
  52.         }
  53.         LOG(LogLevel::INFO) << "bind success";
  54.         // 3. Tcp是面向连接的, 就要求Tcp随时随地等待被客户端连接, Tcp需要将socket设置为监听状态
  55.         //    客户端请求连接时: 将客户端连接请求放入一个监听队列中
  56.         n = ::listen(_listensockfd, BACKLOG);
  57.         if (n < 0)
  58.         {
  59.             LOG(LogLevel::FATAL) << "listen error";
  60.             Die(LISTEN_ERR);
  61.         }
  62.         LOG(LogLevel::INFO) << "listen success";
  63.         // ::signal(SIGCHLD, SIG_IGN); // 子进程退出, 父进程无需wait, 操作系统自动回收资源
  64.     }
  65.     // Tcp也是全双工的: 在同一个文件描述符中, 既可以读又可以写
  66.     void HandlerRequest(int sockfd)
  67.     {
  68.         LOG(LogLevel::INFO) << "开始处理客户端请求...";
  69.         char inbuffer[4096];
  70.         while (true)
  71.         {
  72.             // 约定: 客户端发过来的是一条完整的命令string
  73.             // 1. 读取客户端发送来的消息
  74.             // ssize_t n = ::read(sockfd, inbuffer, sizeof(inbuffer) - 1); // 读取是不完善的
  75.             ssize_t n = ::recv(sockfd, inbuffer, sizeof(inbuffer) - 1, 0);
  76.             if (n > 0)
  77.             {
  78.                 LOG(LogLevel::INFO) << inbuffer;
  79.                 inbuffer[n] = 0;
  80.                 // std::string message = "server echo# ";
  81.                 // message += inbuffer;
  82.                 std::string cmd_reault = _handler(inbuffer);
  83.                 // 2. 向客户端发送消息
  84.                 // ::write(sockfd, message.c_str(), message.size()); // 写入也是不完善的
  85.                 // ::send(sockfd, message.c_str(), message.size(), 0);
  86.                 ::send(sockfd, cmd_reault.c_str(), cmd_reault.size(), 0);
  87.             }
  88.             else if (n == 0)
  89.             {
  90.                 // read如果读取的返回值是0, 表示客户端退出
  91.                 LOG(LogLevel::INFO) << "客户端退出: " << sockfd;
  92.                 break;
  93.             }
  94.             else
  95.             {
  96.                 // 读取失败
  97.                 break;
  98.             }
  99.         }
  100.         ::close(sockfd); // 关闭fd, 防止fd泄漏问题
  101.     }
  102.     void Start()
  103.     {
  104.         _isrunning = true;
  105.         while (_isrunning)
  106.         {
  107.             // 1. Tcp不能直接获取数据: 需要获取新连接
  108.             // 阻塞等待, 直到有客户端连接请求进入监听队列, 然后从队列中取出一个请求, 为该客户端建立连接,
  109.             // 并返回一个新的套接字描述符, 通过这个新的套接字描述符就可以与客户端进行数据的发送和接收
  110.             struct sockaddr_in peer;
  111.             socklen_t peerlen = sizeof(peer);
  112.             int sockfd = ::accept(_listensockfd, CONV(&peer), &peerlen);
  113.             if (sockfd < 0)
  114.             {
  115.                 LOG(LogLevel::WARNING) << "accept error: " << strerror(errno);
  116.                 continue;
  117.             }
  118.             LOG(LogLevel::INFO) << "accept success, sockfd is: " << sockfd;
  119.             // 获取客户端的信息: IP + 端口号
  120.             InetAddr addr(peer);
  121.             LOG(LogLevel::INFO) << "client info: " << addr.Addr();
  122.             // version1->单进程版本: 单客户端访问
  123.             // HandlerRequest(sockfd);
  124.             // version2->多进程版本: 多客户端访问
  125.             // pid_t id = fork();
  126.             // if (id == 0)
  127.             // {
  128.             //     // 子进程: 继承父进程的文件描述符表, 有两张表
  129.             //     ::close(_listensockfd); // 关闭不需要的文件描述符: 监听套接字
  130.             //     if(fork() > 0) exit(0); // 子进程退出
  131.             //     // 孙子进程->孤儿进程: 不断与客户端的数据传输, 退出后被操作系统自动回收
  132.             //     HandlerRequest(sockfd);
  133.             //     exit(0);
  134.             // }
  135.             // // 父进程: 不断与客户端建立连接
  136.             // ::close(sockfd); // 关闭不需要的文件描述符: socket
  137.             // // waitpid不会阻塞
  138.             // int rid = ::waitpid(id, nullptr, 0);
  139.             // if(rid < 0)
  140.             // {
  141.             //     LOG(LogLevel::WARNING) << "waitpid error";
  142.             // }
  143.             // version3->多线程版本: 多客户端访问
  144.             // 主线程和新线程共享同一张文件描述符表
  145.             // pthread_t tid;
  146.             // ThreadData *data = new ThreadData();
  147.             // data->sockfd = sockfd;
  148.             // data->self = this;
  149.             // pthread_create(&tid, nullptr, ThreadEntry, (void *)data);
  150.             // version4->线程池版本: 多客户端访问(适合处理短任务)
  151.             task_t task = std::bind(&TcpServer::HandlerRequest, this, sockfd); // 构建任务
  152.             ThreadPool<task_t>::GetInstance()->Equeue(task);
  153.             // ThreadPool<task_t>::GetInstance()->Equeue([this, &sockfd](){
  154.             //     this->HandlerRequest(sockfd);
  155.             // });
  156.         }
  157.     }
  158.     struct ThreadData
  159.     {
  160.         int sockfd;
  161.         TcpServer *self;
  162.     };
  163.     // 类中ThreadEntry函数带有this指针, 需要加上static
  164.     // 而没有this指针, 又无法调用HandlerReques函数
  165.     // 解决方法: 封装ThreadData结构体
  166.     static void *ThreadEntry(void *argc)
  167.     {
  168.         pthread_detach(pthread_self()); // 线程分离: 线程退出时由操作系统自动回收, 防止类似僵尸进程的问题
  169.         ThreadData *data = (ThreadData *)argc;
  170.         data->self->HandlerRequest(data->sockfd);
  171.         return nullptr;
  172.     }
  173.     void Stop()
  174.     {
  175.         _isrunning = false;
  176.     }
  177. private:
  178.     int _listensockfd; // 监听套接字
  179.     uint16_t _port;
  180.     bool _isrunning;
  181.     handler_t _handler; // 处理客户端发来的任务
  182. };
复制代码
10.TcpServer.cc

  1. #include <memory>
  2. #include "TcpServer.hpp"
  3. #include "CommandExecute.hpp"
  4. int main()
  5. {
  6.     ENABLE_CONSOLE_LOG();
  7.     Command cmd;
  8.     std::shared_ptr<TcpServer> tsvr = std::make_shared<TcpServer>([&cmd](std::string cmdstr){
  9.         return cmd.Execute(cmdstr);
  10.     });
  11.    
  12.     tsvr->InitServer();
  13.     tsvr->Start();
  14.     return 0;
  15. }
复制代码
11.TcpClient.cc

  1. #include <iostream>
  2. #include <string>
  3. #include <cstring>
  4. #include <unistd.h>
  5. #include <sys/types.h>
  6. #include <sys/socket.h>
  7. #include <netinet/in.h>
  8. #include <arpa/inet.h>
  9. #include "Common.hpp"
  10. // ./client_tcp server_ip server_port
  11. int main(int argc, char *argv[])
  12. {
  13.     if (argc != 3)
  14.     {
  15.         std::cout << "Usage: ./client_tcp server_ip server_port" << std::endl;
  16.         return 1;
  17.     }
  18.     std::string server_ip = argv[1];
  19.     int server_port = std::stoi(argv[2]);
  20.     // 1. 创建套接字
  21.     int sockfd = ::socket(AF_INET, SOCK_STREAM, 0);
  22.     if (sockfd < 0)
  23.     {
  24.         std::cout << "socket error" << std::endl;
  25.         return 2;
  26.     }
  27.     // 2. 客户端不需要显示的进行绑定, 但是需要连接服务器, 在建立连接的过程由操作系统进行绑定IP和端口号
  28.     struct sockaddr_in server;
  29.     memset(&server, 0, sizeof(server));
  30.     server.sin_family = AF_INET;
  31.     server.sin_port = ::htons(server_port);
  32.     server.sin_addr.s_addr = ::inet_addr(server_ip.c_str());
  33.     int n = ::connect(sockfd, CONV(&server), sizeof(server));
  34.     if (n < 0)
  35.     {
  36.         std::cout << "connect error" << std::endl;
  37.         return 3;
  38.     }
  39.     std::string message;
  40.     while (true)
  41.     {
  42.         char inbuffer[1024];
  43.         std::cout << "input message: ";
  44.         std::getline(std::cin, message);
  45.         // 3. 向服务器发送消息
  46.         n = ::write(sockfd, message.c_str(), message.size());
  47.         if (n > 0)
  48.         {
  49.             // 4. 读取服务器发送来的消息
  50.             int m = ::read(sockfd, inbuffer, sizeof(inbuffer) - 1);
  51.             if (m > 0)
  52.             {
  53.                 inbuffer[m] = 0;
  54.                 std::cout << inbuffer << std::endl;
  55.             }
  56.             else break;
  57.         }
  58.         else break;
  59.     }
  60.     ::close(sockfd);
  61.     return 0;
  62. }
复制代码
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

小秦哥

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表