/*

 * Copyright (C) 2003-2016 Free Software Foundation, Inc.

 *

 * Authors: Nikos Mavrogiannopoulos, Simon Josefsson, Howard Chu

 *

 * This file is part of GnuTLS.

 *

 * The GnuTLS is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public License

 * as published by the Free Software Foundation; either version 2.1 of

 * the License, or (at your option) any later version.

 *

 * This library 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

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public License

 * along with this program.  If not, see <https://www.gnu.org/licenses/>

 *

 */

#include "gnutls_int.h"

#include <datum.h>

#include <global.h>

#include "errors.h"

#include <common.h>

#include <gnutls/x509-ext.h>

#include <x509.h>

#include <x509_b64.h>

#include <x509_int.h>

#include <libtasn1.h>

#include <pk.h>

#include <pkcs11_int.h>

#include "urls.h"

/**

 * gnutls_x509_tlsfeatures_init:

 * @f: The TLS features

 *

 * This function will initialize a X.509 TLS features extension structure

 *

 * Returns: On success, %GNUTLS_E_SUCCESS (0) is returned,

 *   otherwise a negative error value.

 *

 * Since: 3.5.1

 **/

int gnutls_x509_tlsfeatures_init(gnutls_x509_tlsfeatures_t *f)

{

	*f = gnutls_calloc(1, sizeof(struct gnutls_x509_tlsfeatures_st));

	if (*f == NULL)

		return gnutls_assert_val(GNUTLS_E_MEMORY_ERROR);

	return 0;

}

/**

 * gnutls_x509_tlsfeatures_deinit:

 * @f: The TLS features

 *

 * This function will deinitialize a X.509 TLS features extension structure

 *

 * Since: 3.5.1

 **/

void gnutls_x509_tlsfeatures_deinit(gnutls_x509_tlsfeatures_t f)

{

	gnutls_free(f);

}

/**

 * gnutls_x509_tlsfeatures_get:

 * @f: The TLS features

 * @idx: The index of the feature to get

 * @feature: If the function succeeds, the feature will be stored in this variable

 *

 * This function will get a feature from the X.509 TLS features

 * extension structure.

 *

 * Returns: On success, %GNUTLS_E_SUCCESS (0) is returned,

 *   otherwise a negative error value.

 *

 * Since: 3.5.1

 **/

int gnutls_x509_tlsfeatures_get(gnutls_x509_tlsfeatures_t f, unsigned idx, unsigned int *feature)

{

	if (f == NULL) {

		gnutls_assert();

		return GNUTLS_E_INVALID_REQUEST;

	}

	if (idx >= f->size) {

		return gnutls_assert_val(GNUTLS_E_REQUESTED_DATA_NOT_AVAILABLE);

	}

	*feature = f->feature[idx];

	return 0;

}

/**

 * gnutls_x509_crt_get_tlsfeatures:

 * @crt: A X.509 certificate

 * @features: If the function succeeds, the

 *   features will be stored in this variable.

 * @flags: zero or %GNUTLS_EXT_FLAG_APPEND

 * @critical: the extension status

 *

 * This function will get the X.509 TLS features

 * extension structure from the certificate. The

 * returned structure needs to be freed using

 * gnutls_x509_tlsfeatures_deinit().

 *

 * When the @flags is set to %GNUTLS_EXT_FLAG_APPEND,

 * then if the @features structure is empty this function will behave

 * identically as if the flag was not set. Otherwise if there are elements 

 * in the @features structure then they will be merged with.

 *

 * Note that @features must be initialized prior to calling this function.

 *

 * Returns: On success, %GNUTLS_E_SUCCESS (0) is returned,

 *   otherwise a negative error value.

 *

 * Since: 3.5.1

 **/

int gnutls_x509_crt_get_tlsfeatures(gnutls_x509_crt_t crt,

				    gnutls_x509_tlsfeatures_t features,

				    unsigned int flags,

				    unsigned int *critical)

