toys/pending/bootchartd.c - third_party/toybox - Git at Google

 /* bootchartd.c - bootchartd is commonly used to profile the boot process.
  *
  * Copyright 2014 Bilal Qureshi <bilal.jmi@gmail.com>
  * Copyright 2014 Kyungwan Han <asura321@gmail.com>
  *
  * No Standard

 USE_BOOTCHARTD(NEWTOY(bootchartd, 0, TOYFLAG_STAYROOT|TOYFLAG_USR|TOYFLAG_BIN))

 config BOOTCHARTD
   bool "bootchartd"
   default n
   depends on TOYBOX_FORK
   help
     usage: bootchartd {start [PROG ARGS]}|stop|init

     Create /var/log/bootlog.tgz with boot chart data

     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/xdm is seen
           (for init scripts)

     Under PID 1: as init, then exec $bootchart_init, /init, /sbin/init
 */

 #define FOR_bootchartd
 #include "toys.h"

 GLOBALS(
   char buf[32];
   long smpl_period_usec;
   int proc_accounting;
   int is_login;

   void *head;
 )

 struct pid_list {
   struct pid_list *next, *prev;
   int pid;
 };

 static int push_pids_in_list(pid_t pid, char *name)
 {
   struct pid_list *new = xzalloc(sizeof(struct pid_list));

   new->pid = pid;
   dlist_add_nomalloc((void *)&TT.head, (void *)new);

   return 0;
 }

 static void dump_data_in_file(char *fname, int wfd)
 {
   int rfd = open(fname, O_RDONLY);

   if (rfd != -1) {
     xwrite(wfd, TT.buf, strlen(TT.buf));
     xsendfile(rfd, wfd);
     close(rfd);
     xwrite(wfd, "\n", 1);
   }
 }

 static int dump_proc_data(FILE *fp)
 {
   struct dirent *pid_dir;
   int login_flag = 0;
   pid_t pid;
   DIR *proc_dir = opendir("/proc");

   fputs(TT.buf, fp);
   while ((pid_dir = readdir(proc_dir))) {
     char filename[64];
     int fd;

     if (!isdigit(pid_dir->d_name[0])) continue;
     sscanf(pid_dir->d_name, "%d", &pid);
     sprintf(filename, "/proc/%d/stat", pid);
     if ((fd = open(filename, O_RDONLY)) != -1 ) {
       char *ptr;
       ssize_t len;

       if ((len = readall(fd, toybuf, sizeof(toybuf)-1)) < 0) {
         xclose(fd);
         continue;
       }
       toybuf[len] = '\0';
       close(fd);
       fputs(toybuf, fp);
       if (!TT.is_login) continue;
       if ((ptr = strchr(toybuf, '('))) {
         char *tmp = strchr(++ptr, ')');

         if (tmp) *tmp = '\0';
       }
       // Checks for gdm, kdm or getty
       if (((ptr[0] == 'g' || ptr[0] == 'k' || ptr[0] == 'x') && ptr[1] == 'd'
             && ptr[2] == 'm') || strstr(ptr, "getty")) login_flag = 1;
     }
   }
   closedir(proc_dir);
   fputc('\n', fp);
   return login_flag;
 }

 static int parse_config_file(char *fname)
 {
   size_t len = 0;
   char  *line = NULL;
   FILE *fp = fopen(fname, "r");

   if (!fp) return 0;
   for (;getline(&line, &len, fp) != -1; line = NULL) {
     char *ptr = line;

     while (*ptr == ' ' || *ptr == '\t') ptr++;
     if (!*ptr || *ptr == '#' || *ptr == '\n') continue;
     if (!strncmp(ptr, "SAMPLE_PERIOD", strlen("SAMPLE_PERIOD"))) {
       double smpl_val;

       if ((ptr = strchr(ptr, '='))) ptr += 1;
       else continue;
       sscanf(ptr, "%lf", &smpl_val);
       TT.smpl_period_usec = smpl_val * 1000000;
       if (TT.smpl_period_usec <= 0) TT.smpl_period_usec = 1;
     }
     if (!strncmp(ptr, "PROCESS_ACCOUNTING", strlen("PROCESS_ACCOUNTING"))) {
       if ((ptr = strchr(ptr, '='))) ptr += 1;
       else continue;
       sscanf(ptr, "%s", toybuf);  // string will come with double quotes.
       if (!(strncmp(toybuf+1, "on", strlen("on"))) ||
           !(strncmp(toybuf+1, "yes", strlen("yes")))) TT.proc_accounting = 1;
     }
     free(line);
   }
   fclose(fp);
   return 1;
 }

 static char *create_tmp_dir()
 {
   char *dir_list[] = {"/tmp", "/mnt", "/boot", "/proc"}, **target = dir_list;
   char *dir, dir_path[] = "/tmp/bootchart.XXXXXX";

   if ((dir = mkdtemp(dir_path))) {
     xchdir((dir = xstrdup(dir)));
     return dir;
   }
   while (mount("none", *target, "tmpfs", (1<<15), "size=16m")) //MS_SILENT
     if (!++target) perror_exit("can't mount tmpfs");
   xchdir(*target);
   if (umount2(*target, MNT_DETACH)) perror_exit("Can't unmount tmpfs");
   return *target;
 }

 static void start_logging()
 {
   int proc_stat_fd = xcreate("proc_stat.log",
       O_WRONLY | O_CREAT | O_TRUNC, 0644);
   int proc_diskstats_fd = xcreate("proc_diskstats.log",
       O_WRONLY | O_CREAT | O_TRUNC, 0644);
   FILE *proc_ps_fp = xfopen("proc_ps.log", "w");
   long tcnt = 60 * 1000 * 1000 / TT.smpl_period_usec;

   if (tcnt <= 0) tcnt = 1;
   if (TT.proc_accounting) {
     int kp_fd = xcreate("kernel_procs_acct", O_WRONLY | O_CREAT | O_TRUNC,0666);

     xclose(kp_fd);
     acct("kernel_procs_acct");
   }
   memset(TT.buf, 0, sizeof(TT.buf));
   while (--tcnt && !toys.signal) {
     int i = 0, j = 0, fd = open("/proc/uptime", O_RDONLY);
     if (fd < 0) goto wait_usec;
     char *line = get_line(fd);

     if (!line)  goto wait_usec;
     while (line[i] != ' ') {
       if (line[i] == '.') {
         i++;
         continue;
       }
       TT.buf[j++] = line[i++];
     }
     TT.buf[j++] = '\n';
     TT.buf[j] = '\0';
     free(line);
     close(fd);
     dump_data_in_file("/proc/stat", proc_stat_fd);
     dump_data_in_file("/proc/diskstats", proc_diskstats_fd);
     // stop proc dumping in 2 secs if getty or gdm, kdm, xdm found
     if (dump_proc_data(proc_ps_fp))
       if (tcnt > 2 * 1000 * 1000 / TT.smpl_period_usec)
         tcnt = 2 * 1000 * 1000 / TT.smpl_period_usec;
     fflush(NULL);
 wait_usec:
     usleep(TT.smpl_period_usec);
   }
   xclose(proc_stat_fd);
   xclose(proc_diskstats_fd);
   fclose(proc_ps_fp);
 }

 static void stop_logging(char *tmp_dir, char *prog)
 {
   char host_name[32];
   int kcmd_line_fd;
   time_t t;
   struct tm st;
   struct utsname uts;
   FILE *hdr_fp = xfopen("header", "w");

   if (TT.proc_accounting) acct(NULL);
   if (prog) fprintf(hdr_fp, "profile.process = %s\n", prog);
   gethostname(host_name, sizeof(host_name));
   time(&t);
   localtime_r(&t, &st);
   memset(toybuf, 0, sizeof(toybuf));
   strftime(toybuf, sizeof(toybuf), "%a %b %e %H:%M:%S %Z %Y", &st);
   fprintf(hdr_fp, "version = TBX_BCHARTD_VER 1.0.0\n");
   fprintf(hdr_fp, "title = Boot chart for %s (%s)\n", host_name, toybuf);
   if (uname(&uts) < 0) perror_exit("uname");
   fprintf(hdr_fp, "system.uname = %s %s %s %s\n", uts.sysname, uts.release,
       uts.version, uts.machine);
   memset(toybuf, 0, sizeof(toybuf));
   if ((kcmd_line_fd = open("/proc/cmdline", O_RDONLY)) != -1) {
     ssize_t len;

     if ((len = readall(kcmd_line_fd, toybuf, sizeof(toybuf)-1)) > 0) {
       toybuf[len] = 0;
       while (--len >= 0 && !toybuf[len]) continue;
       for (; len > 0; len--) if (toybuf[len] < ' ') toybuf[len] = ' ';
     } else *toybuf = 0;
   }
   fprintf(hdr_fp, "system.kernel.options = %s", toybuf);
   close(kcmd_line_fd);
   fclose(hdr_fp);
   memset(toybuf, 0, sizeof(toybuf));
   snprintf(toybuf, sizeof(toybuf), "tar -zcf /var/log/bootlog.tgz header %s *.log",
       TT.proc_accounting ? "kernel_procs_acct" : "");
   system(toybuf);
   if (tmp_dir) {
     unlink("header");
     unlink("proc_stat.log");
     unlink("proc_diskstats.log");
     unlink("proc_ps.log");
     if (TT.proc_accounting) unlink("kernel_procs_acct");
     rmdir(tmp_dir);
   }
 }

 void bootchartd_main()
 {
   pid_t lgr_pid, self_pid = getpid();
   int bchartd_opt = 0; // 0=PID1, 1=start, 2=stop, 3=init
   TT.smpl_period_usec = 200 * 1000;

   TT.is_login = (self_pid == 1);
   if (*toys.optargs) {
     if (!strcmp("start", *toys.optargs)) bchartd_opt = 1;
     else if (!strcmp("stop", *toys.optargs)) bchartd_opt = 2;
     else if (!strcmp("init", *toys.optargs)) bchartd_opt = 3;
     else error_exit("Unknown option '%s'", *toys.optargs);

     if (bchartd_opt == 2) {
       struct pid_list *temp;
       char *process_name[] = {"bootchartd", NULL};

       names_to_pid(process_name, push_pids_in_list);
       temp = TT.head;
       if (temp) temp->prev->next = 0;
       for (; temp; temp = temp->next)
         if (temp->pid != self_pid) kill(temp->pid, SIGUSR1);
       llist_traverse(TT.head, free);

       return;
     }
   } else if (!TT.is_login) error_exit("not PID 1");

   // Execute the code below for start or init or PID1
   if (!parse_config_file("bootchartd.conf"))
     parse_config_file("/etc/bootchartd.conf");

   memset(toybuf, 0, sizeof(toybuf));
   if (!(lgr_pid = xfork())) {
     char *tmp_dir = create_tmp_dir();

     sigatexit(generic_signal);
     raise(SIGSTOP);
     if (!bchartd_opt && !getenv("PATH"))
       putenv("PATH=/sbin:/usr/sbin:/bin:/usr/bin");
     start_logging();
     stop_logging(tmp_dir, bchartd_opt == 1 ? toys.optargs[1] : NULL);
     return;
   }
   waitpid(lgr_pid, NULL, WUNTRACED);
   kill(lgr_pid, SIGCONT);

   if (!bchartd_opt) {
     char *pbchart_init = getenv("bootchart_init");

     if (pbchart_init) execl(pbchart_init, pbchart_init, NULL);
     execl("/init", "init", (void *)0);
     execl("/sbin/init", "init", (void *)0);
   }
   if (bchartd_opt == 1 && toys.optargs[1]) {
     pid_t prog_pid;

     if (!(prog_pid = xfork())) xexec(toys.optargs+1);
     waitpid(prog_pid, NULL, 0);
     kill(lgr_pid, SIGUSR1);
   }
 }
	/* bootchartd.c - bootchartd is commonly used to profile the boot process.
	*
	* Copyright 2014 Bilal Qureshi <bilal.jmi@gmail.com>
	* Copyright 2014 Kyungwan Han <asura321@gmail.com>
	*
	* No Standard

	USE_BOOTCHARTD(NEWTOY(bootchartd, 0, TOYFLAG_STAYROOT\|TOYFLAG_USR\|TOYFLAG_BIN))

	config BOOTCHARTD
	bool "bootchartd"
	default n
	depends on TOYBOX_FORK
	help
	usage: bootchartd {start [PROG ARGS]}\|stop\|init

	Create /var/log/bootlog.tgz with boot chart data

	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/xdm is seen
	(for init scripts)

	Under PID 1: as init, then exec $bootchart_init, /init, /sbin/init
	*/

	#define FOR_bootchartd
	#include "toys.h"

	GLOBALS(
	char buf[32];
	long smpl_period_usec;
	int proc_accounting;
	int is_login;

	void *head;
	)

	struct pid_list {
	struct pid_list next, prev;
	int pid;
	};

	static int push_pids_in_list(pid_t pid, char *name)
	{
	struct pid_list *new = xzalloc(sizeof(struct pid_list));

	new->pid = pid;
	dlist_add_nomalloc((void )&TT.head, (void )new);

	return 0;
	}

	static void dump_data_in_file(char *fname, int wfd)
	{
	int rfd = open(fname, O_RDONLY);

	if (rfd != -1) {
	xwrite(wfd, TT.buf, strlen(TT.buf));
	xsendfile(rfd, wfd);
	close(rfd);
	xwrite(wfd, "\n", 1);
	}
	}

	static int dump_proc_data(FILE *fp)
	{
	struct dirent *pid_dir;
	int login_flag = 0;
	pid_t pid;
	DIR *proc_dir = opendir("/proc");

	fputs(TT.buf, fp);
	while ((pid_dir = readdir(proc_dir))) {
	char filename[64];
	int fd;

	if (!isdigit(pid_dir->d_name[0])) continue;
	sscanf(pid_dir->d_name, "%d", &pid);
	sprintf(filename, "/proc/%d/stat", pid);
	if ((fd = open(filename, O_RDONLY)) != -1 ) {
	char *ptr;
	ssize_t len;

	if ((len = readall(fd, toybuf, sizeof(toybuf)-1)) < 0) {
	xclose(fd);
	continue;
	}
	toybuf[len] = '\0';
	close(fd);
	fputs(toybuf, fp);
	if (!TT.is_login) continue;
	if ((ptr = strchr(toybuf, '('))) {
	char *tmp = strchr(++ptr, ')');

	if (tmp) *tmp = '\0';
	}
	// Checks for gdm, kdm or getty
	if (((ptr[0] == 'g' \|\| ptr[0] == 'k' \|\| ptr[0] == 'x') && ptr[1] == 'd'
	&& ptr[2] == 'm') \|\| strstr(ptr, "getty")) login_flag = 1;
	}
	}
	closedir(proc_dir);
	fputc('\n', fp);
	return login_flag;
	}

	static int parse_config_file(char *fname)
	{
	size_t len = 0;
	char *line = NULL;
	FILE *fp = fopen(fname, "r");

	if (!fp) return 0;
	for (;getline(&line, &len, fp) != -1; line = NULL) {
	char *ptr = line;

	while (ptr == ' ' \|\| ptr == '\t') ptr++;
	if (!ptr \|\| ptr == '#' \|\| *ptr == '\n') continue;
	if (!strncmp(ptr, "SAMPLE_PERIOD", strlen("SAMPLE_PERIOD"))) {
	double smpl_val;

	if ((ptr = strchr(ptr, '='))) ptr += 1;
	else continue;
	sscanf(ptr, "%lf", &smpl_val);
	TT.smpl_period_usec = smpl_val * 1000000;
	if (TT.smpl_period_usec <= 0) TT.smpl_period_usec = 1;
	}
	if (!strncmp(ptr, "PROCESS_ACCOUNTING", strlen("PROCESS_ACCOUNTING"))) {
	if ((ptr = strchr(ptr, '='))) ptr += 1;
	else continue;
	sscanf(ptr, "%s", toybuf); // string will come with double quotes.
	if (!(strncmp(toybuf+1, "on", strlen("on"))) \|\|
	!(strncmp(toybuf+1, "yes", strlen("yes")))) TT.proc_accounting = 1;
	}
	free(line);
	}
	fclose(fp);
	return 1;
	}

	static char *create_tmp_dir()
	{
	char dir_list[] = {"/tmp", "/mnt", "/boot", "/proc"}, *target = dir_list;
	char *dir, dir_path[] = "/tmp/bootchart.XXXXXX";

	if ((dir = mkdtemp(dir_path))) {
	xchdir((dir = xstrdup(dir)));
	return dir;
	}
	while (mount("none", *target, "tmpfs", (1<<15), "size=16m")) //MS_SILENT
	if (!++target) perror_exit("can't mount tmpfs");
	xchdir(*target);
	if (umount2(*target, MNT_DETACH)) perror_exit("Can't unmount tmpfs");
	return *target;
	}

	static void start_logging()
	{
	int proc_stat_fd = xcreate("proc_stat.log",
	O_WRONLY \| O_CREAT \| O_TRUNC, 0644);
	int proc_diskstats_fd = xcreate("proc_diskstats.log",
	O_WRONLY \| O_CREAT \| O_TRUNC, 0644);
	FILE *proc_ps_fp = xfopen("proc_ps.log", "w");
	long tcnt = 60 * 1000 * 1000 / TT.smpl_period_usec;

	if (tcnt <= 0) tcnt = 1;
	if (TT.proc_accounting) {
	int kp_fd = xcreate("kernel_procs_acct", O_WRONLY \| O_CREAT \| O_TRUNC,0666);

	xclose(kp_fd);
	acct("kernel_procs_acct");
	}
	memset(TT.buf, 0, sizeof(TT.buf));
	while (--tcnt && !toys.signal) {
	int i = 0, j = 0, fd = open("/proc/uptime", O_RDONLY);
	if (fd < 0) goto wait_usec;
	char *line = get_line(fd);

	if (!line) goto wait_usec;
	while (line[i] != ' ') {
	if (line[i] == '.') {
	i++;
	continue;
	}
	TT.buf[j++] = line[i++];
	}
	TT.buf[j++] = '\n';
	TT.buf[j] = '\0';
	free(line);
	close(fd);
	dump_data_in_file("/proc/stat", proc_stat_fd);
	dump_data_in_file("/proc/diskstats", proc_diskstats_fd);
	// stop proc dumping in 2 secs if getty or gdm, kdm, xdm found
	if (dump_proc_data(proc_ps_fp))
	if (tcnt > 2 * 1000 * 1000 / TT.smpl_period_usec)
	tcnt = 2 * 1000 * 1000 / TT.smpl_period_usec;
	fflush(NULL);
	wait_usec:
	usleep(TT.smpl_period_usec);
	}
	xclose(proc_stat_fd);
	xclose(proc_diskstats_fd);
	fclose(proc_ps_fp);
	}

	static void stop_logging(char tmp_dir, char prog)
	{
	char host_name[32];
	int kcmd_line_fd;
	time_t t;
	struct tm st;
	struct utsname uts;
	FILE *hdr_fp = xfopen("header", "w");

	if (TT.proc_accounting) acct(NULL);
	if (prog) fprintf(hdr_fp, "profile.process = %s\n", prog);
	gethostname(host_name, sizeof(host_name));
	time(&t);
	localtime_r(&t, &st);
	memset(toybuf, 0, sizeof(toybuf));
	strftime(toybuf, sizeof(toybuf), "%a %b %e %H:%M:%S %Z %Y", &st);
	fprintf(hdr_fp, "version = TBX_BCHARTD_VER 1.0.0\n");
	fprintf(hdr_fp, "title = Boot chart for %s (%s)\n", host_name, toybuf);
	if (uname(&uts) < 0) perror_exit("uname");
	fprintf(hdr_fp, "system.uname = %s %s %s %s\n", uts.sysname, uts.release,
	uts.version, uts.machine);
	memset(toybuf, 0, sizeof(toybuf));
	if ((kcmd_line_fd = open("/proc/cmdline", O_RDONLY)) != -1) {
	ssize_t len;

	if ((len = readall(kcmd_line_fd, toybuf, sizeof(toybuf)-1)) > 0) {
	toybuf[len] = 0;
	while (--len >= 0 && !toybuf[len]) continue;
	for (; len > 0; len--) if (toybuf[len] < ' ') toybuf[len] = ' ';
	} else *toybuf = 0;
	}
	fprintf(hdr_fp, "system.kernel.options = %s", toybuf);
	close(kcmd_line_fd);
	fclose(hdr_fp);
	memset(toybuf, 0, sizeof(toybuf));
	snprintf(toybuf, sizeof(toybuf), "tar -zcf /var/log/bootlog.tgz header %s *.log",
	TT.proc_accounting ? "kernel_procs_acct" : "");
	system(toybuf);
	if (tmp_dir) {
	unlink("header");
	unlink("proc_stat.log");
	unlink("proc_diskstats.log");
	unlink("proc_ps.log");
	if (TT.proc_accounting) unlink("kernel_procs_acct");
	rmdir(tmp_dir);
	}
	}

	void bootchartd_main()
	{
	pid_t lgr_pid, self_pid = getpid();
	int bchartd_opt = 0; // 0=PID1, 1=start, 2=stop, 3=init
	TT.smpl_period_usec = 200 * 1000;

	TT.is_login = (self_pid == 1);
	if (*toys.optargs) {
	if (!strcmp("start", *toys.optargs)) bchartd_opt = 1;
	else if (!strcmp("stop", *toys.optargs)) bchartd_opt = 2;
	else if (!strcmp("init", *toys.optargs)) bchartd_opt = 3;
	else error_exit("Unknown option '%s'", *toys.optargs);

	if (bchartd_opt == 2) {
	struct pid_list *temp;
	char *process_name[] = {"bootchartd", NULL};

	names_to_pid(process_name, push_pids_in_list);
	temp = TT.head;
	if (temp) temp->prev->next = 0;
	for (; temp; temp = temp->next)
	if (temp->pid != self_pid) kill(temp->pid, SIGUSR1);
	llist_traverse(TT.head, free);

	return;
	}
	} else if (!TT.is_login) error_exit("not PID 1");

	// Execute the code below for start or init or PID1
	if (!parse_config_file("bootchartd.conf"))
	parse_config_file("/etc/bootchartd.conf");

	memset(toybuf, 0, sizeof(toybuf));
	if (!(lgr_pid = xfork())) {
	char *tmp_dir = create_tmp_dir();

	sigatexit(generic_signal);
	raise(SIGSTOP);
	if (!bchartd_opt && !getenv("PATH"))
	putenv("PATH=/sbin:/usr/sbin:/bin:/usr/bin");
	start_logging();
	stop_logging(tmp_dir, bchartd_opt == 1 ? toys.optargs[1] : NULL);
	return;
	}
	waitpid(lgr_pid, NULL, WUNTRACED);
	kill(lgr_pid, SIGCONT);

	if (!bchartd_opt) {
	char *pbchart_init = getenv("bootchart_init");

	if (pbchart_init) execl(pbchart_init, pbchart_init, NULL);
	execl("/init", "init", (void *)0);
	execl("/sbin/init", "init", (void *)0);
	}
	if (bchartd_opt == 1 && toys.optargs[1]) {
	pid_t prog_pid;

	if (!(prog_pid = xfork())) xexec(toys.optargs+1);
	waitpid(prog_pid, NULL, 0);
	kill(lgr_pid, SIGUSR1);
	}
	}