home *** CD-ROM | disk | FTP | other *** search
/ rtsi.com / 2014.01.www.rtsi.com.tar / www.rtsi.com / OS9 / OSK / CMDS / mtools_3.6.src.lzh / MTOOLS_3.6 / tty.c < prev    next >
C/C++ Source or Header  |  1997-11-13  |  4KB  |  195 lines
  1. #include "sysincludes.h"
  2. #include "mtools.h"
  3.  
  4. static FILE *tty=NULL;
  5. static int notty=0;    
  6. static int ttyfd=-1;
  7. static int tty_mode = -1; /* 1 for raw, 0 for cooked, -1 for initial */
  8. static int need_tty_reset = 0;
  9. static int handlerIsSet = 0;
  10. int    mtools_raw_tty = 1;
  11.  
  12.  
  13. #ifdef TCSANOW
  14.  
  15. /* we have tcsetattr & tcgetattr. Good */
  16. #define stty(a,b)        (void)tcsetattr(a,TCSANOW,b)
  17. #define gtty(a,b)        (void)tcgetattr(a,b)
  18. #ifdef TCIFLUSH
  19. #define discard_input(a) tcflush(a,TCIFLUSH)
  20. #else
  21. #define discard_input(a) /**/
  22. #endif
  23.  
  24. #else /* TCSANOW */
  25.  
  26. #ifndef TCSETS
  27. #define TCSETS TCSETA
  28. #define TCGETS TCGETA
  29. #define termios termio
  30. #endif
  31.  
  32. #define stty(a,b) (void)ioctl(a,TCSETS,(char *)b)
  33. #define gtty(a,b) (void)ioctl(a,TCGETS,(char *)b)
  34.  
  35. #ifdef TCIFLUSH
  36. #define discard_input(a) (void)ioctl(a,TCFLSH,TCIFLUSH)
  37. #else
  38. #define discard_input(a) /**/
  39. #endif
  40.  
  41. #endif /* TCSANOW */
  42.  
  43. #define restore_tty(a) stty(STDIN,a)
  44.  
  45.  
  46. #define STDIN ttyfd
  47. #define FAIL (-1)
  48. #define DONE 0
  49. #ifndef _OSK
  50. static struct termios in_orig;
  51. #endif
  52.  
  53. /*--------------- Signal Handler routines -------------*/
  54.  
  55. static void tty_time_out(void)
  56. {
  57.     int exit_code;
  58. #ifndef _OSK
  59.     signal(SIGALRM, SIG_IGN);
  60.     if(tty && need_tty_reset)
  61.         restore_tty (&in_orig);    
  62. #if future
  63.     if (fail_on_timeout)
  64.         exit_code=SHFAIL;
  65.     else {
  66.         if (default_choice && mode_defined) {
  67.             if (yes_no) {
  68.                 if ('Y' == default_choice)
  69.                     exit_code=0;
  70.                 else
  71.                     exit_code=1;
  72.             } else
  73.                 exit_code=default_choice-minc+1;
  74.         } else
  75.             exit_code=DONE;
  76.     }
  77. #else
  78.     exit_code = DONE;
  79. #endif
  80. #else
  81.     exit_code = DONE;
  82. #endif
  83.     exit(exit_code);
  84. }
  85.  
  86. static void cleanup_tty(void)
  87. { 
  88. #ifndef _OSK
  89.     if(tty && need_tty_reset) {
  90.         restore_tty (&in_orig);
  91.         signal (SIGHUP, SIG_IGN);
  92.         signal (SIGINT, SIG_IGN);
  93.         signal (SIGTERM, SIG_IGN);
  94.         signal (SIGALRM, SIG_IGN);
  95.     }
  96. #endif
  97. }
  98.  
  99. static void fail(void)
  100. {
  101.     exit(1);
  102. }
  103.  
  104. static void set_raw_tty(int mode)
  105. {
  106. #ifndef _OSK
  107.     struct termios in_raw;
  108.  
  109.     if(mode != tty_mode && mode != -1) {
  110.         /* Determine existing TTY settings */
  111.         gtty (STDIN, &in_orig);
  112.         need_tty_reset = 1;
  113.         if(!handlerIsSet) {
  114.             atexit(cleanup_tty);
  115.             handlerIsSet = 1;
  116.         }
  117.  
  118.         /*------ Restore original TTY settings on exit --------*/
  119.         signal (SIGHUP, (SIG_CAST) fail);
  120.         signal (SIGINT, (SIG_CAST) fail);
  121.         signal (SIGTERM, (SIG_CAST) fail);
  122.         signal (SIGALRM, (SIG_CAST) tty_time_out);
  123.     
  124.         /* Change STDIN settings to raw */
  125.  
  126.         gtty (STDIN, &in_raw);
  127.         if(mode) {
  128.             in_raw.c_lflag &= ~ICANON;
  129.             in_raw.c_cc[VMIN]=1;
  130.             in_raw.c_cc[VTIME]=0;
  131.             stty (STDIN, &in_raw);
  132.         } else {
  133.             in_raw.c_lflag |= ICANON;
  134.             stty (STDIN, &in_raw);
  135.         }
  136.         tty_mode = mode;
  137.         discard_input(STDIN);
  138.     }
  139. #endif
  140. }
  141.  
  142. FILE *opentty(int mode)
  143. {
  144. #ifndef _OSK
  145.     if(notty)
  146.         return NULL;
  147. /* BeOS does not have a working /dev/tty, so we use stdin */
  148. #ifndef __BEOS__
  149.     if (tty == NULL) {
  150.         ttyfd = open("/dev/tty", O_RDONLY);
  151.         if(ttyfd >= 0) {
  152.             tty = fdopen(ttyfd, "r");
  153.         }
  154.     }
  155. #endif
  156.     if  (tty == NULL){
  157.         if ( !isatty(0) ){
  158.             notty = 1;
  159.             return NULL;
  160.         }
  161.         ttyfd = 0;
  162.         tty = stdin;
  163.     }
  164.  
  165.     set_raw_tty(mode);
  166.     return tty;
  167. #else
  168.     return stdin;
  169. #endif
  170. }
  171.  
  172. int ask_confirmation(const char *format, const char *p1, const char *p2)
  173. {
  174.     char ans[10];
  175.  
  176.     if(!opentty(-1))
  177.         return 0;
  178.  
  179.     while (1) {
  180.         fprintf(stderr, format, p1, p2);
  181.         fflush(stderr);
  182.         fflush(opentty(-1));
  183.         if (mtools_raw_tty) {
  184.             ans[0] = fgetc(opentty(1));
  185.             fputs("\n", stderr);
  186.         } else {
  187.             fgets(ans,9, opentty(0));
  188.         }
  189.         if (ans[0] == 'y' || ans[0] == 'Y')
  190.             return 0;
  191.         if (ans[0] == 'n' || ans[0] == 'N')
  192.             return -1;
  193.     }
  194. }
  195.