{

	int ret;

	gnutls_datum_t der;

	if (crt == NULL) {

		gnutls_assert();

		return GNUTLS_E_INVALID_REQUEST;

	}

	if ((ret =

		 _gnutls_x509_crt_get_extension(crt, GNUTLS_X509EXT_OID_TLSFEATURES, 0,

						&der, critical)) < 0)

	{

		return ret;

	}

	if (der.size == 0 || der.data == NULL) {

		gnutls_assert();

		return GNUTLS_E_REQUESTED_DATA_NOT_AVAILABLE;

	}

	ret = gnutls_x509_ext_import_tlsfeatures(&der, features, flags);

	if (ret < 0) {

		gnutls_assert();

		goto cleanup;

	}

	ret = 0;

 cleanup:

	gnutls_free(der.data);

	return ret;

}

/**

 * gnutls_x509_crt_set_tlsfeatures:

 * @crt: A X.509 certificate

 * @features: If the function succeeds, the

 *   features will be added to the certificate.

 *

 * This function will set the certificates

 * X.509 TLS extension from the given structure.

 *

 * Returns: On success, %GNUTLS_E_SUCCESS (0) is returned,

 *   otherwise a negative error value.

 *

 * Since: 3.5.1

 **/

int gnutls_x509_crt_set_tlsfeatures(gnutls_x509_crt_t crt,

				    gnutls_x509_tlsfeatures_t features)

{

	int ret;

	gnutls_datum_t der;

	if (crt == NULL || features == NULL) {

		gnutls_assert();

		return GNUTLS_E_INVALID_REQUEST;

	}

	ret = gnutls_x509_ext_export_tlsfeatures(features, &der;;

	if (ret < 0) {

		gnutls_assert();

		return ret;

	}

	ret = _gnutls_x509_crt_set_extension(crt, GNUTLS_X509EXT_OID_TLSFEATURES, &der, 0);

	_gnutls_free_datum(&der;;

	if (ret < 0) {

		gnutls_assert();

	}

	return ret;

}

/**

 * gnutls_x509_tls_features_check_crt:

 * @feat: a set of TLSFeatures

 * @cert: the certificate to be checked

 *

 * This function will check the provided certificate against the TLSFeatures

 * set in @feat using the RFC7633 p.4.2.2 rules. It will check whether the certificate

 * contains the features in @feat or a superset.

 *

 * Returns: non-zero if the provided certificate complies, and zero otherwise.

 *

 * Since: 3.5.1

 **/

unsigned gnutls_x509_tlsfeatures_check_crt(gnutls_x509_tlsfeatures_t feat,

					   gnutls_x509_crt_t cert)

{

	int ret;

	gnutls_x509_tlsfeatures_t cfeat;

	unsigned i, j, uret, found;

	if (feat->size == 0)

		return 1; /* shortcut; no constraints to check */

	ret = gnutls_x509_tlsfeatures_init(&cfeat);

	if (ret < 0)

		return gnutls_assert_val(0);

	ret = gnutls_x509_crt_get_tlsfeatures(cert, cfeat, 0, NULL);

	if (ret < 0) {

		gnutls_assert();

		uret = 0;

		goto cleanup;

	}

	/* if cert's features cannot be a superset */

	if (feat->size > cfeat->size) {

		_gnutls_debug_log("certificate has %u, while issuer has %u tlsfeatures\n", cfeat->size, feat->size);

		gnutls_assert();

		uret = 0;

		goto cleanup;

	}

	for (i=0;i<feat->size;i++) {

		found = 0;

		for (j=0;j<cfeat->size;j++) {

			if (feat->feature[i] == cfeat->feature[j]) {

				found = 1;

				break;

			}

		}

		if (found == 0) {

			_gnutls_debug_log("feature %d was not found in cert\n", (int)feat->feature[i]);

			uret = 0;

			goto cleanup;

		}

	}

	uret = 1;

 cleanup:

	gnutls_x509_tlsfeatures_deinit(cfeat);

	return uret;

}