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: 438\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 while (1) { 69 int nclientfd = ff_accept(sockfd, NULL, NULL); 70 if (nclientfd < 0) { 71 break; 72 } 73 74 /* Add to event list */ 75 ev.data.fd = nclientfd; 76 ev.events = EPOLLIN; 77 if (ff_epoll_ctl(epfd, EPOLL_CTL_ADD, nclientfd, &ev) != 0) { 78 printf("ff_epoll_ctl failed:%d, %s\n", errno, 79 strerror(errno)); 80 break; 81 } 82 } 83 } else { 84 if (events[i].events & EPOLLERR ) { 85 /* Simply close socket */ 86 ff_epoll_ctl(epfd, EPOLL_CTL_DEL, events[i].data.fd, NULL); 87 ff_close(events[i].data.fd); 88 } else if (events[i].events & EPOLLIN) { 89 char buf[256]; 90 size_t readlen = ff_read( events[i].data.fd, buf, sizeof(buf)); 91 if(readlen > 0) { 92 ff_write( events[i].data.fd, html, sizeof(html) - 1); 93 } else { 94 ff_epoll_ctl(epfd, EPOLL_CTL_DEL, events[i].data.fd, NULL); 95 ff_close( events[i].data.fd); 96 } 97 } else { 98 printf("unknown event: %8.8X\n", events[i].events); 99 } 100 } 101 } 102 } 103 104 int main(int argc, char * argv[]) 105 { 106 ff_init(argc, argv); 107 108 sockfd = ff_socket(AF_INET, SOCK_STREAM, 0); 109 printf("sockfd:%d\n", sockfd); 110 if (sockfd < 0) { 111 printf("ff_socket failed\n"); 112 exit(1); 113 } 114 115 int on = 1; 116 ff_ioctl(sockfd, FIONBIO, &on); 117 118 struct sockaddr_in my_addr; 119 bzero(&my_addr, sizeof(my_addr)); 120 my_addr.sin_family = AF_INET; 121 my_addr.sin_port = htons(80); 122 my_addr.sin_addr.s_addr = htonl(INADDR_ANY); 123 124 int ret = ff_bind(sockfd, (struct linux_sockaddr *)&my_addr, sizeof(my_addr)); 125 if (ret < 0) { 126 printf("ff_bind failed\n"); 127 exit(1); 128 } 129 130 ret = ff_listen(sockfd, MAX_EVENTS); 131 if (ret < 0) { 132 printf("ff_listen failed\n"); 133 exit(1); 134 } 135 136 assert((epfd = ff_epoll_create(0)) > 0); 137 ev.data.fd = sockfd; 138 ev.events = EPOLLIN; 139 ff_epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &ev); 140 ff_run(loop, NULL); 141 return 0; 142 } 143