Move to new Build System

MailServer/Server.cpp

@@ -0,0 +1,81 @@
+#include "Server.h"
+#include <Logging.h>
+#include <functional>
+#include <sstream>
+#include <sys/wait.h>
+
+
+namespace MailServer {
+
+Server::Server(asio::io_service& ioService, unsigned short port, const std::string& path, const std::string& url) :
+	m_ioService(ioService),
+	m_signal(ioService, SIGCHLD),
+	m_acceptor(ioService, asio::ip::tcp::endpoint(asio::ip::tcp::v4(), port)),
+	m_socket(ioService)
+{
+	m_context.path = path;
+	m_context.url = url;
+
+	StartSignalWait();
+	StartAccept();
+}
+
+void Server::StartSignalWait()
+{
+	m_signal.async_wait(std::bind(&Server::HandleSignalWait, this));
+}
+
+void Server::HandleSignalWait()
+{
+	// Only the parent process should check for this signal. We can determine
+	// whether we are in the parent by checking if the acceptor is still open.
+	if (m_acceptor.is_open())
+	{
+		// Reap completed child processes so that we don't end up with zombies.
+		int status = 0;
+		while (waitpid(-1, &status, WNOHANG) > 0) {}
+
+		StartSignalWait();
+	}
+}
+
+void Server::StartAccept()
+{
+	m_connection.reset(new Connection(m_ioService, m_context));
+	m_acceptor.async_accept(m_connection->Socket(), std::bind(&Server::HandleAccept, this, std::placeholders::_1));
+}
+
+void Server::HandleAccept(const asio::error_code& ec)
+{
+	if (!ec)
+	{
+		m_ioService.notify_fork(asio::io_service::fork_prepare);
+
+		if (fork() == 0)
+		{
+			m_ioService.notify_fork(asio::io_service::fork_child);
+			m_acceptor.close();
+			m_signal.cancel();
+
+			m_connection->Start();
+		}
+		else
+		{
+			m_ioService.notify_fork(asio::io_service::fork_parent);
+
+			m_connection.reset();
+			m_socket.close();
+			StartAccept();
+		}
+	}
+	else
+	{
+		std::stringstream ss;
+		ss << "Server::HandleAccept() - Error" << std::endl;
+		Logging::Log(Logging::Severity::Error, ss.str());
+
+		StartAccept();
+	}
+}
+
+} // namespace MailServer