mirror of
https://github.com/sheumann/hush.git
synced 2024-08-20 19:29:02 +00:00
function old new delta packed_usage 28706 28623 -83 Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
452 lines
13 KiB
C
452 lines
13 KiB
C
/* vi: set sw=4 ts=4: */
|
|
/*
|
|
* Licensed under GPLv2 or later, see file LICENSE in this source tree.
|
|
*/
|
|
|
|
//applet:IF_BOOTCHARTD(APPLET(bootchartd, BB_DIR_SBIN, BB_SUID_DROP))
|
|
|
|
//kbuild:lib-$(CONFIG_BOOTCHARTD) += bootchartd.o
|
|
|
|
//config:config BOOTCHARTD
|
|
//config: bool "bootchartd"
|
|
//config: default y
|
|
//config: help
|
|
//config: bootchartd is commonly used to profile the boot process
|
|
//config: for the purpose of speeding it up. In this case, it is started
|
|
//config: by the kernel as the init process. This is configured by adding
|
|
//config: the init=/sbin/bootchartd option to the kernel command line.
|
|
//config:
|
|
//config: It can also be used to monitor the resource usage of a specific
|
|
//config: application or the running system in general. In this case,
|
|
//config: bootchartd is started interactively by running bootchartd start
|
|
//config: and stopped using bootchartd stop.
|
|
//config:
|
|
//config:config FEATURE_BOOTCHARTD_BLOATED_HEADER
|
|
//config: bool "Compatible, bloated header"
|
|
//config: default y
|
|
//config: depends on BOOTCHARTD
|
|
//config: help
|
|
//config: Create extended header file compatible with "big" bootchartd.
|
|
//config: "Big" bootchartd is a shell script and it dumps some
|
|
//config: "convenient" info int the header, such as:
|
|
//config: title = Boot chart for `hostname` (`date`)
|
|
//config: system.uname = `uname -srvm`
|
|
//config: system.release = `cat /etc/DISTRO-release`
|
|
//config: system.cpu = `grep '^model name' /proc/cpuinfo | head -1` ($cpucount)
|
|
//config: system.kernel.options = `cat /proc/cmdline`
|
|
//config: This data is not mandatory for bootchart graph generation,
|
|
//config: and is considered bloat. Nevertheless, this option
|
|
//config: makes bootchartd applet to dump a subset of it.
|
|
//config:
|
|
//config:config FEATURE_BOOTCHARTD_CONFIG_FILE
|
|
//config: bool "Support bootchartd.conf"
|
|
//config: default y
|
|
//config: depends on BOOTCHARTD
|
|
//config: help
|
|
//config: Enable reading and parsing of $PWD/bootchartd.conf
|
|
//config: and /etc/bootchartd.conf files.
|
|
|
|
#include "libbb.h"
|
|
/* After libbb.h, since it needs sys/types.h on some systems */
|
|
#include <sys/utsname.h>
|
|
|
|
#ifdef __linux__
|
|
# include <sys/mount.h>
|
|
# ifndef MS_SILENT
|
|
# define MS_SILENT (1 << 15)
|
|
# endif
|
|
# ifndef MNT_DETACH
|
|
# define MNT_DETACH 0x00000002
|
|
# endif
|
|
#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
|
|
|
|
|
|
//$PWD/bootchartd.conf and /etc/bootchartd.conf:
|
|
//supported options:
|
|
//# Sampling period (in seconds)
|
|
//SAMPLE_PERIOD=0.2
|
|
//
|
|
//not yet supported:
|
|
//# tmpfs size
|
|
//# (32 MB should suffice for ~20 minutes worth of log data, but YMMV)
|
|
//TMPFS_SIZE=32m
|
|
//
|
|
//# 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[COMMON_BUFSIZE];
|
|
} 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';
|
|
/* Is it 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(void)
|
|
{
|
|
char template[] = "/tmp/bootchart.XXXXXX";
|
|
char *tempdir = xstrdup(mkdtemp(template));
|
|
if (!tempdir) {
|
|
#ifdef __linux__
|
|
/* /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
|
|
bb_perror_msg_and_die("can't create temporary directory");
|
|
#endif
|
|
} else {
|
|
xchdir(tempdir);
|
|
}
|
|
return tempdir;
|
|
}
|
|
|
|
static void do_logging(unsigned sample_period_us)
|
|
{
|
|
//# 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 / sample_period_us; /* ~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 / sample_period_us)
|
|
count = 2*1000*1000 / sample_period_us;
|
|
}
|
|
fflush_all();
|
|
wait_more:
|
|
usleep(sample_period_us);
|
|
}
|
|
|
|
// [ -e kernel_pacct ] && accton off
|
|
}
|
|
|
|
static void finalize(char *tempdir, const char *prog)
|
|
{
|
|
//# Stop process accounting if configured
|
|
//local pacct=
|
|
//[ -e kernel_pacct ] && pacct=kernel_pacct
|
|
|
|
FILE *header_fp = xfopen("header", "w");
|
|
|
|
if (prog)
|
|
fprintf(header_fp, "profile.process = %s\n", prog);
|
|
|
|
fputs("version = "BC_VERSION_STR"\n", header_fp);
|
|
if (ENABLE_FEATURE_BOOTCHARTD_BLOATED_HEADER) {
|
|
char *hostname;
|
|
char *kcmdline;
|
|
time_t t;
|
|
struct tm tm_time;
|
|
/* x2 for possible localized weekday/month names */
|
|
char date_buf[sizeof("Mon Jun 21 05:29:03 CEST 2010") * 2];
|
|
struct utsname unamebuf;
|
|
|
|
hostname = safe_gethostname();
|
|
time(&t);
|
|
localtime_r(&t, &tm_time);
|
|
strftime(date_buf, sizeof(date_buf), "%a %b %e %H:%M:%S %Z %Y", &tm_time);
|
|
fprintf(header_fp, "title = Boot chart for %s (%s)\n", hostname, date_buf);
|
|
if (ENABLE_FEATURE_CLEAN_UP)
|
|
free(hostname);
|
|
|
|
uname(&unamebuf); /* never fails */
|
|
/* same as uname -srvm */
|
|
fprintf(header_fp, "system.uname = %s %s %s %s\n",
|
|
unamebuf.sysname,
|
|
unamebuf.release,
|
|
unamebuf.version,
|
|
unamebuf.machine
|
|
);
|
|
|
|
//system.release = `cat /etc/DISTRO-release`
|
|
//system.cpu = `grep '^model name' /proc/cpuinfo | head -1` ($cpucount)
|
|
|
|
kcmdline = xmalloc_open_read_close("/proc/cmdline", NULL);
|
|
/* kcmdline includes trailing "\n" */
|
|
fprintf(header_fp, "system.kernel.options = %s", kcmdline);
|
|
if (ENABLE_FEATURE_CLEAN_UP)
|
|
free(kcmdline);
|
|
}
|
|
fclose(header_fp);
|
|
|
|
/* 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:#define bootchartd_trivial_usage
|
|
//usage: "start [PROG ARGS]|stop|init"
|
|
//usage:#define bootchartd_full_usage "\n\n"
|
|
//usage: "Create /var/log/bootchart.tgz with boot chart data\n"
|
|
//usage: "\nstart: start background logging; with PROG, run PROG, then kill logging with USR1"
|
|
//usage: "\nstop: send USR1 to all bootchartd processes"
|
|
//usage: "\ninit: start background logging; stop when getty/xdm is seen (for init scripts)"
|
|
//usage: "\nUnder PID 1: as init, then exec $bootchart_init, /init, /sbin/init"
|
|
|
|
int bootchartd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
|
|
int bootchartd_main(int argc UNUSED_PARAM, char **argv)
|
|
{
|
|
unsigned sample_period_us;
|
|
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, INIT or CMD_PID1 state */
|
|
|
|
/* Read config file: */
|
|
sample_period_us = 200 * 1000;
|
|
if (ENABLE_FEATURE_BOOTCHARTD_CONFIG_FILE) {
|
|
char* token[2];
|
|
parser_t *parser = config_open2("/etc/bootchartd.conf" + 5, fopen_for_read);
|
|
if (!parser)
|
|
parser = config_open2("/etc/bootchartd.conf", fopen_for_read);
|
|
while (config_read(parser, token, 2, 0, "#=", PARSE_NORMAL & ~PARSE_COLLAPSE)) {
|
|
if (strcmp(token[0], "SAMPLE_PERIOD") == 0 && token[1])
|
|
sample_period_us = atof(token[1]) * 1000000;
|
|
}
|
|
config_close(parser);
|
|
}
|
|
if ((int)sample_period_us <= 0)
|
|
sample_period_us = 1; /* prevent division by 0 */
|
|
|
|
/* 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();
|
|
do_logging(sample_period_us);
|
|
finalize(tempdir, cmd == CMD_START ? argv[2] : NULL);
|
|
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 execute '%s'", "/sbin/init");
|
|
}
|
|
|
|
if (cmd == CMD_START && argv[2]) { /* "start PROG ARGS" */
|
|
pid_t pid = xvfork();
|
|
if (pid == 0) { /* child */
|
|
argv += 2;
|
|
BB_EXECVP_or_die(argv);
|
|
}
|
|
/* parent */
|
|
waitpid(pid, NULL, 0);
|
|
kill(logger_pid, SIGUSR1);
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|