#include <openssl/x509v3.h>
#include <sys/stat.h>
#include <unistd.h>
#include <string.h>
#include <strings.h>
#include <fcntl.h>
#include "tls.h"
#include "ssl_timeoutio.h"
#include "qremote.h"
#include "control.h"
#include "netio.h"
#include "sstring.h"
#include "starttlsr.h"

const char *ssl_err_str = NULL;

#warning FIXME: ugly
void __attribute__ ((noreturn))
TLS_QUIT(void)
{
	char *msg = ssl ? "; connected to " : "; connecting to ";

	write(1, msg, strlen(msg));
	write(1, rhost, rhostlen);
	write(1, ".\n", 2);
	quit();
}


void  __attribute__ ((noreturn))
tls_quit(const char *s1, const char *s2)
{
	write(1, s1, strlen(s1));
	if (s2) {
		write(1, ": ", 2);
		write(1, s2, strlen(s2));
	}
	TLS_QUIT();
}
#define tls_quit_error(s) tls_quit(s, ssl_error())

int match_partner(const char *s, size_t len)
{
	if (!strncasecmp(partner_fqdn, s, len) && !partner_fqdn[len])
		return 1;
	/* we also match if the name is *.domainname */
	if (*s == '*') {
		const char *domain = strchr(partner_fqdn, '.');

		if (!domain)
			return 0;
		if (!strncasecmp(domain, ++s, --len) && !domain[len])
			return 1;
	}
	return 0;
}

int tls_init(void)
{
	int i = 0;
	SSL *myssl;
	SSL_CTX *ctx;
	char *saciphbuf, **saciphers, *servercert = NULL;
	const char *ciphers;
	size_t fqlen;

	if (partner_fqdn) {
		char *tmp;
		struct stat st;

		fqlen = strlen(partner_fqdn);
		tmp = malloc(fqlen + 22);
		if (!tmp)
			err_mem(1);
		memcpy(tmp, "control/tlshosts/", 17);
		memcpy(tmp + 17, partner_fqdn, fqlen);
		memcpy(tmp + 17 + fqlen, ".pem", 5);
		if (stat(tmp, &st)) {
			free(tmp);
		} else {
			servercert = tmp;
		}
	}

	SSL_library_init();
	ctx = SSL_CTX_new(SSLv23_client_method());
	if (!ctx) {
		if (!servercert)
			return 0;
		tls_quit_error("ZTLS error initializing ctx");
	}

	if (servercert) {
		if (!SSL_CTX_load_verify_locations(ctx, servercert, NULL)) {
			SSL_CTX_free(ctx);
			write(1, "ZTLS unable to load ", 20);
			tls_quit_error(servercert);
		}
		/* set the callback here; SSL_set_verify didn't work before 0.9.6c */
		SSL_CTX_set_verify(ctx, SSL_VERIFY_NONE, NULL);
	}

	/* let the other side complain if it needs a cert and we don't have one */
	if (SSL_CTX_use_certificate_chain_file(ctx, "control/clientcert.pem"))
		SSL_CTX_use_RSAPrivateKey_file(ctx, "control/clientcert.pem", SSL_FILETYPE_PEM);

	myssl = SSL_new(ctx);
	SSL_CTX_free(ctx);
	if (!myssl) {
		if (!servercert)
			return 0;
		tls_quit_error("ZTLS error initializing ssl");
	}

	netwrite("STARTTLS\r\n");

	/* while the server is preparing a responce, do something else */
	if (loadlistfd(open("control/tlsclientciphers", O_RDONLY), &saciphbuf, &saciphers, NULL) == -1) {
		SSL_free(myssl);
		err_conf("can't open tlsclientciphers");
	}
	if (saciphers) {
		while (saciphers[i + 1]) {
			saciphers[i][strlen(saciphers[i])] = ':';
			i++;
		}
		ciphers = saciphbuf;
		free(saciphers);
	} else {
		 ciphers = "DEFAULT";
	}
	SSL_set_cipher_list(myssl, ciphers);
	free(saciphbuf);

	/* SSL_set_options(myssl, SSL_OP_NO_TLSv1); */
	SSL_set_fd(myssl, socketd);

	/* read the responce to STARTTLS */
	if (netget() != 220) {
		SSL_free(myssl);
		if (!servercert)
			return 0;
		write(1, "ZSTARTTLS rejected while ", 25);
		write(1, servercert, strlen(servercert));
		write(1, " exists", 7);
		TLS_QUIT();
	}

	ssl = myssl;
	ssl_rfd = ssl_wfd = socketd;
	if (ssl_timeoutconn(timeout) <= 0) {
		tls_quit("ZTLS connect failed", ssl_strerror());
	}

	if (servercert) {
		X509 *peercert;
		STACK_OF(GENERAL_NAME) *gens;
		long r = SSL_get_verify_result(ssl);

		if (r != X509_V_OK) {
			write(1, "ZTLS unable to verify server with ", 34);
			tls_quit(servercert, X509_verify_cert_error_string(r));
		}
		free(servercert);
		
		peercert = SSL_get_peer_certificate(ssl);
		if (!peercert) {
			write(1, "ZTLS unable to verify server ", 29);
			tls_quit(partner_fqdn, "no certificate provided");
		}

		/* RFC 2595 section 2.4: find a matching name
		 * first find a match among alternative names */
		gens = X509_get_ext_d2i(peercert, NID_subject_alt_name, 0, 0);
		if (gens) {
			for (i = 0, r = sk_GENERAL_NAME_num(gens); i < r; ++i) {
				const GENERAL_NAME *gn = sk_GENERAL_NAME_value(gens, i);

				if (gn->type == GEN_DNS)
				if (match_partner(gn->d.ia5->data, gn->d.ia5->length)) break;
			}
			sk_GENERAL_NAME_free(gens);
		}

		/* no alternative name matched, look up commonName */
		if (!gens || i >= r) {
			string peer;

			STREMPTY(peer);
			X509_NAME *subj = X509_get_subject_name(peercert);
			i = X509_NAME_get_index_by_NID(subj, NID_commonName, -1);
			if (i >= 0) {
				const ASN1_STRING *s = X509_NAME_get_entry(subj, i)->value;

				if (s) {
					peer.len = s->length > 0 ? s->length : 0;
					peer.s = s->data;
				}
			}
			if (!peer.len) {
				write(1, "ZTLS unable to verify server ", 29);
				tls_quit(partner_fqdn, "certificate contains no valid commonName");
			}
			if (!match_partner(peer.s, peer.len)) {
				char buf[64];
				int idx = 0;
				size_t j;

				write(1, "ZTLS unable to verify server ", 29);
				write(1, partner_fqdn, fqlen);
				write(1, ": received certificate for ", 27);
				for (j = 0; j < peer.len; ++j) {
					if ( (peer.s[j] < 33) || (peer.s[j] > 126) ) {
						buf[idx++] = '?';
					} else {
						buf[idx++] = peer.s[j];
					}
					if (idx == 64) {
						write(1, buf, 64);
						idx = 0;
					}
				}
				write(1, buf, idx);
				TLS_QUIT();
			}
		}

		X509_free(peercert);
	}

	return 1;
}