*/
#include <stdlib.h>
#include <stdio.h>
-#include <malloc.h>
#include <wchar.h>
#include <string.h>
#include <errno.h>
#include "log.h"
#include "utils.h"
#include "module.h"
-#include "tiger.h"
#include "net.h"
#include "sysevents.h"
+#include <tiger.h>
struct scanstate
{
static int conf_share(int argc, wchar_t **argv);
static void freecache(struct sharecache *node);
static void checkhashes(void);
+static void writehashcache(int now);
static struct configvar myvars[] =
{
+ /** The default nick name to use. The nickname can also be
+ * specified for individual hubs, overriding this setting. */
{CONF_VAR_STRING, "defnick", {.str = L"DoldaConnect user"}},
+ /** When scanning shares, this bitmask is consulted for every
+ * regular file. Unless the file's mode has the bits specified by
+ * this mask set, it will not be shared. */
{CONF_VAR_INT, "scanfilemask", {.num = 0004}},
+ /** When scanning shares, this bitmask is consulted for every
+ * directory encountered. Unless the directory's mode has the bits
+ * specified by this mask set, it will be ignored and any files
+ * under it will not be shared. */
{CONF_VAR_INT, "scandirmask", {.num = 0005}},
+ /** The filename to use for the hash cache (see the FILES section
+ * for more information). */
{CONF_VAR_STRING, "hashcache", {.str = L"dc-hashcache"}},
+ /** Writes of the hash cache and file lists are delayed for an
+ * amount of time, in order to minimize the time spent on I/O wait
+ * while hashing many small files. This variable sets the amount
+ * of time, in seconds. */
+ {CONF_VAR_INT, "hashwritedelay", {.num = 300}},
+ /** The amount of time, in seconds, to wait before automatically
+ * rescanning the shared directories for changes. Set to zero (the
+ * default) to disable automatic rescanning. */
+ {CONF_VAR_INT, "rescandelay", {.num = 0}},
{CONF_VAR_END}
};
static struct scanqueue *scanqueue = NULL;
static struct sharepoint *shares = NULL;
static struct hashcache *hashcache = NULL;
+static struct timer *hashwritetimer = NULL;
/* Set initially to -1, but changed to 0 the first time run() is
* called. This is to avoid forking a hash job before daemonizing,
* since that would make the daemon unable to wait() for the hash
* job. */
static pid_t hashjob = -1;
struct sharecache *shareroot = NULL;
+static struct timer *scantimer = NULL;
unsigned long long sharesize = 0;
GCBCHAIN(sharechangecb, unsigned long long);
free(hc);
}
-static char *findhashcachefile(int filldef)
-{
- static char ret[128];
- char *hcname;
-
- if(getenv("HOME") != NULL)
- {
- snprintf(ret, sizeof(ret), "%s/.dc-hashcache", getenv("HOME"));
- if(!access(ret, R_OK))
- return(ret);
- }
- if((hcname = icswcstombs(confgetstr("cli", "hashcache"), NULL, NULL)) == NULL)
- {
- flog(LOG_WARNING, "could not convert hash cache name into local charset: %s", strerror(errno));
- return(NULL);
- }
- if(strchr(hcname, '/') != NULL)
- {
- if(!access(hcname, R_OK))
- {
- strcpy(ret, hcname);
- return(ret);
- }
- } else {
- snprintf(ret, sizeof(ret), "/etc/%s", hcname);
- if(!access(ret, R_OK))
- return(ret);
- snprintf(ret, sizeof(ret), "/usr/etc/%s", hcname);
- if(!access(ret, R_OK))
- return(ret);
- snprintf(ret, sizeof(ret), "/usr/local/etc/%s", hcname);
- if(!access(ret, R_OK))
- return(ret);
- }
- if(filldef)
- {
- if(getenv("HOME") != NULL)
- snprintf(ret, sizeof(ret), "%s/.dc-hashcache", getenv("HOME"));
- else
- snprintf(ret, sizeof(ret), "/etc/%s", hcname);
- return(ret);
- } else {
- return(NULL);
- }
-}
-
static struct hashcache *findhashcache(dev_t dev, ino_t inode)
{
struct hashcache *hc;
struct hashcache *hc;
size_t len;
- if((hcname = findhashcachefile(0)) == NULL)
+ if((hcname = findfile(icswcstombs(confgetstr("cli", "hashcache"), NULL, NULL), NULL, 0)) == NULL)
return;
if((stream = fopen(hcname, "r")) == NULL)
{
fclose(stream);
}
-static void writehashcache(void)
+static void hashtimercb(int cancelled, void *uudata)
+{
+ hashwritetimer = NULL;
+ if(!cancelled)
+ writehashcache(1);
+}
+
+static void writehashcache(int now)
{
char *buf;
char *hcname;
FILE *stream;
struct hashcache *hc;
- hcname = findhashcachefile(1);
+ if(!now)
+ {
+ if(hashwritetimer == NULL)
+ hashwritetimer = timercallback(ntime() + confgetint("cli", "hashwritedelay"), (void (*)(int, void *))hashtimercb, NULL);
+ return;
+ }
+ if(hashwritetimer != NULL)
+ canceltimer(hashwritetimer);
+ hcname = findfile(icswcstombs(confgetstr("cli", "hashcache"), NULL, NULL), NULL, 1);
if((stream = fopen(hcname, "w")) == NULL)
{
flog(LOG_WARNING, "could not write hash cache %s: %s", hcname, strerror(errno));
for(hc = hashcache; hc != NULL; hc = hc->next)
{
buf = base64encode(hc->tth, 24);
- fprintf(stream, "%lli %lli %li tth %s\n", hc->dev, (long long)hc->inode, hc->mtime, buf);
+ fprintf(stream, "%lli %lli %li tth %s\n", (long long)hc->dev, (long long)hc->inode, hc->mtime, buf);
free(buf);
}
fclose(stream);
buf = base64decode(wv[3], NULL);
memcpy(hc->tth, buf, 24);
free(buf);
- writehashcache();
+ writehashcache(0);
}
memmove(hashbuf, lp, hashbufdata -= (lp - hashbuf));
}
}
-static void hashexit(pid_t pid, int status, void *uudata)
+static void hashexit(pid_t pid, int status, struct socket *outsock)
{
if(pid != hashjob)
flog(LOG_ERR, "BUG: hashing process changed PID?! old: %i new %i", hashjob, pid);
flog(LOG_WARNING, "hashing process exited with non-zero status: %i", status);
hashjob = 0;
checkhashes();
+ putsock(outsock);
}
static int hashfile(char *path)
}
synctigertree(&tth);
restigertree(&tth, digest);
- ret = snprintf(buf, sizeof(buf), "%lli %lli %li %s\n", sb.st_dev, (long long)sb.st_ino, sb.st_mtime, base64encode(digest, 24));
+ ret = snprintf(buf, sizeof(buf), "%lli %lli %li %s\n", (long long)sb.st_dev, (long long)sb.st_ino, sb.st_mtime, base64encode(digest, 24));
write(1, buf, ret);
exit(0);
}
close(pfd[1]);
outsock = wrapsock(pfd[0]);
outsock->readcb = hashread;
- childcallback(hashjob, hashexit, NULL);
+ childcallback(hashjob, (void (*)(pid_t, int, void *))hashexit, outsock);
return(0);
}
char *path;
node = shareroot->child;
- while(1)
+ for(node = shareroot->child; node != NULL; node = nextscnode(node))
{
- if(node->child != NULL)
- {
- node = node->child;
+ if(node->f.b.type != FILE_REG)
continue;
- }
if(!node->f.b.hastth)
{
- if((hc = findhashcache(node->dev, node->inode)) != NULL)
+ if(((hc = findhashcache(node->dev, node->inode)) != NULL) && (hc->mtime == node->mtime))
{
memcpy(node->hashtth, hc->tth, 24);
node->f.b.hastth = 1;
{
flog(LOG_WARNING, "could not hash %s, unsharing it", path);
freecache(node);
+ free(path);
+ flog(LOG_INFO, "sharing %lli bytes", sharesize);
+ continue;
}
free(path);
return;
}
}
- while(node->next == NULL)
- {
- if((node = node->parent) == shareroot)
- break;
- }
- if(node == shareroot)
- break;
- node = node->next;
}
}
}
type = FILE_REG;
} else {
- flog(LOG_WARNING, "unhandled file type: %i", sb.st_mode);
+ flog(LOG_WARNING, "unhandled file type: 0%o", sb.st_mode);
free(wcs);
continue;
}
return(1);
}
+static void rescancb(int cancelled, void *uudata)
+{
+ scantimer = NULL;
+ if(!cancelled)
+ {
+ if(scanqueue == NULL)
+ scanshares();
+ else if(confgetint("cli", "rescandelay") > 0)
+ scantimer = timercallback(ntime() + confgetint("cli", "rescandelay"), (void (*)(int, void *))rescancb, NULL);
+ }
+}
+
void scanshares(void)
{
struct sharepoint *cur;
}
queuescan(node);
}
+ if(scantimer != NULL)
+ canceltimer(scantimer);
+ if(confgetint("cli", "rescandelay") > 0)
+ scantimer = timercallback(ntime() + confgetint("cli", "rescandelay"), (void (*)(int, void *))rescancb, NULL);
}
static void preinit(int hup)
}
}
+static int rsdelayupdate(struct configvar *var, void *uudata)
+{
+ if(scantimer != NULL)
+ canceltimer(scantimer);
+ if(confgetint("cli", "rescandelay") > 0)
+ scantimer = timercallback(ntime() + var->val.num, (void (*)(int, void *))rescancb, NULL);
+ return(0);
+}
+
static int init(int hup)
{
struct sharepoint *cur, *next;
}
scanshares();
if(!hup)
+ {
while(doscan(100));
+ CBREG(confgetvar("cli", "rescandelay"), conf_update, rsdelayupdate, NULL, NULL);
+ }
return(0);
}