#include "HTTPSServer.hpp"

namespace httpsserver {


HTTPSServer::HTTPSServer(SSLCert * cert, const uint16_t port, const uint8_t maxConnections, const in_addr_t bindAddress):
	HTTPServer(port, maxConnections, bindAddress),
	_cert(cert) {

	// Configure runtime data
	_sslctx = NULL;
}

HTTPSServer::~HTTPSServer() {

}

/**
 * This method starts the server and begins to listen on the port
 */
uint8_t HTTPSServer::setupSocket() {
	if (!isRunning()) {
		if (!setupSSLCTX()) {
			Serial.println("setupSSLCTX failed");
			return 0;
		}

		if (!setupCert()) {
			Serial.println("setupCert failed");
			SSL_CTX_free(_sslctx);
			_sslctx = NULL;
			return 0;
		}

		if (HTTPServer::setupSocket()) {
			return 1;
		} else {
			Serial.println("setupSockets failed");
			SSL_CTX_free(_sslctx);
			_sslctx = NULL;
			return 0;
		}
	} else {
		return 1;
	}
}

void HTTPSServer::teardownSocket() {

	HTTPServer::teardownSocket();

	// Tear down the SSL context
	SSL_CTX_free(_sslctx);
	_sslctx = NULL;
}

int HTTPSServer::createConnection(int idx) {
	HTTPSConnection * newConnection = new HTTPSConnection(this);
	_connections[idx] = newConnection;
	return newConnection->initialize(_socket, _sslctx, &_defaultHeaders);
}

/**
 * This method configures the ssl context that is used for the server
 */
uint8_t HTTPSServer::setupSSLCTX() {
	_sslctx = SSL_CTX_new(TLSv1_2_server_method());
	if (_sslctx) {
		// Set SSL Timeout to 5 minutes
		SSL_CTX_set_timeout(_sslctx, 300);
		return 1;
	} else {
		_sslctx = NULL;
		return 0;
	}
}

/**
 * This method configures the certificate and private key for the given
 * ssl context
 */
uint8_t HTTPSServer::setupCert() {
	// Configure the certificate first
	uint8_t ret = SSL_CTX_use_certificate_ASN1(
		_sslctx,
		_cert->getCertLength(),
		_cert->getCertData()
	);

	// Then set the private key accordingly
	if (ret) {
		ret = SSL_CTX_use_RSAPrivateKey_ASN1(
			_sslctx,
			_cert->getPKData(),
			_cert->getPKLength()
		);
	}

	return ret;
}

} /* namespace httpsserver */