busybox/procps/pstree.c
Denys Vlasenko 22542eca18 getopt32: remove opt_complementary
function                                             old     new   delta
vgetopt32                                           1318    1392     +74
runsvdir_main                                        703     713     +10
bb_make_directory                                    423     425      +2
collect_cpu                                          546     545      -1
opt_chars                                              3       -      -3
opt_complementary                                      4       -      -4
tftpd_main                                           567     562      -5
ntp_init                                             476     471      -5
zcip_main                                           1266    1256     -10
xxd_main                                             428     418     -10
whois_main                                           140     130     -10
who_main                                             463     453     -10
which_main                                           212     202     -10
wget_main                                           2535    2525     -10
watchdog_main                                        291     281     -10
watch_main                                           222     212     -10
vlock_main                                           399     389     -10
uuencode_main                                        332     322     -10
uudecode_main                                        316     306     -10
unlink_main                                           45      35     -10
udhcpd_main                                         1482    1472     -10
udhcpc_main                                         2762    2752     -10
tune2fs_main                                         290     280     -10
tunctl_main                                          366     356     -10
truncate_main                                        218     208     -10
tr_main                                              518     508     -10
time_main                                           1134    1124     -10
tftp_main                                            286     276     -10
telnetd_main                                        1873    1863     -10
tcpudpsvd_main                                      1785    1775     -10
taskset_main                                         521     511     -10
tar_main                                            1009     999     -10
tail_main                                           1644    1634     -10
syslogd_main                                        1967    1957     -10
switch_root_main                                     368     358     -10
svlogd_main                                         1454    1444     -10
sv                                                  1296    1286     -10
stat_main                                            104      94     -10
start_stop_daemon_main                              1028    1018     -10
split_main                                           542     532     -10
sort_main                                            796     786     -10
slattach_main                                        624     614     -10
shuf_main                                            504     494     -10
setsid_main                                           96      86     -10
setserial_main                                      1132    1122     -10
setfont_main                                         388     378     -10
setconsole_main                                       78      68     -10
sendmail_main                                       1209    1199     -10
sed_main                                             677     667     -10
script_main                                         1077    1067     -10
run_parts_main                                       325     315     -10
rtcwake_main                                         454     444     -10
rm_main                                              175     165     -10
reformime_main                                       119     109     -10
readlink_main                                        123     113     -10
rdate_main                                           246     236     -10
pwdx_main                                            189     179     -10
pstree_main                                          317     307     -10
pscan_main                                           663     653     -10
popmaildir_main                                      818     808     -10
pmap_main                                             80      70     -10
nc_main                                             1042    1032     -10
mv_main                                              558     548     -10
mountpoint_main                                      477     467     -10
mount_main                                          1264    1254     -10
modprobe_main                                        768     758     -10
modinfo_main                                         333     323     -10
mktemp_main                                          200     190     -10
mkswap_main                                          324     314     -10
mkfs_vfat_main                                      1489    1479     -10
microcom_main                                        715     705     -10
md5_sha1_sum_main                                    521     511     -10
man_main                                             867     857     -10
makedevs_main                                       1052    1042     -10
ls_main                                              563     553     -10
losetup_main                                         432     422     -10
loadfont_main                                         89      79     -10
ln_main                                              524     514     -10
link_main                                             75      65     -10
ipcalc_main                                          544     534     -10
iostat_main                                         2397    2387     -10
install_main                                         768     758     -10
id_main                                              480     470     -10
i2cset_main                                         1239    1229     -10
i2cget_main                                          380     370     -10
i2cdump_main                                        1482    1472     -10
i2cdetect_main                                       682     672     -10
hwclock_main                                         406     396     -10
httpd_main                                           741     731     -10
grep_main                                            837     827     -10
getty_main                                          1559    1549     -10
fuser_main                                           297     287     -10
ftpgetput_main                                       345     335     -10
ftpd_main                                           2232    2222     -10
fstrim_main                                          251     241     -10
fsfreeze_main                                         77      67     -10
fsck_minix_main                                     2921    2911     -10
flock_main                                           314     304     -10
flashcp_main                                         740     730     -10
flash_eraseall_main                                  833     823     -10
fdformat_main                                        532     522     -10
expand_main                                          680     670     -10
eject_main                                           335     325     -10
dumpleases_main                                      630     620     -10
du_main                                              314     304     -10
dos2unix_main                                        441     431     -10
diff_main                                           1350    1340     -10
df_main                                             1064    1054     -10
date_main                                           1095    1085     -10
cut_main                                             961     951     -10
cryptpw_main                                         228     218     -10
crontab_main                                         575     565     -10
crond_main                                          1149    1139     -10
cp_main                                              370     360     -10
common_traceroute_main                              3834    3824     -10
common_ping_main                                    1767    1757     -10
comm_main                                            239     229     -10
cmp_main                                             655     645     -10
chrt_main                                            379     369     -10
chpst_main                                           704     694     -10
chpasswd_main                                        308     298     -10
chown_main                                           171     161     -10
chmod_main                                           158     148     -10
cat_main                                             428     418     -10
bzip2_main                                           120     110     -10
blkdiscard_main                                      264     254     -10
base64_main                                          221     211     -10
arping_main                                         1665    1655     -10
ar_main                                              556     546     -10
adjtimex_main                                        406     396     -10
adduser_main                                         882     872     -10
addgroup_main                                        411     401     -10
acpid_main                                          1198    1188     -10
optstring                                             11       -     -11
opt_string                                            18       -     -18
OPT_STR                                               25       -     -25
ubi_tools_main                                      1288    1258     -30
ls_options                                            31       -     -31
------------------------------------------------------------------------------
(add/remove: 0/6 grow/shrink: 3/129 up/down: 86/-1383)      Total: -1297 bytes
   text	   data	    bss	    dec	    hex	filename
 915428	    485	   6876	 922789	  e14a5	busybox_old
 914629	    485	   6872	 921986	  e1182	busybox_unstripped

Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
2017-08-08 21:55:02 +02:00

