18
19
* along with this program. If not, see
19
20
* <http://www.gnu.org/licenses/>.
21
* Contact the authors at <mandos@fukt.bsnet.se>.
22
* Contact the authors at <mandos@recompile.se>.
24
25
#define _GNU_SOURCE /* TEMP_FAILURE_RETRY(), getline(),
26
asprintf(), O_CLOEXEC */
26
27
#include <stddef.h> /* size_t, NULL */
27
#include <stdlib.h> /* malloc(), exit(), EXIT_FAILURE,
28
EXIT_SUCCESS, realloc() */
28
#include <stdlib.h> /* malloc(), exit(), EXIT_SUCCESS,
29
30
#include <stdbool.h> /* bool, true, false */
30
#include <stdio.h> /* perror, popen(), fileno(),
31
fprintf(), stderr, STDOUT_FILENO */
32
#include <sys/types.h> /* DIR, opendir(), stat(), struct
31
#include <stdio.h> /* fileno(), fprintf(),
32
stderr, STDOUT_FILENO */
33
#include <sys/types.h> /* DIR, fdopendir(), stat(), struct
33
34
stat, waitpid(), WIFEXITED(),
34
35
WEXITSTATUS(), wait(), pid_t,
35
36
uid_t, gid_t, getuid(), getgid(),
37
38
#include <sys/select.h> /* fd_set, select(), FD_ZERO(),
38
39
FD_SET(), FD_ISSET(), FD_CLR */
39
40
#include <sys/wait.h> /* wait(), waitpid(), WIFEXITED(),
41
WEXITSTATUS(), WTERMSIG(),
41
43
#include <sys/stat.h> /* struct stat, stat(), S_ISREG() */
42
44
#include <iso646.h> /* and, or, not */
43
#include <dirent.h> /* DIR, struct dirent, opendir(),
45
#include <dirent.h> /* DIR, struct dirent, fdopendir(),
44
46
readdir(), closedir(), dirfd() */
45
47
#include <unistd.h> /* struct stat, stat(), S_ISREG(),
46
48
fcntl(), setuid(), setgid(),
47
49
F_GETFD, F_SETFD, FD_CLOEXEC,
48
50
access(), pipe(), fork(), close()
49
dup2, STDOUT_FILENO, _exit(),
51
dup2(), STDOUT_FILENO, _exit(),
50
52
execv(), write(), read(),
52
54
#include <fcntl.h> /* fcntl(), F_GETFD, F_SETFD,
54
#include <string.h> /* strsep, strlen(), asprintf() */
56
#include <string.h> /* strsep, strlen(), asprintf(),
57
strsignal(), strcmp(), strncmp() */
55
58
#include <errno.h> /* errno */
56
59
#include <argp.h> /* struct argp_option, struct
57
60
argp_state, struct argp,
89
96
size_t buffer_size;
90
97
size_t buffer_length;
92
volatile bool completed;
99
volatile sig_atomic_t completed;
94
101
struct plugin *next;
97
104
static plugin *plugin_list = NULL;
99
/* Gets a existing plugin based on name,
106
/* Gets an existing plugin based on name,
100
107
or if none is found, creates a new one */
108
__attribute__((warn_unused_result))
101
109
static plugin *getplugin(char *name){
102
/* Check for exiting plugin with that name */
103
for (plugin *p = plugin_list; p != NULL; p = p->next){
104
if ((p->name == name)
105
or (p->name and name and (strcmp(p->name, name) == 0))){
110
/* Check for existing plugin with that name */
111
for(plugin *p = plugin_list; p != NULL; p = p->next){
113
or (p->name and name and (strcmp(p->name, name) == 0))){
109
117
/* Create a new plugin */
110
plugin *new_plugin = malloc(sizeof(plugin));
111
if (new_plugin == NULL){
118
plugin *new_plugin = NULL;
120
new_plugin = malloc(sizeof(plugin));
121
} while(new_plugin == NULL and errno == EINTR);
122
if(new_plugin == NULL){
114
125
char *copy_name = NULL;
115
126
if(name != NULL){
116
copy_name = strdup(name);
128
copy_name = strdup(name);
129
} while(copy_name == NULL and errno == EINTR);
117
130
if(copy_name == NULL){
122
*new_plugin = (plugin) { .name = copy_name,
125
.next = plugin_list };
127
new_plugin->argv = malloc(sizeof(char *) * 2);
128
if (new_plugin->argv == NULL){
138
*new_plugin = (plugin){ .name = copy_name,
141
.next = plugin_list };
144
new_plugin->argv = malloc(sizeof(char *) * 2);
145
} while(new_plugin->argv == NULL and errno == EINTR);
146
if(new_plugin->argv == NULL){
130
149
free(new_plugin);
133
153
new_plugin->argv[0] = copy_name;
134
154
new_plugin->argv[1] = NULL;
136
new_plugin->environ = malloc(sizeof(char *));
157
new_plugin->environ = malloc(sizeof(char *));
158
} while(new_plugin->environ == NULL and errno == EINTR);
137
159
if(new_plugin->environ == NULL){
139
162
free(new_plugin->argv);
140
163
free(new_plugin);
143
167
new_plugin->environ[0] = NULL;
145
169
/* Append the new plugin to the list */
146
170
plugin_list = new_plugin;
147
171
return new_plugin;
150
174
/* Helper function for add_argument and add_environment */
175
__attribute__((nonnull, warn_unused_result))
151
176
static bool add_to_char_array(const char *new, char ***array,
153
178
/* Resize the pointed-to array to hold one more pointer */
154
*array = realloc(*array, sizeof(char *)
155
* (size_t) ((*len) + 2));
179
char **new_array = NULL;
181
new_array = realloc(*array, sizeof(char *)
182
* (size_t) ((*len) + 2));
183
} while(new_array == NULL and errno == EINTR);
156
184
/* Malloc check */
185
if(new_array == NULL){
160
189
/* Make a copy of the new string */
161
char *copy = strdup(new);
193
} while(copy == NULL and errno == EINTR);
162
194
if(copy == NULL){
181
214
/* Add to a plugin's environment */
182
static bool add_environment(plugin *p, const char *def){
215
__attribute__((nonnull(2), warn_unused_result))
216
static bool add_environment(plugin *p, const char *def, bool replace){
220
/* namelen = length of name of environment variable */
221
size_t namelen = (size_t)(strchrnul(def, '=') - def);
222
/* Search for this environment variable */
223
for(char **envdef = p->environ; *envdef != NULL; envdef++){
224
if(strncmp(*envdef, def, namelen + 1) == 0){
225
/* It already exists */
229
new_envdef = realloc(*envdef, strlen(def) + 1);
230
} while(new_envdef == NULL and errno == EINTR);
231
if(new_envdef == NULL){
234
*envdef = new_envdef;
235
strcpy(*envdef, def);
186
240
return add_to_char_array(def, &(p->environ), &(p->envc));
190
244
* Based on the example in the GNU LibC manual chapter 13.13 "File
191
245
* Descriptor Flags".
192
* *Note File Descriptor Flags:(libc)Descriptor Flags.
246
| [[info:libc:Descriptor%20Flags][File Descriptor Flags]] |
194
static int set_cloexec_flag(int fd)
196
int ret = fcntl(fd, F_GETFD, 0);
248
__attribute__((warn_unused_result))
249
static int set_cloexec_flag(int fd){
250
int ret = (int)TEMP_FAILURE_RETRY(fcntl(fd, F_GETFD, 0));
197
251
/* If reading the flags failed, return error indication now. */
201
255
/* Store modified flag word in the descriptor. */
202
return fcntl(fd, F_SETFD, ret | FD_CLOEXEC);
256
return (int)TEMP_FAILURE_RETRY(fcntl(fd, F_SETFD,
206
261
/* Mark processes as completed when they exit, and save their exit
208
void handle_sigchld(__attribute__((unused)) int sig){
263
static void handle_sigchld(__attribute__((unused)) int sig){
264
int old_errno = errno;
210
266
plugin *proc = plugin_list;
327
384
{ .name = "global-options", .key = 'g',
328
385
.arg = "OPTION[,OPTION[,...]]",
329
386
.doc = "Options passed to all plugins" },
330
{ .name = "global-envs", .key = 'e',
387
{ .name = "global-env", .key = 'G',
331
388
.arg = "VAR=value",
332
389
.doc = "Environment variable passed to all plugins" },
333
390
{ .name = "options-for", .key = 'o',
334
391
.arg = "PLUGIN:OPTION[,OPTION[,...]]",
335
392
.doc = "Options passed only to specified plugin" },
336
{ .name = "envs-for", .key = 'f',
393
{ .name = "env-for", .key = 'E',
337
394
.arg = "PLUGIN:ENV=value",
338
395
.doc = "Environment variable passed to specified plugin" },
339
396
{ .name = "disable", .key = 'd',
341
398
.doc = "Disable a specific plugin", .group = 1 },
399
{ .name = "enable", .key = 'e',
401
.doc = "Enable a specific plugin", .group = 1 },
342
402
{ .name = "plugin-dir", .key = 128,
343
403
.arg = "DIRECTORY",
344
404
.doc = "Specify a different plugin directory", .group = 2 },
353
413
.doc = "Group ID the plugins will run as", .group = 3 },
354
414
{ .name = "debug", .key = 132,
355
415
.doc = "Debug mode", .group = 4 },
417
* These reproduce what we would get without ARGP_NO_HELP
419
{ .name = "help", .key = '?',
420
.doc = "Give this help list", .group = -1 },
421
{ .name = "usage", .key = -3,
422
.doc = "Give a short usage message", .group = -1 },
423
{ .name = "version", .key = 'V',
424
.doc = "Print program version", .group = -1 },
359
error_t parse_opt (int key, char *arg, __attribute__((unused))
360
struct argp_state *state) {
361
/* Get the INPUT argument from `argp_parse', which we know is a
362
pointer to our plugin list pointer. */
428
__attribute__((nonnull(3)))
429
error_t parse_opt(int key, char *arg, struct argp_state *state){
364
434
case 'g': /* --global-options */
367
while((p = strsep(&arg, ",")) != NULL){
371
if(not add_argument(getplugin(NULL), p)){
372
perror("add_argument");
373
return ARGP_ERR_UNKNOWN;
378
case 'e': /* --global-envs */
383
char *envdef = strdup(arg);
387
if(not add_environment(getplugin(NULL), envdef)){
388
perror("add_environment");
437
while((plugin_option = strsep(&arg, ",")) != NULL){
438
if(not add_argument(getplugin(NULL), plugin_option)){
445
case 'G': /* --global-env */
446
if(add_environment(getplugin(NULL), arg, true)){
392
450
case 'o': /* --options-for */
394
char *p_name = strsep(&arg, ":");
395
if(p_name[0] == '\0'){
398
char *opt = strsep(&arg, ":");
404
while((p = strsep(&opt, ",")) != NULL){
408
if(not add_argument(getplugin(p_name), p)){
409
perror("add_argument");
410
return ARGP_ERR_UNKNOWN;
452
char *option_list = strchr(arg, ':');
453
if(option_list == NULL){
454
argp_error(state, "No colon in \"%s\"", arg);
461
argp_error(state, "Empty plugin name");
466
while((option = strsep(&option_list, ",")) != NULL){
467
if(not add_argument(getplugin(arg), option)){
416
case 'f': /* --envs-for */
474
case 'E': /* --env-for */
421
476
char *envdef = strchr(arg, ':');
422
477
if(envdef == NULL){
425
char *p_name = strndup(arg, (size_t) (envdef-arg));
478
argp_error(state, "No colon in \"%s\"", arg);
430
if(not add_environment(getplugin(p_name), envdef)){
431
perror("add_environment");
485
argp_error(state, "Empty plugin name");
489
if(add_environment(getplugin(arg), envdef, true)){
435
494
case 'd': /* --disable */
437
plugin *p = getplugin(arg);
439
return ARGP_ERR_UNKNOWN;
496
plugin *p = getplugin(arg);
503
case 'e': /* --enable */
505
plugin *p = getplugin(arg);
444
512
case 128: /* --plugin-dir */
445
514
plugindir = strdup(arg);
446
if(plugindir == NULL){
450
case 129: /* --config-file */
515
if(plugindir != NULL){
519
case 129: /* --config-file */
520
/* This is already done by parse_opt_config_file() */
522
case 130: /* --userid */
523
tmp_id = strtoimax(arg, &tmp, 10);
524
if(errno != 0 or tmp == arg or *tmp != '\0'
525
or tmp_id != (uid_t)tmp_id){
526
argp_error(state, "Bad user ID number: \"%s\", using %"
527
PRIdMAX, arg, (intmax_t)uid);
533
case 131: /* --groupid */
534
tmp_id = strtoimax(arg, &tmp, 10);
535
if(errno != 0 or tmp == arg or *tmp != '\0'
536
or tmp_id != (gid_t)tmp_id){
537
argp_error(state, "Bad group ID number: \"%s\", using %"
538
PRIdMAX, arg, (intmax_t)gid);
544
case 132: /* --debug */
548
* These reproduce what we would get without ARGP_NO_HELP
550
case '?': /* --help */
551
state->flags &= ~(unsigned int)ARGP_NO_EXIT; /* force exit */
552
argp_state_help(state, state->out_stream, ARGP_HELP_STD_HELP);
553
case -3: /* --usage */
554
state->flags &= ~(unsigned int)ARGP_NO_EXIT; /* force exit */
555
argp_state_help(state, state->out_stream,
556
ARGP_HELP_USAGE | ARGP_HELP_EXIT_OK);
557
case 'V': /* --version */
558
fprintf(state->out_stream, "%s\n", argp_program_version);
562
* When adding more options before this line, remember to also add a
563
* "case" to the "parse_opt_config_file" function below.
566
/* Cryptsetup always passes an argument, which is an empty
567
string if "none" was specified in /etc/crypttab. So if
568
argument was empty, we ignore it silently. */
573
return ARGP_ERR_UNKNOWN;
575
return errno; /* Set to 0 at start */
578
/* This option parser is the same as parse_opt() above, except it
579
ignores everything but the --config-file option. */
580
error_t parse_opt_config_file(int key, char *arg,
581
__attribute__((unused))
582
struct argp_state *state){
585
case 'g': /* --global-options */
586
case 'G': /* --global-env */
587
case 'o': /* --options-for */
588
case 'E': /* --env-for */
589
case 'd': /* --disable */
590
case 'e': /* --enable */
591
case 128: /* --plugin-dir */
593
case 129: /* --config-file */
451
595
argfile = strdup(arg);
456
600
case 130: /* --userid */
457
uid = (uid_t)strtol(arg, NULL, 10);
459
601
case 131: /* --groupid */
460
gid = (gid_t)strtol(arg, NULL, 10);
462
602
case 132: /* --debug */
603
case '?': /* --help */
604
case -3: /* --usage */
605
case 'V': /* --version */
465
606
case ARGP_KEY_ARG:
466
fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
471
609
return ARGP_ERR_UNKNOWN;
476
struct argp argp = { .options = options, .parser = parse_opt,
477
.args_doc = "[+PLUS_SEPARATED_OPTIONS]",
614
struct argp argp = { .options = options,
615
.parser = parse_opt_config_file,
478
617
.doc = "Mandos plugin runner -- Run plugins" };
480
ret = argp_parse (&argp, argc, argv, 0, 0, NULL);
481
if (ret == ARGP_ERR_UNKNOWN){
482
fprintf(stderr, "Unknown error while parsing arguments\n");
483
exitstatus = EXIT_FAILURE;
619
/* Parse using parse_opt_config_file() in order to get the custom
620
config file location, if any. */
621
ret = argp_parse(&argp, argc, argv,
622
ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
630
error(0, errno, "argp_parse");
631
exitstatus = EX_OSERR;
634
exitstatus = EX_USAGE;
487
/* Opens the configfile if aviable */
488
if (argfile == NULL){
638
/* Reset to the normal argument parser */
639
argp.parser = parse_opt;
641
/* Open the configfile if available */
489
643
conffp = fopen(AFILE, "r");
491
645
conffp = fopen(argfile, "r");
493
647
if(conffp != NULL){
494
648
char *org_line = NULL;
495
649
char *p, *arg, *new_arg, *line;
498
651
const char whitespace_delims[] = " \r\t\f\v\n";
499
652
const char comment_delim[] = "#";
502
655
custom_argv = malloc(sizeof(char*) * 2);
503
656
if(custom_argv == NULL){
505
exitstatus = EXIT_FAILURE;
657
error(0, errno, "malloc");
658
exitstatus = EX_OSERR;
508
661
custom_argv[0] = argv[0];
509
662
custom_argv[1] = NULL;
511
664
/* for each line in the config file, strip whitespace and ignore
512
665
commented text */
525
678
new_arg = strdup(p);
526
679
if(new_arg == NULL){
528
exitstatus = EXIT_FAILURE;
680
error(0, errno, "strdup");
681
exitstatus = EX_OSERR;
533
686
custom_argc += 1;
534
custom_argv = realloc(custom_argv, sizeof(char *)
535
* ((unsigned int) custom_argc + 1));
536
if(custom_argv == NULL){
538
exitstatus = EXIT_FAILURE;
688
char **new_argv = realloc(custom_argv, sizeof(char *)
691
if(new_argv == NULL){
692
error(0, errno, "realloc");
693
exitstatus = EX_OSERR;
698
custom_argv = new_argv;
542
701
custom_argv[custom_argc-1] = new_arg;
543
custom_argv[custom_argc] = NULL;
702
custom_argv[custom_argc] = NULL;
706
ret = fclose(conffp);
707
} while(ret == EOF and errno == EINTR);
709
error(0, errno, "fclose");
710
exitstatus = EX_IOERR;
548
715
/* Check for harmful errors and go to fallback. Other errors might
549
716
not affect opening plugins */
550
if (errno == EMFILE or errno == ENFILE or errno == ENOMEM){
552
exitstatus = EXIT_FAILURE;
717
if(errno == EMFILE or errno == ENFILE or errno == ENOMEM){
718
error(0, errno, "fopen");
719
exitstatus = EX_OSERR;
556
/* If there was any arguments from configuration file,
557
pass them to parser as command arguments */
723
/* If there were any arguments from the configuration file, pass
724
them to parser as command line arguments */
558
725
if(custom_argv != NULL){
559
ret = argp_parse (&argp, custom_argc, custom_argv, 0, 0, NULL);
560
if (ret == ARGP_ERR_UNKNOWN){
561
fprintf(stderr, "Unknown error while parsing arguments\n");
562
exitstatus = EXIT_FAILURE;
726
ret = argp_parse(&argp, custom_argc, custom_argv,
727
ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
735
error(0, errno, "argp_parse");
736
exitstatus = EX_OSERR;
739
exitstatus = EX_CONFIG;
744
/* Parse actual command line arguments, to let them override the
746
ret = argp_parse(&argp, argc, argv,
747
ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
755
error(0, errno, "argp_parse");
756
exitstatus = EX_OSERR;
759
exitstatus = EX_USAGE;
568
764
for(plugin *p = plugin_list; p != NULL; p=p->next){
569
765
fprintf(stderr, "Plugin: %s has %d arguments\n",
571
767
for(char **a = p->argv; *a != NULL; a++){
572
768
fprintf(stderr, "\tArg: %s\n", *a);
574
fprintf(stderr, "...and %u environment variables\n", p->envc);
770
fprintf(stderr, "...and %d environment variables\n", p->envc);
575
771
for(char **a = p->environ; *a != NULL; a++){
576
772
fprintf(stderr, "\t%s\n", *a);
581
/* Strip permissions down to nobody */
778
/* Work around Debian bug #633582:
779
<http://bugs.debian.org/633582> */
780
int plugindir_fd = open(/* plugindir or */ PDIR, O_RDONLY);
781
if(plugindir_fd == -1){
782
error(0, errno, "open");
784
ret = (int)TEMP_FAILURE_RETRY(fstat(plugindir_fd, &st));
786
error(0, errno, "fstat");
788
if(S_ISDIR(st.st_mode) and st.st_uid == 0 and st.st_gid == 0){
789
ret = fchown(plugindir_fd, uid, gid);
791
error(0, errno, "fchown");
795
TEMP_FAILURE_RETRY(close(plugindir_fd));
799
/* Lower permissions */
802
error(0, errno, "setgid");
582
804
ret = setuid(uid);
591
if (plugindir == NULL){
594
dir = opendir(plugindir);
598
perror("Could not open plugin dir");
599
exitstatus = EXIT_FAILURE;
603
/* Set the FD_CLOEXEC flag on the directory, if possible */
806
error(0, errno, "setuid");
809
/* Open plugin directory with close_on_exec flag */
605
int dir_fd = dirfd(dir);
607
ret = set_cloexec_flag(dir_fd);
609
perror("set_cloexec_flag");
610
exitstatus = EXIT_FAILURE;
812
if(plugindir == NULL){
813
dir_fd = open(PDIR, O_RDONLY |
816
#else /* not O_CLOEXEC */
818
#endif /* not O_CLOEXEC */
821
dir_fd = open(plugindir, O_RDONLY |
824
#else /* not O_CLOEXEC */
826
#endif /* not O_CLOEXEC */
830
error(0, errno, "Could not open plugin dir");
831
exitstatus = EX_UNAVAILABLE;
836
/* Set the FD_CLOEXEC flag on the directory */
837
ret = set_cloexec_flag(dir_fd);
839
error(0, errno, "set_cloexec_flag");
840
TEMP_FAILURE_RETRY(close(dir_fd));
841
exitstatus = EX_OSERR;
844
#endif /* O_CLOEXEC */
846
dir = fdopendir(dir_fd);
848
error(0, errno, "Could not open plugin dir");
849
TEMP_FAILURE_RETRY(close(dir_fd));
850
exitstatus = EX_OSERR;
616
855
FD_ZERO(&rfds_all);
618
857
/* Read and execute any executable in the plugin directory*/
620
dirst = readdir(dir);
860
dirst = readdir(dir);
861
} while(dirst == NULL and errno == EINTR);
622
// All directory entries have been processed
863
/* All directory entries have been processed */
623
864
if(dirst == NULL){
626
exitstatus = EXIT_FAILURE;
866
error(0, errno, "readdir");
867
exitstatus = EX_IOERR;
734
1000
process, too. */
735
1001
if(p->environ[0] != NULL){
736
1002
for(char **e = environ; *e != NULL; e++){
737
char *copy = strdup(*e);
742
if(not add_environment(p, copy)){
743
perror("add_environment");
1003
if(not add_environment(p, *e, false)){
1004
error(0, errno, "add_environment");
752
exitstatus = EXIT_FAILURE;
1010
ret = (int)TEMP_FAILURE_RETRY(pipe(pipefd));
1012
error(0, errno, "pipe");
1013
exitstatus = EX_OSERR;
755
1016
/* Ask OS to automatic close the pipe on exec */
756
1017
ret = set_cloexec_flag(pipefd[0]);
758
perror("set_cloexec_flag");
759
exitstatus = EXIT_FAILURE;
1019
error(0, errno, "set_cloexec_flag");
1020
exitstatus = EX_OSERR;
762
1023
ret = set_cloexec_flag(pipefd[1]);
764
perror("set_cloexec_flag");
765
exitstatus = EXIT_FAILURE;
1025
error(0, errno, "set_cloexec_flag");
1026
exitstatus = EX_OSERR;
768
1029
/* Block SIGCHLD until process is safely in process list */
769
ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
1030
ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
1031
&sigchld_action.sa_mask,
771
perror("sigprocmask");
772
exitstatus = EXIT_FAILURE;
1034
error(0, errno, "sigprocmask");
1035
exitstatus = EX_OSERR;
775
// Starting a new process to be watched
1038
/* Starting a new process to be watched */
1042
} while(pid == -1 and errno == EINTR);
779
exitstatus = EXIT_FAILURE;
1044
error(0, errno, "fork");
1045
exitstatus = EX_OSERR;
783
1049
/* this is the child process */
784
1050
ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
1052
error(0, errno, "sigaction");
789
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
1055
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
791
perror("sigprocmask");
1057
error(0, errno, "sigprocmask");
795
1061
ret = dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */
1063
error(0, errno, "dup2");
801
1067
if(dirfd(dir) < 0){
863
1147
free_plugin_list();
867
1151
/* Main loop while running plugins exist */
868
1152
while(plugin_list){
869
1153
fd_set rfds = rfds_all;
870
1154
int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
871
if (select_ret == -1){
873
exitstatus = EXIT_FAILURE;
1155
if(select_ret == -1 and errno != EINTR){
1156
error(0, errno, "select");
1157
exitstatus = EX_OSERR;
876
1160
/* OK, now either a process completed, or something can be read
877
1161
from one of them */
878
for(plugin *proc = plugin_list; proc != NULL; proc = proc->next){
1162
for(plugin *proc = plugin_list; proc != NULL;){
879
1163
/* Is this process completely done? */
880
if(proc->eof and proc->completed){
1164
if(proc->completed and proc->eof){
881
1165
/* Only accept the plugin output if it exited cleanly */
882
1166
if(not WIFEXITED(proc->status)
883
1167
or WEXITSTATUS(proc->status) != 0){
884
1168
/* Bad exit by plugin */
887
1171
if(WIFEXITED(proc->status)){
888
fprintf(stderr, "Plugin %u exited with status %d\n",
889
(unsigned int) (proc->pid),
1172
fprintf(stderr, "Plugin %s [%" PRIdMAX "] exited with"
1173
" status %d\n", proc->name,
1174
(intmax_t) (proc->pid),
890
1175
WEXITSTATUS(proc->status));
891
} else if(WIFSIGNALED(proc->status)) {
892
fprintf(stderr, "Plugin %u killed by signal %d\n",
893
(unsigned int) (proc->pid),
894
WTERMSIG(proc->status));
1176
} else if(WIFSIGNALED(proc->status)){
1177
fprintf(stderr, "Plugin %s [%" PRIdMAX "] killed by"
1178
" signal %d: %s\n", proc->name,
1179
(intmax_t) (proc->pid),
1180
WTERMSIG(proc->status),
1181
strsignal(WTERMSIG(proc->status)));
895
1182
} else if(WCOREDUMP(proc->status)){
896
fprintf(stderr, "Plugin %d dumped core\n",
897
(unsigned int) (proc->pid));
1183
fprintf(stderr, "Plugin %s [%" PRIdMAX "] dumped"
1184
" core\n", proc->name, (intmax_t) (proc->pid));
901
1188
/* Remove the plugin */
902
FD_CLR(proc->fd, &rfds_all);
1189
#if defined (__GNUC__) and defined (__GLIBC__)
1190
#if not __GLIBC_PREREQ(2, 16)
1191
#pragma GCC diagnostic push
1192
#pragma GCC diagnostic ignored "-Wsign-conversion"
1195
FD_CLR(proc->fd, &rfds_all); /* Spurious warning from
1196
-Wconversion in GNU libc
1198
#if defined (__GNUC__) and defined (__GLIBC__)
1199
#if not __GLIBC_PREREQ(2, 16)
1200
#pragma GCC diagnostic pop
904
1204
/* Block signal while modifying process_list */
905
ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
1205
ret = (int)TEMP_FAILURE_RETRY(sigprocmask
1207
&sigchld_action.sa_mask,
907
perror("sigprocmask");
908
exitstatus = EXIT_FAILURE;
1210
error(0, errno, "sigprocmask");
1211
exitstatus = EX_OSERR;
1215
plugin *next_plugin = proc->next;
911
1216
free_plugin(proc);
912
1219
/* We are done modifying process list, so unblock signal */
913
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,
1220
ret = (int)(TEMP_FAILURE_RETRY
1221
(sigprocmask(SIG_UNBLOCK,
1222
&sigchld_action.sa_mask, NULL)));
916
perror("sigprocmask");
917
exitstatus = EXIT_FAILURE;
1224
error(0, errno, "sigprocmask");
1225
exitstatus = EX_OSERR;
921
1229
if(plugin_list == NULL){
927
1236
/* This process exited nicely, so print its buffer */
929
1238
bool bret = print_out_password(proc->buffer,
930
1239
proc->buffer_length);
932
perror("print_out_password");
933
exitstatus = EXIT_FAILURE;
1241
error(0, errno, "print_out_password");
1242
exitstatus = EX_IOERR;
938
1247
/* This process has not completed. Does it have any output? */
939
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
1248
#if defined (__GNUC__) and defined (__GLIBC__)
1249
#if not __GLIBC_PREREQ(2, 16)
1250
#pragma GCC diagnostic push
1251
#pragma GCC diagnostic ignored "-Wsign-conversion"
1254
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious
1260
#if defined (__GNUC__) and defined (__GLIBC__)
1261
#if not __GLIBC_PREREQ(2, 16)
1262
#pragma GCC diagnostic pop
940
1265
/* This process had nothing to say at this time */
943
1269
/* Before reading, make the process' data buffer large enough */
944
1270
if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
945
proc->buffer = realloc(proc->buffer, proc->buffer_size
946
+ (size_t) BUFFER_SIZE);
947
if (proc->buffer == NULL){
949
exitstatus = EXIT_FAILURE;
1271
char *new_buffer = realloc(proc->buffer, proc->buffer_size
1272
+ (size_t) BUFFER_SIZE);
1273
if(new_buffer == NULL){
1274
error(0, errno, "malloc");
1275
exitstatus = EX_OSERR;
1278
proc->buffer = new_buffer;
952
1279
proc->buffer_size += BUFFER_SIZE;
954
1281
/* Read from the process */
955
ret = read(proc->fd, proc->buffer + proc->buffer_length,
1282
sret = TEMP_FAILURE_RETRY(read(proc->fd,
1284
+ proc->buffer_length,
958
1287
/* Read error from this process; ignore the error */
963
1293
proc->eof = true;
965
proc->buffer_length += (size_t) ret;
1295
proc->buffer_length += (size_t) sret;
973
if(plugin_list == NULL or exitstatus != EXIT_SUCCESS){
1303
if(plugin_list == NULL or (exitstatus != EXIT_SUCCESS
1304
and exitstatus != EX_OK)){
974
1305
/* Fallback if all plugins failed, none are found or an error
977
1308
fprintf(stderr, "Going to fallback mode using getpass(3)\n");
978
1309
char *passwordbuffer = getpass("Password: ");
979
bret = print_out_password(passwordbuffer, strlen(passwordbuffer));
1310
size_t len = strlen(passwordbuffer);
1311
/* Strip trailing newline */
1312
if(len > 0 and passwordbuffer[len-1] == '\n'){
1313
passwordbuffer[len-1] = '\0'; /* not strictly necessary */
1316
bret = print_out_password(passwordbuffer, len);
981
perror("print_out_password");
982
exitstatus = EXIT_FAILURE;
1318
error(0, errno, "print_out_password");
1319
exitstatus = EX_IOERR;
986
1323
/* Restore old signal handler */
987
1324
ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
990
exitstatus = EXIT_FAILURE;
1326
error(0, errno, "sigaction");
1327
exitstatus = EX_OSERR;
993
1330
if(custom_argv != NULL){
994
1331
for(char **arg = custom_argv+1; *arg != NULL; arg++){