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,
69
70
#define PDIR "/lib/mandos/plugins.d"
70
71
#define AFILE "/conf/conf.d/mandos/plugin-runner.conf"
72
const char *argp_program_version = "plugin-runner 1.0";
73
const char *argp_program_version = "plugin-runner " VERSION;
73
74
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
77
76
typedef struct plugin{
78
77
char *name; /* can be NULL or any plugin name */
97
96
static plugin *plugin_list = NULL;
99
/* Gets a existing plugin based on name,
98
/* Gets an existing plugin based on name,
100
99
or if none is found, creates a new one */
101
100
static plugin *getplugin(char *name){
102
101
/* 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))){
102
for(plugin *p = plugin_list; p != NULL; p = p->next){
104
or (p->name and name and (strcmp(p->name, name) == 0))){
109
108
/* Create a new plugin */
110
109
plugin *new_plugin = malloc(sizeof(plugin));
111
if (new_plugin == NULL){
110
if(new_plugin == NULL){
114
113
char *copy_name = NULL;
187
186
size_t namelen = (size_t)(strchrnul(def, '=') - def);
188
187
/* Search for this environment variable */
189
188
for(char **e = p->environ; *e != NULL; e++){
190
if(strncmp(*e, def, namelen+1) == 0){
191
/* Refuse to add an existing variable */
189
if(strncmp(*e, def, namelen + 1) == 0){
190
/* It already exists */
192
char *new = realloc(*e, strlen(def) + 1);
200
207
* Descriptor Flags".
201
208
* *Note File Descriptor Flags:(libc)Descriptor Flags.
203
static int set_cloexec_flag(int fd)
210
static int set_cloexec_flag(int fd){
205
211
int ret = fcntl(fd, F_GETFD, 0);
206
212
/* If reading the flags failed, return error indication now. */
247
253
/* Prints out a password to stdout */
248
bool print_out_password(const char *buffer, size_t length){
254
static bool print_out_password(const char *buffer, size_t length){
250
if(length>0 and buffer[length-1] == '\n'){
253
256
for(size_t written = 0; written < length; written += (size_t)ret){
254
257
ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO, buffer + written,
255
258
length - written));
336
340
{ .name = "global-options", .key = 'g',
337
341
.arg = "OPTION[,OPTION[,...]]",
338
342
.doc = "Options passed to all plugins" },
339
{ .name = "global-env", .key = 'e',
343
{ .name = "global-env", .key = 'G',
340
344
.arg = "VAR=value",
341
345
.doc = "Environment variable passed to all plugins" },
342
346
{ .name = "options-for", .key = 'o',
343
347
.arg = "PLUGIN:OPTION[,OPTION[,...]]",
344
348
.doc = "Options passed only to specified plugin" },
345
{ .name = "env-for", .key = 'f',
349
{ .name = "env-for", .key = 'E',
346
350
.arg = "PLUGIN:ENV=value",
347
351
.doc = "Environment variable passed to specified plugin" },
348
352
{ .name = "disable", .key = 'd',
350
354
.doc = "Disable a specific plugin", .group = 1 },
355
{ .name = "enable", .key = 'e',
357
.doc = "Enable a specific plugin", .group = 1 },
351
358
{ .name = "plugin-dir", .key = 128,
352
359
.arg = "DIRECTORY",
353
360
.doc = "Specify a different plugin directory", .group = 2 },
368
error_t parse_opt (int key, char *arg, __attribute__((unused))
369
struct argp_state *state) {
370
/* Get the INPUT argument from `argp_parse', which we know is a
371
pointer to our plugin list pointer. */
375
error_t parse_opt(int key, char *arg, __attribute__((unused))
376
struct argp_state *state) {
373
378
case 'g': /* --global-options */
376
381
while((p = strsep(&arg, ",")) != NULL){
377
382
if(p[0] == '\0'){
387
case 'e': /* --global-env */
392
case 'G': /* --global-env */
392
char *envdef = strdup(arg);
396
if(not add_environment(getplugin(NULL), envdef)){
397
perror("add_environment");
396
if(not add_environment(getplugin(NULL), arg, true)){
397
perror("add_environment");
401
400
case 'o': /* --options-for */
403
402
char *p_name = strsep(&arg, ":");
404
if(p_name[0] == '\0'){
403
if(p_name[0] == '\0' or arg == NULL){
407
406
char *opt = strsep(&arg, ":");
407
if(opt[0] == '\0' or opt == NULL){
413
while((p = strsep(&opt, ",")) != NULL){
417
if(not add_argument(getplugin(p_name), p)){
418
perror("add_argument");
419
return ARGP_ERR_UNKNOWN;
411
while((p = strsep(&opt, ",")) != NULL){
415
if(not add_argument(getplugin(p_name), p)){
416
perror("add_argument");
417
return ARGP_ERR_UNKNOWN;
425
case 'f': /* --env-for */
422
case 'E': /* --env-for */
431
428
if(envdef == NULL){
434
char *p_name = strndup(arg, (size_t) (envdef-arg));
439
if(not add_environment(getplugin(p_name), envdef)){
432
if(not add_environment(getplugin(arg), envdef+1, true)){
440
433
perror("add_environment");
444
437
case 'd': /* --disable */
446
439
plugin *p = getplugin(arg);
448
441
return ARGP_ERR_UNKNOWN;
450
443
p->disabled = true;
446
case 'e': /* --enable */
448
plugin *p = getplugin(arg);
450
return ARGP_ERR_UNKNOWN;
453
455
case 128: /* --plugin-dir */
454
457
plugindir = strdup(arg);
455
458
if(plugindir == NULL){
456
459
perror("strdup");
459
462
case 129: /* --config-file */
463
/* This is already done by parse_opt_config_file() */
465
case 130: /* --userid */
466
/* In the GNU C library, uid_t is always unsigned int */
467
ret = sscanf(arg, "%u", &uid);
469
fprintf(stderr, "Bad user ID number: \"%s\", using %u\n", arg,
473
case 131: /* --groupid */
474
/* In the GNU C library, gid_t is always unsigned int */
475
ret = sscanf(arg, "%u", &gid);
477
fprintf(stderr, "Bad group ID number: \"%s\", using %u\n",
481
case 132: /* --debug */
485
* When adding more options before this line, remember to also add a
486
* "case" to the "parse_opt_config_file" function below.
489
/* Cryptsetup always passes an argument, which is an empty
490
string if "none" was specified in /etc/crypttab. So if
491
argument was empty, we ignore it silently. */
493
fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
499
return ARGP_ERR_UNKNOWN;
504
/* This option parser is the same as parse_opt() above, except it
505
ignores everything but the --config-file option. */
506
error_t parse_opt_config_file(int key, char *arg,
507
__attribute__((unused))
508
struct argp_state *state) {
510
case 'g': /* --global-options */
511
case 'G': /* --global-env */
512
case 'o': /* --options-for */
513
case 'E': /* --env-for */
514
case 'd': /* --disable */
515
case 'e': /* --enable */
516
case 128: /* --plugin-dir */
518
case 129: /* --config-file */
460
520
argfile = strdup(arg);
461
521
if(argfile == NULL){
462
522
perror("strdup");
465
525
case 130: /* --userid */
466
uid = (uid_t)strtol(arg, NULL, 10);
468
526
case 131: /* --groupid */
469
gid = (gid_t)strtol(arg, NULL, 10);
471
527
case 132: /* --debug */
474
528
case ARGP_KEY_ARG:
475
fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
477
529
case ARGP_KEY_END:
485
struct argp argp = { .options = options, .parser = parse_opt,
486
.args_doc = "[+PLUS_SEPARATED_OPTIONS]",
537
struct argp argp = { .options = options,
538
.parser = parse_opt_config_file,
487
540
.doc = "Mandos plugin runner -- Run plugins" };
489
ret = argp_parse (&argp, argc, argv, 0, 0, NULL);
490
if (ret == ARGP_ERR_UNKNOWN){
542
/* Parse using parse_opt_config_file() in order to get the custom
543
config file location, if any. */
544
ret = argp_parse(&argp, argc, argv, ARGP_IN_ORDER, 0, NULL);
545
if(ret == ARGP_ERR_UNKNOWN){
491
546
fprintf(stderr, "Unknown error while parsing arguments\n");
492
547
exitstatus = EXIT_FAILURE;
496
/* Opens the configfile if aviable */
497
if (argfile == NULL){
551
/* Reset to the normal argument parser */
552
argp.parser = parse_opt;
554
/* Open the configfile if available */
498
556
conffp = fopen(AFILE, "r");
500
558
conffp = fopen(argfile, "r");
565
622
/* If there was any arguments from configuration file,
566
623
pass them to parser as command arguments */
567
624
if(custom_argv != NULL){
568
ret = argp_parse (&argp, custom_argc, custom_argv, 0, 0, NULL);
569
if (ret == ARGP_ERR_UNKNOWN){
625
ret = argp_parse(&argp, custom_argc, custom_argv, ARGP_IN_ORDER,
627
if(ret == ARGP_ERR_UNKNOWN){
570
628
fprintf(stderr, "Unknown error while parsing arguments\n");
571
629
exitstatus = EXIT_FAILURE;
634
/* Parse actual command line arguments, to let them override the
636
ret = argp_parse(&argp, argc, argv, ARGP_IN_ORDER, 0, NULL);
637
if(ret == ARGP_ERR_UNKNOWN){
638
fprintf(stderr, "Unknown error while parsing arguments\n");
639
exitstatus = EXIT_FAILURE;
577
644
for(plugin *p = plugin_list; p != NULL; p=p->next){
578
645
fprintf(stderr, "Plugin: %s has %d arguments\n",
687
ret = asprintf(&filename, "%s/%s", plugindir, dirst->d_name);
755
if(plugindir == NULL){
756
ret = asprintf(&filename, PDIR "/%s", dirst->d_name);
758
ret = asprintf(&filename, "%s/%s", plugindir, dirst->d_name);
689
761
perror("asprintf");
693
765
ret = stat(filename, &st);
700
772
/* Ignore non-executable files */
701
if (not S_ISREG(st.st_mode) or (access(filename, X_OK) != 0)){
773
if(not S_ISREG(st.st_mode) or (access(filename, X_OK) != 0)){
703
775
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
704
776
" with bad type or mode\n", filename);
733
805
/* Add global environment variables */
734
806
for(char **e = g->environ; *e != NULL; e++){
735
if(not add_environment(p, *e)){
807
if(not add_environment(p, *e, false)){
736
808
perror("add_environment");
744
816
if(p->environ[0] != NULL){
745
817
for(char **e = environ; *e != NULL; e++){
746
char *copy = strdup(*e);
751
if(not add_environment(p, copy)){
818
if(not add_environment(p, *e, false)){
752
819
perror("add_environment");
777
844
/* Block SIGCHLD until process is safely in process list */
778
ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
845
ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
780
847
perror("sigprocmask");
781
848
exitstatus = EXIT_FAILURE;
784
// Starting a new process to be watched
851
/* Starting a new process to be watched */
785
852
pid_t pid = fork();
795
862
perror("sigaction");
796
863
_exit(EXIT_FAILURE);
798
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
865
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
800
867
perror("sigprocmask");
801
868
_exit(EXIT_FAILURE);
804
871
ret = dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */
829
896
close(pipefd[1]); /* Close unused write end of pipe */
831
898
plugin *new_plugin = getplugin(dirst->d_name);
832
if (new_plugin == NULL){
899
if(new_plugin == NULL){
833
900
perror("getplugin");
834
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
901
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
836
903
perror("sigprocmask");
845
912
/* Unblock SIGCHLD so signal handler can be run if this process
846
913
has already completed */
847
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
914
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
849
916
perror("sigprocmask");
850
917
exitstatus = EXIT_FAILURE;
872
938
free_plugin_list();
876
942
/* Main loop while running plugins exist */
877
943
while(plugin_list){
878
944
fd_set rfds = rfds_all;
879
945
int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
880
if (select_ret == -1){
946
if(select_ret == -1){
881
947
perror("select");
882
948
exitstatus = EXIT_FAILURE;
885
951
/* OK, now either a process completed, or something can be read
886
952
from one of them */
887
for(plugin *proc = plugin_list; proc != NULL; proc = proc->next){
953
for(plugin *proc = plugin_list; proc != NULL;){
888
954
/* Is this process completely done? */
889
955
if(proc->eof and proc->completed){
890
956
/* Only accept the plugin output if it exited cleanly */
902
968
(unsigned int) (proc->pid),
903
969
WTERMSIG(proc->status));
904
970
} else if(WCOREDUMP(proc->status)){
905
fprintf(stderr, "Plugin %d dumped core\n",
971
fprintf(stderr, "Plugin %u dumped core\n",
906
972
(unsigned int) (proc->pid));
917
983
exitstatus = EXIT_FAILURE;
987
plugin *next_plugin = proc->next;
920
988
free_plugin(proc);
921
991
/* We are done modifying process list, so unblock signal */
922
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,
992
ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask,
925
995
perror("sigprocmask");
926
996
exitstatus = EXIT_FAILURE;
947
1018
/* This process has not completed. Does it have any output? */
948
1019
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
949
1020
/* This process had nothing to say at this time */
952
1024
/* Before reading, make the process' data buffer large enough */
953
1025
if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
954
1026
proc->buffer = realloc(proc->buffer, proc->buffer_size
955
1027
+ (size_t) BUFFER_SIZE);
956
if (proc->buffer == NULL){
1028
if(proc->buffer == NULL){
957
1029
perror("malloc");
958
1030
exitstatus = EXIT_FAILURE;
961
1033
proc->buffer_size += BUFFER_SIZE;
963
1035
/* Read from the process */
964
ret = read(proc->fd, proc->buffer + proc->buffer_length,
1036
sret = read(proc->fd, proc->buffer + proc->buffer_length,
967
1039
/* Read error from this process; ignore the error */
972
1045
proc->eof = true;
974
proc->buffer_length += (size_t) ret;
1047
proc->buffer_length += (size_t) sret;
986
1059
fprintf(stderr, "Going to fallback mode using getpass(3)\n");
987
1060
char *passwordbuffer = getpass("Password: ");
988
bret = print_out_password(passwordbuffer, strlen(passwordbuffer));
1061
size_t len = strlen(passwordbuffer);
1062
/* Strip trailing newline */
1063
if(len > 0 and passwordbuffer[len-1] == '\n'){
1064
passwordbuffer[len-1] = '\0'; /* not strictly necessary */
1067
bret = print_out_password(passwordbuffer, len);
990
1069
perror("print_out_password");
991
1070
exitstatus = EXIT_FAILURE;