From 2831c51d3386d12a97a6ce5f3d795cb51e02beec Mon Sep 17 00:00:00 2001 From: David Oberhollenzer Date: Thu, 20 Apr 2023 18:43:43 +0200 Subject: Collect and print statistics about the kind of files we are packing Using depth-first search, we collect some crude statistics about directory tree types (e.g. regular files, directories, device special files and so on) and print them out after serializing the tree. Signed-off-by: David Oberhollenzer --- lib/fstree/src/stats.c | 49 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) create mode 100644 lib/fstree/src/stats.c (limited to 'lib/fstree/src') diff --git a/lib/fstree/src/stats.c b/lib/fstree/src/stats.c new file mode 100644 index 0000000..2dbf043 --- /dev/null +++ b/lib/fstree/src/stats.c @@ -0,0 +1,49 @@ +/* SPDX-License-Identifier: GPL-3.0-or-later */ +/* + * stats.c + * + * Copyright (C) 2023 David Oberhollenzer + */ +#include "config.h" + +#include "fstree.h" +#include + +static void count_dfs(const tree_node_t *n, fstree_stats_t *out) +{ + switch (n->mode & S_IFMT) { + case S_IFSOCK: + case S_IFIFO: + out->num_ipc += 1; + break; + case S_IFLNK: + if (n->flags & FLAG_LINK_IS_HARD) { + out->num_links += 1; + } else { + out->num_slinks += 1; + } + break; + case S_IFREG: + out->num_files += 1; + break; + case S_IFBLK: + case S_IFCHR: + out->num_devices += 1; + break; + case S_IFDIR: + out->num_dirs += 1; + for (n = n->data.children; n != NULL; n = n->next) { + count_dfs(n, out); + } + break; + default: + break; + } +} + +void fstree_collect_stats(const fstree_t *fs, fstree_stats_t *out) +{ + memset(out, 0, sizeof(*out)); + + count_dfs(fs->root, out); +} -- cgit v1.2.3