#include <sys/socket.h>
#include <sys/wait.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <sys/epoll.h>
#include <sys/sendfile.h>
#include <sys/stat.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <fcntl.h>
#include <errno.h>
#include <arpa/inet.h>
#include <sys/signal.h>
#define MAX_EVENTS 200
#define PORT 8888
int sum = 0;
//设置socket连接为非阻塞模式
void setnonblocking(int sockfd)
{
int opts;
opts = fcntl(sockfd, F_GETFL);
if(opts < 0)
{
perror("fcntl(F_GETFL)\n");
exit(1);
}
opts = (opts | O_NONBLOCK);
if(fcntl(sockfd, F_SETFL, opts) < 0)
{
perror("fcntl(F_SETFL)\n");
exit(1);
}
}
void sig(int sig)
{
printf("sum is %d\n",sum);
}
int main()
{
signal(SIGINT,sig);
struct epoll_event ev, events[MAX_EVENTS];
int addrlen, listenfd, conn_sock, nfds, epfd, fd, i, nread, n;
struct sockaddr_in local, remote;
char buf[BUFSIZ];
//创建listen socket
if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
perror("sockfd\n");
exit(1);
}
setnonblocking(listenfd);
bzero(&local, sizeof(local));
local.sin_family = AF_INET;
//local.sin_addr.s_addr = htonl(INADDR_ANY);;
local.sin_addr.s_addr = inet_addr("49.4.132.245");
// local.sin_addr.s_addr = inet_addr("172.16.29.52");
local.sin_port = htons(PORT);
if( bind(listenfd, (struct sockaddr *) &local, sizeof(local)) < 0)
{
perror("bind\n");
exit(1);
}
listen(listenfd, 512);
epfd = epoll_create(MAX_EVENTS);
if (epfd == -1)
{
perror("epoll_create");
exit(EXIT_FAILURE);
}
//setnonblocking(epfd);
//ev.events = EPOLLIN ;
ev.events = EPOLLIN | EPOLLET;
ev.data.fd = listenfd;
if (epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev) == -1)
{
perror("epoll_ctl: listen_sock");
exit(EXIT_FAILURE);
}
int ii = 0;
int iii = 0;
for (;;)
{
nfds = epoll_wait(epfd, events, MAX_EVENTS, -1);
if (nfds == -1)
{
if(errno == EINTR)
{
continue;
}
else
{
perror("epoll_pwait");
exit(EXIT_FAILURE);
}
}
for (i = 0; i < nfds; ++i)
{
fd = events[i].data.fd;
if (fd == listenfd)
{
while ((conn_sock = accept(listenfd,(struct sockaddr *) &remote,(socklen_t*)&addrlen)) > 0)
//if((conn_sock = accept(listenfd,(struct sockaddr *) &remote,(socklen_t*)&addrlen)) > 0)
{
printf("accept while loop times %d,",++iii);
setnonblocking(conn_sock);
ev.events = EPOLLIN | EPOLLET;
ev.data.fd = conn_sock;
if (epoll_ctl(epfd, EPOLL_CTL_ADD, conn_sock,&ev) == -1)
{
perror("epoll_ctl: add");
exit(EXIT_FAILURE);
}
printf("accept client times %d\n",++ii);
}
sum+=iii;
iii = 0;
if (conn_sock == -1)
{
if (errno != EAGAIN && errno != ECONNABORTED && errno != EPROTO && errno != EINTR)
perror("accept");
}
continue;
}
if (events[i].events & EPOLLIN)
{
n = 0;
while ((nread = read(fd, buf + n, BUFSIZ-1)) > 0)
{
n += nread;
}
if (nread == -1 && errno != EAGAIN)
{
perror("read error");
}
ev.data.fd = fd;
ev.events = events[i].events | EPOLLOUT;
if (epoll_ctl(epfd, EPOLL_CTL_MOD, fd, &ev) == -1)
{
perror("epoll_ctl: mod");
}
}
if (events[i].events & EPOLLOUT)
{
sprintf(buf, "HTTP/1.1 200 OK\r\nContent-Length: %d\r\n\r\nHello World", 11);
int nwrite, data_size = strlen(buf);
n = data_size;
while (n > 0)
{
nwrite = write(fd, buf + data_size - n, n);
if (nwrite < n)
{
if (nwrite == -1 && errno != EAGAIN)
{
perror("write error");
}
break;
}
n -= nwrite;
}
close(fd);
}
}
}
return 0;
}