/mandos/release

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/release

« back to all changes in this revision

Viewing changes to plugin-runner.c

  • Committer: Teddy Hogeborn
  • Date: 2008-08-29 05:53:59 UTC
  • Revision ID: teddy@fukt.bsnet.se-20080829055359-wkdasnyxtylmnxus
* mandos.xml (EXAMPLE): Replaced all occurences of command name with
                        "&COMMANDNAME;".

* plugins.d/password-prompt.c (main): Improved some documentation
                                      strings.  Do perror() of
                                      tcgetattr() fails.  Add debug
                                      output if interrupted by signal.
                                      Loop over write() instead of
                                      using fwrite() when outputting
                                      password.  Add debug output if
                                      getline() returns 0, unless it
                                      was caused by a signal.  Add
                                      exit status code to debug
                                      output.

* plugins.d/password-prompt.xml: Changed all single quotes to double
                                 quotes for consistency.  Removed
                                 <?xml-stylesheet>.
  (ENTITY TIMESTAMP): New.  Automatically updated by Emacs time-stamp
                      by using Emacs local variables.
  (/refentry/refentryinfo/title): Changed to "Mandos Manual".
  (/refentry/refentryinfo/productname): Changed to "Mandos".
  (/refentry/refentryinfo/date): New; set to "&TIMESTAMP;".
  (/refentry/refentryinfo/copyright): Split copyright holders.
  (/refentry/refnamediv/refpurpose): Improved wording.
  (SYNOPSIS): Fix to use correct markup.  Add short options.
  (DESCRIPTION, OPTIONS): Improved wording.
  (OPTIONS): Improved wording.  Use more correct markup.  Document
             short options.
  (EXIT STATUS): Add text.
  (ENVIRONMENT): Document use of "cryptsource" and "crypttarget".
  (FILES): REMOVED.
  (BUGS): Add text.
  (EXAMPLE): Added some examples.
  (SECURITY): Added text.
  (SEE ALSO): Remove reference to mandos(8).  Add reference to
              crypttab(5).

Show diffs side-by-side

added added

removed removed

Lines of Context:
21
21
 * Contact the authors at <mandos@fukt.bsnet.se>.
22
22
 */
23
23
 
24
 
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY() */
25
 
 
 
24
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), getline(),
 
25
                                   asprintf() */
26
26
#include <stddef.h>             /* size_t, NULL */
27
27
#include <stdlib.h>             /* malloc(), exit(), EXIT_FAILURE,
28
28
                                   EXIT_SUCCESS, realloc() */
51
51
                                   close() */
52
52
#include <fcntl.h>              /* fcntl(), F_GETFD, F_SETFD,
53
53
                                   FD_CLOEXEC */
54
 
#include <string.h>             /* strtok, strlen(), strcpy(),
55
 
                                   strcat() */
 
54
#include <string.h>             /* strsep, strlen(), asprintf() */
56
55
#include <errno.h>              /* errno */
57
56
#include <argp.h>               /* struct argp_option, struct
58
57
                                   argp_state, struct argp,
59
58
                                   argp_parse(), ARGP_ERR_UNKNOWN,
60
 
                                   ARGP_KEY_END, ARGP_KEY_ARG, error_t */
 
59
                                   ARGP_KEY_END, ARGP_KEY_ARG,
 
60
                                   error_t */
61
61
#include <signal.h>             /* struct sigaction, sigemptyset(),
62
62
                                   sigaddset(), sigaction(),
63
63
                                   sigprocmask(), SIG_BLOCK, SIGCHLD,
66
66
 
67
67
#define BUFFER_SIZE 256
68
68
 
 
69
#define PDIR "/lib/mandos/plugins.d"
 
70
#define AFILE "/conf/conf.d/mandos/plugin-runner.conf"
 
71
 
69
72
const char *argp_program_version = "plugin-runner 1.0";
70
73
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
71
74
 
78
81
  size_t buffer_size;
79
82
  size_t buffer_length;
80
83
  bool eof;
81
 
  bool completed;
82
 
  int status;
 
84
  volatile bool completed;
 
85
  volatile int status;
83
86
  struct process *next;
84
87
} process;
85
88
 
87
90
  char *name;                   /* can be NULL or any plugin name */
88
91
  char **argv;
89
92
  int argc;
 
93
  char **environ;
 
94
  int envc;
90
95
  bool disabled;
91
96
  struct plugin *next;
