3
3
* Mandos plugin runner - Run Mandos plugins
5
* Copyright © 2007-2008 Teddy Hogeborn & Björn Påhlsson
5
* Copyright © 2008,2009 Teddy Hogeborn
6
* Copyright © 2008,2009 Björn Påhlsson
7
8
* This program is free software: you can redistribute it and/or
8
9
* modify it under the terms of the GNU General Public License as
27
28
#include <stdlib.h> /* malloc(), exit(), EXIT_FAILURE,
28
29
EXIT_SUCCESS, realloc() */
29
30
#include <stdbool.h> /* bool, true, false */
30
#include <stdio.h> /* perror, popen(), fileno(),
31
fprintf(), stderr, STDOUT_FILENO */
31
#include <stdio.h> /* perror, fileno(), fprintf(),
32
stderr, STDOUT_FILENO */
32
33
#include <sys/types.h> /* DIR, opendir(), stat(), struct
33
34
stat, waitpid(), WIFEXITED(),
34
35
WEXITSTATUS(), wait(), pid_t,
46
47
fcntl(), setuid(), setgid(),
47
48
F_GETFD, F_SETFD, FD_CLOEXEC,
48
49
access(), pipe(), fork(), close()
49
dup2, STDOUT_FILENO, _exit(),
50
dup2(), STDOUT_FILENO, _exit(),
50
51
execv(), write(), read(),
52
53
#include <fcntl.h> /* fcntl(), F_GETFD, F_SETFD,
61
62
#include <signal.h> /* struct sigaction, sigemptyset(),
62
63
sigaddset(), sigaction(),
63
64
sigprocmask(), SIG_BLOCK, SIGCHLD,
64
SIG_UNBLOCK, kill() */
65
SIG_UNBLOCK, kill(), sig_atomic_t
65
67
#include <errno.h> /* errno, EBADF */
68
#include <inttypes.h> /* intmax_t, PRIdMAX, strtoimax() */
67
70
#define BUFFER_SIZE 256
69
72
#define PDIR "/lib/mandos/plugins.d"
70
73
#define AFILE "/conf/conf.d/mandos/plugin-runner.conf"
72
const char *argp_program_version = "plugin-runner 1.0";
75
const char *argp_program_version = "plugin-runner " VERSION;
73
76
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
77
78
typedef struct plugin{
78
79
char *name; /* can be NULL or any plugin name */
89
90
size_t buffer_size;
90
91
size_t buffer_length;
92
volatile bool completed;
93
volatile sig_atomic_t completed;
94
95
struct plugin *next;
97
98
static plugin *plugin_list = NULL;
99
/* Gets a existing plugin based on name,
100
/* Gets an existing plugin based on name,
100
101
or if none is found, creates a new one */
101
102
static plugin *getplugin(char *name){
102
103
/* 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))){
104
for(plugin *p = plugin_list; p != NULL; p = p->next){
106
or (p->name and name and (strcmp(p->name, name) == 0))){
109
110
/* Create a new plugin */
110
111
plugin *new_plugin = malloc(sizeof(plugin));
111
if (new_plugin == NULL){
112
if(new_plugin == NULL){
114
115
char *copy_name = NULL;
115
116
if(name != NULL){
116
117
copy_name = strdup(name);
117
118
if(copy_name == NULL){
122
*new_plugin = (plugin) { .name = copy_name,
125
.next = plugin_list };
124
*new_plugin = (plugin){ .name = copy_name,
127
.next = plugin_list };
127
129
new_plugin->argv = malloc(sizeof(char *) * 2);
128
if (new_plugin->argv == NULL){
130
if(new_plugin->argv == NULL){
130
132
free(new_plugin);
187
189
size_t namelen = (size_t)(strchrnul(def, '=') - def);
188
190
/* Search for this environment variable */
189
191
for(char **e = p->environ; *e != NULL; e++){
190
if(strncmp(*e, def, namelen+1) == 0){
192
if(strncmp(*e, def, namelen + 1) == 0){
191
193
/* It already exists */
193
char *new = realloc(*e, strlen(def));
195
char *new = realloc(*e, strlen(def) + 1);
208
210
* Descriptor Flags".
209
211
* *Note File Descriptor Flags:(libc)Descriptor Flags.
211
static int set_cloexec_flag(int fd)
213
static int set_cloexec_flag(int fd){
213
214
int ret = fcntl(fd, F_GETFD, 0);
214
215
/* If reading the flags failed, return error indication now. */
250
252
proc->status = status;
251
proc->completed = true;
255
258
/* Prints out a password to stdout */
256
bool print_out_password(const char *buffer, size_t length){
259
static bool print_out_password(const char *buffer, size_t length){
258
if(length>0 and buffer[length-1] == '\n'){
261
261
for(size_t written = 0; written < length; written += (size_t)ret){
262
262
ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO, buffer + written,
263
263
length - written));
344
346
{ .name = "global-options", .key = 'g',
345
347
.arg = "OPTION[,OPTION[,...]]",
346
348
.doc = "Options passed to all plugins" },
347
{ .name = "global-env", .key = 'e',
349
{ .name = "global-env", .key = 'G',
348
350
.arg = "VAR=value",
349
351
.doc = "Environment variable passed to all plugins" },
350
352
{ .name = "options-for", .key = 'o',
351
353
.arg = "PLUGIN:OPTION[,OPTION[,...]]",
352
354
.doc = "Options passed only to specified plugin" },
353
{ .name = "env-for", .key = 'f',
355
{ .name = "env-for", .key = 'E',
354
356
.arg = "PLUGIN:ENV=value",
355
357
.doc = "Environment variable passed to specified plugin" },
356
358
{ .name = "disable", .key = 'd',
358
360
.doc = "Disable a specific plugin", .group = 1 },
361
{ .name = "enable", .key = 'e',
363
.doc = "Enable a specific plugin", .group = 1 },
359
364
{ .name = "plugin-dir", .key = 128,
360
365
.arg = "DIRECTORY",
361
366
.doc = "Specify a different plugin directory", .group = 2 },
376
error_t parse_opt (int key, char *arg, __attribute__((unused))
377
struct argp_state *state) {
378
/* Get the INPUT argument from `argp_parse', which we know is a
379
pointer to our plugin list pointer. */
381
error_t parse_opt(int key, char *arg, __attribute__((unused))
382
struct argp_state *state){
381
385
case 'g': /* --global-options */
384
while((p = strsep(&arg, ",")) != NULL){
388
while((plugin_option = strsep(&arg, ",")) != NULL){
389
if(plugin_option[0] == '\0'){
388
if(not add_argument(getplugin(NULL), p)){
392
if(not add_argument(getplugin(NULL), plugin_option)){
389
393
perror("add_argument");
390
394
return ARGP_ERR_UNKNOWN;
395
case 'e': /* --global-env */
399
case 'G': /* --global-env */
400
char *envdef = strdup(arg);
404
if(not add_environment(getplugin(NULL), envdef, true)){
405
perror("add_environment");
403
if(not add_environment(getplugin(NULL), arg, true)){
404
perror("add_environment");
409
407
case 'o': /* --options-for */
411
char *p_name = strsep(&arg, ":");
412
if(p_name[0] == '\0'){
415
char *opt = strsep(&arg, ":");
421
while((p = strsep(&opt, ",")) != NULL){
425
if(not add_argument(getplugin(p_name), p)){
426
perror("add_argument");
427
return ARGP_ERR_UNKNOWN;
409
char *plugin_name = strsep(&arg, ":");
410
if(plugin_name[0] == '\0'){
414
while((plugin_option = strsep(&arg, ",")) != NULL){
415
if(not add_argument(getplugin(plugin_name), plugin_option)){
416
perror("add_argument");
417
return ARGP_ERR_UNKNOWN;
433
case 'f': /* --env-for */
422
case 'E': /* --env-for */
439
428
if(envdef == NULL){
442
char *p_name = strndup(arg, (size_t) (envdef-arg));
447
if(not add_environment(getplugin(p_name), envdef, true)){
432
if(not add_environment(getplugin(arg), envdef+1, true)){
448
433
perror("add_environment");
452
437
case 'd': /* --disable */
454
439
plugin *p = getplugin(arg);
456
441
return ARGP_ERR_UNKNOWN;
458
443
p->disabled = true;
446
case 'e': /* --enable */
448
plugin *p = getplugin(arg);
450
return ARGP_ERR_UNKNOWN;
461
455
case 128: /* --plugin-dir */
462
457
plugindir = strdup(arg);
463
458
if(plugindir == NULL){
464
459
perror("strdup");
467
462
case 129: /* --config-file */
463
/* This is already done by parse_opt_config_file() */
465
case 130: /* --userid */
467
tmpmax = strtoimax(arg, &tmp, 10);
468
if(errno != 0 or tmp == arg or *tmp != '\0'
469
or tmpmax != (uid_t)tmpmax){
470
fprintf(stderr, "Bad user ID number: \"%s\", using %"
471
PRIdMAX "\n", arg, (intmax_t)uid);
476
case 131: /* --groupid */
478
tmpmax = strtoimax(arg, &tmp, 10);
479
if(errno != 0 or tmp == arg or *tmp != '\0'
480
or tmpmax != (gid_t)tmpmax){
481
fprintf(stderr, "Bad group ID number: \"%s\", using %"
482
PRIdMAX "\n", arg, (intmax_t)gid);
487
case 132: /* --debug */
491
* When adding more options before this line, remember to also add a
492
* "case" to the "parse_opt_config_file" function below.
495
/* Cryptsetup always passes an argument, which is an empty
496
string if "none" was specified in /etc/crypttab. So if
497
argument was empty, we ignore it silently. */
499
fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
505
return ARGP_ERR_UNKNOWN;
510
/* This option parser is the same as parse_opt() above, except it
511
ignores everything but the --config-file option. */
512
error_t parse_opt_config_file(int key, char *arg,
513
__attribute__((unused))
514
struct argp_state *state){
516
case 'g': /* --global-options */
517
case 'G': /* --global-env */
518
case 'o': /* --options-for */
519
case 'E': /* --env-for */
520
case 'd': /* --disable */
521
case 'e': /* --enable */
522
case 128: /* --plugin-dir */
524
case 129: /* --config-file */
468
526
argfile = strdup(arg);
469
527
if(argfile == NULL){
470
528
perror("strdup");
473
531
case 130: /* --userid */
474
uid = (uid_t)strtol(arg, NULL, 10);
476
532
case 131: /* --groupid */
477
gid = (gid_t)strtol(arg, NULL, 10);
479
533
case 132: /* --debug */
482
534
case ARGP_KEY_ARG:
483
fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
485
535
case ARGP_KEY_END:
493
struct argp argp = { .options = options, .parser = parse_opt,
543
struct argp argp = { .options = options,
544
.parser = parse_opt_config_file,
495
546
.doc = "Mandos plugin runner -- Run plugins" };
548
/* Parse using parse_opt_config_file() in order to get the custom
549
config file location, if any. */
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;
557
/* Reset to the normal argument parser */
558
argp.parser = parse_opt;
497
560
/* Open the configfile if available */
498
if (argfile == NULL){
499
562
conffp = fopen(AFILE, "r");
501
564
conffp = fopen(argfile, "r");
566
628
/* If there was any arguments from configuration file,
567
629
pass them to parser as command arguments */
568
630
if(custom_argv != NULL){
569
ret = argp_parse (&argp, custom_argc, custom_argv, ARGP_IN_ORDER,
571
if (ret == ARGP_ERR_UNKNOWN){
631
ret = argp_parse(&argp, custom_argc, custom_argv, ARGP_IN_ORDER,
633
if(ret == ARGP_ERR_UNKNOWN){
572
634
fprintf(stderr, "Unknown error while parsing arguments\n");
573
635
exitstatus = EXIT_FAILURE;
578
640
/* Parse actual command line arguments, to let them override the
580
ret = argp_parse (&argp, argc, argv, ARGP_IN_ORDER, 0, NULL);
581
if (ret == ARGP_ERR_UNKNOWN){
642
ret = argp_parse(&argp, argc, argv, ARGP_IN_ORDER, 0, NULL);
643
if(ret == ARGP_ERR_UNKNOWN){
582
644
fprintf(stderr, "Unknown error while parsing arguments\n");
583
645
exitstatus = EXIT_FAILURE;
591
653
for(char **a = p->argv; *a != NULL; a++){
592
654
fprintf(stderr, "\tArg: %s\n", *a);
594
fprintf(stderr, "...and %u environment variables\n", p->envc);
656
fprintf(stderr, "...and %d environment variables\n", p->envc);
595
657
for(char **a = p->environ; *a != NULL; a++){
596
658
fprintf(stderr, "\t%s\n", *a);
698
ret = asprintf(&filename, "%s/%s", plugindir, dirst->d_name);
761
if(plugindir == NULL){
762
ret = asprintf(&filename, PDIR "/%s", dirst->d_name);
764
ret = asprintf(&filename, "%s/%s", plugindir, dirst->d_name);
700
767
perror("asprintf");
704
771
ret = stat(filename, &st);
711
778
/* Ignore non-executable files */
712
if (not S_ISREG(st.st_mode) or (access(filename, X_OK) != 0)){
779
if(not S_ISREG(st.st_mode) or (access(filename, X_OK) != 0)){
714
781
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
715
782
" with bad type or mode\n", filename);
878
945
free_plugin_list();
882
949
/* Main loop while running plugins exist */
883
950
while(plugin_list){
884
951
fd_set rfds = rfds_all;
885
952
int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
886
if (select_ret == -1){
953
if(select_ret == -1){
887
954
perror("select");
888
955
exitstatus = EXIT_FAILURE;
891
958
/* OK, now either a process completed, or something can be read
892
959
from one of them */
893
for(plugin *proc = plugin_list; proc != NULL; proc = proc->next){
960
for(plugin *proc = plugin_list; proc != NULL;){
894
961
/* Is this process completely done? */
895
if(proc->eof and proc->completed){
962
if(proc->completed and proc->eof){
896
963
/* Only accept the plugin output if it exited cleanly */
897
964
if(not WIFEXITED(proc->status)
898
965
or WEXITSTATUS(proc->status) != 0){
899
966
/* Bad exit by plugin */
902
969
if(WIFEXITED(proc->status)){
903
fprintf(stderr, "Plugin %u exited with status %d\n",
904
(unsigned int) (proc->pid),
970
fprintf(stderr, "Plugin %s [%" PRIdMAX "] exited with"
971
" status %d\n", proc->name,
972
(intmax_t) (proc->pid),
905
973
WEXITSTATUS(proc->status));
906
} else if(WIFSIGNALED(proc->status)) {
907
fprintf(stderr, "Plugin %u killed by signal %d\n",
908
(unsigned int) (proc->pid),
974
} else if(WIFSIGNALED(proc->status)){
975
fprintf(stderr, "Plugin %s [%" PRIdMAX "] killed by"
976
" signal %d\n", proc->name,
977
(intmax_t) (proc->pid),
909
978
WTERMSIG(proc->status));
910
979
} else if(WCOREDUMP(proc->status)){
911
fprintf(stderr, "Plugin %d dumped core\n",
912
(unsigned int) (proc->pid));
980
fprintf(stderr, "Plugin %s [%" PRIdMAX "] dumped"
981
" core\n", proc->name, (intmax_t) (proc->pid));
916
985
/* Remove the plugin */
917
986
FD_CLR(proc->fd, &rfds_all);
919
988
/* Block signal while modifying process_list */
920
989
ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
923
992
exitstatus = EXIT_FAILURE;
996
plugin *next_plugin = proc->next;
926
997
free_plugin(proc);
927
1000
/* We are done modifying process list, so unblock signal */
928
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,
1001
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask,
931
1004
perror("sigprocmask");
932
1005
exitstatus = EXIT_FAILURE;
953
1027
/* This process has not completed. Does it have any output? */
954
1028
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
955
1029
/* This process had nothing to say at this time */
958
1033
/* Before reading, make the process' data buffer large enough */
959
1034
if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
960
1035
proc->buffer = realloc(proc->buffer, proc->buffer_size
961
1036
+ (size_t) BUFFER_SIZE);
962
if (proc->buffer == NULL){
1037
if(proc->buffer == NULL){
963
1038
perror("malloc");
964
1039
exitstatus = EXIT_FAILURE;
967
1042
proc->buffer_size += BUFFER_SIZE;
969
1044
/* Read from the process */
970
ret = read(proc->fd, proc->buffer + proc->buffer_length,
1045
sret = read(proc->fd, proc->buffer + proc->buffer_length,
973
1048
/* Read error from this process; ignore the error */
978
1054
proc->eof = true;
980
proc->buffer_length += (size_t) ret;
1056
proc->buffer_length += (size_t) sret;
988
1064
if(plugin_list == NULL or exitstatus != EXIT_SUCCESS){
992
1068
fprintf(stderr, "Going to fallback mode using getpass(3)\n");
993
1069
char *passwordbuffer = getpass("Password: ");
994
bret = print_out_password(passwordbuffer, strlen(passwordbuffer));
1070
size_t len = strlen(passwordbuffer);
1071
/* Strip trailing newline */
1072
if(len > 0 and passwordbuffer[len-1] == '\n'){
1073
passwordbuffer[len-1] = '\0'; /* not strictly necessary */
1076
bret = print_out_password(passwordbuffer, len);
996
1078
perror("print_out_password");
997
1079
exitstatus = EXIT_FAILURE;