~drizzle-trunk/drizzle/development

/* Copyright (C) 2000 MySQL AB

   This program is free software; you can redistribute it and/or modify

   it under the terms of the GNU General Public License as published by

   the Free Software Foundation; version 2 of the License.

   This program is distributed in the hope that it will be useful,

   but WITHOUT ANY WARRANTY; without even the implied warranty of

   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License

   along with this program; if not, write to the Free Software

   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

/*

** Ask for a password from tty

** This is an own file to avoid conflicts with curses

*/

#include <my_global.h>

#include <my_sys.h>

#include "mysql.h"

#include <m_string.h>

#include <m_ctype.h>

#if defined(HAVE_BROKEN_GETPASS) && !defined(HAVE_GETPASSPHRASE)

#undef HAVE_GETPASS

#endif

#ifdef HAVE_GETPASS

#ifdef HAVE_PWD_H

#include <pwd.h>

#endif /* HAVE_PWD_H */

#else /* ! HAVE_GETPASS */

 #include <sys/ioctl.h>

 #ifdef HAVE_TERMIOS_H				/* For tty-password */

  #include	<termios.h>

  #define TERMIO	struct termios

 #else

  #ifdef HAVE_TERMIO_H				/* For tty-password */

   #include	<termio.h>

   #define TERMIO	struct termio

  #else

   #include	<sgtty.h>

   #define TERMIO	struct sgttyb

  #endif

 #endif

 #ifdef alpha_linux_port

  #include <asm/ioctls.h>				/* QQ; Fix this in configure */

  #include <asm/termiobits.h>

 #endif

#endif /* HAVE_GETPASS */

#ifdef HAVE_GETPASSPHRASE			/* For Solaris */

#define getpass(A) getpassphrase(A)

#endif

#ifndef HAVE_GETPASS

/*

** Can't use fgets, because readline will get confused

** length is max number of chars in to, not counting \0

*  to will not include the eol characters.

*/

static void get_password(char *to,uint length,int fd, my_bool echo)

{

  char *pos=to,*end=to+length;

  for (;;)

  {

    char tmp;

    if (my_read(fd,&tmp,1,MYF(0)) != 1)

      break;

    if (tmp == '\b' || (int) tmp == 127)

    {

      if (pos != to)

      {

	if (echo)

	{

	  fputs("\b \b",stderr);

	  fflush(stderr);

	}

	pos--;

	continue;

      }

    }

    if (tmp == '\n' || tmp == '\r' || tmp == 3)

      break;

    if (iscntrl(tmp) || pos == end)

      continue;

    if (echo)

    {

      fputc('*',stderr);

      fflush(stderr);

    }

    *(pos++) = tmp;

  }

  while (pos != to && isspace(pos[-1]) == ' ')

    pos--;					/* Allow dummy space at end */

  *pos=0;

  return;

}

#endif /* ! HAVE_GETPASS */

char *get_tty_password(const char *opt_message)

{

#ifdef HAVE_GETPASS

  char *passbuff;

#else /* ! HAVE_GETPASS */

  TERMIO org,tmp;

#endif /* HAVE_GETPASS */

  char buff[80];

  DBUG_ENTER("get_tty_password");

#ifdef HAVE_GETPASS

  passbuff = getpass(opt_message ? opt_message : "Enter password: ");

  /* copy the password to buff and clear original (static) buffer */

  strnmov(buff, passbuff, sizeof(buff) - 1);

#ifdef _PASSWORD_LEN

  memset(passbuff, 0, _PASSWORD_LEN);

#endif

#else 

  if (isatty(fileno(stderr)))

  {

    fputs(opt_message ? opt_message : "Enter password: ",stderr);

    fflush(stderr);

  }

#if defined(HAVE_TERMIOS_H)

  tcgetattr(fileno(stdin), &org);

  tmp = org;

  tmp.c_lflag &= ~(ECHO | ISIG | ICANON);

  tmp.c_cc[VMIN] = 1;

  tmp.c_cc[VTIME] = 0;

  tcsetattr(fileno(stdin), TCSADRAIN, &tmp);

  get_password(buff, sizeof(buff)-1, fileno(stdin), isatty(fileno(stderr)));

  tcsetattr(fileno(stdin), TCSADRAIN, &org);

#elif defined(HAVE_TERMIO_H)

  ioctl(fileno(stdin), (int) TCGETA, &org);

  tmp=org;

  tmp.c_lflag &= ~(ECHO | ISIG | ICANON);

  tmp.c_cc[VMIN] = 1;

  tmp.c_cc[VTIME]= 0;

  ioctl(fileno(stdin),(int) TCSETA, &tmp);

  get_password(buff,sizeof(buff)-1,fileno(stdin),isatty(fileno(stderr)));

  ioctl(fileno(stdin),(int) TCSETA, &org);

#else

  gtty(fileno(stdin), &org);

  tmp=org;

  tmp.sg_flags &= ~ECHO;

  tmp.sg_flags |= RAW;

  stty(fileno(stdin), &tmp);

  get_password(buff,sizeof(buff)-1,fileno(stdin),isatty(fileno(stderr)));

  stty(fileno(stdin), &org);

#endif

  if (isatty(fileno(stderr)))

    fputc('\n',stderr);

#endif /* HAVE_GETPASS */

  DBUG_RETURN(my_strdup(buff,MYF(MY_FAE)));

}