From 5e61115ea45c621867941e52e6ac016680415656 Mon Sep 17 00:00:00 2001 From: Denys Vlasenko Date: Tue, 19 May 2009 17:36:16 +0200 Subject: [PATCH] scriptreplay: new applet. +423 bytes Signed-off-by: Pascal Bellard Signed-off-by: Denys Vlasenko --- include/applets.h | 1 + include/usage.h | 10 ++++++- util-linux/Config.in | 7 +++++ util-linux/Kbuild | 1 + util-linux/script.c | 59 ++++++++++++++++++++++++++------------- util-linux/scriptreplay.c | 38 +++++++++++++++++++++++++ 6 files changed, 95 insertions(+), 21 deletions(-) create mode 100644 util-linux/scriptreplay.c diff --git a/include/applets.h b/include/applets.h index 359440def..7838757f5 100644 --- a/include/applets.h +++ b/include/applets.h @@ -326,6 +326,7 @@ IF_RUNSV(APPLET(runsv, _BB_DIR_USR_BIN, _BB_SUID_NEVER)) IF_RUNSVDIR(APPLET(runsvdir, _BB_DIR_USR_BIN, _BB_SUID_NEVER)) IF_RX(APPLET(rx, _BB_DIR_USR_BIN, _BB_SUID_NEVER)) IF_SCRIPT(APPLET(script, _BB_DIR_USR_BIN, _BB_SUID_NEVER)) +IF_SCRIPTREPLAY(APPLET(scriptreplay, _BB_DIR_BIN, _BB_SUID_NEVER)) IF_SED(APPLET(sed, _BB_DIR_BIN, _BB_SUID_NEVER)) IF_SELINUXENABLED(APPLET(selinuxenabled, _BB_DIR_USR_SBIN, _BB_SUID_NEVER)) IF_SENDMAIL(APPLET(sendmail, _BB_DIR_USR_SBIN, _BB_SUID_NEVER)) diff --git a/include/usage.h b/include/usage.h index 123462a02..1e327fb97 100644 --- a/include/usage.h +++ b/include/usage.h @@ -3512,6 +3512,11 @@ "\n -g Process group id(s)" \ "\n -u Process user name(s) and/or id(s)" \ +#define scriptreplay_trivial_usage \ + "timingfile [typescript [divisor]]" +#define scriptreplay_full_usage "\n\n" \ + "Play back typescripts, using timing information" + #define reset_trivial_usage \ "" #define reset_full_usage "\n\n" \ @@ -3706,13 +3711,16 @@ "$ rx /tmp/foo\n" #define script_trivial_usage \ - "[-afq] [-c COMMAND] [OUTFILE]" + "[-afq" IF_SCRIPTREPLAY("t") "] [-c COMMAND] [OUTFILE]" #define script_full_usage "\n\n" \ "Options:" \ "\n -a Append output" \ "\n -c Run COMMAND, not shell" \ "\n -f Flush output after each write" \ "\n -q Quiet" \ + IF_SCRIPTREPLAY( \ + "\n -t Send timing to stderr" \ + ) #define sed_trivial_usage \ "[-efinr] pattern [files...]" diff --git a/util-linux/Config.in b/util-linux/Config.in index e5c053fcf..024550172 100644 --- a/util-linux/Config.in +++ b/util-linux/Config.in @@ -764,6 +764,13 @@ config SCRIPT help The script makes typescript of terminal session. +config SCRIPTREPLAY + bool "scriptreplay" + default n + help + This program replays a typescript, using timing information + given by script -t. + config SETARCH bool "setarch" default n diff --git a/util-linux/Kbuild b/util-linux/Kbuild index ac071c620..eaad3319d 100644 --- a/util-linux/Kbuild +++ b/util-linux/Kbuild @@ -33,6 +33,7 @@ lib-$(CONFIG_RDEV) += rdev.o lib-$(CONFIG_READPROFILE) += readprofile.o lib-$(CONFIG_RTCWAKE) += rtcwake.o lib-$(CONFIG_SCRIPT) += script.o +lib-$(CONFIG_SCRIPTREPLAY) += scriptreplay.o lib-$(CONFIG_SETARCH) += setarch.o lib-$(CONFIG_SWAPONOFF) += swaponoff.o lib-$(CONFIG_SWITCH_ROOT) += switch_root.o diff --git a/util-linux/script.c b/util-linux/script.c index a9f24b10e..d16a2914a 100644 --- a/util-linux/script.c +++ b/util-linux/script.c @@ -10,16 +10,8 @@ * * Licensed under GPLv2 or later, see file License in this tarball for details. */ - #include "libbb.h" -static smallint fd_count = 2; - -static void handle_sigchld(int sig UNUSED_PARAM) -{ - fd_count = 0; -} - int script_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int script_main(int argc UNUSED_PARAM, char **argv) { @@ -36,6 +28,15 @@ int script_main(int argc UNUSED_PARAM, char **argv) const char *shell; char shell_opt[] = "-i"; char *shell_arg = NULL; + enum { + OPT_a = (1 << 0), + OPT_c = (1 << 1), + OPT_f = (1 << 2), + OPT_q = (1 << 3), +#if ENABLE_SCRIPTREPLAY + OPT_t = (1 << 4), +#endif + }; #if ENABLE_GETOPT_LONG static const char getopt_longopts[] ALIGN1 = @@ -43,25 +44,28 @@ int script_main(int argc UNUSED_PARAM, char **argv) "command\0" Required_argument "c" "flush\0" No_argument "f" "quiet\0" No_argument "q" +# if ENABLE_SCRIPTREPLAY + "timing\0" No_argument "t" +# endif ; applet_long_options = getopt_longopts; #endif opt_complementary = "?1"; /* max one arg */ - opt = getopt32(argv, "ac:fq", &shell_arg); + opt = getopt32(argv, "ac:fq" IF_SCRIPTREPLAY("t") , &shell_arg); //argc -= optind; argv += optind; if (argv[0]) { fname = argv[0]; } mode = O_CREAT|O_TRUNC|O_WRONLY; - if (opt & 1) { + if (opt & OPT_a) { mode = O_CREAT|O_APPEND|O_WRONLY; } - if (opt & 2) { + if (opt & OPT_c) { shell_opt[1] = 'c'; } - if (!(opt & 8)) { /* not -q */ + if (!(opt & OPT_q)) { printf("Script started, file is %s\n", fname); } shell = getenv("SHELL"); @@ -83,7 +87,7 @@ int script_main(int argc UNUSED_PARAM, char **argv) /* "script" from util-linux exits when child exits, * we wouldn't wait for EOF from slave pty * (output may be produced by grandchildren of child) */ - signal(SIGCHLD, handle_sigchld); + signal(SIGCHLD, record_signo); /* TODO: SIGWINCH? pass window size changes down to slave? */ @@ -97,19 +101,23 @@ int script_main(int argc UNUSED_PARAM, char **argv) #define buf bb_common_bufsiz1 struct pollfd pfd[2]; int outfd, count, loop; +#if ENABLE_SCRIPTREPLAY + double oldtime = time(NULL); +#endif + smallint fd_count = 2; outfd = xopen(fname, mode); pfd[0].fd = pty; pfd[0].events = POLLIN; - pfd[1].fd = 0; + pfd[1].fd = STDIN_FILENO; pfd[1].events = POLLIN; ndelay_on(pty); /* this descriptor is not shared, can do this */ - /* ndelay_on(0); - NO, stdin can be shared! Pity :( */ + /* ndelay_on(STDIN_FILENO); - NO, stdin can be shared! Pity :( */ /* copy stdin to pty master input, * copy pty master output to stdout and file */ /* TODO: don't use full_write's, use proper write buffering */ - while (fd_count) { + while (fd_count && !bb_got_signal) { /* not safe_poll! we want SIGCHLD to EINTR poll */ if (poll(pfd, fd_count, -1) < 0 && errno != EINTR) { /* If child exits too quickly, we may get EIO: @@ -124,9 +132,20 @@ int script_main(int argc UNUSED_PARAM, char **argv) goto restore; } if (count > 0) { +#if ENABLE_SCRIPTREPLAY + if (opt & OPT_t) { + struct timeval tv; + double newtime; + + gettimeofday(&tv, NULL); + newtime = tv.tv_sec + (double) tv.tv_usec / 1000000; + fprintf(stderr, "%f %u\n", newtime - oldtime, count); + oldtime = newtime; + } +#endif full_write(STDOUT_FILENO, buf, count); full_write(outfd, buf, count); - if (opt & 4) { /* -f */ + if (opt & OPT_f) { fsync(outfd); } } @@ -142,8 +161,8 @@ int script_main(int argc UNUSED_PARAM, char **argv) } } } - /* If loop was exited because SIGCHLD handler set fd_count to 0, - * there still can be some buffered output. But not loop forever: + /* If loop was exited because SIGCHLD handler set bb_got_signal, + * there still can be some buffered output. But dont loop forever: * we won't pump orphaned grandchildren's output indefinitely. * Testcase: running this in script: * exec dd if=/dev/zero bs=1M count=1 @@ -158,7 +177,7 @@ int script_main(int argc UNUSED_PARAM, char **argv) restore: if (attr_ok == 0) tcsetattr(0, TCSAFLUSH, &tt); - if (!(opt & 8)) /* not -q */ + if (!(opt & OPT_q)) printf("Script done, file is %s\n", fname); return EXIT_SUCCESS; } diff --git a/util-linux/scriptreplay.c b/util-linux/scriptreplay.c new file mode 100644 index 000000000..038dbdfe1 --- /dev/null +++ b/util-linux/scriptreplay.c @@ -0,0 +1,38 @@ +/* vi: set sw=4 ts=4: */ +/* + * scriptreplay - play back typescripts, using timing information + * + * pascal.bellard@ads-lu.com + * + * Licensed under GPLv2 or later, see file License in this tarball for details. + * + */ +#include "libbb.h" + +int scriptreplay_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; +int scriptreplay_main(int argc UNUSED_PARAM, char **argv) +{ + const char *script = "typescript"; + double delay, factor = 1000000.0; + int fd; + unsigned long count; + FILE *tfp; + + if (argv[2]) { + script = argv[2]; + if (argv[3]) + factor /= atof(argv[3]); + } + + tfp = xfopen_for_read(argv[1]); + fd = xopen(script, O_RDONLY); + while (fscanf(tfp, "%lf %lu\n", &delay, &count) == 2) { + usleep(delay * factor); + bb_copyfd_exact_size(fd, STDOUT_FILENO, count); + } +#if ENABLE_FEATURE_CLEAN_UP + close(fd); + fclose(tfp); +#endif + return EXIT_SUCCESS; +}