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         close(term_fd);
  18         term_fd = -1;
  19 }
  20
  21 static void restore_term_on_signal(int sig)
  22 {
  23         restore_term();
  24         sigchain_pop(sig);
  25         raise(sig);
  26 }
  27
  28 static int disable_echo(void)
  29 {
  30         struct termios t;
  31
  32         term_fd = open("/dev/tty", O_RDWR);
  33         if (tcgetattr(term_fd, &t) < 0)
  34                 goto error;
  35
  36         old_term = t;
  37         sigchain_push_common(restore_term_on_signal);
  38
  39         t.c_lflag &= ~ECHO;
  40         if (!tcsetattr(term_fd, TCSAFLUSH, &t))
  41                 return 0;
  42
  43 error:
  44         close(term_fd);
  45         term_fd = -1;
  46         return -1;
  47 }
  48
  49 char *git_terminal_prompt(const char *prompt, int echo)
  50 {
  51         static struct strbuf buf = STRBUF_INIT;
  52         int r;
  53         FILE *fh;
  54
  55         fh = fopen("/dev/tty", "w+");
  56         if (!fh)
  57                 return NULL;
  58
  59         if (!echo && disable_echo()) {
  60                 fclose(fh);
  61                 return NULL;
  62         }
  63
  64         fputs(prompt, fh);
  65         fflush(fh);
  66
  67         r = strbuf_getline(&buf, fh, '\n');
  68         if (!echo) {
  69                 fseek(fh, SEEK_CUR, 0);
  70                 putc('\n', fh);
  71                 fflush(fh);
  72         }
  73
  74         restore_term();
  75         fclose(fh);
  76
  77         if (r == EOF)
  78                 return NULL;
  79         return buf.buf;
  80 }
  81
  82 #else
  83
  84 char *git_terminal_prompt(const char *prompt, int echo)
  85 {
  86         return getpass(prompt);
  87 }
  88
  89 #endif