#include "includes.h"
#define NBTKEY_FMT "NBT/%s#%02X"
BOOL namecache_enable(void)
{
if (lp_name_cache_timeout() == 0) {
DEBUG(5, ("namecache_enable: disabling netbios name cache\n"));
return False;
}
if (!gencache_init()) {
DEBUG(2, ("namecache_enable: Couldn't initialise namecache on top of gencache.\n"));
return False;
}
DEBUG(5, ("namecache_enable: enabling netbios namecache, timeout %d "
"seconds\n", lp_name_cache_timeout()));
return True;
}
BOOL namecache_shutdown(void)
{
if (!gencache_shutdown()) {
DEBUG(2, ("namecache_shutdown: Couldn't close namecache on top of gencache.\n"));
return False;
}
DEBUG(5, ("namecache_shutdown: netbios namecache closed successfully.\n"));
return True;
}
static char* namecache_key(const char *name, int name_type)
{
char *keystr;
asprintf(&keystr, NBTKEY_FMT, strupper_static(name), name_type);
return keystr;
}
BOOL namecache_store(const char *name, int name_type,
int num_names, struct ip_service *ip_list)
{
time_t expiry;
char *key, *value_string;
int i;
BOOL ret;
if (!gencache_init()) return False;
if (name_type > 255) {
return False;
}
if ( DEBUGLEVEL >= 5 ) {
DEBUG(5, ("namecache_store: storing %d address%s for %s#%02x: ",
num_names, num_names == 1 ? "": "es", name, name_type));
for (i = 0; i < num_names; i++)
DEBUGADD(5, ("%s:%d%s", inet_ntoa(ip_list[i].ip),
ip_list[i].port, (i == (num_names - 1) ? "" : ",")));
DEBUGADD(5, ("\n"));
}
key = namecache_key(name, name_type);
if (!key) {
return False;
}
expiry = time(NULL) + lp_name_cache_timeout();
if (!ipstr_list_make(&value_string, ip_list, num_names)) {
SAFE_FREE(key);
SAFE_FREE(value_string);
return False;
}
ret = gencache_set(key, value_string, expiry);
SAFE_FREE(key);
SAFE_FREE(value_string);
return ret;
}
BOOL namecache_fetch(const char *name, int name_type, struct ip_service **ip_list,
int *num_names)
{
char *key, *value;
time_t timeout;
if (!ip_list || !num_names) return False;
if (!gencache_init())
return False;
if (name_type > 255) {
return False;
}
*num_names = 0;
key = namecache_key(name, name_type);
if (!key) {
return False;
}
if (!gencache_get(key, &value, &timeout)) {
DEBUG(5, ("no entry for %s#%02X found.\n", name, name_type));
SAFE_FREE(key);
return False;
} else {
DEBUG(5, ("name %s#%02X found.\n", name, name_type));
}
*num_names = ipstr_list_parse(value, ip_list);
SAFE_FREE(key);
SAFE_FREE(value);
return *num_names > 0;
}
BOOL namecache_delete(const char *name, int name_type)
{
BOOL ret;
char *key;
if (!gencache_init())
return False;
if (name_type > 255) {
return False;
}
key = namecache_key(name, name_type);
if (!key) {
return False;
}
ret = gencache_del(key);
SAFE_FREE(key);
return ret;
}
static void flush_netbios_name(const char* key, const char *value, time_t timeout, void* dptr)
{
gencache_del(key);
DEBUG(5, ("Deleting entry %s\n", key));
}
void namecache_flush(void)
{
if (!gencache_init())
return;
gencache_iterate(flush_netbios_name, NULL, "NBT/*");
DEBUG(5, ("Namecache flushed\n"));
}
static char *namecache_status_record_key(const char *name, int name_type1,
int name_type2, struct in_addr keyip)
{
char *keystr;
asprintf(&keystr, "NBT/%s#%02X.%02X.%s",
strupper_static(name), name_type1, name_type2, inet_ntoa(keyip));
return keystr;
}
BOOL namecache_status_store(const char *keyname, int keyname_type,
int name_type, struct in_addr keyip,
const char *srvname)
{
char *key;
time_t expiry;
BOOL ret;
if (!gencache_init())
return False;
key = namecache_status_record_key(keyname, keyname_type, name_type, keyip);
if (!key)
return False;
expiry = time(NULL) + lp_name_cache_timeout();
ret = gencache_set(key, srvname, expiry);
if (ret)
DEBUG(5, ("namecache_status_store: entry %s -> %s\n", key, srvname ));
else
DEBUG(5, ("namecache_status_store: entry %s store failed.\n", key ));
SAFE_FREE(key);
return ret;
}
BOOL namecache_status_fetch(const char *keyname, int keyname_type,
int name_type, struct in_addr keyip, char *srvname_out)
{
char *key = NULL;
char *value = NULL;
time_t timeout;
if (!gencache_init())
return False;
key = namecache_status_record_key(keyname, keyname_type, name_type, keyip);
if (!key)
return False;
if (!gencache_get(key, &value, &timeout)) {
DEBUG(5, ("namecache_status_fetch: no entry for %s found.\n", key));
SAFE_FREE(key);
return False;
} else {
DEBUG(5, ("namecache_status_fetch: key %s -> %s\n", key, value ));
}
strlcpy(srvname_out, value, 16);
SAFE_FREE(key);
SAFE_FREE(value);
return True;
}