From 51fa147c9bab41cc1f7bf5b2e3bbeddf0fdaf5ca Mon Sep 17 00:00:00 2001 From: Denys Vlasenko Date: Fri, 25 Jun 2010 00:57:57 +0200 Subject: [PATCH] conspy: code shrink function old new delta conspy_main 1385 1380 -5 screen_dump 215 202 -13 Signed-off-by: Denys Vlasenko --- miscutils/conspy.c | 49 ++++++++++++++++++++++++++-------------------- 1 file changed, 28 insertions(+), 21 deletions(-) diff --git a/miscutils/conspy.c b/miscutils/conspy.c index b443c9133..b5adce930 100644 --- a/miscutils/conspy.c +++ b/miscutils/conspy.c @@ -9,9 +9,9 @@ * * Licensed under GPLv2 or later, see file License in this tarball for details. * - * example : conspy num shared access to console num - * or conspy -d num screenshot of console num - * or conspy -cs num poor man's GNU screen like + * example: conspy num shared access to console num + * or conspy -d num screenshot of console num + * or conspy -cs num poor man's GNU screen like */ //applet:IF_CONSPY(APPLET(conspy, _BB_DIR_BIN, _BB_SUID_DROP)) @@ -28,7 +28,7 @@ //config: or conspy -cs num poor man's GNU screen like //usage:#define conspy_trivial_usage -//usage: "[-vcsndf] [-x ROW] [-y LINE] [CONSOLE_NO]" +//usage: "[-vcsndf] [-x ROW] [-y LINE] [CONSOLE_NO]" //usage:#define conspy_full_usage "\n\n" //usage: "A text-mode VNC like program for Linux virtual consoles." //usage: "\nTo exit, quickly press ESC 3 times." @@ -50,10 +50,10 @@ struct screen_info { unsigned char lines, rows, cursor_x, cursor_y; }; -#define CHAR(x) ((x)[0]) -#define ATTR(x) ((x)[1]) +#define CHAR(x) ((uint8_t)((x)[0])) +#define ATTR(x) ((uint8_t)((x)[1])) #define NEXT(x) ((x)+=2) -#define DATA(x) (* (short *) (x)) +#define DATA(x) (*(uint16_t*)(x)) struct globals { char* data; @@ -102,7 +102,7 @@ static void screen_read_close(void) unsigned y = i - G.y; // if will catch i < G.y too if (CHAR(data) < ' ') - CHAR(data) = ' '; + *data = ' '; // CHAR(data) = ' '; if (y >= G.height || x >= G.width) DATA(data) = 0; } @@ -151,8 +151,15 @@ static void screen_char(char *data) putchar(CHAR(data)); } -#define clrscr() printf("\033[1;1H" "\033[0J") -#define curoff() printf("\033[?25l") +static void clrscr(void) +{ + printf("\033[1;1H" "\033[0J"); +} + +static void curoff(void) +{ + printf("\033[?25l"); +} static void curon(void) { @@ -180,10 +187,10 @@ static void screen_dump(void) if (tty_row >= G.width) continue; space_cnt++; - if (BW && (CHAR(data) | ' ') == ' ') + if (BW && CHAR(data) == ' ') continue; while (linefeed_cnt != 0) { - bb_putchar('\r'); + //bb_putchar('\r'); - tty driver does it for us bb_putchar('\n'); linefeed_cnt--; } @@ -229,12 +236,11 @@ static void cleanup(int code) static void get_initial_data(const char* vcsa_name) { - int size; G.vcsa_fd = xopen(vcsa_name, O_RDONLY); xread(G.vcsa_fd, &G.info, 4); - G.size = size = G.info.rows * G.info.lines * 2; + G.size = G.info.rows * G.info.lines * 2; G.width = G.height = UINT_MAX; - G.data = xzalloc(2 * size); + G.data = xzalloc(2 * G.size); screen_read_close(); } @@ -280,11 +286,11 @@ static NOINLINE void start_shell_in_child(const char* tty_name) } } -int conspy_main(int argc UNUSED_PARAM, char **argv) MAIN_EXTERNALLY_VISIBLE; +int conspy_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int conspy_main(int argc UNUSED_PARAM, char **argv) { - char vcsa_name[sizeof("/dev/vcsa") + 2]; - char tty_name[sizeof("/dev/tty") + 2]; + char vcsa_name[sizeof("/dev/vcsaNN")]; + char tty_name[sizeof("/dev/ttyNN")]; #define keybuf bb_common_bufsiz1 struct termios termbuf; unsigned opts; @@ -339,7 +345,7 @@ int conspy_main(int argc UNUSED_PARAM, char **argv) tcgetattr(G.kbd_fd, &G.term_orig); termbuf = G.term_orig; termbuf.c_iflag &= ~(BRKINT|INLCR|ICRNL|IXON|IXOFF|IUCLC|IXANY|IMAXBEL); - termbuf.c_oflag &= ~(OPOST); + //termbuf.c_oflag &= ~(OPOST); - no, we still want \n -> \r\n termbuf.c_lflag &= ~(ISIG|ICANON|ECHO); termbuf.c_cc[VMIN] = 1; termbuf.c_cc[VTIME] = 0; @@ -354,7 +360,7 @@ int conspy_main(int argc UNUSED_PARAM, char **argv) old = G.data + G.current; G.current = G.size - G.current; data = G.data + G.current; - + // Close & re-open vcsa in case they have // swapped virtual consoles G.vcsa_fd = xopen(vcsa_name, O_RDONLY); @@ -448,7 +454,8 @@ int conspy_main(int argc UNUSED_PARAM, char **argv) // Do exit processing for (i = 0; i < bytes_read; i++) { - if (k[i] != '\033') G.escape_count = 0; + if (k[i] != '\033') + G.escape_count = 0; else if (++G.escape_count >= 3) cleanup(0); }