| #include <stdio.h> #include <stdlib.h> #include <errno.h> #include <string.h> #include <sys/types.h> #include <netinet/in.h> #include <sys/socket.h> #include <sys/wait.h> #include <unistd.h> #include <arpa/inet.h> #include <openssl/ssl.h> #include <openssl/err.h> #include <fcntl.h> #include <sys/epoll.h> #include <sys/time.h> #include <sys/resource.h> #define MAXBUF 1024 #define MAXEPOLLSIZE 10000 /* setnonblocking - 設(shè)置句柄為非阻塞方式 */ int setnonblocking(int sockfd) { if (fcntl(sockfd, F_SETFL, fcntl(sockfd, F_GETFD, 0)|O_NONBLOCK) == -1) { return -1; } return 0; } /* handle_message - 處理每個 socket 上的消息收發(fā) */ int handle_message(int new_fd) { char buf[MAXBUF + 1]; int len; /* 開始處理每個新連接上的數(shù)據(jù)收發(fā) */ bzero(buf, MAXBUF + 1); /* 接收客戶端的消息 */ len = recv(new_fd, buf, MAXBUF, 0); if (len > 0) printf ("%d接收消息成功:'%s'，共%d個字節(jié)的數(shù)據(jù)\n", new_fd, buf, len); else { if (len < 0) printf ("消息接收失敗！錯誤代碼是%d，錯誤信息是'%s'\n", errno, strerror(errno)); close(new_fd); return -1; } /* 處理每個新連接上的數(shù)據(jù)收發(fā)結(jié)束 */ return len; } /************關(guān)于本文檔******************************************** *filename: epoll-server.c *purpose: 演示epoll處理海量socket連接的方法 *wrote by: zhoulifa(zhoulifa@163.com) 周立發(fā)(http://zhoulifa.bokee.com) Linux愛好者 Linux知識傳播者 SOHO族 開發(fā)者 最擅長C語言 *date time:2007-01-31 21:00 *Note: 任何人可以任意復(fù)制代碼并運用這些文檔，當(dāng)然包括你的商業(yè)用途 * 但請遵循GPL *Thanks to:Google *Hope:希望越來越多的人貢獻(xiàn)自己的力量，為科學(xué)技術(shù)發(fā)展出力 * 科技站在巨人的肩膀上進(jìn)步更快！感謝有開源前輩的貢獻(xiàn)！ *********************************************************************/ int main(int argc, char **argv) { int listener, new_fd, kdpfd, nfds, n, ret, curfds; socklen_t len; struct sockaddr_in my_addr, their_addr; unsigned int myport, lisnum; struct epoll_event ev; struct epoll_event events[MAXEPOLLSIZE]; struct rlimit rt; if (argv[1]) myport = atoi(argv[1]); else myport = 7838; if (argv[2]) lisnum = atoi(argv[2]); else lisnum = 2; /* 設(shè)置每個進(jìn)程允許打開的最大文件數(shù) */ rt.rlim_max = rt.rlim_cur = MAXEPOLLSIZE; if (setrlimit(RLIMIT_NOFILE, &rt) == -1) { perror("setrlimit"); exit(1); } else printf("設(shè)置系統(tǒng)資源參數(shù)成功！\n"); /* 開啟 socket 監(jiān)聽 */ if ((listener = socket(PF_INET, SOCK_STREAM, 0)) == -1) { perror("socket"); exit(1); } else pintf("socket 創(chuàng)建成功！\n"); setnonblocking(listener); bzero(&my_addr, sizeof(my_addr)); my_addr.sin_family = PF_INET; my_addr.sin_port = htons(myport); if (argv[3]) my_addr.sin_addr.s_addr = inet_addr(argv[3]); else my_addr.sin_addr.s_addr = INADDR_ANY; if (bind (listener, (struct sockaddr *) &my_addr, sizeof(struct sockaddr)) == -1) { perror("bind"); exit(1); } else printf("IP 地址和端口綁定成功\n"); if (listen(listener, lisnum) == -1) { perror("listen"); exit(1); } else printf("開啟服務(wù)成功！\n"); /* 創(chuàng)建 epoll 句柄，把監(jiān)聽 socket 加入到 epoll 集合里 */ kdpfd = epoll_create(MAXEPOLLSIZE); len = sizeof(struct sockaddr_in); ev.events = EPOLLIN | EPOLLET; ev.data.fd = listener; if (epoll_ctl(kdpfd, EPOLL_CTL_ADD, listener, &ev) < 0) { fprintf(stderr, "epoll set insertion error: fd=%d\n", listener); return -1; } else printf("監(jiān)聽 socket 加入 epoll 成功！\n"); curfds = 1; while (1) { /* 等待有事件發(fā)生 */ nfds = epoll_wait(kdpfd, events, curfds, -1); if (nfds == -1) { perror("epoll_wait"); break; } /* 處理所有事件 */ for (n = 0; n < nfds; ++n) { if (events[n].data.fd == listener) { new_fd = accept(listener, (struct sockaddr *) &their_addr, &len); if (new_fd < 0) { perror("accept"); continue; } else printf("有連接來自于： %d:%d， 分配的 socket 為:%d\n", inet_ntoa(their_addr.sin_addr), ntohs(their_addr.sin_port), new_fd); setnonblocking(new_fd); ev.events = EPOLLIN | EPOLLET; ev.data.fd = new_fd; if (epoll_ctl(kdpfd, EPOLL_CTL_ADD, new_fd, &ev) < 0) { fprintf(stderr, "把 socket '%d' 加入 epoll 失?。?s\n", new_fd, strerror(errno)); return -1; } curfds++; } else { ret = handle_message(events[n].data.fd); if (ret < 1 && errno != 11) { epoll_ctl(kdpfd, EPOLL_CTL_DEL, events[n].data.fd, &ev); curfds--; } } } } close(listener); return 0; } |
