Quick and dirty terminal_size() and yesno() functions, both of which need to be improved.

This commit is contained in:
Rob Landley 2012-02-02 07:27:35 -06:00
parent ce8a267189
commit 26e7b5ece4
2 changed files with 53 additions and 1 deletions

View File

@ -174,7 +174,8 @@ void xexec(char **argv)
{
toy_exec(argv);
execvp(argv[0], argv);
error_exit("No %s", argv[0]);
perror_exit("exec %s", argv[0]);
}
void xaccess(char *path, int flags)
@ -756,3 +757,51 @@ void crc_init(unsigned int *crc_table, int little_endian)
crc_table[i] = c;
}
}
// Quick and dirty query size of terminal, doesn't do ANSI probe fallback.
// set *x=0 and *y=0 before calling to detect failure to set either, or
// x=80 y=25 to provide defaults
void terminal_size(unsigned *x, unsigned *y)
{
struct winsize ws;
int i;
//memset(&ws, 0, sizeof(ws));
for (i=0; i<3; i++) {
if (ioctl(i, TIOCGWINSZ, &ws)) continue;
if (x) *x = ws.ws_col;
if (y) *y = ws.ws_row;
}
if (x) {
char *s = getenv("COLUMNS");
i = s ? atoi(s) : 0;
if (i>0) *x = i;
}
if (y) {
char *s = getenv("ROWS");
i = s ? atoi(s) : 0;
if (i>0) *y = i;
}
}
// This should use a raw tty, fixit later.
int yesno(int def)
{
char buf[16];
int i;
for (i=0; i<3 && !isatty(i); i++);
if (i == 3) return 1;
sprintf(buf, "(%c/%c):", def ? 'Y' : 'y', def ? 'n' : 'N');
write(i, buf, 6);
while (read(i, buf, 1)) {
if (isspace(*buf)) break;
if (tolower(*buf) == 'y') return 1;
if (tolower(*buf) == 'n') return 0;
}
return def;
}

View File

@ -108,6 +108,9 @@ int copy_tempfile(int fdin, char *name, char **tempname);
void delete_tempfile(int fdin, int fdout, char **tempname);
void replace_tempfile(int fdin, int fdout, char **tempname);
void crc_init(unsigned int *crc_table, int little_endian);
void terminal_size(unsigned *x, unsigned *y);
int yesno(int def);
// getmountlist.c
struct mtab_list {