1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111#include "libbb.h"
112#include "common_bufsiz.h"
113#ifdef __linux__
114# include <sys/sysinfo.h>
115#endif
116
117
118enum { MAX_WIDTH = 2*1024 };
119
120#if ENABLE_FEATURE_PS_TIME || ENABLE_FEATURE_PS_LONG
121static unsigned long get_uptime(void)
122{
123#ifdef __linux__
124 struct sysinfo info;
125 if (sysinfo(&info) < 0)
126 return 0;
127 return info.uptime;
128#elif 1
129 unsigned long uptime;
130 char buf[sizeof(uptime)*3 + 2];
131
132
133
134 if (open_read_close("/proc/uptime", buf, sizeof(buf)) <= 0)
135 bb_perror_msg_and_die("can't read '%s'", "/proc/uptime");
136 buf[sizeof(buf)-1] = '\0';
137 sscanf(buf, "%lu", &uptime);
138 return uptime;
139#else
140 struct timespec ts;
141 if (clock_gettime(CLOCK_MONOTONIC, &ts) < 0)
142 return 0;
143 return ts.tv_sec;
144#endif
145}
146#endif
147
148#if ENABLE_DESKTOP
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169typedef struct {
170 uint16_t width;
171 char name6[6];
172 const char *header;
173 void (*f)(char *buf, int size, const procps_status_t *ps);
174 int ps_flags;
175} ps_out_t;
176
177struct globals {
178 ps_out_t* out;
179 int out_cnt;
180 int print_header;
181 int need_flags;
182 char *buffer;
183 unsigned terminal_width;
184#if ENABLE_FEATURE_PS_TIME
185# if ENABLE_FEATURE_PS_UNUSUAL_SYSTEMS || !defined(__linux__)
186 unsigned kernel_HZ;
187# endif
188 unsigned long seconds_since_boot;
189#endif
190} FIX_ALIASING;
191#define G (*(struct globals*)bb_common_bufsiz1)
192#define out (G.out )
193#define out_cnt (G.out_cnt )
194#define print_header (G.print_header )
195#define need_flags (G.need_flags )
196#define buffer (G.buffer )
197#define terminal_width (G.terminal_width )
198#define INIT_G() do { setup_common_bufsiz(); } while (0)
199
200#if ENABLE_FEATURE_PS_TIME
201# if ENABLE_FEATURE_PS_UNUSUAL_SYSTEMS || !defined(__linux__)
202# define get_kernel_HZ() (G.kernel_HZ)
203# else
204
205# define get_kernel_HZ() ((unsigned)100)
206# endif
207#endif
208
209
210
211static void func_user(char *buf, int size, const procps_status_t *ps)
212{
213#if 1
214 safe_strncpy(buf, get_cached_username(ps->uid), size+1);
215#else
216
217
218
219
220
221
222 const char *user = get_cached_username(ps->uid);
223 if (strlen(user) <= size)
224 safe_strncpy(buf, user, size+1);
225 else
226 sprintf(buf, "%*u", size, (unsigned)ps->uid);
227#endif
228}
229
230static void func_group(char *buf, int size, const procps_status_t *ps)
231{
232 safe_strncpy(buf, get_cached_groupname(ps->gid), size+1);
233}
234
235static void func_comm(char *buf, int size, const procps_status_t *ps)
236{
237 safe_strncpy(buf, ps->comm, size+1);
238}
239
240static void func_state(char *buf, int size, const procps_status_t *ps)
241{
242 safe_strncpy(buf, ps->state, size+1);
243}
244
245static void func_args(char *buf, int size, const procps_status_t *ps)
246{
247 read_cmdline(buf, size+1, ps->pid, ps->comm);
248}
249
250static void func_pid(char *buf, int size, const procps_status_t *ps)
251{
252 sprintf(buf, "%*u", size, ps->pid);
253}
254
255static void func_ppid(char *buf, int size, const procps_status_t *ps)
256{
257 sprintf(buf, "%*u", size, ps->ppid);
258}
259
260static void func_pgid(char *buf, int size, const procps_status_t *ps)
261{
262 sprintf(buf, "%*u", size, ps->pgid);
263}
264
265static void func_sid(char *buf, int size, const procps_status_t *ps)
266{
267 sprintf(buf, "%*u", size, ps->sid);
268}
269
270static void put_lu(char *buf, int size, unsigned long u)
271{
272 char buf4[5];
273
274
275 smart_ulltoa4(u, buf4, " mgtpezy")[0] = '\0';
276 sprintf(buf, "%.*s", size, buf4);
277}
278
279static void func_vsz(char *buf, int size, const procps_status_t *ps)
280{
281 put_lu(buf, size, ps->vsz);
282}
283
284static void func_rss(char *buf, int size, const procps_status_t *ps)
285{
286 put_lu(buf, size, ps->rss);
287}
288
289static void func_tty(char *buf, int size, const procps_status_t *ps)
290{
291 buf[0] = '?';
292 buf[1] = '\0';
293 if (ps->tty_major)
294 snprintf(buf, size+1, "%u,%u", ps->tty_major, ps->tty_minor);
295}
296
297#if ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS
298static void func_rgroup(char *buf, int size, const procps_status_t *ps)
299{
300 safe_strncpy(buf, get_cached_groupname(ps->rgid), size+1);
301}
302static void func_ruser(char *buf, int size, const procps_status_t *ps)
303{
304 safe_strncpy(buf, get_cached_username(ps->ruid), size+1);
305}
306static void func_nice(char *buf, int size, const procps_status_t *ps)
307{
308 sprintf(buf, "%*d", size, ps->niceness);
309}
310#endif
311
312#if ENABLE_FEATURE_PS_TIME
313static void format_time(char *buf, int size, unsigned long tt)
314{
315 unsigned ff;
316
317
318
319
320
321
322
323
324
325
326 ff = tt % 60;
327 tt /= 60;
328 if (tt < 60) {
329 snprintf(buf, size+1, "%2u:%02u", (unsigned)tt, ff);
330 return;
331 }
332 ff = tt % 60;
333 tt /= 60;
334 if (tt < 24) {
335 snprintf(buf, size+1, "%2uh%02u", (unsigned)tt, ff);
336 return;
337 }
338 ff = tt % 24;
339 tt /= 24;
340 if (tt < 100) {
341 snprintf(buf, size+1, "%2ud%02u", (unsigned)tt, ff);
342 return;
343 }
344 snprintf(buf, size+1, "%4lud", tt);
345}
346static void func_etime(char *buf, int size, const procps_status_t *ps)
347{
348
349 unsigned long mm;
350
351 mm = ps->start_time / get_kernel_HZ();
352 mm = G.seconds_since_boot - mm;
353 format_time(buf, size, mm);
354}
355static void func_time(char *buf, int size, const procps_status_t *ps)
356{
357
358 unsigned long mm;
359
360 mm = (ps->utime + ps->stime) / get_kernel_HZ();
361 format_time(buf, size, mm);
362}
363#endif
364
365#if ENABLE_SELINUX
366static void func_label(char *buf, int size, const procps_status_t *ps)
367{
368 safe_strncpy(buf, ps->context ? ps->context : "unknown", size+1);
369}
370#endif
371
372
373
374
375
376
377
378static const ps_out_t out_spec[] ALIGN_PTR = {
379
380 { 8 , "user" ,"USER" ,func_user ,PSSCAN_UIDGID },
381 { 8 , "group" ,"GROUP" ,func_group ,PSSCAN_UIDGID },
382 { 16 , "comm" ,"COMMAND",func_comm ,PSSCAN_COMM },
383 { MAX_WIDTH , "args" ,"COMMAND",func_args ,PSSCAN_COMM },
384 { 5 , "pid" ,"PID" ,func_pid ,PSSCAN_PID },
385 { 5 , "ppid" ,"PPID" ,func_ppid ,PSSCAN_PPID },
386 { 5 , "pgid" ,"PGID" ,func_pgid ,PSSCAN_PGID },
387#if ENABLE_FEATURE_PS_TIME
388 { sizeof("ELAPSED")-1, "etime" ,"ELAPSED",func_etime ,PSSCAN_START_TIME },
389#endif
390#if ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS
391 { 5 , "nice" ,"NI" ,func_nice ,PSSCAN_NICE },
392 { 8 , "rgroup","RGROUP" ,func_rgroup,PSSCAN_RUIDGID },
393 { 8 , "ruser" ,"RUSER" ,func_ruser ,PSSCAN_RUIDGID },
394
395#endif
396#if ENABLE_FEATURE_PS_TIME
397 { 5 , "time" ,"TIME" ,func_time ,PSSCAN_STIME | PSSCAN_UTIME },
398#endif
399 { 6 , "tty" ,"TT" ,func_tty ,PSSCAN_TTY },
400 { 4 , "vsz" ,"VSZ" ,func_vsz ,PSSCAN_VSZ },
401
402 { 5 , "sid" ,"SID" ,func_sid ,PSSCAN_SID },
403 { 4 , "stat" ,"STAT" ,func_state ,PSSCAN_STATE },
404 { 4 , "rss" ,"RSS" ,func_rss ,PSSCAN_RSS },
405#if ENABLE_SELINUX
406 { 35 , "label" ,"LABEL" ,func_label ,PSSCAN_CONTEXT },
407#endif
408};
409
410static ps_out_t* new_out_t(void)
411{
412 out = xrealloc_vector(out, 2, out_cnt);
413 return &out[out_cnt++];
414}
415
416static const ps_out_t* find_out_spec(const char *name)
417{
418 unsigned i;
419 char buf[ARRAY_SIZE(out_spec)*7 + 1];
420 char *p = buf;
421
422 for (i = 0; i < ARRAY_SIZE(out_spec); i++) {
423 if (strncmp(name, out_spec[i].name6, 6) == 0)
424 return &out_spec[i];
425 p += sprintf(p, "%.6s,", out_spec[i].name6);
426 }
427 p[-1] = '\0';
428 bb_error_msg_and_die("bad -o argument '%s', supported arguments: %s", name, buf);
429}
430
431static void parse_o(char* opt)
432{
433 ps_out_t* new;
434
435 char *comma, *equal;
436 while (1) {
437 comma = strchr(opt, ',');
438 equal = strchr(opt, '=');
439 if (comma && (!equal || equal > comma)) {
440 *comma = '\0';
441 *new_out_t() = *find_out_spec(opt);
442 *comma = ',';
443 opt = comma + 1;
444 continue;
445 }
446 break;
447 }
448
449
450 new = new_out_t();
451 if (equal)
452 *equal = '\0';
453 *new = *find_out_spec(opt);
454 if (equal) {
455 *equal = '=';
456 new->header = equal + 1;
457
458
459
460
461
462 if (new->header[0]) {
463 new->width = strlen(new->header);
464 print_header = 1;
465 }
466 } else
467 print_header = 1;
468}
469
470static void alloc_line_buffer(void)
471{
472 int i;
473 int width = 0;
474 for (i = 0; i < out_cnt; i++) {
475 need_flags |= out[i].ps_flags;
476 if (out[i].header[0]) {
477 print_header = 1;
478 }
479 width += out[i].width + 1;
480 if ((int)(width - terminal_width) > 0) {
481
482
483 out_cnt = i + 1;
484 break;
485 }
486 }
487#if ENABLE_SELINUX
488 if (!is_selinux_enabled())
489 need_flags &= ~PSSCAN_CONTEXT;
490#endif
491 buffer = xmalloc(width + 1);
492}
493
494static void format_header(void)
495{
496 int i;
497 ps_out_t* op;
498 char *p;
499
500 if (!print_header)
501 return;
502 p = buffer;
503 i = 0;
504 if (out_cnt) {
505 while (1) {
506 op = &out[i];
507 if (++i == out_cnt)
508 break;
509 p += sprintf(p, "%-*s ", op->width, op->header);
510 }
511 strcpy(p, op->header);
512 }
513 printf("%.*s\n", terminal_width, buffer);
514}
515
516static void format_process(const procps_status_t *ps)
517{
518 int i, len;
519 char *p = buffer;
520 i = 0;
521 if (out_cnt) while (1) {
522 out[i].f(p, out[i].width, ps);
523
524
525
526 if (!p[0]) {
527 p[0] = '-';
528 p[1] = '\0';
529 }
530 len = strlen(p);
531 p += len;
532 len = out[i].width - len + 1;
533 if (++i == out_cnt)
534 break;
535 p += sprintf(p, "%*s", len, " ");
536 }
537 printf("%.*s\n", terminal_width, buffer);
538}
539
540#if ENABLE_SELINUX
541# define SELINUX_O_PREFIX "label,"
542# define DEFAULT_O_STR (SELINUX_O_PREFIX "pid,user" IF_FEATURE_PS_TIME(",time") ",args")
543#else
544# define DEFAULT_O_STR ("pid,user" IF_FEATURE_PS_TIME(",time") ",args")
545#endif
546
547int ps_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
548int ps_main(int argc UNUSED_PARAM, char **argv)
549{
550 procps_status_t *p;
551 llist_t* opt_o = NULL;
552 char default_o[sizeof(DEFAULT_O_STR)];
553#if ENABLE_SELINUX || ENABLE_FEATURE_SHOW_THREADS
554 int opt;
555#endif
556 enum {
557 OPT_Z = (1 << 0),
558 OPT_o = (1 << 1),
559 OPT_a = (1 << 2),
560 OPT_A = (1 << 3),
561 OPT_d = (1 << 4),
562 OPT_e = (1 << 5),
563 OPT_f = (1 << 6),
564 OPT_l = (1 << 7),
565 OPT_T = (1 << 8) * ENABLE_FEATURE_SHOW_THREADS,
566 };
567
568 INIT_G();
569#if ENABLE_FEATURE_PS_TIME
570 G.seconds_since_boot = get_uptime();
571# if ENABLE_FEATURE_PS_UNUSUAL_SYSTEMS || !defined(__linux__)
572 G.kernel_HZ = bb_clk_tck();
573# endif
574#endif
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591#if ENABLE_SELINUX || ENABLE_FEATURE_SHOW_THREADS
592 opt =
593#endif
594 getopt32(argv, "Zo:*aAdefl"IF_FEATURE_SHOW_THREADS("T"), &opt_o);
595
596 if (opt_o) {
597 do {
598 parse_o(llist_pop(&opt_o));
599 } while (opt_o);
600 } else {
601
602
603
604#if ENABLE_SELINUX
605 if (!(opt & OPT_Z) || !is_selinux_enabled()) {
606
607 strcpy(default_o, DEFAULT_O_STR + sizeof(SELINUX_O_PREFIX)-1);
608 } else
609#endif
610 {
611 strcpy(default_o, DEFAULT_O_STR);
612 }
613 parse_o(default_o);
614 }
615#if ENABLE_FEATURE_SHOW_THREADS
616 if (opt & OPT_T)
617 need_flags |= PSSCAN_TASKS;
618#endif
619
620
621
622 terminal_width = MAX_WIDTH;
623 if (isatty(1)) {
624 terminal_width = get_terminal_width(0);
625 if (--terminal_width > MAX_WIDTH)
626 terminal_width = MAX_WIDTH;
627 }
628 alloc_line_buffer();
629 format_header();
630
631 p = NULL;
632 while ((p = procps_scan(p, need_flags)) != NULL) {
633 format_process(p);
634 }
635
636 return EXIT_SUCCESS;
637}
638
639
640#else
641
642
643int ps_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
644int ps_main(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
645{
646 procps_status_t *p;
647 int psscan_flags = PSSCAN_PID | PSSCAN_UIDGID
648 | PSSCAN_STATE | PSSCAN_VSZ | PSSCAN_COMM;
649 unsigned terminal_width IF_NOT_FEATURE_PS_WIDE(= 79);
650 enum {
651 OPT_Z = (1 << 0) * ENABLE_SELINUX,
652 OPT_T = (1 << ENABLE_SELINUX) * ENABLE_FEATURE_SHOW_THREADS,
653 OPT_l = (1 << ENABLE_SELINUX) * (1 << ENABLE_FEATURE_SHOW_THREADS) * ENABLE_FEATURE_PS_LONG,
654 };
655#if ENABLE_FEATURE_PS_LONG
656 time_t now = now;
657 unsigned long uptime = uptime;
658#endif
659
660#if ENABLE_SELINUX || ENABLE_FEATURE_SHOW_THREADS || ENABLE_FEATURE_PS_WIDE || ENABLE_FEATURE_PS_LONG
661 int opts = 0;
662# if ENABLE_FEATURE_PS_WIDE
663
664 int w_count = 0;
665 make_all_argv_opts(argv);
666 opts = getopt32(argv, "^"
667 IF_SELINUX("Z")IF_FEATURE_SHOW_THREADS("T")IF_FEATURE_PS_LONG("l")"w"
668 "\0" "ww",
669 &w_count
670 );
671
672
673
674 if (w_count) {
675 terminal_width = (w_count == 1) ? 132 : MAX_WIDTH;
676 } else {
677 terminal_width = get_terminal_width(0);
678
679 if (--terminal_width > MAX_WIDTH)
680 terminal_width = MAX_WIDTH;
681 }
682# else
683
684 make_all_argv_opts(argv);
685 opts = getopt32(argv, IF_SELINUX("Z")IF_FEATURE_SHOW_THREADS("T")IF_FEATURE_PS_LONG("l"));
686# endif
687
688# if ENABLE_SELINUX
689 if ((opts & OPT_Z) && is_selinux_enabled()) {
690 psscan_flags = PSSCAN_PID | PSSCAN_CONTEXT
691 | PSSCAN_STATE | PSSCAN_COMM;
692 puts(" PID CONTEXT STAT COMMAND");
693 } else
694# endif
695 if (opts & OPT_l) {
696 psscan_flags = PSSCAN_STATE | PSSCAN_UIDGID | PSSCAN_PID | PSSCAN_PPID
697 | PSSCAN_TTY | PSSCAN_STIME | PSSCAN_UTIME | PSSCAN_COMM
698 | PSSCAN_VSZ | PSSCAN_RSS;
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717 puts("S UID PID PPID VSZ RSS TTY STIME TIME CMD");
718# if ENABLE_FEATURE_PS_LONG
719 now = time(NULL);
720 uptime = get_uptime();
721# endif
722 }
723 else {
724 puts(" PID USER VSZ STAT COMMAND");
725 }
726 if (opts & OPT_T) {
727 psscan_flags |= PSSCAN_TASKS;
728 }
729#endif
730
731 p = NULL;
732 while ((p = procps_scan(p, psscan_flags)) != NULL) {
733 int len;
734#if ENABLE_SELINUX
735 if (psscan_flags & PSSCAN_CONTEXT) {
736 len = printf("%5u %-32.32s %s ",
737 p->pid,
738 p->context ? p->context : "unknown",
739 p->state);
740 } else
741#endif
742 {
743 char buf6[6];
744 smart_ulltoa5(p->vsz, buf6, " mgtpezy")[0] = '\0';
745#if ENABLE_FEATURE_PS_LONG
746 if (opts & OPT_l) {
747 char bufr[6], stime_str[6];
748 char tty[2 * sizeof(int)*3 + 2];
749 char *endp;
750 unsigned sut = (p->stime + p->utime) / 100;
751 unsigned elapsed = uptime - (p->start_time / 100);
752 time_t start = now - elapsed;
753 struct tm *tm = localtime(&start);
754
755 smart_ulltoa5(p->rss, bufr, " mgtpezy")[0] = '\0';
756
757 if (p->tty_major == 136)
758
759
760
761 endp = stpcpy(tty, "pts");
762 else
763 if (p->tty_major == 4) {
764 endp = stpcpy(tty, "tty");
765 if (p->tty_minor >= 64) {
766 p->tty_minor -= 64;
767 *endp++ = 'S';
768 }
769 }
770 else
771 endp = tty + sprintf(tty, "%d:", p->tty_major);
772 strcpy(endp, utoa(p->tty_minor));
773
774 strftime(stime_str, 6, (elapsed >= (24 * 60 * 60)) ? "%b%d" : "%H:%M", tm);
775 stime_str[5] = '\0';
776
777 len = printf("%c %5u %5u %5u %5s %5s %-5s %s %02u:%02u:%02u ",
778 p->state[0], p->uid, p->pid, p->ppid, buf6, bufr, tty,
779 stime_str, sut / 3600, (sut % 3600) / 60, sut % 60);
780 } else
781#endif
782 {
783 const char *user = get_cached_username(p->uid);
784 len = printf("%5u %-8.8s %s %s ",
785 p->pid, user, buf6, p->state);
786 }
787 }
788
789 {
790 int sz = terminal_width - len;
791 if (sz >= 0) {
792 char buf[sz + 1];
793 read_cmdline(buf, sz, p->pid, p->comm);
794 puts(buf);
795 }
796 }
797 }
798 if (ENABLE_FEATURE_CLEAN_UP)
799 clear_username_cache();
800 return EXIT_SUCCESS;
801}
802
803#endif
804