/mandos/trunk

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

« back to all changes in this revision

Viewing changes to plugbasedclient.c

  • Committer: Teddy Hogeborn
  • Date: 2008-07-20 06:33:48 UTC
  • Revision ID: teddy@fukt.bsnet.se-20080720063348-jscgy5p0itrgvlo8
* mandos-clients.conf ([foo]): Uncommented.
  ([foo]/secret): New.
  ([foo]/secfile): Commented out.
  ([foo]/checker): Changed to "fping -q -- %%(fqdn)s".
  ([foo]/timeout): New.

* server.py: New modeline for Python and Emacs.  Set a logging format.
  (Client.__init__): Bug fix: Choose either the value from the options
                     object or pass the argument through string_to_delta
                     for both "timeout" and "interval".
  (Client.checker_callback): Bug fix: Do not log spurious "Checker for
                             <foo> failed" messages.
  (Client.start_checker): Moved "Starting checker" log message down to
                          just before actually starting the subprocess.
                          Do not redirect the subprocesses' stdout to a
                          pipe.
  (peer_certificate, fingerprint): Added docstrings.
  (entry_group_state_changed): Call "killme()" instead of
                               "main_loop.quit()".
  (daemon, killme): New functions.
  (exitstatus, main_loop_started): New global variables.
  (__main__): Removed the "--cert", "--key", "--ca", and "--crl"
              options.  Removed the sleep command from the default
              checker.  Add a console logger in debug mode.  Call
              "killme()" instead of "main_loop.quit()" when there are no
              more clients.  Call "daemon()" if not in debug mode.
              Register "cleanup()" to run at exit.  Ignore some
              signals.  Catch DBusException to detect another running
              server and exit cleanly.  Exit with "exitstatus".
  (cleanup): New function.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/*  -*- coding: utf-8; mode: c; mode: orgtbl -*- */
2
 
/*
3
 
 * Mandos plugin runner - Run Mandos plugins
4
 
 *
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
9
 
 * modify it under the terms of the GNU General Public License as
10
 
 * published by the Free Software Foundation, either version 3 of the
11
 
 * License, or (at your option) any later version.
12
 
 * 
13
 
 * This program is distributed in the hope that it will be useful, but
14
 
 * WITHOUT ANY WARRANTY; without even the implied warranty of
15
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16
 
 * General Public License for more details.
17
 
 * 
18
 
 * You should have received a copy of the GNU General Public License
19
 
 * along with this program.  If not, see
20
 
 * <http://www.gnu.org/licenses/>.
21
 
 * 
22
 
 * Contact the authors at <mandos@fukt.bsnet.se>.
23
 
 */
24
 
 
25
 
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), getline(),
26
 
                                   asprintf(), O_CLOEXEC */
27
 
#include <stddef.h>             /* size_t, NULL */
28
 
#include <stdlib.h>             /* malloc(), exit(), EXIT_FAILURE,
29
 
                                   EXIT_SUCCESS, realloc() */
30
 
#include <stdbool.h>            /* bool, true, false */
31
 
#include <stdio.h>              /* perror, fileno(), fprintf(),
32
 
                                   stderr, STDOUT_FILENO */
33
 
#include <sys/types.h>          /* DIR, fdopendir(), stat(), struct
34
 
                                   stat, waitpid(), WIFEXITED(),
35
 
                                   WEXITSTATUS(), wait(), pid_t,
36
 
                                   uid_t, gid_t, getuid(), getgid(),
37
 
                                   dirfd() */
38
 
#include <sys/select.h>         /* fd_set, select(), FD_ZERO(),
39
 
                                   FD_SET(), FD_ISSET(), FD_CLR */
40
 
#include <sys/wait.h>           /* wait(), waitpid(), WIFEXITED(),
41
 
                                   WEXITSTATUS(), WTERMSIG(),
42
 
                                   WCOREDUMP() */
43
 
#include <sys/stat.h>           /* struct stat, stat(), S_ISREG() */
44
 
#include <iso646.h>             /* and, or, not */
45
 
#include <dirent.h>             /* DIR, struct dirent, fdopendir(),
46
 
                                   readdir(), closedir(), dirfd() */
47
 
#include <unistd.h>             /* struct stat, stat(), S_ISREG(),
48
 
                                   fcntl(), setuid(), setgid(),
49
 
                                   F_GETFD, F_SETFD, FD_CLOEXEC,
50
 
                                   access(), pipe(), fork(), close()
51
 
                                   dup2(), STDOUT_FILENO, _exit(),
52
 
                                   execv(), write(), read(),
53
 
                                   close() */
54
 
#include <fcntl.h>              /* fcntl(), F_GETFD, F_SETFD,
55
 
                                   FD_CLOEXEC */
56
 
#include <string.h>             /* strsep, strlen(), asprintf(),
57
 
                                   strsignal() */
58
 
#include <errno.h>              /* errno */
59
 
#include <argp.h>               /* struct argp_option, struct
60
 
                                   argp_state, struct argp,
61
 
                                   argp_parse(), ARGP_ERR_UNKNOWN,
62
 
                                   ARGP_KEY_END, ARGP_KEY_ARG,
63
 
                                   error_t */
64
 
#include <signal.h>             /* struct sigaction, sigemptyset(),
65
 
                                   sigaddset(), sigaction(),
66
 
                                   sigprocmask(), SIG_BLOCK, SIGCHLD,
67
 
                                   SIG_UNBLOCK, kill(), sig_atomic_t
68
 
                                */
69
 
#include <errno.h>              /* errno, EBADF */
70
 
#include <inttypes.h>           /* intmax_t, PRIdMAX, strtoimax() */
71
 
#include <sysexits.h>           /* EX_OSERR, EX_USAGE */
72
 
 
73
 
#define BUFFER_SIZE 256
74
 
 
75
 
#define PDIR "/lib/mandos/plugins.d"
76
 
#define AFILE "/conf/conf.d/mandos/plugin-runner.conf"
77
 
 
78
 
const char *argp_program_version = "plugin-runner " VERSION;
79
 
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
80
 
 
81
 
