busybox/init/bootchartd.c

/* vi: set sw=4 ts=4: */
/*
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */

//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>
#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


//$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) {
		/* /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);
	}
	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:     "\nOptions:"
//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;
			execvp(argv[0], argv);
			bb_perror_msg_and_die("can't execute '%s'", argv[0]);
		}
		/* parent */
		waitpid(pid, NULL, 0);
		kill(logger_pid, SIGUSR1);
	}

	return EXIT_SUCCESS;
}
1	/* vi: set sw=4 ts=4: */
2	/*
3	* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
4	*/
5
6	//config:config BOOTCHARTD
7	//config: bool "bootchartd"
8	//config: default y
9	//config: help
10	//config: bootchartd is commonly used to profile the boot process
11	//config: for the purpose of speeding it up. In this case, it is started
12	//config: by the kernel as the init process. This is configured by adding
13	//config: the init=/sbin/bootchartd option to the kernel command line.
14	//config:
15	//config: It can also be used to monitor the resource usage of a specific
16	//config: application or the running system in general. In this case,
17	//config: bootchartd is started interactively by running bootchartd start
18	//config: and stopped using bootchartd stop.
19	//config:
20	//config:config FEATURE_BOOTCHARTD_BLOATED_HEADER
21	//config: bool "Compatible, bloated header"
22	//config: default y
23	//config: depends on BOOTCHARTD
24	//config: help
25	//config: Create extended header file compatible with "big" bootchartd.
26	//config: "Big" bootchartd is a shell script and it dumps some
27	//config: "convenient" info int the header, such as:
28	//config: title = Boot chart for `hostname` (`date`)
29	//config: system.uname = `uname -srvm`
30	//config: system.release = `cat /etc/DISTRO-release`
31	//config: system.cpu = `grep '^model name' /proc/cpuinfo \| head -1` ($cpucount)
32	//config: system.kernel.options = `cat /proc/cmdline`
33	//config: This data is not mandatory for bootchart graph generation,
34	//config: and is considered bloat. Nevertheless, this option
35	//config: makes bootchartd applet to dump a subset of it.
36	//config:
37	//config:config FEATURE_BOOTCHARTD_CONFIG_FILE
38	//config: bool "Support bootchartd.conf"
39	//config: default y
40	//config: depends on BOOTCHARTD
41	//config: help
42	//config: Enable reading and parsing of $PWD/bootchartd.conf
43	//config: and /etc/bootchartd.conf files.
44
45	#include "libbb.h"
46	/* After libbb.h, since it needs sys/types.h on some systems */
47	#include <sys/utsname.h>
48	#include <sys/mount.h>
49	#ifndef MS_SILENT
50	# define MS_SILENT (1 << 15)
51	#endif
52	#ifndef MNT_DETACH
53	# define MNT_DETACH 0x00000002
54	#endif
55
56	#define BC_VERSION_STR "0.8"
57
58	/* For debugging, set to 0:
59	* strace won't work with DO_SIGNAL_SYNC set to 1.
60	*/
61	#define DO_SIGNAL_SYNC 1
62
63
64	//$PWD/bootchartd.conf and /etc/bootchartd.conf:
65	//supported options:
66	//# Sampling period (in seconds)
67	//SAMPLE_PERIOD=0.2
68	//
69	//not yet supported:
70	//# tmpfs size
71	//# (32 MB should suffice for ~20 minutes worth of log data, but YMMV)
72	//TMPFS_SIZE=32m
73	//
74	//# Whether to enable and store BSD process accounting information. The
75	//# kernel needs to be configured to enable v3 accounting
76	//# (CONFIG_BSD_PROCESS_ACCT_V3). accton from the GNU accounting utilities
77	//# is also required.
78	//PROCESS_ACCOUNTING="no"
79	//
80	//# Tarball for the various boot log files
81	//BOOTLOG_DEST=/var/log/bootchart.tgz
82	//
83	//# Whether to automatically stop logging as the boot process completes.
84	//# The logger will look for known processes that indicate bootup completion
85	//# at a specific runlevel (e.g. gdm-binary, mingetty, etc.).
86	//AUTO_STOP_LOGGER="yes"
87	//
88	//# Whether to automatically generate the boot chart once the boot logger
89	//# completes. The boot chart will be generated in $AUTO_RENDER_DIR.
90	//# Note that the bootchart package must be installed.
91	//AUTO_RENDER="no"
92	//
93	//# Image format to use for the auto-generated boot chart
94	//# (choose between png, svg and eps).
95	//AUTO_RENDER_FORMAT="png"
96	//
97	//# Output directory for auto-generated boot charts
98	//AUTO_RENDER_DIR="/var/log"
99
100
101	/* Globals */
102	struct globals {
103	char jiffy_line[COMMON_BUFSIZE];
104	} FIX_ALIASING;
105	#define G ((struct globals)&bb_common_bufsiz1)
106	#define INIT_G() do { } while (0)
107
108	static void dump_file(FILE fp, const char filename)
109	{
110	int fd = open(filename, O_RDONLY);
111	if (fd >= 0) {
112	fputs(G.jiffy_line, fp);
113	fflush(fp);
114	bb_copyfd_eof(fd, fileno(fp));
115	close(fd);
116	fputc('\n', fp);
117	}
118	}
119
120	static int dump_procs(FILE *fp, int look_for_login_process)
121	{
122	struct dirent *entry;
123	DIR *dir = opendir("/proc");
124	int found_login_process = 0;
125
126	fputs(G.jiffy_line, fp);
127	while ((entry = readdir(dir)) != NULL) {
128	char name[sizeof("/proc/%u/cmdline") + sizeof(int)*3];
129	int stat_fd;
130	unsigned pid = bb_strtou(entry->d_name, NULL, 10);
131	if (errno)
132	continue;
133
134	/* Android's version reads /proc/PID/cmdline and extracts
135	* non-truncated process name. Do we want to do that? */
136
137	sprintf(name, "/proc/%u/stat", pid);
138	stat_fd = open(name, O_RDONLY);
139	if (stat_fd >= 0) {
140	char *p;
141	char stat_line[4*1024];
142	int rd = safe_read(stat_fd, stat_line, sizeof(stat_line)-2);
143
144	close(stat_fd);
145	if (rd < 0)
146	continue;
147	stat_line[rd] = '\0';
148	p = strchrnul(stat_line, '\n');
149	*p++ = '\n';
150	*p = '\0';
151	fputs(stat_line, fp);
152	if (!look_for_login_process)
153	continue;
154	p = strchr(stat_line, '(');
155	if (!p)
156	continue;
157	p++;
158	strchrnul(p, ')')[0] = '\0';
159	/* Is it gdm, kdm or a getty? */
160	if (((p[0] == 'g' \|\| p[0] == 'k' \|\| p[0] == 'x') && p[1] == 'd' && p[2] == 'm')
161	\|\| strstr(p, "getty")
162	) {
163	found_login_process = 1;
164	}
165	}
166	}
167	closedir(dir);
168	fputc('\n', fp);
169	return found_login_process;
170	}
171
172	static char *make_tempdir(void)
173	{
174	char template[] = "/tmp/bootchart.XXXXXX";
175	char *tempdir = xstrdup(mkdtemp(template));
176	if (!tempdir) {
177	/* /tmp is not writable (happens when we are used as init).
178	* Try to mount a tmpfs, them cd and lazily unmount it.
179	* Since we unmount it at once, we can mount it anywhere.
180	* Try a few locations which are likely ti exist.
181	*/
182	static const char dirs[] = "/mnt\0""/tmp\0""/boot\0""/proc\0";
183	const char *try_dir = dirs;
184	while (mount("none", try_dir, "tmpfs", MS_SILENT, "size=16m") != 0) {
185	try_dir += strlen(try_dir) + 1;
186	if (!try_dir[0])
187	bb_perror_msg_and_die("can't %smount tmpfs", "");
188	}
189	//bb_error_msg("mounted tmpfs on %s", try_dir);
190	xchdir(try_dir);
191	if (umount2(try_dir, MNT_DETACH) != 0) {
192	bb_perror_msg_and_die("can't %smount tmpfs", "un");
193	}
194	} else {
195	xchdir(tempdir);
196	}
197	return tempdir;
198	}
199
200	static void do_logging(unsigned sample_period_us)
201	{
202	//# Enable process accounting if configured
203	//if [ "$PROCESS_ACCOUNTING" = "yes" ]; then
204	// [ -e kernel_pacct ] \|\| : > kernel_pacct
205	// accton kernel_pacct
206	//fi
207
208	FILE *proc_stat = xfopen("proc_stat.log", "w");
209	FILE *proc_diskstats = xfopen("proc_diskstats.log", "w");
210	//FILE *proc_netdev = xfopen("proc_netdev.log", "w");
211	FILE *proc_ps = xfopen("proc_ps.log", "w");
212	int look_for_login_process = (getppid() == 1);
213	unsigned count = 6010001000 / sample_period_us; /* ~1 minute */
214
215	while (--count && !bb_got_signal) {
216	char *p;
217	int len = open_read_close("/proc/uptime", G.jiffy_line, sizeof(G.jiffy_line)-2);
218	if (len < 0)
219	goto wait_more;
220	/* /proc/uptime has format "NNNNNN.MM NNNNNNN.MM" */
221	/* we convert it to "NNNNNNMM\n" (using first value) */
222	G.jiffy_line[len] = '\0';
223	p = strchr(G.jiffy_line, '.');
224	if (!p)
225	goto wait_more;
226	while (isdigit(*++p))
227	p[-1] = *p;
228	p[-1] = '\n';
229	p[0] = '\0';
230
231	dump_file(proc_stat, "/proc/stat");
232	dump_file(proc_diskstats, "/proc/diskstats");
233	//dump_file(proc_netdev, "/proc/net/dev");
234	if (dump_procs(proc_ps, look_for_login_process)) {
235	/* dump_procs saw a getty or {g,k,x}dm
236	* stop logging in 2 seconds:
237	*/
238	if (count > 210001000 / sample_period_us)
239	count = 210001000 / sample_period_us;
240	}
241	fflush_all();
242	wait_more:
243	usleep(sample_period_us);
244	}
245
246	// [ -e kernel_pacct ] && accton off
247	}
248
249	static void finalize(char tempdir, const char prog)
250	{
251	//# Stop process accounting if configured
252	//local pacct=
253	//[ -e kernel_pacct ] && pacct=kernel_pacct
254
255	FILE *header_fp = xfopen("header", "w");
256
257	if (prog)
258	fprintf(header_fp, "profile.process = %s\n", prog);
259
260	fputs("version = "BC_VERSION_STR"\n", header_fp);
261	if (ENABLE_FEATURE_BOOTCHARTD_BLOATED_HEADER) {
262	char *hostname;
263	char *kcmdline;
264	time_t t;
265	struct tm tm_time;
266	/* x2 for possible localized weekday/month names */
267	char date_buf[sizeof("Mon Jun 21 05:29:03 CEST 2010") * 2];
268	struct utsname unamebuf;
269
270	hostname = safe_gethostname();
271	time(&t);
272	localtime_r(&t, &tm_time);
273	strftime(date_buf, sizeof(date_buf), "%a %b %e %H:%M:%S %Z %Y", &tm_time);
274	fprintf(header_fp, "title = Boot chart for %s (%s)\n", hostname, date_buf);
275	if (ENABLE_FEATURE_CLEAN_UP)
276	free(hostname);
277
278	uname(&unamebuf); /* never fails */
279	/* same as uname -srvm */
280	fprintf(header_fp, "system.uname = %s %s %s %s\n",
281	unamebuf.sysname,
282	unamebuf.release,
283	unamebuf.version,
284	unamebuf.machine
285	);
286
287	//system.release = `cat /etc/DISTRO-release`
288	//system.cpu = `grep '^model name' /proc/cpuinfo \| head -1` ($cpucount)
289
290	kcmdline = xmalloc_open_read_close("/proc/cmdline", NULL);
291	/* kcmdline includes trailing "\n" */
292	fprintf(header_fp, "system.kernel.options = %s", kcmdline);
293	if (ENABLE_FEATURE_CLEAN_UP)
294	free(kcmdline);
295	}
296	fclose(header_fp);
297
298	/* Package log files */
299	system("tar -zcf /var/log/bootchart.tgz header *.log"); // + $pacct
300	/* Clean up (if we are not in detached tmpfs) */
301	if (tempdir) {
302	unlink("header");
303	unlink("proc_stat.log");
304	unlink("proc_diskstats.log");
305	//unlink("proc_netdev.log");
306	unlink("proc_ps.log");
307	rmdir(tempdir);
308	}
309
310	/* shell-based bootchartd tries to run /usr/bin/bootchart if $AUTO_RENDER=yes:
311	* /usr/bin/bootchart -o "$AUTO_RENDER_DIR" -f $AUTO_RENDER_FORMAT "$BOOTLOG_DEST"
312	*/
313	}
314
315	//usage:#define bootchartd_trivial_usage
316	//usage: "start [PROG ARGS]\|stop\|init"
317	//usage:#define bootchartd_full_usage "\n\n"
318	//usage: "Create /var/log/bootchart.tgz with boot chart data\n"
319	//usage: "\nOptions:"
320	//usage: "\nstart: start background logging; with PROG, run PROG, then kill logging with USR1"
321	//usage: "\nstop: send USR1 to all bootchartd processes"
322	//usage: "\ninit: start background logging; stop when getty/xdm is seen (for init scripts)"
323	//usage: "\nUnder PID 1: as init, then exec $bootchart_init, /init, /sbin/init"
324
325	int bootchartd_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
326	int bootchartd_main(int argc UNUSED_PARAM, char **argv)
327	{
328	unsigned sample_period_us;
329	pid_t parent_pid, logger_pid;
330	smallint cmd;
331	enum {
332	CMD_STOP = 0,
333	CMD_START,
334	CMD_INIT,
335	CMD_PID1, /* used to mark pid 1 case */
336	};
337
338	INIT_G();
339
340	parent_pid = getpid();
341	if (argv[1]) {
342	cmd = index_in_strings("stop\0""start\0""init\0", argv[1]);
343	if (cmd < 0)
344	bb_show_usage();
345	if (cmd == CMD_STOP) {
346	pid_t *pidList = find_pid_by_name("bootchartd");
347	while (*pidList != 0) {
348	if (*pidList != parent_pid)
349	kill(*pidList, SIGUSR1);
350	pidList++;
351	}
352	return EXIT_SUCCESS;
353	}
354	} else {
355	if (parent_pid != 1)
356	bb_show_usage();
357	cmd = CMD_PID1;
358	}
359
360	/* Here we are in START, INIT or CMD_PID1 state */
361
362	/* Read config file: */
363	sample_period_us = 200 * 1000;
364	if (ENABLE_FEATURE_BOOTCHARTD_CONFIG_FILE) {
365	char* token[2];
366	parser_t *parser = config_open2("/etc/bootchartd.conf" + 5, fopen_for_read);
367	if (!parser)
368	parser = config_open2("/etc/bootchartd.conf", fopen_for_read);
369	while (config_read(parser, token, 2, 0, "#=", PARSE_NORMAL & ~PARSE_COLLAPSE)) {
370	if (strcmp(token[0], "SAMPLE_PERIOD") == 0 && token[1])
371	sample_period_us = atof(token[1]) * 1000000;
372	}
373	config_close(parser);
374	}
375	if ((int)sample_period_us <= 0)
376	sample_period_us = 1; /* prevent division by 0 */
377
378	/* Create logger child: */
379	logger_pid = fork_or_rexec(argv);
380
381	if (logger_pid == 0) { /* child */
382	char *tempdir;
383
384	bb_signals(0
385	+ (1 << SIGUSR1)
386	+ (1 << SIGUSR2)
387	+ (1 << SIGTERM)
388	+ (1 << SIGQUIT)
389	+ (1 << SIGINT)
390	+ (1 << SIGHUP)
391	, record_signo);
392
393	if (DO_SIGNAL_SYNC)
394	/* Inform parent that we are ready */
395	raise(SIGSTOP);
396
397	/* If we are started by kernel, PATH might be unset.
398	* In order to find "tar", let's set some sane PATH:
399	*/
400	if (cmd == CMD_PID1 && !getenv("PATH"))
401	putenv((char*)bb_PATH_root_path);
402
403	tempdir = make_tempdir();
404	do_logging(sample_period_us);
405	finalize(tempdir, cmd == CMD_START ? argv[2] : NULL);
406	return EXIT_SUCCESS;
407	}
408
409	/* parent */
410
411	if (DO_SIGNAL_SYNC) {
412	/* Wait for logger child to set handlers, then unpause it.
413	* Otherwise with short-lived PROG (e.g. "bootchartd start true")
414	* we might send SIGUSR1 before logger sets its handler.
415	*/
416	waitpid(logger_pid, NULL, WUNTRACED);
417	kill(logger_pid, SIGCONT);
418	}
419
420	if (cmd == CMD_PID1) {
421	char *bootchart_init = getenv("bootchart_init");
422	if (bootchart_init)
423	execl(bootchart_init, bootchart_init, NULL);
424	execl("/init", "init", NULL);
425	execl("/sbin/init", "init", NULL);
426	bb_perror_msg_and_die("can't execute '%s'", "/sbin/init");
427	}
428
429	if (cmd == CMD_START && argv[2]) { /* "start PROG ARGS" */
430	pid_t pid = xvfork();
431	if (pid == 0) { /* child */
432	argv += 2;
433	execvp(argv[0], argv);
434	bb_perror_msg_and_die("can't execute '%s'", argv[0]);
435	}
436	/* parent */
437	waitpid(pid, NULL, 0);
438	kill(logger_pid, SIGUSR1);
439	}
440
441	return EXIT_SUCCESS;
442	}