actube/src/cw/msgset.c

#include <errno.h>

#include "mavl.h"
#include "debug.h"
#include "dbg.h"
#include "log.h"

#include "msget.h"

static int cmp_cw_elemhandler_by_id(const void *elem1, const void *elem2)
{
	const struct cw_ElemHandler *e1 = elem1;
	const struct cw_ElemHandler *e2 = elem2;
	int r;
	r = e1->id - e2->id;
	if (r != 0)
		return r;
	r = e1->vendor - e2->vendor;
	if (r != 0)
		return r;
	r = e1->proto - e2->proto;
	if (r != 0)
		return r;
	return 0;
}

static int cmp_cw_elemhandler_by_key(const void *elem1, const void *elem2)
{
	const struct cw_ElemHandler *e1 = elem1;
	const struct cw_ElemHandler *e2 = elem2;
	return strcmp(e1->key, e2->key);
}

static int cmp_msgdata(const void *elem1, const void *elem2)
{
	const struct cw_MsgData *e1 = elem1;
	const struct cw_MsgData *e2 = elem2;
	return e1->type - e2->type;
}

static int cmp_elemdata(const void *elem1, const void *elem2)
{
	const struct cw_ElemData *e1 = elem1;
	const struct cw_ElemData *e2 = elem2;
	int r;
	r = e1->id - e2->id;
	if (r != 0)
		return r;
	r = e1->vendor - e2->vendor;
	if (r != 0)
		return r;
	r = e1->proto - e2->proto;
	if (r != 0)
		return r;
	return 0;

}

static void msgdata_destroy(struct cw_MsgData *data)
{
	if (!data)
		return;
/*	if (data->elements_list)
		mlist_destroy(data->elements_list);
*/
	if (data->elements_tree)
		mavl_destroy(data->elements_tree);
	free(data);
}


static void del_mavl_msdgdata(void *d)
{
	msgdata_destroy(d);
}


/**
 * @brief Destroy a message set
 * @param set Message set to destroy
 */
void cw_msgset_destroy(struct cw_MsgSet *set)
{
	if (set->msgdata)
		mavl_destroy(set->msgdata);
	if (set->handlers_by_id)
		mavl_destroy(set->handlers_by_id);
	if (set->handlers_by_key)
		mavl_destroy(set->handlers_by_key);
	free(set);
}


/**
 * @brief Create a message set
 * @return Message set create, NULL if an error has occured
 */
struct cw_MsgSet *cw_msgset_create()
{

	/* allocate memory for a message_set */
	struct cw_MsgSet *set = malloc(sizeof(struct cw_MsgSet));
	if (set == NULL)
		return NULL;

	memset(set, 0, sizeof(struct cw_MsgSet));

	/* create mavl for all_elems by id */
	set->handlers_by_id = mavl_create(cmp_cw_elemhandler_by_id, NULL,
					  sizeof(struct cw_ElemHandler));
	if (set->handlers_by_id == NULL) {
		cw_msgset_destroy(set);
		return NULL;
	}

	/* create mavl for all_elems by key */
	set->handlers_by_key = mavl_create(cmp_cw_elemhandler_by_key, NULL,
					   sizeof(struct cw_ElemHandler));
	if (set->handlers_by_key == NULL) {
		cw_msgset_destroy(set);
		return NULL;
	}

	/* create mavl for messages */
	set->msgdata = mavl_create(cmp_msgdata, del_mavl_msdgdata,
				   sizeof(struct cw_MsgData));
	if (set->msgdata == NULL) {
		cw_msgset_destroy(set);
		return NULL;
	}

	return set;
}

struct cw_ElemHandler *cw_msgset_get_elemhandler(struct cw_MsgSet *set,
						 int proto, int vendor, int id)
{
	struct cw_ElemHandler search;
	search.proto = proto;
	search.vendor = vendor;
	search.id = id;
	return mavl_find(set->handlers_by_id, &search);
}


static int update_msgdata(struct cw_MsgSet *set, struct cw_MsgData *msgdata,
			  struct cw_MsgDef *msgdef)
{
	struct cw_ElemDef *elemdef;
	struct cw_ElemData ed, *result;

	/* iterate through all defined elements */
	for (elemdef = msgdef->elements; elemdef->id; elemdef++) {
		struct cw_ElemHandler *handler;
		int replaced;

		handler = cw_msgset_get_elemhandler(set,
						    elemdef->proto,
						    elemdef->vendor, elemdef->id);
		/* check if a handler for our element already exists */
		if (!handler) {
			cw_log(LOG_ERR, "No handler for message element: %d %d %d",
			       elemdef->proto, elemdef->vendor, elemdef->id);
			continue;
		}

		ed.id = elemdef->id;
		ed.proto = elemdef->proto;
		ed.vendor = elemdef->vendor;
		ed.mand = elemdef->mand;

		result = mavl_replace(msgdata->elements_tree, &ed, &replaced);

		if (!replaced) {
			cw_dbg(DBG_MOD, "  adding message element %d %d %d - %s",
			       elemdef->proto,
			       elemdef->vendor, elemdef->id, handler->name);
		} else {
			cw_dbg(DBG_MOD, "  replacing message element %d %d %d - %s",
			       elemdef->proto,
			       elemdef->vendor, elemdef->id, handler->name);
		}
	}

	return 0;
}


int cw_msgset_add(struct cw_MsgSet *set,
		  struct cw_MsgDef messages[], struct cw_ElemHandler handlers[]
    )
{

	struct cw_ElemHandler *handler;
	struct cw_MsgDef *msgdef;

	/* Create mavl for all handlers */
	for (handler = handlers; handler->id; handler++) {
		cw_dbg(DBG_MOD, "Adding handler for element %d - %s - with key: %s",
		       handler->id, handler->name, handler->key);
		mavl_replace(set->handlers_by_id, handler, NULL);
		mavl_replace(set->handlers_by_key, handler, NULL);
	}

	for (msgdef = messages; msgdef->type != 0; msgdef++) {
		struct cw_MsgData search;
		struct cw_MsgData *msg;
		int exists;

		/* Look if message already exists */
		search.type = msgdef->type;
		msg = mavl_add(set->msgdata, &search, &exists);

		if (msg == NULL) {
			cw_log(LOG_ERR, "Can't create messae");
			return 0;
		}

		if (!exists) {
			msg->elements_tree = mavl_create(cmp_elemdata, NULL,
							 sizeof(struct cw_ElemData));
		}

		/* Overwrite the found message */
		if (msgdef->name)
			msg->name = msgdef->name;
		if (msgdef->states)
			msg->states = msgdef->states;
		msg->receiver = msgdef->receiver;


		cw_dbg(DBG_MOD, "Add message Type:%d - %s ", msgdef->type, msgdef->name);


		update_msgdata(set, msg, msgdef);
	}

	return 0;
}

/**
 * @brief Find message data to a specific message
 * @param set message set
 * @param type message type to search for
 * @return message data or NULL if not found
 */
struct cw_MsgData *cw_msgset_get_msgdata(struct cw_MsgSet *set, int type)
{
	struct cw_MsgData search;
	search.type = type;
	return mavl_find_ptr(set->msgdata, &search);
}