147 lines
2.9 KiB
C
147 lines
2.9 KiB
C
/*
|
|
This file is part of actube.
|
|
|
|
actube 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/>.
|
|
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
|
|
#include <pthread.h>
|
|
#include <sys/types.h>
|
|
#include <netinet/in.h>
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include "mbag.h"
|
|
#include "conn.h"
|
|
#include "connlist.h"
|
|
|
|
#include "sock.h"
|
|
|
|
|
|
|
|
|
|
static int cmp_by_addr(const mavldata_t * d1,const mavldata_t *d2)
|
|
{
|
|
struct conn * c1=(struct conn *) d1->ptr;
|
|
struct conn * c2=(struct conn *) d2->ptr;
|
|
|
|
return sock_cmpaddr((struct sockaddr*)&c1->addr,(struct sockaddr*)&c2->addr,1);
|
|
}
|
|
|
|
static int cmp_by_session_id(const mavldata_t *d1, const mavldata_t *d2)
|
|
{
|
|
struct conn * c1=(struct conn *) d1->ptr;
|
|
struct conn * c2=(struct conn *) d2->ptr;
|
|
return memcmp(c1->session_id,c2->session_id,16);
|
|
}
|
|
|
|
|
|
struct connlist * connlist_create(int len)
|
|
{
|
|
|
|
struct connlist * cl = malloc(sizeof(struct connlist));
|
|
if (!cl)
|
|
return 0;
|
|
|
|
|
|
cl->by_addr = mavl_create(cmp_by_addr,0);
|
|
|
|
if (!cl->by_addr){
|
|
free(cl);
|
|
return 0;
|
|
}
|
|
|
|
cl->by_session_id = mavl_create(cmp_by_session_id,0);
|
|
|
|
|
|
if (pthread_mutex_init(&cl->connlist_mutex,NULL)){
|
|
mavl_destroy(cl->by_addr);
|
|
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->by_addr)
|
|
mavl_destroy(cl->by_addr);
|
|
pthread_mutex_destroy(&cl->connlist_mutex);
|
|
free(cl);
|
|
|
|
}
|
|
|
|
|
|
struct conn * connlist_get(struct connlist * cl, const struct sockaddr * addr)
|
|
{
|
|
struct conn search;
|
|
sock_copyaddr(&search.addr,addr);
|
|
return mavl_get_ptr(cl->by_addr,&search);
|
|
}
|
|
|
|
|
|
struct conn * connlist_add(struct connlist * cl, struct conn * conn)
|
|
{
|
|
if ( cl->len!=0)
|
|
if (cl->by_addr->count>=cl->len)
|
|
return NULL;
|
|
conn->connlist=cl;
|
|
return mavl_add_ptr(cl->by_addr,conn);
|
|
}
|
|
|
|
struct conn * connlist_get_by_session_id(struct connlist *cl, struct conn * conn)
|
|
{
|
|
return mavl_get_ptr(cl->by_session_id,conn);
|
|
}
|
|
|
|
struct conn * connlist_add_by_session_id(struct connlist * cl, struct conn * conn)
|
|
{
|
|
return mavl_add_ptr(cl->by_session_id,conn);
|
|
}
|
|
|
|
void connlist_remove(struct connlist *cl,struct conn * conn)
|
|
{
|
|
mavldata_t md;
|
|
md.ptr=conn;
|
|
mavl_del(cl->by_session_id,&md);
|
|
md.ptr=conn;
|
|
mavl_del(cl->by_addr,&md);
|
|
}
|
|
|
|
|
|
|