/*
 * uhub - A tiny ADC p2p connection hub
 * Copyright (C) 2007-2012, Jan Vidar Krey
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "uhub.h"
#include "network/common.h"
#include "network/tls.h"

#ifdef SSL_SUPPORT
#ifdef SSL_USE_OPENSSL

#define NETWORK_DUMP_DEBUG

struct net_ssl_openssl
{
	SSL* ssl;
	enum ssl_state state;
	uint32_t flags;
	BIO* biow;
	BIO* bior;
};

struct net_context_openssl
{
	SSL_METHOD* ssl_method;
	SSL_CTX* ssl_ctx;
};

static struct net_ssl_openssl* get_handle(struct net_connection* con)
{
	uhub_assert(con);
	return (struct net_ssl_openssl*) con->ssl;
}

const char* net_ssl_get_provider()
{
	return OPENSSL_VERSION_TEXT;
}


/**
 * Create a new SSL context.
 */
struct ssl_context_handle* net_ssl_context_create()
{

	struct net_context_openssl* ctx = (struct net_context_openssl*) hub_malloc_zero(sizeof(struct net_context_openssl));
	ctx->ssl_method = (SSL_METHOD*) SSLv23_method(); /* TLSv1_method() */
	ctx->ssl_ctx = SSL_CTX_new(ctx->ssl_method);

	/* Disable SSLv2 */
	SSL_CTX_set_options(ctx->ssl_ctx, SSL_OP_NO_SSLv2);
	SSL_CTX_set_quiet_shutdown(ctx->ssl_ctx, 1);

	return (struct ssl_context_handle*) ctx;
}

extern void net_ssl_context_destroy(struct ssl_context_handle* ctx_)
{
	struct net_context_openssl* ctx = (struct net_context_openssl*) ctx_;
	SSL_CTX_free(ctx->ssl_ctx);
	hub_free(ctx);
}

int ssl_load_certificate(struct ssl_context_handle* ctx_, const char* pem_file)
{
	struct net_context_openssl* ctx = (struct net_context_openssl*) ctx_;
	if (SSL_CTX_use_certificate_file(ctx->ssl_ctx, pem_file, SSL_FILETYPE_PEM) < 0)
	{
		LOG_ERROR("SSL_CTX_use_certificate_file: %s", ERR_error_string(ERR_get_error(), NULL));
		return 0;
	}

	return 1;
}

int ssl_load_private_key(struct ssl_context_handle* ctx_, const char* pem_file)
{
	struct net_context_openssl* ctx = (struct net_context_openssl*) ctx_;
	if (SSL_CTX_use_PrivateKey_file(ctx->ssl_ctx, pem_file, SSL_FILETYPE_PEM) < 0)
	{
		LOG_ERROR("SSL_CTX_use_PrivateKey_file: %s", ERR_error_string(ERR_get_error(), NULL));
		return 0;
	}
	return 1;
}

int ssl_check_private_key(struct ssl_context_handle* ctx_)
{
	struct net_context_openssl* ctx = (struct net_context_openssl*) ctx_;
	if (SSL_CTX_check_private_key(ctx->ssl_ctx) != 1)
	{
		LOG_FATAL("SSL_CTX_check_private_key: Private key does not match the certificate public key: %s", ERR_error_string(ERR_get_error(), NULL));
		return 0;
	}
	return 1;
}

ssize_t net_con_ssl_accept(struct net_connection* con)
{
	struct net_ssl_openssl* handle = get_handle(con);
	handle->state = tls_st_accepting;
	ssize_t ret;

	ret = SSL_accept(handle->ssl);
	LOG_PROTO("SSL_accept() ret=%d", ret);
	if (ret > 0)
	{
		net_con_update(con, NET_EVENT_READ);
		handle->state = tls_st_connected;
	}
	else
	{
		int err = SSL_get_error(handle->ssl, ret);
		switch (err)
		{
			case SSL_ERROR_ZERO_RETURN:
				// Not really an error, but SSL was shut down.
				return -1;

			case SSL_ERROR_WANT_READ:
				net_con_update(con, NET_EVENT_READ);
				return 0;

			case SSL_ERROR_WANT_WRITE:
				net_con_update(con, NET_EVENT_WRITE);
				return 0;

			case SSL_ERROR_SYSCALL:
				return -2;
		}
	}
	return ret;
}

ssize_t net_con_ssl_connect(struct net_connection* con)
{
	struct net_ssl_openssl* handle = get_handle(con);
	ssize_t ret;
	handle->state = tls_st_connecting;

	ret = SSL_connect(handle->ssl);
#ifdef NETWORK_DUMP_DEBUG
	LOG_PROTO("SSL_connect() ret=%d", ret);
#endif /* NETWORK_DUMP_DEBUG */

	if (ret > 0)
	{
		handle->state = tls_st_connected;
		net_con_update(con, NET_EVENT_READ);
	}
	else
	{
		int err = SSL_get_error(handle->ssl, ret);
		switch (err)
		{
			case SSL_ERROR_ZERO_RETURN:
				// Not really an error, but SSL was shut down.
				return -1;

			case SSL_ERROR_WANT_READ:
				net_con_update(con, NET_EVENT_READ);
				return 0;

			case SSL_ERROR_WANT_WRITE:
				net_con_update(con, NET_EVENT_WRITE);
				return 0;

			case SSL_ERROR_SYSCALL:
				return -2;
		}
	}
	return ret;
}