92
97
} plugin;
101
106
  /* Create a new plugin */
102
107
  plugin *new_plugin = malloc(sizeof(plugin));
103
108
  if (new_plugin == NULL){
104
 
    perror("malloc");
105
 
    exit(EXIT_FAILURE);
106
 
  }
107
 
  new_plugin->name = name;
 
109
    return NULL;
 
110
  }
 
111
  char *copy_name = NULL;
 
112
  if(name != NULL){
 
113
    copy_name = strdup(name);
 
114
    if(copy_name == NULL){
 
115
      return NULL;
 
116
    }
 
117
  }
 
118
  
 
119
  *new_plugin = (plugin) { .name = copy_name,
 
120
                           .argc = 1,
 
121
                           .envc = 0,
 
122
                           .disabled = false,
 
123
                           .next = *plugin_list };
 
124
  
108
125
  new_plugin->argv = malloc(sizeof(char *) * 2);
109
126
  if (new_plugin->argv == NULL){
110
 
    perror("malloc");
111
 
    exit(EXIT_FAILURE);
 
127
    free(copy_name);
 
128
    free(new_plugin);
 
129
    return NULL;
112
130
  }
113
 
  new_plugin->argv[0] = name;
 
131
  new_plugin->argv[0] = copy_name;
114
132
  new_plugin->argv[1] = NULL;
115
 
  new_plugin->argc = 1;
116
 
  new_plugin->disabled = false;
117
 
  new_plugin->next = *plugin_list;
 
133
 
 
134
  new_plugin->environ = malloc(sizeof(char *));
 
135
  if(new_plugin->environ == NULL){
 
136
    free(copy_name);
 
137
    free(new_plugin->argv);
 
138
    free(new_plugin);
 
139
    return NULL;
 
140
  }
 
141
  new_plugin->environ[0] = NULL;
118
142
  /* Append the new plugin to the list */
119
143
  *plugin_list = new_plugin;
120
144
  return new_plugin;
121
145
}
122
146
 
123
 
static void addargument(plugin *p, char *arg){
124
 
  p->argv[p->argc] = arg;
125
 
  p->argv = realloc(p->argv, sizeof(char *) * (size_t)(p->argc + 2));
126
 
  if (p->argv == NULL){
127
 
    perror("malloc");
128
 
    exit(EXIT_FAILURE);
129
 
  }
130
 
  p->argc++;
131
 
  p->argv[p->argc] = NULL;
132
 
}
 
147
/* Helper function for add_argument and add_environment */
 
148
static bool add_to_char_array(const char *new, char ***array,
 
149
                              int *len){
 
150
  /* Resize the pointed-to array to hold one more pointer */
 
151
  *array = realloc(*array, sizeof(char *)
 
152
                   * (size_t) ((*len) + 2));
 
153
  /* Malloc check */
 
154
  if(*array == NULL){
 
155
    return false;
 
156
  }
 
157
  /* Make a copy of the new string */
 
158
  char *copy = strdup(new);
 
159
  if(copy == NULL){
 
160
    return false;
 
161
  }
 
162
  /* Insert the copy */
 
163
  (*array)[*len] = copy;
 
164
  (*len)++;
 
165
  /* Add a new terminating NULL pointer to the last element */
 
166
  (*array)[*len] = NULL;
 
167
  return true;
 
168
}
 
169
 
 
170
/* Add to a plugin's argument vector */
 
171
static bool add_argument(plugin *p, const char *arg){
 
172
  if(p == NULL){
 
173
    return false;
 
174
  }
 
175
  return add_to_char_array(arg, &(p->argv), &(p->argc));
 
176
}
 
177
 
 
178
/* Add to a plugin's environment */
 
179
static bool add_environment(plugin *p, const char *def){
 
180
  if(p == NULL){
 
181
    return false;
 
182
  }
 
183
  return add_to_char_array(def, &(p->environ), &(p->envc));
 
184
}
 
185
 
133
186
 
134
187
/*
135
188
 * Based on the example in the GNU LibC manual chapter 13.13 "File
149
202
 
150
203
process *process_list = NULL;
151
204
 
152
 
/* Mark a process as completed when it exits, and save its exit
 
205
/* Mark processes as completed when they exit, and save their exit
153
206
   status. */
