tris/recoll
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fsocc

Browse Source
This commit is contained in:
dockes 2007-06-08 15:30:01 +00:00
parent 61004757c4
commit 20f5363a70
2 changed files with 66 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
src/utils/pathut.cpp
View File

@ -1,5 +1,5 @@
#ifndef lint #ifndef lint
static char rcsid[] = "@(#$Id: pathut.cpp,v 1.15 2007-02-06 14:16:43 dockes Exp $ (C) 2004 J.F.Dockes"; static char rcsid[] = "@(#$Id: pathut.cpp,v 1.16 2007-06-08 15:30:01 dockes Exp $ (C) 2004 J.F.Dockes";
#endif #endif
/* /*
* This program is free software; you can redistribute it and/or modify * This program is free software; you can redistribute it and/or modify
@ -23,6 +23,7 @@ static char rcsid[] = "@(#$Id: pathut.cpp,v 1.15 2007-02-06 14:16:43 dockes Exp
#include <fcntl.h> #include <fcntl.h>
#include <sys/param.h> #include <sys/param.h>
#include <pwd.h> #include <pwd.h>
#include <math.h>
#include <iostream> #include <iostream>
#include <list> #include <list>
@ -35,6 +36,40 @@ using std::stack;
#include "pathut.h" #include "pathut.h"
#ifndef STATFS_INCLUDE
#define STATFS_INCLUDE <sys/vfs.h>
#endif
#include STATFS_INCLUDE
bool fsocc(const string &path, int *pc, long *blocks)
{
struct statfs buf;
if (statfs(path.c_str(), &buf) != 0) {
return false;
}
// used blocks
double fpc = 0.0;
#define FSOCC_USED (double(buf.f_blocks - buf.f_bfree))
#define FSOCC_TOTAVAIL (FSOCC_USED + double(buf.f_bavail))
if (FSOCC_TOTAVAIL > 0) {
fpc = 100.0 * FSOCC_USED / FSOCC_TOTAVAIL;
}
*pc = int(fpc);
if (blocks) {
*blocks = 0;
#define FSOCC_MB (1024*1024)
if (buf.f_bsize > 0) {
int ratio = buf.f_bsize > FSOCC_MB ? buf.f_bsize / FSOCC_MB :
FSOCC_MB / buf.f_bsize;
*blocks = buf.f_bsize > FSOCC_MB ? long(buf.f_bavail) * ratio :
long(buf.f_bavail) / ratio;
}
}
return true;
}
static const char *tmplocation() static const char *tmplocation()
{ {
const char *tmpdir = getenv("RECOLL_TMPDIR"); const char *tmpdir = getenv("RECOLL_TMPDIR");
@ -352,8 +387,12 @@ const string ttvec[] = {"/dir", "", "~", "~/sub", "~root", "~root/sub",
"~nosuch", "~nosuch/sub"}; "~nosuch", "~nosuch/sub"};
int nttvec = sizeof(ttvec) / sizeof(string); int nttvec = sizeof(ttvec) / sizeof(string);
const char *thisprog;
int main(int argc, const char **argv) int main(int argc, const char **argv)
{ {
thisprog = *argv++;argc--;
string s; string s;
list<string>::const_iterator it; list<string>::const_iterator it;
#if 0 #if 0
@ -388,18 +427,34 @@ int main(int argc, const char **argv)
path_canon(canontst[i]) << "'" << endl; path_canon(canontst[i]) << "'" << endl;
} }
#endif #endif
#if 1 #if 0
if (argc != 3) { if (argc != 2) {
fprintf(stderr, "Usage: trpathut <dir> <pattern>\n"); fprintf(stderr, "Usage: trpathut <dir> <pattern>\n");
exit(1); exit(1);
} }
string dir=argv[1], pattern=argv[2]; string dir = *argv++;argc--;
string pattern = *argv++;argc--;
list<string> matched = path_dirglob(dir, pattern); list<string> matched = path_dirglob(dir, pattern);
for (it = matched.begin(); it != matched.end();it++) { for (it = matched.begin(); it != matched.end();it++) {
cout << *it << endl; cout << *it << endl;
} }
#endif #endif
#if 1
if (argc != 1) {
fprintf(stderr, "Usage: fsocc: trpathut <path>\n");
exit(1);
}
string path = *argv++;argc--;
int pc;
long blocks;
if (!fsocc(path, &pc, &blocks)) {
fprintf(stderr, "fsocc failed\n");
return 1;
}
printf("pc %d, megabytes %ld\n", pc, blocks);
#endif
return 0; return 0;
} }

8
src/utils/pathut.h
View File

@ -16,7 +16,7 @@
*/ */
#ifndef _PATHUT_H_INCLUDED_ #ifndef _PATHUT_H_INCLUDED_
#define _PATHUT_H_INCLUDED_ #define _PATHUT_H_INCLUDED_
/* @(#$Id: pathut.h,v 1.12 2007-02-06 14:16:43 dockes Exp $ (C) 2004 J.F.Dockes */ /* @(#$Id: pathut.h,v 1.13 2007-06-08 15:30:01 dockes Exp $ (C) 2004 J.F.Dockes */
#include <string> #include <string>
#include <list> #include <list>
@ -55,6 +55,12 @@ extern string url_encode(const string url,
/// Stat parameter and check if it's a directory /// Stat parameter and check if it's a directory
extern bool path_isdir(const string& path); extern bool path_isdir(const string& path);
/** A small wrapper around statfs et al, to return percentage of disk
occupation */
bool fsocc(const string &path, int *pc, // Percent occupied
long *avmbs = 0 // Mbs available to non-superuser
);
/// Create temporary directory /// Create temporary directory
extern bool maketmpdir(string& tdir, string& reason); extern bool maketmpdir(string& tdir, string& reason);