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