154
207
void handle_sigchld(__attribute__((unused)) int sig){
155
 
  process *proc = process_list;
156
 
  int status;
157
 
  pid_t pid = wait(&status);
158
 
  if(pid == -1){
159
 
    perror("wait");
160
 
    return;
161
 
  }
162
 
  while(proc != NULL and proc->pid != pid){
163
 
    proc = proc->next;
164
 
  }
165
 
  if(proc == NULL){
166
 
    /* Process not found in process list */
167
 
    return;
168
 
  }
169
 
  proc->status = status;
170
 
  proc->completed = true;
 
208
  while(true){
 
209
    process *proc = process_list;
 
210
    int status;
 
211
    pid_t pid = waitpid(-1, &status, WNOHANG);
 
212
    if(pid == 0){
 
213
      /* Only still running child processes */
 
214
      break;
 
215
    }
 
216
    if(pid == -1){
 
217
      if (errno != ECHILD){
 
218
        perror("waitpid");
 
219
      }
 
220
      /* No child processes */
 
221
      break;
 
222
    }
 
223
 
 
224
    /* A child exited, find it in process_list */
 
225
    while(proc != NULL and proc->pid != pid){
 
226
      proc = proc->next;
 
227
    }
 
228
    if(proc == NULL){
 
229
      /* Process not found in process list */
 
230
      continue;
 
231
    }
 
232
    proc->status = status;
 
233
    proc->completed = true;
 
234
  }
171
235
}
172
236
 
173
237
bool print_out_password(const char *buffer, size_t length){
185
249
  return true;
186
250
}
187
251
 
 
252
static void free_plugin_list(plugin *plugin_list){
 
253
  for(plugin *next; plugin_list != NULL; plugin_list = next){
 
254
    next = plugin_list->next;
 
255
    for(char **arg = plugin_list->argv; *arg != NULL; arg++){
 
256
      free(*arg);
 
257
    }
 
258
    free(plugin_list->argv);
 
259
    for(char **env = plugin_list->environ; *env != NULL; env++){
 
260
      free(*env);
 
261
    }
 
262
    free(plugin_list->environ);
 
263
    free(plugin_list);
 
264
  }
 
265
}
 
266
 
