网络套接字-TCP服务器

一 前言

        前面已经写过udp服务器的实现了,那里说了很多编写服务器的所需知识,在tcp服务器实现中就不再赘述了。

二 服务端编写

        大致接口如下。

./server + port端口号 启动时指明端口号
void usage(const std::string proc)
{
    std::cout<<"Usage "<<proc<<" port"<<std::endl;
}

int main(int argc,char*argv[]) 解析命令行参数,第二个就是端口号
{
    if(argc != 2)
    {
        usage(argv[1]);
        std::cout<<"port error"<<strerror(errno)<<std::endl;
        exit(USAGE_ERR);
    }
    u_int16_t port = atoi(argv[1]);将字符串的端口号转为数字,后续传给服务端类。

    tcp->InitServer();
    tcp->start();
    return 0;
}

        然后我们就去实现类了,首先服务端必定要有端口号,而且一个公司部署的服务端的端口号是分配好的。还有个成员是保存套接字的,这个后面具体实现再提。

namespace server
{


    class TcpServer
    {
    public:
        TcpServer()
        {
            ;
        }
        ~TcpServer()
        {
            ;
        }
        void InitServer()
        {
 
        }
        void start()
        {
      
        }
 
    private:
        int listensocket_;
        int port_;
    };
};

1 创建套接字

        初始化,显然就是要打开网络文件,创建套接字,直接复制udp的实现。

             // 1 创建套接字
            listensocket_ = socket(AF_INET, SOCK_STREAM, 0);
            if (listensocket_ < 0)
            {
                std::cout << "socket err" << std::endl;
                exit(SOCKET_ERR);
            }
            std::cout << "socket success" << std::endl;

2  开始绑定

        注意:recv和sendto函数内部不会实现大小端转化,需要我们自己主机转网络序列。


            // 2 绑定端口号和ip地址
            struct sockaddr_in sock; // 头文件<netinet/in.h>
            bzero(&sock, sizeof(sock));
            sock.sin_addr.s_addr = INADDR_ANY; // 设置ip地址 表示所有的ip的地址
            sock.sin_port = htons(port_);
            sock.sin_family = AF_INET;
            if (bind(listensocket_, (sockaddr *)(&sock), sizeof(sock)) < 0)
            {
                std::cout << "bind error " << strerror(errno) << std::endl;
                exit(BIND_ERR);
            }
            std::cout << "bind success" << std::endl;

3 监听

        参数2后面提及协议再说。可以设个不大不小的整数。

        listensocket_是我们先前创建套接字返回的文件描述符,这个描述符被用来监听了,监听的是客户端的链接请求,因为请求也是通信,所以我们要先创建套接字,如果有链接请求从网络中来,os会把链接请求转成数据保存在监听套接字对应的文件中,上层就从这个文件中读取链接。

             // 3 开始监听
            if (listen(listensocket_, backlog)) // 返回0,监听成功
            {
                std::cout << "listen error " << strerror(errno) << std::endl;
                exit(LISTEN_ERR);
            }

