1 /* prompt.c -- prompt routines for vlock,
2 * the VT locking program for linux
4 * This program is copyright (C) 2007 Frank Benkstein, and is free
5 * software which is freely distributable under the terms of the
6 * GNU General Public License version 2, included as the file COPYING in this
7 * distribution. It is NOT public domain software, and any
8 * redistribution not permitted by the GNU General Public License is
9 * expressly forbidden without prior written permission from
13 * The prompt functions (prompt and prompt_echo_off) were
14 * inspired by/copied from openpam's openpam_ttyconv.c:
16 * Copyright (c) 2002-2003 Networks Associates Technology, Inc.
18 * Redistribution and use in source and binary forms, with or without
19 * modification, are permitted provided that the following conditions
21 * 1. Redistributions of source code must retain the above copyright
22 * notice, this list of conditions and the following disclaimer.
23 * 2. Redistributions in binary form must reproduce the above copyright
24 * notice, this list of conditions and the following disclaimer in the
25 * documentation and/or other materials provided with the distribution.
26 * 3. The name of the author may not be used to endorse or promote
27 * products derived from this software without specific prior written
30 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
31 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
32 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
33 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
34 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
35 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
36 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
37 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
38 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
39 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
49 #include <sys/select.h>
54 #define PROMPT_BUFFER_SIZE 512
56 char *prompt(const char *msg
, const struct timespec
*timeout
) {
57 char buffer
[PROMPT_BUFFER_SIZE
];
61 struct timeval
*timeout_val
= NULL
;
66 && (timeout_val
= malloc(sizeof *timeout_val
)) != NULL
) {
67 timeout_val
->tv_sec
= timeout
->tv_sec
;
68 timeout_val
->tv_usec
= timeout
->tv_nsec
/ 1000;
72 /* Write out the prompt. */
73 (void) fputs(msg
, stderr
); fflush(stderr
);
76 /* Get the current terminal attributes. */
77 (void) tcgetattr(STDIN_FILENO
, &term
);
78 /* Save the lflag value. */
80 /* Enable canonical mode. We're only interested in line buffering. */
81 term
.c_lflag
|= ICANON
;
82 /* Disable terminal signals. */
83 term
.c_lflag
&= ~ISIG
;
84 /* Set the terminal attributes. */
85 (void) tcsetattr(STDIN_FILENO
, TCSAFLUSH
, &term
);
86 /* Discard all unread input characters. */
87 (void) tcflush(STDIN_FILENO
, TCIFLUSH
);
89 /* Initialize file descriptor set. */
91 FD_SET(STDIN_FILENO
, &readfds
);
95 /* Wait until a string was entered. */
96 if (select(STDIN_FILENO
+1, &readfds
, NULL
, NULL
, timeout_val
) != 1) {
98 perror("vlock-auth: select() on stdin failed");
100 fprintf(stderr
, "timeout!\n");
106 /* Read the string from stdin. At most buffer length - 1 bytes, to
107 * leave room for the terminating zero byte. */
108 if ((len
= read(STDIN_FILENO
, buffer
, sizeof buffer
- 1)) < 0) {
113 /* Terminate the string. */
116 /* Strip trailing newline characters. */
117 for (len
= strlen(buffer
); len
> 0; --len
) {
118 if (buffer
[len
-1] != '\r' && buffer
[len
-1] != '\n')
122 /* Terminate the string, again. */
125 /* Copy the string. */
126 result
= strdup(buffer
);
128 /* Clear our buffer. */
129 memset(buffer
, 0, sizeof buffer
);
134 /* Restore original terminal attributes. */
135 term
.c_lflag
= lflag
;
136 (void) tcsetattr(STDIN_FILENO
, TCSAFLUSH
, &term
);
141 char *prompt_echo_off(const char *msg
, const struct timespec
*timeout
) {
146 (void) tcgetattr(STDIN_FILENO
, &term
);
147 lflag
= term
.c_lflag
;
148 term
.c_lflag
&= ~ECHO
;
149 (void) tcsetattr(STDIN_FILENO
, TCSAFLUSH
, &term
);
151 result
= prompt(msg
, timeout
);
153 term
.c_lflag
= lflag
;
154 (void) tcsetattr(STDIN_FILENO
, TCSAFLUSH
, &term
);