/*
 * Copyright (C) 2000-2012 Free Software Foundation, Inc.
 * Author: Nikos Mavrogiannopoulos
 *
 * This file is part of GnuTLS.
 *
 * GnuTLS 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.
 *
 * GnuTLS 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 <https://www.gnu.org/licenses/>.
 */

#include <config.h>

/* Work around problem reported in
   <https://permalink.gmane.org/gmane.comp.lib.gnulib.bugs/15755>.*/
#if GETTIMEOFDAY_CLOBBERS_LOCALTIME
#undef localtime
#endif

#include <getpass.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <gnutls/gnutls.h>
#include <gnutls/x509.h>
#include <gnutls/crypto.h>
#include <time.h>
#include <common.h>
#include <unistd.h>

#ifndef _WIN32
# include <signal.h>
#else
#include <ws2tcpip.h>
#endif

#ifdef ENABLE_PKCS11
#include <gnutls/pkcs11.h>
#endif

#define SU(x) (x!=NULL?x:"Unknown")

const char str_unknown[] = "(unknown)";

static FILE *logfile = NULL;
/* Hex encodes the given data adding a semicolon between hex bytes.
 */
const char *raw_to_string(const unsigned char *raw, size_t raw_size)
{
	static char buf[1024];
	size_t i;
	if (raw_size == 0)
		return "(empty)";

	if (raw_size * 3 + 1 >= sizeof(buf))
		return "(too large)";

	for (i = 0; i < raw_size; i++) {
		sprintf(&(buf[i * 3]), "%02X%s", raw[i],
			(i == raw_size - 1) ? "" : ":");
	}
	buf[sizeof(buf) - 1] = '\0';

	return buf;
}

/* Hex encodes the given data.
 */
const char *raw_to_hex(const unsigned char *raw, size_t raw_size)
{
	static char buf[1024];
	size_t i;
	if (raw_size == 0)
		return "(empty)";

	if (raw_size * 2 + 1 >= sizeof(buf))
		return "(too large)";

	for (i = 0; i < raw_size; i++) {
		sprintf(&(buf[i * 2]), "%02x", raw[i]);
	}
	buf[sizeof(buf) - 1] = '\0';

	return buf;
}

const char *raw_to_base64(const unsigned char *raw, size_t raw_size)
{
	static char buf[1024];
	gnutls_datum_t data = {(unsigned char*)raw, raw_size};
	size_t buf_size;
	int ret;

	if (raw_size == 0)
		return "(empty)";

	buf_size = sizeof(buf);
	ret = gnutls_pem_base64_encode(NULL, &data, buf, &buf_size);
	if (ret < 0)
		return "(error)";

	buf[sizeof(buf) - 1] = '\0';

	return buf;
}

static void
print_x509_info(gnutls_session_t session, FILE *out, int flag, int print_cert, int print_crt_status)
{
	gnutls_x509_crt_t crt;
	const gnutls_datum_t *cert_list;
	unsigned int cert_list_size = 0, j;
	int ret;

	cert_list = gnutls_certificate_get_peers(session, &cert_list_size);
	if (cert_list_size == 0) {
		if (print_crt_status)
			fprintf(stderr, "No certificates found!\n");
		return;
	}

	log_msg(out, "- Certificate type: X.509\n");
	log_msg(out, "- Got a certificate list of %d certificates.\n",
	       cert_list_size);

	for (j = 0; j < cert_list_size; j++) {
		gnutls_datum_t cinfo;

		ret = gnutls_x509_crt_init(&crt);
		if (ret < 0) {
			fprintf(stderr, "Memory error\n");
			return;
		}

		ret =
		    gnutls_x509_crt_import(crt, &cert_list[j],
					   GNUTLS_X509_FMT_DER);
		if (ret < 0) {
			fprintf(stderr, "Decoding error: %s\n",
				gnutls_strerror(ret));
			return;
		}

		log_msg(out, "- Certificate[%d] info:\n - ", j);
		if (flag == GNUTLS_CRT_PRINT_COMPACT && j > 0)
			flag = GNUTLS_CRT_PRINT_ONELINE;

		ret = gnutls_x509_crt_print(crt, flag, &cinfo);
		if (ret == 0) {
			log_msg(out, "%s\n", cinfo.data);
			gnutls_free(cinfo.data);
		}

		if (print_cert) {
			gnutls_datum_t pem;

			ret =
			    gnutls_x509_crt_export2(crt,
						   GNUTLS_X509_FMT_PEM, &pem);
			if (ret < 0) {
				fprintf(stderr, "Encoding error: %s\n",
					gnutls_strerror(ret));
				return;
			}

			log_msg(out, "\n%s\n", (char*)pem.data);

			gnutls_free(pem.data);
		}

		gnutls_x509_crt_deinit(crt);
	}
}

