多程序伺服器中,epoll的建立應該在建立子程序之後
阿新 • • 發佈:2018-12-27
#include <iostream>
#include <sys/socket.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>usingnamespace std;
#define MAXLINE 5#define OPEN_MAX 100#define LISTENQ 20#define SERV_PORT 5000#define INFTIM 1000
typedef struct task_t
{
int fd;
char buffer[100];
int n;
}task_t;
int CreateWorker(int nWorker)
{
if (0< nWorker)
{
bool bIsChild;
pid_t nPid;
while (!bIsChild)
{
if (0< nWorker)
{
nPid = ::fork();
if (nPid >0)
{
bIsChild =false;
--nWorker;
}
elseif (0== nPid)
{
bIsChild =true;
printf( "create worker %d success!\n", ::getpid());
}
else
{
printf("fork error: %s\n", ::strerror(errno));
return-1;
}
}
else
{
int nStatus;
if (-1== ::wait(&nStatus))
{
++nWorker;
}
}
}
}
return0;
}
void setnonblocking(int sock)
{
int opts;
opts=fcntl(sock,F_GETFL);
if(opts<0)
{
perror("fcntl(sock,GETFL)");
exit(1);
}
opts = opts|O_NONBLOCK;
if(fcntl(sock,F_SETFL,opts)<0)
{
perror("fcntl(sock,SETFL,opts)");
exit(1);
}
}
int main()
{
int i, maxi, listenfd, connfd, sockfd,epfd,nfds;
ssize_t n;
char line[MAXLINE];
socklen_t clilen;
struct epoll_event ev,events[20];
struct sockaddr_in clientaddr;
struct sockaddr_in serveraddr;
listenfd = socket(AF_INET, SOCK_STREAM, 0);
bzero(&serveraddr, sizeof(serveraddr));
serveraddr.sin_family = AF_INET;
char*local_addr="127.0.0.1";
inet_aton(local_addr,&(serveraddr.sin_addr));//htons(SERV_PORT); serveraddr.sin_port=htons(SERV_PORT);
// 地址重用int nOptVal =1;
socklen_t nOptLen =sizeof(int);
if (-1== ::setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &nOptVal, nOptLen))
{
return-1;
}
setnonblocking(listenfd);
bind(listenfd,(sockaddr *)&serveraddr, sizeof(serveraddr));
listen(listenfd, LISTENQ);
CreateWorker(5);
//把socket設定為非阻塞方式
//生成用於處理accept的epoll專用的檔案描述符 epfd=epoll_create(256);
//設定與要處理的事件相關的檔案描述符 ev.data.fd=listenfd;
//設定要處理的事件型別 ev.events=EPOLLIN|EPOLLET;
//ev.events=EPOLLIN;
//註冊epoll事件 epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&ev);
//CreateWorker(5);
maxi =0;
task_t task;
task_t *ptask;
while(true)
{
//等待epoll事件的發生 nfds=epoll_wait(epfd,events,20,500);
//處理所發生的所有事件 for(i=0;i<nfds;++i)
{
if(events[i].data.fd==listenfd)
{
connfd = accept(listenfd,NULL, NULL);
if(connfd<0){
printf("connfd<0, listenfd = %d\n", listenfd);
printf("error = %s\n", strerror(errno));
exit(1);
}
setnonblocking(connfd);
//設定用於讀操作的檔案描述符 memset(&task, 0, sizeof(task));
task.fd = connfd;
ev.data.ptr =&task;
//設定用於註冊的讀操作事件 ev.events=EPOLLIN|EPOLLET;
//ev.events=EPOLLIN;
//註冊ev epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);
}
elseif(events[i].events&EPOLLIN)
{
cout <<"EPOLLIN"<< endl;
ptask = (task_t*)events[i].data.ptr;
sockfd = ptask->fd;
if ( (ptask->n = read(sockfd, ptask->buffer, 100)) <0) {
if (errno == ECONNRESET) {
close(sockfd);
events[i].data.ptr = NULL;
} else
std::cout<<"readline error"<<std::endl;
} elseif (ptask->n ==0) {
close(sockfd);
events[i].data.ptr = NULL;
}
ptask->buffer[ptask->n] ='\0';
cout <<"read "<< ptask->buffer << endl;
//設定用於寫操作的檔案描述符 ev.data.ptr = ptask;
//設定用於注測的寫操作事件 ev.events=EPOLLOUT|EPOLLET;
//修改sockfd上要處理的事件為EPOLLOUT epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);
}
elseif(events[i].events&EPOLLOUT)
{
cout <<"EPOLLOUT"<< endl;
ptask = (task_t*)events[i].data.ptr;
sockfd = ptask->fd;
write(sockfd, ptask->buffer, ptask->n);
//設定用於讀操作的檔案描述符 ev.data.ptr = ptask;
//修改sockfd上要處理的事件為EPOLIN epoll_ctl(epfd,EPOLL_CTL_DEL,sockfd,&ev);
cout <<"write "<< ptask->buffer;
memset(ptask, 0, sizeof(*ptask));
close(sockfd);
}
}
}
return0;
}
#include <sys/socket.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/wait.h>usingnamespace std;
typedef struct task_t
{
int fd;
char buffer[100];
int n;
}task_t;
int CreateWorker(int nWorker)
{
if (0< nWorker)
{
bool bIsChild;
pid_t nPid;
while (!bIsChild)
{
{
nPid = ::fork();
if (nPid >0)
{
bIsChild =false;
--nWorker;
}
elseif (0== nPid)
{
bIsChild =true;
printf(
}
else
{
printf("fork error: %s\n", ::strerror(errno));
return-1;
}
}
else
{
int nStatus;
if (-1== ::wait(&nStatus))
{
++nWorker;
}
}
}
}
return0;
}
void setnonblocking(int sock)
{
int opts;
opts=fcntl(sock,F_GETFL);
if(opts<0)
{
perror("fcntl(sock,GETFL)");
exit(1);
}
opts = opts|O_NONBLOCK;
if(fcntl(sock,F_SETFL,opts)<0)
{
perror("fcntl(sock,SETFL,opts)");
exit(1);
}
}
int main()
{
int i, maxi, listenfd, connfd, sockfd,epfd,nfds;
ssize_t n;
char line[MAXLINE];
socklen_t clilen;
struct epoll_event ev,events[20];
struct sockaddr_in clientaddr;
struct sockaddr_in serveraddr;
listenfd = socket(AF_INET, SOCK_STREAM, 0);
bzero(&serveraddr, sizeof(serveraddr));
serveraddr.sin_family = AF_INET;
char*local_addr="127.0.0.1";
inet_aton(local_addr,&(serveraddr.sin_addr));//htons(SERV_PORT); serveraddr.sin_port=htons(SERV_PORT);
// 地址重用int nOptVal =1;
socklen_t nOptLen =sizeof(int);
if (-1== ::setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &nOptVal, nOptLen))
{
return-1;
}
setnonblocking(listenfd);
bind(listenfd,(sockaddr *)&serveraddr, sizeof(serveraddr));
listen(listenfd, LISTENQ);
CreateWorker(5);
//把socket設定為非阻塞方式
//生成用於處理accept的epoll專用的檔案描述符 epfd=epoll_create(256);
//設定與要處理的事件相關的檔案描述符 ev.data.fd=listenfd;
//設定要處理的事件型別 ev.events=EPOLLIN|EPOLLET;
//ev.events=EPOLLIN;
//註冊epoll事件 epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&ev);
//CreateWorker(5);
maxi =0;
task_t task;
task_t *ptask;
while(true)
{
//等待epoll事件的發生 nfds=epoll_wait(epfd,events,20,500);
//處理所發生的所有事件 for(i=0;i<nfds;++i)
{
if(events[i].data.fd==listenfd)
{
connfd = accept(listenfd,NULL, NULL);
if(connfd<0){
printf("connfd<0, listenfd = %d\n", listenfd);
printf("error = %s\n", strerror(errno));
exit(1);
}
setnonblocking(connfd);
//設定用於讀操作的檔案描述符 memset(&task, 0, sizeof(task));
task.fd = connfd;
ev.data.ptr =&task;
//設定用於註冊的讀操作事件 ev.events=EPOLLIN|EPOLLET;
//ev.events=EPOLLIN;
//註冊ev epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);
}
elseif(events[i].events&EPOLLIN)
{
cout <<"EPOLLIN"<< endl;
ptask = (task_t*)events[i].data.ptr;
sockfd = ptask->fd;
if ( (ptask->n = read(sockfd, ptask->buffer, 100)) <0) {
if (errno == ECONNRESET) {
close(sockfd);
events[i].data.ptr = NULL;
} else
std::cout<<"readline error"<<std::endl;
} elseif (ptask->n ==0) {
close(sockfd);
events[i].data.ptr = NULL;
}
ptask->buffer[ptask->n] ='\0';
cout <<"read "<< ptask->buffer << endl;
//設定用於寫操作的檔案描述符 ev.data.ptr = ptask;
//設定用於注測的寫操作事件 ev.events=EPOLLOUT|EPOLLET;
//修改sockfd上要處理的事件為EPOLLOUT epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);
}
elseif(events[i].events&EPOLLOUT)
{
cout <<"EPOLLOUT"<< endl;
ptask = (task_t*)events[i].data.ptr;
sockfd = ptask->fd;
write(sockfd, ptask->buffer, ptask->n);
//設定用於讀操作的檔案描述符 ev.data.ptr = ptask;
//修改sockfd上要處理的事件為EPOLIN epoll_ctl(epfd,EPOLL_CTL_DEL,sockfd,&ev);
cout <<"write "<< ptask->buffer;
memset(ptask, 0, sizeof(*ptask));
close(sockfd);
}
}
}
return0;
}