188
267
int main(int argc, char *argv[]){
189
 
  const char *plugindir = "/lib/mandos/plugins.d";
 
268
  char *plugindir = NULL;
 
269
  char *argfile = NULL;
 
270
  FILE *conffp;
190
271
  size_t d_name_len;
191
272
  DIR *dir = NULL;
192
273
  struct dirent *dirst;
200
281
  struct sigaction old_sigchld_action;
201
282
  struct sigaction sigchld_action = { .sa_handler = handle_sigchld,
202
283
                                      .sa_flags = SA_NOCLDSTOP };
203
 
  char *plus_options = NULL;
204
 
  char **plus_argv = NULL;
205
 
 
 
284
  char **custom_argv = NULL;
 
285
  int custom_argc = 0;
 
286
  
206
287
  /* Establish a signal handler */
207
288
  sigemptyset(&sigchld_action.sa_mask);
208
289
  ret = sigaddset(&sigchld_action.sa_mask, SIGCHLD);
209
 
  if(ret < 0){
 
290
  if(ret == -1){
210
291
    perror("sigaddset");
211
 
    exit(EXIT_FAILURE);
 
292
    exitstatus = EXIT_FAILURE;
 
293
    goto fallback;
212
294
  }
213
295
  ret = sigaction(SIGCHLD, &sigchld_action, &old_sigchld_action);
214
 
  if(ret < 0){
 
296
  if(ret == -1){
215
297
    perror("sigaction");
216
 
    exit(EXIT_FAILURE);
 
298
    exitstatus = EXIT_FAILURE;
 
299
    goto fallback;
217
300
  }
218
301
  
219
302
  /* The options we understand. */
221
304
    { .name = "global-options", .key = 'g',
222
305
      .arg = "OPTION[,OPTION[,...]]",
223
306
      .doc = "Options passed to all plugins" },
 
307
    { .name = "global-envs", .key = 'e',
 
308
      .arg = "VAR=value",
 
309
      .doc = "Environment variable passed to all plugins" },
224
310
    { .name = "options-for", .key = 'o',
225
311
      .arg = "PLUGIN:OPTION[,OPTION[,...]]",
226
312
      .doc = "Options passed only to specified plugin" },
 
313
    { .name = "envs-for", .key = 'f',
 
314
      .arg = "PLUGIN:ENV=value",
 
315
      .doc = "Environment variable passed to specified plugin" },
227
316
    { .name = "disable", .key = 'd',
228
317
      .arg = "PLUGIN",
229
318
      .doc = "Disable a specific plugin", .group = 1 },
230
319
    { .name = "plugin-dir", .key = 128,
231
320
      .arg = "DIRECTORY",
232
321
      .doc = "Specify a different plugin directory", .group = 2 },
233
 
    { .name = "userid", .key = 129,
234
 
      .arg = "ID", .flags = 0,
235
 
      .doc = "User ID the plugins will run as", .group = 2 },
236
 
    { .name = "groupid", .key = 130,
237
 
      .arg = "ID", .flags = 0,
238
 
      .doc = "Group ID the plugins will run as", .group = 2 },
239
 
    { .name = "debug", .key = 131,
240
 
      .doc = "Debug mode", .group = 3 },
 
322
    { .name = "config-file", .key = 129,
 
323
      .arg = "FILE",
 
324
      .doc = "Specify a different configuration file", .group = 2 },
 
325
    { .name = "userid", .key = 130,
 
326
      .arg = "ID", .flags = 0,
 
327
      .doc = "User ID the plugins will run as", .group = 3 },
 
328
    { .name = "groupid", .key = 131,
 
329
      .arg = "ID", .flags = 0,
 
330
      .doc = "Group ID the plugins will run as", .group = 3 },
 
331
    { .name = "debug", .key = 132,
 
332
      .doc = "Debug mode", .group = 4 },
241
333
    { .name = NULL }
242
334
  };
243
335
  
248
340
    switch (key) {
249
341
    case 'g':
250
342
      if (arg != NULL){
251
 
        char *p = strtok(arg, ",");
252
 
        do{
253
 
          addargument(getplugin(NULL, plugins), p);
254
 
          p = strtok(NULL, ",");
255
 
        } while (p != NULL);
 
343
        char *p;
 
344
        while((p = strsep(&arg, ",")) != NULL){
 
345
          if(p[0] == '\0'){
 
346
            continue;
 
347
          }
 
348
          if(not add_argument(getplugin(NULL, plugins), p)){
 
349
            perror("add_argument");
 
350
            return ARGP_ERR_UNKNOWN;
 
351
          }
 
352
        }
 
353
      }
 
354
      break;
 
355
    case 'e':
 
356
      if(arg == NULL){
 
357
        break;
 
358
      }
 
359
      {
 
360
        char *envdef = strdup(arg);
 
361
        if(envdef == NULL){
 
362
          break;
 
363
        }
 
364
        if(not add_environment(getplugin(NULL, plugins), envdef)){
 
365
          perror("add_environment");
 
366
        }
256
367
      }
257
368
      break;
258
369
    case 'o':
259
370
      if (arg != NULL){
260
 
        char *name = strtok(arg, ":");
261
 
        char *p = strtok(NULL, ":");
262
 
        if(p != NULL){
263
 
          p = strtok(p, ",");
264
 
          do{
265
 
            addargument(getplugin(name, plugins), p);
266
 
            p = strtok(NULL, ",");
267
 
          } while (p != NULL);
 
371
        char *p_name = strsep(&arg, ":");
 
372
        if(p_name[0] == '\0'){
 
373
          break;
 
374
        }
 
375
        char *opt = strsep(&arg, ":");
 
376
        if(opt[0] == '\0'){
 
377
          break;
 
378
        }
 
379
        if(opt != NULL){
 
380
          char *p;
 
381
          while((p = strsep(&opt, ",")) != NULL){
 
382
            if(p[0] == '\0'){
 
383
              continue;
 
384
            }
 
385
            if(not add_argument(getplugin(p_name, plugins), p)){
 
386
              perror("add_argument");
 
387
              return ARGP_ERR_UNKNOWN;
 
388
            }
 
389
          }
 
390
        }
 
391
      }
 
392
      break;
 
393
    case 'f':
 
394
      if(arg == NULL){
 
395
        break;
 
396
      }
 
397
      {
 
398
        char *envdef = strchr(arg, ':');
 
399
        if(envdef == NULL){
 
400
          break;
 
401
        }
 
402
        char *p_name = strndup(arg, (size_t) (envdef-arg));
 
403
        if(p_name == NULL){
 
404
          break;
 
405
        }
 
406
        envdef++;
 
407
        if(not add_environment(getplugin(p_name, plugins), envdef)){
 
408
          perror("add_environment");
268
409
        }
269
410
      }
270
411
      break;
271
412
    case 'd':
272
413
      if (arg != NULL){
273
 
        getplugin(arg, plugins)->disabled = true;
 
414
        plugin *p = getplugin(arg, plugins);
 
415
        if(p == NULL){
 
416
          return ARGP_ERR_UNKNOWN;
 
417
        }
 
418
        p->disabled = true;
274
419
      }
275
420
      break;
276
421
    case 128:
277
 
      plugindir = arg;
 
422
      plugindir = strdup(arg);
 
423
      if(plugindir == NULL){
 
424
        perror("strdup");
 
425
      }      
278
426
      break;
279
427
    case 129:
 
428
      argfile = strdup(arg);
 
429
      if(argfile == NULL){
 
430
        perror("strdup");
 
431
      }
 
432
      break;      
 
433
    case 130:
280
434
      uid = (uid_t)strtol(arg, NULL, 10);
281
435
      break;
282
 
    case 130:
 
436
    case 131:
283
437
      gid = (gid_t)strtol(arg, NULL, 10);
284
438
      break;
285
 
    case 131:
 
439
    case 132:
286
440
      debug = true;
287
441
      break;
288
442
    case ARGP_KEY_ARG:
289
 
      if(plus_options != NULL or arg == NULL or arg[0] != '+'){
290
 
        argp_usage (state);
291
 
      }
292
 
      plus_options = arg;
 
443
      fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
293
444
      break;
294
445
    case ARGP_KEY_END:
295
446
      break;
309
460
  if (ret == ARGP_ERR_UNKNOWN){
310
461
    fprintf(stderr, "Unknown error while parsing arguments\n");
311
462
    exitstatus = EXIT_FAILURE;
312
 
    goto end;
 
463
    goto fallback;
 
464
  }
 
465
 
 
466
  if (argfile == NULL){
 
467
    conffp = fopen(AFILE, "r");
 
468
  } else {
 
469
    conffp = fopen(argfile, "r");
313
470
  }
314
471
  
315
 
  if(plus_options){
316
 
    /* This is a mangled argument in the form of
317
 
     "+--option+--other-option=parameter+--yet-another-option", etc */
318
 
    /* Make new argc and argv vars, and call argp_parse() again. */
319
 
    plus_options++;             /* skip the first '+' character */
320
 
    const char delims[] = "+";
321
 
    char *arg;
322
 
    int new_argc = 1;
323
 
    plus_argv = malloc(sizeof(char*) * 2);
324
 
    if(plus_argv == NULL){
 
472
  if(conffp != NULL){
 
473
    char *org_line = NULL;
 
474
    char *p, *arg, *new_arg, *line;
 
475
    size_t size = 0;
 
476
    ssize_t sret;
 
477
    const char whitespace_delims[] = " \r\t\f\v\n";
 
478
    const char comment_delim[] = "#";
 
479
 
 
480
    custom_argc = 1;
 
481
    custom_argv = malloc(sizeof(char*) * 2);
 
482
    if(custom_argv == NULL){
325
483
      perror("malloc");
326
484
      exitstatus = EXIT_FAILURE;
327
 
      goto end;
328
 
    }
329
 
    plus_argv[0] = argv[0];
330
 
    plus_argv[1] = NULL;
331
 
    arg = strtok(plus_options, delims); /* Get first argument */
332
 
    while(arg != NULL){
333
 
      new_argc++;
334
 
      plus_argv = realloc(plus_argv, sizeof(char *)
335
 
                         * ((unsigned int) new_argc + 1));
336
 
      if(plus_argv == NULL){
337
 
        perror("realloc");
338
 
        exitstatus = EXIT_FAILURE;
339
 
        goto end;
340
 
      }
341
 
      plus_argv[new_argc-1] = arg;
342
 
      plus_argv[new_argc] = NULL;
343
 
      arg = strtok(NULL, delims); /* Get next argument */
344
 
    }
345
 
    ret = argp_parse (&argp, new_argc, plus_argv, 0, 0, &plugin_list);
 
485
      goto fallback;
 
486
    }
 
487
    custom_argv[0] = argv[0];
 
488
    custom_argv[1] = NULL;
 
489
    
 
490
    while(true){
 
491
      sret = getline(&org_line, &size, conffp);
 
492
      if(sret == -1){
 
493
        break;
 
494
      }
 
495
 
 
496
      line = org_line;
 
497
      arg = strsep(&line, comment_delim);
 
498
      while((p = strsep(&arg, whitespace_delims)) != NULL){
 
499
        if(p[0] == '\0'){
 
500
          continue;
 
501
        }
 
502
        new_arg = strdup(p);
 
503
        if(new_arg == NULL){
 
504
          perror("strdup");
 
505
          exitstatus = EXIT_FAILURE;
 
506
          free(org_line);
 
507
          goto fallback;
 
508
        }
 
509
        
 
510
        custom_argc += 1;
 
511
        custom_argv = realloc(custom_argv, sizeof(char *)
 
512
                              * ((unsigned int) custom_argc + 1));
 
513
        if(custom_argv == NULL){
 
514
          perror("realloc");
 
515
          exitstatus = EXIT_FAILURE;
 
516
          free(org_line);
 
517
          goto fallback;
 
518
        }
 
519
        custom_argv[custom_argc-1] = new_arg;
 
520
        custom_argv[custom_argc] = NULL;        
 
521
      }
 
522
    }
 
523
    free(org_line);
 
524
  } else{
 
525
    /* Check for harmful errors and go to fallback. Other errors might
 
526
       not affect opening plugins */
 
527
    if (errno == EMFILE or errno == ENFILE or errno == ENOMEM){
 
528
      perror("fopen");
 
529
      exitstatus = EXIT_FAILURE;
 
530
      goto fallback;
 
531
    }
 
532
  }
 
533
 
 
534
  if(custom_argv != NULL){
 
535
    ret = argp_parse (&argp, custom_argc, custom_argv, 0, 0, &plugin_list);
346
536
    if (ret == ARGP_ERR_UNKNOWN){
347
537
      fprintf(stderr, "Unknown error while parsing arguments\n");
348
538
      exitstatus = EXIT_FAILURE;
349
 
      goto end;
 
539
      goto fallback;
350
540
    }
351
541
  }
352
542
  
357
547
      for(char **a = p->argv; *a != NULL; a++){
358
548
        fprintf(stderr, "\tArg: %s\n", *a);
359
549
      }
 
550
      fprintf(stderr, "...and %u environment variables\n", p->envc);
 
551
      for(char **a = p->environ; *a != NULL; a++){
 
552
        fprintf(stderr, "\t%s\n", *a);
 
553
      }
360
554
    }
361
555
  }
362
556
  
369
563
  if (ret == -1){
370
564
    perror("setgid");
371
565
  }
 
566
 
 
567
  if (plugindir == NULL){
 
568
    dir = opendir(PDIR);
 
569
  } else {
 
570
    dir = opendir(plugindir);
 
571
  }
372
572
  
373
 
  dir = opendir(plugindir);
374
573
  if(dir == NULL){
375
574
    perror("Could not open plugin dir");
376
575
    exitstatus = EXIT_FAILURE;
377
 
    goto end;
 
576
    goto fallback;
378
577
  }
379
578
  
380
579
  /* Set the FD_CLOEXEC flag on the directory, if possible */
385
584
      if(ret < 0){
386
585
        perror("set_cloexec_flag");
387
586
        exitstatus = EXIT_FAILURE;
388
 
        goto end;
 
587
        goto fallback;
389
588
      }
390
589
    }
391
590
  }
400
599
      if (errno == EBADF){
401
600
        perror("readdir");
402
601
        exitstatus = EXIT_FAILURE;
403
 
        goto end;
 
602
        goto fallback;
404
603
      }
405
604
      break;
406
605
    }
