基于EPOLL写的HTTP服务器(加入了线程池)_没落都城_新浪博客
基于EPOLL写的HTTP服务器(加入了线程池)
(2010-12-07 19:02:51)
转载▼
标签：
杂谈
分类： EPOLL
#include<fcntl.h>
#include<cstdio>
#include<unistd.h>
#include<cstdlib>
#include<sys/socket.h>
#include<sys/epoll.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<errno.h>
#include<cstring>
#include<pthread.h>
const int EPOLL_SIZE=5000;
const int EVENT_ARR=5000;
const int PORT=8002;
const int BUF_SIZE=5000;
const int BACK_QUEUE=100;
const int THREAD_MAX=100;
static unsigned int s_thread_para[THREAD_MAX][8]; //线程参数
static pthread_t s_tid[THREAD_MAX]; //线程ID
pthread_mutex_t s_mutex[THREAD_MAX]; //线程锁
int epFd; //epoll
struct epoll_event ev,evs[EVENT_ARR];
char *get_type(char *url,char *buf)
{
const char *t=url+strlen(url);
char type[64];
for(;t>=url&&*t!='.';--t) ;
strcpy(type,t+1);
if(strcmp(type,"html")==0||strcmp(type,"htm")==0)
sprintf(buf,"text/%s",type);
else if(strcmp(type,"gif")==0||
strcmp(type,"jpg")==0||
strcmp(type,"jpeg")==0||
strcmp(type,"png")==0)
sprintf(buf,"image/%s",type);
else if(strcmp(type,"/")==0)
sprintf(buf,"text/html");
else if(strcmp(type,"css")==0)
sprintf(buf,"text/css");
else if(strcmp(type,"js")==0)
sprintf(buf,"application/x-javascript");
else
{
sprintf( buf, "unknown" );
}
return buf;
}
void* http_server(int thread_para[])
{
int pool_index; //thread pool ID
int clientFd; //client socket
char buf[BUF_SIZE];
pthread_detach(pthread_self());
pool_index=thread_para[7];
wait_unlock:
pthread_mutex_lock(s_mutex+pool_index); //wait for thread unlock
clientFd=thread_para[1]; //client socket ID
//先进行试探性读取
int len=read(clientFd,buf,BUF_SIZE);
printf("%s",buf);
if(len>0)
{
char *token=strtok(buf," "); //GET
printf("token:%s",token);
char type[64];
char *url=strtok(NULL," "); //URL
while(*url=='.'||*url=='/')++url;
printf("url:%s",url);
char file[1280000];
sprintf(file,"%s",url);
printf("file:%s",file);
FILE *fp=fopen(file,"rb");
if(fp==0)
{
char response[]="HTTP/1.1 404 NOT FOUND\r\n\r\n";
printf("HTTP/1.1 404 NOT FOUND\r\n\r\n");
write(clientFd,response,strlen(response));
}
else
{
int file_size;
char *content;
char *response;
fseek(fp,0,SEEK_END);
file_size=ftell(fp);
fseek(fp,0,SEEK_SET);
content=(char *)malloc(file_size+1);
response=(char*)malloc(200);
fread(content,file_size,1,fp);
content[file_size]=0;
sprintf(response,"HTTP/1.1 200 OK\r\nContent-Length:%d\r\nContent-Type:%s\r\n\r\n",file_size,get_type(url,type));
// printf("HTTP/1.1 200 OK\r\nContent-Type:%s\r\nContent-Length:%d\r\n\r\n%s",get_type(url,type),file_size,content);
write(clientFd,response,strlen(response));
write(clientFd,content,file_size);
free(content);
free(response);
}
}
else if(len==0)
{
//触发了EPOLL事件，却没有读取，表示断线
//printf("Client closed at %s\n",inet_ntoa(clientAddr.sin_addr));
epoll_ctl(epFd,EPOLL_CTL_DEL,clientFd,&ev);
close(clientFd);
int i=thread_para[3];
evs[i].data.fd=-1;
}
else if(len==EAGAIN)
{
printf("socket huan cun man le!\n");
}
else
{
//client读取出错
printf("Client read failed!\n");
}
thread_para[0] = 0;//设置线程占用标志为"空闲"
goto wait_unlock;
printf("pthread exit!\n");
pthread_exit(NULL);
}
static int init_thread_pool(void)
{
int i,rc;
for(i=0;i<THREAD_MAX;i++)
{
s_thread_para[i][0]=0; //idle
s_thread_para[i][7]=i; //thread pool ID
pthread_mutex_lock(s_mutex+i); //thread lock
}
// create thread pool
for(i=0;i<THREAD_MAX;i++)
{ rc=pthread_create(s_tid+i,0,(void *(*)(void*))&http_server,(void *)(s_thread_para[i]));
if(0!=rc)
{ fprintf(stderr,"Create thread failed!\n");
return -1;
}
}
return 0;
}
void setnoblock(int sockFd) //设置非阻塞模式
{
int opt;
if((opt=fcntl(sockFd,F_GETFL))<0) //获取原来的flag;
{
printf("GET FL failed!\n");
exit(-1);
}
opt|=O_NONBLOCK;
if(fcntl(sockFd,F_SETFL,opt)<0)
printf("SET FL failed!\n");
}
int main()
{
int serverFd,j;
serverFd=socket(AF_INET,SOCK_STREAM,0); //创建服务器fd
setnoblock(serverFd); //设置为非阻塞模式
unsigned int optval;
struct linger optval1;
//设置SO_REUSEADDR选项(服务器快速重起)
optval = 0x1;
setsockopt(serverFd, SOL_SOCKET, SO_REUSEADDR, &optval, 4);
// 设置SO_LINGER选项(防范CLOSE_WAIT挂住所有套接字)
optval1.l_onoff = 1;
optval1.l_linger = 60;
setsockopt(serverFd, SOL_SOCKET, SO_LINGER, &optval1, sizeof(struct linger));
//创建epoll，并将serverFd放入监听队列
epFd=epoll_create(EPOLL_SIZE);
ev.data.fd=serverFd;
ev.events=EPOLLIN|EPOLLET;
epoll_ctl(epFd,EPOLL_CTL_ADD,serverFd,&ev);
//绑定服务器端口
struct sockaddr_in serverAddr;
socklen_t serverlen=sizeof(struct sockaddr_in);
serverAddr.sin_addr.s_addr=htonl(INADDR_ANY);
serverAddr.sin_port=htons(PORT);
if(bind(serverFd,(struct sockaddr*)&serverAddr,serverlen))
{
printf("BIND failed!\n");
exit(-1);
}
//线程池初始化
int rc = init_thread_pool();
if (0 != rc) exit(-1);
//打开监听
if(listen(serverFd,BACK_QUEUE))
{
printf("Listen failed!\n");
exit(-1);
}
//服务处理
int clientFd;
sockaddr_in clientAddr;
socklen_t clientlen;
for(;;)
{
//等待epoll事件到来，最多取EVENT_ARR个事件
int nfds=epoll_wait(epFd,evs,EVENT_ARR,-1);
//处理事件
for(int i=0;i<nfds;i++)
{
if(evs[i].data.fd==serverFd&&evs[i].events&EPOLLIN)
{
//如果是serverFd，表明有新连接连入
if((clientFd=accept(serverFd,(struct sockaddr*)&clientAddr,&clientlen))<0)
{
printf("ACCEPT failed\n");
}
printf("Connect from %s:%d\n",inet_ntoa(clientAddr.sin_addr),htons(clientAddr.sin_port));
setnoblock(clientFd);
//注册accept()到的连接
ev.data.fd=clientFd;
ev.events=EPOLLIN|EPOLLET;
epoll_ctl(epFd,EPOLL_CTL_ADD,clientFd,&ev);
}
else if(evs[i].events&EPOLLIN)
{
//如果不是serverFd,则是client的可读
printf("client can write!\n");
if((clientFd=evs[i].data.fd)>0)
{
//查询空闲线程对
for(j = 0; j < THREAD_MAX; j++) {
if (0 == s_thread_para[j][0]) break;
}
if (j >= THREAD_MAX) {
fprintf(stderr, "线程池已满, 连接将被放弃\r\n");
shutdown(clientFd, SHUT_RDWR);
close(clientFd);
continue;
}
//复制有关参数
s_thread_para[j][0] = 1;//设置活动标志为"活动"
s_thread_para[j][1] = clientFd;//客户端连接
s_thread_para[j][2] = serverFd;//服务索引
s_thread_para[j][3]=i; //epoll event id;
//线程解锁
pthread_mutex_unlock(s_mutex + j); }
else printf("other error!\n");
}
}
}
return 0;
}