/*

 * Copyright (C) 2000-2012 Free Software Foundation, Inc.

 *

 * Author: Nikos Mavrogiannopoulos

 *

 * 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/>

 *

 */

#ifndef GNUTLS_LIB_RECORD_H

#define GNUTLS_LIB_RECORD_H

#include <gnutls/gnutls.h>

#include <buffers.h>

#include <constate.h>

ssize_t _gnutls_send_tlen_int(gnutls_session_t session,

			      content_type_t type,

			      gnutls_handshake_description_t htype,

			      unsigned int epoch_rel, const void *data,

			      size_t sizeofdata, size_t min_pad,

			      unsigned int mflags);

inline static ssize_t

_gnutls_send_int(gnutls_session_t session, content_type_t type,

		 gnutls_handshake_description_t htype,

		 unsigned int epoch_rel, const void *_data,

		 size_t data_size, unsigned int mflags)

{

	return _gnutls_send_tlen_int(session, type, htype, epoch_rel,

				     _data, data_size, 0, mflags);

}

ssize_t _gnutls_recv_int(gnutls_session_t session, content_type_t type,

			 uint8_t * data,

			 size_t sizeofdata, void *seq, unsigned int ms);

inline static unsigned max_record_recv_size(gnutls_session_t session)

{

	unsigned size;

	if (session->internals.max_recv_size == 0) {

		size = session->security_parameters.max_record_recv_size + RECORD_HEADER_SIZE(session);

		if (session->internals.allow_large_records != 0)

			size += EXTRA_COMP_SIZE;

	} else {

		size = session->internals.max_recv_size;

	}

	return size;

}

inline static unsigned max_decrypted_size(gnutls_session_t session)

{

	unsigned size = 0;

	if (session->internals.allow_large_records != 0)

		size += EXTRA_COMP_SIZE;

	size += session->security_parameters.max_record_recv_size;

	return size;

}

/* Returns the headers + any IV that the ciphersuite

 * requires */

inline static

unsigned int get_total_headers(gnutls_session_t session)

{

	int ret;

	record_parameters_st *params;

	unsigned total = RECORD_HEADER_SIZE(session);

	ret = _gnutls_epoch_get(session, EPOCH_WRITE_CURRENT, &params);

	if (ret < 0) {

		return total;

	}

	return total + _gnutls_cipher_get_explicit_iv_size(params->cipher);

}

inline static

unsigned int get_total_headers2(gnutls_session_t session, record_parameters_st *params)

{

	unsigned total = RECORD_HEADER_SIZE(session);

	return total + _gnutls_cipher_get_explicit_iv_size(params->cipher);

}

inline static void session_invalidate(gnutls_session_t session)

{

	session->internals.invalid_connection = 1;

}

#endif /* GNUTLS_LIB_RECORD_H */