123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356 |
- /* vi: set sw=4 ts=4: */
- /*
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
- */
- #include "libbb.h"
- #include <sys/mount.h>
- #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 are started by kernel, PATH might be unset.
- * In order to find "tar", let's set some sane PATH:
- */
- 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;
- }
|