poll() replaced by epoll()

2022-07-23 07:01:41 +02:00
parent cb8cc22c67
commit 72d9709ca7
3 changed files with 147 additions and 107 deletions
--- a/srcs/Webserv.cpp
+++ b/srcs/Webserv.cpp
@@ -4,12 +4,12 @@
 Webserv::Webserv()
 {
 	std::cout << "Server init\n";
-	_socket_fd = ::socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
-	//_socket_fd = ::socket(AF_INET, SOCK_STREAM, 0);
-	if (_socket_fd == -1)
+
+	_epfd = ::epoll_create1(0); // (EPOLL_CLOEXEC) for CGI fork ?
+	if (_epfd == -1)
 	{
-		std::perror("err socket(): ");
-		throw std::runtime_error("Socket init");
+		std::perror("err epoll_create1(): ");
+		throw std::runtime_error("Epoll init");
 	}
 }

@@ -20,6 +20,8 @@ Webserv::Webserv()

 Webserv::~Webserv()
 {
+	close(_socket_fd);
+	close(_epfd);
 	std::cout << "Server destroyed\n";
 }

@@ -31,7 +33,71 @@ Webserv::~Webserv()
  ///////////////
 // Functions //

-void	Webserv::bind(in_port_t port)
+void	Webserv::init_virtual_servers() // ADD config param
+{
+	_socket_fd = ::socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0); // (SOCK_CLOEXEC) for CGI fork ?
+	if (_socket_fd == -1)
+	{
+		std::perror("err socket(): ");
+		throw std::runtime_error("Socket init");
+	}
+
+	_bind(_socket_fd, 80);
+	_listen(_socket_fd, 512); // 512 arbitrary
+
+	struct epoll_event	ev;
+	std::memset(&ev, 0, sizeof ev);
+	ev.events = EPOLLIN;
+	ev.data.fd = _socket_fd;
+	if (::epoll_ctl(_epfd, EPOLL_CTL_ADD, _socket_fd, &ev) == -1)
+	{
+		std::perror("err epoll_ctl(): ");
+		throw std::runtime_error("Socket init");
+	}
+}
+
+void	Webserv::start()
+{
+	std::cout << "Server started\n";
+	struct epoll_event	events[MAX_EVENTS];
+	int	nfds;
+	int	i;
+	int	count_loop = 0;
+	std::cout << ++count_loop << "----loop epoll()\n";
+
+	while ( (nfds = ::epoll_wait(_epfd, events, MAX_EVENTS, TIMEOUT)) != -1)
+	{
+		if (nfds == 0)
+		{
+			(void)0;
+			// TODO : parcourir les "Clients" encore ouvert et les close() tous
+		}
+		i = 0;
+		while (i < nfds)
+		{
+			if ((events[i].data.fd == _socket_fd) && (events[i].events & EPOLLIN))
+				_accept_connection(events[i].data.fd);
+			else if (events[i].events & EPOLLIN)
+				_read_request(events[i].data.fd);
+			else if (events[i].events & EPOLLOUT)
+				_send_response(events[i].data.fd);
+			++i;
+		}
+		std::cout << ++count_loop << "----loop epoll()\n";
+	}
+
+	if (nfds == -1)
+	{
+		std::perror("err epoll_wait(): ");
+		throw std::runtime_error("Epoll wait");
+	}
+}
+
+
+  ///////////////////////
+ // Private Functions //
+
+void	Webserv::_bind(int socket_fd, in_port_t port)
 {
 	// cast invalid ? how to ?
 	// const struct sockaddr* cast_test = static_cast<const struct sockaddr*>(addr);
@@ -42,118 +108,79 @@ void	Webserv::bind(in_port_t port)
 	addr.sin_port = ::htons(port);
 	addr.sin_addr.s_addr = ::htonl(INADDR_ANY); // htonl useless with 0 value (INADDR_ANY) ?

-	if (::bind(_socket_fd, (const sockaddr*)&addr, sizeof addr) == -1)
+	if (::bind(socket_fd, (const sockaddr*)&addr, sizeof addr) == -1)
 	{
 		std::perror("err bind(): ");
 		throw std::runtime_error("Socket bind");		
 	}
 }

-void	Webserv::listen(unsigned int max_connections)
+void	Webserv::_listen(int socket_fd, unsigned int max_connections)
 {
-	if (::listen(_socket_fd, max_connections) == -1)
+	if (::listen(socket_fd, max_connections) == -1)
 	{
 		std::perror("err listen(): ");
 		throw std::runtime_error("Socket listen");
 	}
 }

