compat/terminal.c

   1 #include "git-compat-util.h"
   2 #include "compat/terminal.h"
   3 #include "sigchain.h"
   4 #include "strbuf.h"
   5
   6 #ifdef HAVE_DEV_TTY
   7
   8 static int term_fd = -1;
   9 static struct termios old_term;
  10
  11 static void restore_term(void)
  12 {
  13         if (term_fd < 0)
  14                 return;
  15
  16         tcsetattr(term_fd, TCSAFLUSH, &old_term);
  17         term_fd = -1;
  18 }
  19
  20 static void restore_term_on_signal(int sig)
  21 {
  22         restore_term();
  23         sigchain_pop(sig);
  24         raise(sig);
  25 }
  26
  27 char *git_terminal_prompt(const char *prompt, int echo)
  28 {
  29         static struct strbuf buf = STRBUF_INIT;
  30         int r;
  31         FILE *fh;
  32
  33         fh = fopen("/dev/tty", "w+");
  34         if (!fh)
  35                 return NULL;
  36
  37         if (!echo) {
  38                 struct termios t;
  39
  40                 if (tcgetattr(fileno(fh), &t) < 0) {
  41                         fclose(fh);
  42                         return NULL;
  43                 }
  44
  45                 old_term = t;
  46                 term_fd = fileno(fh);
  47                 sigchain_push_common(restore_term_on_signal);
  48
  49                 t.c_lflag &= ~ECHO;
  50                 if (tcsetattr(fileno(fh), TCSAFLUSH, &t) < 0) {
  51                         term_fd = -1;
  52                         fclose(fh);
  53                         return NULL;
  54                 }
  55         }
  56
  57         fputs(prompt, fh);
  58         fflush(fh);
  59
  60         r = strbuf_getline(&buf, fh, '\n');
  61         if (!echo) {
  62                 fseek(fh, SEEK_CUR, 0);
  63                 putc('\n', fh);
  64                 fflush(fh);
  65         }
  66
  67         restore_term();
  68         fclose(fh);
  69
  70         if (r == EOF)
  71                 return NULL;
  72         return buf.buf;
  73 }
  74
  75 #else
  76
  77 char *git_terminal_prompt(const char *prompt, int echo)
  78 {
  79         return getpass(prompt);
  80 }
  81
  82 #endif