4 启动服务器

        当有请求来了,我们当然要调用函数去处理,参数很熟悉,sockfd是我们先前用的listensockt_,因为请求都在这个文件内。

        void start()
        {
            while (true)
            {
                // 获取链接
                struct sockaddr_in sock; // 头文件<netinet/in.h>
    
                socklen_t len = sizeof(sock);
                int socket = accept(listensocket_, (sockaddr *)&sock, &len); // 不保存吗
                if (socket < 0)
                {
                    // std::cout << "accept err" << std::endl;
                    exit(SOCKET_ERR);
                }
                std::cout << "accept success" << std::endl;
                // 开始发消息,此时我们不能让主线程去发消息,不然就无法链接其它的客户端了
                std::string clientip = inet_ntoa(sock.sin_addr);
                u_int16_t clientport = ntohs(sock.sin_port);
                server(sock,clientip, clientport);

            }
        }
        void server(int sock, std::string ip, uint16_t port)
        {
            ;
        }

        此时收到请求后,我们也就获得了客户端的ip和端口,接下来就用ip和端口传给server函数实现通信。可是为什么accept又返回一个文件描述符,为什么要有两个套接字呢?在udp服务器中我们只创建了一个文件,意味着所有客户端发来的消息都在这个文件中,读取的时候其实很容易读取到a客户端的数据处理完发给了b客户端,那对于a客户端来说,数据就缺失了,所以udp的通信确实是不考虑完不完整的,所以tcp为了保证客户端数据的独立性,就每接收一次链接就创建一个文件,内部一定会把客户端的ip和端口号和这个文件绑定,这样下次客户端的数据来的时候就会根据ip和端口号判断放在哪个文件中了。

        接下来看看server函数内如何通信。直接调用read函数读,调用write函数写,奇怪,为什么先前udp服务器不可以用read,而是用recvfrom,也不是用write,而是用sendto,据我了解,read和write是面向字节流的,udp服务端我们打开文件的时候是指明面向数据报的,read用面向字节流读取面向数据报的文件会出问题。后面讲tcp,udp协议,了解了面向字节流和面向数据报就理解得更深刻了。

        void server(int sock, std::string ip, uint16_t port)
        {

            char buffer[1024] = {0};
            int n = read(sock, buffer, sizeof(buffer) - 1);
            if (n > 0)
            {
                buffer[n] = 0;
                // 写回给客户端
                write(sock, ret.c_str(), ret.size());
            }
            else if (n == 0)
            {
                close(sock);
                std::cout << name << " 断开连接" << std::endl;
            }
            else
            {
                close(sock);
                exit(READ_ERR);
            }

        }

   可是直接输出又有点单调,所以我们做了些修改。调用回调方法对数据做处理。

    

         func_就是func_t定义的变量,可以接收一个可调用对象,这个可调用对象的类型是string(string),也就是返回值是string,参数也是string。 

   

       数据处理方法由外部定义。如下就是完整的定义服务端对象和启动。

std::string echo(std::string message)
{
    return message;
}
int main(int argc,char*argv[])
{
    if(argc != 2)
    {
        usage(argv[1]);
        std::cout<<"port error"<<strerror(errno)<<std::endl;
        exit(USAGE_ERR);
    }
    u_int16_t port = atoi(argv[1]);
    std::shared_ptr<TcpServer> tcp = std::make_shared<TcpServer>(echo,port);
    tcp->InitServer();
    tcp->start();
    return 0;
}

相应构造函数也要修改。

三 客户端编写

        同样客户端不用自己bind。那要不要listen,accept呢? 我认为是不需要的,这两个函数的意义在于等待别人来链接,一般来说都是客户端主动连接服务端,很少有服务端来主动找客户端的,并且获取链接方便后面给对方发数据,前面还说了可以保证服务端收到来自客户端数据的独立和安全。

        而客户端本身不用担心多个服务端发来的数据混杂,虽然我们手机上有多个客户端,但是我们的客户端都是一个个进程,创建的套接字文件是独立的,服务端是因为会有多个客户端和它通信才要创建多个文件,我们这里是一个客户端对应一个服务端。所以通信步骤如下。

1 创建套接字

2 发起链接

        用的是下面这个connect函数,参数列表也是非常熟悉。

        直接开始准备初始化sock结构体。

        因为我们不能一链接失败就退出,要尝试重连几次,就像我们打游戏没网,也是会有尝试重连选项的。

