#include "includes.h"
static TDB_CONTEXT *tdb_stat_cache;
void stat_cache_add( const char *full_orig_name, const char *orig_translated_path, BOOL case_sensitive)
{
char *translated_path;
size_t translated_path_length;
TDB_DATA data_val;
char *original_path;
size_t original_path_length;
if (!lp_stat_cache())
return;
ZERO_STRUCT(data_val);
if((*full_orig_name == '\0') || (full_orig_name[0] == '.' &&
((full_orig_name[1] == '\0') ||
(full_orig_name[1] == '.' && full_orig_name[1] == '\0'))))
return;
if(case_sensitive && (strcmp(full_orig_name, orig_translated_path) == 0))
return;
translated_path = SMB_STRDUP(orig_translated_path);
if (!translated_path)
return;
translated_path_length = strlen(translated_path);
if(translated_path[translated_path_length-1] == '/') {
translated_path[translated_path_length-1] = '\0';
translated_path_length--;
}
if(case_sensitive) {
original_path = SMB_STRDUP(full_orig_name);
} else {
original_path = strdup_upper(full_orig_name);
}
if (!original_path) {
SAFE_FREE(translated_path);
return;
}
original_path_length = strlen(original_path);
if(original_path[original_path_length-1] == '/') {
original_path[original_path_length-1] = '\0';
original_path_length--;
}
if (original_path_length != translated_path_length) {
if (original_path_length < translated_path_length) {
DEBUG(0, ("OOPS - tried to store stat cache entry for weird length paths [%s] %lu and [%s] %lu)!\n",
original_path, (unsigned long)original_path_length, translated_path, (unsigned long)translated_path_length));
SAFE_FREE(original_path);
SAFE_FREE(translated_path);
return;
}
original_path[translated_path_length] = '\0';
original_path_length = translated_path_length;
}
data_val.dsize = translated_path_length + 1;
data_val.dptr = translated_path;
if (tdb_store_bystring(tdb_stat_cache, original_path, data_val, TDB_REPLACE) != 0) {
DEBUG(0,("stat_cache_add: Error storing entry %s -> %s\n", original_path, translated_path));
} else {
DEBUG(5,("stat_cache_add: Added entry (%x:size%x) %s -> %s\n",
(unsigned int)data_val.dptr, data_val.dsize, original_path, translated_path));
}
SAFE_FREE(original_path);
SAFE_FREE(translated_path);
}
BOOL stat_cache_lookup(connection_struct *conn, pstring name, pstring dirpath,
char **start, SMB_STRUCT_STAT *pst)
{
char *chk_name;
size_t namelen;
BOOL sizechanged = False;
unsigned int num_components = 0;
if (!lp_stat_cache())
return False;
namelen = strlen(name);
*start = name;
DO_PROFILE_INC(statcache_lookups);
if((*name == '\0') || (name[0] == '.' &&
((name[1] == '\0') ||
(name[1] == '.' && name[1] == '\0'))))
return False;
if (conn->case_sensitive) {
chk_name = SMB_STRDUP(name);
if (!chk_name) {
DEBUG(0, ("stat_cache_lookup: strdup failed!\n"));
return False;
}
} else {
chk_name = strdup_upper(name);
if (!chk_name) {
DEBUG(0, ("stat_cache_lookup: strdup_upper failed!\n"));
return False;
}
if (strlen(chk_name) != namelen)
sizechanged = True;
}
while (1) {
TDB_DATA data_val;
char *sp;
data_val = tdb_fetch_bystring(tdb_stat_cache, chk_name);
if(data_val.dptr == NULL || data_val.dsize == 0) {
DEBUG(10,("stat_cache_lookup: lookup failed for name [%s]\n", chk_name ));
sp = strrchr_m(chk_name, '/');
if (sp) {
*sp = '\0';
if (sizechanged)
num_components++;
} else {
DO_PROFILE_INC(statcache_misses);
SAFE_FREE(chk_name);
return False;
}
if((*chk_name == '\0') || (strcmp(chk_name, ".") == 0)
|| (strcmp(chk_name, "..") == 0)) {
DO_PROFILE_INC(statcache_misses);
SAFE_FREE(chk_name);
return False;
}
} else {
BOOL retval;
char *translated_path = data_val.dptr;
size_t translated_path_length = data_val.dsize - 1;
DEBUG(10,("stat_cache_lookup: lookup succeeded for name [%s] -> [%s]\n", chk_name, translated_path ));
DO_PROFILE_INC(statcache_hits);
if(SMB_VFS_STAT(conn,translated_path, pst) != 0) {
tdb_delete_bystring(tdb_stat_cache, chk_name);
SAFE_FREE(chk_name);
SAFE_FREE(data_val.dptr);
return False;
}
if (!sizechanged) {
memcpy(name, translated_path, MIN(sizeof(pstring)-1, translated_path_length));
} else if (num_components == 0) {
pstrcpy(name, translated_path);
} else {
sp = strnrchr_m(name, '/', num_components);
if (sp) {
pstring last_component;
pstrcpy(last_component, sp);
pstrcpy(name, translated_path);
pstrcat(name, last_component);
} else {
pstrcpy(name, translated_path);
}
}
*start = &name[translated_path_length];
if(**start == '/')
++*start;
pstrcpy(dirpath, translated_path);
retval = (namelen == translated_path_length) ? True : False;
SAFE_FREE(chk_name);
SAFE_FREE(data_val.dptr);
return retval;
}
}
}
u32 fast_string_hash(TDB_DATA *key)
{
u32 n = 0;
const char *p;
for (p = key->dptr; *p != '\0'; p++) {
n = ((n << 5) + n) ^ (u32)(*p);
}
return n;
}
BOOL reset_stat_cache( void )
{
if (!lp_stat_cache())
return True;
if (tdb_stat_cache) {
tdb_close(tdb_stat_cache);
}
tdb_stat_cache = tdb_open_ex("statcache", 1031, TDB_INTERNAL,
(O_RDWR|O_CREAT), 0644, NULL, fast_string_hash);
if (!tdb_stat_cache)
return False;
return True;
}