ssize_t net_con_ssl_handshake(struct net_connection* con, enum net_con_ssl_mode ssl_mode, struct ssl_context_handle* ssl_ctx)
{
	uhub_assert(con);

	struct net_context_openssl* ctx = (struct net_context_openssl*) ssl_ctx;
	struct net_ssl_openssl* handle = (struct net_ssl_openssl*) hub_malloc_zero(sizeof(struct net_ssl_openssl));

	if (ssl_mode == net_con_ssl_mode_server)
	{
		handle->ssl = SSL_new(ctx->ssl_ctx);
		if (!handle->ssl)
		{
			LOG_ERROR("Unable to create new SSL stream\n");
			return -1;
		}
		SSL_set_fd(handle->ssl, con->sd);
		handle->bior = SSL_get_rbio(handle->ssl);
		handle->biow = SSL_get_wbio(handle->ssl);
		con->ssl = (struct ssl_handle*) handle;
		return net_con_ssl_accept(con);
	}
	else
	{
		handle->ssl = SSL_new(SSL_CTX_new(TLSv1_method()));
		SSL_set_fd(handle->ssl, con->sd);
		handle->bior = SSL_get_rbio(handle->ssl);
		handle->biow = SSL_get_wbio(handle->ssl);
		con->ssl = (struct ssl_handle*) handle;
		return net_con_ssl_connect(con);
	}

}

ssize_t net_ssl_send(struct net_connection* con, const void* buf, size_t len)
{
	struct net_ssl_openssl* handle = get_handle(con);

	uhub_assert(handle->state == tls_st_connected || handle->state == tls_st_need_write);

	ERR_clear_error();
	ssize_t ret = SSL_write(handle->ssl, buf, len);
	LOG_PROTO("SSL_write(con=%p, buf=%p, len=" PRINTF_SIZE_T ") => %d", con, buf, len, ret);
	if (ret > 0)
	{
		handle->state = tls_st_connected;
		return ret;
	}
	else if (ret <= 0)
	{
		int err = SSL_get_error(handle->ssl, ret);
		switch (err)
		{
			case SSL_ERROR_ZERO_RETURN:
				// Not really an error, but SSL was shut down.
				return -1;

			case SSL_ERROR_WANT_READ:
				handle->state = tls_st_need_write;
				net_con_update(con, NET_EVENT_READ);
				return 0;

			case SSL_ERROR_WANT_WRITE:
				handle->state = tls_st_need_write;
				net_con_update(con, NET_EVENT_WRITE);
				return 0;

			case SSL_ERROR_SYSCALL:
				return -2;
		}
	}
}

ssize_t net_ssl_recv(struct net_connection* con, void* buf, size_t len)
{
	struct net_ssl_openssl* handle = get_handle(con);
	ssize_t ret;

	if (handle->state == tls_st_error)
		return -2;

	uhub_assert(handle->state == tls_st_connected || handle->state == tls_st_need_read);

	ERR_clear_error();

	ret = SSL_read(handle->ssl, buf, len);
	LOG_PROTO("SSL_read(con=%p, buf=%p, len=" PRINTF_SIZE_T ") => %d", con, buf, len, ret);
	if (ret > 0)
	{
		handle->state = tls_st_connected;
		return ret;
	}
	else if (ret <= 0)
	{
		int err = SSL_get_error(handle->ssl, ret);
		switch (err)
		{
			case SSL_ERROR_ZERO_RETURN:
				// Not really an error, but SSL was shut down.
				return -1;

			case SSL_ERROR_WANT_READ:
				handle->state = tls_st_need_read;
				net_con_update(con, NET_EVENT_READ);
				return 0;

			case SSL_ERROR_WANT_WRITE:
				handle->state = tls_st_need_read;
				net_con_update(con, NET_EVENT_WRITE);
				return 0;

			case SSL_ERROR_SYSCALL:
				return -2;
		}
	}
}

void net_ssl_shutdown(struct net_connection* con)
{
	struct net_ssl_openssl* handle = get_handle(con);
	SSL_shutdown(handle->ssl);
	SSL_clear(handle->ssl);
}

void net_ssl_destroy(struct net_connection* con)
{
	struct net_ssl_openssl* handle = get_handle(con);
	SSL_free(handle->ssl);
}

void net_ssl_callback(struct net_connection* con, int events)
{
	struct net_ssl_openssl* handle = get_handle(con);

	switch (handle->state)
	{
		case tls_st_none:
			con->callback(con, events, con->ptr);
			break;

		case tls_st_error:
			con->callback(con, NET_EVENT_READ, con->ptr);
			break;

		case tls_st_accepting:
			if (net_con_ssl_accept(con) != 0)
				con->callback(con, NET_EVENT_READ, con->ptr);
			break;

		case tls_st_connecting:
			if (net_con_ssl_connect(con) != 0)
				con->callback(con, NET_EVENT_READ, con->ptr);
			break;

		case tls_st_need_read:
			con->callback(con, NET_EVENT_READ, con->ptr);
			break;

		case tls_st_need_write:
			con->callback(con, NET_EVENT_WRITE, con->ptr);
			break;

		case tls_st_connected:
			con->callback(con, events, con->ptr);
			break;

		case tls_st_disconnecting:
			return;
	}
}


#endif /* SSL_USE_OPENSSL */
#endif /* SSL_SUPPORT */