1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
/**
* @file
*
* @author Sebastian Huber <sebastian.huber@embedded-brains.de>
*
* @brief Shell login prompt functions.
*/
/*
* Copyright (c) 2009
* embedded brains GmbH
* Obere Lagerstr. 30
* D-82178 Puchheim
* Germany
* <rtems@embedded-brains.de>
*
* Based on work from Chris Johns, Fernando Ruiz and Till Straumann.
*
* The license and distribution terms for this file may be
* found in the file LICENSE in this distribution or at
* http://www.rtems.com/license/LICENSE.
*/
#include <stdio.h>
#include <termios.h>
#include <unistd.h>
#include <ctype.h>
#include <rtems/login.h>
static int rtems_shell_discard( int c, FILE *stream)
{
return c;
}
static void rtems_shell_get_text(
FILE *in,
FILE *out,
char *line,
size_t size
)
{
int fd_in = fileno( in);
int (*put)( int, FILE *) =
out != NULL && isatty( fd_in)
? fputc
: rtems_shell_discard;
size_t i = 0;
if (size < 1) {
return;
}
tcdrain( fd_in);
if (out != NULL){
tcdrain( fileno( out));
}
while (true) {
int c = fgetc( in);
switch (c) {
case EOF:
/* Here comes an ugly hack: The Termios driver's read() handler returns
* 0 to the C library's fgets() if it times out. fgets() interprets
* this (correctly) as EOF, a condition we want to undo since it's not
* really true since we really have a read error (Termios bug?).
*
* As a workaround we push something back and read it again. This
* should simply reset the EOF condition.
*/
if (ungetc( '?', in) == '?') {
fgetc( in);
}
break;
case '\n':
case '\r':
put( '\n', out);
line [i] = '\0';
return;
case 127:
case '\b':
if (i > 0) {
put( '\b', out);
put( ' ', out);
put( '\b', out);
--i;
} else {
put( '\a', out);
}
break;
default:
if (!iscntrl( c)) {
if (i < size - 1) {
line [i] = (char) c;
++i;
put( c, out);
} else {
put( '\a', out);
}
} else {
put( '\a', out);
}
break;
}
}
}
bool rtems_shell_login_prompt(
FILE *in,
FILE *out,
const char *device,
rtems_login_check check
)
{
int fd_in = fileno( in);
struct termios termios_previous;
bool restore_termios = false;
int i = 0;
bool result = false;
if (tcgetattr( fd_in, &termios_previous) == 0) {
struct termios termios_new = termios_previous;
termios_new.c_lflag &= ~ECHO;
termios_new.c_lflag &= ~ICANON;
termios_new.c_cc [VTIME] = 255;
termios_new.c_cc [VMIN] = 0;
restore_termios = tcsetattr( fd_in, TCSANOW, &termios_new) == 0;
}
for (i = 0; i < 3; ++i) {
char user [32];
char passphrase [128];
fprintf( out, "%s login: ", device);
fflush( out);
rtems_shell_get_text( in, out, user, sizeof( user));
fflush( in);
fprintf( out, "Password: ");
fflush( out);
rtems_shell_get_text( in, NULL, passphrase, sizeof( passphrase));
fputc( '\n', out);
result = check( user, passphrase);
if (result) {
break;
}
fprintf( out, "Login incorrect\n\n");
sleep( 2);
}
if (restore_termios) {
/* What to do if restoring the flags fails? */
tcsetattr( fd_in, TCSANOW, &termios_previous);
}
return result;
}
|
