To get this branch, use:
bzr branch
http://bzr.recompile.se/loggerhead/mandos/trunk
« back to all changes in this revision
Viewing changes to plugins.d/password-prompt.c
- browse files at revision 305
- compare with another revision
- download diff
- download tarball
- view history from revision 305
- Committer: Teddy Hogeborn
- Date: 2009-02-09 02:13:58 UTC
- Revision ID: teddy@fukt.bsnet.se-20090209021358-6nrlc91oi08hn5hb
* initramfs-tools-hook: Whitespace change only.
- files removed:
- DBUS-API
- debian/source
- files modified:
- .bzrignore
added
removed
plugins.d/password-prompt.c
1
/* -*- coding: utf-8; mode: c; mode: orgtbl -*- */
1
/* -*- coding: utf-8 -*- */
2
2
/*
3
3
* Password-prompt - Read a password from the terminal and print it
4
4
*
5
* Copyright © 2008-2010 Teddy Hogeborn
6
* Copyright © 2008-2010 Björn Påhlsson
5
* Copyright © 2008,2009 Teddy Hogeborn
6
* Copyright © 2008,2009 Björn Påhlsson
7
7
*
8
8
* This program is free software: you can redistribute it and/or
9
9
* modify it under the terms of the GNU General Public License as
19
19
* along with this program. If not, see
20
20
* <http://www.gnu.org/licenses/>.
21
21
*
22
* Contact the authors at <mandos@fukt.bsnet.se>.
22
* Contact the authors at <https://www.fukt.bsnet.se/~belorn/> and
23
* <https://www.fukt.bsnet.se/~teddy/>.
23
24
*/
24
25
25
#define _GNU_SOURCE /* getline(), asprintf() */
26
#define _GNU_SOURCE /* getline() */
26
27
27
28
#include <termios.h> /* struct termios, tcsetattr(),
28
29
TCSAFLUSH, tcgetattr(), ECHO */
29
30
#include <unistd.h> /* struct termios, tcsetattr(),
30
31
STDIN_FILENO, TCSAFLUSH,
31
tcgetattr(), ECHO, readlink() */
32
tcgetattr(), ECHO */
32
33
#include <signal.h> /* sig_atomic_t, raise(), struct
33
34
sigaction, sigemptyset(),
34
35
sigaction(), sigaddset(), SIGINT,
35
SIGQUIT, SIGHUP, SIGTERM,
36
raise() */
36
SIGQUIT, SIGHUP, SIGTERM */
37
37
#include <stddef.h> /* NULL, size_t, ssize_t */
38
#include <sys/types.h> /* ssize_t, struct dirent, pid_t,
39
ssize_t, open() */
38
#include <sys/types.h> /* ssize_t */
40
39
#include <stdlib.h> /* EXIT_SUCCESS, EXIT_FAILURE,
41
getenv(), free() */
42
#include <dirent.h> /* scandir(), alphasort() */
40
getopt_long, getenv() */
43
41
#include <stdio.h> /* fprintf(), stderr, getline(),
44
stdin, feof(), fputc()
45
*/
46
#include <errno.h> /* errno, EBADF, ENOTTY, EINVAL,
47
EFAULT, EFBIG, EIO, ENOSPC, EINTR
48
*/
49
#include <error.h> /* error() */
42
stdin, feof(), perror(), fputc(),
43
stdout, getopt_long */
44
#include <errno.h> /* errno, EINVAL */
50
45
#include <iso646.h> /* or, not */
51
46
#include <stdbool.h> /* bool, false, true */
52
#include <inttypes.h> /* strtoumax() */
53
#include <sys/stat.h> /* struct stat, lstat(), open() */
54
#include <string.h> /* strlen, rindex, memcmp */
47
#include <string.h> /* strlen, rindex, strncmp, strcmp */
55
48
#include <argp.h> /* struct argp_option, struct
56
49
argp_state, struct argp,
57
50
argp_parse(), error_t,
58
51
ARGP_KEY_ARG, ARGP_KEY_END,
59
52
ARGP_ERR_UNKNOWN */
60
#include <sysexits.h> /* EX_SOFTWARE, EX_OSERR,
61
EX_UNAVAILABLE, EX_IOERR, EX_OK */
62
#include <fcntl.h> /* open() */
63
53
64
54
volatile sig_atomic_t quit_now = 0;
65
int signal_received;
66
55
bool debug = false;
67
56
const char *argp_program_version = "password-prompt " VERSION;
68
57
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
69
58
70
/* Needed for conflic resolution */
71
const char plymouthd_name[] = "plymouthd";
72
73
74
static void termination_handler(int signum){
75
if(quit_now){
76
return;
77
}
59
static void termination_handler(__attribute__((unused))int signum){
78
60
quit_now = 1;
79
signal_received = signum;
80
}
81
82
bool conflict_detection(void){
83
84
/* plymouth conflicts with password-prompt since both want to read
85
from the terminal. Password-prompt will exit if it detects
86
plymouth since plymouth performs the same functionality.
87
*/
88
int is_plymouth(const struct dirent *proc_entry){
89
int ret;
90
int cl_fd;
91
{
92
uintmax_t maxvalue;
93
char *tmp;
94
errno = 0;
95
maxvalue = strtoumax(proc_entry->d_name, &tmp, 10);
96
97
if(errno != 0 or *tmp != '\0'
98
or maxvalue != (uintmax_t)((pid_t)maxvalue)){
99
return 0;
100
}
101
}
102
103
char *cmdline_filename;
104
ret = asprintf(&cmdline_filename, "/proc/%s/cmdline", proc_entry->d_name);
105
if(ret == -1){
106
error(0, errno, "asprintf");
107
return 0;
108
}
109
110
/* Open /proc/<pid>/cmdline */
111
cl_fd = open(cmdline_filename, O_RDONLY);
112
free(cmdline_filename);
113
if(cl_fd == -1){
114
error(0, errno, "open");
115
return 0;
116
}
117
118
char *cmdline = NULL;
119
{
120
size_t cmdline_len = 0;
121
size_t cmdline_allocated = 0;
122
char *tmp;
123
const size_t blocksize = 1024;
124
ssize_t sret;
125
do {
126
/* Allocate more space? */
127
if(cmdline_len + blocksize + 1 > cmdline_allocated){
128
tmp = realloc(cmdline, cmdline_allocated + blocksize + 1);
129
if(tmp == NULL){
130
error(0, errno, "realloc");
131
free(cmdline);
132
close(cl_fd);
133
return 0;
134
}
135
cmdline = tmp;
136
cmdline_allocated += blocksize;
137
}
138
139
/* Read data */
140
sret = read(cl_fd, cmdline + cmdline_len,
141
cmdline_allocated - cmdline_len);
142
if(sret == -1){
143
error(0, errno, "read");
144
free(cmdline);
145
close(cl_fd);
146
return 0;
147
}
148
cmdline_len += (size_t)sret;
149
} while(sret != 0);
150
ret = close(cl_fd);
151
if(ret == -1){
152
error(0, errno, "close");
153
free(cmdline);
154
return 0;
155
}
156
cmdline[cmdline_len] = '\0'; /* Make sure it is terminated */
157
}
158
/* we now have cmdline */
159
160
/* get basename */
161
char *cmdline_base = strrchr(cmdline, '/');
162
if(cmdline_base != NULL){
163
cmdline_base += 1; /* skip the slash */
164
} else {
165
cmdline_base = cmdline;
166
}
167
168
if(strcmp(cmdline_base, plymouthd_name) != 0){
169
free(cmdline);
170
return 0;
171
}
172
free(cmdline);
173
return 1;
174
}
175
176
struct dirent **direntries;
177
int ret;
178
ret = scandir("/proc", &direntries, is_plymouth, alphasort);
179
if (ret == -1){
180
error(1, errno, "scandir");
181
}
182
return ret > 0;
183
}
184
61
}
185
62
186
63
int main(int argc, char **argv){
187
ssize_t sret;
188
int ret;
64
ssize_t ret;
189
65
size_t n;
190
66
struct termios t_new, t_old;
191
67
char *buffer = NULL;
201
77
.doc = "Prefix shown before the prompt", .group = 2 },
202
78
{ .name = "debug", .key = 128,
203
79
.doc = "Debug mode", .group = 3 },
204
/*
205
* These reproduce what we would get without ARGP_NO_HELP
206
*/
207
{ .name = "help", .key = '?',
208
.doc = "Give this help list", .group = -1 },
209
{ .name = "usage", .key = -3,
210
.doc = "Give a short usage message", .group = -1 },
211
{ .name = "version", .key = 'V',
212
.doc = "Print program version", .group = -1 },
213
80
{ .name = NULL }
214
81
};
215
82
216
83
error_t parse_opt (int key, char *arg, struct argp_state *state){
217
errno = 0;
218
84
switch (key){
219
85
case 'p':
220
86
prefix = arg;
222
88
case 128:
223
89
debug = true;
224
90
break;
225
/*
226
* These reproduce what we would get without ARGP_NO_HELP
227
*/
228
case '?': /* --help */
229
argp_state_help(state, state->out_stream,
230
(ARGP_HELP_STD_HELP | ARGP_HELP_EXIT_ERR)
231
& ~(unsigned int)ARGP_HELP_EXIT_OK);
232
case -3: /* --usage */
233
argp_state_help(state, state->out_stream,
234
ARGP_HELP_USAGE | ARGP_HELP_EXIT_ERR);
235
case 'V': /* --version */
236
fprintf(state->out_stream, "%s\n", argp_program_version);
237
exit(argp_err_exit_status);
91
case ARGP_KEY_ARG:
92
argp_usage(state);
93
break;
94
case ARGP_KEY_END:
238
95
break;
239
96
default:
240
97
return ARGP_ERR_UNKNOWN;
241
98
}
242
return errno;
99
return 0;
243
100
}
244
101
245
102
struct argp argp = { .options = options, .parser = parse_opt,
246
103
.args_doc = "",
247
104
.doc = "Mandos password-prompt -- Read and"
248
105
" output a password" };
249
ret = argp_parse(&argp, argc, argv,
250
ARGP_IN_ORDER | ARGP_NO_HELP, NULL, NULL);
251
switch(ret){
252
case 0:
253
break;
254
case ENOMEM:
255
default:
256
errno = ret;
257
error(0, errno, "argp_parse");
258
return EX_OSERR;
259
case EINVAL:
260
return EX_USAGE;
106
ret = argp_parse(&argp, argc, argv, 0, 0, NULL);
107
if(ret == ARGP_ERR_UNKNOWN){
108
fprintf(stderr, "Unknown error while parsing arguments\n");
109
return EXIT_FAILURE;
261
110
}
262
111
}
263
112
264
113
if(debug){
265
114
fprintf(stderr, "Starting %s\n", argv[0]);
266
115
}
267
268
if (conflict_detection()){
269
if(debug){
270
fprintf(stderr, "Stopping %s because of conflict", argv[0]);
271
}
272
return EXIT_FAILURE;
273
}
274
275
116
if(debug){
276
117
fprintf(stderr, "Storing current terminal attributes\n");
277
118
}
278
119
279
120
if(tcgetattr(STDIN_FILENO, &t_old) != 0){
280
int e = errno;
281
error(0, errno, "tcgetattr");
282
switch(e){
283
case EBADF:
284
case ENOTTY:
285
return EX_UNAVAILABLE;
286
default:
287
return EX_OSERR;
288
}
121
perror("tcgetattr");
122
return EXIT_FAILURE;
289
123
}
290
124
291
125
sigemptyset(&new_action.sa_mask);
292
ret = sigaddset(&new_action.sa_mask, SIGINT);
293
if(ret == -1){
294
error(0, errno, "sigaddset");
295
return EX_OSERR;
296
}
297
ret = sigaddset(&new_action.sa_mask, SIGHUP);
298
if(ret == -1){
299
error(0, errno, "sigaddset");
300
return EX_OSERR;
301
}
302
ret = sigaddset(&new_action.sa_mask, SIGTERM);
303
if(ret == -1){
304
error(0, errno, "sigaddset");
305
return EX_OSERR;
306
}
307
/* Need to check if the handler is SIG_IGN before handling:
308
| [[info:libc:Initial Signal Actions]] |
309
| [[info:libc:Basic Signal Handling]] |
310
*/
126
sigaddset(&new_action.sa_mask, SIGINT);
127
sigaddset(&new_action.sa_mask, SIGHUP);
128
sigaddset(&new_action.sa_mask, SIGTERM);
311
129
ret = sigaction(SIGINT, NULL, &old_action);
312
130
if(ret == -1){
313
error(0, errno, "sigaction");
314
return EX_OSERR;
131
perror("sigaction");
132
return EXIT_FAILURE;
315
133
}
316
134
if(old_action.sa_handler != SIG_IGN){
317
135
ret = sigaction(SIGINT, &new_action, NULL);
318
136
if(ret == -1){
319
error(0, errno, "sigaction");
320
return EX_OSERR;
137
perror("sigaction");
138
return EXIT_FAILURE;
321
139
}
322
140
}
323
141
ret = sigaction(SIGHUP, NULL, &old_action);
324
142
if(ret == -1){
325
error(0, errno, "sigaction");
326
return EX_OSERR;
143
perror("sigaction");
144
return EXIT_FAILURE;
327
145
}
328
146
if(old_action.sa_handler != SIG_IGN){
329
147
ret = sigaction(SIGHUP, &new_action, NULL);
330
148
if(ret == -1){
331
error(0, errno, "sigaction");
332
return EX_OSERR;
149
perror("sigaction");
150
return EXIT_FAILURE;
333
151
}
334
152
}
335
153
ret = sigaction(SIGTERM, NULL, &old_action);
336
154
if(ret == -1){
337
error(0, errno, "sigaction");
338
return EX_OSERR;
155
perror("sigaction");
156
return EXIT_FAILURE;
339
157
}
340
158
if(old_action.sa_handler != SIG_IGN){
341
159
ret = sigaction(SIGTERM, &new_action, NULL);
342
160
if(ret == -1){
343
error(0, errno, "sigaction");
344
return EX_OSERR;
161
perror("sigaction");
162
return EXIT_FAILURE;
345
163
}
346
164
}
347
165
351
169
}
352
170
353
171
t_new = t_old;
354
t_new.c_lflag &= ~(tcflag_t)ECHO;
172
t_new.c_lflag &= ~ECHO;
355
173
if(tcsetattr(STDIN_FILENO, TCSAFLUSH, &t_new) != 0){
356
int e = errno;
357
error(0, errno, "tcsetattr-echo");
358
switch(e){
359
case EBADF:
360
case ENOTTY:
361
return EX_UNAVAILABLE;
362
case EINVAL:
363
default:
364
return EX_OSERR;
365
}
174
perror("tcsetattr-echo");
175
return EXIT_FAILURE;
366
176
}
367
177
368
178
if(debug){
369
179
fprintf(stderr, "Waiting for input from stdin \n");
370
180
}
381
191
fprintf(stderr, "%s ", prefix);
382
192
}
383
193
{
384
const char *cryptsource = getenv("CRYPTTAB_SOURCE");
385
const char *crypttarget = getenv("CRYPTTAB_NAME");
386
/* Before cryptsetup 1.1.0~rc2 */
387
if(cryptsource == NULL){
388
cryptsource = getenv("cryptsource");
389
}
390
if(crypttarget == NULL){
391
crypttarget = getenv("crypttarget");
392
}
393
const char *const prompt1 = "Unlocking the disk";
394
const char *const prompt2 = "Enter passphrase";
194
const char *cryptsource = getenv("cryptsource");
195
const char *crypttarget = getenv("crypttarget");
196
const char *const prompt
197
= "Enter passphrase to unlock the disk";
395
198
if(cryptsource == NULL){
396
199
if(crypttarget == NULL){
397
fprintf(stderr, "%s to unlock the disk: ", prompt2);
200
fprintf(stderr, "%s: ", prompt);
398
201
} else {
399
fprintf(stderr, "%s (%s)\n%s: ", prompt1, crypttarget,
400
prompt2);
202
fprintf(stderr, "%s (%s): ", prompt, crypttarget);
401
203
}
402
204
} else {
403
205
if(crypttarget == NULL){
404
fprintf(stderr, "%s %s\n%s: ", prompt1, cryptsource,
405
prompt2);
206
fprintf(stderr, "%s %s: ", prompt, cryptsource);
406
207
} else {
407
fprintf(stderr, "%s %s (%s)\n%s: ", prompt1, cryptsource,
408
crypttarget, prompt2);
208
fprintf(stderr, "%s %s (%s): ", prompt, cryptsource,
209
crypttarget);
409
210
}
410
211
}
411
212
}
412
sret = getline(&buffer, &n, stdin);
413
if(sret > 0){
213
ret = getline(&buffer, &n, stdin);
214
if(ret > 0){
414
215
status = EXIT_SUCCESS;
415
216
/* Make n = data size instead of allocated buffer size */
416
n = (size_t)sret;
217
n = (size_t)ret;
417
218
/* Strip final newline */
418
if(n > 0 and buffer[n-1] == '\n'){
219
if(n>0 and buffer[n-1] == '\n'){
419
220
buffer[n-1] = '\0'; /* not strictly necessary */
420
221
n--;
421
222
}
422
223
size_t written = 0;
423
224
while(written < n){
424
sret = write(STDOUT_FILENO, buffer + written, n - written);
425
if(sret < 0){
426
int e = errno;
427
error(0, errno, "write");
428
switch(e){
429
case EBADF:
430
case EFAULT:
431
case EINVAL:
432
case EFBIG:
433
case EIO:
434
case ENOSPC:
435
default:
436
status = EX_IOERR;
437
break;
438
case EINTR:
439
status = EXIT_FAILURE;
440
break;
441
}
442
break;
443
}
444
written += (size_t)sret;
445
}
446
sret = close(STDOUT_FILENO);
447
if(sret == -1){
448
int e = errno;
449
error(0, errno, "close");
450
switch(e){
451
case EBADF:
452
status = EX_OSFILE;
453
break;
454
case EIO:
455
default:
456
status = EX_IOERR;
457
break;
458
}
225
ret = write(STDOUT_FILENO, buffer + written, n - written);
226
if(ret < 0){
227
perror("write");
228
status = EXIT_FAILURE;
229
break;
230
}
231
written += (size_t)ret;
459
232
}
460
233
break;
461
234
}
462
if(sret < 0){
463
int e = errno;
235
if(ret < 0){
464
236
if(errno != EINTR and not feof(stdin)){
465
error(0, errno, "getline");
466
switch(e){
467
case EBADF:
468
status = EX_UNAVAILABLE;
469
case EIO:
470
case EINVAL:
471
default:
472
status = EX_IOERR;
473
break;
474
}
237
perror("getline");
238
status = EXIT_FAILURE;
475
239
break;
476
240
}
477
241
}
478
/* if(sret == 0), then the only sensible thing to do is to retry to
242
/* if(ret == 0), then the only sensible thing to do is to retry to
479
243
read from stdin */
480
244
fputc('\n', stderr);
481
if(debug and not quit_now){
245
if(debug and quit_now == 0){
482
246
/* If quit_now is nonzero, we were interrupted by a signal, and
483
247
will print that later, so no need to show this too. */
484
248
fprintf(stderr, "getline() returned 0, retrying.\n");
491
255
fprintf(stderr, "Restoring terminal attributes\n");
492
256
}
493
257
if(tcsetattr(STDIN_FILENO, TCSAFLUSH, &t_old) != 0){
494
error(0, errno, "tcsetattr+echo");
495
}
496
497
if(quit_now){
498
sigemptyset(&old_action.sa_mask);
499
old_action.sa_handler = SIG_DFL;
500
ret = sigaction(signal_received, &old_action, NULL);
501
if(ret == -1){
502
error(0, errno, "sigaction");
503
}
504
raise(signal_received);
258
perror("tcsetattr+echo");
505
259
}
506
260
507
261
if(debug){
508
262
fprintf(stderr, "%s is exiting with status %d\n", argv[0],
509
263
status);
510
264
}
511
if(status == EXIT_SUCCESS or status == EX_OK){
265
if(status == EXIT_SUCCESS){
512
266
fputc('\n', stderr);
513
267
}
514
268
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0