diff options
Diffstat (limited to 'miscutils/tree.c')
-rw-r--r-- | miscutils/tree.c | 118 |
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 | |||
25 | static 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 | |||
100 | int tree_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
101 | int 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 | } | ||