一个简短的epoll服务器示例, 监听5000个端口, 使用线程池

ehuangmy

[转帖]

运行这个程序需要预先设置栈内存和文件描述符上限, 否则运行失败
ulimit -n 16384
ulimit -s 4096
文件名:server.c
编译: gcc server.c -Wall -O2 -pthread -o server
程序源码如下(请自行编辑宏定义SERVER_IP为自己的IP):
/*Linux 2.6 x86_64 only*/
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define THREAD_MAX 4096
#define LISTEN_MAX 5000
#define SERVER_IP "192.168.1.103"
typedef struct {
    char    ip4[128];
    int    port;
    int    fd;
} LISTEN_INFO;
//服务器参数
static LISTEN_INFO s_listens[LISTEN_MAX];
//线程池参数
static unsigned int s_thread_para[THREAD_MAX][8];//线程参数
static pthread_t s_tid[THREAD_MAX];//线程ID
pthread_mutex_t s_mutex[THREAD_MAX];//线程锁
//私有函数
static int init_thread_pool(void);
static int init_listen4(char *ip4, int port, int max_link);
//线程函数
void * test_server4(unsigned int thread_para[]);
int main(int argc, char *argv[])//客户端驱动
{
    //临时变量
    int            i, j, rc;
   
    int            sock_listen;    //监听套接字
    int            sock_cli;    //客户端连接
    int            listen_index;
   
    int            epfd;
    int             nfds;
    struct epoll_event    ev;
    struct epoll_event    events[LISTEN_MAX];
    socklen_t        addrlen;    //地址信息长度
    struct sockaddr_in    addr4;        //IPv4地址结构
    //线程池初始化
    rc = init_thread_pool();
    if (0 != rc) exit(-1);
   
    //初始化服务监听
    for(i = 0; i  rc) {
            fprintf(stderr, "无法创建服务器监听于%s:%d\r\n", s_listens.ip4, s_listens.port);
            exit(-1);
        }
        s_listens.fd = rc;
    }
   
    //设置集合
    epfd = epoll_create(8192);
    for (i = 0; i  sock_cli) {
                fprintf(stderr, "接收客户端连接失败\n");
                continue;
            }
            //查询空闲线程对
            for(j = 0; j = THREAD_MAX) {
                fprintf(stderr, "线程池已满, 连接将被放弃\r\n");
                shutdown(sock_cli, SHUT_RDWR);
                close(sock_cli);
                continue;
            }
            //复制有关参数
            s_thread_para[j][0] = 1;//设置活动标志为"活动"
            s_thread_para[j][1] = sock_cli;//客户端连接
            s_thread_para[j][2] = listen_index;//服务索引
            //线程解锁
            pthread_mutex_unlock(s_mutex + j);
        }//end of for(i;;)
    }//end of for(;;)
    exit(0);
}
static int init_thread_pool(void)
{
    int    i, rc;
    //初始化线程池参数
    for(i = 0; i  sock_listen4) return(-1);
    //设置SO_REUSEADDR选项(服务器快速重起)
    optval = 0x1;
    setsockopt(sock_listen4, SOL_SOCKET, SO_REUSEADDR, &optval, 4);
   
    //设置SO_LINGER选项(防范CLOSE_WAIT挂住所有套接字)
    optval1.l_onoff = 1;
    optval1.l_linger = 60;
    setsockopt(sock_listen4, SOL_SOCKET, SO_LINGER, &optval1, sizeof(struct linger));
    if (0 > bind(sock_listen4, (struct sockaddr *)&addr4, sizeof(addr4))) {
        close(sock_listen4);
        return(-1);
    }
    if (0 > listen(sock_listen4, max_link)) {
        close(sock_listen4);
        return(-1);
    }
    return(sock_listen4);
}
void * test_server4(unsigned int thread_para[])
{
    //临时变量
    int        pool_index;    //线程池索引
    int        sock_cli;    //客户端连接
    int        listen_index;    //监听索引
    char        buff[32768];    //传输缓冲区
    char        *p;
    int        i, j, len;
   
    //线程脱离创建者
    pthread_detach(pthread_self());
    pool_index = thread_para[7];
wait_unlock:
    pthread_mutex_lock(s_mutex + pool_index);//等待线程解锁
   
    //线程变量内容复制
    sock_cli = thread_para[1];//客户端连接
    listen_index = thread_para[2];//监听索引
   
    //接收请求
    len = recv(sock_cli, buff, 32768, MSG_NOSIGNAL);
    //构造响应
    p = buff;
    //HTTP头
    p += sprintf(p, "HTTP/1.1 200 OK\r\n");
    p += sprintf(p, "Content-Type: text/html\r\n");
    p += sprintf(p, "Connection: closed\r\n\r\n");
    //页面
    p += sprintf(p, "\r\n\r\n");
    p += sprintf(p, "\r\n");
    p += sprintf(p, "\r\n");
    p += sprintf(p, "\r\n");
   
    p += sprintf(p, "\r\n");
    p += sprintf(p, "连接状态\r\n");
    p += sprintf(p, "服务器地址 %s:%d\r\n", s_listens[listen_index].ip4, s_listens[listen_index].port);
    j = 0;
    for(i = 0; i 线程池状态\r\n");
    p += sprintf(p, "线程池总数 %d 活动线程总数 %d\r\n", THREAD_MAX, j);
    p += sprintf(p, "\r\n");
    len = p - buff;
    //发送响应
    send(sock_cli, buff, len, MSG_NOSIGNAL);
    //释放连接
    shutdown(sock_cli, SHUT_RDWR);
    close(sock_cli);
   
    //线程任务结束
    thread_para[0] = 0;//设置线程占用标志为"空闲"
    goto wait_unlock;
    pthread_exit(NULL);
}


本文来自ChinaUnix博客，如果查看原文请点：http://blog.chinaunix.net/u/14063/showart_262447.html