SMB_BIG_UINT get_dfree_info(connection_struct *conn,
const char *path,
bool small_query,
SMB_BIG_UINT *bsize,
SMB_BIG_UINT *dfree,
SMB_BIG_UINT *dsize)
{
int dfree_cache_time = lp_dfree_cache_time(SNUM(conn));
struct dfree_cached_info *dfc = conn->dfree_info;
SMB_BIG_UINT dfree_ret;
if (!dfree_cache_time) {
return SMB_VFS_DISK_FREE(conn,path,small_query,bsize,dfree,dsize);
}
if (dfc && (conn->lastused - dfc->last_dfree_time < dfree_cache_time)) {
/* Return cached info. */
*bsize = dfc->bsize;
*dfree = dfc->dfree;
*dsize = dfc->dsize;
return dfc->dfree_ret;
}
dfree_ret = SMB_VFS_DISK_FREE(conn,path,small_query,bsize,dfree,dsize);
if (dfree_ret == (SMB_BIG_UINT)-1) {
/* Don't cache bad data. */
return dfree_ret;
}
/* No cached info or time to refresh. */
if (!dfc) {
dfc = TALLOC_P(conn->mem_ctx, struct dfree_cached_info);
if (!dfc) {
return dfree_ret;
}
conn->dfree_info = dfc;
}
static NTSTATUS cmd_disk_free(struct vfs_state *vfs, TALLOC_CTX *mem_ctx, int argc, const char **argv)
{
SMB_BIG_UINT diskfree, bsize, dfree, dsize;
if (argc != 2) {
printf("Usage: disk_free <path>\n");
return NT_STATUS_OK;
}
diskfree = SMB_VFS_DISK_FREE(vfs->conn, argv[1], False, &bsize, &dfree, &dsize);
printf("disk_free: %lu, bsize = %lu, dfree = %lu, dsize = %lu\n",
(unsigned long)diskfree,
(unsigned long)bsize,
(unsigned long)dfree,
(unsigned long)dsize);
return NT_STATUS_OK;
}
uint64_t sys_disk_free(connection_struct *conn, struct smb_filename *fname,
uint64_t *bsize, uint64_t *dfree, uint64_t *dsize)
{
uint64_t dfree_retval;
uint64_t dfree_q = 0;
uint64_t bsize_q = 0;
uint64_t dsize_q = 0;
const char *dfree_command;
static bool dfree_broken = false;
const char *path = fname->base_name;
(*dfree) = (*dsize) = 0;
(*bsize) = 512;
/*
* If external disk calculation specified, use it.
*/
dfree_command = lp_dfree_command(talloc_tos(), SNUM(conn));
if (dfree_command && *dfree_command) {
const char *p;
char **lines = NULL;
char *syscmd = NULL;
syscmd = talloc_asprintf(talloc_tos(),
"%s %s",
dfree_command,
path);
if (!syscmd) {
return (uint64_t)-1;
}
DEBUG (3, ("disk_free: Running command '%s'\n", syscmd));
lines = file_lines_pload(talloc_tos(), syscmd, NULL);
if (lines != NULL) {
char *line = lines[0];
DEBUG (3, ("Read input from dfree, \"%s\"\n", line));
*dsize = STR_TO_SMB_BIG_UINT(line, &p);
while (p && *p && isspace(*p))
p++;
if (p && *p)
*dfree = STR_TO_SMB_BIG_UINT(p, &p);
while (p && *p && isspace(*p))
p++;
if (p && *p)
*bsize = STR_TO_SMB_BIG_UINT(p, NULL);
else
*bsize = 1024;
TALLOC_FREE(lines);
DEBUG (3, ("Parsed output of dfree, dsize=%u, dfree=%u, bsize=%u\n",
(unsigned int)*dsize, (unsigned int)*dfree, (unsigned int)*bsize));
if (!*dsize)
*dsize = 2048;
if (!*dfree)
*dfree = 1024;
goto dfree_done;
}
DEBUG (0, ("disk_free: file_lines_load() failed for "
"command '%s'. Error was : %s\n",
syscmd, strerror(errno) ));
}
if (SMB_VFS_DISK_FREE(conn, path, bsize, dfree, dsize) ==
(uint64_t)-1) {
DBG_ERR("VFS disk_free failed. Error was : %s\n",
strerror(errno));
return (uint64_t)-1;
}
if (disk_quotas(conn, fname, &bsize_q, &dfree_q, &dsize_q)) {
uint64_t min_bsize = MIN(*bsize, bsize_q);
(*dfree) = (*dfree) * (*bsize) / min_bsize;
(*dsize) = (*dsize) * (*bsize) / min_bsize;
dfree_q = dfree_q * bsize_q / min_bsize;
dsize_q = dsize_q * bsize_q / min_bsize;
(*bsize) = min_bsize;
(*dfree) = MIN(*dfree,dfree_q);
(*dsize) = MIN(*dsize,dsize_q);
}
/* FIXME : Any reason for this assumption ? */
if (*bsize < 256) {
DEBUG(5,("disk_free:Warning: bsize == %d < 256 . Changing to assumed correct bsize = 512\n",(int)*bsize));
*bsize = 512;
}
if ((*dsize)<1) {
if (!dfree_broken) {
DEBUG(0,("WARNING: dfree is broken on this system\n"));
dfree_broken=true;
}
*dsize = 20*1024*1024/(*bsize);
*dfree = MAX(1,*dfree);
}
dfree_done:
disk_norm(bsize, dfree, dsize);
if ((*bsize) < 1024) {
dfree_retval = (*dfree)/(1024/(*bsize));
} else {
dfree_retval = ((*bsize)/1024)*(*dfree);
}
return(dfree_retval);
}