/* returns false (0) if not verified, or true (1) otherwise 
 */
int cert_verify(gnutls_session_t session, const char *hostname, const char *purpose)
{
	int rc;
	unsigned int status = 0;
	gnutls_datum_t out;
	int type;
	gnutls_typed_vdata_st data[2];
	unsigned elements = 0;

	memset(data, 0, sizeof(data));

	if (hostname) {
		data[elements].type = GNUTLS_DT_DNS_HOSTNAME;
		data[elements].data = (void*)hostname;
		elements++;
	}

	if (purpose) {
		data[elements].type = GNUTLS_DT_KEY_PURPOSE_OID;
		data[elements].data = (void*)purpose;
		elements++;
	}

	rc = gnutls_certificate_verify_peers(session, data, elements, &status);
	if (rc == GNUTLS_E_NO_CERTIFICATE_FOUND) {
		log_msg(stdout, "- Peer did not send any certificate.\n");
		return 0;
	}

	if (rc < 0) {
		log_msg(stdout, "- Could not verify certificate (err: %s)\n",
		       gnutls_strerror(rc));
		return 0;
	}

	type = gnutls_certificate_type_get(session);
	rc = gnutls_certificate_verification_status_print(status, type,
							  &out, 0);
	if (rc < 0) {
		log_msg(stdout, "- Could not print verification flags (err: %s)\n",
		       gnutls_strerror(rc));
		return 0;
	}

	log_msg(stdout, "- Status: %s\n", out.data);

	gnutls_free(out.data);

	if (status)
		return 0;

	return 1;
}

static void
print_dh_info(gnutls_session_t session, const char *str, int print)
{
#if defined(ENABLE_DHE) || defined(ENABLE_ANON)
	unsigned group;
	int ret;
	gnutls_datum_t raw_gen = { NULL, 0 };
	gnutls_datum_t raw_prime = { NULL, 0 };
	gnutls_dh_params_t dh_params = NULL;
	unsigned char *params_data = NULL;
	size_t params_data_size = 0;

	if (!print)
		return;

	group = gnutls_group_get(session);
	if (group != 0) {
		return;
	}

	log_msg(stdout, "- %sDiffie-Hellman parameters\n", str);
	log_msg(stdout, " - Using prime: %d bits\n",
	       gnutls_dh_get_prime_bits(session));
	log_msg(stdout, " - Secret key: %d bits\n",
	       gnutls_dh_get_secret_bits(session));
	log_msg(stdout, " - Peer's public key: %d bits\n",
	       gnutls_dh_get_peers_public_bits(session));

	ret = gnutls_dh_get_group(session, &raw_gen, &raw_prime);
	if (ret) {
		fprintf(stderr, "gnutls_dh_get_group %d\n", ret);
		goto out;
	}

	ret = gnutls_dh_params_init(&dh_params);
	if (ret) {
		fprintf(stderr, "gnutls_dh_params_init %d\n", ret);
		goto out;
	}

	ret =
	    gnutls_dh_params_import_raw(dh_params, &raw_prime,
						&raw_gen);
	if (ret) {
		fprintf(stderr, "gnutls_dh_params_import_raw %d\n",
			ret);
		goto out;
	}

	ret = gnutls_dh_params_export_pkcs3(dh_params,
					    GNUTLS_X509_FMT_PEM,
					    params_data,
					    &params_data_size);
	if (ret != GNUTLS_E_SHORT_MEMORY_BUFFER) {
		fprintf(stderr,
			"gnutls_dh_params_export_pkcs3 %d\n", ret);
		goto out;
	}

	params_data = gnutls_malloc(params_data_size);
	if (!params_data) {
		fprintf(stderr, "gnutls_malloc %d\n", ret);
		goto out;
	}

	ret = gnutls_dh_params_export_pkcs3(dh_params,
					    GNUTLS_X509_FMT_PEM,
					    params_data,
					    &params_data_size);
	if (ret) {
		fprintf(stderr,
			"gnutls_dh_params_export_pkcs3-2 %d\n",
			ret);
		goto out;
	}

	log_msg(stdout, " - PKCS#3 format:\n\n%.*s\n",
	       (int) params_data_size, params_data);

      out:
	gnutls_free(params_data);
	gnutls_free(raw_prime.data);
	gnutls_free(raw_gen.data);
	gnutls_dh_params_deinit(dh_params);
#endif
}