-#define BUFSIZE 8192
-#define MSG_TEST "Le Webserv / 20 =D\n"
-#define MSG_BOUNCE "bounced properly ;)\n" // placeholder
-
-void	Webserv::start()
+void	Webserv::_accept_connection(int fd)
 {
 	struct sockaddr_in	addr;
 	socklen_t 			addr_len;
 	int					accepted_fd;
-	char				buf[BUFSIZE+1]; // WIP buffer. need to try with std::vector or std::string.
-	int					ret;

-	int	timeout = 3000;
-	int	i = 1;
-	int	nfds = 1;
-	int	ret_nfds;
-
-	_poll_s[0].fd = _socket_fd;
-	_poll_s[0].events = POLLIN;
-
-	std::cout << "Server started\n";
-	int i_loop = 0;
-	++i_loop;
-	std::cout << i_loop << "----------\n";
-	std::cout << "loop poll()\n";
-	while ( (ret_nfds = ::poll(_poll_s, nfds, timeout)) != -1)
+	std::cout << "accept()\n";
+	addr_len = sizeof addr;
+	accepted_fd = ::accept(fd, (sockaddr*)&addr, &addr_len);
+	if (accepted_fd == -1)
 	{
-		i = 1;
-		while (i < nfds)
-		{
-			if (ret_nfds == 0)
-			{
-				std::cout << "timeout\n";
-				::close(_poll_s[i].fd);
-				_poll_s[i].fd = -1;
-				--nfds;
-			}
-
-			else if (_poll_s[i].revents & POLLIN) // READ
-			{
-				std::cout << "recv()\n";
-				ret = ::recv(_poll_s[i].fd, buf, BUFSIZE, 0);
-				if (ret == -1)
-				{
-					std::perror("err recv(): ");
-					if (::send(_poll_s[i].fd, MSG_BOUNCE, sizeof MSG_BOUNCE - 1, 0) == -1)
-						std::perror("err send(): ");
-
-					::close(_poll_s[i].fd);
-					_poll_s[i].fd = -1;
-					--nfds;
-					continue;
-				}
-				/*
-				if (ret == BUFSIZE)
-					// send error like "request too long" to client
-				*/
-				buf[ret] = '\0';
-				_poll_s[i].events = POLLOUT;
-			}
-			
-			else if (_poll_s[i].revents & POLLOUT) // WRITE
-			{
-				std::cout << "send()\n";
-				if (::send(_poll_s[i].fd, buf, ret, 0) == -1) // echo the read
-					std::perror("err send(): ");
-				if (::send(_poll_s[i].fd, MSG_TEST, sizeof MSG_TEST - 1, 0) == -1)
-					std::perror("err send(): ");
-
-				::close(_poll_s[i].fd);
-				_poll_s[i].fd = -1;
-				--nfds;
-			}
-
-			++i;
-		}
-
-
-		if ((_poll_s[0].fd == _socket_fd) && (_poll_s[0].revents & POLLIN)) // ACCEPT
-		{
-			std::cout << "accept()\n";
-			addr_len = sizeof addr;
-			accepted_fd = ::accept(_socket_fd, (sockaddr*)&addr, &addr_len);
-			if (accepted_fd == -1)
-			{
-				std::perror("err accept(): ");
-				continue;		
-			}
-			::fcntl(accepted_fd, F_SETFL, O_NONBLOCK);
-			_poll_s[nfds].fd = accepted_fd;
-			_poll_s[nfds].events = POLLIN;
-			++nfds;
-		}
-
-		++i_loop;
-		std::cout << i_loop << "----------\n";
-		std::cout << "loop poll()\n";
+		std::perror("err accept(): ");
+		return;
 	}
+	::fcntl(accepted_fd, F_SETFL, O_NONBLOCK);
+
+	struct epoll_event	ev;
+	std::memset(&ev, 0, sizeof ev);
+	ev.events = EPOLLIN;
+	ev.data.fd = accepted_fd;
+	if (::epoll_ctl(_epfd, EPOLL_CTL_ADD, accepted_fd, &ev) == -1)
+		std::perror("err accept() epoll_ctl(): ");
+}
+
+void	Webserv::_read_request(int fd)
+{
+	std::cout << "recv()\n";
+	_read_ret = ::recv(fd, _buf, BUFSIZE, 0);
+	if (_read_ret == -1)
+	{
+		std::perror("err recv(): ");
+		if (::send(fd, MSG_BOUNCE, sizeof MSG_BOUNCE - 1, 0) == -1)
+			std::perror("err send(): ");
+
+		::close(fd);
+		return;
+	}
+	/*
+	if (_read_ret == BUFSIZE)
+		// send error like "request too long" to client
+	*/
+	_buf[_read_ret] = '\0';
+
+	struct epoll_event	ev;
+	std::memset(&ev, 0, sizeof ev);
+	ev.events = EPOLLOUT;
+	ev.data.fd = fd;
+	if (::epoll_ctl(_epfd, EPOLL_CTL_MOD, fd, &ev) == -1)
+		std::perror("err accept() epoll_ctl(): ");
+}
+
+void	Webserv::_send_response(int fd)
+{
+	std::cout << "send()\n";
+	if (::send(fd, _buf, _read_ret, 0) == -1)
+		std::perror("err send(): ");
+	if (::send(fd, MSG_TEST, sizeof MSG_TEST - 1, 0) == -1)
+		std::perror("err send(): ");
+	::close(fd);
 }