/*
 * 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 <http://www.gnu.org/licenses/>.
 */

#include <config.h>

/* Work around problem reported in
   <http://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)";

/* 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_compact(gnutls_session_t session)
{
	gnutls_x509_crt_t crt;
	const gnutls_datum_t *cert_list;
	unsigned int cert_list_size = 0;
	int ret;
	gnutls_datum_t cinfo;

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

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

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

	ret = gnutls_x509_crt_print(crt, GNUTLS_CRT_PRINT_COMPACT, &cinfo);
	if (ret == 0) {
		printf("- X.509 cert: %s\n", cinfo.data);
		gnutls_free(cinfo.data);
	}

	gnutls_x509_crt_deinit(crt);
}

static void
print_x509_info(gnutls_session_t session, FILE *out, int flag, int print_cert)
{
	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) {
		fprintf(stderr, "No certificates found!\n");
		return;
	}

	fprintf(out, "- Certificate type: X.509\n");
	fprintf(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;
		}

		fprintf(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) {
			fprintf(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;
			}

			fputs("\n", out);
			fputs((char*)pem.data, out);
			fputs("\n", out);

			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) {
		printf("- Peer did not send any certificate.\n");
		return 0;
	}

	if (rc < 0) {
		printf("- 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) {
		printf("- Could not print verification flags (err: %s)\n",
		       gnutls_strerror(rc));
		return 0;
	}

	printf("- 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;

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

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

	if (print) {
		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;

		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;
		}

		printf(" - 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 curve;

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

	curve = gnutls_ecc_curve_get(session);

	printf(" - Using curve: %s\n", gnutls_ecc_curve_get_name(curve));
	printf(" - 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;
	int rc;

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

	/* print session ID */
	gnutls_session_get_id(session, session_id, &session_id_size);
	printf("- 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 ");
		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)
			printf("- 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)
			printf("- 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)
			printf("- 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 ");
		break;
#endif
	case GNUTLS_CRD_IA:
		printf("- 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) {
				printf("- Given server name[%d]: %s\n",
				       type, dns);
			}
		}

		if ((flags & P_WAIT_FOR_CERT) && gnutls_certificate_get_ours(session) == 0)
			printf("- 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 ");
	}

	tmp =
	    SU(gnutls_protocol_get_name
	       (gnutls_protocol_get_version(session)));
	printf("- Version: %s\n", tmp);

	tmp = SU(gnutls_kx_get_name(kx));
	printf("- 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)));
		printf("- 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)));
		printf("- Client Signature: %s\n", tmp);
	}

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

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

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

#ifdef ENABLE_DTLS_SRTP
	rc = gnutls_srtp_get_selected_profile(session, &srtp_profile);
	if (rc == 0)
		printf("- 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)
		printf("- 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;

			printf("- Channel binding 'tls-unique': ");
			for (i = 0; i < cb.size; i++)
				printf("%02x", cb.data[i]);
			printf("\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;

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

	if (gnutls_certificate_client_get_request_status(session) != 0)
		printf("- Server has requested a certificate.\n");

	switch (gnutls_certificate_type_get(session)) {
	case GNUTLS_CRT_X509:
		print_x509_info(session, out, flag, print_cert);
		break;
	default:
		printf("Unknown type\n");
		break;
	}
}

void print_cert_info_compact(gnutls_session_t session)
{

	if (gnutls_certificate_client_get_request_status(session) != 0)
		printf("- Server has requested a certificate.\n");

	switch (gnutls_certificate_type_get(session)) {
	case GNUTLS_CRT_X509:
		print_x509_info_compact(session);
		break;
	default:
		printf("Unknown type\n");
		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) {
		printf("Cipher suites for %s\n", priorities);

		ret = gnutls_priority_init(&pcache, priorities, &err);
		if (ret < 0) {
			fprintf(stderr, "Syntax error at: %s\n", err);
			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)
				printf("%-50s\t0x%02x, 0x%02x\t%s\n",
				       name, (unsigned char) id[0],
				       (unsigned char) id[1],
				       gnutls_protocol_get_name(version));
		}

		printf("\n");
		{
			ret =
			    gnutls_priority_certificate_type_list(pcache,
								  &list);

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

		{
			ret = gnutls_priority_protocol_list(pcache, &list);

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

		{
			ret = gnutls_priority_cipher_list(pcache, &list);

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

		{
			ret = gnutls_priority_mac_list(pcache, &list);

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

		{
			ret = gnutls_priority_kx_list(pcache, &list);

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

		{
			ret =
			    gnutls_priority_compression_list(pcache,
							     &list);

			printf("Compression: ");
			if (ret == 0)
				printf("none\n");
			for (i = 0; i < (unsigned) ret; i++) {
				printf("COMP-%s",
				       gnutls_compression_get_name(list
								   [i]));
				if (i + 1 != (unsigned) ret)
					printf(", ");
				else
					printf("\n");
			}
		}

		{
			ret =
			    gnutls_priority_group_list(pcache, &list);

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

		{
			ret = gnutls_priority_sign_list(pcache, &list);

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

		gnutls_priority_deinit(pcache);
		return;
	}

	printf("Cipher suites:\n");
	for (i = 0; (name = gnutls_cipher_suite_info
		     (i, id, &kx, &cipher, &mac, &version)); i++) {
		printf("%-50s\t0x%02x, 0x%02x\t%s\n",
		       name,
		       (unsigned char) id[0], (unsigned char) id[1],
		       gnutls_protocol_get_name(version));
		if (verbose)
			printf
			    ("\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));
	}

	printf("\n");
	{
		const gnutls_certificate_type_t *p =
		    gnutls_certificate_type_list();

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

	{
		const gnutls_protocol_t *p = gnutls_protocol_list();

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

	{
		const gnutls_cipher_algorithm_t *p = gnutls_cipher_list();

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

	{
		const gnutls_mac_algorithm_t *p = gnutls_mac_list();

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

	{
		const gnutls_digest_algorithm_t *p = gnutls_digest_list();

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

	{
		const gnutls_kx_algorithm_t *p = gnutls_kx_list();

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

	{
		const gnutls_compression_method_t *p =
		    gnutls_compression_list();

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

	{
		const gnutls_group_t *p = gnutls_group_list();

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

	{
		const gnutls_pk_algorithm_t *p = gnutls_pk_list();

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

	{
		const gnutls_sign_algorithm_t *p = gnutls_sign_list();

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

int check_command(gnutls_session_t session, const char *str)
{
	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, "**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;
}

#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)
{
	const char *password = NULL;
	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[32] = "";
	const char *env;

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

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

	if (flags & GNUTLS_PIN_WRONG) {
		cache = 0;
		printf("*** 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 == NULL) {
		password = getenv(env);
		if (password == NULL) /* compatibility */
			password = getenv("GNUTLS_PIN");
	}

	if (password == NULL && (info == NULL || info->batch == 0)) {
		fprintf(stderr, "Token '%s' with URL '%s' ", token_label, token_url);
		fprintf(stderr, "requires %s PIN\n", desc);

		password = getpass("Enter PIN: ");
	} else {
		if (flags & GNUTLS_PIN_WRONG) {
			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 == NULL || 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;
	}
	printf("Please insert token '%s' in slot and press enter\n",
	       label);
	fgets(buf, sizeof(buf), stdin);

	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
}