typedef struct plugin{
82
 
  char *name;                   /* can be NULL or any plugin name */
83
 
  char **argv;
84
 
  int argc;
85
 
  char **environ;
86
 
  int envc;
87
 
  bool disabled;
88
 
  
89
 
  /* Variables used for running processes*/
 
1
#include <stdio.h>      /* popen, fileno */
 
2
#include <iso646.h>     /* and, or, not */
 
3
#include <sys/types.h>  /* DIR, opendir, stat, struct stat, waitpid,
 
4
                           WIFEXITED, WEXITSTATUS, wait */
 
5
#include <sys/wait.h>   /* wait */
 
6
#include <dirent.h>     /* DIR, opendir */
 
7
#include <sys/stat.h>   /* stat, struct stat */
 
8
#include <unistd.h>     /* stat, struct stat, chdir */
 
9
#include <stdlib.h>     /* EXIT_FAILURE */
 
10
#include <sys/select.h> /* fd_set, select, FD_ZERO, FD_SET, FD_ISSET */
 
11
#include <string.h>     /* strlen, strcpy, strcat */
 
12
#include <stdbool.h>    /* true */
 
13
#include <sys/wait.h>   /* waitpid, WIFEXITED, WEXITSTATUS */
 
14
#include <errno.h>      /* errno */
 
15
 
 
16
struct process;
 
17
 
 
18
typedef struct process{
90
19
  pid_t pid;
91
20
  int fd;
92
21
  char *buffer;
93
 
  size_t buffer_size;
94
 
  size_t buffer_length;
95
 
  bool eof;
96
 
  volatile sig_atomic_t completed;
97
 
  int status;
98
 
  struct plugin *next;
99
 
} plugin;
100
 
 
101
 
static plugin *plugin_list = NULL;
102
 
 
103
 
/* Gets an existing plugin based on name,
104
 
   or if none is found, creates a new one */
105
 
static plugin *getplugin(char *name){
106
 
  /* Check for existing plugin with that name */
107
 
  for(plugin *p = plugin_list; p != NULL; p = p->next){
108
 
    if((p->name == name)
109
 
       or (p->name and name and (strcmp(p->name, name) == 0))){
110
 
      return p;
111
 
    }
112
 
  }
113
 
  /* Create a new plugin */
114
 
  plugin *new_plugin = NULL;
115
 
  do {
116
 
    new_plugin = malloc(sizeof(plugin));
117
 
  } while(new_plugin == NULL and errno == EINTR);
118
 
  if(new_plugin == NULL){
119
 
    return NULL;
120
 
  }
121
 
  char *copy_name = NULL;
122
 
  if(name != NULL){
123
 
    do {
124
 
      copy_name = strdup(name);
125
 
    } while(copy_name == NULL and errno == EINTR);
126
 
    if(copy_name == NULL){
127
 
      int e = errno;
128
 
      free(new_plugin);
129
 
      errno = e;
130
 
      return NULL;
131
 
    }
132
 
  }
133
 
  
134
 
  *new_plugin = (plugin){ .name = copy_name,
135
 
                          .argc = 1,
136
 
                          .disabled = false,
137
 
                          .next = plugin_list };
138
 
  
139
 
  do {
140
 
    new_plugin->argv = malloc(sizeof(char *) * 2);
141
 
  } while(new_plugin->argv == NULL and errno == EINTR);
142
 
  if(new_plugin->argv == NULL){
143
 
    int e = errno;
144
 
    free(copy_name);
145
 
    free(new_plugin);
146
 
    errno = e;
147
 
    return NULL;
148
 
  }
149
 
  new_plugin->argv[0] = copy_name;
150
 
  new_plugin->argv[1] = NULL;
151
 
  
152
 
  do {
153
 
    new_plugin->environ = malloc(sizeof(char *));
154
 
  } while(new_plugin->environ == NULL and errno == EINTR);
155
 
  if(new_plugin->environ == NULL){
156
 
    int e = errno;
157
 
    free(copy_name);
158
 
    free(new_plugin->argv);
159
 
    free(new_plugin);
160
 
    errno = e;
161
 
    return NULL;
162
 
  }
163
 
  new_plugin->environ[0] = NULL;
164
 
  
165
 
  /* Append the new plugin to the list */
166
 
  plugin_list = new_plugin;
167
 
  return new_plugin;
168
 
}
169
 
 
170
 
/* Helper function for add_argument and add_environment */
171
 
static bool add_to_char_array(const char *new, char ***array,
172
 
                              int *len){
173
 
  /* Resize the pointed-to array to hold one more pointer */
174
 
  do {
175
 
    *array = realloc(*array, sizeof(char *)
176
 
                     * (size_t) ((*len) + 2));
177
 
  } while(*array == NULL and errno == EINTR);
178
 
  /* Malloc check */
179
 
  if(*array == NULL){
180
 
    return false;
181
 
  }
182
 
  /* Make a copy of the new string */
183
 
  char *copy;
184
 
  do {
185
 
    copy = strdup(new);
186
 
  } while(copy == NULL and errno == EINTR);
187
 
  if(copy == NULL){
188
 
    return false;
189
 
  }
190
 
  /* Insert the copy */
191
 
  (*array)[*len] = copy;
192
 
  (*len)++;
193
 
  /* Add a new terminating NULL pointer to the last element */
194
 
  (*array)[*len] = NULL;
195
 
  return true;
196
 
}
197
 
 
198
 
/* Add to a plugin's argument vector */
199
 
static bool add_argument(plugin *p, const char *arg){
200
 
  if(p == NULL){
201
 
    return false;
202
 
  }
203
 
  return add_to_char_array(arg, &(p->argv), &(p->argc));
204
 
}
205
 
 
206
 
/* Add to a plugin's environment */
207
 
static bool add_environment(plugin *p, const char *def, bool replace){
208
 
  if(p == NULL){
209
 
    return false;
210
 
  }
211
 
  /* namelen = length of name of environment variable */
212
 
  size_t namelen = (size_t)(strchrnul(def, '=') - def);
213
 
  /* Search for this environment variable */
214
 
  for(char **e = p->environ; *e != NULL; e++){
215
 
    if(strncmp(*e, def, namelen + 1) == 0){
216
 
      /* It already exists */
217
 
      if(replace){
218
 
        char *new;
219
 
        do {
220
 
          new = realloc(*e, strlen(def) + 1);
221
 
        } while(new == NULL and errno == EINTR);
222
 
        if(new == NULL){
223
 
          return false;
224
 
        }
225
 
        *e = new;
226
 
        strcpy(*e, def);
227
 
      }
228
 
      return true;
229
 
    }
230
 
  }
231
 
  return add_to_char_array(def, &(p->environ), &(p->envc));
232
 
}
233
 
 
234
 
/*
235
 
 * Based on the example in the GNU LibC manual chapter 13.13 "File
236
 
 * Descriptor Flags".
237
 
 | [[info:libc:Descriptor%20Flags][File Descriptor Flags]] |
238
 
 */
239
 
static int set_cloexec_flag(int fd){
240
 
  int ret = (int)TEMP_FAILURE_RETRY(fcntl(fd, F_GETFD, 0));
241
 
  /* If reading the flags failed, return error indication now. */
242
 
  if(ret < 0){
243
 
    return ret;
244
 
  }
245
 
  /* Store modified flag word in the descriptor. */
246
 
  return (int)TEMP_FAILURE_RETRY(fcntl(fd, F_SETFD,
247
 
                                       ret | FD_CLOEXEC));
248
 
}
249
 
 
250
 
 
251
 
/* Mark processes as completed when they exit, and save their exit
252
 
   status. */
253
 
static void handle_sigchld(__attribute__((unused)) int sig){
254
 
  int old_errno = errno;
255
 
  while(true){
256
 
    plugin *proc = plugin_list;
257
 
    int status;
258
 
    pid_t pid = waitpid(-1, &status, WNOHANG);
259
 
    if(pid == 0){
260
 
      /* Only still running child processes */
261
 
      break;
262
 
    }
263
 
    if(pid == -1){
264
 
      if(errno == ECHILD){
265
 
        /* No child processes */
266
 
        break;
267
 
      }
268
 
      perror("waitpid");
269
 
    }
270
 
    
271
 
    /* A child exited, find it in process_list */
272
 
    while(proc != NULL and proc->pid != pid){
273
 
      proc = proc->next;
274
 
    }
275
 
    if(proc == NULL){
276
 
      /* Process not found in process list */
277
 
      continue;
278
 
    }
279
 
    proc->status = status;
280
 
    proc->completed = 1;
281
 
  }
282
 
  errno = old_errno;
283
 
}
284
 
 
285
 
/* Prints out a password to stdout */
286
 
static bool print_out_password(const char *buffer, size_t length){
287
 
  ssize_t ret;
288
 
  for(size_t written = 0; written < length; written += (size_t)ret){
289
 
    ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO, buffer + written,
290
 
                                   length - written));
291
 
    if(ret < 0){
292
 
      return false;
293
 
    }
294
 
  }