451
650
        continue;
452
651
      }
453
652
    }
454
 
    
455
 
    char *filename = malloc(d_name_len + strlen(plugindir) + 2);
456
 
    if (filename == NULL){
457
 
      perror("malloc");
 
653
 
 
654
    char *filename;
 
655
    ret = asprintf(&filename, "%s/%s", plugindir, dirst->d_name);
 
656
    if(ret < 0){
 
657
      perror("asprintf");
458
658
      continue;
459
659
    }
460
 
    strcpy(filename, plugindir); /* Spurious warning */
461
 
    strcat(filename, "/");      /* Spurious warning */
462
 
    strcat(filename, dirst->d_name); /* Spurious warning */
463
660
    
464
661
    ret = stat(filename, &st);
465
662
    if (ret == -1){
476
673
      free(filename);
477
674
      continue;
478
675
    }
479
 
    if(getplugin(dirst->d_name, &plugin_list)->disabled){
 
676
    plugin *p = getplugin(dirst->d_name, &plugin_list);
 
677
    if(p == NULL){
 
678
      perror("getplugin");
 
679
      free(filename);
 
680
      continue;
 
681
    }
 
682
    if(p->disabled){
480
683
      if(debug){
481
684
        fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
482
685
                dirst->d_name);
484
687
      free(filename);
485
688
      continue;
486
689
    }
487
 
    plugin *p = getplugin(dirst->d_name, &plugin_list);
488
690
    {
489
691
      /* Add global arguments to argument list for this plugin */
490
692
      plugin *g = getplugin(NULL, &plugin_list);
491
 
      for(char **a = g->argv + 1; *a != NULL; a++){
492
 
        addargument(p, *a);
493
 
      }
494
 
    }
495
 
    int pipefd[2]; 
 
693
      if(g != NULL){
 
694
        for(char **a = g->argv + 1; *a != NULL; a++){
 
695
          if(not add_argument(p, *a)){
 
696
            perror("add_argument");
 
697
          }
 
698
        }
 
699
        /* Add global environment variables */
 
700
        for(char **e = g->environ; *e != NULL; e++){
 
701
          if(not add_environment(p, *e)){
 
702
            perror("add_environment");
 
703
          }
 
704
        }
 
705
      }
 
706
    }
 
707
    /* If this plugin has any environment variables, we will call
 
708
       using execve and need to duplicate the environment from this
 
709
       process, too. */
 
710
    if(p->environ[0] != NULL){
 
711
      for(char **e = environ; *e != NULL; e++){
 
712
        char *copy = strdup(*e);
 
713
        if(copy == NULL){
 
714
          perror("strdup");
 
715
          continue;
 
716
        }
 
717
        if(not add_environment(p, copy)){
 
718
          perror("add_environment");
 
719
        }
 
720
      }
 
721
    }
 
722
    
 
723
    int pipefd[2];
496
724
    ret = pipe(pipefd);
497
725
    if (ret == -1){
498
726
      perror("pipe");
499
727
      exitstatus = EXIT_FAILURE;
500
 
      goto end;
 
728
      goto fallback;
501
729
    }
502
730
    ret = set_cloexec_flag(pipefd[0]);
503
731
    if(ret < 0){
504
732
      perror("set_cloexec_flag");
505
733
      exitstatus = EXIT_FAILURE;
506
 
      goto end;
 
734
      goto fallback;
507
735
    }
508
736
    ret = set_cloexec_flag(pipefd[1]);
509
737
    if(ret < 0){
510
738
      perror("set_cloexec_flag");
511
739
      exitstatus = EXIT_FAILURE;
512
 
      goto end;
 
740
      goto fallback;
513
741
    }
514
742
    /* Block SIGCHLD until process is safely in process list */
515
743
    ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
516
744
    if(ret < 0){
517
745
      perror("sigprocmask");
518
746
      exitstatus = EXIT_FAILURE;
519
 
      goto end;
 
747
      goto fallback;
520
748
    }
521
749
    // Starting a new process to be watched
522
750
    pid_t pid = fork();
523
751
    if(pid == -1){
524
752
      perror("fork");
525
753
      exitstatus = EXIT_FAILURE;
526
 
      goto end;
 
754
      goto fallback;
527
755
    }
528
756
    if(pid == 0){
529
757
      /* this is the child process */
549
777
           above and must now close it manually here. */
550
778
        closedir(dir);
551
779
      }
552
 
      if(execv(filename, p->argv) < 0){
553
 
        perror("execv");
554
 
        _exit(EXIT_FAILURE);
 
780
      if(p->environ[0] == NULL){
 
781
        if(execv(filename, p->argv) < 0){
 
782
          perror("execv");
 
783
          _exit(EXIT_FAILURE);
 
784
        }
 
785
      } else {
 
786
        if(execve(filename, p->argv, p->environ) < 0){
 
787
          perror("execve");
 
788
          _exit(EXIT_FAILURE);
 
789
        }
555
790
      }
556
791
      /* no return */
557
792
    }
566
801
        perror("sigprocmask");
567
802
      }