int main(int argc, char *argv[])  ip地址不能是任意的,必须知道服务端的ip地址和端口号
{
    if (argc != 3)
    {
        usage(argv[0]);
        std::cout << "stage error" << strerror(errno) << std::endl;
        exit(USAGE_ERR);
    }
    const std::string ip_ = argv[1];
    u_int16_t port = atoi(argv[2]);

    int socket_ = socket(AF_INET, SOCK_STREAM, 0);
    if (socket_ < 0)
    {
        std::cout << "create socket error" << strerror(errno) << std::endl;
        exit(SOCKET_ERR);
    }
    std::cout << "create socket successs" << std::endl;
    // 打开网络文件

    // 不用客户端自己绑定端口号和ip地址
    struct sockaddr_in sock;
    sock.sin_addr.s_addr = inet_addr(ip_.c_str()); // 将字符串类型的地址转为四字节地址,而且是网络字节序了
    sock.sin_port = htons(port);
    sock.sin_family = AF_INET;
    int len = sizeof(sock);
    // 开始连接
    int timenum = 5;
    while (connect(socket_, (sockaddr *)&sock, len) < 0)
    {
        sleep(1);
        std::cout << "开始重连:" << std::endl;
        timenum--;
        if (timenum == 0)
            break;
    }
    if (timenum <= 0)
    {
        std::cout << "链接失败" << std::endl;
        exit(CON_ERR);
    }
    return 0;
}

        接下来就是我们客户端直接把消息写到文件中,然后read读取服务端返回的数据。

       

四 实验测试 

        测试1

      服务端bug:如果客户端不给我们发消息,我们就会阻塞在read这里,就不能调用accept接收新连接了,所以我们应该安排一个线程去调用server函数来收发消息

          注意:我服务端代码里的server函数只能读一次和发一次消息给客户端,如果写成死循环后面引进线程池还会出问题,因为线程数量有限,多个客户端来链接,线程就会不够,这里不会,因为此时我们是一个线程负责收发消息给一个客户端。

    
        
void *recv(void *arg)
{
    // 收消息
    int socket_ = *((int *)arg);
    while (true)
    {
        char buffer[1024] = {0};
        int n = read(socket_, buffer, sizeof(buffer) - 1);
        if (n > 0)
        {
            buffer[n] = '\0';
            std::cout << "server echo# " << buffer << std::endl;
        }
        else if (n == 0)
        {
            close(socket_);
            std::cout << "server quit" << std::endl;
        }
        else
        {
            close(socket_);
            std::cout << "client read err" << strerror(errno) << std::endl;
            exit(READ_ERR);
        }
    }
}

         void start()
        {
            while (true)
            {
                // 获取链接
                struct sockaddr_in sock; // 头文件<netinet/in.h>
                bzero(&sock, sizeof(sock));
                socklen_t len = sizeof(sock);
                int socket = accept(listensocket_, (sockaddr *)&sock, &len); // 不保存吗
                if (socket < 0)
                {
                    lg_(ErrorLevel::Info,"accept err");
                    // std::cout << "accept err" << std::endl;
                    exit(SOCKET_ERR);
                }
                std::cout << "accept success" << std::endl;
                // 开始发消息,此时我们不能让主线程去发消息,不然就无法链接其它的客户端了
                std::string clientip = inet_ntoa(sock.sin_addr);
                u_int16_t clientport = ntohs(sock.sin_port);

                pthread_t id;
                pthread_create(&id,nullptr,recv,ThreadDta);
                                            
            }
        } 
    

        ThreadData类内包含ip和端口,server函数实现也被放入了静态成员函数recv中,我这里只是演示一下,只会把最后版本放出来下面我们引入线程池来做优化,我们希望在accept链接前就创建好线程了,提高响应速度。

引进线程池

        线程池:内部负责创建线程,我们外部构建任务,放入线程池中,内部线程池会去执行。

 先来看看任务构建,外部传入一个套接字,ip,端口号,可调用对象构建任务,线程拿到任务后调用可调用对象。


