baked hugo version of webserv

srcs/bak__Webserv.cpp

@@ -0,0 +1,163 @@
+
+#include "bak__Webserv.hpp"
+
+Webserv::Webserv()
+{
+	int	on = 1;
+
+	std::cout << "Server init\n";
+
+	// create socket descriptor
+	_socket_fd = ::socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
+	if (_socket_fd == -1)
+	{
+		::perror("err socket()");
+		throw std::runtime_error("Socket init");
+	}
+
+	// allow socket descriptor to be reuseable
+	if (setsockopt(_socket_fd, SOL_SOCKET,  SO_REUSEADDR, (char *)&on, sizeof(on)) < 0)
+	{
+		::perror("setsockopt() failed");
+		throw std::runtime_error("Socket init");
+	}
+
+}
+
+Webserv::~Webserv()
+{
+	std::cout << "Server destroyed\n";
+}
+
+void	Webserv::bind(in_port_t port)
+{
+	struct sockaddr_in		addr;
+
+	addr.sin_family			= AF_INET;
+	addr.sin_port			= ::htons(port);
+	addr.sin_addr.s_addr	= ::htonl(INADDR_ANY);
+	if (::bind(_socket_fd, (const sockaddr*)&addr, sizeof addr) == -1)
+	{
+		::perror("err bind()");
+		throw std::runtime_error("Socket bind");		
+	}
+}
+
+void	Webserv::listen(unsigned int max_connections)
+{
+	if (::listen(_socket_fd, max_connections) == -1)
+	{
+		::perror("err listen(): ");
+		throw std::runtime_error("Socket listen");
+	}
+}
+
+void	Webserv::_add_fd(int sd, short event)
+{
+	struct pollfd	new_poll_fd;
+
+	new_poll_fd.fd = sd;
+	new_poll_fd.events = event;
+	new_poll_fd.revents = 0;
+	_fds.push_back(new_poll_fd);
+}
+
+void	Webserv::_close_fd(int i)
+{
+	close(_fds[i].fd);
+	_fds[i].fd = -1;
+	_fds.erase(_fds.begin() + i);
+}
+
+void	Webserv::_accept_connection()
+{
+	int	new_sd = 0;
+
+	printf("  Listening socket is readable\n");
+	while (new_sd != -1)
+	{
+		new_sd = accept(_socket_fd, NULL, NULL);
+		if (new_sd < 0)
+		{
+			if (errno != EWOULDBLOCK && errno != EAGAIN)
+			{
+				::perror("err accept()");
+				throw std::runtime_error("Socket accept");
+			}
+			break;
+		}
+		printf("  New incoming connection - %d\n", new_sd);
+		_add_fd(new_sd, POLLIN);
+	}
+}
+
+void	Webserv::_connect_socket(int i, int bufsize)
+{
+	int					len;
+	int					ret;
+	std::vector<char>	buffer(bufsize);
+
+	printf("  Descriptor %d is readable\n", _fds[i].fd);
+	while (TRUE)
+	{
+		ret = recv(_fds[i].fd, &buffer[0], buffer.size(), 0);
+		if (ret < 0)
+		{
+			if (errno != EWOULDBLOCK)
+			{
+				::perror("  recv() failed");
+				_close_fd(i);
+			}
+			break;
+		}
+
+		if (ret == 0)
+		{
+			printf("  Connection closed\n");
+			_close_fd(i);
+			break;
+		}
+
+		len = ret;
+		printf("  %d bytes received\n", len);
+
+		ret = send(_fds[i].fd, &buffer[0], len, 0);
+		if (ret < 0)
+		{
+			::perror("  send() failed");
+			_close_fd(i);
+			break;
+		}
+	}
+}
+
+void	Webserv::start(int timeout, int bufsize)
+{
+	_add_fd(_socket_fd, POLLIN);
+
+	std::cout << "Server started\n";
+	while (TRUE)
+	{
+		printf("-----------\npoll\n");
+		poll(&_fds[0], _fds.size(), timeout);
+
+		for (unsigned int i = 0; i < _fds.size(); i++)
+		{
+			if(_fds[i].revents == 0)
+				continue;
+
+			if(_fds[i].revents != POLLIN)
+			{
+				::perror("err revents");
+				throw std::runtime_error("Socket poll");
+			}
+
+			if (_fds[i].fd == _socket_fd)
+				_accept_connection();
+
+			else
+				_connect_socket(i, bufsize);
+		}
+	}
+}
+