568
803
      exitstatus = EXIT_FAILURE;
569
 
      goto end;
 
804
      goto fallback;
570
805
    }
571
806
    
572
807
    *new_process = (struct process){ .pid = pid,
580
815
    if(ret < 0){
581
816
      perror("sigprocmask");
582
817
      exitstatus = EXIT_FAILURE;
583
 
      goto end;
 
818
      goto fallback;
584
819
    }
585
820
    
586
821
    FD_SET(new_process->fd, &rfds_all);
590
825
    }
591
826
    
592
827
  }
593
 
  
594
 
  /* Free the plugin list */
595
 
  for(plugin *next; plugin_list != NULL; plugin_list = next){
596
 
    next = plugin_list->next;
597
 
    free(plugin_list->argv);
598
 
    free(plugin_list);
599
 
  }
 
828
 
 
829
  free_plugin_list(plugin_list);
 
830
  plugin_list = NULL;
600
831
  
601
832
  closedir(dir);
602
833
  dir = NULL;
612
843
    if (select_ret == -1){
613
844
      perror("select");
614
845
      exitstatus = EXIT_FAILURE;
615
 
      goto end;
 
846
      goto fallback;
616
847
    }
617
848
    /* OK, now either a process completed, or something can be read
618
849
       from one of them */
640
871
          /* Remove the plugin */