#define NUM 5
class Task
{
public:
    using func_t = std::function<void(int,std::string,u_int16_t)>;
    Task()
    {
        ;
    }
    Task(int sock,std::string ip,uint16_t port,func_t func)
        : socket_(sock), ip_(ip), port_(port),func_(func)
    {
        ;
    }
    void operator()()
    {
        func_(socket_,ip_,port_);
    }
    int socket_;
    func_t func_;
    std::string ip_;
    u_int16_t port_;
};

        使用如下,构建任务,并且入队列,这里面还用了bind语法。

        然后我大致说一下线程池内部实现,以及我们什么时候控制线程池去执行任务。成员如下,有任务队列保存任务,还有vector<Thread>保存多个线程。

        我们线程池内存的也不是线程id,而是封装后的Thread类。代码如下,可以不关心内部实现。

class Thread
{
public:
    typedef enum
    {
        NEW = 1,
        RUNING,
        EXIT
    }status;
    typedef void* (*fun_t)(void*);
    Thread()
    {
        ;
    }
    Thread(int num, fun_t fun, void* arg)
    :id_(0),fun_(fun),arg_(arg),status_(NEW)
    {
        name_ = "thread->" + std::to_string(num);
    }
    ~Thread()
    {
        ;
    }
    static void * threadRun(void*arg)
    {
        Thread* th = (Thread*) arg;
        th->fun_(th->arg_);
        return nullptr;
    }
    void Run()
    {
        int n = pthread_create(&id_,nullptr,threadRun,(void*)this);
        if(n != 0)//成功返回0,不成功返回错误码
            exit(4);
        status_ = RUNING;    
    }
    void join()
    {
        pthread_join(id_,nullptr);
        status_ = EXIT;
    }
    std::string getname()
    {
        return name_;
    }
    int getstatus()
    {
        return status_;
    }
    pthread_t  getid()
    {
        if(status_ == RUNING)
            return id_;
        else
        {
            std::cout<<name_<<" not create ";
            return 1;
        }    
    }
    pthread_t id_;
    std::string name_;//线程名
    status status_;//线程状态
    fun_t fun_;//线程执行函数
    void* arg_;//线程参数
};

        我们只需要知道我们要线程池初始化thread对象时要传一个执行函数和参数即可,i就是内部用来构建线程名的,不用关心。

        外部通过线程池的静态函数获取单例对象。

        线程池内调用Thread类内方法创建线程。

        Thread类内的Run方法。

        线程池内部实现。

template <class T>
class threadPool
{
    threadPool(int size = NUM) // vp_存的自定义类型要有默认构造,不然这里初始化会找不到默认构造!
        : vp_(size)
    {
        pthread_mutex_init(&mutex_, nullptr);
        pthread_cond_init(&Consumer, nullptr);
        pthread_cond_init(&Productor, nullptr);
    }
      ~threadPool()
    {
        for (auto &e : vp_) // 复用Thread join方法回收线程
        {
            e.join();
        }
        pthread_mutex_destroy(&mutex_);
        pthread_cond_destroy(&Consumer);
        pthread_cond_destroy(&Productor);
    }
   threadPool(const threadPool<T>& sh) = delete;
	threadPool<T> operator=(const threadPool<T>& sh) = delete;
public:
    void init()
    {
        for (int i = 0; i < NUM; i++)
        {
            vp_[i] = (Thread(i, threadRun, this));//this指针是给内部传参数的
        }
    }
    void Lock()
    {
        pthread_mutex_lock(&mutex_);
    }
    void Unlock()
    {
        pthread_mutex_unlock(&mutex_);
    }
    static void *threadRun(void *arg)
    {
        // pthread_detach(pthread_self());
        threadPool<T> *tp = static_cast<threadPool<T> *>(arg);
        // 执行任务
        while (true)
        {
            T data;
            tp->pop(data);
            data();
        }
    }
    bool Full()
    {
        return tasks_.size() == NUM;
    }
    bool Empty()
    {
        return tasks_.size() == 0;
    }
    void start()
    {
        for (auto &e : vp_) // 复用Thread Run方法创建线程
        {
            e.Run();
        }
    }
  
