about summary refs log tree commit diff
path: root/src/libutil
diff options
context:
space:
mode:
Diffstat (limited to 'src/libutil')
-rw-r--r--src/libutil/util.cc23
-rw-r--r--src/libutil/util.hh3
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. */