actube/src/cw/connlist.c

102 lines
1.5 KiB
C
Raw Normal View History

#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);
}