#include "server.hpp"

using namespace boost::asio::ip;
using namespace boost::asio;

Server::Server(io_service& io_service, const ListenSettings& listen_settings)
: v4_acceptor(io_service, address_v4::from_string(listen_settings.ipv4_inaddr), listen_settings.port)
, v6_acceptor(io_service, address_v6::from_string(listen_settings.ipv6_inaddr), listen_settings.port)
{
	/*
	 * We can't bind both a v4 and v6 socket to both inaddr4_any and inaddr6_any.
	 * So in case both ipv4 and ipv6 are enabled and both want to bind to their 
	 * respective inaddr_any we need to bind to v6 any and disable ipv6 only on
	 * the acceptor socket, else we can just bind to the interfaces normally.
	 */
	bool bind_v4_any = v4_acceptor.get_address().to_v4() == address_v4::any() && listen_settings.enable_ipv4;
	bool bind_v6_any = v6_acceptor.get_address().to_v6() == address_v6::any() && listen_settings.enable_ipv6;
	
	if(bind_v4_any && bind_v6_any) {
		v6_acceptor.listen_v6_and_v4_any();
	} else if((bind_v4_any ^ bind_v6_any) && listen_settings.enable_ipv4 && listen_settings.enable_ipv6) {
		throw std::runtime_error("Cannot bind an INADDR_ANY and a non INADDR_ANY address on ipv4 and ipv6");
	} else {
		if(listen_settings.enable_ipv4) {
			v4_acceptor.listen_socket();
		}
		if(listen_settings.enable_ipv6) {
			v6_acceptor.listen_socket();
		}
	}
}

Server::Server(io_service& io_service, const ListenSettings& listen_settings, AcceptHandler accept_handler)
: Server(io_service, listen_settings)
{
	if(v4_acceptor.is_open()) {
		v4_acceptor.start_accepting(accept_handler);
	}
	if(v6_acceptor.is_open()) {
		v6_acceptor.start_accepting(accept_handler);
	}
}

Server::Server(io_service& io_service, const ListenSettings& listen_settings, std::shared_ptr<ssl::context> ctx, SSLAcceptHandler accept_handler)
: Server(io_service, listen_settings)
{
	ctx->set_options(
		  boost::asio::ssl::context::default_workarounds
		| boost::asio::ssl::context::no_sslv2
		| boost::asio::ssl::context::single_dh_use
	);

	ctx->use_certificate_chain_file(listen_settings.cert);
	ctx->use_private_key_file(listen_settings.key, boost::asio::ssl::context::pem);
	ctx->use_tmp_dh_file(listen_settings.dhparam);
	ctx->set_default_verify_paths();

	if(v4_acceptor.is_open()) {
		v4_acceptor.start_accepting(ctx, accept_handler);
	}
	if(v6_acceptor.is_open()) {
		v6_acceptor.start_accepting(ctx, accept_handler);
	}
}

void Server::close()
{
	if(v4_acceptor.is_open()) {
		v4_acceptor.close();
	}
	if(v6_acceptor.is_open()) {
		v6_acceptor.close();
	}
}