From 462d413d2e8ef4e8196ac99c142dd165086f7b69 Mon Sep 17 00:00:00 2001
From: Jonathan Lebon <jonathan@jlebon.com>
Date: Thu, 13 Apr 2023 17:22:42 -0400
Subject: libotutil: add utility functions for calculating directory size

Prep for future patch.
---
 src/libotutil/ot-fs-utils.c | 44 ++++++++++++++++++++++++++++++++++++++++++++
 src/libotutil/ot-fs-utils.h |  7 +++++++
 2 files changed, 51 insertions(+)

diff --git a/src/libotutil/ot-fs-utils.c b/src/libotutil/ot-fs-utils.c
index 64ebc966..935b1bbc 100644
--- a/src/libotutil/ot-fs-utils.c
+++ b/src/libotutil/ot-fs-utils.c
@@ -245,3 +245,47 @@ ot_parse_file_by_line (const char    *path,
 
   return TRUE;
 }
+
+/* Calculate the size of the files contained in a directory. Symlinks are not
+ * followed. */
+gboolean
+ot_get_dir_size (int            dfd,
+                 const char    *path,
+                 guint64       *out_size,
+                 GCancellable  *cancellable,
+                 GError       **error)
+{
+  g_auto(GLnxDirFdIterator) dfd_iter = { 0, };
+  if (!glnx_dirfd_iterator_init_at (dfd, path, FALSE, &dfd_iter, error))
+    return FALSE;
+
+  *out_size = 0;
+  while (TRUE)
+    {
+      struct dirent *dent;
+      if (!glnx_dirfd_iterator_next_dent_ensure_dtype (&dfd_iter, &dent, cancellable, error))
+        return FALSE;
+
+      if (dent == NULL)
+        break;
+
+      if (dent->d_type == DT_REG)
+        {
+          struct stat stbuf;
+          if (!glnx_fstatat (dfd_iter.fd, dent->d_name, &stbuf, AT_SYMLINK_NOFOLLOW, error))
+            return FALSE;
+
+          *out_size += stbuf.st_size;
+        }
+      else if (dent->d_type == DT_DIR)
+        {
+          guint64 subdir_size;
+          if (!ot_get_dir_size (dfd_iter.fd, dent->d_name, &subdir_size, cancellable, error))
+            return FALSE;
+
+          *out_size += subdir_size;
+        }
+    }
+
+  return TRUE;
+}
diff --git a/src/libotutil/ot-fs-utils.h b/src/libotutil/ot-fs-utils.h
index fad4c53d..f719381d 100644
--- a/src/libotutil/ot-fs-utils.h
+++ b/src/libotutil/ot-fs-utils.h
@@ -95,4 +95,11 @@ ot_parse_file_by_line (const char    *path,
                        GCancellable  *cancellable,
                        GError       **error);
 
+gboolean
+ot_get_dir_size (int            dfd,
+                 const char    *path,
+                 guint64       *out_size,
+                 GCancellable  *cancellable,
+                 GError       **error);
+
 G_END_DECLS
-- 
cgit v1.2.1