epoll實現的net_echo程序（轉）

epoll實現的net_echo程序

這是我前兩天所做的一個小練習，用epoll寫個echo程序，裏面用共享內存存儲訪問信息，貼在這裏，哪天生疏了還可以過來查查~~ 更多內容請訪問： http://lmlf001.blog.sohu.com/

//net_echo.cpp
//寫一個程序，支持同時打開10w個文件句柄，申請1G共享內存，是一個tcp echo的server，採用select或epoll管理多連接
#include<sys/socket.h>
#include<sys/resource.h>
#include<stdio.h>
#include<sys/epoll.h>
#include<arpa/inet.h>
#include<strings.h>
#include<unistd.h>
#include<fcntl.h>
#include<errno.h>
#include<sys/shm.h>
#include<string.h>
#include<time.h>

#define SHM_MAX 1000000000UL  //共享內存大小
#define SHM_KEY 7896   //共享內存申請時的key
#define SERV_PORT 4466   //服務端口號
#define MAX_RLIMIT 100000  //最大訪問量
#define LISTENQ  5    //監聽隊列長度
#define MAX_LINE 128   //緩存長度
const char *local_addr="127.0.0.1";//綁定服務地址

struct access_info{  //記錄客戶訪問信息
time_t a_time;  //客戶訪問時間
in_addr_t a_ip;  //客戶ip
int a_errno;  //是否訪問成功，成功爲0，否則爲其錯誤號
};

bool setnonblocking(int fd); //設置fd爲非阻塞模式
bool set_fd_limit(unsigned int max); //設置系統允許的進程所能打開的文件描述符的最大值

int main(int argc,char **argv)
{
    int listenfd=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);//建立serv socket
    if(listenfd<0){
        perror("create socket failed!");
        return -1;
    }

    struct sockaddr_in servaddr,clientaddr;
    bzero(&servaddr,sizeof(servaddr));
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(SERV_PORT);
    inet_aton(local_addr,&(servaddr.sin_addr));
    if(bind(listenfd,(sockaddr *)&servaddr,sizeof(servaddr))<0)  //綁定本機地址
    {
        perror("bind error!");
        return -1;
    }

if(!set_fd_limit(MAX_RLIMIT)){
  perror("setrlimit failed!");
  return -1;
}

struct epoll_event ev,events[20];
int epfd=epoll_create(MAX_RLIMIT);  //epoll_create
if(!setnonblocking(listenfd))return -1;
ev.data.fd=listenfd;
ev.events=EPOLLIN|EPOLLET;
epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&ev);//把listenfd加入到epoll監聽隊列

int shm_id=shmget(SHM_KEY,SHM_MAX,IPC_CREAT|0600); //申請共享內存
if(shm_id==-1){
  perror("shmget");
  return -2;
}
struct access_info  client_info,*pos;  //客戶信息
int *head;
head=(int *)shmat(shm_id,0,0);
if(int(head)==-1){
  perror("shmat");
  return -1;
}
*head=1;      //服務器在運行狀態，若該值變爲0，則關閉服務器
if(*(head+1)!=1){    // head+1服務器是否第一次運行，head+2共享內存存儲的信息數量
  *(head+1)=1;             //  ___________
  *(head+2)=0;   //head-->|___ 0/1___|  服務器的運行狀態
}                            //   |___ 0/1___|  共享內存是否使用過，是爲1，否則爲0
                               //   |___  n____|   共享內存存儲信息數量  0~SHM_MAX/(3*4*8)-1
pos=(struct access_info *)(head)+1+*(head+2); //記錄信息的開始位置

listen(listenfd,LISTENQ);  //監聽客戶端請求
int nfds,i,connfd,sockfd,n;
socklen_t len;
char line[MAX_LINE];
while(*head)
{
  nfds=epoll_wait(epfd,events,20,500); //檢測活躍連接
  for(i=0;i<nfds;i++)
  {
   if(events[i].data.fd==listenfd)    //有新連接到來
   {
    len=sizeof(clientaddr);
    connfd=accept(listenfd,(sockaddr *)&clientaddr,&len);

    client_info.a_time=time(NULL);     //註冊客戶信息
    client_info.a_ip=clientaddr.sin_addr.s_addr;
    client_info.a_errno=0;

    if(connfd<0){
     perror("connfd<0!");
     client_info.a_errno=errno;
     continue;
    }

    memcpy(pos,&client_info,sizeof(client_info));
    pos++;         //共享內存指針後移，並把信息數量加1
    if((*(head+2))++>4*SHM_MAX/(3*8*4*5))  //共享內存剩餘不足1/5時發出警告信息
     fprintf(stderr,"Warning:share memory is being not enough

/n Left:%d/n",SHM_MAX-*(head+2)*3*4*8);


    if(!setnonblocking(connfd))continue;
    ev.data.fd=connfd;
    ev.events=EPOLLIN|EPOLLET;
    epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);//新連接加入epoll_wait
   }
   else if(events[i].events&EPOLLIN)   //連接可讀
   {
    if((sockfd=events[i].data.fd)<0)continue;
    while((n=read(sockfd,line,MAX_LINE))==MAX_LINE)
     write(sockfd,line,n);
    if(n<0)
    {
     if (errno == ECONNRESET) {
      events[i].data.fd = -1;
      epoll_ctl(epfd,EPOLL_CTL_DEL,sockfd,NULL);
      close(sockfd);
      continue;
     }
     else continue;  //恰好讀完MAX_LINE後無數據
    }
    else if(n==0){      //客戶端關閉連接
     epoll_ctl(epfd,EPOLL_CTL_DEL,sockfd,NULL);
     close(sockfd);
     events[i].data.fd = -1;
     continue;
    }
    else write(sockfd,line,n);
   }
/*   else if(events[i].events&EPOLLOUT)
   {
    sockfd=events[i].data.fd;
    if(sockfd<0)continue;
    write(sockfd,line,n);
    ev.data.fd=sockfd;
    ev.events=EPOLLIN|EPOLLET;
    epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);
   }
*/ }
    }
