get_password.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:5k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000-2004 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation.
  5.    There are special exceptions to the terms and conditions of the GPL as it
  6.    is applied to this software. View the full text of the exception in file
  7.    EXCEPTIONS-CLIENT in the directory of this software distribution.
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  12.    You should have received a copy of the GNU General Public License
  13.    along with this program; if not, write to the Free Software
  14.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  15. /*
  16. ** Ask for a password from tty
  17. ** This is an own file to avoid conflicts with curses
  18. */
  19. #include <my_global.h>
  20. #include <my_sys.h>
  21. #include "mysql.h"
  22. #include <m_string.h>
  23. #include <m_ctype.h>
  24. #if defined(HAVE_BROKEN_GETPASS) && !defined(HAVE_GETPASSPHRASE)
  25. #undef HAVE_GETPASS
  26. #endif
  27. #ifdef HAVE_GETPASS
  28. #ifdef HAVE_PWD_H
  29. #include <pwd.h>
  30. #endif /* HAVE_PWD_H */
  31. #else /* ! HAVE_GETPASS */
  32. #if !defined(__WIN__) && !defined(OS2) && !defined(__NETWARE__)
  33. #include <sys/ioctl.h>
  34. #ifdef HAVE_TERMIOS_H /* For tty-password */
  35. #include <termios.h>
  36. #define TERMIO struct termios
  37. #else
  38. #ifdef HAVE_TERMIO_H /* For tty-password */
  39. #include <termio.h>
  40. #define TERMIO struct termio
  41. #else
  42. #include <sgtty.h>
  43. #define TERMIO struct sgttyb
  44. #endif
  45. #endif
  46. #ifdef alpha_linux_port
  47. #include <asm/ioctls.h> /* QQ; Fix this in configure */
  48. #include <asm/termiobits.h>
  49. #endif
  50. #else
  51. #ifndef __NETWARE__
  52. #include <conio.h>
  53. #endif /* __NETWARE__ */
  54. #endif /* __WIN__ */
  55. #endif /* HAVE_GETPASS */
  56. #ifdef HAVE_GETPASSPHRASE /* For Solaris */
  57. #define getpass(A) getpassphrase(A)
  58. #endif
  59. #if defined( __WIN__) || defined(OS2) || defined(__NETWARE__)
  60. /* were just going to fake it here and get input from the keyboard */
  61. #ifdef __NETWARE__
  62. #undef _getch
  63. #undef _cputs
  64. #define _getch getcharacter
  65. #define _cputs(A) putstring(A)
  66. #endif
  67. char *get_tty_password(char *opt_message)
  68. {
  69.   char to[80];
  70.   char *pos=to,*end=to+sizeof(to)-1;
  71.   int i=0;
  72.   DBUG_ENTER("get_tty_password");
  73.   _cputs(opt_message ? opt_message : "Enter password: ");
  74.   for (;;)
  75.   {
  76.     char tmp;
  77.     tmp=_getch();
  78.     if (tmp == 'b' || (int) tmp == 127)
  79.     {
  80.       if (pos != to)
  81.       {
  82. _cputs("b b");
  83. pos--;
  84. continue;
  85.       }
  86.     }
  87.     if (tmp == 'n' || tmp == 'r' || tmp == 3)
  88.       break;
  89.     if (iscntrl(tmp) || pos == end)
  90.       continue;
  91.     _cputs("*");
  92.     *(pos++) = tmp;
  93.   }
  94.   while (pos != to && isspace(pos[-1]) == ' ')
  95.     pos--; /* Allow dummy space at end */
  96.   *pos=0;
  97.   _cputs("n");
  98.   DBUG_RETURN(my_strdup(to,MYF(MY_FAE)));
  99. }
  100. #else
  101. #ifndef HAVE_GETPASS
  102. /*
  103.   Can't use fgets, because readline will get confused
  104.   length is max number of chars in to, not counting 
  105.   to will not include the eol characters.
  106. */
  107. static void get_password(char *to,uint length,int fd,bool echo)
  108. {
  109.   char *pos=to,*end=to+length;
  110.   for (;;)
  111.   {
  112.     char tmp;
  113.     if (my_read(fd,&tmp,1,MYF(0)) != 1)
  114.       break;
  115.     if (tmp == 'b' || (int) tmp == 127)
  116.     {
  117.       if (pos != to)
  118.       {
  119. if (echo)
  120. {
  121.   fputs("b b",stdout);
  122.   fflush(stdout);
  123. }
  124. pos--;
  125. continue;
  126.       }
  127.     }
  128.     if (tmp == 'n' || tmp == 'r' || tmp == 3)
  129.       break;
  130.     if (iscntrl(tmp) || pos == end)
  131.       continue;
  132.     if (echo)
  133.     {
  134.       fputc('*',stdout);
  135.       fflush(stdout);
  136.     }
  137.     *(pos++) = tmp;
  138.   }
  139.   while (pos != to && isspace(pos[-1]) == ' ')
  140.     pos--; /* Allow dummy space at end */
  141.   *pos=0;
  142.   return;
  143. }
  144. #endif /* ! HAVE_GETPASS */
  145. char *get_tty_password(char *opt_message)
  146. {
  147. #ifdef HAVE_GETPASS
  148.   char *passbuff;
  149. #else /* ! HAVE_GETPASS */
  150.   TERMIO org,tmp;
  151. #endif /* HAVE_GETPASS */
  152.   char buff[80];
  153.   DBUG_ENTER("get_tty_password");
  154. #ifdef HAVE_GETPASS
  155.   passbuff = getpass(opt_message ? opt_message : "Enter password: ");
  156.   /* copy the password to buff and clear original (static) buffer */
  157.   strnmov(buff, passbuff, sizeof(buff) - 1);
  158. #ifdef _PASSWORD_LEN
  159.   memset(passbuff, 0, _PASSWORD_LEN);
  160. #endif
  161. #else 
  162.   if (isatty(fileno(stdout)))
  163.   {
  164.     fputs(opt_message ? opt_message : "Enter password: ",stdout);
  165.     fflush(stdout);
  166.   }
  167. #if defined(HAVE_TERMIOS_H)
  168.   tcgetattr(fileno(stdin), &org);
  169.   tmp = org;
  170.   tmp.c_lflag &= ~(ECHO | ISIG | ICANON);
  171.   tmp.c_cc[VMIN] = 1;
  172.   tmp.c_cc[VTIME] = 0;
  173.   tcsetattr(fileno(stdin), TCSADRAIN, &tmp);
  174.   get_password(buff, sizeof(buff)-1, fileno(stdin), isatty(fileno(stdout)));
  175.   tcsetattr(fileno(stdin), TCSADRAIN, &org);
  176. #elif defined(HAVE_TERMIO_H)
  177.   ioctl(fileno(stdin), (int) TCGETA, &org);
  178.   tmp=org;
  179.   tmp.c_lflag &= ~(ECHO | ISIG | ICANON);
  180.   tmp.c_cc[VMIN] = 1;
  181.   tmp.c_cc[VTIME]= 0;
  182.   ioctl(fileno(stdin),(int) TCSETA, &tmp);
  183.   get_password(buff,sizeof(buff)-1,fileno(stdin),isatty(fileno(stdout)));
  184.   ioctl(fileno(stdin),(int) TCSETA, &org);
  185. #else
  186.   gtty(fileno(stdin), &org);
  187.   tmp=org;
  188.   tmp.sg_flags &= ~ECHO;
  189.   tmp.sg_flags |= RAW;
  190.   stty(fileno(stdin), &tmp);
  191.   get_password(buff,sizeof(buff)-1,fileno(stdin),isatty(fileno(stdout)));
  192.   stty(fileno(stdin), &org);
  193. #endif
  194.   if (isatty(fileno(stdout)))
  195.     fputc('n',stdout);
  196. #endif /* HAVE_GETPASS */
  197.   DBUG_RETURN(my_strdup(buff,MYF(MY_FAE)));
  198. }
  199. #endif /*__WIN__*/