quick patchs: drop founded memory leak, more libbb, more C-compatibility, size reduction

This commit is contained in:
"Vladimir N. Oleynik" 2005-09-19 10:28:43 +00:00
parent 4cbe45934e
commit a0ae6de848

View File

@ -77,7 +77,7 @@
#define NORMAL "\033[0m"
/* The escape code to clear the screen */
#define CLEAR "\033[2J"
#define CLEAR "\033[H\033[J"
/* Maximum number of lines in a file */
#define MAXLINES 10000
@ -142,13 +142,6 @@ static void set_tty_cooked(void) {
/* Set terminal input to raw mode (taken from vi.c) */
static void set_tty_raw(void) {
tcgetattr(0, &term_orig);
term_vi = term_orig;
term_vi.c_lflag &= (~ICANON & ~ECHO);
term_vi.c_iflag &= (~IXON & ~ICRNL);
term_vi.c_oflag &= (~ONLCR);
term_vi.c_cc[VMIN] = 1;
term_vi.c_cc[VTIME] = 0;
tcsetattr(0, TCSANOW, &term_vi);
}
@ -169,36 +162,34 @@ static void tless_exit(int code) {
special return codes. Note that this function works best with raw input. */
static int tless_getch(void) {
set_tty_raw();
char input_key[3];
int input;
input_key[0] = getc(inp);
set_tty_raw();
input = getc(inp);
/* Detect escape sequences (i.e. arrow keys) and handle
them accordingly */
if (input_key[0] == '\033') {
input_key[1] = getc(inp);
input_key[2] = getc(inp);
if (input == '\033' && getc(inp) == '[') {
input = getc(inp);
set_tty_cooked();
if (input_key[1] == '[') {
if (input_key[2] == REAL_KEY_UP)
if (input == REAL_KEY_UP)
return KEY_UP;
else if (input_key[2] == REAL_KEY_DOWN)
else if (input == REAL_KEY_DOWN)
return KEY_DOWN;
else if (input_key[2] == REAL_KEY_RIGHT)
else if (input == REAL_KEY_RIGHT)
return KEY_RIGHT;
else if (input_key[2] == REAL_KEY_LEFT)
else if (input == REAL_KEY_LEFT)
return KEY_LEFT;
else if (input_key[2] == REAL_PAGE_UP)
else if (input == REAL_PAGE_UP)
return PAGE_UP;
else if (input_key[2] == REAL_PAGE_DOWN)
else if (input == REAL_PAGE_DOWN)
return PAGE_DOWN;
}
}
/* The input is a normal ASCII value */
else {
set_tty_cooked();
return input_key[0];
return input;
}
return 0;
}
@ -222,8 +213,7 @@ static void add_linenumbers(void) {
for (i = 0; i <= num_flines; i++) {
safe_strncpy(current_line, flines[i], 256);
flines[i] = xrealloc(flines[i], strlen(current_line) + 7);
sprintf(flines[i],"%5d %s", i + 1, current_line);
bb_xasprintf(&flines[i],"%5d %s", i + 1, current_line);
}
}
@ -246,7 +236,7 @@ static void data_readlines(void) {
strcpy(current_line, "");
fgets(current_line, 256, fp);
bb_xferror(fp, filename);
flines[i] = (char *) bb_xstrndup(current_line, (strlen(current_line) + 1) * sizeof(char));
flines[i] = bb_xstrdup(current_line);
}
num_flines = i - 2;
@ -307,7 +297,7 @@ static void m_status_print(void) {
}
else {
percentage = calc_percent();
printf("%i%s %s", percentage, "%", NORMAL);
printf("%i%% %s", percentage, NORMAL);
}
}
else {
@ -325,11 +315,11 @@ static void medium_status_print(void) {
percentage = calc_percent();
if (!line_pos)
printf("%s%s %i%s%s", HIGHLIGHT, filename, percentage, "%", NORMAL);
printf("%s%s %i%%%s", HIGHLIGHT, filename, percentage, NORMAL);
else if (line_pos == num_flines - height + 2)
printf("%s(END)%s", HIGHLIGHT, NORMAL);
else
printf("%s%i%s%s", HIGHLIGHT, percentage, "%", NORMAL);
printf("%s%i%%%s", HIGHLIGHT, percentage, NORMAL);
}
#endif
@ -368,15 +358,12 @@ static void buffer_print(void) {
int i;
if (num_flines >= height - 2) {
printf("%s", CLEAR);
move_cursor(0,0);
if (num_flines >= height - 2) {
for (i = 0; i < height - 1; i++)
printf("%s", buffer[i]);
}
else {
printf("%s", CLEAR);
move_cursor(0,0);
for (i = 1; i < (height - 1 - num_flines); i++)
putchar('\n');
for (i = 0; i < height - 1; i++)
@ -391,8 +378,13 @@ static void buffer_init(void) {
int i;
if(buffer == NULL) {
/* malloc the number of lines needed for the buffer */
buffer = xrealloc(buffer, height * sizeof(char *));
} else {
for (i = 0; i < (height - 1); i++)
free(buffer[i]);
}
/* Fill the buffer until the end of the file or the
end of the buffer is reached */
@ -402,7 +394,7 @@ static void buffer_init(void) {
/* If the buffer still isn't full, fill it with blank lines */
for (; i < (height - 1); i++) {
buffer[i] = "";
buffer[i] = bb_xstrdup("");
}
}
@ -414,18 +406,22 @@ static void buffer_down(int nlines) {
if (!past_eof) {
if (line_pos + (height - 3) + nlines < num_flines) {
line_pos += nlines;
for (i = 0; i < (height - 1); i++)
for (i = 0; i < (height - 1); i++) {
free(buffer[i]);
buffer[i] = (char *) bb_xstrdup(flines[line_pos + i]);
}
}
else {
/* As the number of lines requested was too large, we just move
to the end of the file */
while (line_pos + (height - 3) + 1 < num_flines) {
line_pos += 1;
for (i = 0; i < (height - 1); i++)
for (i = 0; i < (height - 1); i++) {
free(buffer[i]);
buffer[i] = (char *) bb_xstrdup(flines[line_pos + i]);
}
}
}
/* We exit if the -E flag has been set */
if ((flags & FLAG_E) && (line_pos + (height - 2) == num_flines))
@ -441,19 +437,23 @@ static void buffer_up(int nlines) {
if (!past_eof) {
if (line_pos - nlines >= 0) {
line_pos -= nlines;
for (i = 0; i < (height - 1); i++)
for (i = 0; i < (height - 1); i++) {
free(buffer[i]);
buffer[i] = (char *) bb_xstrdup(flines[line_pos + i]);
}
}
else {
/* As the requested number of lines to move was too large, we
move one line up at a time until we can't. */
while (line_pos != 0) {
line_pos -= 1;
for (i = 0; i < (height - 1); i++)
for (i = 0; i < (height - 1); i++) {
free(buffer[i]);
buffer[i] = (char *) bb_xstrdup(flines[line_pos + i]);
}
}
}
}
else {
/* Work out where the tildes start */
tilde_line = num_flines - line_pos + 3;
@ -469,11 +469,12 @@ static void buffer_up(int nlines) {
/* We only move part of the buffer, as the rest
is past the EOF */
for (i = 0; i < (height - 1); i++) {
free(buffer[i]);
if (i < tilde_line - nlines + 1)
buffer[i] = (char *) bb_xstrdup(flines[line_pos + i]);
else {
if (line_pos >= num_flines - height + 2)
buffer[i] = "~\n";
buffer[i] = bb_xstrdup("~\n");
}
}
}
@ -491,12 +492,15 @@ static void buffer_line(int linenum) {
printf("%s%s%i%s", HIGHLIGHT, "Cannot seek to line number ", linenum, NORMAL);
}
else if (linenum < (num_flines - height - 2)) {
for (i = 0; i < (height - 1); i++)
for (i = 0; i < (height - 1); i++) {
free(buffer[i]);
buffer[i] = (char *) bb_xstrdup(flines[linenum + i]);
}
line_pos = linenum;
}
else {
for (i = 0; i < (height - 1); i++) {
free(buffer[i]);
if (linenum + i < num_flines + 2)
buffer[i] = (char *) bb_xstrdup(flines[linenum + i]);
else
@ -629,15 +633,10 @@ static void colon_process(void) {
static char *insert_highlights (char *line, int start, int end) {
char *new_line = (char *) malloc((sizeof(char) * (strlen(line) + 1)) + 10);
memset(new_line, 0, ((sizeof(char) * (strlen(line) + 1)) + 10));
strncat(new_line, line, start);
strcat(new_line, HIGHLIGHT);
strncat(new_line, line + start, end - start);
strcat(new_line, NORMAL);
strncat(new_line, line + end, strlen(line) - end);
char *new_line;
bb_xasprintf(&new_line, "%.*s%s%.*s%s%s", start, line, HIGHLIGHT,
end - start, line + start, NORMAL, line + end);
return new_line;
}
@ -869,7 +868,7 @@ static void show_flag_status(void) {
}
clear_line();
printf("%s%s%i%s", HIGHLIGHT, "The status of the flag is: ", flag_val, NORMAL);
printf("%s%s%i%s", HIGHLIGHT, "The status of the flag is: ", flag_val != 0, NORMAL);
}
#endif
@ -1168,6 +1167,14 @@ int less_main(int argc, char **argv) {
strcpy(filename, (inp_stdin) ? "stdin" : files[0]);
tty_width_height();
tcgetattr(0, &term_orig);
term_vi = term_orig;
term_vi.c_lflag &= (~ICANON & ~ECHO);
term_vi.c_iflag &= (~IXON & ~ICRNL);
term_vi.c_oflag &= (~ONLCR);
term_vi.c_cc[VMIN] = 1;
term_vi.c_cc[VTIME] = 0;
data_readlines();
buffer_init();
buffer_print();