shmdt(head);  //卸載共享內存
close(epfd);
close(listenfd);
return 0;
}

bool setnonblocking(int sock)
{
    int opts;
    opts=fcntl(sock,F_GETFL);
    if(opts<0)
{
  perror("fcntl(sock,GETFL)");
  return false;
    }
opts = opts|O_NONBLOCK;
if(fcntl(sock,F_SETFL,opts)<0)
{
  perror("fcntl(sock,SETFL,opts)");
  return false;
    }
return true;
}

bool set_fd_limit(unsigned int max)
{
    struct rlimit rlim,rlim_new;
    if (getrlimit(RLIMIT_NOFILE, &rlim)!=0)
        return false;
    if(rlim.rlim_cur>=max) return true;

    if(rlim.rlim_max==RLIM_INFINITY||rlim.rlim_max>=max){
        rlim_new.rlim_max = rlim.rlim_max;
        rlim_new.rlim_cur = max;
    }
    else{
        if(geteuid()!=0){errno=1;return false; }
  rlim_new.rlim_max=rlim_new.rlim_cur=max;
    }
if (setrlimit(RLIMIT_NOFILE, &rlim_new)!=0) {/* failed. try raising just to the old max */
  int err=errno;
  setrlimit(RLIMIT_NOFILE, &rlim);
  errno=err;
  return false;
    }
return true;
}

/*----------------------------------------------------------------*/

//net_echo_shutdown.cpp
//啓動該進程時，關閉net_echo服務進程

#include<sys/shm.h>
#include<stdio.h>
#include<unistd.h>
#include<errno.h>
#define SHM_MAX 1000000000UL  //共享內存大小
#define SHM_KEY 7896   //共享內存申請時的key

#ifndef IPC_ALLOC
#define IPC_ALLOC IPC_CREAT
#endif

int main(int argc,char **argv)
{
if(geteuid()!=0){
  errno=1;
  perror("net_echo_shutdown:");
  return -1;
}
int shmid;
if((shmid=shmget(SHM_KEY,SHM_MAX,IPC_ALLOC|0600))==-1)
{
  perror("shmget()");
  return -1;
}
int *head=(int *)shmat(shmid,0,0);
if(int(head)==-1){
  perror("shmat()");
  return -1;
}
if(*head!=1){     //服務器並未運行
  fprintf(stderr,"Net_echo server is not running

/n");
return -1;
}
*head=0; //設置關閉標誌
printf("Shutdown the echo server

/n");
sleep(2);
shmdt(head);
return 0;
}

/******************************************************************/

//print_shm.cpp
//讀取並打印共享內存信息
#include<stdio.h>
#include<sys/shm.h>
#include<unistd.h>
#include<errno.h>
#include<time.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<string.h>
#define SHM_KEY 7896
#define SHM_MAX 1000000000UL
#ifndef IPC_ALLOC
#define IPC_ALLOC IPC_CREAT
#endif

struct access_info{  //記錄客戶訪問信息
time_t a_time;  //客戶訪問時間
in_addr_t a_ip;  //客戶ip
int a_errno;  //是否訪問成功，成功爲0，否則爲其錯誤號
};

int main(int argc,char **argv)
{
if(geteuid()!=0){
  errno=1;
  perror("print_shm:");
  return -1;
}
int shmid=shmget(SHM_KEY,SHM_MAX,IPC_ALLOC|0600);
if(shmid==-1)
{
  perror("shmget()");
  return -1;
}
int *head=(int *)shmat(shmid,0,0);
if(int(head)==-1){
  perror("shmat()");
  return -1;
}
if(*(head+1)!=1){
  fprintf(stderr,"SHM have not be used!/n");
  return -1;
}
struct access_info *pos=(access_info *)(head)+1;
for(int i=0;i<*(head+2);i++,pos++)
  printf("%-15s%-10s%20s%s",inet_ntoa(*(in_addr *)&(pos->a_ip)),pos->a_errno==0?"Success!":"Failed:",
   pos->a_errno==0?"":strerror(pos->a_errno),ctime(&pos->a_time));

shmdt(head);
return 0;
}

epoll實現的net_echo程序（轉）