295
 
  return true;
296
 
}
297
 
 
298
 
/* Removes and free a plugin from the plugin list */
299
 
static void free_plugin(plugin *plugin_node){
300
 
  
301
 
  for(char **arg = plugin_node->argv; *arg != NULL; arg++){
302
 
    free(*arg);
303
 
  }
304
 
  free(plugin_node->argv);
305
 
  for(char **env = plugin_node->environ; *env != NULL; env++){
306
 
    free(*env);
307
 
  }
308
 
  free(plugin_node->environ);
309
 
  free(plugin_node->buffer);
310
 
  
311
 
  /* Removes the plugin from the singly-linked list */
312
 
  if(plugin_node == plugin_list){
313
 
    /* First one - simple */
314
 
    plugin_list = plugin_list->next;
315
 
  } else {
316
 
    /* Second one or later */
317
 
    for(plugin *p = plugin_list; p != NULL; p = p->next){
318
 
      if(p->next == plugin_node){
319
 
        p->next = plugin_node->next;
320
 
        break;
321
 
      }
322
 
    }
323
 
  }
324
 
  
325
 
  free(plugin_node);
326
 
}
327
 
 
328
 
static void free_plugin_list(void){
329
 
  while(plugin_list != NULL){
330
 
    free_plugin(plugin_list);
331
 
  }
332
 
}
 
22
  int buffer_size;
 
23
  int buffer_length;
 
24
  struct process *next;
 
25
} process;
 
26
 
 
27
#define BUFFER_SIZE 256
333
28
 
