/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 plugin-runner.c

  • Committer: Teddy Hogeborn
  • Date: 2008-08-16 03:29:08 UTC
  • Revision ID: teddy@fukt.bsnet.se-20080816032908-ihw7c05r2mnyk389
Add feature to specify custom environment variables for plugins.

* plugin-runner.c (plugin): New members "environ" and "envc" to
                            contain possible custom environment.
  (getplugin): Return NULL on failure instead of doing exit(); all
               callers changed.
  (add_to_char_array): New helper function for "add_argument" and
                       "add_environment".
  (addargument): Renamed to "add_argument".  Return bool.  Call
                 "add_to_char_array" to actually do things.
  (add_environment): New; analogous to "add_argument".
  (addcustomargument): Renamed to "add_to_argv" to avoid confusion
                       with "add_argument".
  (main): New options "--global-envs" and "--envs-for" to specify
          custom environment for plugins.  Print environment for
          plugins in debug mode.  Use asprintf instead of strcpy and
          strcat.  Use execve() for plugins with custom environments.
          Free environment for plugin when freeing plugin list.

Show diffs side-by-side

added added

removed removed

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