static void print_ecdh_info(gnutls_session_t session, const char *str, int print)
{
	int curve;

	if (!print)
		return;

	log_msg(stdout, "- %sEC Diffie-Hellman parameters\n", str);

	curve = gnutls_ecc_curve_get(session);

	log_msg(stdout, " - Using curve: %s\n", gnutls_ecc_curve_get_name(curve));
	log_msg(stdout, " - Curve size: %d bits\n",
	       gnutls_ecc_curve_get_size(curve) * 8);

}

int print_info(gnutls_session_t session, int verbose, int flags)
{
	const char *tmp;
	gnutls_credentials_type_t cred;
	gnutls_kx_algorithm_t kx;
	unsigned char session_id[33];
	size_t session_id_size = sizeof(session_id);
	gnutls_srtp_profile_t srtp_profile;
	gnutls_datum_t p;
	char *desc;
	gnutls_protocol_t version;
	int rc;

	desc = gnutls_session_get_desc(session);
	log_msg(stdout, "- Description: %s\n", desc);
	gnutls_free(desc);

	/* print session ID */
	gnutls_session_get_id(session, session_id, &session_id_size);
	if (session_id_size > 0) {
		log_msg(stdout, "- Session ID: %s\n",
		       raw_to_string(session_id, session_id_size));
	}

	/* print the key exchange's algorithm name
	 */
	kx = gnutls_kx_get(session);

	cred = gnutls_auth_get_type(session);
	switch (cred) {
#ifdef ENABLE_ANON
	case GNUTLS_CRD_ANON:
		if (kx == GNUTLS_KX_ANON_ECDH)
			print_ecdh_info(session, "Anonymous ", verbose);
		else
			print_dh_info(session, "Anonymous ", verbose);
		break;
#endif
#ifdef ENABLE_SRP
	case GNUTLS_CRD_SRP:
		/* This should be only called in server
		 * side.
		 */
		if (gnutls_srp_server_get_username(session) != NULL)
			log_msg(stdout, "- SRP authentication. Connected as '%s'\n",
			       gnutls_srp_server_get_username(session));
		break;
#endif
#ifdef ENABLE_PSK
	case GNUTLS_CRD_PSK:
		/* This returns NULL in server side.
		 */
		if (gnutls_psk_client_get_hint(session) != NULL)
			log_msg(stdout, "- PSK authentication. PSK hint '%s'\n",
			       gnutls_psk_client_get_hint(session));
		/* This returns NULL in client side.
		 */
		if (gnutls_psk_server_get_username(session) != NULL)
			log_msg(stdout, "- PSK authentication. Connected as '%s'\n",
			       gnutls_psk_server_get_username(session));
		if (kx == GNUTLS_KX_DHE_PSK)
			print_dh_info(session, "Ephemeral ", verbose);
		if (kx == GNUTLS_KX_ECDHE_PSK)
			print_ecdh_info(session, "Ephemeral ", verbose);
		break;
#endif
	case GNUTLS_CRD_IA:
		log_msg(stdout, "- TLS/IA authentication\n");
		break;
	case GNUTLS_CRD_CERTIFICATE:
		{
			char dns[256];
			size_t dns_size = sizeof(dns);
			unsigned int type;

			/* This fails in client side */
			if (gnutls_server_name_get
			    (session, dns, &dns_size, &type, 0) == 0) {
				log_msg(stdout, "- Given server name[%d]: %s\n",
				       type, dns);
			}
		}

		if ((flags & P_WAIT_FOR_CERT) && gnutls_certificate_get_ours(session) == 0)
			log_msg(stdout, "- No certificate was sent to peer\n");

		if (flags& P_PRINT_CERT)
			print_cert_info(session, verbose, (flags&P_PRINT_CERT));

		if (kx == GNUTLS_KX_DHE_RSA || kx == GNUTLS_KX_DHE_DSS)
			print_dh_info(session, "Ephemeral ", verbose);
		else if (kx == GNUTLS_KX_ECDHE_RSA
			 || kx == GNUTLS_KX_ECDHE_ECDSA)
			print_ecdh_info(session, "Ephemeral ", verbose);
	}


	if (verbose) {
		version = gnutls_protocol_get_version(session);
		tmp =
		    SU(gnutls_protocol_get_name(version));
		log_msg(stdout, "- Version: %s\n", tmp);

		if (version < GNUTLS_TLS1_3) {
			tmp = SU(gnutls_kx_get_name(kx));
			log_msg(stdout, "- Key Exchange: %s\n", tmp);
		}

		if (gnutls_sign_algorithm_get(session) != GNUTLS_SIGN_UNKNOWN) {
			tmp =
			    SU(gnutls_sign_get_name
			       (gnutls_sign_algorithm_get(session)));
			log_msg(stdout, "- Server Signature: %s\n", tmp);
		}

		if (gnutls_sign_algorithm_get_client(session) !=
		    GNUTLS_SIGN_UNKNOWN) {
			tmp =
			    SU(gnutls_sign_get_name
			       (gnutls_sign_algorithm_get_client(session)));
			log_msg(stdout, "- Client Signature: %s\n", tmp);
		}

		tmp = SU(gnutls_cipher_get_name(gnutls_cipher_get(session)));
		log_msg(stdout, "- Cipher: %s\n", tmp);

		tmp = SU(gnutls_mac_get_name(gnutls_mac_get(session)));
		log_msg(stdout, "- MAC: %s\n", tmp);
	}

	log_msg(stdout, "- Options:");
	if (gnutls_session_ext_master_secret_status(session)!=0)
		log_msg(stdout, " extended master secret,");
	if (gnutls_safe_renegotiation_status(session)!=0)
		log_msg(stdout, " safe renegotiation,");
	if (gnutls_session_etm_status(session)!=0)
		log_msg(stdout, " EtM,");
#ifdef ENABLE_OCSP
	if (gnutls_ocsp_status_request_is_checked(session, GNUTLS_OCSP_SR_IS_AVAIL)!=0) {
		log_msg(stdout, " OCSP status request%s,", gnutls_ocsp_status_request_is_checked(session,0)!=0?"":"[ignored]");
	}
#endif
	log_msg(stdout, "\n");

#ifdef ENABLE_DTLS_SRTP
	rc = gnutls_srtp_get_selected_profile(session, &srtp_profile);
	if (rc == 0)
		log_msg(stdout, "- SRTP profile: %s\n",
		       gnutls_srtp_get_profile_name(srtp_profile));
#endif

#ifdef ENABLE_ALPN
	rc = gnutls_alpn_get_selected_protocol(session, &p);
	if (rc == 0)
		log_msg(stdout, "- Application protocol: %.*s\n", p.size, p.data);
#endif

	if (verbose) {
		gnutls_datum_t cb;

		rc = gnutls_session_channel_binding(session,
						    GNUTLS_CB_TLS_UNIQUE,
						    &cb);
		if (rc)
			fprintf(stderr, "Channel binding error: %s\n",
				gnutls_strerror(rc));
		else {
			size_t i;

			log_msg(stdout, "- Channel binding 'tls-unique': ");
			for (i = 0; i < cb.size; i++)
				log_msg(stdout, "%02x", cb.data[i]);
			log_msg(stdout, "\n");
			gnutls_free(cb.data);
		}
	}

	fflush(stdout);

	return 0;
}