334
29
int main(int argc, char *argv[]){
335
 
  char *plugindir = NULL;
336
 
  char *argfile = NULL;
337
 
  FILE *conffp;
338
 
  size_t d_name_len;
339
 
  DIR *dir = NULL;
 
30
  char plugindir[] = "plugins.d";
 
31
  size_t d_name_len, plugindir_len = sizeof(plugindir)-1;
 
32
  DIR *dir;
340
33
  struct dirent *dirst;
341
34
  struct stat st;
342
 
  fd_set rfds_all;
 
35
  fd_set rfds_orig;
343
36
  int ret, maxfd = 0;
344
 
  ssize_t sret;
345
 
  uid_t uid = 65534;
346
 
  gid_t gid = 65534;
347
 
  bool debug = false;
348
 
  int exitstatus = EXIT_SUCCESS;
349
 
  struct sigaction old_sigchld_action;
350
 
  struct sigaction sigchld_action = { .sa_handler = handle_sigchld,
351
 
                                      .sa_flags = SA_NOCLDSTOP };
352
 
  char **custom_argv = NULL;
353
 
  int custom_argc = 0;
354
 
  
355
 
  /* Establish a signal handler */
356
 
  sigemptyset(&sigchld_action.sa_mask);
357
 
  ret = sigaddset(&sigchld_action.sa_mask, SIGCHLD);
358
 
  if(ret == -1){
359
 
    perror("sigaddset");
360
 
    exitstatus = EXIT_FAILURE;
361
 
    goto fallback;
362
 
  }
363
 
  ret = sigaction(SIGCHLD, &sigchld_action, &old_sigchld_action);
364
 
  if(ret == -1){
365
 
    perror("sigaction");
366
 
    exitstatus = EXIT_FAILURE;
367
 
    goto fallback;
368
 
  }
369
 
  
370
 
  /* The options we understand. */
371
 
  struct argp_option options[] = {
372
 
    { .name = "global-options", .key = 'g',
373
 
      .arg = "OPTION[,OPTION[,...]]",
374
 
      .doc = "Options passed to all plugins" },
375
 
    { .name = "global-env", .key = 'G',
376
 
      .arg = "VAR=value",
377
 
      .doc = "Environment variable passed to all plugins" },
378
 
    { .name = "options-for", .key = 'o',
379
 
      .arg = "PLUGIN:OPTION[,OPTION[,...]]",
380
 
      .doc = "Options passed only to specified plugin" },
381
 
    { .name = "env-for", .key = 'E',
382
 
      .arg = "PLUGIN:ENV=value",
383
 
      .doc = "Environment variable passed to specified plugin" },
384
 
    { .name = "disable", .key = 'd',
385
 
      .arg = "PLUGIN",
386
 
      .doc = "Disable a specific plugin", .group = 1 },
387
 
    { .name = "enable", .key = 'e',
388
 
      .arg = "PLUGIN",
389
 
      .doc = "Enable a specific plugin", .group = 1 },
390
 
    { .name = "plugin-dir", .key = 128,
391
 
      .arg = "DIRECTORY",
392
 
      .doc = "Specify a different plugin directory", .group = 2 },
393
 
    { .name = "config-file", .key = 129,
394
 
      .arg = "FILE",
395
 
      .doc = "Specify a different configuration file", .group = 2 },
396
 
    { .name = "userid", .key = 130,
397
 
      .arg = "ID", .flags = 0,
398
 
      .doc = "User ID the plugins will run as", .group = 3 },
399
 
    { .name = "groupid", .key = 131,
400
 
      .arg = "ID", .flags = 0,
401
 
      .doc = "Group ID the plugins will run as", .group = 3 },
402
 
    { .name = "debug", .key = 132,
403
 
      .doc = "Debug mode", .group = 4 },
404
 
    /*
405
 
     * These reproduce what we would get without ARGP_NO_HELP
406
 
     */
407
 
    { .name = "help", .key = '?',
408
 
      .doc = "Give this help list", .group = -1 },
409
 
    { .name = "usage", .key = -3,
410
 
      .doc = "Give a short usage message", .group = -1 },
411
 
    { .name = "version", .key = 'V',
412
 
      .doc = "Print program version", .group = -1 },
413
 
    { .name = NULL }
414
 
  };
415
 
  
416
 
  error_t parse_opt(int key, char *arg, struct argp_state *state){
417
 
    errno = 0;
418
 
    switch(key){
419
 
      char *tmp;
420
 
      intmax_t tmpmax;
421
 
    case 'g':                   /* --global-options */
422
 
      {
423
 
        char *plugin_option;
424
 
        while((plugin_option = strsep(&arg, ",")) != NULL){
425
 
          if(not add_argument(getplugin(NULL), plugin_option)){
426
 
            break;
427
 
          }
428
 
        }
429
 
      }
430
 
      break;
431
 
    case 'G':                   /* --global-env */
432
 
      add_environment(getplugin(NULL), arg, true);
433
 
      break;
434
 
    case 'o':                   /* --options-for */
435
 
      {
436
 
        char *option_list = strchr(arg, ':');
437
 
        if(option_list == NULL){
438
 
          argp_error(state, "No colon in \"%s\"", arg);
439
 
          errno = EINVAL;
440
 
          break;
441
 
        }
442
 
        *option_list = '\0';
443
 
        option_list++;
444
 
        if(arg[0] == '\0'){
445
 
          argp_error(state, "Empty plugin name");
446
 
          errno = EINVAL;
447
 
          break;
448
 
        }
449
 
        char *option;
450
 
        while((option = strsep(&option_list, ",")) != NULL){
451
 
          if(not add_argument(getplugin(arg), option)){
452
 
            break;
453
 
          }
454
 
        }
455
 
      }
456
 
      break;
457
 
    case 'E':                   /* --env-for */
458
 
      {
459
 
        char *envdef = strchr(arg, ':');
460
 
        if(envdef == NULL){
461
 
          argp_error(state, "No colon in \"%s\"", arg);
462
 
          errno = EINVAL;
463
 
          break;
464
 
        }
465
 
        *envdef = '\0';
466
 
        envdef++;
467
 
        if(arg[0] == '\0'){
468
 
          argp_error(state, "Empty plugin name");
469
 
          errno = EINVAL;
470
 
          break;
471
 
        }
472
 
        add_environment(getplugin(arg), envdef, true);
473
 
      }
474
 
      break;
475
 
    case 'd':                   /* --disable */
476
 
      {
477
 
        plugin *p = getplugin(arg);
478
 
        if(p != NULL){
479
 
          p->disabled = true;
480
 
        }
481
 
      }
482
 
      break;
483
 
    case 'e':                   /* --enable */
484
 
      {
485
 
        plugin *p = getplugin(arg);
486
 
        if(p != NULL){
487
 
          p->disabled = false;
488
 
        }
489
 
      }
490
 
      break;
491
 
    case 128:                   /* --plugin-dir */
492
 
      free(plugindir);
493
 
      plugindir = strdup(arg);
494
 
      break;
495
 
    case 129:                   /* --config-file */
496
 
      /* This is already done by parse_opt_config_file() */
497
 
      break;
498
 
    case 130:                   /* --userid */
499
 
      tmpmax = strtoimax(arg, &tmp, 10);
500
 
      if(errno != 0 or tmp == arg or *tmp != '\0'
501
 
         or tmpmax != (uid_t)tmpmax){
502
 
        argp_error(state, "Bad user ID number: \"%s\", using %"
503
 
                   PRIdMAX, arg, (intmax_t)uid);
504
 
        break;
505
 
      }
506
 
      uid = (uid_t)tmpmax;
507
 
      break;
508
 
    case 131:                   /* --groupid */
509
 
      tmpmax = strtoimax(arg, &tmp, 10);
510
 
      if(errno != 0 or tmp == arg or *tmp != '\0'
511
 
         or tmpmax != (gid_t)tmpmax){
512
 
        argp_error(state, "Bad group ID number: \"%s\", using %"
513
 
                   PRIdMAX, arg, (intmax_t)gid);
514
 
        break;
515
 
      }
516
 
      gid = (gid_t)tmpmax;
517
 
      break;
518
 
    case 132:                   /* --debug */
519
 
      debug = true;
520
 
      break;
521
 
      /*
522
 
       * These reproduce what we would get without ARGP_NO_HELP
523
 
       */
524
 
    case '?':                   /* --help */
525
 
      state->flags &= ~(unsigned int)ARGP_NO_EXIT; /* force exit */
526
 
      argp_state_help(state, state->out_stream, ARGP_HELP_STD_HELP);
527
 
    case -3:                    /* --usage */
528
 
      state->flags &= ~(unsigned int)ARGP_NO_EXIT; /* force exit */
529
 
      argp_state_help(state, state->out_stream,
530
 
                      ARGP_HELP_USAGE | ARGP_HELP_EXIT_OK);
531
 
    case 'V':                   /* --version */
532
 
      fprintf(state->out_stream, "%s\n", argp_program_version);
533
 
      exit(EXIT_SUCCESS);
534
 
      break;
535
 
/*
536
 
 * When adding more options before this line, remember to also add a
537
 
 * "case" to the "parse_opt_config_file" function below.
538
 
 */
539
 
    case ARGP_KEY_ARG:
540
 
      /* Cryptsetup always passes an argument, which is an empty
541
 
         string if "none" was specified in /etc/crypttab.  So if
542
 
         argument was empty, we ignore it silently. */
543
 
      if(arg[0] == '\0'){
544
 
        break;
545
 
      }
546
 
    default:
547
 
      return ARGP_ERR_UNKNOWN;
548
 
    }
549
 
    return errno;               /* Set to 0 at start */
550
 
  }
551
 
  
552
 
  /* This option parser is the same as parse_opt() above, except it
553
 
     ignores everything but the --config-file option. */
554
 
  error_t parse_opt_config_file(int key, char *arg,
555
 
                                __attribute__((unused))
556
 
                                struct argp_state *state){
557
 
    errno = 0;
558
 
    switch(key){
559
 
    case 'g':                   /* --global-options */
560
 
    case 'G':                   /* --global-env */
561
 
    case 'o':                   /* --options-for */
562
 
    case 'E':                   /* --env-for */
563
 
    case 'd':                   /* --disable */
564
 
    case 'e':                   /* --enable */
565
 
    case 128:                   /* --plugin-dir */
566
 
      break;
567
 
    case 129:                   /* --config-file */
568
 
      free(argfile);
569
 
      argfile = strdup(arg);
570
 
      break;
571
 
    case 130:                   /* --userid */
572
 
    case 131:                   /* --groupid */
573
 
    case 132:                   /* --debug */
574
 
    case '?':                   /* --help */
575
 
    case -3:                    /* --usage */
576
 
    case 'V':                   /* --version */
577
 
    case ARGP_KEY_ARG:
578
 
      break;
579
 
    default:
580
 
      return ARGP_ERR_UNKNOWN;
581
 
    }
582
 
    return errno;
583
 
  }
584
 
  
585
 
  struct argp argp = { .options = options,
586
 
                       .parser = parse_opt_config_file,
587
 
                       .args_doc = "",
588
 
                       .doc = "Mandos plugin runner -- Run plugins" };
589
 
  
590
 
  /* Parse using parse_opt_config_file() in order to get the custom
591
 
     config file location, if any. */
592
 
  ret = argp_parse(&argp, argc, argv,
593
 
                   ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
594
 
                   NULL, NULL);
595
 
  switch(ret){
596
 
  case 0:
597
 
    break;
598
 
  case ENOMEM:
599
 
  default:
600
 
    errno = ret;
601
 
    perror("argp_parse");
602
 
    exitstatus = EX_OSERR;
603
 
    goto fallback;
604
 
  case EINVAL:
605
 
    exitstatus = EX_USAGE;
606
 
    goto fallback;
607
 
  }
608
 
  
609
 
  /* Reset to the normal argument parser */
610
 
  argp.parser = parse_opt;
611
 
  
612
 
  /* Open the configfile if available */
613
 
  if(argfile == NULL){
614
 
    conffp = fopen(AFILE, "r");
615
 
  } else {
616
 
    conffp = fopen(argfile, "r");
617
 
  }
618
 
  if(conffp != NULL){
619
 
    char *org_line = NULL;
620
 
    char *p, *arg, *new_arg, *line;
621
 
    size_t size = 0;
622
 
    const char whitespace_delims[] = " \r\t\f\v\n";
623
 
    const char comment_delim[] = "#";
624
 
    
625
 
    custom_argc = 1;
626
 
    custom_argv = malloc(sizeof(char*) * 2);
627
 
    if(custom_argv == NULL){
628
 
      perror("malloc");
629
 
      exitstatus = EXIT_FAILURE;
630
 
      goto fallback;
631
 
    }
632
 
    custom_argv[0] = argv[0];
633
 
    custom_argv[1] = NULL;
634
 
    
635
 
    /* for each line in the config file, strip whitespace and ignore
636
 
       commented text */
637
 
    while(true){
638
 
      sret = getline(&org_line, &size, conffp);
639
 
      if(sret == -1){
640
 
        break;
641
 
      }
642
 
      
643
 
      line = org_line;
644
 
      arg = strsep(&line, comment_delim);
645
 
      while((p = strsep(&arg, whitespace_delims)) != NULL){
646
 
        if(p[0] == '\0'){
647
 
          continue;
648
 
        }
649
 
        new_arg = strdup(p);
650
 
        if(new_arg == NULL){
651
 
          perror("strdup");
652
 
          exitstatus = EXIT_FAILURE;
653
 
          free(org_line);
654
 
          goto fallback;
655
 
        }
656
 
        
657
 
        custom_argc += 1;
658
 
        custom_argv = realloc(custom_argv, sizeof(char *)
659
 
                              * ((unsigned int) custom_argc + 1));
660
 
        if(custom_argv == NULL){
661
 
          perror("realloc");
662
 
          exitstatus = EXIT_FAILURE;
663
 
          free(org_line);
664
 
          goto fallback;
665
 
        }
666
 
        custom_argv[custom_argc-1] = new_arg;
667
 
        custom_argv[custom_argc] = NULL;
668
 
      }
669
 
    }
670
 
    do {
671
 
      ret = fclose(conffp);
672
 
    } while(ret == EOF and errno == EINTR);
673
 
    if(ret == EOF){
674
 
      perror("fclose");
675
 
      exitstatus = EXIT_FAILURE;
676
 
      goto fallback;
677
 
    }
678
 
    free(org_line);
679
 
  } else {
680
 
    /* Check for harmful errors and go to fallback. Other errors might
681
 
       not affect opening plugins */
682
 
    if(errno == EMFILE or errno == ENFILE or errno == ENOMEM){
683
 
      perror("fopen");
684
 
      exitstatus = EXIT_FAILURE;
685
 
      goto fallback;
686
 
    }
687
 
  }
688
 
  /* If there were any arguments from the configuration file, pass
689
 
     them to parser as command line arguments */
690
 
  if(custom_argv != NULL){
691
 
    ret = argp_parse(&argp, custom_argc, custom_argv,
692
 
                     ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
693
 
                     NULL, NULL);
694
 
    switch(ret){
695
 
    case 0:
696
 
      break;
697
 
    case ENOMEM:
698
 
    default:
699
 
      errno = ret;
700
 
      perror("argp_parse");
701
 
      exitstatus = EX_OSERR;
702
 
      goto fallback;
703
 
    case EINVAL:
704
 
      exitstatus = EX_CONFIG;
705
 
      goto fallback;
706
 
    }
707
 
  }
708
 
  
709
 
  /* Parse actual command line arguments, to let them override the
710
 
     config file */
711
 
  ret = argp_parse(&argp, argc, argv,
712
 
                   ARGP_IN_ORDER | ARGP_NO_EXIT | ARGP_NO_HELP,
713
 
                   NULL, NULL);
714
 
  switch(ret){
715
 
  case 0:
716
 
    break;
717
 
  case ENOMEM:
718
 
  default:
719
 
    errno = ret;
720
 
    perror("argp_parse");
721
 
    exitstatus = EX_OSERR;
722
 
    goto fallback;
723
 
  case EINVAL:
724
 
    exitstatus = EX_USAGE;
725
 
    goto fallback;
726
 
  }
727
 
  
728
 
  if(debug){
729
 
    for(plugin *p = plugin_list; p != NULL; p=p->next){
730
 
      fprintf(stderr, "Plugin: %s has %d arguments\n",
731
 
              p->name ? p->name : "Global", p->argc - 1);
732
 
      for(char **a = p->argv; *a != NULL; a++){
733
 
        fprintf(stderr, "\tArg: %s\n", *a);
734
 
      }
735
 
      fprintf(stderr, "...and %d environment variables\n", p->envc);
736
 
      for(char **a = p->environ; *a != NULL; a++){
737
 
        fprintf(stderr, "\t%s\n", *a);
738
 
      }
739
 
    }
740
 
  }
741
 
  
742
 
  /* Strip permissions down to nobody */
743
 
  setgid(gid);
744
 
  if(ret == -1){
745
 
    perror("setgid");
746
 
  }
747
 
  ret = setuid(uid);
748
 
  if(ret == -1){
749
 
    perror("setuid");
750
 
  }
751
 
  
752
 
  /* Open plugin directory with close_on_exec flag */
753
 
  {
754
 
    int dir_fd = -1;
755
 
    if(plugindir == NULL){
756
 
      dir_fd = open(PDIR, O_RDONLY |
757
 
#ifdef O_CLOEXEC
758
 
                    O_CLOEXEC
759
 
#else  /* not O_CLOEXEC */
760
 
                    0
761
 
#endif  /* not O_CLOEXEC */
762
 
                    );
763
 
    } else {
764
 
      dir_fd = open(plugindir, O_RDONLY |
765
 
#ifdef O_CLOEXEC
766
 
                    O_CLOEXEC
767
 
#else  /* not O_CLOEXEC */
768
 
                    0
769
 
#endif  /* not O_CLOEXEC */
770
 
                    );
771
 
    }
772
 
    if(dir_fd == -1){
773
 
      perror("Could not open plugin dir");
774
 
      exitstatus = EXIT_FAILURE;
775
 
      goto fallback;
776
 
    }
777
 
    
778
 
#ifndef O_CLOEXEC
779
 
  /* Set the FD_CLOEXEC flag on the directory */
780
 
    ret = set_cloexec_flag(dir_fd);
781
 
    if(ret < 0){
782
 
      perror("set_cloexec_flag");
783
 
      TEMP_FAILURE_RETRY(close(dir_fd));
784
 
      exitstatus = EXIT_FAILURE;
785
 
      goto fallback;
786
 
    }
787
 
#endif  /* O_CLOEXEC */
788
 
    
789
 
    dir = fdopendir(dir_fd);
790
 
    if(dir == NULL){
791
 
      perror("Could not open plugin dir");
792
 
      TEMP_FAILURE_RETRY(close(dir_fd));
793
 
      exitstatus = EXIT_FAILURE;
794
 
      goto fallback;
795
 
    }
796
 
  }
797
 
  
798
 
  FD_ZERO(&rfds_all);
799
 
  
800
 
  /* Read and execute any executable in the plugin directory*/
 
37
  process *process_list = NULL;
 
38
  
 
39
  dir = opendir(plugindir);
 
40
 
 
41
  if(dir == NULL){
 
42
    fprintf(stderr, "Can not open directory\n");
 
43
    return EXIT_FAILURE;
 
44
  }
 
45
  
 
46
  FD_ZERO(&rfds_orig);
 
47
  
801
48
  while(true){
802
 
    do {
803
 
      dirst = readdir(dir);
804
 
    } while(dirst == NULL and errno == EINTR);
 
49
    dirst = readdir(dir);
805
50
    
806
 
    /* All directory entries have been processed */
 
51
    // All directory entries have been processed
807
52
    if(dirst == NULL){
808
 
      if(errno == EBADF){
809
 
        perror("readdir");
810
 
        exitstatus = EXIT_FAILURE;
811
 
        goto fallback;
812
 
      }
813
53
      break;
814
54
    }
815
55
    
816
56
    d_name_len = strlen(dirst->d_name);
817
57
    
818
 
    /* Ignore dotfiles, backup files and other junk */
819
 
    {
820
 
      bool bad_name = false;
821
 
      
822
 
      const char const *bad_prefixes[] = { ".", "#", NULL };
823
 
      
824
 
      const char const *bad_suffixes[] = { "~", "#", ".dpkg-new",
825
 
                                           ".dpkg-old",
826
 
                                           ".dpkg-bak",
827
 
                                           ".dpkg-divert", NULL };
828
 
      for(const char **pre = bad_prefixes; *pre != NULL; pre++){
829
 
        size_t pre_len = strlen(*pre);
830
 
        if((d_name_len >= pre_len)
831
 
           and strncmp((dirst->d_name), *pre, pre_len) == 0){
832
 
          if(debug){
833
 
            fprintf(stderr, "Ignoring plugin dir entry \"%s\""
834
 
                    " with bad prefix %s\n", dirst->d_name, *pre);
835
 
          }
836
 
          bad_name = true;
837
 
          break;
838
 
        }
839
 
      }
840
 
      if(bad_name){
841
 
        continue;
842
 
      }
843
 
      for(const char **suf = bad_suffixes; *suf != NULL; suf++){
844
 
        size_t suf_len = strlen(*suf);
845
 
        if((d_name_len >= suf_len)
846
 
           and (strcmp((dirst->d_name)+d_name_len-suf_len, *suf)
847
 
                == 0)){
848
 
          if(debug){
849
 
            fprintf(stderr, "Ignoring plugin dir entry \"%s\""
850
 
                    " with bad suffix %s\n", dirst->d_name, *suf);
851
 
          }
852
 
          bad_name = true;
853
 
          break;
854
 
        }
855
 
      }
856
 
      
857
 
      if(bad_name){
858
 
        continue;
859
 
      }
860
 
    }
861
 
    
862
 
    char *filename;
863
 
    if(plugindir == NULL){
864
 
      ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, PDIR "/%s",
865
 
                                             dirst->d_name));
866
 
    } else {
867
 
      ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, "%s/%s",
868
 
                                             plugindir,
869
 
                                             dirst->d_name));