417 lines
8.1 KiB
C

/*
* pstree.c - display process tree
*
* Copyright (C) 1993-2002 Werner Almesberger
* Copyright (C) 2002-2009 Craig Small
* Copyright (C) 2010 Lauri Kasanen
*
* Based on pstree (PSmisc) 22.13.
*
* Licensed under GPLv2, see file LICENSE in this source tree.
*/
//config:config PSTREE
//config: bool "pstree (9.4 kb)"
//config: default y
//config: help
//config: Display a tree of processes.
//applet:IF_PSTREE(APPLET_NOEXEC(pstree, pstree, BB_DIR_USR_BIN, BB_SUID_DROP, pstree))
//kbuild:lib-$(CONFIG_PSTREE) += pstree.o
//usage:#define pstree_trivial_usage
//usage: "[-p] [PID|USER]"
//usage:#define pstree_full_usage "\n\n"
//usage: "Display process tree, optionally start from USER or PID\n"
//usage: "\n -p Show pids"
#include "libbb.h"
#define PROC_BASE "/proc"
#define OPT_PID (1 << 0)
struct child;
#if ENABLE_FEATURE_SHOW_THREADS
/* For threads, we add {...} around the comm, so we need two extra bytes */
# define COMM_DISP_LEN (COMM_LEN + 2)
#else
# define COMM_DISP_LEN COMM_LEN
#endif
typedef struct proc {
char comm[COMM_DISP_LEN + 1];
// char flags; - unused, delete?
pid_t pid;
uid_t uid;
struct child *children;
struct proc *parent;
struct proc *next;
} PROC;
/* For flags above */
//#define PFLAG_THREAD 0x01
typedef struct child {
PROC *child;
struct child *next;
} CHILD;
#define empty_2 " "
#define branch_2 "|-"
#define vert_2 "| "
#define last_2 "`-"
#define single_3 "---"
#define first_3 "-+-"
struct globals {
/* 0-based. IOW: the number of chars we printed on current line */
unsigned cur_x;
unsigned output_width;
/* The buffers will be dynamically increased in size as needed */
unsigned capacity;
unsigned *width;
uint8_t *more;
PROC *list;
smallint dumped; /* used by dump_by_user */
};
#define G (*ptr_to_globals)
#define INIT_G() do { \
SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
} while (0)
/*
* Allocates additional buffer space for width and more as needed.
* The first call will allocate the first buffer.
*
* bufindex the index that will be used after the call to this function.
*/
static void ensure_buffer_capacity(int bufindex)
{
if (bufindex >= G.capacity) {
G.capacity += 0x100;
G.width = xrealloc(G.width, G.capacity * sizeof(G.width[0]));
G.more = xrealloc(G.more, G.capacity * sizeof(G.more[0]));
}
}
/* NB: this function is never called with "bad" chars
* (control chars or chars >= 0x7f)
*/
static void out_char(char c)
{
G.cur_x++;
if (G.cur_x > G.output_width)
return;
if (G.cur_x == G.output_width)
c = '+';
putchar(c);
}
/* NB: this function is never called with "bad" chars
* (control chars or chars >= 0x7f)
*/
static void out_string(const char *str)
{
while (*str)
out_char(*str++);
}
static void out_newline(void)
{
putchar('\n');
G.cur_x = 0;
}
static PROC *find_proc(pid_t pid)
{
PROC *walk;
for (walk = G.list; walk; walk = walk->next)
if (walk->pid == pid)
break;
return walk;
}
static PROC *new_proc(const char *comm, pid_t pid, uid_t uid)
{
PROC *new = xzalloc(sizeof(*new));
strcpy(new->comm, comm);
new->pid = pid;
new->uid = uid;
new->next = G.list;
G.list = new;
return G.list;
}
static void add_child(PROC *parent, PROC *child)
{
CHILD *new, **walk;
int cmp;
new = xmalloc(sizeof(*new));
new->child = child;
for (walk = &parent->children; *walk; walk = &(*walk)->next) {
cmp = strcmp((*walk)->child->comm, child->comm);
if (cmp > 0)
break;
if (cmp == 0 && (*walk)->child->uid > child->uid)
break;
}
new->next = *walk;
*walk = new;
}
static void add_proc(const char *comm, pid_t pid, pid_t ppid,
uid_t uid /*, char isthread*/)
{
PROC *this, *parent;
this = find_proc(pid);
if (!this)
this = new_proc(comm, pid, uid);
else {
strcpy(this->comm, comm);
this->uid = uid;
}
if (pid == ppid)
ppid = 0;
// if (isthread)
// this->flags |= PFLAG_THREAD;
parent = find_proc(ppid);
if (!parent)
parent = new_proc("?", ppid, 0);
add_child(parent, this);
this->parent = parent;
}
static int tree_equal(const PROC *a, const PROC *b)
{
const CHILD *walk_a, *walk_b;
if (strcmp(a->comm, b->comm) != 0)
return 0;
if ((option_mask32 /*& OPT_PID*/) && a->pid != b->pid)
return 0;
for (walk_a = a->children, walk_b = b->children;
walk_a && walk_b;
walk_a = walk_a->next, walk_b = walk_b->next
) {
if (!tree_equal(walk_a->child, walk_b->child))
return 0;
}
return !(walk_a || walk_b);
}
static int out_args(const char *mystr)
{
const char *here;
int strcount = 0;
char tmpstr[5];
for (here = mystr; *here; here++) {
if (*here == '\\') {
out_string("\\\\");
strcount += 2;
} else if (*here >= ' ' && *here < 0x7f) {
out_char(*here);
strcount++;
} else {
sprintf(tmpstr, "\\%03o", (unsigned char) *here);
out_string(tmpstr);
strcount += 4;
}
}
return strcount;
}
static void
dump_tree(PROC *current, int level, int rep, int leaf, int last, int closing)
{
CHILD *walk, *next, **scan;
int lvl, i, add, offset, count, comm_len, first;
char tmp[sizeof(int)*3 + 4];
if (!current)
return;
if (!leaf) {
for (lvl = 0; lvl < level; lvl++) {
i = G.width[lvl] + 1;
while (--i >= 0)
out_char(' ');
if (lvl == level - 1) {
if (last) {
out_string(last_2);
} else {
out_string(branch_2);
}
} else {
if (G.more[lvl + 1]) {
out_string(vert_2);
} else {
out_string(empty_2);
}
}
}
}
add = 0;
if (rep > 1) {
add += sprintf(tmp, "%d*[", rep);
out_string(tmp);
}
comm_len = out_args(current->comm);
if (option_mask32 /*& OPT_PID*/) {
comm_len += sprintf(tmp, "(%d)", (int)current->pid);
out_string(tmp);
}
offset = G.cur_x;
if (!current->children) {
while (closing--)
out_char(']');
out_newline();
}
ensure_buffer_capacity(level);
G.more[level] = !last;
G.width[level] = comm_len + G.cur_x - offset + add;
if (G.cur_x >= G.output_width) {
//out_string(first_3); - why? it won't print anything
//out_char('+');
out_newline();
return;
}
first = 1;
for (walk = current->children; walk; walk = next) {
count = 0;
next = walk->next;
scan = &walk->next;
while (*scan) {
if (!tree_equal(walk->child, (*scan)->child))
scan = &(*scan)->next;
else {
if (next == *scan)
next = (*scan)->next;
count++;
*scan = (*scan)->next;
}
}
if (first) {
out_string(next ? first_3 : single_3);
first = 0;
}
dump_tree(walk->child, level + 1, count + 1,
walk == current->children, !next,
closing + (count ? 1 : 0));
}
}
static void dump_by_user(PROC *current, uid_t uid)
{
const CHILD *walk;
if (!current)
return;
if (current->uid == uid) {
if (G.dumped)
putchar('\n');
dump_tree(current, 0, 1, 1, 1, 0);
G.dumped = 1;
return;
}
for (walk = current->children; walk; walk = walk->next)
dump_by_user(walk->child, uid);
}
#if ENABLE_FEATURE_SHOW_THREADS
static void handle_thread(const char *comm, pid_t pid, pid_t ppid, uid_t uid)
{
char threadname[COMM_DISP_LEN + 1];
sprintf(threadname, "{%.*s}", (int)sizeof(threadname) - 3, comm);
add_proc(threadname, pid, ppid, uid/*, 1*/);
}
#endif
static void mread_proc(void)
{
procps_status_t *p = NULL;
#if ENABLE_FEATURE_SHOW_THREADS
pid_t parent = 0;
#endif
int flags = PSSCAN_COMM | PSSCAN_PID | PSSCAN_PPID | PSSCAN_UIDGID | PSSCAN_TASKS;
while ((p = procps_scan(p, flags)) != NULL) {
#if ENABLE_FEATURE_SHOW_THREADS
if (p->pid != p->main_thread_pid)
handle_thread(p->comm, p->pid, parent, p->uid);
else
#endif
{
add_proc(p->comm, p->pid, p->ppid, p->uid/*, 0*/);
#if ENABLE_FEATURE_SHOW_THREADS
parent = p->pid;
#endif
}
}
}
int pstree_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int pstree_main(int argc UNUSED_PARAM, char **argv)
{
pid_t pid = 1;
long uid = 0;
INIT_G();
G.output_width = get_terminal_width(0);
getopt32(argv, "^" "p" "\0" "?1");
argv += optind;
if (argv[0]) {
if (argv[0][0] >= '0' && argv[0][0] <= '9') {
pid = xatoi(argv[0]);
} else {
uid = xuname2uid(argv[0]);
}
}
mread_proc();
if (!uid)
dump_tree(find_proc(pid), 0, 1, 1, 1, 0);
else {
dump_by_user(find_proc(1), uid);
if (!G.dumped) {
bb_error_msg_and_die("no processes found");
}
}
if (ENABLE_FEATURE_CLEAN_UP) {
free(G.width);
free(G.more);
}
return 0;
}