mirror of https://github.com/sipwise/rtpengine.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.0 KiB
79 lines
2.0 KiB
#include "cookie_cache.h"
|
|
|
|
#include <time.h>
|
|
#include <glib.h>
|
|
|
|
#include "compat.h"
|
|
#include "aux.h"
|
|
#include "poller.h"
|
|
#include "str.h"
|
|
|
|
static const char *cookie_in_use = "MAGIC";
|
|
|
|
INLINE void cookie_cache_state_init(struct cookie_cache_state *s) {
|
|
s->cookies = g_hash_table_new(str_hash, str_equal);
|
|
s->chunks = g_string_chunk_new(4 * 1024);
|
|
}
|
|
|
|
void cookie_cache_init(struct cookie_cache *c) {
|
|
cookie_cache_state_init(&c->current);
|
|
cookie_cache_state_init(&c->old);
|
|
c->swap_time = poller_now;
|
|
mutex_init(&c->lock);
|
|
cond_init(&c->cond);
|
|
}
|
|
|
|
/* lock must be held */
|
|
static void __cookie_cache_check_swap(struct cookie_cache *c) {
|
|
if (poller_now - c->swap_time >= 30) {
|
|
g_hash_table_remove_all(c->old.cookies);
|
|
g_string_chunk_clear(c->old.chunks);
|
|
swap_ptrs(&c->old.chunks, &c->current.chunks);
|
|
swap_ptrs(&c->old.cookies, &c->current.cookies);
|
|
c->swap_time = poller_now;
|
|
}
|
|
}
|
|
|
|
str *cookie_cache_lookup(struct cookie_cache *c, const str *s) {
|
|
str *ret;
|
|
|
|
mutex_lock(&c->lock);
|
|
|
|
__cookie_cache_check_swap(c);
|
|
|
|
restart:
|
|
ret = g_hash_table_lookup(c->current.cookies, s);
|
|
if (!ret)
|
|
ret = g_hash_table_lookup(c->old.cookies, s);
|
|
if (ret) {
|
|
if (ret == (void *) cookie_in_use) {
|
|
/* another thread is working on this right now */
|
|
cond_wait(&c->cond, &c->lock);
|
|
goto restart;
|
|
}
|
|
ret = str_dup(ret);
|
|
mutex_unlock(&c->lock);
|
|
return ret;
|
|
}
|
|
g_hash_table_replace(c->current.cookies, (void *) s, (void *) cookie_in_use);
|
|
mutex_unlock(&c->lock);
|
|
return NULL;
|
|
}
|
|
|
|
void cookie_cache_insert(struct cookie_cache *c, const str *s, const str *r) {
|
|
mutex_lock(&c->lock);
|
|
g_hash_table_replace(c->current.cookies, str_chunk_insert(c->current.chunks, s),
|
|
str_chunk_insert(c->current.chunks, r));
|
|
g_hash_table_remove(c->old.cookies, s);
|
|
cond_broadcast(&c->cond);
|
|
mutex_unlock(&c->lock);
|
|
}
|
|
|
|
void cookie_cache_remove(struct cookie_cache *c, const str *s) {
|
|
mutex_lock(&c->lock);
|
|
g_hash_table_remove(c->current.cookies, s);
|
|
g_hash_table_remove(c->old.cookies, s);
|
|
cond_broadcast(&c->cond);
|
|
mutex_unlock(&c->lock);
|
|
}
|