    void push(const T &data)
    {
        {
            LockGuard lg(&mutex_);
            while (Full())
            {
                pthread_cond_wait(&Productor, &mutex_);
            }
            tasks_.push(data);
        }
        pthread_cond_signal(&Consumer);

    }
    void pop(T &data)
    {

        {
            LockGuard lg(&mutex_);
            // 检查是否有任务
            while (Empty())
            {
                pthread_cond_wait(&Consumer, &mutex_);
            }
            data = tasks_.front();
            tasks_.pop();
        }
        pthread_cond_signal(&Productor);
    }
    static threadPool<T>* getthreadPool()
    {
        if(tp_ == nullptr)//减少加锁次数
        {
            LockGuard lg(&Poolmutex_);
            if (tp_ == nullptr)
            {    
                tp_ = new threadPool<T>();
                tp_->init();
                tp_->start();
            }

        }
        return tp_;    
    }
    std::queue<T> tasks_;    // 任务队列
    std::vector<Thread> vp_; // 线程池,不能存指针,内部要解引用访问的
    pthread_mutex_t mutex_;
    pthread_cond_t Consumer;
    pthread_cond_t Productor;
    static threadPool<T>* tp_;
    static pthread_mutex_t Poolmutex_;
};
template<class T>
threadPool<T>*threadPool<T>::tp_ = nullptr;
template<class T>
pthread_mutex_t threadPool<T>::Poolmutex_ = PTHREAD_MUTEX_INITIALIZER;

        综上,外部线程获取单例线程池对象并且push任务到队列,内部线程pop任务出来并执行。由于我们的线程池并没有设计扩容功能,所以线程是有限的,而线程池内部线程执行的任务是我们外部传入的server函数,如果server函数是个死循环,就会导致无法执行其它客户端的任务,所以我们把server函数内的死循环改了。

五 守护进程

1 概念介绍

        为什么要有守护进程呢?

        因为我们目前起的进程如果把中端窗口一关闭,此时就会导致服务停了,那总不能让屏幕一直开着吧,所以我们让这个进程在后台运行,也就是守护进程化。

我们以前在讲进程状态的时候,用过下面这条命令,但是有几个成员一直没提及,那就是PGID和SID以及TTY。

        PGID表是进程组,我们在命令行可以一次性起多个进程,这些进程都被归属于某个进程组被管理起来,进程组id一般是第一个创建的进程id,例如我们./server起了一个进程,这个进程内部也起了多个进程,此时这些进程都会被划分在一个进程组中,此时组长是server。

        tty表示终端文件,终端就是我们打开xshell显示的那个窗口,所以当我们在命令行输入指令,都是云服务将结果输出到文件,文件经过网络发送到我们的主机上。

在一个终端下起的进程的终端文件是一样的,父进程是一样的,都是bash,而且还都属于某个进程组,进程组以第一个创建的进程id命名,不是bash。例如sleep 100 | sleep 100 &我们就在后台一次性起了两个进程,它们属于一个进程组。

        所以./server不再是启动一个进程了,我们应该说是启动一个任务,因为serve程序内部可能也会创建进程,所以需要一个更大的概念-任务来描述,jobs可以查看当前会话的任务。

        左侧的是任务码,我们可以把一个后台拉到前台运行,fg+ 任务码,然后ctrl+c就可以结束这个进程了。

        将前台的任务回退为后台,Ctrl+z。

        细节,当我们把sleep拉回前台运行时,此时bash命令就没了,因为我们./server就是把一个任务变成了前台,bash就变成后台,一个终端只能有一个前台进程,可以简单理解前台进程就是要和我们的键盘交互的,我们只能给一个进程喂指令,所以一个终端只有一个前台进程,当我们ctrl+z把前台进程变成后台进程后,此时bash命令行又回来了,因为它自动把自己变成前台进程了。

        那什么是SID呢,我们称为会话id,22103其实是我们的bash进程,会话id以一个会话内的首进程id命名,什么是会话呢?当我们登录xshell时,linux就会建立一个会话,将bash以及bash创建进程管理在其中,有时候bash创建的进程(例如./server)内部又创建了许多进程,为了将server和它创建的进程关联起来,就有了进程组。       

        注销的理解:以前我们的电脑上是有个注销选项的,注销和重启是不同的,注销是删掉会话内的所有进程,重启则是整个系统进程都要重启了。守护进程化,就是对server进程独立开一个会话,这样用户注销就不会影响我这个守护进程了。使用如下接口。

