To get this branch, use:
bzr branch
http://bzr.recompile.se/loggerhead/mandos/trunk
« back to all changes in this revision
Viewing changes to plugin-runner.c
- browse files at revision 274
- compare with another revision
- download diff
- download tarball
- view history from revision 274
- Committer: Teddy Hogeborn
- Date: 2009-01-17 09:46:54 UTC
- Revision ID: teddy@fukt.bsnet.se-20090117094654-ydvanis7caqoxxn7
* mandos-keygen (password): Remove bashism "${PIPESTATUS}".
added
removed
plugin-runner.c
1
/* -*- coding: utf-8; mode: c; mode: orgtbl -*- */
1
/* -*- coding: utf-8 -*- */
2
2
/*
3
3
* Mandos plugin runner - Run Mandos plugins
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
23
23
*/
24
24
25
25
#define _GNU_SOURCE /* TEMP_FAILURE_RETRY(), getline(),
26
asprintf(), O_CLOEXEC */
26
asprintf() */
27
27
#include <stddef.h> /* size_t, NULL */
28
#include <stdlib.h> /* malloc(), exit(), EXIT_SUCCESS,
29
realloc() */
28
#include <stdlib.h> /* malloc(), exit(), EXIT_FAILURE,
29
EXIT_SUCCESS, realloc() */
30
30
#include <stdbool.h> /* bool, true, false */
31
#include <stdio.h> /* fileno(), fprintf(),
31
#include <stdio.h> /* perror, fileno(), fprintf(),
32
32
stderr, STDOUT_FILENO */
33
#include <sys/types.h> /* DIR, fdopendir(), stat(), struct
33
#include <sys/types.h> /* DIR, opendir(), stat(), struct
34
34
stat, waitpid(), WIFEXITED(),
35
35
WEXITSTATUS(), wait(), pid_t,
36
36
uid_t, gid_t, getuid(), getgid(),
38
38
#include <sys/select.h> /* fd_set, select(), FD_ZERO(),
39
39
FD_SET(), FD_ISSET(), FD_CLR */
40
40
#include <sys/wait.h> /* wait(), waitpid(), WIFEXITED(),
41
WEXITSTATUS(), WTERMSIG(),
42
WCOREDUMP() */
41
WEXITSTATUS() */
43
42
#include <sys/stat.h> /* struct stat, stat(), S_ISREG() */
44
43
#include <iso646.h> /* and, or, not */
45
#include <dirent.h> /* DIR, struct dirent, fdopendir(),
44
#include <dirent.h> /* DIR, struct dirent, opendir(),
46
45
readdir(), closedir(), dirfd() */
47
46
#include <unistd.h> /* struct stat, stat(), S_ISREG(),
48
47
fcntl(), setuid(), setgid(),
53
52
close() */
54
53
#include <fcntl.h> /* fcntl(), F_GETFD, F_SETFD,
55
54
FD_CLOEXEC */
56
#include <string.h> /* strsep, strlen(), asprintf(),
57
strsignal(), strcmp(), strncmp() */
55
#include <string.h> /* strsep, strlen(), asprintf() */
58
56
#include <errno.h> /* errno */
59
57
#include <argp.h> /* struct argp_option, struct
60
58
argp_state, struct argp,
64
62
#include <signal.h> /* struct sigaction, sigemptyset(),
65
63
sigaddset(), sigaction(),
66
64
sigprocmask(), SIG_BLOCK, SIGCHLD,
67
SIG_UNBLOCK, kill(), sig_atomic_t
68
*/
65
SIG_UNBLOCK, kill() */
69
66
#include <errno.h> /* errno, EBADF */
70
#include <inttypes.h> /* intmax_t, PRIdMAX, strtoimax() */
71
#include <sysexits.h> /* EX_OSERR, EX_USAGE, EX_IOERR,
72
EX_CONFIG, EX_UNAVAILABLE, EX_OK */
73
#include <errno.h> /* errno */
74
#include <error.h> /* error() */
67
#include <inttypes.h> /* intmax_t, SCNdMAX, PRIdMAX, */
75
68
76
69
#define BUFFER_SIZE 256
77
70
88
81
char **environ;
89
82
int envc;
90
83
bool disabled;
91
84
92
85
/* Variables used for running processes*/
93
86
pid_t pid;
94
87
int fd;
96
89
size_t buffer_size;
97
90
size_t buffer_length;
98
91
bool eof;
99
volatile sig_atomic_t completed;
100
int status;
92
volatile bool completed;
93
volatile int status;
101
94
struct plugin *next;
102
95
} plugin;
103
96
106
99
/* Gets an existing plugin based on name,
107
100
or if none is found, creates a new one */
108
101
static plugin *getplugin(char *name){
109
/* Check for existing plugin with that name */
102
/* Check for exiting plugin with that name */
110
103
for(plugin *p = plugin_list; p != NULL; p = p->next){
111
104
if((p->name == name)
112
105
or (p->name and name and (strcmp(p->name, name) == 0))){
114
107
}
115
108
}
116
109
/* Create a new plugin */
117
plugin *new_plugin = NULL;
118
do {
119
new_plugin = malloc(sizeof(plugin));
120
} while(new_plugin == NULL and errno == EINTR);
110
plugin *new_plugin = malloc(sizeof(plugin));
121
111
if(new_plugin == NULL){
122
112
return NULL;
123
113
}
124
114
char *copy_name = NULL;
125
115
if(name != NULL){
126
do {
127
copy_name = strdup(name);
128
} while(copy_name == NULL and errno == EINTR);
116
copy_name = strdup(name);
129
117
if(copy_name == NULL){
130
int e = errno;
131
118
free(new_plugin);
132
errno = e;
133
119
return NULL;
134
120
}
135
121
}
136
122
137
*new_plugin = (plugin){ .name = copy_name,
138
.argc = 1,
139
.disabled = false,
140
.next = plugin_list };
123
*new_plugin = (plugin) { .name = copy_name,
124
.argc = 1,
125
.disabled = false,
126
.next = plugin_list };
141
127
142
do {
143
new_plugin->argv = malloc(sizeof(char *) * 2);
144
} while(new_plugin->argv == NULL and errno == EINTR);
128
new_plugin->argv = malloc(sizeof(char *) * 2);
145
129
if(new_plugin->argv == NULL){
146
int e = errno;
147
130
free(copy_name);
148
131
free(new_plugin);
149
errno = e;
150
132
return NULL;
151
133
}
152
134
new_plugin->argv[0] = copy_name;
153
135
new_plugin->argv[1] = NULL;
154
136
155
do {
156
new_plugin->environ = malloc(sizeof(char *));
157
} while(new_plugin->environ == NULL and errno == EINTR);
137
new_plugin->environ = malloc(sizeof(char *));
158
138
if(new_plugin->environ == NULL){
159
int e = errno;
160
139
free(copy_name);
161
140
free(new_plugin->argv);
162
141
free(new_plugin);
163
errno = e;
164
142
return NULL;
165
143
}
166
144
new_plugin->environ[0] = NULL;
174
152
static bool add_to_char_array(const char *new, char ***array,
175
153
int *len){
176
154
/* Resize the pointed-to array to hold one more pointer */
177
do {
178
*array = realloc(*array, sizeof(char *)
179
* (size_t) ((*len) + 2));
180
} while(*array == NULL and errno == EINTR);
155
*array = realloc(*array, sizeof(char *)
156
* (size_t) ((*len) + 2));
181
157
/* Malloc check */
182
158
if(*array == NULL){
183
159
return false;
184
160
}
185
161
/* Make a copy of the new string */
186
char *copy;
187
do {
188
copy = strdup(new);
189
} while(copy == NULL and errno == EINTR);
162
char *copy = strdup(new);
190
163
if(copy == NULL){
191
164
return false;
192
165
}
218
191
if(strncmp(*e, def, namelen + 1) == 0){
219
192
/* It already exists */
220
193
if(replace){
221
char *new;
222
do {
223
new = realloc(*e, strlen(def) + 1);
224
} while(new == NULL and errno == EINTR);
194
char *new = realloc(*e, strlen(def) + 1);
225
195
if(new == NULL){
226
196
return false;
227
197
}
237
207
/*
238
208
* Based on the example in the GNU LibC manual chapter 13.13 "File
239
209
* Descriptor Flags".
240
| [[info:libc:Descriptor%20Flags][File Descriptor Flags]] |
210
* *Note File Descriptor Flags:(libc)Descriptor Flags.
241
211
*/
242
212
static int set_cloexec_flag(int fd){
243
int ret = (int)TEMP_FAILURE_RETRY(fcntl(fd, F_GETFD, 0));
213
int ret = fcntl(fd, F_GETFD, 0);
244
214
/* If reading the flags failed, return error indication now. */
245
215
if(ret < 0){
246
216
return ret;
247
217
}
248
218
/* Store modified flag word in the descriptor. */
249
return (int)TEMP_FAILURE_RETRY(fcntl(fd, F_SETFD,
250
ret | FD_CLOEXEC));
219
return fcntl(fd, F_SETFD, ret | FD_CLOEXEC);
251
220
}
252
221
253
222
254
223
/* Mark processes as completed when they exit, and save their exit
255
224
status. */
256
225
static void handle_sigchld(__attribute__((unused)) int sig){
257
int old_errno = errno;
258
226
while(true){
259
227
plugin *proc = plugin_list;
260
228
int status;
264
232
break;
265
233
}
266
234
if(pid == -1){
267
if(errno == ECHILD){
268
/* No child processes */
269
break;
235
if(errno != ECHILD){
236
perror("waitpid");
270
237
}
271
error(0, errno, "waitpid");
238
/* No child processes */
239
break;
272
240
}
273
241
274
242
/* A child exited, find it in process_list */
280
248
continue;
281
249
}
282
250
proc->status = status;
283
proc->completed = 1;
251
proc->completed = true;
284
252
}
285
errno = old_errno;
286
253
}
287
254
288
255
/* Prints out a password to stdout */
310
277
}
311
278
free(plugin_node->environ);
312
279
free(plugin_node->buffer);
313
280
314
281
/* Removes the plugin from the singly-linked list */
315
282
if(plugin_node == plugin_list){
316
283
/* First one - simple */
343
310
struct dirent *dirst;
344
311
struct stat st;
345
312
fd_set rfds_all;
346
int ret, maxfd = 0;
313
int ret, numchars, maxfd = 0;
347
314
ssize_t sret;
315
intmax_t tmpmax;
348
316
uid_t uid = 65534;
349
317
gid_t gid = 65534;
350
318
bool debug = false;
359
327
sigemptyset(&sigchld_action.sa_mask);
360
328
ret = sigaddset(&sigchld_action.sa_mask, SIGCHLD);
361
329
if(ret == -1){
362
error(0, errno, "sigaddset");
363
exitstatus = EX_OSERR;
330
perror("sigaddset");
331
exitstatus = EXIT_FAILURE;
364
332
goto fallback;
365
333
}
366
334
ret = sigaction(SIGCHLD, &sigchld_action, &old_sigchld_action);
367
335
if(ret == -1){
368
error(0, errno, "sigaction");
369
exitstatus = EX_OSERR;
336
perror("sigaction");
337
exitstatus = EXIT_FAILURE;
370
338
goto fallback;
371
339
}
372
340
404
372
.doc = "Group ID the plugins will run as", .group = 3 },
405
373
{ .name = "debug", .key = 132,
406
374
.doc = "Debug mode", .group = 4 },
407
/*
408
* These reproduce what we would get without ARGP_NO_HELP
409
*/
410
{ .name = "help", .key = '?',
411
.doc = "Give this help list", .group = -1 },
412
{ .name = "usage", .key = -3,
413
.doc = "Give a short usage message", .group = -1 },
414
{ .name = "version", .key = 'V',
415
.doc = "Print program version", .group = -1 },
416
375
{ .name = NULL }
417
376
};
418
377
419
error_t parse_opt(int key, char *arg, struct argp_state *state){
420
errno = 0;
421
switch(key){
422
char *tmp;
423
intmax_t tmpmax;
378
error_t parse_opt(int key, char *arg, __attribute__((unused))
379
struct argp_state *state) {
380
switch(key) {
424
381
case 'g': /* --global-options */
425
{
426
char *plugin_option;
427
while((plugin_option = strsep(&arg, ",")) != NULL){
428
if(not add_argument(getplugin(NULL), plugin_option)){
429
break;
382
if(arg != NULL){
383
char *p;
384
while((p = strsep(&arg, ",")) != NULL){
385
if(p[0] == '\0'){
386
continue;
387
}
388
if(not add_argument(getplugin(NULL), p)){
389
perror("add_argument");
390
return ARGP_ERR_UNKNOWN;
430
391
}
431
392
}
432
393
}
433
394
break;
434
395
case 'G': /* --global-env */
435
add_environment(getplugin(NULL), arg, true);
396
if(arg == NULL){
397
break;
398
}
399
if(not add_environment(getplugin(NULL), arg, true)){
400
perror("add_environment");
401
}
436
402
break;
437
403
case 'o': /* --options-for */
438
{
439
char *option_list = strchr(arg, ':');
440
if(option_list == NULL){
441
argp_error(state, "No colon in \"%s\"", arg);
442
errno = EINVAL;
443
break;
444
}
445
*option_list = '\0';
446
option_list++;
447
if(arg[0] == '\0'){
448
argp_error(state, "Empty plugin name");
449
errno = EINVAL;
450
break;
451
}
452
char *option;
453
while((option = strsep(&option_list, ",")) != NULL){
454
if(not add_argument(getplugin(arg), option)){
455
break;
404
if(arg != NULL){
405
char *p_name = strsep(&arg, ":");
406
if(p_name[0] == '\0' or arg == NULL){
407
break;
408
}
409
char *opt = strsep(&arg, ":");
410
if(opt[0] == '\0' or opt == NULL){
411
break;
412
}
413
char *p;
414
while((p = strsep(&opt, ",")) != NULL){
415
if(p[0] == '\0'){
416
continue;
417
}
418
if(not add_argument(getplugin(p_name), p)){
419
perror("add_argument");
420
return ARGP_ERR_UNKNOWN;
456
421
}
457
422
}
458
423
}
459
424
break;
460
425
case 'E': /* --env-for */
426
if(arg == NULL){
427
break;
428
}
461
429
{
462
430
char *envdef = strchr(arg, ':');
463
431
if(envdef == NULL){
464
argp_error(state, "No colon in \"%s\"", arg);
465
errno = EINVAL;
466
432
break;
467
433
}
468
434
*envdef = '\0';
469
envdef++;
470
if(arg[0] == '\0'){
471
argp_error(state, "Empty plugin name");
472
errno = EINVAL;
473
break;
435
if(not add_environment(getplugin(arg), envdef+1, true)){
436
perror("add_environment");
474
437
}
475
add_environment(getplugin(arg), envdef, true);
476
438
}
477
439
break;
478
440
case 'd': /* --disable */
479
{
441
if(arg != NULL){
480
442
plugin *p = getplugin(arg);
481
if(p != NULL){
482
p->disabled = true;
443
if(p == NULL){
444
return ARGP_ERR_UNKNOWN;
483
445
}
446
p->disabled = true;
484
447
}
485
448
break;
486
449
case 'e': /* --enable */
487
{
450
if(arg != NULL){
488
451
plugin *p = getplugin(arg);
489
if(p != NULL){
490
p->disabled = false;
452
if(p == NULL){
453
return ARGP_ERR_UNKNOWN;
491
454
}
455
p->disabled = false;
492
456
}
493
457
break;
494
458
case 128: /* --plugin-dir */
495
459
free(plugindir);
496
460
plugindir = strdup(arg);
461
if(plugindir == NULL){
462
perror("strdup");
463
}
497
464
break;
498
465
case 129: /* --config-file */
499
466
/* This is already done by parse_opt_config_file() */
500
467
break;
501
468
case 130: /* --userid */
502
tmpmax = strtoimax(arg, &tmp, 10);
503
if(errno != 0 or tmp == arg or *tmp != '\0'
504
or tmpmax != (uid_t)tmpmax){
505
argp_error(state, "Bad user ID number: \"%s\", using %"
506
PRIdMAX, arg, (intmax_t)uid);
507
break;
469
ret = sscanf(arg, "%" SCNdMAX "%n", &tmpmax, &numchars);
470
if(ret < 1 or tmpmax != (uid_t)tmpmax
471
or arg[numchars] != '\0'){
472
fprintf(stderr, "Bad user ID number: \"%s\", using %"
473
PRIdMAX "\n", arg, (intmax_t)uid);
474
} else {
475
uid = (uid_t)tmpmax;
508
476
}
509
uid = (uid_t)tmpmax;
510
477
break;
511
478
case 131: /* --groupid */
512
tmpmax = strtoimax(arg, &tmp, 10);
513
if(errno != 0 or tmp == arg or *tmp != '\0'
514
or tmpmax != (gid_t)tmpmax){
515
argp_error(state, "Bad group ID number: \"%s\", using %"
516
PRIdMAX, arg, (intmax_t)gid);
517
break;
479
ret = sscanf(arg, "%" SCNdMAX "%n", &tmpmax, &numchars);
480
if(ret < 1 or tmpmax != (gid_t)tmpmax
481
or arg[numchars] != '\0'){
482
fprintf(stderr, "Bad group ID number: \"%s\", using %"
483
PRIdMAX "\n", arg, (intmax_t)gid);
484
} else {
485
gid = (gid_t)tmpmax;
518
486
}
519
gid = (gid_t)tmpmax;
520
487
break;
521
488
case 132: /* --debug */
522
489
debug = true;
523
490
break;
524
/*
525
* These reproduce what we would get without ARGP_NO_HELP
526
*/
527
case '?': /* --help */
528
state->flags &= ~(unsigned int)ARGP_NO_EXIT; /* force exit */
529
argp_state_help(state, state->out_stream, ARGP_HELP_STD_HELP);
530
case -3: /* --usage */
531
state->flags &= ~(unsigned int)ARGP_NO_EXIT; /* force exit */
532
argp_state_help(state, state->out_stream,
533
ARGP_HELP_USAGE | ARGP_HELP_EXIT_OK);
534
case 'V': /* --version */
535
fprintf(state->out_stream, "%s\n", argp_program_version);
536
exit(EXIT_SUCCESS);
537
break;
538
491
/*
539
492
* When adding more options before this line, remember to also add a
540
493
* "case" to the "parse_opt_config_file" function below.
543
496
/* Cryptsetup always passes an argument, which is an empty
544
497
string if "none" was specified in /etc/crypttab. So if
545
498
argument was empty, we ignore it silently. */
546
if(arg[0] == '\0'){
547
break;
499
if(arg[0] != '\0'){
500
fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
548
501
}
502
break;
503
case ARGP_KEY_END:
504
break;
549
505
default:
550
506
return ARGP_ERR_UNKNOWN;
551
507
}
552
return errno; /* Set to 0 at start */
508
return 0;
553
509
}
554
510
555
511
/* This option parser is the same as parse_opt() above, except it
556
512
ignores everything but the --config-file option. */
557
513
error_t parse_opt_config_file(int key, char *arg,
558
514
__attribute__((unused))
559
struct argp_state *state){
560
errno = 0;
561
switch(key){
515
struct argp_state *state) {
516
switch(key) {
562
517
case 'g': /* --global-options */
563
518
case 'G': /* --global-env */
564
519
case 'o': /* --options-for */
570
525
case 129: /* --config-file */
571
526
free(argfile);
572
527
argfile = strdup(arg);
573
break;
528
if(argfile == NULL){
529
perror("strdup");
530
}
531
break;
574
532
case 130: /* --userid */
575
533
case 131: /* --groupid */
576
534
case 132: /* --debug */
577
case '?': /* --help */
578
case -3: /* --usage */
579
case 'V': /* --version */
580
535
case ARGP_KEY_ARG:
536
case ARGP_KEY_END:
581
537
break;
582
538
default:
583
539
return ARGP_ERR_UNKNOWN;
584
540
}
585
return errno;
541
return 0;
586
542
}
587
543
588
544
struct argp argp = { .options = options,
592
548
593
549
/* Parse using parse_opt_config_file() in order to get the custom
594
550
config file location, if any. */
595
ret = argp_parse(&argp, argc, argv,
596
ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
597
NULL, NULL);
598
switch(ret){
599
case 0:
600
break;
601
case ENOMEM:
602
default:
603
errno = ret;
604
error(0, errno, "argp_parse");
605
exitstatus = EX_OSERR;
606
goto fallback;
607
case EINVAL:
608
exitstatus = EX_USAGE;
551
ret = argp_parse(&argp, argc, argv, ARGP_IN_ORDER, 0, NULL);
552
if(ret == ARGP_ERR_UNKNOWN){
553
fprintf(stderr, "Unknown error while parsing arguments\n");
554
exitstatus = EXIT_FAILURE;
609
555
goto fallback;
610
556
}
611
557
617
563
conffp = fopen(AFILE, "r");
618
564
} else {
619
565
conffp = fopen(argfile, "r");
620
}
566
}
621
567
if(conffp != NULL){
622
568
char *org_line = NULL;
623
569
char *p, *arg, *new_arg, *line;
624
570
size_t size = 0;
625
571
const char whitespace_delims[] = " \r\t\f\v\n";
626
572
const char comment_delim[] = "#";
627
573
628
574
custom_argc = 1;
629
575
custom_argv = malloc(sizeof(char*) * 2);
630
576
if(custom_argv == NULL){
631
error(0, errno, "malloc");
632
exitstatus = EX_OSERR;
577
perror("malloc");
578
exitstatus = EXIT_FAILURE;
633
579
goto fallback;
634
580
}
635
581
custom_argv[0] = argv[0];
636
582
custom_argv[1] = NULL;
637
583
638
584
/* for each line in the config file, strip whitespace and ignore
639
585
commented text */
640
586
while(true){
642
588
if(sret == -1){
643
589
break;
644
590
}
645
591
646
592
line = org_line;
647
593
arg = strsep(&line, comment_delim);
648
594
while((p = strsep(&arg, whitespace_delims)) != NULL){
651
597
}
652
598
new_arg = strdup(p);
653
599
if(new_arg == NULL){
654
error(0, errno, "strdup");
655
exitstatus = EX_OSERR;
600
perror("strdup");
601
exitstatus = EXIT_FAILURE;
656
602
free(org_line);
657
603
goto fallback;
658
604
}
661
607
custom_argv = realloc(custom_argv, sizeof(char *)
662
608
* ((unsigned int) custom_argc + 1));
663
609
if(custom_argv == NULL){
664
error(0, errno, "realloc");
665
exitstatus = EX_OSERR;
610
perror("realloc");
611
exitstatus = EXIT_FAILURE;
666
612
free(org_line);
667
613
goto fallback;
668
614
}
669
615
custom_argv[custom_argc-1] = new_arg;
670
custom_argv[custom_argc] = NULL;
616
custom_argv[custom_argc] = NULL;
671
617
}
672
618
}
673
do {
674
ret = fclose(conffp);
675
} while(ret == EOF and errno == EINTR);
676
if(ret == EOF){
677
error(0, errno, "fclose");
678
exitstatus = EX_IOERR;
679
goto fallback;
680
}
681
619
free(org_line);
682
620
} else {
683
621
/* Check for harmful errors and go to fallback. Other errors might
684
622
not affect opening plugins */
685
623
if(errno == EMFILE or errno == ENFILE or errno == ENOMEM){
686
error(0, errno, "fopen");
687
exitstatus = EX_OSERR;
624
perror("fopen");
625
exitstatus = EXIT_FAILURE;
688
626
goto fallback;
689
627
}
690
628
}
691
/* If there were any arguments from the configuration file, pass
692
them to parser as command line arguments */
629
/* If there was any arguments from configuration file,
630
pass them to parser as command arguments */
693
631
if(custom_argv != NULL){
694
ret = argp_parse(&argp, custom_argc, custom_argv,
695
ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
696
NULL, NULL);
697
switch(ret){
698
case 0:
699
break;
700
case ENOMEM:
701
default:
702
errno = ret;
703
error(0, errno, "argp_parse");
704
exitstatus = EX_OSERR;
705
goto fallback;
706
case EINVAL:
707
exitstatus = EX_CONFIG;
632
ret = argp_parse(&argp, custom_argc, custom_argv, ARGP_IN_ORDER,
633
0, NULL);
634
if(ret == ARGP_ERR_UNKNOWN){
635
fprintf(stderr, "Unknown error while parsing arguments\n");
636
exitstatus = EXIT_FAILURE;
708
637
goto fallback;
709
638
}
710
639
}
711
640
712
641
/* Parse actual command line arguments, to let them override the
713
642
config file */
714
ret = argp_parse(&argp, argc, argv,
715
ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
716
NULL, NULL);
717
switch(ret){
718
case 0:
719
break;
720
case ENOMEM:
721
default:
722
errno = ret;
723
error(0, errno, "argp_parse");
724
exitstatus = EX_OSERR;
725
goto fallback;
726
case EINVAL:
727
exitstatus = EX_USAGE;
643
ret = argp_parse(&argp, argc, argv, ARGP_IN_ORDER, 0, NULL);
644
if(ret == ARGP_ERR_UNKNOWN){
645
fprintf(stderr, "Unknown error while parsing arguments\n");
646
exitstatus = EXIT_FAILURE;
728
647
goto fallback;
729
648
}
730
649
743
662
}
744
663
745
664
/* Strip permissions down to nobody */
665
ret = setuid(uid);
666
if(ret == -1){
667
perror("setuid");
668
}
746
669
setgid(gid);
747
670
if(ret == -1){
748
error(0, errno, "setgid");
749
}
750
ret = setuid(uid);
751
if(ret == -1){
752
error(0, errno, "setuid");
753
}
754
755
/* Open plugin directory with close_on_exec flag */
671
perror("setgid");
672
}
673
674
if(plugindir == NULL){
675
dir = opendir(PDIR);
676
} else {
677
dir = opendir(plugindir);
678
}
679
680
if(dir == NULL){
681
perror("Could not open plugin dir");
682
exitstatus = EXIT_FAILURE;
683
goto fallback;
684
}
685
686
/* Set the FD_CLOEXEC flag on the directory, if possible */
756
687
{
757
int dir_fd = -1;
758
if(plugindir == NULL){
759
dir_fd = open(PDIR, O_RDONLY |
760
#ifdef O_CLOEXEC
761
O_CLOEXEC
762
#else /* not O_CLOEXEC */
763
0
764
#endif /* not O_CLOEXEC */
765
);
766
} else {
767
dir_fd = open(plugindir, O_RDONLY |
768
#ifdef O_CLOEXEC
769
O_CLOEXEC
770
#else /* not O_CLOEXEC */
771
0
772
#endif /* not O_CLOEXEC */
773
);
774
}
775
if(dir_fd == -1){
776
error(0, errno, "Could not open plugin dir");
777
exitstatus = EX_UNAVAILABLE;
778
goto fallback;
779
}
780
781
#ifndef O_CLOEXEC
782
/* Set the FD_CLOEXEC flag on the directory */
783
ret = set_cloexec_flag(dir_fd);
784
if(ret < 0){
785
error(0, errno, "set_cloexec_flag");
786
TEMP_FAILURE_RETRY(close(dir_fd));
787
exitstatus = EX_OSERR;
788
goto fallback;
789
}
790
#endif /* O_CLOEXEC */
791
792
dir = fdopendir(dir_fd);
793
if(dir == NULL){
794
error(0, errno, "Could not open plugin dir");
795
TEMP_FAILURE_RETRY(close(dir_fd));
796
exitstatus = EX_OSERR;
797
goto fallback;
688
int dir_fd = dirfd(dir);
689
if(dir_fd >= 0){
690
ret = set_cloexec_flag(dir_fd);
691
if(ret < 0){
692
perror("set_cloexec_flag");
693
exitstatus = EXIT_FAILURE;
694
goto fallback;
695
}
798
696
}
799
697
}
800
698
802
700
803
701
/* Read and execute any executable in the plugin directory*/
804
702
while(true){
805
do {
806
dirst = readdir(dir);
807
} while(dirst == NULL and errno == EINTR);
703
dirst = readdir(dir);
808
704
809
705
/* All directory entries have been processed */
810
706
if(dirst == NULL){
811
707
if(errno == EBADF){
812
error(0, errno, "readdir");
813
exitstatus = EX_IOERR;
708
perror("readdir");
709
exitstatus = EXIT_FAILURE;
814
710
goto fallback;
815
711
}
816
712
break;
846
742
for(const char **suf = bad_suffixes; *suf != NULL; suf++){
847
743
size_t suf_len = strlen(*suf);
848
744
if((d_name_len >= suf_len)
849
and (strcmp((dirst->d_name) + d_name_len-suf_len, *suf)
745
and (strcmp((dirst->d_name)+d_name_len-suf_len, *suf)
850
746
== 0)){
851
747
if(debug){
852
748
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
861
757
continue;
862
758
}
863
759
}
864
760
865
761
char *filename;
866
762
if(plugindir == NULL){
867
ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, PDIR "/%s",
868
dirst->d_name));
763
ret = asprintf(&filename, PDIR "/%s", dirst->d_name);
869
764
} else {
870
ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, "%s/%s",
871
plugindir,
872
dirst->d_name));
765
ret = asprintf(&filename, "%s/%s", plugindir, dirst->d_name);
873
766
}
874
767
if(ret < 0){
875
error(0, errno, "asprintf");
768
perror("asprintf");
876
769
continue;
877
770
}
878
771
879
ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
772
ret = stat(filename, &st);
880
773
if(ret == -1){
881
error(0, errno, "stat");
774
perror("stat");
882
775
free(filename);
883
776
continue;
884
777
}
885
778
886
779
/* Ignore non-executable files */
887
if(not S_ISREG(st.st_mode)
888
or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
780
if(not S_ISREG(st.st_mode) or (access(filename, X_OK) != 0)){
889
781
if(debug){
890
782
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
891
783
" with bad type or mode\n", filename);
896
788
897
789
plugin *p = getplugin(dirst->d_name);
898
790
if(p == NULL){
899
error(0, errno, "getplugin");
791
perror("getplugin");
900
792
free(filename);
901
793
continue;
902
794
}
914
806
if(g != NULL){
915
807
for(char **a = g->argv + 1; *a != NULL; a++){
916
808
if(not add_argument(p, *a)){
917
error(0, errno, "add_argument");
809
perror("add_argument");
918
810
}
919
811
}
920
812
/* Add global environment variables */
921
813
for(char **e = g->environ; *e != NULL; e++){
922
814
if(not add_environment(p, *e, false)){
923
error(0, errno, "add_environment");
815
perror("add_environment");
924
816
}
925
817
}
926
818
}
931
823
if(p->environ[0] != NULL){
932
824
for(char **e = environ; *e != NULL; e++){
933
825
if(not add_environment(p, *e, false)){
934
error(0, errno, "add_environment");
826
perror("add_environment");
935
827
}
936
828
}
937
829
}
938
830
939
831
int pipefd[2];
940
ret = (int)TEMP_FAILURE_RETRY(pipe(pipefd));
832
ret = pipe(pipefd);
941
833
if(ret == -1){
942
error(0, errno, "pipe");
943
exitstatus = EX_OSERR;
834
perror("pipe");
835
exitstatus = EXIT_FAILURE;
944
836
goto fallback;
945
837
}
946
838
/* Ask OS to automatic close the pipe on exec */
947
839
ret = set_cloexec_flag(pipefd[0]);
948
840
if(ret < 0){
949
error(0, errno, "set_cloexec_flag");
950
exitstatus = EX_OSERR;
841
perror("set_cloexec_flag");
842
exitstatus = EXIT_FAILURE;
951
843
goto fallback;
952
844
}
953
845
ret = set_cloexec_flag(pipefd[1]);
954
846
if(ret < 0){
955
error(0, errno, "set_cloexec_flag");
956
exitstatus = EX_OSERR;
847
perror("set_cloexec_flag");
848
exitstatus = EXIT_FAILURE;
957
849
goto fallback;
958
850
}
959
851
/* Block SIGCHLD until process is safely in process list */
960
ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
961
&sigchld_action.sa_mask,
962
NULL));
852
ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
963
853
if(ret < 0){
964
error(0, errno, "sigprocmask");
965
exitstatus = EX_OSERR;
854
perror("sigprocmask");
855
exitstatus = EXIT_FAILURE;
966
856
goto fallback;
967
857
}
968
858
/* Starting a new process to be watched */
969
pid_t pid;
970
do {
971
pid = fork();
972
} while(pid == -1 and errno == EINTR);
859
pid_t pid = fork();
973
860
if(pid == -1){
974
error(0, errno, "fork");
975
exitstatus = EX_OSERR;
861
perror("fork");
862
exitstatus = EXIT_FAILURE;
976
863
goto fallback;
977
864
}
978
865
if(pid == 0){
979
866
/* this is the child process */
980
867
ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
981
868
if(ret < 0){
982
error(0, errno, "sigaction");
983
_exit(EX_OSERR);
869
perror("sigaction");
870
_exit(EXIT_FAILURE);
984
871
}
985
872
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
986
873
if(ret < 0){
987
error(0, errno, "sigprocmask");
988
_exit(EX_OSERR);
874
perror("sigprocmask");
875
_exit(EXIT_FAILURE);
989
876
}
990
877
991
878
ret = dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */
992
879
if(ret == -1){
993
error(0, errno, "dup2");
994
_exit(EX_OSERR);
880
perror("dup2");
881
_exit(EXIT_FAILURE);
995
882
}
996
883
997
884
if(dirfd(dir) < 0){
1001
888
}
1002
889
if(p->environ[0] == NULL){
1003
890
if(execv(filename, p->argv) < 0){
1004
error(0, errno, "execv for %s", filename);
1005
_exit(EX_OSERR);
891
perror("execv");
892
_exit(EXIT_FAILURE);
1006
893
}
1007
894
} else {
1008
895
if(execve(filename, p->argv, p->environ) < 0){
1009
error(0, errno, "execve for %s", filename);
1010
_exit(EX_OSERR);
896
perror("execve");
897
_exit(EXIT_FAILURE);
1011
898
}
1012
899
}
1013
900
/* no return */
1014
901
}
1015
902
/* Parent process */
1016
TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
1017
pipe */
903
close(pipefd[1]); /* Close unused write end of pipe */
1018
904
free(filename);
1019
905
plugin *new_plugin = getplugin(dirst->d_name);
1020
906
if(new_plugin == NULL){
1021
error(0, errno, "getplugin");
1022
ret = (int)(TEMP_FAILURE_RETRY
1023
(sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask,
1024
NULL)));
907
perror("getplugin");
908
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
1025
909
if(ret < 0){
1026
error(0, errno, "sigprocmask");
910
perror("sigprocmask");
1027
911
}
1028
exitstatus = EX_OSERR;
912
exitstatus = EXIT_FAILURE;
1029
913
goto fallback;
1030
914
}
1031
915
1034
918
1035
919
/* Unblock SIGCHLD so signal handler can be run if this process
1036
920
has already completed */
1037
ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_UNBLOCK,
1038
&sigchld_action.sa_mask,
1039
NULL));
921
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
1040
922
if(ret < 0){
1041
error(0, errno, "sigprocmask");
1042
exitstatus = EX_OSERR;
923
perror("sigprocmask");
924
exitstatus = EXIT_FAILURE;
1043
925
goto fallback;
1044
926
}
1045
927
1046
FD_SET(new_plugin->fd, &rfds_all); /* Spurious warning from
1047
-Wconversion */
928
FD_SET(new_plugin->fd, &rfds_all);
1048
929
1049
930
if(maxfd < new_plugin->fd){
1050
931
maxfd = new_plugin->fd;
1051
932
}
1052
933
}
1053
934
1054
TEMP_FAILURE_RETRY(closedir(dir));
935
closedir(dir);
1055
936
dir = NULL;
1056
free_plugin(getplugin(NULL));
1057
937
1058
938
for(plugin *p = plugin_list; p != NULL; p = p->next){
1059
939
if(p->pid != 0){
1070
950
while(plugin_list){
1071
951
fd_set rfds = rfds_all;
1072
952
int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
1073
if(select_ret == -1 and errno != EINTR){
1074
error(0, errno, "select");
1075
exitstatus = EX_OSERR;
953
if(select_ret == -1){
954
perror("select");
955
exitstatus = EXIT_FAILURE;
1076
956
goto fallback;
1077
957
}
1078
958
/* OK, now either a process completed, or something can be read
1079
959
from one of them */
1080
960
for(plugin *proc = plugin_list; proc != NULL;){
1081
961
/* Is this process completely done? */
1082
if(proc->completed and proc->eof){
962
if(proc->eof and proc->completed){
1083
963
/* Only accept the plugin output if it exited cleanly */
1084
964
if(not WIFEXITED(proc->status)
1085
965
or WEXITSTATUS(proc->status) != 0){
1086
966
/* Bad exit by plugin */
1087
967
1088
968
if(debug){
1089
969
if(WIFEXITED(proc->status)){
1090
fprintf(stderr, "Plugin %s [%" PRIdMAX "] exited with"
1091
" status %d\n", proc->name,
1092
(intmax_t) (proc->pid),
970
fprintf(stderr, "Plugin %" PRIdMAX " exited with status"
971
" %d\n", (intmax_t) (proc->pid),
1093
972
WEXITSTATUS(proc->status));
1094
} else if(WIFSIGNALED(proc->status)){
1095
fprintf(stderr, "Plugin %s [%" PRIdMAX "] killed by"
1096
" signal %d: %s\n", proc->name,
1097
(intmax_t) (proc->pid),
1098
WTERMSIG(proc->status),
1099
strsignal(WTERMSIG(proc->status)));
973
} else if(WIFSIGNALED(proc->status)) {
974
fprintf(stderr, "Plugin %" PRIdMAX " killed by signal"
975
" %d\n", (intmax_t) (proc->pid),
976
WTERMSIG(proc->status));
1100
977
} else if(WCOREDUMP(proc->status)){
1101
fprintf(stderr, "Plugin %s [%" PRIdMAX "] dumped"
1102
" core\n", proc->name, (intmax_t) (proc->pid));
978
fprintf(stderr, "Plugin %" PRIdMAX " dumped core\n",
979
(intmax_t) (proc->pid));
1103
980
}
1104
981
}
1105
982
1106
983
/* Remove the plugin */
1107
FD_CLR(proc->fd, &rfds_all); /* Spurious warning from
1108
-Wconversion */
1109
984
FD_CLR(proc->fd, &rfds_all);
985
1110
986
/* Block signal while modifying process_list */
1111
ret = (int)TEMP_FAILURE_RETRY(sigprocmask
1112
(SIG_BLOCK,
1113
&sigchld_action.sa_mask,
1114
NULL));
987
ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
1115
988
if(ret < 0){
1116
error(0, errno, "sigprocmask");
1117
exitstatus = EX_OSERR;
989
perror("sigprocmask");
990
exitstatus = EXIT_FAILURE;
1118
991
goto fallback;
1119
992
}
1120
993
1123
996
proc = next_plugin;
1124
997
1125
998
/* We are done modifying process list, so unblock signal */
1126
ret = (int)(TEMP_FAILURE_RETRY
1127
(sigprocmask(SIG_UNBLOCK,
1128
&sigchld_action.sa_mask, NULL)));
999
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask,
1000
NULL);
1129
1001
if(ret < 0){
1130
error(0, errno, "sigprocmask");
1131
exitstatus = EX_OSERR;
1002
perror("sigprocmask");
1003
exitstatus = EXIT_FAILURE;
1132
1004
goto fallback;
1133
1005
}
1134
1006
1144
1016
bool bret = print_out_password(proc->buffer,
1145
1017
proc->buffer_length);
1146
1018
if(not bret){
1147
error(0, errno, "print_out_password");
1148
exitstatus = EX_IOERR;
1019
perror("print_out_password");
1020
exitstatus = EXIT_FAILURE;
1149
1021
}
1150
1022
goto fallback;
1151
1023
}
1152
1024
1153
1025
/* This process has not completed. Does it have any output? */
1154
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious
1155
warning from
1156
-Wconversion */
1026
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
1157
1027
/* This process had nothing to say at this time */
1158
1028
proc = proc->next;
1159
1029
continue;
1163
1033
proc->buffer = realloc(proc->buffer, proc->buffer_size
1164
1034
+ (size_t) BUFFER_SIZE);
1165
1035
if(proc->buffer == NULL){
1166
error(0, errno, "malloc");
1167
exitstatus = EX_OSERR;
1036
perror("malloc");
1037
exitstatus = EXIT_FAILURE;
1168
1038
goto fallback;
1169
1039
}
1170
1040
proc->buffer_size += BUFFER_SIZE;
1171
1041
}
1172
1042
/* Read from the process */
1173
sret = TEMP_FAILURE_RETRY(read(proc->fd,
1174
proc->buffer
1175
+ proc->buffer_length,
1176
BUFFER_SIZE));
1043
sret = read(proc->fd, proc->buffer + proc->buffer_length,
1044
BUFFER_SIZE);
1177
1045
if(sret < 0){
1178
1046
/* Read error from this process; ignore the error */
1179
1047
proc = proc->next;
1187
1055
}
1188
1056
}
1189
1057
}
1190
1191
1058
1059
1192
1060
fallback:
1193
1061
1194
if(plugin_list == NULL or (exitstatus != EXIT_SUCCESS
1195
and exitstatus != EX_OK)){
1062
if(plugin_list == NULL or exitstatus != EXIT_SUCCESS){
1196
1063
/* Fallback if all plugins failed, none are found or an error
1197
1064
occured */
1198
1065
bool bret;
1206
1073
}
1207
1074
bret = print_out_password(passwordbuffer, len);
1208
1075
if(not bret){
1209
error(0, errno, "print_out_password");
1210
exitstatus = EX_IOERR;
1076
perror("print_out_password");
1077
exitstatus = EXIT_FAILURE;
1211
1078
}
1212
1079
}
1213
1080
1214
1081
/* Restore old signal handler */
1215
1082
ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
1216
1083
if(ret == -1){
1217
error(0, errno, "sigaction");
1218
exitstatus = EX_OSERR;
1084
perror("sigaction");
1085
exitstatus = EXIT_FAILURE;
1219
1086
}
1220
1087
1221
1088
if(custom_argv != NULL){
1236
1103
ret = kill(p->pid, SIGTERM);
1237
1104
if(ret == -1 and errno != ESRCH){
1238
1105
/* Set-uid proccesses might not get closed */
1239
error(0, errno, "kill");
1106
perror("kill");
1240
1107
}
1241
1108
}
1242
1109
}
1243
1110
1244
1111
/* Wait for any remaining child processes to terminate */
1245
do {
1112
do{
1246
1113
ret = wait(NULL);
1247
1114
} while(ret >= 0);
1248
1115
if(errno != ECHILD){
1249
error(0, errno, "wait");
1116
perror("wait");
1250
1117
}
1251
1118
1252
1119
free_plugin_list();
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0