870
 
    }
871
 
    if(ret < 0){
872
 
      perror("asprintf");
873
 
      continue;
874
 
    }
875
 
    
876
 
    ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
877
 
    if(ret == -1){
878
 
      perror("stat");
879
 
      free(filename);
880
 
      continue;
881
 
    }
882
 
    
883
 
    /* Ignore non-executable files */
884
 
    if(not S_ISREG(st.st_mode)
885
 
       or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
886
 
      if(debug){
887
 
        fprintf(stderr, "Ignoring plugin dir entry \"%s\""
888
 
                " with bad type or mode\n", filename);
889
 
      }
890
 
      free(filename);
891
 
      continue;
892
 
    }
893
 
    
894
 
    plugin *p = getplugin(dirst->d_name);
895
 
    if(p == NULL){
896
 
      perror("getplugin");
897
 
      free(filename);
898
 
      continue;
899
 
    }
900
 
    if(p->disabled){
901
 
      if(debug){
902
 
        fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
903
 
                dirst->d_name);
904
 
      }
905
 
      free(filename);
906
 
      continue;
907
 
    }
908
 
    {
909
 
      /* Add global arguments to argument list for this plugin */
910
 
      plugin *g = getplugin(NULL);
911
 
      if(g != NULL){
912
 
        for(char **a = g->argv + 1; *a != NULL; a++){
913
 
          if(not add_argument(p, *a)){
914
 
            perror("add_argument");
915
 
          }
916
 
        }
917
 
        /* Add global environment variables */
918
 
        for(char **e = g->environ; *e != NULL; e++){
919
 
          if(not add_environment(p, *e, false)){
920
 
            perror("add_environment");
921
 
          }
922
 
        }
923
 
      }
924
 
    }
