aboutsummaryrefslogtreecommitdiff
path: root/miscutils/tree.c
diff options
context:
space:
mode:
Diffstat (limited to 'miscutils/tree.c')
-rw-r--r--miscutils/tree.c118
1 files changed, 118 insertions, 0 deletions
diff --git a/miscutils/tree.c b/miscutils/tree.c
new file mode 100644
index 000000000..8b16c5383
--- /dev/null
+++ b/miscutils/tree.c
@@ -0,0 +1,118 @@
1/* vi: set sw=4 ts=4: */
2/*
3 * Copyright (C) 2022 Roger Knecht <rknecht@pm.me>
4 *
5 * Licensed under GPLv2, see file LICENSE in this source tree.
6 */
7//config:config TREE
8//config: bool "tree (0.6 kb)"
9//config: default y
10//config: help
11//config: List files and directories in a tree structure.
12
13//applet:IF_TREE(APPLET(tree, BB_DIR_USR_BIN, BB_SUID_DROP))
14
15//kbuild:lib-$(CONFIG_TREE) += tree.o
16
17//usage:#define tree_trivial_usage NOUSAGE_STR
18//usage:#define tree_full_usage ""
19
20#include "libbb.h"
21#include "common_bufsiz.h"
22
23#define prefix_buf bb_common_bufsiz1
24
25static void tree_print(unsigned count[2], const char* directory_name, char* prefix_pos)
26{
27 struct dirent **entries;
28 int index, size;
29
30 // read directory entries
31 size = scandir(directory_name, &entries, NULL, alphasort);
32
33 if (size < 0) {
34 fputs_stdout(directory_name);
35 puts(" [error opening dir]");
36 return;
37 }
38
39 // print directory name
40 puts(directory_name);
41
42 // switch to sub directory
43 xchdir(directory_name);
44
45 // print all directory entries
46 for (index = 0; index < size;) {
47 struct dirent *dirent = entries[index++];
48
49 // filter hidden files and directories
50 if (dirent->d_name[0] != '.') {
51 int status;
52 struct stat statBuf;
53
54//TODO: when -l is implemented, use stat, not lstat, if -l
55 status = lstat(dirent->d_name, &statBuf);
56
57 if (index == size) {
58 strcpy(prefix_pos, "└── ");
59 } else {
60 strcpy(prefix_pos, "├── ");
61 }
62 fputs_stdout(prefix_buf);
63
64 if (status == 0 && S_ISLNK(statBuf.st_mode)) {
65 // handle symlink
66 char* symlink_path = xmalloc_readlink(dirent->d_name);
67 printf("%s -> %s\n", dirent->d_name, symlink_path);
68 free(symlink_path);
69 count[1]++;
70 } else if (status == 0 && S_ISDIR(statBuf.st_mode)
71 && (prefix_pos - prefix_buf) < (COMMON_BUFSIZE - 16)
72 ) {
73 // handle directory
74 char* pos;
75 if (index == size) {
76 pos = stpcpy(prefix_pos, " ");
77 } else {
78 pos = stpcpy(prefix_pos, "│   ");
79 }
80 tree_print(count, dirent->d_name, pos);
81 count[0]++;
82 } else {
83 // handle file
84 puts(dirent->d_name);
85 count[1]++;
86 }
87 }
88
89 // release directory entry
90 free(dirent);
91 }
92
93 // release directory array
94 free(entries);
95
96 // switch to parent directory
97 xchdir("..");
98}
99
100int tree_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
101int tree_main(int argc UNUSED_PARAM, char **argv)
102{
103 unsigned count[2] = { 0, 0 };
104
105 setup_common_bufsiz();
106
107 if (!argv[1])
108 *argv-- = (char*)".";
109
110 // list directories given as command line arguments
111 while (*(++argv))
112 tree_print(count, *argv, prefix_buf);
113
114 // print statistic
115 printf("\n%u directories, %u files\n", count[0], count[1]);
116
117 return EXIT_SUCCESS;
118}