void print_cert_info(gnutls_session_t session, int verbose, int print_cert)
{
	print_cert_info2(session, verbose, stdout, print_cert);
}

void print_cert_info2(gnutls_session_t session, int verbose, FILE *out, int print_cert)
{
	int flag, print_crt_status = 0;

	if (verbose)
		flag = GNUTLS_CRT_PRINT_FULL;
	else
		flag = GNUTLS_CRT_PRINT_COMPACT;

	if (gnutls_certificate_client_get_request_status(session) != 0) {
		log_msg(stdout, "- Server has requested a certificate.\n");
		print_crt_status = 1;
	}

	switch (gnutls_certificate_type_get(session)) {
	case GNUTLS_CRT_X509:
		print_x509_info(session, out, flag, print_cert, print_crt_status);
		break;
	default:
		break;
	}
}

void print_list(const char *priorities, int verbose)
{
	size_t i;
	int ret;
	unsigned int idx;
	const char *name;
	const char *err;
	unsigned char id[2];
	gnutls_kx_algorithm_t kx;
	gnutls_cipher_algorithm_t cipher;
	gnutls_mac_algorithm_t mac;
	gnutls_protocol_t version;
	gnutls_priority_t pcache;
	const unsigned int *list;

	if (priorities != NULL) {
		log_msg(stdout, "Cipher suites for %s\n", priorities);

		ret = gnutls_priority_init(&pcache, priorities, &err);
		if (ret < 0) {
			if (ret == GNUTLS_E_INVALID_REQUEST)
				fprintf(stderr, "Syntax error at: %s\n", err);
			else
				fprintf(stderr, "Error in priorities: %s\n", gnutls_strerror(ret));
			exit(1);
		}

		for (i = 0;; i++) {
			ret =
			    gnutls_priority_get_cipher_suite_index(pcache,
								   i,
								   &idx);
			if (ret == GNUTLS_E_REQUESTED_DATA_NOT_AVAILABLE)
				break;
			if (ret == GNUTLS_E_UNKNOWN_CIPHER_SUITE)
				continue;

			name =
			    gnutls_cipher_suite_info(idx, id, NULL, NULL,
						     NULL, &version);

			if (name != NULL)
				log_msg(stdout, "%-50s\t0x%02x, 0x%02x\t%s\n",
				       name, (unsigned char) id[0],
				       (unsigned char) id[1],
				       gnutls_protocol_get_name(version));
		}

		log_msg(stdout, "\n");
#if 0
		{
			ret =
			    gnutls_priority_certificate_type_list2(pcache,
								  &list,
								  GNUTLS_CTYPE_CLIENT);

			log_msg(stdout, "Certificate types: ");
			if (ret == 0)
				log_msg(stdout, "none\n");
			for (i = 0; i < (unsigned) ret; i++) {
				log_msg(stdout, "CTYPE-%s",
				       gnutls_certificate_type_get_name
				       (list[i]));
				if (i + 1 != (unsigned) ret)
					log_msg(stdout, ", ");
				else
					log_msg(stdout, "\n");
			}
		}
#endif

		{
			ret = gnutls_priority_protocol_list(pcache, &list);

			log_msg(stdout, "Protocols: ");
			if (ret == 0)
				log_msg(stdout, "none\n");
			for (i = 0; i < (unsigned) ret; i++) {
				log_msg(stdout, "VERS-%s",
				       gnutls_protocol_get_name(list[i]));
				if (i + 1 != (unsigned) ret)
					log_msg(stdout, ", ");
				else
					log_msg(stdout, "\n");
			}
		}

		{
			ret = gnutls_priority_cipher_list(pcache, &list);

			log_msg(stdout, "Ciphers: ");
			if (ret == 0)
				log_msg(stdout, "none\n");
			for (i = 0; i < (unsigned) ret; i++) {
				log_msg(stdout, "%s",
				       gnutls_cipher_get_name(list[i]));
				if (i + 1 != (unsigned) ret)
					log_msg(stdout, ", ");
				else
					log_msg(stdout, "\n");
			}
		}

		{
			ret = gnutls_priority_mac_list(pcache, &list);

			log_msg(stdout, "MACs: ");
			if (ret == 0)
				log_msg(stdout, "none\n");
			for (i = 0; i < (unsigned) ret; i++) {
				log_msg(stdout, "%s",
				       gnutls_mac_get_name(list[i]));
				if (i + 1 != (unsigned) ret)
					log_msg(stdout, ", ");
				else
					log_msg(stdout, "\n");
			}
		}

		{
			ret = gnutls_priority_kx_list(pcache, &list);

			log_msg(stdout, "Key Exchange Algorithms: ");
			if (ret == 0)
				log_msg(stdout, "none\n");
			for (i = 0; i < (unsigned) ret; i++) {
				log_msg(stdout, "%s",
				       gnutls_kx_get_name(list[i]));
				if (i + 1 != (unsigned) ret)
					log_msg(stdout, ", ");
				else
					log_msg(stdout, "\n");
			}
		}

		{
			ret =
			    gnutls_priority_group_list(pcache, &list);

			log_msg(stdout, "Groups: ");
			if (ret == 0)
				log_msg(stdout, "none\n");
			for (i = 0; i < (unsigned) ret; i++) {
				log_msg(stdout, "GROUP-%s",
				       gnutls_group_get_name(list[i]));
				if (i + 1 != (unsigned) ret)
					log_msg(stdout, ", ");
				else
					log_msg(stdout, "\n");
			}
		}

		{
			ret = gnutls_priority_sign_list(pcache, &list);

			log_msg(stdout, "PK-signatures: ");
			if (ret == 0)
				log_msg(stdout, "none\n");
			for (i = 0; i < (unsigned) ret; i++) {
				log_msg(stdout, "SIGN-%s",
				       gnutls_sign_algorithm_get_name(list
								      [i]));
				if (i + 1 != (unsigned) ret)
					log_msg(stdout, ", ");
				else
					log_msg(stdout, "\n");
			}
		}

		gnutls_priority_deinit(pcache);
		return;
	}

	log_msg(stdout, "Cipher suites:\n");
	for (i = 0; (name = gnutls_cipher_suite_info
		     (i, id, &kx, &cipher, &mac, &version)); i++) {
		log_msg(stdout, "%-50s\t0x%02x, 0x%02x\t%s\n",
		       name,
		       (unsigned char) id[0], (unsigned char) id[1],
		       gnutls_protocol_get_name(version));
		if (verbose)
			log_msg
			    (stdout, "\tKey exchange: %s\n\tCipher: %s\n\tMAC: %s\n\n",
			     gnutls_kx_get_name(kx),
			     gnutls_cipher_get_name(cipher),
			     gnutls_mac_get_name(mac));
	}

	log_msg(stdout, "\n");
	{
		const gnutls_certificate_type_t *p =
		    gnutls_certificate_type_list();

		log_msg(stdout, "Certificate types: ");
		for (; *p; p++) {
			log_msg(stdout, "CTYPE-%s",
			       gnutls_certificate_type_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_protocol_t *p = gnutls_protocol_list();

		log_msg(stdout, "Protocols: ");
		for (; *p; p++) {
			log_msg(stdout, "VERS-%s", gnutls_protocol_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_cipher_algorithm_t *p = gnutls_cipher_list();

		log_msg(stdout, "Ciphers: ");
		for (; *p; p++) {
			log_msg(stdout, "%s", gnutls_cipher_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_mac_algorithm_t *p = gnutls_mac_list();

		log_msg(stdout, "MACs: ");
		for (; *p; p++) {
			log_msg(stdout, "%s", gnutls_mac_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_digest_algorithm_t *p = gnutls_digest_list();

		log_msg(stdout, "Digests: ");
		for (; *p; p++) {
			log_msg(stdout, "%s", gnutls_digest_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_kx_algorithm_t *p = gnutls_kx_list();

		log_msg(stdout, "Key exchange algorithms: ");
		for (; *p; p++) {
			log_msg(stdout, "%s", gnutls_kx_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_compression_method_t *p =
		    gnutls_compression_list();

		log_msg(stdout, "Compression: ");
		for (; *p; p++) {
			log_msg(stdout, "COMP-%s", gnutls_compression_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_group_t *p = gnutls_group_list();

		log_msg(stdout, "Groups: ");
		for (; *p; p++) {
			log_msg(stdout, "GROUP-%s", gnutls_group_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_pk_algorithm_t *p = gnutls_pk_list();

		log_msg(stdout, "Public Key Systems: ");
		for (; *p; p++) {
			log_msg(stdout, "%s", gnutls_pk_algorithm_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}

	{
		const gnutls_sign_algorithm_t *p = gnutls_sign_list();

		log_msg(stdout, "PK-signatures: ");
		for (; *p; p++) {
			log_msg(stdout, "SIGN-%s",
			       gnutls_sign_algorithm_get_name(*p));
			if (*(p + 1))
				log_msg(stdout, ", ");
			else
				log_msg(stdout, "\n");
		}
	}
}

void
print_key_material(gnutls_session_t session, const char *label, size_t size)
{
	gnutls_datum_t bin = { NULL, 0 }, hex = { NULL, 0 };
	int ret;

	bin.data = gnutls_malloc(size);
	if (!bin.data) {
		fprintf(stderr, "Error in gnutls_malloc: %s\n",
			gnutls_strerror(GNUTLS_E_MEMORY_ERROR));
		goto out;
	}

	bin.size = size;

	ret = gnutls_prf_rfc5705(session, strlen(label), label,
				 0, NULL, size, (char *)bin.data);
	if (ret < 0) {
		fprintf(stderr, "Error in gnutls_prf_rfc5705: %s\n",
			gnutls_strerror(ret));
		goto out;
	}

	ret = gnutls_hex_encode2(&bin, &hex);
	if (ret < 0) {
		fprintf(stderr, "Error in hex encoding: %s\n",
			gnutls_strerror(ret));
		goto out;
	}
	log_msg(stdout, "- Key material: %s\n", hex.data);
	fflush(stdout);

 out:
	gnutls_free(bin.data);
	gnutls_free(hex.data);
}

int check_command(gnutls_session_t session, const char *str, unsigned no_cli_cert)
{
	size_t len = strnlen(str, 128);
	int ret;

	fprintf(stderr, "*** Processing %u bytes command: %s\n", (unsigned)len,
		str);
	if (len > 2 && str[0] == str[1] && str[0] == '*') {
		if (strncmp
		    (str, "**REHANDSHAKE**",
		     sizeof("**REHANDSHAKE**") - 1) == 0) {
			fprintf(stderr,
				"*** Sending rehandshake request\n");
			gnutls_rehandshake(session);
			return 1;
		} else if (strncmp
		    (str, "**REAUTH**",
		     sizeof("**REAUTH**") - 1) == 0) {
			/* in case we have a re-auth cmd prepare for it */
			if (no_cli_cert)
				gnutls_certificate_server_set_request(session, GNUTLS_CERT_REQUIRE);

			fprintf(stderr,
				"*** Sending re-auth request\n");
			do {
				ret = gnutls_reauth(session, 0);
			} while(ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED);
			if (ret < 0) {
				fprintf(stderr, "reauth: %s\n",
					gnutls_strerror(ret));
				exit(1);
			}
			return 1;
		} else
		    if (strncmp
			(str, "**HEARTBEAT**",
			 sizeof("**HEARTBEAT**") - 1) == 0) {
			ret =
			    gnutls_heartbeat_ping(session, 300, 5,
						  GNUTLS_HEARTBEAT_WAIT);
			if (ret < 0) {
				if (ret == GNUTLS_E_INVALID_REQUEST) {
					fprintf(stderr,
						"No heartbeat in this session\n");
				} else {
					fprintf(stderr, "ping: %s\n",
						gnutls_strerror(ret));
					exit(1);
				}
			}
			return 2;
		}
	}
	return 0;
}

/* error is indicated by returning an empty string */
void getpass_copy(char *pass, size_t max_pass_size, const char *prompt)
{
	char *tmp;
	size_t len;

	tmp = getpass(prompt);
	if (tmp == NULL) {
		pass[0] = 0;
		return;
	}

	len = strlen(tmp);
	if (len >= max_pass_size) {
		gnutls_memset(tmp, 0, len);
		pass[0] = 0;
		return;
	}

	strcpy(pass, tmp);
	gnutls_memset(tmp, 0, len);

	return;
}

/* error is indicated by returning an empty string */
void getenv_copy(char *str, size_t max_str_size, const char *envvar)
{
	char *tmp;
	size_t len;

	tmp = getenv(envvar);
	if (tmp == NULL) {
		str[0] = 0;
		return;
	}

	len = strlen(tmp);
	if (len >= max_str_size) {
		str[0] = 0;
		return;
	}

	strcpy(str, tmp);

	return;
}

#define MIN(x,y) ((x)<(y))?(x):(y)
#define MAX_CACHE_TRIES 5
int
pin_callback(void *user, int attempt, const char *token_url,
	     const char *token_label, unsigned int flags, char *pin,
	     size_t pin_max)
{
	char password[MAX_PIN_LEN] = "";
	common_info_st *info = user;
	const char *desc;
	int cache = MAX_CACHE_TRIES;
	unsigned len;
/* allow caching of PIN */
	static char *cached_url = NULL;
	static char cached_pin[MAX_PIN_LEN] = "";
	const char *env;

	if (flags & GNUTLS_PIN_SO) {
		env = "GNUTLS_SO_PIN";
		desc = "security officer";
		if (info && info->so_pin)
			snprintf(password, sizeof(password), "%s", info->so_pin);
	} else {
		env = "GNUTLS_PIN";
		desc = "user";
		if (info && info->pin)
			snprintf(password, sizeof(password), "%s", info->pin);
	}

	if (flags & GNUTLS_PIN_FINAL_TRY) {
		cache = 0;
		log_msg(stdout, "*** This is the final try before locking!\n");
	}
	if (flags & GNUTLS_PIN_COUNT_LOW) {
		cache = 0;
		log_msg(stdout, "*** Only few tries left before locking!\n");
	}

	if (flags & GNUTLS_PIN_WRONG) {
		cache = 0;
		log_msg(stdout, "*** Wrong PIN has been provided!\n");
	}

	if (cache > 0 && cached_url != NULL) {
		if (token_url != NULL
		    && strcmp(cached_url, token_url) == 0) {
			if (strlen(cached_pin) >= pin_max) {
				fprintf(stderr, "Too long PIN given\n");
				exit(1);
			}

			if (info && info->verbose) {
				fprintf(stderr,
					"Re-using cached PIN for token '%s'\n",
					token_label);
			}
			strcpy(pin, cached_pin);
			cache--;
			return 0;
		}
	}

	if (password[0] == 0) {
		getenv_copy(password, sizeof(password), env);
		if (password[0] == 0) /* compatibility */
			getenv_copy(password, sizeof(password), "GNUTLS_PIN");
	}

	if (password[0] == 0 && (info == NULL || info->batch == 0 || info->ask_pass != 0)) {
		if (token_label && token_label[0] != 0) {
			fprintf(stderr, "Token '%s' with URL '%s' ", token_label, token_url);
			fprintf(stderr, "requires %s PIN\n", desc);
			getpass_copy(password, sizeof(password), "Enter PIN: ");
		} else {
			getpass_copy(password, sizeof(password), "Enter password: ");
		}

	} else {
		if (flags & GNUTLS_PIN_WRONG) {
			if (token_label && token_label[0] != 0) {
				fprintf(stderr, "Token '%s' with URL '%s' ", token_label, token_url);
				fprintf(stderr, "requires %s PIN\n", desc);
			}
			fprintf(stderr, "Cannot continue with a wrong password in the environment.\n");
			exit(1);
		}
	}

	if (password[0] == 0 || password[0] == '\n') {
		fprintf(stderr, "No PIN given.\n");
		if (info != NULL && info->batch != 0) {
			fprintf(stderr, "note: when operating in batch mode, set the GNUTLS_PIN or GNUTLS_SO_PIN environment variables\n");
		}
		exit(1);
	}

	len = MIN(pin_max - 1, strlen(password));
	memcpy(pin, password, len);
	pin[len] = 0;

	/* cache */
	if (len < sizeof(cached_pin)) {
		memcpy(cached_pin, pin, len);
		cached_pin[len] = 0;
	} else
		cached_pin[0] = 0;

	free(cached_url);
	if (token_url)
		cached_url = strdup(token_url);
	else
		cached_url = NULL;

	return 0;
}

#ifdef ENABLE_PKCS11

static int
token_callback(void *user, const char *label, const unsigned retry)
{
	char buf[32];
	common_info_st *info = user;

	if (retry > 0 || (info != NULL && info->batch != 0)) {
		fprintf(stderr, "Could not find token %s\n", label);
		return -1;
	}
	log_msg(stdout, "Please insert token '%s' in slot and press enter\n",
	       label);
	if (fgets(buf, sizeof(buf), stdin) == NULL) {
		fprintf(stderr, "error reading input\n");
		return -1;
	}

	return 0;
}

void pkcs11_common(common_info_st *c)
{

	gnutls_pkcs11_set_pin_function(pin_callback, c);
	gnutls_pkcs11_set_token_function(token_callback, c);

}

#endif

void sockets_init(void)
{
#ifdef _WIN32
	WORD wVersionRequested;
	WSADATA wsaData;

	wVersionRequested = MAKEWORD(1, 1);
	if (WSAStartup(wVersionRequested, &wsaData) != 0) {
		perror("WSA_STARTUP_ERROR");
	}
#else
	signal(SIGPIPE, SIG_IGN);
#endif
}


int log_msg(FILE *file, const char *message, ...)
{
	va_list args;
	int rv;

	va_start(args, message);

	rv = vfprintf(logfile ? logfile : file, message, args);

	va_end(args);

	return rv;
}

void log_set(FILE *file)
{
	logfile = file;
}

/* This is very similar to ctime() but it does not force a newline.
 */
char *simple_ctime(const time_t *t, char out[SIMPLE_CTIME_BUF_SIZE])
{
	struct tm tm;

	if (localtime_r(t, &tm) == NULL)
		goto error;

	if (!strftime(out, SIMPLE_CTIME_BUF_SIZE, "%c", &tm))
		goto error;

	return out;

 error:
	snprintf(out, SIMPLE_CTIME_BUF_SIZE, "[error]");
	return out;
}