925
 
    /* If this plugin has any environment variables, we will call
926
 
       using execve and need to duplicate the environment from this
927
 
       process, too. */
928
 
    if(p->environ[0] != NULL){
929
 
      for(char **e = environ; *e != NULL; e++){
930
 
        if(not add_environment(p, *e, false)){
931
 
          perror("add_environment");
932
 
        }
933
 
      }
934
 
    }
935
 
    
936
 
    int pipefd[2];
937
 
    ret = (int)TEMP_FAILURE_RETRY(pipe(pipefd));
938
 
    if(ret == -1){
939
 
      perror("pipe");
940
 
      exitstatus = EXIT_FAILURE;
941
 
      goto fallback;
942
 
    }
943
 
    /* Ask OS to automatic close the pipe on exec */
944
 
    ret = set_cloexec_flag(pipefd[0]);
945
 
    if(ret < 0){
946
 
      perror("set_cloexec_flag");
947
 
      exitstatus = EXIT_FAILURE;
948
 
      goto fallback;
949
 
    }
950
 
    ret = set_cloexec_flag(pipefd[1]);
951
 
    if(ret < 0){
952
 
      perror("set_cloexec_flag");
953
 
      exitstatus = EXIT_FAILURE;
954
 
      goto fallback;
955
 
    }
956
 
    /* Block SIGCHLD until process is safely in process list */
957
 
    ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
958
 
                                              &sigchld_action.sa_mask,
959
 
                                              NULL));
960
 
    if(ret < 0){
961
 
      perror("sigprocmask");
962
 
      exitstatus = EXIT_FAILURE;
963
 
      goto fallback;
964
 
    }
965
 
    /* Starting a new process to be watched */
966
 
    pid_t pid;
967
 
    do {
968
 
      pid = fork();
969
 
    } while(pid == -1 and errno == EINTR);
970
 
    if(pid == -1){
971
 
      perror("fork");
972
 
      exitstatus = EXIT_FAILURE;
973
 
      goto fallback;
974
 
    }