返回值:会话ID。

2 守护进程化实现

void Daemon()
{

    pid_t ret = setsid();   
    if((int)ret == -1)
    {
        cout<<"setsid err:错误码: "<<errno<<" 错误信息:"<<strerror(errno)<<endl;
        exit(SETSID_ERR);
    }    
}

        然后我们在server.cc中调用一下,我们整个进程就变成守护进程啦。对进程组的其它进程无影响?如果我们是像这里一样实现的话,大概率会出错,首先我们这里只有一个server进程,

       

        server进程就必定是组长,组长进程不能从会话中独立出去,为什么不能走呢,我想是因为组长进程内有着管理组内进程的方法,如果走了就管理不了其它的进程了,深究的话就得讨论为什么组长走了就不能管理其它进程了,难道我们不能再选一个组长吗? 当然我说的容易,实现起来可能比较冗余,所以就禁止我们对组长进程进行守护进程化。

        优化如下。

        我们在Daemon()函数中创建子进程,让父进程退出,此时子进程就变成孤儿进程了,被os接管了,此时这个孤儿进程不会是组长,此时孤儿进程再将自己独立出去。

        变成守护进程后,一直在后台执行start函数,这个函数内是在接收链接,并且创建任务到队列中,让内部的线程池去执行通信任务。

然后我们还要忽略一些常见错误信号,让我们的守护进程不会随随便便就退出。

 signal(SIGPIPE,SIG_IGN);
    signal(SIGCHLD,SIG_IGN);

        已经成为守护进程了,就不应该和键盘显示器关联了,我们可以直接关闭显示器和键盘文件,但是这样我们使用cout,cin会直接出错。所以我们打开一个特殊文件,让cin读不到数据,直接返回,写入的数据也会被丢弃。

//守护进程化
void Daemon()
{

    子进程去创建新会话
    if(fork() > 0)
        exit(0);
    pid_t ret = setsid();   
    if((int)ret == -1)
    {
        cout<<"setsid err:错误码: "<<errno<<" 错误信息:"<<strerror(errno)<<endl;
        exit(SETSID_ERR);
    }    
    int fd = open("/dev/null",O_RDWR);
    if(fd < 0)
    {
        cout<<"open err:错误码: "<<errno<<" 错误信息:"<<strerror(errno)<<endl;
        exit(OPEN_ERR);
    }
    //关闭输入输出
    dup2(fd,0);
    dup2(fd,1);
    dup2(fd,2);
}

        守护进程只能kill掉。

相关推荐

最近更新

  1. TCP协议是安全的吗?

    2024-03-25 09:46:02       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-03-25 09:46:02       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-03-25 09:46:02       18 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-03-25 09:46:02       20 阅读

热门阅读

  1. Python Flask 使用数据库

    2024-03-25 09:46:02       18 阅读
  2. AtCoder Beginner Contest 346

    2024-03-25 09:46:02       19 阅读
  3. 【QT问题】 Qt信号函数如果重名,调用怎么处理

    2024-03-25 09:46:02       16 阅读
  4. 想注册滴滴司机驾龄不够怎么办?

    2024-03-25 09:46:02       18 阅读
  5. 大学 Python 程序设计实验报告:替换敏感词

    2024-03-25 09:46:02       21 阅读
  6. 24计算机考研调剂 | 西安石油大学

    2024-03-25 09:46:02       20 阅读
  7. 单例模式---饿汉模式和懒汉模式

    2024-03-25 09:46:02       18 阅读