emo/busybox
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

conspy: code shrink

Browse Source 
function                                             old     new   delta
screen_read_close                                    114     187     +73
set_cursor                                             -      53     +53
cleanup                                               93      91      -2
screen_dump                                          202     197      -5
curon                                                 34       -     -34
screen_char                                          351     308     -43
conspy_main                                         1444    1264    -180
------------------------------------------------------------------------------
(add/remove: 1/1 grow/shrink: 1/4 up/down: 126/-264)         Total: -138 bytes

Signed-off-by: Pascal Bellard <pascal.bellard@ads-lu.com>
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
This commit is contained in:
Pascal Bellard 2010-06-30 07:05:31 +02:00 committed by Denys Vlasenko
parent c03fb3cd7d
commit fa5ea17b5f
1 changed files with 100 additions and 109 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

203
miscutils/conspy.c
View File

@ -46,9 +46,9 @@ struct screen_info {
unsigned char lines, cols, cursor_x, cursor_y; unsigned char lines, cols, cursor_x, cursor_y;
}; };
#define CHAR(x) ((uint8_t)((x)[0])) #define CHAR(x) (*(uint8_t*)(x))
#define ATTR(x) ((uint8_t)((x)[1])) #define ATTR(x) (((uint8_t*)(x))[1])
#define NEXT(x) ((x)+=2) #define NEXT(x) ((x) += 2)
#define DATA(x) (*(uint16_t*)(x)) #define DATA(x) (*(uint16_t*)(x))
struct globals { struct globals {
@ -56,25 +56,25 @@ struct globals {
int size; int size;
int x, y; int x, y;
int kbd_fd; int kbd_fd;
int vcsa_fd;
int ioerror_count; int ioerror_count;
int key_count; int key_count;
int escape_count; int escape_count;
int nokeys; int nokeys;
int current; int current;
int first_line_offset;
int last_attr;
// cached local tty parameters // cached local tty parameters
unsigned width; unsigned width;
unsigned height; unsigned height;
unsigned col; unsigned col;
unsigned line; unsigned line;
smallint curoff; smallint curoff; // unknown:0 cursor on:-1 cursor off:1
uint8_t last_attr;
uint8_t force_attr_change;
char attrbuf[sizeof("\033[0;1;5;30;40m")]; char attrbuf[sizeof("\033[0;1;5;30;40m")];
// remote console // remote console
struct screen_info remote; struct screen_info remote;
// saved local tty terminfo // saved local tty terminfo
struct termios term_orig; struct termios term_orig;
char vcsa_name[sizeof("/dev/vcsaNN")];
}; };
#define G (*ptr_to_globals) #define G (*ptr_to_globals)
@ -82,7 +82,8 @@ struct globals {
SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \ SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
G.attrbuf[0] = '\033'; \ G.attrbuf[0] = '\033'; \
G.attrbuf[1] = '['; \ G.attrbuf[1] = '['; \
G.force_attr_change = 0xff; \ G.width = G.height = UINT_MAX; \
G.last_attr--; \
} while (0) } while (0)
enum { enum {
@ -96,24 +97,80 @@ enum {
#define FLAG(x) (1 << FLAG_##x) #define FLAG(x) (1 << FLAG_##x)
#define BW (option_mask32 & FLAG(n)) #define BW (option_mask32 & FLAG(n))
static void clrscr(void)
{
// Home, clear till end of screen
fputs("\033[1;1H" "\033[J", stdout);
G.col = G.line = 0;
}
static void set_cursor(int state)
{
if (G.curoff != state) {
G.curoff = state;
fputs("\033[?25", stdout);
bb_putchar("h?l"[1 + state]);
}
}
static void gotoxy(int col, int line)
{
if (G.col != col || G.line != line) {
G.col = col;
G.line = line;
printf("\033[%u;%uH", line + 1, col + 1);
}
}
static void cleanup(int code)
{
set_cursor(-1); // cursor on
tcsetattr(G.kbd_fd, TCSANOW, &G.term_orig);
if (ENABLE_FEATURE_CLEAN_UP) {
close(G.kbd_fd);
}
// Reset attributes
if (!BW)
fputs("\033[0m", stdout);
bb_putchar('\n');
if (code > 1)
kill_myself_with_sig(code);
exit(code);
}
static void screen_read_close(void) static void screen_read_close(void)
{ {
unsigned i, j; unsigned i, j;
char *data = G.data + G.current; int vcsa_fd;
char *data;
xread(G.vcsa_fd, data, G.size); // Close & re-open vcsa in case they have swapped virtual consoles
vcsa_fd = xopen(G.vcsa_name, O_RDONLY);
xread(vcsa_fd, &G.remote, 4);
i = G.remote.cols * 2;
G.first_line_offset = G.y * i;
i *= G.remote.lines;
if (G.data == NULL) {
G.size = i;
G.data = xzalloc(2 * i);
}
else if (G.size != i) {
cleanup(1);
}
data = G.data + G.current;
xread(vcsa_fd, data, G.size);
close(vcsa_fd);
for (i = 0; i < G.remote.lines; i++) { for (i = 0; i < G.remote.lines; i++) {
for (j = 0; j < G.remote.cols; j++, NEXT(data)) { for (j = 0; j < G.remote.cols; j++, NEXT(data)) {
unsigned x = j - G.x; // if will catch j < G.x too unsigned x = j - G.x; // if will catch j < G.x too
unsigned y = i - G.y; // if will catch i < G.y too unsigned y = i - G.y; // if will catch i < G.y too
if (CHAR(data) < ' ') if (CHAR(data) < ' ')
*data = ' '; // CHAR(data) = ' '; CHAR(data) = ' ';
if (y >= G.height || x >= G.width) if (y >= G.height || x >= G.width)
DATA(data) = 0; DATA(data) = 0;
} }
} }
close(G.vcsa_fd);
} }
static void screen_char(char *data) static void screen_char(char *data)
@ -121,7 +178,7 @@ static void screen_char(char *data)
if (!BW) { if (!BW) {
uint8_t attr = ATTR(data); uint8_t attr = ATTR(data);
//uint8_t attr = ATTR(data) >> 1; // for framebuffer console //uint8_t attr = ATTR(data) >> 1; // for framebuffer console
uint8_t attr_diff = (G.last_attr ^ attr) | G.force_attr_change; uint8_t attr_diff = G.last_attr ^ attr;
if (attr_diff) { if (attr_diff) {
// Attribute layout for VGA compatible text videobuffer: // Attribute layout for VGA compatible text videobuffer:
@ -153,19 +210,20 @@ static void screen_char(char *data)
const uint8_t bg_mask = 0x70, blink_mask = 0x80; const uint8_t bg_mask = 0x70, blink_mask = 0x80;
char *ptr; char *ptr;
ptr = G.attrbuf + 2; /* skip "ESC [" */ ptr = G.attrbuf + 2; // skip "ESC ["
// (G.last_attr & ~attr) has 1 only where // (G.last_attr & ~attr) has 1 only where
// G.last_attr has 1 but attr has 0. // G.last_attr has 1 but attr has 0.
// Here we check whether we have transition // Here we check whether we have transition
// bold->non-bold or blink->non-blink: // bold->non-bold or blink->non-blink:
if ((G.last_attr & ~attr) & (bold_mask | blink_mask)) { if (G.last_attr < 0 // initial value
|| ((G.last_attr & ~attr) & (bold_mask | blink_mask)) != 0
) {
*ptr++ = '0'; // "reset all attrs" *ptr++ = '0'; // "reset all attrs"
*ptr++ = ';'; *ptr++ = ';';
// must set fg & bg, maybe need to set bold or blink: // must set fg & bg, maybe need to set bold or blink:
attr_diff = attr | ~(bold_mask | blink_mask); attr_diff = attr | ~(bold_mask | blink_mask);
} }
G.force_attr_change = 0;
G.last_attr = attr; G.last_attr = attr;
if (attr_diff & bold_mask) { if (attr_diff & bold_mask) {
*ptr++ = '1'; *ptr++ = '1';
@ -196,44 +254,12 @@ static void screen_char(char *data)
G.col++; G.col++;
} }
static void clrscr(void)
{
// Home, clear till end of src, cursor on
fputs("\033[1;1H" "\033[J" "\033[?25h", stdout);
G.curoff = G.col = G.line = 0;
}
static void curoff(void)
{
if (!G.curoff) {
G.curoff = 1;
fputs("\033[?25l", stdout);
}
}
static void curon(void)
{
if (G.curoff) {
G.curoff = 0;
fputs("\033[?25h", stdout);
}
}
static void gotoxy(int col, int line)
{
if (G.col != col || G.line != line) {
G.col = col;
G.line = line;
printf("\033[%u;%uH", line + 1, col + 1);
}
}
static void screen_dump(void) static void screen_dump(void)
{ {
int linefeed_cnt; int linefeed_cnt;
int line, col; int line, col;
int linecnt = G.remote.lines - G.y; int linecnt = G.remote.lines - G.y;
char *data = G.data + G.current + (2 * G.y * G.remote.cols); char *data = G.data + G.current + G.first_line_offset;
linefeed_cnt = 0; linefeed_cnt = 0;
for (line = 0; line < linecnt && line < G.height; line++) { for (line = 0; line < linecnt && line < G.height; line++) {
@ -263,41 +289,13 @@ static void curmove(void)
{ {
unsigned cx = G.remote.cursor_x - G.x; unsigned cx = G.remote.cursor_x - G.x;
unsigned cy = G.remote.cursor_y - G.y; unsigned cy = G.remote.cursor_y - G.y;
int cursor = 1;
if (cx >= G.width || cy >= G.height) { if (cx < G.width && cy < G.height) {
curoff();
} else {
curon();
gotoxy(cx, cy); gotoxy(cx, cy);
cursor = -1;
} }
fflush_all(); set_cursor(cursor);
}
static void cleanup(int code)
{
curon();
fflush_all();
tcsetattr(G.kbd_fd, TCSANOW, &G.term_orig);
if (ENABLE_FEATURE_CLEAN_UP) {
close(G.kbd_fd);
}
// Reset attributes
if (!BW)
fputs("\033[0m", stdout);
bb_putchar('\n');
if (code > 1)
kill_myself_with_sig(code); // does not return
exit(code);
}
static void get_initial_data(const char* vcsa_name)
{
G.vcsa_fd = xopen(vcsa_name, O_RDONLY);
xread(G.vcsa_fd, &G.remote, 4);
G.size = G.remote.cols * G.remote.lines * 2;
G.width = G.height = UINT_MAX;
G.data = xzalloc(2 * G.size);
screen_read_close();
} }
static void create_cdev_if_doesnt_exist(const char* name, dev_t dev) static void create_cdev_if_doesnt_exist(const char* name, dev_t dev)
@ -345,7 +343,6 @@ static NOINLINE void start_shell_in_child(const char* tty_name)
int conspy_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int conspy_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int conspy_main(int argc UNUSED_PARAM, char **argv) int conspy_main(int argc UNUSED_PARAM, char **argv)
{ {
char vcsa_name[sizeof("/dev/vcsaNN")];
char tty_name[sizeof("/dev/ttyNN")]; char tty_name[sizeof("/dev/ttyNN")];
#define keybuf bb_common_bufsiz1 #define keybuf bb_common_bufsiz1
struct termios termbuf; struct termios termbuf;
@ -365,7 +362,7 @@ int conspy_main(int argc UNUSED_PARAM, char **argv)
applet_long_options = getopt_longopts; applet_long_options = getopt_longopts;
#endif #endif
INIT_G(); INIT_G();
strcpy(vcsa_name, "/dev/vcsa"); strcpy(G.vcsa_name, "/dev/vcsa");
opt_complementary = "x+:y+"; // numeric params opt_complementary = "x+:y+"; // numeric params
opts = getopt32(argv, "vcsndfx:y:", &G.x, &G.y); opts = getopt32(argv, "vcsndfx:y:", &G.x, &G.y);
@ -373,19 +370,19 @@ int conspy_main(int argc UNUSED_PARAM, char **argv)
ttynum = 0; ttynum = 0;
if (argv[0]) { if (argv[0]) {
ttynum = xatou_range(argv[0], 0, 63); ttynum = xatou_range(argv[0], 0, 63);
sprintf(vcsa_name + sizeof("/dev/vcsa")-1, "%u", ttynum); sprintf(G.vcsa_name + sizeof("/dev/vcsa")-1, "%u", ttynum);
} }
sprintf(tty_name, "%s%u", "/dev/tty", ttynum); sprintf(tty_name, "%s%u", "/dev/tty", ttynum);
if (opts & FLAG(c)) { if (opts & FLAG(c)) {
if ((opts & (FLAG(s)|FLAG(v))) != FLAG(v)) if ((opts & (FLAG(s)|FLAG(v))) != FLAG(v))
create_cdev_if_doesnt_exist(tty_name, makedev(4, ttynum)); create_cdev_if_doesnt_exist(tty_name, makedev(4, ttynum));
create_cdev_if_doesnt_exist(vcsa_name, makedev(7, 128 + ttynum)); create_cdev_if_doesnt_exist(G.vcsa_name, makedev(7, 128 + ttynum));
} }
if ((opts & FLAG(s)) && ttynum) { if ((opts & FLAG(s)) && ttynum) {
start_shell_in_child(tty_name); start_shell_in_child(tty_name);
} }
get_initial_data(vcsa_name); screen_read_close();
if (opts & FLAG(d)) { if (opts & FLAG(d)) {
screen_dump(); screen_dump();
bb_putchar('\n'); bb_putchar('\n');
@ -412,17 +409,11 @@ int conspy_main(int argc UNUSED_PARAM, char **argv)
int i, j; int i, j;
char *data, *old; char *data, *old;
// Close & re-open vcsa in case they have // in the first loop G.width = G.height = 0: refresh
// swapped virtual consoles
G.vcsa_fd = xopen(vcsa_name, O_RDONLY);
xread(G.vcsa_fd, &G.remote, 4);
if (G.size != (G.remote.cols * G.remote.lines * 2)) {
cleanup(1);
}
i = G.width; i = G.width;
j = G.height; j = G.height;
get_terminal_width_height(G.kbd_fd, &G.width, &G.height); get_terminal_width_height(G.kbd_fd, &G.width, &G.height);
if ((option_mask32 & FLAG(f))) { if (option_mask32 & FLAG(f)) {
int nx = G.remote.cursor_x - G.width + 1; int nx = G.remote.cursor_x - G.width + 1;
int ny = G.remote.cursor_y - G.height + 1; int ny = G.remote.cursor_y - G.height + 1;
@ -454,8 +445,8 @@ int conspy_main(int argc UNUSED_PARAM, char **argv)
screen_dump(); screen_dump();
} else { } else {
// For each remote line // For each remote line
old += G.y * G.remote.cols * 2; old += G.first_line_offset;
data += G.y * G.remote.cols * 2; data += G.first_line_offset;
for (i = G.y; i < G.remote.lines; i++) { for (i = G.y; i < G.remote.lines; i++) {
char *first = NULL; // first char which needs updating char *first = NULL; // first char which needs updating
char *last = last; // last char which needs updating char *last = last; // last char which needs updating
@ -463,10 +454,8 @@ int conspy_main(int argc UNUSED_PARAM, char **argv)
if (iy >= G.height) if (iy >= G.height)
break; break;
old += G.x * 2; for (j = 0; j < G.remote.cols; j++, NEXT(old), NEXT(data)) {
data += G.x * 2; unsigned jx = j - G.x; // if will catch j >= G.x too
for (j = G.x; j < G.remote.cols; j++, NEXT(old), NEXT(data)) {
unsigned jx = j - G.x;
if (jx < G.width && DATA(data) != DATA(old)) { if (jx < G.width && DATA(data) != DATA(old)) {
last = data; last = data;
@ -486,6 +475,7 @@ int conspy_main(int argc UNUSED_PARAM, char **argv)
curmove(); curmove();
// Wait for local user keypresses // Wait for local user keypresses
fflush_all();
pfd.fd = G.kbd_fd; pfd.fd = G.kbd_fd;
pfd.events = POLLIN; pfd.events = POLLIN;
bytes_read = 0; bytes_read = 0;
@ -532,16 +522,17 @@ int conspy_main(int argc UNUSED_PARAM, char **argv)
else if (kbd_mode != K_XLATE && kbd_mode != K_UNICODE) else if (kbd_mode != K_XLATE && kbd_mode != K_UNICODE)
G.key_count = 0; // scan code mode G.key_count = 0; // scan code mode
else { else {
for (i = 0; i < G.key_count && result != -1; i++) char *p = keybuf;
result = ioctl(handle, TIOCSTI, keybuf + i); for (; G.key_count != 0 && result != -1; p++, G.key_count--) {
G.key_count -= i; result = ioctl(handle, TIOCSTI, p);
if (G.key_count)
memmove(keybuf, keybuf + i, G.key_count);
// If there is an application on console which reacts // If there is an application on console which reacts
// to keypresses, we need to make our first sleep // to keypresses, we need to make our first sleep
// shorter to quickly redraw whatever it printed there. // shorter to quickly redraw whatever it printed there.
poll_timeout_ms = 20; poll_timeout_ms = 20;
} }
if (G.key_count)
memmove(keybuf, p, G.key_count);
}
// Close & re-open tty in case they have // Close & re-open tty in case they have
// swapped virtual consoles // swapped virtual consoles
close(handle); close(handle);
@ -553,5 +544,5 @@ int conspy_main(int argc UNUSED_PARAM, char **argv)
else if (errno != EIO || ++G.ioerror_count > 4) else if (errno != EIO || ++G.ioerror_count > 4)
cleanup(1); cleanup(1);
} }
} } /* while (1) */
} }