975
 
    if(pid == 0){
976
 
      /* this is the child process */
977
 
      ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
978
 
      if(ret < 0){
979
 
        perror("sigaction");
980
 
        _exit(EXIT_FAILURE);
981
 
      }
982
 
      ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
983
 
      if(ret < 0){
984
 
        perror("sigprocmask");
985
 
        _exit(EXIT_FAILURE);
986
 
      }
987
 
      
988
 
      ret = dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */
989
 
      if(ret == -1){
990
 
        perror("dup2");
991
 
        _exit(EXIT_FAILURE);
992
 
      }
993
 
      
994
 
      if(dirfd(dir) < 0){
995
 
        /* If dir has no file descriptor, we could not set FD_CLOEXEC
996
 
           above and must now close it manually here. */
 
58
    // Ignore dotfiles and backup files
 
59
    if (dirst->d_name[0] == '.'
 
60
        or dirst->d_name[d_name_len - 1] == '~'){
 
61
      continue;
 
62
    }
 
63
 
 
64
    char *filename = malloc(d_name_len + plugindir_len + 2);
 
65
    strcpy(filename, plugindir);
 
66
    strcat(filename, "/");
 
67
    strcat(filename, dirst->d_name);    
 
68
 
 
69
    stat(filename, &st);
 
70
 
 
71
    if (S_ISREG(st.st_mode) and (access(filename, X_OK) == 0)){
 
72
      // Starting a new process to be watched
 
73
      process *new_process = malloc(sizeof(process));
 
74
      int pipefd[2];
 
75
      pipe(pipefd);
 
76
      new_process->pid = fork();
 
77
      if(new_process->pid == 0){
 
78
        /* this is the child process */
997
79
        closedir(dir);
998
 
      }
999
 
      if(p->environ[0] == NULL){
1000
 
        if(execv(filename, p->argv) < 0){
1001
 
          perror("execv");
1002
 
          _exit(EXIT_FAILURE);
1003
 
        }
1004
 
      } else {
1005
 
        if(execve(filename, p->argv, p->environ) < 0){
1006
 
          perror("execve");
1007
 
          _exit(EXIT_FAILURE);
1008
 
        }
1009
 
      }
1010
 
      /* no return */
1011
 
    }
1012
 
    /* Parent process */
1013
 
    TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
1014
 
                                             pipe */
1015
 
    free(filename);
1016
 
    plugin *new_plugin = getplugin(dirst->d_name);
1017
 
    if(new_plugin == NULL){
1018
 
      perror("getplugin");
1019
 
      ret = (int)(TEMP_FAILURE_RETRY
1020
 
                  (sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask,
1021
 
                               NULL)));
1022
 
      if(ret < 0){
1023
 
        perror("sigprocmask");
1024
 
      }
1025
 
      exitstatus = EXIT_FAILURE;
1026
 
      goto fallback;
1027
 
    }
1028
 
    
1029
 
    new_plugin->pid = pid;
1030
 
    new_plugin->fd = pipefd[0];
1031
 
    
1032
 
    /* Unblock SIGCHLD so signal handler can be run if this process
1033
 
       has already completed */
1034
 
    ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_UNBLOCK,
1035
 
                                              &sigchld_action.sa_mask,
1036
 
                                              NULL));
1037
 
    if(ret < 0){
1038
 
      perror("sigprocmask");
1039
 
      exitstatus = EXIT_FAILURE;
1040
 
      goto fallback;
1041
 
    }
1042
 
    
1043
 
    FD_SET(new_plugin->fd, &rfds_all); /* Spurious warning from
1044
 
                                          -Wconversion */
1045
 
    
1046
 
    if(maxfd < new_plugin->fd){
1047
 
      maxfd = new_plugin->fd;
1048
 
    }
1049
 
  }
1050
 
  
1051
 
  TEMP_FAILURE_RETRY(closedir(dir));
1052
 
  dir = NULL;
1053
 
  free_plugin(getplugin(NULL));
1054
 
  
1055
 
  for(plugin *p = plugin_list; p != NULL; p = p->next){
1056
 
    if(p->pid != 0){
 
80
        close(pipefd[0]);       /* close unused read end of pipe */
 
81
        dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */
 
82
        /* create a new modified argument list */
 
83
        char **new_argv = malloc(sizeof(char *) * (argc + 1));
 
84
        new_argv[0] = filename;
 
85
        for(int i = 1; i < argc; i++){
 
86
          new_argv[i] = argv[i];
 
87
        }
 
88
        new_argv[argc] = NULL;
 
89
        if(execv(filename, new_argv) < 0){
 
90
          perror(argv[0]);
 
91
          close(pipefd[1]);
 
92
          exit(EXIT_FAILURE);
 
93
        }
 
94
        /* no return */
 
95
      }
 
96
      close(pipefd[1]);         /* close unused write end of pipe */
 
97
      new_process->fd = pipefd[0];
 
98
      new_process->buffer = malloc(BUFFER_SIZE);
 
99
      if (new_process->buffer == NULL){
 
100
        perror(argv[0]);
 
101
        goto end;
 
102
      }
 
103
      new_process->buffer_size = BUFFER_SIZE;
 
104
      new_process->buffer_length = 0;
 
105
      FD_SET(new_process->fd, &rfds_orig);
 
106
      
 
107
      if (maxfd < new_process->fd){
 
108
        maxfd = new_process->fd;
 
109
      }
 
110
      
 
111
      //List handling
 
112
      new_process->next = process_list;
 
113
      process_list = new_process;
 
114
    }
 
115
  }
 
116
  
 
117
  closedir(dir);
 
118
  
 
119
  if (process_list != NULL){
 
120
    while(true){
 
121
      fd_set rfds = rfds_orig;
 
122
      int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
 
123
      if (select_ret == -1){
 
124
        perror(argv[0]);
 
125
        goto end;
 
126
      }else{    
 
127
        for(process *process_itr = process_list; process_itr != NULL;
 
128
            process_itr = process_itr->next){
 
129
          if(FD_ISSET(process_itr->fd, &rfds)){
 
130
            if(process_itr->buffer_length + BUFFER_SIZE
 
131
               > process_itr->buffer_size){
 
132
                process_itr->buffer = realloc(process_itr->buffer,
 
133
                                              process_itr->buffer_size
 
134
                                              + BUFFER_SIZE);
 
135
                if (process_itr->buffer == NULL){
 
136
                  perror(argv[0]);
 
137
                  goto end;
 
138
                }
 
139
                process_itr->buffer_size += BUFFER_SIZE;
 
140
            }
 
141
            ret = read(process_itr->fd, process_itr->buffer
 
142
                       + process_itr->buffer_length, BUFFER_SIZE);
 
143
            process_itr->buffer_length+=ret;
 
144
            if(ret == 0){
 
145
              /* got EOF */
 
146
              /* wait for process exit */
 
147
              int status;
 
148
              waitpid(process_itr->pid, &status, 0);
 
149
              if(WIFEXITED(status) and WEXITSTATUS(status) == 0){
 
150
                write(STDOUT_FILENO, process_itr->buffer,
 
151
                      process_itr->buffer_length);
 
152
                goto end;
 
153
              } else {
 
154
                FD_CLR(process_itr->fd, &rfds_orig);
 
155
              }
 
156
            }
 
157
          }
 
158
        }
 
159
      }
 
160
    }
 
161
  }
 
162
  
 
163
 end:
 
164
  for(process *process_itr = process_list; process_itr != NULL;
 
165
      process_itr = process_itr->next){
 
166
    close(process_itr->fd);
 
167
    kill(process_itr->pid, SIGTERM);
 
168
    free(process_itr->buffer);
 
169
  }
 
170
  
 
171
  while(true){
 
172
    int status;
 
173
    ret = wait(&status);
 
174
    if (ret == -1){
 
175
      if(errno != ECHILD){
 
176
        perror("wait");
 
177
      }
1057
178
      break;
1058
179
    }
1059
 
    if(p->next == NULL){
1060
 
      fprintf(stderr, "No plugin processes started. Incorrect plugin"
1061
 
              " directory?\n");
1062
 
      free_plugin_list();
1063
 
    }
1064
 
  }
1065
 
  
1066
 
  /* Main loop while running plugins exist */
