#include "capwap.h"
#include "element.h"

/********************************************************************

 0                   1                   2                   3
 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|    Reason     |    Length     |       Message Element...
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+

Type:   34 for Returned Message Element

Length:  >= 6

********************************************************************/

/* */
static void capwap_returnedmessage_element_create(void* data, capwap_message_elements_handle handle, struct capwap_write_message_elements_ops* func) {
	struct capwap_returnedmessage_element* element = (struct capwap_returnedmessage_element*)data;

	ASSERT(data != NULL);
	ASSERT(element->length >= 4);
	ASSERT((element->reason == CAPWAP_RETURNED_MESSAGE_UNKNOWN_MESSAGE_ELEMENT) || (element->reason == CAPWAP_RETURNED_MESSAGE_UNSUPPORTED_MESSAGE_ELEMENT) || 
		(element->reason == CAPWAP_RETURNED_MESSAGE_UNKNOWN_MESSAGE_ELEMENT_VALUE) || (element->reason == CAPWAP_RETURNED_MESSAGE_UNSUPPORTED_MESSAGE_ELEMENT_VALUE));

	func->write_u8(handle, element->reason);
	func->write_u8(handle, element->length);
	func->write_block(handle, element->message, element->length);
}

/* */
static void* capwap_returnedmessage_element_clone(void* data) {
	struct capwap_returnedmessage_element* cloneelement;

	ASSERT(data != NULL);

	cloneelement = capwap_clone(data, sizeof(struct capwap_returnedmessage_element));
	if (cloneelement->length > 0) {
		cloneelement->message = capwap_clone(((struct capwap_returnedmessage_element*)data)->message, cloneelement->length);
	}

	return cloneelement;
}

/* */
static void capwap_returnedmessage_element_free(void* data) {
	struct capwap_returnedmessage_element* element = (struct capwap_returnedmessage_element*)data;

	ASSERT(data != NULL);

	if (element->message) {
		capwap_free(element->message);
	}

	capwap_free(data);
}

/* */
static void* capwap_returnedmessage_element_parsing(capwap_message_elements_handle handle, struct capwap_read_message_elements_ops* func) {
	unsigned short length;
	struct capwap_returnedmessage_element* data;

	ASSERT(handle != NULL);
	ASSERT(func != NULL);

	length = func->read_ready(handle);
	if (length < 6) {
		log_printf(LOG_DEBUG, "Invalid Returned Message element: underbuffer");
		return NULL;
	}

	length -= 2;
	if (length > CAPWAP_RETURNED_MESSAGE_MAX_LENGTH) {
		log_printf(LOG_DEBUG, "Invalid Returned Message element: overbuffer");
		return NULL;
	}

	/* */
	data = (struct capwap_returnedmessage_element*)capwap_alloc(sizeof(struct capwap_returnedmessage_element));
	memset(data, 0, sizeof(struct capwap_returnedmessage_element));

	/* Retrieve data */
	func->read_u8(handle, &data->reason);
	func->read_u8(handle, &data->length);

	if ((data->reason != CAPWAP_RETURNED_MESSAGE_UNKNOWN_MESSAGE_ELEMENT) &&
		(data->reason != CAPWAP_RETURNED_MESSAGE_UNSUPPORTED_MESSAGE_ELEMENT) &&
		(data->reason != CAPWAP_RETURNED_MESSAGE_UNKNOWN_MESSAGE_ELEMENT_VALUE) &&
		(data->reason != CAPWAP_RETURNED_MESSAGE_UNSUPPORTED_MESSAGE_ELEMENT_VALUE)) {
		capwap_returnedmessage_element_free((void*)data);
		log_printf(LOG_DEBUG, "Invalid Returned Message element: invalid reason");
		return NULL;
	} else if (data->length != length) {
		capwap_returnedmessage_element_free((void*)data);
		log_printf(LOG_DEBUG, "Invalid Returned Message element: invalid length");
		return NULL;
	}

	data->message = (uint8_t*)capwap_alloc(data->length);
	func->read_block(handle, data->message, data->length);

	return data;
}

/* */
const struct capwap_message_elements_ops capwap_element_returnedmessage_ops = {
	.category = CAPWAP_MESSAGE_ELEMENT_ARRAY,
	.create = capwap_returnedmessage_element_create,
	.parse = capwap_returnedmessage_element_parsing,
	.clone = capwap_returnedmessage_element_clone,
	.free = capwap_returnedmessage_element_free
};