diff options
Diffstat (limited to 'src/libutil')
-rw-r--r-- | src/libutil/util.cc | 23 | ||||
-rw-r--r-- | src/libutil/util.hh | 3 |
2 files changed, 26 insertions, 0 deletions
diff --git a/src/libutil/util.cc b/src/libutil/util.cc index 38e32bfa9b58..f80fb7dbf766 100644 --- a/src/libutil/util.cc +++ b/src/libutil/util.cc @@ -238,6 +238,29 @@ void writeFile(const Path & path, const string & s) } +unsigned long long computePathSize(const Path & path) +{ + unsigned long long size = 0; + + checkInterrupt(); + + struct stat st; + if (lstat(path.c_str(), &st)) + throw SysError(format("getting attributes of path `%1%'") % path); + + size += st.st_size; + + if (S_ISDIR(st.st_mode)) { + Strings names = readDirectory(path); + + for (Strings::iterator i = names.begin(); i != names.end(); ++i) + size += computePathSize(path + "/" + *i); + } + + return size; +} + + static void _deletePath(const Path & path, unsigned long long & bytesFreed) { checkInterrupt(); diff --git a/src/libutil/util.hh b/src/libutil/util.hh index 52ef2c6eb20e..7f3d41e76ef3 100644 --- a/src/libutil/util.hh +++ b/src/libutil/util.hh @@ -56,6 +56,9 @@ string readFile(const Path & path); /* Write a string to a file. */ void writeFile(const Path & path, const string & s); +/* Compute the sum of the sizes of all files in `path'. */ +unsigned long long computePathSize(const Path & path); + /* Delete a path; i.e., in the case of a directory, it is deleted recursively. Don't use this at home, kids. The second variant returns the number of bytes freed. */ |