githubmirror/libuser

lib/prompt.c
133 строки3 KB
Miloslav Trmač
Allow passing passwords using a pipe.
08 фев 2010, 14:38
08 фев 2010, 14:384454e75
/* Copyright (C) 2000-2002 Red Hat, Inc.
 *
 * This is free software; you can redistribute it and/or modify it under 
 * the terms of the GNU Library General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * 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 Library General Public 
 * License along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include <config.h>
#include <libintl.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <termios.h>
#include <unistd.h>
#include "user.h"
#include "user_private.h"

gboolean
lu_prompt_console(struct lu_prompt *prompts, int count, gpointer calldata,
		  struct lu_error **error)
{
	int i, is_tty;

	(void)calldata;
	LU_ERROR_CHECK(error);

	if (count > 0) {
		g_assert(prompts != NULL);
	}

	is_tty = isatty(fileno(stdin));
	for (i = 0; i < count; i++) {
		char buf[LINE_MAX], *p;
		struct termios otermios, ntermios;

		if (prompts[i].prompt) {
			g_print("%s",
				prompts[i].domain ?
			       	dgettext(prompts[i].domain, prompts[i].prompt) :
				prompts[i].prompt);
		}
		if (prompts[i].visible && prompts[i].default_value) {
			g_print(" [%s]: ", prompts[i].default_value);
		} else {
			g_print(": ");
		}

		prompts[i].value = NULL;
		prompts[i].free_value = NULL;

		if (prompts[i].visible == FALSE && is_tty) {
			if (tcgetattr(fileno(stdin), &otermios) == -1) {
				lu_error_new(error, lu_error_terminal,
					     _("error reading terminal attributes"));
				return FALSE;
			}
			ntermios = otermios;
			ntermios.c_lflag &= ~ECHO;
			if (tcsetattr(fileno(stdin), TCSADRAIN, &ntermios) == -1) {
				lu_error_new(error, lu_error_terminal,
					     _("error setting terminal attributes"));
				return FALSE;
			}
		}
		if (fgets(buf, sizeof(buf), stdin) == NULL) {
			lu_error_new(error, lu_error_terminal,
				     _("error reading from terminal"));
			return FALSE;
		}
		if (prompts[i].visible == FALSE && is_tty) {
			if (tcsetattr(fileno(stdin), TCSADRAIN, &otermios) == -1) {
				lu_error_new(error, lu_error_terminal,
					     _("error setting terminal attributes"));
				return FALSE;
			}
			g_print("\n");
		}

		p = strchr(buf, '\r');
		if (p != NULL)
			*p = '\0';
		p = strchr(buf, '\n');
		if (p != NULL)
			*p = '\0';

		prompts[i].value = (strlen(buf) > 0) ?
			g_strdup(buf) :
			(prompts[i].default_value ?
			 g_strdup(prompts[i].default_value) :
			 g_strdup(""));
		prompts[i].free_value = (void *) g_free;
	}
	return TRUE;
}

gboolean
lu_prompt_console_quiet(struct lu_prompt * prompts, int count,
			gpointer calldata, struct lu_error ** error)
{
	int i;
	gboolean ret = TRUE;

	LU_ERROR_CHECK(error);

	if (count > 0) {
		g_return_val_if_fail(prompts != NULL, FALSE);
	}

	for (i = 0; (i < count) && ret; i++) {
		if (prompts[i].default_value) {
			prompts[i].value =
			    g_strdup(prompts[i].default_value);
			prompts[i].free_value = (void *) g_free;
		} else {
			ret = ret &&
			      lu_prompt_console(&prompts[i], 1, calldata,
					        error);
		}
	}

	return ret;
}