From ff027d6f50bfa24228e230b84a3297c51d37d000 Mon Sep 17 00:00:00 2001 From: Denys Vlasenko Date: Tue, 4 May 2010 15:45:25 +0200 Subject: bootchartd: new applet Usage: bootchartd start [PROG ARGS]|init|stop Create /var/log/bootchart.tgz with boot chart data Options: start: start background logging; with PROG, run PROG, then kill logging with USR1 stop: send USR1 to all bootchartd processes init: start background logging; stop when getty/gdm is seen (for init scripts) Under PID 1: as init, then exec $bootchart_init, /init, /sbin/init function old new delta bootchartd_main - 907 +907 dump_procs - 353 +353 packed_usage 26566 26735 +169 dump_file - 91 +91 static.dirs - 23 +23 applet_names 2176 2187 +11 applet_main 1284 1288 +4 applet_nameofs 642 644 +2 ------------------------------------------------------------------------------ (add/remove: 5/0 grow/shrink: 4/0 up/down: 1560/0) Total: 1560 bytes Signed-off-by: Denys Vlasenko --- init/bootchartd.c | 355 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 355 insertions(+) create mode 100644 init/bootchartd.c (limited to 'init/bootchartd.c') diff --git a/init/bootchartd.c b/init/bootchartd.c new file mode 100644 index 000000000..fccaaceb4 --- /dev/null +++ b/init/bootchartd.c @@ -0,0 +1,355 @@ +/* vi: set sw=4 ts=4: */ +/* + * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. + */ +#include "libbb.h" +#include +#ifndef MS_SILENT +# define MS_SILENT (1 << 15) +#endif +#ifndef MNT_DETACH +# define MNT_DETACH 0x00000002 +#endif + +#define BC_VERSION_STR "0.8" + +/* For debugging, set to 0: + * strace won't work with DO_SIGNAL_SYNC set to 1. + */ +#define DO_SIGNAL_SYNC 1 + + +//Not supported: $PWD/bootchartd.conf and /etc/bootchartd.conf + +//# tmpfs size +//# (32 MB should suffice for ~20 minutes worth of log data, but YMMV) +//TMPFS_SIZE=32m +// +//# Sampling period (in seconds) +//SAMPLE_PERIOD=0.2 +// +//# Whether to enable and store BSD process accounting information. The +//# kernel needs to be configured to enable v3 accounting +//# (CONFIG_BSD_PROCESS_ACCT_V3). accton from the GNU accounting utilities +//# is also required. +//PROCESS_ACCOUNTING="no" +// +//# Tarball for the various boot log files +//BOOTLOG_DEST=/var/log/bootchart.tgz +// +//# Whether to automatically stop logging as the boot process completes. +//# The logger will look for known processes that indicate bootup completion +//# at a specific runlevel (e.g. gdm-binary, mingetty, etc.). +//AUTO_STOP_LOGGER="yes" +// +//# Whether to automatically generate the boot chart once the boot logger +//# completes. The boot chart will be generated in $AUTO_RENDER_DIR. +//# Note that the bootchart package must be installed. +//AUTO_RENDER="no" +// +//# Image format to use for the auto-generated boot chart +//# (choose between png, svg and eps). +//AUTO_RENDER_FORMAT="png" +// +//# Output directory for auto-generated boot charts +//AUTO_RENDER_DIR="/var/log" + + +/* Globals */ +struct globals { + char jiffy_line[sizeof(bb_common_bufsiz1)]; +} FIX_ALIASING; +#define G (*(struct globals*)&bb_common_bufsiz1) +#define INIT_G() do { } while (0) + +static void dump_file(FILE *fp, const char *filename) +{ + int fd = open(filename, O_RDONLY); + if (fd >= 0) { + fputs(G.jiffy_line, fp); + fflush(fp); + bb_copyfd_eof(fd, fileno(fp)); + close(fd); + fputc('\n', fp); + } +} + +static int dump_procs(FILE *fp, int look_for_login_process) +{ + struct dirent *entry; + DIR *dir = opendir("/proc"); + int found_login_process = 0; + + fputs(G.jiffy_line, fp); + while ((entry = readdir(dir)) != NULL) { + char name[sizeof("/proc/%u/cmdline") + sizeof(int)*3]; + int stat_fd; + unsigned pid = bb_strtou(entry->d_name, NULL, 10); + if (errno) + continue; + + /* Android's version reads /proc/PID/cmdline and extracts + * non-truncated process name. Do we want to do that? */ + + sprintf(name, "/proc/%u/stat", pid); + stat_fd = open(name, O_RDONLY); + if (stat_fd >= 0) { + char *p; + char stat_line[4*1024]; + int rd = safe_read(stat_fd, stat_line, sizeof(stat_line)-2); + + close(stat_fd); + if (rd < 0) + continue; + stat_line[rd] = '\0'; + p = strchrnul(stat_line, '\n'); + *p++ = '\n'; + *p = '\0'; + fputs(stat_line, fp); + if (!look_for_login_process) + continue; + p = strchr(stat_line, '('); + if (!p) + continue; + p++; + strchrnul(p, ')')[0] = '\0'; + /* If is gdm, kdm or a getty? */ + if (((p[0] == 'g' || p[0] == 'k' || p[0] == 'x') && p[1] == 'd' && p[2] == 'm') + || strstr(p, "getty") + ) { + found_login_process = 1; + } + } + } + closedir(dir); + fputc('\n', fp); + return found_login_process; +} + +static char *make_tempdir(const char *prog) +{ + char template[] = "/tmp/bootchart.XXXXXX"; + char *tempdir = xstrdup(mkdtemp(template)); + if (!tempdir) { + /* /tmp is not writable (happens when we are used as init). + * Try to mount a tmpfs, them cd and lazily unmount it. + * Since we unmount it at once, we can mount it anywhere. + * Try a few locations which are likely ti exist. + */ + static const char dirs[] = "/mnt\0""/tmp\0""/boot\0""/proc\0"; + const char *try_dir = dirs; + while (mount("none", try_dir, "tmpfs", MS_SILENT, "size=16m") != 0) { + try_dir += strlen(try_dir) + 1; + if (!try_dir[0]) + bb_perror_msg_and_die("can't %smount tmpfs", ""); + } + //bb_error_msg("mounted tmpfs on %s", try_dir); + xchdir(try_dir); + if (umount2(try_dir, MNT_DETACH) != 0) { + bb_perror_msg_and_die("can't %smount tmpfs", "un"); + } + } else { + xchdir(tempdir); + } + { + FILE *header_fp = xfopen("header", "w"); + if (prog) + fprintf(header_fp, "profile.process = %s\n", prog); + fputs("version = "BC_VERSION_STR"\n", header_fp); + fclose(header_fp); + } + + return tempdir; +} + +static void do_logging(void) +{ + //# Enable process accounting if configured + //if [ "$PROCESS_ACCOUNTING" = "yes" ]; then + // [ -e kernel_pacct ] || : > kernel_pacct + // accton kernel_pacct + //fi + + FILE *proc_stat = xfopen("proc_stat.log", "w"); + FILE *proc_diskstats = xfopen("proc_diskstats.log", "w"); + //FILE *proc_netdev = xfopen("proc_netdev.log", "w"); + FILE *proc_ps = xfopen("proc_ps.log", "w"); + int look_for_login_process = (getppid() == 1); + unsigned count = 60*1000*1000 / 200*1000; /* ~1 minute */ + + while (--count && !bb_got_signal) { + char *p; + int len = open_read_close("/proc/uptime", G.jiffy_line, sizeof(G.jiffy_line)-2); + if (len < 0) + goto wait_more; + /* /proc/uptime has format "NNNNNN.MM NNNNNNN.MM" */ + /* we convert it to "NNNNNNMM\n" (using first value) */ + G.jiffy_line[len] = '\0'; + p = strchr(G.jiffy_line, '.'); + if (!p) + goto wait_more; + while (isdigit(*++p)) + p[-1] = *p; + p[-1] = '\n'; + p[0] = '\0'; + + dump_file(proc_stat, "/proc/stat"); + dump_file(proc_diskstats, "/proc/diskstats"); + //dump_file(proc_netdev, "/proc/net/dev"); + if (dump_procs(proc_ps, look_for_login_process)) { + /* dump_procs saw a getty or {g,k,x}dm + * stop logging in 2 seconds: + */ + if (count > 2*1000*1000 / 200*1000) + count = 2*1000*1000 / 200*1000; + } + fflush_all(); + wait_more: + usleep(200*1000); + } + + // [ -e kernel_pacct ] && accton off +} + +static void finalize(char *tempdir) +{ + //# Stop process accounting if configured + //local pacct= + //[ -e kernel_pacct ] && pacct=kernel_pacct + + //( + // echo "version = $VERSION" + // echo "title = Boot chart for $( hostname | sed q ) ($( date ))" + // echo "system.uname = $( uname -srvm | sed q )" + // echo "system.release = $( sed q /etc/SuSE-release )" + // echo "system.cpu = $( grep '^model name' /proc/cpuinfo | sed q ) ($cpucount)" + // echo "system.kernel.options = $( sed q /proc/cmdline )" + //) >> header + + /* Package log files */ + system("tar -zcf /var/log/bootchart.tgz header *.log"); // + $pacct + /* Clean up (if we are not in detached tmpfs) */ + if (tempdir) { + unlink("header"); + unlink("proc_stat.log"); + unlink("proc_diskstats.log"); + //unlink("proc_netdev.log"); + unlink("proc_ps.log"); + rmdir(tempdir); + } + + /* shell-based bootchartd tries to run /usr/bin/bootchart if $AUTO_RENDER=yes: + * /usr/bin/bootchart -o "$AUTO_RENDER_DIR" -f $AUTO_RENDER_FORMAT "$BOOTLOG_DEST" + */ +} + +/* Usage: + * bootchartd start [PROG ARGS]: start logging in background, USR1 stops it. + * With PROG, runs PROG, then kills background logging. + * bootchartd stop: same as "killall -USR1 bootchartd" + * bootchartd init: start logging in background + * Stop when getty/gdm is seen (if AUTO_STOP_LOGGER = yes). + * Meant to be used from init scripts. + * bootchartd (pid==1): as init, but then execs $bootchart_init, /init, /sbin/init + * Meant to be used as kernel's init process. + */ +int bootchartd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; +int bootchartd_main(int argc UNUSED_PARAM, char **argv) +{ + pid_t parent_pid, logger_pid; + smallint cmd; + enum { + CMD_STOP = 0, + CMD_START, + CMD_INIT, + CMD_PID1, /* used to mark pid 1 case */ + }; + + INIT_G(); + + parent_pid = getpid(); + if (argv[1]) { + cmd = index_in_strings("stop\0""start\0""init\0", argv[1]); + if (cmd < 0) + bb_show_usage(); + if (cmd == CMD_STOP) { + pid_t *pidList = find_pid_by_name("bootchartd"); + while (*pidList != 0) { + if (*pidList != parent_pid) + kill(*pidList, SIGUSR1); + pidList++; + } + return EXIT_SUCCESS; + } + } else { + if (parent_pid != 1) + bb_show_usage(); + cmd = CMD_PID1; + } + + /* Here we are in START or INIT state. Create logger child: */ + logger_pid = fork_or_rexec(argv); + + if (logger_pid == 0) { /* child */ + char *tempdir; + + bb_signals(0 + + (1 << SIGUSR1) + + (1 << SIGUSR2) + + (1 << SIGTERM) + + (1 << SIGQUIT) + + (1 << SIGINT) + + (1 << SIGHUP) + , record_signo); + + if (DO_SIGNAL_SYNC) + /* Inform parent that we are ready */ + raise(SIGSTOP); + + /* If we started by kernel, PATH might be not set. + * And in order to run tar we may need PATH to be set: + */ + if (cmd == CMD_PID1 && !getenv("PATH")) + putenv((char*)bb_PATH_root_path); + tempdir = make_tempdir(cmd == CMD_START ? argv[2] : NULL); + do_logging(); + finalize(tempdir); + return EXIT_SUCCESS; + } + + /* parent */ + + if (DO_SIGNAL_SYNC) { + /* Wait for logger child to set handlers, then unpause it. + * Otherwise with short-lived PROG (e.g. "bootchartd start true") + * we might send SIGUSR1 before logger sets its handler. + */ + waitpid(logger_pid, NULL, WUNTRACED); + kill(logger_pid, SIGCONT); + } + + if (cmd == CMD_PID1) { + char *bootchart_init = getenv("bootchart_init"); + if (bootchart_init) + execl(bootchart_init, bootchart_init, NULL); + execl("/init", "init", NULL); + execl("/sbin/init", "init", NULL); + bb_perror_msg_and_die("can't exec '%s'", "/sbin/init"); + } + + if (cmd == CMD_START && argv[2]) { /* "start PROG ARGS" */ + pid_t pid = vfork(); + if (pid < 0) + bb_perror_msg_and_die("vfork"); + if (pid == 0) { /* child */ + argv += 2; + execvp(argv[0], argv); + bb_perror_msg_and_die("can't exec '%s'", argv[0]); + } + /* parent */ + waitpid(pid, NULL, 0); + kill(logger_pid, SIGUSR1); + } + + return EXIT_SUCCESS; +} -- cgit v1.2.3