1 #include <stdio.h> 2 #include <sys/ioctl.h> 3 #include <stdlib.h> 4 #include <stdint.h> 5 #include <string.h> 6 #include <strings.h> 7 #include <sys/types.h> 8 #include <sys/socket.h> 9 #include <arpa/inet.h> 10 #include <errno.h> 11 #include <assert.h> 12 13 #include "ff_config.h" 14 #include "ff_api.h" 15 #include "ff_epoll.h" 16 17 18 #define MAX_EVENTS 512 19 20 struct epoll_event ev; 21 struct epoll_event events[MAX_EVENTS]; 22 23 int epfd; 24 int sockfd; 25 26 char html[] = 27 "HTTP/1.1 200 OK\r\n" 28 "Server: F-Stack\r\n" 29 "Date: Sat, 25 Feb 2017 09:26:33 GMT\r\n" 30 "Content-Type: text/html\r\n" 31 "Content-Length: 439\r\n" 32 "Last-Modified: Tue, 21 Feb 2017 09:44:03 GMT\r\n" 33 "Connection: keep-alive\r\n" 34 "Accept-Ranges: bytes\r\n" 35 "\r\n" 36 "<!DOCTYPE html>\r\n" 37 "<html>\r\n" 38 "<head>\r\n" 39 "<title>Welcome to F-Stack!</title>\r\n" 40 "<style>\r\n" 41 " body { \r\n" 42 " width: 35em;\r\n" 43 " margin: 0 auto; \r\n" 44 " font-family: Tahoma, Verdana, Arial, sans-serif;\r\n" 45 " }\r\n" 46 "</style>\r\n" 47 "</head>\r\n" 48 "<body>\r\n" 49 "<h1>Welcome to F-Stack!</h1>\r\n" 50 "\r\n" 51 "<p>For online documentation and support please refer to\r\n" 52 "<a href=\"http://F-Stack.org/\">F-Stack.org</a>.<br/>\r\n" 53 "\r\n" 54 "<p><em>Thank you for using F-Stack.</em></p>\r\n" 55 "</body>\r\n" 56 "</html>"; 57 58 int loop(void *arg) 59 { 60 /* Wait for events to happen */ 61 62 int nevents = ff_epoll_wait(epfd, events, MAX_EVENTS, 0); 63 int i; 64 65 for (i = 0; i < nevents; ++i) { 66 /* Handle new connect */ 67 if (events[i].data.fd == sockfd) { 68 int nclientfd = ff_accept(sockfd, NULL, NULL); 69 if (nclientfd < 0) { 70 printf("ff_accept failed:%d, %s\n", errno, strerror(errno)); 71 continue; 72 } 73 74 /* Add to event list */ 75 ev.data.fd = nclientfd; 76 ev.events = EPOLLIN; 77 assert(ff_epoll_ctl(epfd, EPOLL_CTL_ADD, nclientfd, &ev) == 0); 78 //printf("A new client connected to the server..., fd:%d\n", nclientfd); 79 } else { 80 if (events[i].events & EPOLLERR ) { 81 /* Simply close socket */ 82 ff_epoll_ctl(epfd, EPOLL_CTL_DEL, events[i].data.fd, NULL); 83 ff_close(events[i].data.fd); 84 //printf("A client has left the server...,fd:%d\n", events[i].data.fd); 85 } else if (events[i].events & EPOLLIN) { 86 char buf[256]; 87 size_t readlen = ff_read( events[i].data.fd, buf, sizeof(buf)); 88 //printf("bytes are available to read..., readlen:%d, fd:%d\n", readlen, events[i].data.fd); 89 if(readlen > 0) { 90 ff_write( events[i].data.fd, html, sizeof(html)); 91 } else { 92 ff_epoll_ctl(epfd, EPOLL_CTL_DEL, events[i].data.fd, NULL); 93 ff_close( events[i].data.fd); 94 //printf("A client has left the server...,fd:%d\n", events[i].data.fd); 95 } 96 } else { 97 printf("unknown event: %8.8X\n", events[i].events); 98 } 99 } 100 } 101 } 102 103 int main(int argc, char * argv[]) 104 { 105 ff_init(argc, argv); 106 107 sockfd = ff_socket(AF_INET, SOCK_STREAM, 0); 108 printf("sockfd:%d\n", sockfd); 109 if (sockfd < 0) { 110 printf("ff_socket failed\n"); 111 exit(1); 112 } 113 114 int on = 1; 115 ff_ioctl(sockfd, FIONBIO, &on); 116 117 struct sockaddr_in my_addr; 118 bzero(&my_addr, sizeof(my_addr)); 119 my_addr.sin_family = AF_INET; 120 my_addr.sin_port = htons(80); 121 my_addr.sin_addr.s_addr = htonl(INADDR_ANY); 122 123 int ret = ff_bind(sockfd, (struct linux_sockaddr *)&my_addr, sizeof(my_addr)); 124 if (ret < 0) { 125 printf("ff_bind failed\n"); 126 exit(1); 127 } 128 129 ret = ff_listen(sockfd, MAX_EVENTS); 130 if (ret < 0) { 131 printf("ff_listen failed\n"); 132 exit(1); 133 } 134 135 assert((epfd = ff_epoll_create(0)) > 0); 136 ev.data.fd = sockfd; 137 ev.events = EPOLLIN; 138 ff_epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &ev); 139 ff_run(loop, NULL); 140 return 0; 141 } 142