actube/src/cw/connlist.c

#include <stdlib.h>
#include <errno.h>
#include <string.h>

#include <pthread.h>
#include <sys/types.h>
#include <netinet/in.h>



#include <stdio.h>

#include "conn.h"
#include "sock.h"


static int connlist_cmp(const void * d1,const void *d2)
{
	struct conn * c1=(struct conn *) d1;
	struct conn * c2=(struct conn *) d2;

	return sock_cmpaddr((struct sockaddr*)&c1->addr,(struct sockaddr*)&c2->addr,1);
}


struct connlist * connlist_create(int len)
{

	struct connlist * cl = malloc(sizeof(struct connlist));
	if (!cl)
		return 0;


	cl->t = avltree_create(connlist_cmp,0);

	if (!cl->t){
		free(cl);
		return 0;
	}

	if (pthread_mutex_init(&cl->connlist_mutex,NULL)){
		avltree_destroy(cl->t);
		free(cl);
		return 0;
	};

	cl->len=len;

	return cl;
}

void connlist_lock(struct connlist * cl)
{
	pthread_mutex_lock(&cl->connlist_mutex);
}

void connlist_unlock(struct connlist * cl)
{
	pthread_mutex_unlock(&cl->connlist_mutex);
}


void connlist_destroy(struct connlist * cl)
{
	if (!cl)
		return;

	if (cl->t)
		avltree_destroy(cl->t);
	pthread_mutex_destroy(&cl->connlist_mutex);
	free(cl);
	
}


struct conn * connlist_get(struct connlist * cl, const struct sockaddr * addr)
{
	struct conn dummy;
	sock_copyaddr(&dummy.addr,addr);
	return avltree_get(cl->t,&dummy);
}


struct conn * connlist_add(struct connlist * cl, struct conn * conn)
{
	if ( cl->len!=0)
		if (cl->t->count>=cl->len)
			return 0;

	return avltree_add(cl->t,conn);
}


void connlist_remove(struct connlist *cl,struct conn * conn)
{
	avltree_del(cl->t,conn);
}