641
872
          FD_CLR(proc->fd, &rfds_all);
642
873
          /* Block signal while modifying process_list */
643
 
          ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
 
874
          ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
644
875
          if(ret < 0){
645
876
            perror("sigprocmask");
646
877
            exitstatus = EXIT_FAILURE;
647
 
            goto end;
 
878
            goto fallback;
648
879
          }
649
880
          /* Delete this process entry from the list */
650
881
          if(process_list == proc){
674
905
        }
675
906
        /* This process exited nicely, so print its buffer */
676
907
 
677
 
        bool bret = print_out_password(proc->buffer, proc->buffer_length);
 
908
        bool bret = print_out_password(proc->buffer,
 
909
                                       proc->buffer_length);
678
910
        if(not bret){
679
911
          perror("print_out_password");
680
912
          exitstatus = EXIT_FAILURE;
681
913
        }
682
 
        goto end;
 
914
        goto fallback;
683
915
      }
684
916
      /* This process has not completed.  Does it have any output? */
685
917
      if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
693
925
        if (proc->buffer == NULL){
694
926
          perror("malloc");
695
927
          exitstatus = EXIT_FAILURE;
696
 
          goto end;
 
928
          goto fallback;
697
929
        }
698
930
        proc->buffer_size += BUFFER_SIZE;
