actube/src/cw/action.c

/*
    This file is part of libcapwap.

    libcapwap 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.

    libcapwap 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 Foobar.  If not, see <http://www.gnu.org/licenses/>.

*/

/**
 * @file
 * @brief Implementation of methods for actionlists.
 */

#include "mbag.h"
#include <stdlib.h>
#include <string.h>

#include "action.h"

/* ---------------------------------------------
 * cw_actionlist_in stuff 
 */


static int cw_action_in_cmp(const void *elem1, const void *elem2)
{
	struct cw_action_in *e1 = (struct cw_action_in *) elem1;
	struct cw_action_in *e2 = (struct cw_action_in *) elem2;
	int r;

	r = e1->capwap_state - e2->capwap_state;
	if (r != 0)
		return r;

	r = e1->msg_id - e2->msg_id;
	if (r != 0)
		return r;

	r = e1->elem_id - e2->elem_id;
	if (r != 0)
		return r;

	r = e1->vendor_id - e2->vendor_id;
	if (r != 0)
		return r;

	r = e1->proto - e2->proto;
	if (r != 0)
		return r;

	return 0;
}

/**
 * Create an action list for incomming messages
 * @return the create list or NULL if an error has occured
 */
cw_actionlist_in_t cw_actionlist_in_create()
{
	return mavl_create(cw_action_in_cmp, free);
}

/**
 * Add an element to an "in-list"
 * @param t action list to elem to
 * @param a element to add
 * @rturn a pointer to the added element 
 */
cw_action_in_t *cw_actionlist_in_add(cw_actionlist_in_t t, struct cw_action_in * a)
{
	int s = sizeof(struct cw_action_in);

	void *r = mavl_replace_data(t, a, s);
	if (r) 
		return r;

	void *an = malloc(s);
	if (!an)
		return NULL;

	memcpy(an, a, s);
	return mavl_add(t, an);
}

/**
 * Get an element from an actionlist_in
 * @param t action list
 * @param a element to search for
 * @return the elemen or NULL if not found
 */
struct cw_action_in *cw_actionlist_in_get(cw_actionlist_in_t t, struct cw_action_in *a)
{
	return avltree_get(t, a);
}


/**
 * Register actions in an action list for incommin messages 
 * @param t action list, where messaes will be registered
 * @param actions an array of actions to reggister
 * @return the number of registred actions
 */
int cw_actionlist_in_register_actions(cw_actionlist_in_t t, cw_action_in_t * actions)
{
	int n=0;
	while (actions->capwap_state) {
		cw_action_in_t *rc = cw_actionlist_in_add(t, actions);
		if (!rc)
			return 0;
		actions++;
		n++;
	}
	return n;
}

/* ---------------------------------------------
 * cw_actionlist_out stuff 
 */



struct outelem{
	uint32_t msg_id;
	mlist_t * mlist;
};


static int mout_cmp(void *elem1,void *elem2)
{

	struct cw_action_out *e1 = (struct cw_action_out *) elem1;
	struct cw_action_out *e2 = (struct cw_action_out *) elem2;
	int r;

	r = e1->msg_id - e2->msg_id;
	if (r )
		return r;

	r = e1->elem_id - e2->elem_id;
	if (r )
		return r;


	r = e1->vendor_id - e2->vendor_id;
	if (r )
		return r;

	if (e1->item_id == e2->item_id)
		return 0;

	if (e1->item_id && e2->item_id)
		return strcmp(e1->item_id,e2->item_id);
	
	return 1;	
}



struct outelem * cw_actionlist_mout_create(int msg_id)
{
	struct outelem * o = malloc(sizeof(struct outelem));
	if (!o)
		return NULL;

	o->mlist= mlist_create(mout_cmp);
	if (!o->mlist){
		free(o);
		return NULL;
	}
	o->msg_id=msg_id;
	return o;
}

static struct outelem * cw_actionlist_out_get_outelem(cw_actionlist_out_t t, int msg_id)
{
	struct outelem search;
	search.msg_id=msg_id;
	return mavl_get(t,&search);
}



/* 
 * Compare function for actionlist_out_t lists
 */
static int cw_action_out_cmp(const void *elem1, const void *elem2)
{
	struct outelem *e1 = (struct outelem *) elem1;
	struct outelem *e2 = (struct outelem *) elem2;
	return e1->msg_id - e2->msg_id;
}



/**
 * Add an element to an actionlist_out
 * @param t action list
 * @param a element to add
 */
cw_action_out_t *cw_actionlist_out_add(cw_actionlist_out_t t, struct cw_action_out * a)
{
	struct outelem * o =  cw_actionlist_out_get_outelem(t,a->msg_id);


	if (!o){
		o = cw_actionlist_mout_create(a->msg_id);
		if (!o) {
			return NULL;
		}
		mavl_add(t,o);
	}

	struct mlist_elem * e = mlist_replace(o->mlist,NULL,a);
	if (!e)
		e = mlist_append(o->mlist,a);

	if (e)
		return a;

	return 0;
}


/**
 * Register actions in an action list for incommin messages 
 * @param t action list, where messaes will be registered
 * @param actions an array of actions to reggister
 * @return the number of registred actions
 */

int cw_actionlist_out_register_actions(cw_actionlist_out_t t, cw_action_out_t * actions)
{
	int n=0;
	while (actions->msg_id != 0) {
		cw_action_out_t *rc = cw_actionlist_out_add(t, actions);
		if (rc == 0)
			return 0;
		actions++;
		n++;
	}
	return n;
}



/**
 * Add an action to an action 
 * @param t actionlist
 * @param a action to add
 * @param s size of element to add
 * @return pointer to added element or NULL if an error has opccured
 */
void *cw_actionlist_add(struct avltree *t, void *a, size_t s)
{

	void *r = mavl_replace_data(t, a, s); //sizeof(struct cw_action_in));
	if (r) {
		return r;
	}

	void *an = malloc(s);
	if (!an)
		return NULL;

	memcpy(an, a, s);
	return avltree_add(t, an);

}


/**
 * Create an action list for outgoing message lements
 * @return the created action list or NULL if an erro has occured.
 */
cw_actionlist_out_t cw_actionlist_out_create()
{
	return mavl_create(cw_action_out_cmp, free);
}



mlist_t * cw_actionlist_out_get(cw_actionlist_out_t t,int msg_id)
{
	struct outelem *o = cw_actionlist_out_get_outelem(t,msg_id);
	if (!o)
		return NULL;
	return o->mlist;
}






/* misc stuff */


cw_action_fun_t cw_set_msg_end_callback(struct cw_actiondef *actions, 
		int capwap_state,int msg_id, cw_action_fun_t callback)
{
	cw_action_in_t as;
	as.capwap_state = capwap_state;
	as.msg_id = msg_id;
	as.vendor_id = 0;
	as.elem_id = 0;
	as.proto = 0;
	
	cw_action_in_t *af;
	

	af = cw_actionlist_in_get(actions->in, &as);
	if (!af) 
		return NULL;

	cw_action_fun_t  old = af->end;
	af->end =callback;
	return old;

}