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