test

#include<stdio.h> #include<sys/socket.h> #include<sys/types.h> #include <arpa/inet.h> #include <netinet/in.h> #include <sys/epoll.h> #include <stdlib.h> #define SIZE 1024 void usege(const char * arg) { printf("%s ip port\n",arg); } typedef struct epoll_buff{ int fd; char buf [SIZE]; } epoll_buf,*epoll_buf_p; epoll_buf_p alloc(int fd) { epoll_buf_p ret =(epoll_buf_p) malloc(sizeof(struct epoll_buff)); if (ret ==NULL) { perror("malloc"); exit(10); } return ret; } int startup(const char * ip,const char * port) { int fd = socket(AF_INET,SOCK_STREAM,0); if(fd<0) { perror("socket"); exit(1); } struct sockaddr_in sock_addr; sock_addr.sin_family=AF_INET; sock_addr.sin_port = htons(atoi(port)); sock_addr.sin_addr.s_addr=inet_addr(ip); socklen_t len=sizeof(sock_addr); if((bind(fd,(struct sockaddr*)&sock_addr,len))<0) { perror("bind"); exit(2); } if(listen(fd,128)<0) { perror("listen"); exit(3); } return fd; } int main(int argv,const char * args[]) { if(argv != 3) { usege(args[0]); exit(4); } int listen_sock=startup(args[1],args[2]); int epfd=epoll_create(256); if(epfd<0) { perror("epoll_create");} struct epoll_event ev; struct epoll_event env[32]; ev.events=EPOLLIN; ev.data.ptr=alloc(listen_sock); if(epoll_ctl(epfd, EPOLL_CTL_ADD,listen_sock,&ev)<0) { perror("epoll_ctl"); exit(5); } while(1) { int timeout = 1000; int reve_n = epoll_wait(epfd,env,sizeof(env)/sizeof(env[0]),timeout); switch (reve_n) { case 0: printf("time out \n"); break; case -1: perror("epoll_wait"); exit(6); default: { int idx=0; for(idx;idx<reve_n;idx++) { epoll_buf_p p=(epoll_buf_p)env[idx].data.ptr; if(p->fd == listen_sock&&env[idx].events==EPOLLIN) { struct sockaddr_in cilent ; socklen_t len=sizeof(cilent); int sock=0; while((sock=accept(listen_sock,(struct sockaddr*)&cilent,&len))>0) { const char * cilent_ip=inet_ntoa(cilent.sin_addr); short int cilent_port=ntohs(cilent.sin_port); printf("cilent connect ip:%s port:%d \n",cilent_ip,cilent_port); } if(sock<0) { perror("accept"); } ev.events=EPOLLIN; ev.data.ptr=alloc(sock); if(epoll_ctl(epfd, EPOLL_CTL_ADD,sock,&ev)<0) { perror("epoll_ctl"); exit(7); } continue; } else if(env[idx].events==EPOLLIN) { int res=read(p->fd,p->buf,SIZE); if(res<0){ perror("read"); exit(8); } else if(res==0) { printf("cilent quit!!\n"); close(p->fd); free(p); epoll_ctl(epfd,EPOLL_CTL_MOD,p->fd,NULL); } else { (p->buf)[res-1]=0; printf("####cilent : %s",p->buf); fflush(stdout); } ev.events=(env[idx]).events|EPOLLOUT; if(epoll_ctl(epfd,EPOLL_CTL_MOD,p->fd,&ev)<0) { perror("epoll_ctl"); exit(9); } } else{ // end read int ret= sprintf(p->buf,"HTTP/1.1 200 OK\r\n Content-Length :%s \r\n\r\n hello world"); write(p->fd,p->buf,ret); } } } break; }// end switch } // end while(1) return 0; }