1067
 
  while(plugin_list){
1068
 
    fd_set rfds = rfds_all;
1069
 
    int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
1070
 
    if(select_ret == -1 and errno != EINTR){
1071
 
      perror("select");
1072
 
      exitstatus = EXIT_FAILURE;
1073
 
      goto fallback;
1074
 
    }
1075
 
    /* OK, now either a process completed, or something can be read
1076
 
       from one of them */
1077
 
    for(plugin *proc = plugin_list; proc != NULL;){
1078
 
      /* Is this process completely done? */
1079
 
      if(proc->completed and proc->eof){
1080
 
        /* Only accept the plugin output if it exited cleanly */
1081
 
        if(not WIFEXITED(proc->status)
1082
 
           or WEXITSTATUS(proc->status) != 0){
1083
 
          /* Bad exit by plugin */
1084
 
          
1085
 
          if(debug){
1086
 
            if(WIFEXITED(proc->status)){
1087
 
              fprintf(stderr, "Plugin %s [%" PRIdMAX "] exited with"
1088
 
                      " status %d\n", proc->name,
1089
 
                      (intmax_t) (proc->pid),
1090
 
                      WEXITSTATUS(proc->status));
1091
 
            } else if(WIFSIGNALED(proc->status)){
1092
 
              fprintf(stderr, "Plugin %s [%" PRIdMAX "] killed by"
1093
 
                      " signal %d: %s\n", proc->name,
1094
 
                      (intmax_t) (proc->pid),
1095
 
                      WTERMSIG(proc->status),
1096
 
                      strsignal(WTERMSIG(proc->status)));
1097
 
            } else if(WCOREDUMP(proc->status)){
1098
 
              fprintf(stderr, "Plugin %s [%" PRIdMAX "] dumped"
1099
 
                      " core\n", proc->name, (intmax_t) (proc->pid));
1100
 
            }
1101
 
          }
1102
 
          
1103
 
          /* Remove the plugin */
1104
 
          FD_CLR(proc->fd, &rfds_all); /* Spurious warning from
1105
 
                                          -Wconversion */
1106
 
          
1107
 
          /* Block signal while modifying process_list */
1108
 
          ret = (int)TEMP_FAILURE_RETRY(sigprocmask
1109
 
                                        (SIG_BLOCK,
1110
 
                                         &sigchld_action.sa_mask,
1111
 
                                         NULL));
1112
 
          if(ret < 0){
1113
 
            perror("sigprocmask");
1114
 
            exitstatus = EXIT_FAILURE;
1115
 
            goto fallback;
1116
 
          }
1117
 
          
1118
 
          plugin *next_plugin = proc->next;
1119
 
          free_plugin(proc);
1120
 
          proc = next_plugin;
1121
 
          
1122
 
          /* We are done modifying process list, so unblock signal */
1123
 
          ret = (int)(TEMP_FAILURE_RETRY
1124
 
                      (sigprocmask(SIG_UNBLOCK,
1125
 
                                   &sigchld_action.sa_mask, NULL)));
1126
 
          if(ret < 0){
1127
 
            perror("sigprocmask");
1128
 
            exitstatus = EXIT_FAILURE;
1129
 
            goto fallback;
1130
 
          }
1131
 
          
1132
 
          if(plugin_list == NULL){
1133
 
            break;
1134
 
          }
1135
 
          
1136
 
          continue;
1137
 
        }
1138
 
        
1139
 
        /* This process exited nicely, so print its buffer */
1140
 
        
1141
 
        bool bret = print_out_password(proc->buffer,
1142
 
                                       proc->buffer_length);
1143
 
        if(not bret){
1144
 
          perror("print_out_password");
1145
 
          exitstatus = EXIT_FAILURE;
1146
 
        }
1147
 
        goto fallback;
1148
 
      }
1149
 
      
1150
 
      /* This process has not completed.  Does it have any output? */
1151
 
      if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious
1152
 
                                                         warning from
1153
 
                                                         -Wconversion */
1154
 
        /* This process had nothing to say at this time */
1155
 
        proc = proc->next;
1156
 
        continue;
1157
 
      }
1158
 
      /* Before reading, make the process' data buffer large enough */
1159
 
      if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
1160
 
        proc->buffer = realloc(proc->buffer, proc->buffer_size
1161
 
                               + (size_t) BUFFER_SIZE);
1162
 
        if(proc->buffer == NULL){
1163
 
          perror("malloc");
1164
 
          exitstatus = EXIT_FAILURE;
1165
 
          goto fallback;
1166
 
        }
1167
 
        proc->buffer_size += BUFFER_SIZE;
1168
 
      }
1169
 
      /* Read from the process */
1170
 
      sret = TEMP_FAILURE_RETRY(read(proc->fd,
1171
 
                                     proc->buffer
1172
 
                                     + proc->buffer_length,
1173
 
                                     BUFFER_SIZE));
1174
 
      if(sret < 0){
1175
 
        /* Read error from this process; ignore the error */
1176
 
        proc = proc->next;
1177
 
        continue;
1178
 
      }
1179
 
      if(sret == 0){
1180
 
        /* got EOF */
1181
 
        proc->eof = true;
1182
 
      } else {
1183
 
        proc->buffer_length += (size_t) sret;
1184
 
      }
1185
 
    }
1186
 
  }
1187
 
  
1188
 
  
1189
 
 fallback:
1190
 
  
1191
 
  if(plugin_list == NULL or exitstatus != EXIT_SUCCESS){
1192
 
    /* Fallback if all plugins failed, none are found or an error
1193
 
       occured */
1194
 
    bool bret;
1195
 
    fprintf(stderr, "Going to fallback mode using getpass(3)\n");
1196
 
    char *passwordbuffer = getpass("Password: ");
1197
 
    size_t len = strlen(passwordbuffer);
1198
 
    /* Strip trailing newline */
1199
 
    if(len > 0 and passwordbuffer[len-1] == '\n'){
1200
 
      passwordbuffer[len-1] = '\0'; /* not strictly necessary */
1201
 
      len--;
1202
 
    }
1203
 
    bret = print_out_password(passwordbuffer, len);
1204
 
    if(not bret){
1205
 
      perror("print_out_password");
1206
 
      exitstatus = EXIT_FAILURE;
1207
 
    }
1208
 
  }
1209
 
  
1210
 
  /* Restore old signal handler */
1211
 
  ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
1212
 
  if(ret == -1){
1213
 
    perror("sigaction");
1214
 
    exitstatus = EXIT_FAILURE;
1215
 
  }
1216
 
  
1217
 
  if(custom_argv != NULL){
1218
 
    for(char **arg = custom_argv+1; *arg != NULL; arg++){
1219
 
      free(*arg);
1220
 
    }
1221
 
    free(custom_argv);
1222
 
  }
1223
 
  
1224
 
  if(dir != NULL){
1225
 
    closedir(dir);
1226
 
  }
1227
 
  
1228
 
  /* Kill the processes */
1229
 
  for(plugin *p = plugin_list; p != NULL; p = p->next){
1230
 
    if(p->pid != 0){
1231
 
      close(p->fd);
1232
 
      ret = kill(p->pid, SIGTERM);
1233
 
      if(ret == -1 and errno != ESRCH){
1234
 
        /* Set-uid proccesses might not get closed */
1235
 
        perror("kill");
1236
 
      }
1237
 
    }
1238
 
  }
1239
 
  
1240
 
  /* Wait for any remaining child processes to terminate */
1241
 
  do {
1242
 
    ret = wait(NULL);
1243
 
  } while(ret >= 0);
1244
 
  if(errno != ECHILD){
1245
 
    perror("wait");
1246
 
  }
1247
 
  
1248
 
  free_plugin_list();
1249
 
  
1250
 
  free(plugindir);
1251
 
  free(argfile);
1252
 
  
1253
 
  return exitstatus;
 
180
  }  
 
181
  return EXIT_SUCCESS;
1254
182
}