--- /dev/null
+/* 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 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;
+}
projects / oweals / busybox.git / commitdiff