27
27
#include <stdlib.h>             /* malloc(), exit(), EXIT_FAILURE,
 
28
28
                                   EXIT_SUCCESS, realloc() */
 
29
29
#include <stdbool.h>            /* bool, true, false */
 
30
 
#include <stdio.h>              /* perror, popen(), fileno(),
 
31
 
                                   fprintf(), stderr, STDOUT_FILENO */
 
 
30
#include <stdio.h>              /* perror, fileno(), fprintf(),
 
 
31
                                   stderr, STDOUT_FILENO */
 
32
32
#include <sys/types.h>          /* DIR, opendir(), stat(), struct
 
33
33
                                   stat, waitpid(), WIFEXITED(),
 
34
34
                                   WEXITSTATUS(), wait(), pid_t,
 
 
170
179
/* Add to a plugin's environment */
 
171
 
static bool add_environment(plugin *p, const char *def){
 
 
180
static bool add_environment(plugin *p, const char *def, bool replace){
 
 
184
  /* namelen = length of name of environment variable */
 
 
185
  size_t namelen = (size_t)(strchrnul(def, '=') - def);
 
 
186
  /* Search for this environment variable */
 
 
187
  for(char **e = p->environ; *e != NULL; e++){
 
 
188
    if(strncmp(*e, def, namelen + 1) == 0){
 
 
189
      /* It already exists */
 
 
191
        char *new = realloc(*e, strlen(def) + 1);
 
175
201
  return add_to_char_array(def, &(p->environ), &(p->envc));
 
180
205
 * Based on the example in the GNU LibC manual chapter 13.13 "File
 
181
206
 * Descriptor Flags".
 
182
207
 * *Note File Descriptor Flags:(libc)Descriptor Flags.
 
184
 
static int set_cloexec_flag(int fd)
 
 
209
static int set_cloexec_flag(int fd){
 
186
210
  int ret = fcntl(fd, F_GETFD, 0);
 
187
211
  /* If reading the flags failed, return error indication now. */
 
 
192
216
  return fcntl(fd, F_SETFD, ret | FD_CLOEXEC);
 
195
 
process *process_list = NULL;
 
197
 
/* Mark a process as completed when it exits, and save its exit
 
 
220
/* Mark processes as completed when they exit, and save their exit
 
199
 
void handle_sigchld(__attribute__((unused)) int sig){
 
200
 
  process *proc = process_list;
 
202
 
  pid_t pid = wait(&status);
 
207
 
  while(proc != NULL and proc->pid != pid){
 
211
 
    /* Process not found in process list */
 
214
 
  proc->status = status;
 
215
 
  proc->completed = true;
 
 
222
static void handle_sigchld(__attribute__((unused)) int sig){
 
 
224
    plugin *proc = plugin_list;
 
 
226
    pid_t pid = waitpid(-1, &status, WNOHANG);
 
 
228
      /* Only still running child processes */
 
 
232
      if (errno != ECHILD){
 
 
235
      /* No child processes */
 
 
239
    /* A child exited, find it in process_list */
 
 
240
    while(proc != NULL and proc->pid != pid){
 
 
244
      /* Process not found in process list */
 
 
247
    proc->status = status;
 
 
248
    proc->completed = true;
 
218
 
bool print_out_password(const char *buffer, size_t length){
 
 
252
/* Prints out a password to stdout */
 
 
253
static bool print_out_password(const char *buffer, size_t length){
 
220
 
  if(length>0 and buffer[length-1] == '\n'){
 
223
255
  for(size_t written = 0; written < length; written += (size_t)ret){
 
224
256
    ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO, buffer + written,
 
225
257
                                   length - written));
 
 
233
 
char **add_to_argv(char **argv, int *argc, char *arg){
 
236
 
    argv = malloc(sizeof(char*) * 2);
 
 
265
/* Removes and free a plugin from the plugin list */
 
 
266
static void free_plugin(plugin *plugin_node){
 
 
268
  for(char **arg = plugin_node->argv; *arg != NULL; arg++){
 
 
271
  free(plugin_node->argv);
 
 
272
  for(char **env = plugin_node->environ; *env != NULL; env++){
 
 
275
  free(plugin_node->environ);
 
 
276
  free(plugin_node->buffer);
 
 
278
  /* Removes the plugin from the singly-linked list */
 
 
279
  if(plugin_node == plugin_list){
 
 
280
    /* First one - simple */
 
 
281
    plugin_list = plugin_list->next;
 
 
283
    /* Second one or later */
 
 
284
    for(plugin *p = plugin_list; p != NULL; p = p->next){
 
 
285
      if(p->next == plugin_node){
 
 
286
        p->next = plugin_node->next;
 
240
 
    argv[0] = NULL;     /* Will be set to argv[0] in main before parsing */
 
244
 
  argv = realloc(argv, sizeof(char *)
 
245
 
                  * ((unsigned int) *argc + 1));
 
254
 
static void free_plugin_list(plugin *plugin_list){
 
255
 
  for(plugin *next = plugin_list; plugin_list != NULL; plugin_list = next){
 
256
 
    next = plugin_list->next;
 
257
 
    free(plugin_list->name);
 
258
 
    for(char **arg = plugin_list->argv; *arg != NULL; arg++){
 
261
 
    free(plugin_list->argv);
 
262
 
    for(char **env = plugin_list->environ; *env != NULL; env++){
 
265
 
    free(plugin_list->environ);
 
 
295
static void free_plugin_list(void){
 
 
296
  while(plugin_list != NULL){
 
 
297
    free_plugin(plugin_list);
 
270
301
int main(int argc, char *argv[]){
 
271
 
  const char *plugindir = "/lib/mandos/plugins.d";
 
272
 
  const char *argfile = ARGFILE;
 
 
302
  char *plugindir = NULL;
 
 
303
  char *argfile = NULL;
 
274
305
  size_t d_name_len;
 
 
307
338
    { .name = "global-options", .key = 'g',
 
308
339
      .arg = "OPTION[,OPTION[,...]]",
 
309
340
      .doc = "Options passed to all plugins" },
 
310
 
    { .name = "global-envs", .key = 'e',
 
 
341
    { .name = "global-env", .key = 'G',
 
311
342
      .arg = "VAR=value",
 
312
343
      .doc = "Environment variable passed to all plugins" },
 
313
344
    { .name = "options-for", .key = 'o',
 
314
345
      .arg = "PLUGIN:OPTION[,OPTION[,...]]",
 
315
346
      .doc = "Options passed only to specified plugin" },
 
316
 
    { .name = "envs-for", .key = 'f',
 
 
347
    { .name = "env-for", .key = 'E',
 
317
348
      .arg = "PLUGIN:ENV=value",
 
318
349
      .doc = "Environment variable passed to specified plugin" },
 
319
350
    { .name = "disable", .key = 'd',
 
321
352
      .doc = "Disable a specific plugin", .group = 1 },
 
 
353
    { .name = "enable", .key = 'e',
 
 
355
      .doc = "Enable a specific plugin", .group = 1 },
 
322
356
    { .name = "plugin-dir", .key = 128,
 
323
357
      .arg = "DIRECTORY",
 
324
358
      .doc = "Specify a different plugin directory", .group = 2 },
 
325
 
    { .name = "userid", .key = 129,
 
326
 
      .arg = "ID", .flags = 0,
 
327
 
      .doc = "User ID the plugins will run as", .group = 2 },
 
328
 
    { .name = "groupid", .key = 130,
 
329
 
      .arg = "ID", .flags = 0,
 
330
 
      .doc = "Group ID the plugins will run as", .group = 2 },
 
331
 
    { .name = "debug", .key = 131,
 
332
 
      .doc = "Debug mode", .group = 3 },
 
 
359
    { .name = "config-file", .key = 129,
 
 
361
      .doc = "Specify a different configuration file", .group = 2 },
 
 
362
    { .name = "userid", .key = 130,
 
 
363
      .arg = "ID", .flags = 0,
 
 
364
      .doc = "User ID the plugins will run as", .group = 3 },
 
 
365
    { .name = "groupid", .key = 131,
 
 
366
      .arg = "ID", .flags = 0,
 
 
367
      .doc = "Group ID the plugins will run as", .group = 3 },
 
 
368
    { .name = "debug", .key = 132,
 
 
369
      .doc = "Debug mode", .group = 4 },
 
336
 
  error_t parse_opt (int key, char *arg, struct argp_state *state) {
 
337
 
    /* Get the INPUT argument from `argp_parse', which we know is a
 
338
 
       pointer to our plugin list pointer. */
 
339
 
    plugin **plugins = state->input;
 
 
373
  error_t parse_opt (int key, char *arg, __attribute__((unused))
 
 
374
                     struct argp_state *state) {
 
 
376
    case 'g':                   /* --global-options */
 
342
377
      if (arg != NULL){
 
344
379
        while((p = strsep(&arg, ",")) != NULL){
 
345
380
          if(p[0] == '\0'){
 
348
 
          if(not add_argument(getplugin(NULL, plugins), p)){
 
 
383
          if(not add_argument(getplugin(NULL), p)){
 
349
384
            perror("add_argument");
 
350
385
            return ARGP_ERR_UNKNOWN;
 
 
390
    case 'G':                   /* --global-env */
 
360
 
        char *envdef = strdup(arg);
 
364
 
        if(not add_environment(getplugin(NULL, plugins), envdef)){
 
365
 
          perror("add_environment");
 
 
394
      if(not add_environment(getplugin(NULL), arg, true)){
 
 
395
        perror("add_environment");
 
 
398
    case 'o':                   /* --options-for */
 
370
399
      if (arg != NULL){
 
371
400
        char *p_name = strsep(&arg, ":");
 
372
 
        if(p_name[0] == '\0'){
 
 
401
        if(p_name[0] == '\0' or arg == NULL){
 
375
404
        char *opt = strsep(&arg, ":");
 
 
405
        if(opt[0] == '\0' or opt == NULL){
 
381
 
          while((p = strsep(&opt, ",")) != NULL){
 
385
 
            if(not add_argument(getplugin(p_name, plugins), p)){
 
386
 
              perror("add_argument");
 
387
 
              return ARGP_ERR_UNKNOWN;
 
 
409
        while((p = strsep(&opt, ",")) != NULL){
 
 
413
          if(not add_argument(getplugin(p_name), p)){
 
 
414
            perror("add_argument");
 
 
415
            return ARGP_ERR_UNKNOWN;
 
 
420
    case 'E':                   /* --env-for */
 
 
399
426
        if(envdef == NULL){
 
402
 
        char *p_name = strndup(arg, (size_t) (envdef-arg));
 
407
 
        if(not add_environment(getplugin(p_name, plugins), envdef)){
 
 
430
        if(not add_environment(getplugin(arg), envdef+1, true)){
 
408
431
          perror("add_environment");
 
 
435
    case 'd':                   /* --disable */
 
413
436
      if (arg != NULL){
 
414
 
        plugin *p = getplugin(arg, plugins);
 
 
437
        plugin *p = getplugin(arg);
 
416
439
          return ARGP_ERR_UNKNOWN;
 
418
441
        p->disabled = true;
 
 
444
    case 'e':                   /* --enable */
 
 
446
        plugin *p = getplugin(arg);
 
 
448
          return ARGP_ERR_UNKNOWN;
 
 
453
    case 128:                   /* --plugin-dir */
 
 
455
      plugindir = strdup(arg);
 
 
456
      if(plugindir == NULL){
 
 
460
    case 129:                   /* --config-file */
 
 
461
      /* This is already done by parse_opt_config_file() */
 
 
463
    case 130:                   /* --userid */
 
425
464
      uid = (uid_t)strtol(arg, NULL, 10);
 
 
466
    case 131:                   /* --groupid */
 
428
467
      gid = (gid_t)strtol(arg, NULL, 10);
 
 
469
    case 132:                   /* --debug */
 
433
472
    case ARGP_KEY_ARG:
 
434
 
      fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
 
439
 
      return ARGP_ERR_UNKNOWN;
 
444
 
  plugin *plugin_list = NULL;
 
446
 
  struct argp argp = { .options = options, .parser = parse_opt,
 
447
 
                       .args_doc = "[+PLUS_SEPARATED_OPTIONS]",
 
 
473
      /* Cryptsetup always passes an argument, which is an empty
 
 
474
         string if "none" was specified in /etc/crypttab.  So if
 
 
475
         argument was empty, we ignore it silently. */
 
 
477
        fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
 
 
483
      return ARGP_ERR_UNKNOWN;
 
 
488
  /* This option parser is the same as parse_opt() above, except it
 
 
489
     ignores everything but the --config-file option. */
 
 
490
  error_t parse_opt_config_file (int key, char *arg,
 
 
491
                                 __attribute__((unused))
 
 
492
                                 struct argp_state *state) {
 
 
494
    case 'g':                   /* --global-options */
 
 
495
    case 'G':                   /* --global-env */
 
 
496
    case 'o':                   /* --options-for */
 
 
497
    case 'E':                   /* --env-for */
 
 
498
    case 'd':                   /* --disable */
 
 
499
    case 'e':                   /* --enable */
 
 
500
    case 128:                   /* --plugin-dir */
 
 
502
    case 129:                   /* --config-file */
 
 
504
      argfile = strdup(arg);
 
 
509
    case 130:                   /* --userid */
 
 
510
    case 131:                   /* --groupid */
 
 
511
    case 132:                   /* --debug */
 
 
516
      return ARGP_ERR_UNKNOWN;
 
 
521
  struct argp argp = { .options = options,
 
 
522
                       .parser = parse_opt_config_file,
 
448
524
                       .doc = "Mandos plugin runner -- Run plugins" };
 
450
 
  ret = argp_parse (&argp, argc, argv, 0, 0, &plugin_list);
 
 
526
  /* Parse using the parse_opt_config_file in order to get the custom
 
 
527
     config file location, if any. */
 
 
528
  ret = argp_parse (&argp, argc, argv, ARGP_IN_ORDER, 0, NULL);
 
451
529
  if (ret == ARGP_ERR_UNKNOWN){
 
452
530
    fprintf(stderr, "Unknown error while parsing arguments\n");
 
453
531
    exitstatus = EXIT_FAILURE;
 
457
 
  conffp = fopen(argfile, "r");
 
 
535
  /* Reset to the normal argument parser */
 
 
536
  argp.parser = parse_opt;
 
 
538
  /* Open the configfile if available */
 
 
539
  if (argfile == NULL){
 
 
540
    conffp = fopen(AFILE, "r");
 
 
542
    conffp = fopen(argfile, "r");
 
458
544
  if(conffp != NULL){
 
459
545
    char *org_line = NULL;
 
460
546
    char *p, *arg, *new_arg, *line;
 
 
827
956
                      (unsigned int) (proc->pid));
 
830
960
          /* Remove the plugin */
 
831
961
          FD_CLR(proc->fd, &rfds_all);
 
832
963
          /* Block signal while modifying process_list */
 
833
 
          ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
 
 
964
          ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
 
835
966
            perror("sigprocmask");
 
836
967
            exitstatus = EXIT_FAILURE;
 
839
 
          /* Delete this process entry from the list */
 
840
 
          if(process_list == proc){
 
841
 
            /* First one - simple */
 
842
 
            process_list = proc->next;
 
844
 
            /* Second one or later */
 
845
 
            for(process *p = process_list; p != NULL; p = p->next){
 
847
 
                p->next = proc->next;
 
 
971
          plugin *next_plugin = proc->next;
 
852
975
          /* We are done modifying process list, so unblock signal */
 
853
976
          ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,
 
856
979
            perror("sigprocmask");
 
860
 
          /* We deleted this process from the list, so we can't go
 
861
 
             proc->next.  Therefore, start over from the beginning of
 
 
980
            exitstatus = EXIT_FAILURE;
 
 
984
          if(plugin_list == NULL){
 
865
991
        /* This process exited nicely, so print its buffer */
 
867
 
        bool bret = print_out_password(proc->buffer, proc->buffer_length);
 
 
993
        bool bret = print_out_password(proc->buffer,
 
 
994
                                       proc->buffer_length);
 
869
996
          perror("print_out_password");
 
870
997
          exitstatus = EXIT_FAILURE;
 
874
1002
      /* This process has not completed.  Does it have any output? */
 
875
1003
      if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
 
876
1004
        /* This process had nothing to say at this time */
 
879
1008
      /* Before reading, make the process' data buffer large enough */
 
 
909
 
  if(process_list == NULL or exitstatus != EXIT_SUCCESS){
 
910
 
    /* Fallback if all plugins failed, none are found or an error occured */
 
 
1039
  if(plugin_list == NULL or exitstatus != EXIT_SUCCESS){
 
 
1040
    /* Fallback if all plugins failed, none are found or an error
 
912
1043
    fprintf(stderr, "Going to fallback mode using getpass(3)\n");
 
913
1044
    char *passwordbuffer = getpass("Password: ");
 
914
 
    bret = print_out_password(passwordbuffer, strlen(passwordbuffer));
 
 
1045
    size_t len = strlen(passwordbuffer);
 
 
1046
    /* Strip trailing newline */
 
 
1047
    if(len > 0 and passwordbuffer[len-1] == '\n'){
 
 
1048
      passwordbuffer[len-1] = '\0'; /* not strictly necessary */
 
 
1051
    bret = print_out_password(passwordbuffer, len);
 
916
1053
      perror("print_out_password");
 
917
1054
      exitstatus = EXIT_FAILURE;
 
 
924
1061
    perror("sigaction");
 
925
1062
    exitstatus = EXIT_FAILURE;
 
928
1065
  if(custom_argv != NULL){
 
929
 
    for(char **arg = custom_argv; *arg != NULL; arg++){
 
 
1066
    for(char **arg = custom_argv+1; *arg != NULL; arg++){
 
932
1069
    free(custom_argv);
 
934
 
  free_plugin_list(plugin_list);
 
936
1072
  if(dir != NULL){
 
940
 
  /* Free the process list and kill the processes */
 
941
 
  for(process *next; process_list != NULL; process_list = next){
 
942
 
    next = process_list->next;
 
943
 
    close(process_list->fd);
 
944
 
    ret = kill(process_list->pid, SIGTERM);
 
945
 
    if(ret == -1 and errno != ESRCH){
 
946
 
      /* set-uid proccesses migth not get closed */
 
 
1076
  /* Kill the processes */
 
 
1077
  for(plugin *p = plugin_list; p != NULL; p = p->next){
 
 
1080
      ret = kill(p->pid, SIGTERM);
 
 
1081
      if(ret == -1 and errno != ESRCH){
 
 
1082
        /* Set-uid proccesses might not get closed */
 
949
 
    free(process_list->buffer);
 
953
1088
  /* Wait for any remaining child processes to terminate */