#include "includes.h"
#define _SMBC_INTERNAL
#include "include/libsmbclient.h"
struct smbc_server_cache {
char *server_name;
char *share_name;
char *workgroup;
char *username;
SMBCSRV *server;
struct smbc_server_cache *next, *prev;
};
static int smbc_add_cached_server(SMBCCTX * context, SMBCSRV * new,
const char * server, const char * share,
const char * workgroup, const char * username)
{
struct smbc_server_cache * srvcache = NULL;
if (!(srvcache = SMB_MALLOC_P(struct smbc_server_cache))) {
errno = ENOMEM;
DEBUG(3, ("Not enough space for server cache allocation\n"));
return 1;
}
ZERO_STRUCTP(srvcache);
srvcache->server = new;
srvcache->server_name = SMB_STRDUP(server);
if (!srvcache->server_name) {
errno = ENOMEM;
goto failed;
}
srvcache->share_name = SMB_STRDUP(share);
if (!srvcache->share_name) {
errno = ENOMEM;
goto failed;
}
srvcache->workgroup = SMB_STRDUP(workgroup);
if (!srvcache->workgroup) {
errno = ENOMEM;
goto failed;
}
srvcache->username = SMB_STRDUP(username);
if (!srvcache->username) {
errno = ENOMEM;
goto failed;
}
DLIST_ADD((context->server_cache), srvcache);
return 0;
failed:
SAFE_FREE(srvcache->server_name);
SAFE_FREE(srvcache->share_name);
SAFE_FREE(srvcache->workgroup);
SAFE_FREE(srvcache->username);
return 1;
}
static SMBCSRV * smbc_get_cached_server(SMBCCTX * context, const char * server,
const char * share, const char * workgroup, const char * user)
{
struct smbc_server_cache * srv = NULL;
for (srv=((struct smbc_server_cache *)context->server_cache);srv;srv=srv->next) {
if (strcmp(server,srv->server_name) == 0 &&
strcmp(share,srv->share_name) == 0 &&
strcmp(workgroup,srv->workgroup) == 0 &&
strcmp(user, srv->username) == 0)
return srv->server;
}
return NULL;
}
static int smbc_remove_cached_server(SMBCCTX * context, SMBCSRV * server)
{
struct smbc_server_cache * srv = NULL;
for (srv=((struct smbc_server_cache *)context->server_cache);srv;srv=srv->next) {
if (server == srv->server) {
DLIST_REMOVE(context->server_cache, srv);
SAFE_FREE(srv->server_name);
SAFE_FREE(srv->share_name);
SAFE_FREE(srv->workgroup);
SAFE_FREE(srv->username);
SAFE_FREE(srv);
return 0;
}
}
return 1;
}
static int smbc_purge_cached(SMBCCTX * context)
{
struct smbc_server_cache * srv;
struct smbc_server_cache * next;
int could_not_purge_all = 0;
for (srv = ((struct smbc_server_cache *) context->server_cache),
next = (srv ? srv->next :NULL);
srv;
srv = next, next = (srv ? srv->next : NULL)) {
if (smbc_remove_unused_server(context, srv->server)) {
could_not_purge_all = 1;
}
}
return could_not_purge_all;
}
int smbc_default_cache_functions(SMBCCTX * context)
{
context->callbacks.add_cached_srv_fn = smbc_add_cached_server;
context->callbacks.get_cached_srv_fn = smbc_get_cached_server;
context->callbacks.remove_cached_srv_fn = smbc_remove_cached_server;
context->callbacks.purge_cached_fn = smbc_purge_cached;
return 0;
}