minix/commands/i386/mtools-3.9.7/tty.c
2005-04-21 14:53:53 +00:00

201 lines
3.8 KiB
C
Executable file

#include "sysincludes.h"
#include "mtools.h"
static FILE *tty=NULL;
static int notty=0;
static int ttyfd=-1;
#ifdef USE_RAWTERM
int mtools_raw_tty = 1;
#else
int mtools_raw_tty = 0;
#endif
#ifdef USE_RAWTERM
# if defined TCSANOW && defined HAVE_TCSETATTR
/* we have tcsetattr & tcgetattr. Good */
typedef struct termios Terminal;
# define stty(a,b) (void)tcsetattr(a,TCSANOW,b)
# define gtty(a,b) (void)tcgetattr(a,b)
# define USE_TCIFLUSH
# elif defined TCSETS && defined TCGETS
typedef struct termios Terminal;
# define stty(a,b) (void)ioctl(a,TCSETS,(char *)b)
# define gtty(a,b) (void)ioctl(a,TCGETS,(char *)b)
# define USE_TCIFLUSH
# elif defined TCSETA && defined TCGETA
typedef struct termio Terminal;
# define stty(a,b) (void)ioctl(a,TCSETA,(char *)b)
# define gtty(a,b) (void)ioctl(a,TCGETA,(char *)b)
# define USE_TCIFLUSH
# elif defined(HAVE_SGTTY_H) && defined(TIOCSETP) && defined(TIOCGETP)
typedef struct sgttyb Terminal;
# define stty(a,b) (void)ioctl(a,TIOCSETP,(char *)b)
# define gtty(a,b) (void)ioctl(a,TIOCGETP,(char *)b)
# define USE_SGTTY
# define discard_input(a) /**/
# else
/* no way to use raw terminal */
/*
# warning Cannot use raw terminal code (disabled)
*/
# undef USE_RAWTERM
# endif
#endif
#ifdef USE_TCIFLUSH
# if defined TCIFLUSH && defined HAVE_TCFLUSH
# define discard_input(a) tcflush(a,TCIFLUSH)
# else
# define discard_input(a) /**/
# endif
#endif
#ifdef USE_RAWTERM
static int tty_mode = -1; /* 1 for raw, 0 for cooked, -1 for initial */
static int need_tty_reset = 0;
static int handlerIsSet = 0;
#define restore_tty(a) stty(STDIN,a)
#define STDIN ttyfd
#define FAIL (-1)
#define DONE 0
static Terminal in_orig;
/*--------------- Signal Handler routines -------------*/
static void tty_time_out(void)
{
int exit_code;
signal(SIGALRM, SIG_IGN);
if(tty && need_tty_reset)
restore_tty (&in_orig);
#if future
if (fail_on_timeout)
exit_code=SHFAIL;
else {
if (default_choice && mode_defined) {
if (yes_no) {
if ('Y' == default_choice)
exit_code=0;
else
exit_code=1;
} else
exit_code=default_choice-minc+1;
} else
exit_code=DONE;
}
#else
exit_code = DONE;
#endif
exit(exit_code);
}
static void cleanup_tty(void)
{
if(tty && need_tty_reset) {
restore_tty (&in_orig);
setup_signal();
}
}
static void set_raw_tty(int mode)
{
Terminal in_raw;
if(mode != tty_mode && mode != -1) {
if(!handlerIsSet) {
/* Determine existing TTY settings */
gtty (STDIN, &in_orig);
need_tty_reset = 1;
/* Restore original TTY settings on exit */
atexit(cleanup_tty);
handlerIsSet = 1;
}
setup_signal();
signal (SIGALRM, (SIG_CAST) tty_time_out);
/* Change STDIN settings to raw */
gtty (STDIN, &in_raw);
if(mode) {
#ifdef USE_SGTTY
in_raw.sg_flags |= CBREAK;
#else
in_raw.c_lflag &= ~ICANON;
in_raw.c_cc[VMIN]=1;
in_raw.c_cc[VTIME]=0;
#endif
stty (STDIN, &in_raw);
} else {
#ifdef USE_SGTTY
in_raw.sg_flags &= ~CBREAK;
#else
in_raw.c_lflag |= ICANON;
#endif
stty (STDIN, &in_raw);
}
tty_mode = mode;
discard_input(STDIN);
}
}
#endif
FILE *opentty(int mode)
{
if(notty)
return NULL;
if (tty == NULL) {
ttyfd = open("/dev/tty", O_RDONLY);
if(ttyfd >= 0) {
tty = fdopen(ttyfd, "r");
}
}
if (tty == NULL){
if ( !isatty(0) ){
notty = 1;
return NULL;
}
ttyfd = 0;
tty = stdin;
}
#ifdef USE_RAWTERM
if(mtools_raw_tty)
set_raw_tty(mode);
#endif
return tty;
}
int ask_confirmation(const char *format, const char *p1, const char *p2)
{
char ans[10];
if(!opentty(-1))
return 0;
while (1) {
fprintf(stderr, format, p1, p2);
fflush(stderr);
fflush(opentty(-1));
if (mtools_raw_tty) {
ans[0] = fgetc(opentty(1));
fputs("\n", stderr);
} else {
fgets(ans,9, opentty(0));
}
if (ans[0] == 'y' || ans[0] == 'Y')
return 0;
if (ans[0] == 'n' || ans[0] == 'N')
return -1;
}
}