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); }