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