main.cpp

#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
#include <sys/epoll.h>
#include <signal.h>
#include "locker.h"
#include "threadpool.h"
#include "http_conn.h"
#include "util.h"

#define MAX_FD 65535        // 最大的文件描述符个数
#define MAX_EVENT_NUM 10000 // 监听的最大的事件数量

int main(int argc, char *argv[])
{
  if (argc <= 1)
  {
    printf("按照如下格式允许：%s port_number\n", basename(argv[0]));
    exit(-1);
  }

  // 获取端口号
  int port = atoi(argv[1]);

  // 对sigpipe信号进行处理
  addsig(SIGPIPE, SIG_IGN);

  // 初始化线程池
  threadpool<http_conn> *pool = NULL;
  try
  {
    pool = new threadpool<http_conn>;
  }
  catch (...)
  {
    exit(-1);
  }

  // 创建一个数组用于保存所有的客户端信息
  http_conn *users = new http_conn[MAX_FD];

  // 创建监听的套接字
  int listenfd = socket(PF_INET, SOCK_STREAM, 0);

  // 设置端口复用
  int reuse = 1;
  setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse));

  // 绑定
  struct sockaddr_in address;
  address.sin_family = AF_INET;
  address.sin_addr.s_addr = INADDR_ANY;
  address.sin_port = htons(port);
  bind(listenfd, (struct sockaddr *)&address, sizeof(address));

  // 监听
  listen(listenfd, 16);

  // 创建epoll对象，事件数组，添加
  epoll_event events[MAX_EVENT_NUM];
  int epollfd = epoll_create(1);

  // 将监听的文件描述符到epoll对象中
  addfd(epollfd, listenfd, false);
  http_conn::m_epollfd = epollfd;

  while (true)
  {
    int num = epoll_wait(epollfd, events, MAX_EVENT_NUM, -1);
    if ((num < 0) && (errno != EINTR))
    {
      printf("epoll failure: %s (errno=%d)\n", strerror(errno), errno);
      break;
    }

    // 循环遍历事件数组
    for (int i = 0; i < num; i++)
    {
      int sockfd = events[i].data.fd;
      if (sockfd == listenfd)
      {
        // 有客户端连接
        struct sockaddr_in client_address;
        socklen_t client_addrlen = sizeof(client_address);
        int connfd = accept(listenfd, (struct sockaddr *)&client_address, &client_addrlen);

        if (http_conn::m_user_count >= MAX_FD)
        {
          // 目前连接满了
          // 告诉客户端服务器正忙
          close(connfd);
          continue;
        }
        // 将新的客户数据初始化加入数组
        users[connfd].init(connfd, client_address);
      }
      else if (events[i].events & (EPOLLRDHUP | EPOLLHUP | EPOLLERR))
      {
        // 对方异常断开或者错误事件
        users[sockfd].close_conn();
      }
      else if (events[i].events & EPOLLIN)
      {
        if (users[sockfd].read())
        {
          // 一次性读完所有数据
          pool->append(users + sockfd);
        }
        else
        {
          users[sockfd].close_conn();
        }
      }
      else if (events[i].events & EPOLLOUT)
      {
        if (!users[sockfd].write()) // 一次性写完所有数据
        {
          users[sockfd].close_conn();
        }
      }
    }
  }

  close(epollfd);
  close(listenfd);
  delete[] users;
  delete pool;

  return 0;
}