699
931
      }
714
946
  }
715
947
 
716
948
 
717
 
 end:
 
949
 fallback:
718
950
  
719
951
  if(process_list == NULL or exitstatus != EXIT_SUCCESS){
720
 
    /* Fallback if all plugins failed, none are found or an error occured */
 
952
    /* Fallback if all plugins failed, none are found or an error
 
953
       occured */
721
954
    bool bret;
722
955
    fprintf(stderr, "Going to fallback mode using getpass(3)\n");
723
956
    char *passwordbuffer = getpass("Password: ");
725
958
    if(not bret){
726
959
      perror("print_out_password");
727
960
      exitstatus = EXIT_FAILURE;
728
 
      goto end;
729
961
    }
730
962
  }
731
963
  
732
964
  /* Restore old signal handler */
733
 
  sigaction(SIGCHLD, &old_sigchld_action, NULL);
734
 
  
735
 
  free(plus_argv);
736
 
  
737
 
  /* Free the plugin list */
738
 
  for(plugin *next; plugin_list != NULL; plugin_list = next){
739
 
    next = plugin_list->next;
740
 
    free(plugin_list->argv);
741
 
    free(plugin_list);
742
 
  }
 
965
  ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
 
966
  if(ret == -1){
 
967
    perror("sigaction");
 
968
    exitstatus = EXIT_FAILURE;
 
969
  }
 
970
 
 
971
  if(custom_argv != NULL){
 
972
    for(char **arg = custom_argv+1; *arg != NULL; arg++){
 
973
      free(*arg);
 
974
    }
 
975
    free(custom_argv);
 
976
  }
 
977
  free_plugin_list(plugin_list);
743
978
  
744
979
  if(dir != NULL){
745
980
    closedir(dir);
765
1000
  if(errno != ECHILD){
766
1001
    perror("wait");
767
1002
  }
 
1003
 
 
1004
  free(plugindir);
 
1005
  free(argfile);
768
1006
  
769
1007
  return exitstatus;
770
1008
}