/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: 2019-03-18 22:29:25 UTC
  • Revision ID: teddy@recompile.se-20190318222925-jvhek84dgcfgj6g3
mandos-ctl: Refactor tests

* mandos-ctl: Where the clients names "foo" and "barbar" do not refer
              to the actual mock clients in the TestCommand class,
              change all occurrences of these names to "client1" and
              "client2" (or just "client" when only one is used) .
              Also change all test doubles to use correct terminology;
              some things called mocks are actually stubs or spies,
              and rename all true mocks to have "mock" in their names.
              Also eliminate duplicate values in tests; derive values
              from previously defined values whenever possible.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/*  -*- coding: utf-8 -*- */
 
1
/*  -*- coding: utf-8; mode: c; mode: orgtbl -*- */
2
2
/*
3
3
 * Mandos plugin runner - Run Mandos plugins
4
4
 *
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
 
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
13
16
 * WITHOUT ANY WARRANTY; without even the implied warranty of
14
17
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
18
 * General Public License for more details.
16
19
 * 
17
20
 * You should have received a copy of the GNU General Public License
18
 
 * along with this program.  If not, see
19
 
 * <http://www.gnu.org/licenses/>.
 
21
 * along with Mandos.  If not, see <http://www.gnu.org/licenses/>.
20
22
 * 
21
 
 * Contact the authors at <mandos@fukt.bsnet.se>.
 
23
 * Contact the authors at <mandos@recompile.se>.
22
24
 */
23
25
 
24
26
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), getline(),
25
 
                                   asprintf() */
 
27
                                   O_CLOEXEC, pipe2() */
26
28
#include <stddef.h>             /* size_t, NULL */
27
 
#include <stdlib.h>             /* malloc(), exit(), EXIT_FAILURE,
28
 
                                   EXIT_SUCCESS, realloc() */
 
29
#include <stdlib.h>             /* malloc(), exit(), EXIT_SUCCESS,
 
30
                                   realloc() */
29
31
#include <stdbool.h>            /* bool, true, false */
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() */
 
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() */
37
38
#include <sys/select.h>         /* fd_set, select(), FD_ZERO(),
38
39
                                   FD_SET(), FD_ISSET(), FD_CLR */
39
40
#include <sys/wait.h>           /* wait(), waitpid(), WIFEXITED(),
40
 
                                   WEXITSTATUS() */
41
 
#include <sys/stat.h>           /* struct stat, stat(), S_ISREG() */
 
41
                                   WEXITSTATUS(), WTERMSIG() */
 
42
#include <sys/stat.h>           /* struct stat, fstat(), S_ISREG() */
42
43
#include <iso646.h>             /* and, or, not */
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() */
 
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
52
#include <fcntl.h>              /* fcntl(), F_GETFD, F_SETFD,
53
 
                                   FD_CLOEXEC */
54
 
#include <string.h>             /* strsep, strlen(), asprintf() */
 
53
                                   FD_CLOEXEC, openat(), scandirat(),
 
54
                                   pipe2() */
 
55
#include <string.h>             /* strsep, strlen(), strsignal(),
 
56
                                   strcmp(), strncmp() */
55
57
#include <errno.h>              /* errno */
56
58
#include <argp.h>               /* struct argp_option, struct
57
59
                                   argp_state, struct argp,
58
60
                                   argp_parse(), ARGP_ERR_UNKNOWN,
59
 
                                   ARGP_KEY_END, ARGP_KEY_ARG, error_t */
 
61
                                   ARGP_KEY_END, ARGP_KEY_ARG,
 
62
                                   error_t */
60
63
#include <signal.h>             /* struct sigaction, sigemptyset(),
61
64
                                   sigaddset(), sigaction(),
62
65
                                   sigprocmask(), SIG_BLOCK, SIGCHLD,
63
 
                                   SIG_UNBLOCK, kill() */
 
66
                                   SIG_UNBLOCK, kill(), sig_atomic_t
 
67
                                */
64
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() */
65
75
 
66
76
#define BUFFER_SIZE 256
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;
 
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>";
85
84
 
86
85
typedef struct plugin{
87
86
  char *name;                   /* can be NULL or any plugin name */
90
89
  char **environ;
91
90
  int envc;
92
91
  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;
93
102
  struct plugin *next;
94
103
} plugin;
95
104
 
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))){
 
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))){
100
115
      return p;
101
116
    }
102
117
  }
103
118
  /* Create a new plugin */
104
 
  plugin *new_plugin = malloc(sizeof(plugin));
105
 
  if (new_plugin == NULL){
 
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){
106
124
    return NULL;
107
125
  }
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){
 
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);
116
150
    free(new_plugin);
 
151
    errno = e;
117
152
    return NULL;
118
153
  }
119
 
  new_plugin->argv[0] = name;
 
154
  new_plugin->argv[0] = copy_name;
120
155
  new_plugin->argv[1] = NULL;
121
 
 
122
 
  new_plugin->environ = malloc(sizeof(char *));
 
156
  
 
157
  do {
 
158
    new_plugin->environ = malloc(sizeof(char *));
 
159
  } while(new_plugin->environ == NULL and errno == EINTR);
123
160
  if(new_plugin->environ == NULL){
 
161
    int e = errno;
 
162
    free(copy_name);
124
163
    free(new_plugin->argv);
125
164
    free(new_plugin);
 
165
    errno = e;
126
166
    return NULL;
127
167
  }
128
168
  new_plugin->environ[0] = NULL;
 
169
  
129
170
  /* Append the new plugin to the list */
130
 
  *plugin_list = new_plugin;
 
171
  plugin_list = new_plugin;
131
172
  return new_plugin;
132
173
}
133
174
 
134
175
/* Helper function for add_argument and add_environment */
 
176
__attribute__((nonnull, warn_unused_result))
135
177
static bool add_to_char_array(const char *new, char ***array,
136
178
                              int *len){
137
179
  /* Resize the pointed-to array to hold one more pointer */
138
 
  *array = realloc(*array, sizeof(char *)
139
 
                   * (size_t) ((*len) + 2));
 
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);
140
185
  /* Malloc check */
141
 
  if(*array == NULL){
 
186
  if(new_array == NULL){
142
187
    return false;
143
188
  }
 
189
  *array = new_array;
144
190
  /* Make a copy of the new string */
145
 
  char *copy = strdup(new);
 
191
  char *copy;
 
192
  do {
 
193
    copy = strdup(new);
 
194
  } while(copy == NULL and errno == EINTR);
146
195
  if(copy == NULL){
147
196
    return false;
148
197
  }
155
204
}
156
205
 
157
206
/* Add to a plugin's argument vector */
 
207
__attribute__((nonnull(2), warn_unused_result))
158
208
static bool add_argument(plugin *p, const char *arg){
159
209
  if(p == NULL){
160
210
    return false;
163
213
}
164
214
 
165
215
/* Add to a plugin's environment */
166
 
static bool add_environment(plugin *p, const char *def){
 
216
__attribute__((nonnull(2), warn_unused_result))
 
217
static bool add_environment(plugin *p, const char *def, bool replace){
167
218
  if(p == NULL){
168
219
    return false;
169
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
  }
170
241
  return add_to_char_array(def, &(p->environ), &(p->envc));
171
242
}
172
243
 
173
 
 
 
244
#ifndef O_CLOEXEC
174
245
/*
175
246
 * Based on the example in the GNU LibC manual chapter 13.13 "File
176
247
 * Descriptor Flags".
177
 
 * *Note File Descriptor Flags:(libc)Descriptor Flags.
 
248
 | [[info:libc:Descriptor%20Flags][File Descriptor Flags]] |
178
249
 */
179
 
static int set_cloexec_flag(int fd)
180
 
{
181
 
  int ret = fcntl(fd, F_GETFD, 0);
 
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));
182
253
  /* If reading the flags failed, return error indication now. */
183
254
  if(ret < 0){
184
255
    return ret;
185
256
  }
186
257
  /* Store modified flag word in the descriptor. */
187
 
  return fcntl(fd, F_SETFD, ret | FD_CLOEXEC);
 
258
  return (int)TEMP_FAILURE_RETRY(fcntl(fd, F_SETFD,
 
259
                                       ret | FD_CLOEXEC));
188
260
}
189
 
 
190
 
process *process_list = NULL;
191
 
 
192
 
/* Mark a process as completed when it exits, and save its exit
 
261
#endif  /* not O_CLOEXEC */
 
262
 
 
263
 
 
264
/* Mark processes as completed when they exit, and save their exit
193
265
   status. */
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;
 
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;
211
296
}
212
297
 
213
 
bool print_out_password(const char *buffer, size_t length){
 
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){
214
301
  ssize_t ret;
215
 
  if(length>0 and buffer[length-1] == '\n'){
216
 
    length--;
217
 
  }
218
302
  for(size_t written = 0; written < length; written += (size_t)ret){
219
303
    ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO, buffer + written,
220
304
                                   length - written));
225
309
  return true;
226
310
}
227
311
 
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;
 
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
      }
234
337
    }
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;
 
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
  }
247
347
}
248
348
 
249
349
int main(int argc, char *argv[]){
250
 
  const char *plugindir = "/lib/mandos/plugins.d";
251
 
  const char *argfile = ARGFILE;
 
350
  char *plugindir = NULL;
 
351
  char *pluginhelperdir = NULL;
 
352
  char *argfile = NULL;
252
353
  FILE *conffp;
253
 
  size_t d_name_len;
254
 
  DIR *dir = NULL;
255
 
  struct dirent *dirst;
 
354
  struct dirent **direntries = NULL;
256
355
  struct stat st;
257
356
  fd_set rfds_all;
258
357
  int ret, maxfd = 0;
 
358
  ssize_t sret;
259
359
  uid_t uid = 65534;
260
360
  gid_t gid = 65534;
261
361
  bool debug = false;
265
365
                                      .sa_flags = SA_NOCLDSTOP };
266
366
  char **custom_argv = NULL;
267
367
  int custom_argc = 0;
 
368
  int dir_fd = -1;
268
369
  
269
370
  /* Establish a signal handler */
270
371
  sigemptyset(&sigchld_action.sa_mask);
271
372
  ret = sigaddset(&sigchld_action.sa_mask, SIGCHLD);
272
 
  if(ret < 0){
273
 
    perror("sigaddset");
274
 
    exit(EXIT_FAILURE);
 
373
  if(ret == -1){
 
374
    error(0, errno, "sigaddset");
 
375
    exitstatus = EX_OSERR;
 
376
    goto fallback;
275
377
  }
276
378
  ret = sigaction(SIGCHLD, &sigchld_action, &old_sigchld_action);
277
 
  if(ret < 0){
278
 
    perror("sigaction");
279
 
    exit(EXIT_FAILURE);
 
379
  if(ret == -1){
 
380
    error(0, errno, "sigaction");
 
381
    exitstatus = EX_OSERR;
 
382
    goto fallback;
280
383
  }
281
384
  
282
385
  /* The options we understand. */
284
387
    { .name = "global-options", .key = 'g',
285
388
      .arg = "OPTION[,OPTION[,...]]",
286
389
      .doc = "Options passed to all plugins" },
287
 
    { .name = "global-envs", .key = 'e',
 
390
    { .name = "global-env", .key = 'G',
288
391
      .arg = "VAR=value",
289
392
      .doc = "Environment variable passed to all plugins" },
290
393
    { .name = "options-for", .key = 'o',
291
394
      .arg = "PLUGIN:OPTION[,OPTION[,...]]",
292
395
      .doc = "Options passed only to specified plugin" },
293
 
    { .name = "envs-for", .key = 'f',
 
396
    { .name = "env-for", .key = 'E',
294
397
      .arg = "PLUGIN:ENV=value",
295
398
      .doc = "Environment variable passed to specified plugin" },
296
399
    { .name = "disable", .key = 'd',
297
400
      .arg = "PLUGIN",
298
401
      .doc = "Disable a specific plugin", .group = 1 },
 
402
    { .name = "enable", .key = 'e',
 
403
      .arg = "PLUGIN",
 
404
      .doc = "Enable a specific plugin", .group = 1 },
299
405
    { .name = "plugin-dir", .key = 128,
300
406
      .arg = "DIRECTORY",
301
407
      .doc = "Specify a different plugin directory", .group = 2 },
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 },
 
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 },
310
432
    { .name = NULL }
311
433
  };
312
434
  
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
 
      }
 
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 */
374
482
      {
375
483
        char *envdef = strchr(arg, ':');
376
484
        if(envdef == NULL){
377
 
          break;
378
 
        }
379
 
        char *p_name = strndup(arg, (size_t) (envdef-arg));
380
 
        if(p_name == NULL){
381
 
          break;
382
 
        }
 
485
          argp_error(state, "No colon in \"%s\"", arg);
 
486
          errno = EINVAL;
 
487
          break;
 
488
        }
 
489
        *envdef = '\0';
383
490
        envdef++;
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:
 
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 */
408
552
      debug = true;
409
553
      break;
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]",
 
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 = "",
425
635
                       .doc = "Mandos plugin runner -- Run plugins" };
426
636
  
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");
 
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
  }
435
665
  if(conffp != NULL){
436
666
    char *org_line = NULL;
 
667
    char *p, *arg, *new_arg, *line;
437
668
    size_t size = 0;
438
 
    ssize_t sret;
439
 
    char *p, *arg, *new_arg, *line;
440
669
    const char whitespace_delims[] = " \r\t\f\v\n";
441
670
    const char comment_delim[] = "#";
442
 
 
 
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 */
443
684
    while(true){
444
685
      sret = getline(&org_line, &size, conffp);
445
686
      if(sret == -1){
446
687
        break;
447
688
      }
448
 
 
 
689
      
449
690
      line = org_line;
450
691
      arg = strsep(&line, comment_delim);
451
692
      while((p = strsep(&arg, whitespace_delims)) != NULL){
453
694
          continue;
454
695
        }
455
696
        new_arg = strdup(p);
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
 
        }
 
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;
462
720
      }
463
721
    }
 
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
    }
464
730
    free(org_line);
465
 
  } else{
 
731
  } else {
466
732
    /* Check for harmful errors and go to fallback. Other errors might
467
733
       not affect opening plugins */
468
 
    if (errno == EMFILE or errno == ENFILE or errno == ENOMEM){
469
 
      perror("fopen");
470
 
      exitstatus = EXIT_FAILURE;
471
 
      goto end;
 
734
    if(errno == EMFILE or errno == ENFILE or errno == ENOMEM){
 
735
      error(0, errno, "fopen");
 
736
      exitstatus = EX_OSERR;
 
737
      goto fallback;
472
738
    }
473
739
  }
474
 
 
 
740
  /* If there were any arguments from the configuration file, pass
 
741
     them to parser as command line arguments */
475
742
  if(custom_argv != NULL){
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;
 
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);
482
795
    }
483
796
  }
484
797
  
485
798
  if(debug){
486
 
    for(plugin *p = plugin_list; p != NULL; p=p->next){
 
799
    for(plugin *p = plugin_list; p != NULL; p = p->next){
487
800
      fprintf(stderr, "Plugin: %s has %d arguments\n",
488
801
              p->name ? p->name : "Global", p->argc - 1);
489
802
      for(char **a = p->argv; *a != NULL; a++){
490
803
        fprintf(stderr, "\tArg: %s\n", *a);
491
804
      }
492
 
      fprintf(stderr, "...and %u environment variables\n", p->envc);
 
805
      fprintf(stderr, "...and %d environment variables\n", p->envc);
493
806
      for(char **a = p->environ; *a != NULL; a++){
494
807
        fprintf(stderr, "\t%s\n", *a);
495
808
      }
496
809
    }
497
810
  }
498
811
  
 
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
  }
499
841
  ret = setuid(uid);
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 */
 
842
  if(ret == -1){
 
843
    error(0, errno, "setuid");
 
844
  }
 
845
  
 
846
  /* Open plugin directory with close_on_exec flag */
517
847
  {
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;
 
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;
525
892
      }
526
893
    }
 
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;
527
904
  }
528
905
  
529
906
  FD_ZERO(&rfds_all);
530
907
  
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)){
 
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)){
606
929
      if(debug){
607
 
        fprintf(stderr, "Ignoring plugin dir entry \"%s\""
608
 
                " with bad type or mode\n", filename);
 
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);
609
934
      }
610
 
      free(filename);
 
935
      close(plugin_fd);
 
936
      free(direntries[i]);
611
937
      continue;
612
938
    }
613
 
    plugin *p = getplugin(dirst->d_name, &plugin_list);
 
939
    
 
940
    plugin *p = getplugin(direntries[i]->d_name);
614
941
    if(p == NULL){
615
 
      perror("getplugin");
616
 
      free(filename);
 
942
      error(0, errno, "getplugin");
 
943
      close(plugin_fd);
 
944
      free(direntries[i]);
617
945
      continue;
618
946
    }
619
947
    if(p->disabled){
620
948
      if(debug){
621
949
        fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
622
 
                dirst->d_name);
 
950
                direntries[i]->d_name);
623
951
      }
624
 
      free(filename);
 
952
      close(plugin_fd);
 
953
      free(direntries[i]);
625
954
      continue;
626
955
    }
627
956
    {
628
957
      /* Add global arguments to argument list for this plugin */
629
 
      plugin *g = getplugin(NULL, &plugin_list);
 
958
      plugin *g = getplugin(NULL);
630
959
      if(g != NULL){
631
960
        for(char **a = g->argv + 1; *a != NULL; a++){
632
961
          if(not add_argument(p, *a)){
633
 
            perror("add_argument");
 
962
            error(0, errno, "add_argument");
634
963
          }
635
964
        }
636
965
        /* Add global environment variables */
637
966
        for(char **e = g->environ; *e != NULL; e++){
638
 
          if(not add_environment(p, *e)){
639
 
            perror("add_environment");
 
967
          if(not add_environment(p, *e, false)){
 
968
            error(0, errno, "add_environment");
640
969
          }
641
970
        }
642
971
      }
643
972
    }
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. */
 
973
    /* If this plugin has any environment variables, we need to
 
974
       duplicate the environment from this process, too. */
647
975
    if(p->environ[0] != NULL){
648
976
      for(char **e = environ; *e != NULL; e++){
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");
 
977
        if(not add_environment(p, *e, false)){
 
978
          error(0, errno, "add_environment");
656
979
        }
657
980
      }
658
981
    }
659
982
    
660
 
    int pipefd[2]; 
661
 
    ret = pipe(pipefd);
662
 
    if (ret == -1){
663
 
      perror("pipe");
664
 
      exitstatus = EXIT_FAILURE;
665
 
      goto end;
666
 
    }
 
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 */
667
1006
    ret = set_cloexec_flag(pipefd[0]);
668
1007
    if(ret < 0){
669
 
      perror("set_cloexec_flag");
670
 
      exitstatus = EXIT_FAILURE;
671
 
      goto end;
 
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;
672
1014
    }
673
1015
    ret = set_cloexec_flag(pipefd[1]);
674
1016
    if(ret < 0){
675
 
      perror("set_cloexec_flag");
676
 
      exitstatus = EXIT_FAILURE;
677
 
      goto end;
 
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;
678
1023
    }
 
1024
#endif  /* not O_CLOEXEC */
679
1025
    /* Block SIGCHLD until process is safely in process list */
680
 
    ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
 
1026
    ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
 
1027
                                              &sigchld_action.sa_mask,
 
1028
                                              NULL));
681
1029
    if(ret < 0){
682
 
      perror("sigprocmask");
683
 
      exitstatus = EXIT_FAILURE;
684
 
      goto end;
 
1030
      error(0, errno, "sigprocmask");
 
1031
      exitstatus = EX_OSERR;
 
1032
      free(direntries[i]);
 
1033
      goto fallback;
685
1034
    }
686
 
    // Starting a new process to be watched
687
 
    pid_t pid = fork();
 
1035
    /* Starting a new process to be watched */
 
1036
    pid_t pid;
 
1037
    do {
 
1038
      pid = fork();
 
1039
    } while(pid == -1 and errno == EINTR);
688
1040
    if(pid == -1){
689
 
      perror("fork");
690
 
      exitstatus = EXIT_FAILURE;
691
 
      goto end;
 
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;
692
1049
    }
693
1050
    if(pid == 0){
694
1051
      /* this is the child process */
695
1052
      ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
696
1053
      if(ret < 0){
697
 
        perror("sigaction");
698
 
        _exit(EXIT_FAILURE);
 
1054
        error(0, errno, "sigaction");
 
1055
        _exit(EX_OSERR);
699
1056
      }
700
 
      ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
 
1057
      ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
701
1058
      if(ret < 0){
702
 
        perror("sigprocmask");
703
 
        _exit(EXIT_FAILURE);
 
1059
        error(0, errno, "sigprocmask");
 
1060
        _exit(EX_OSERR);
704
1061
      }
705
 
 
 
1062
      
706
1063
      ret = dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */
707
1064
      if(ret == -1){
708
 
        perror("dup2");
709
 
        _exit(EXIT_FAILURE);
 
1065
        error(0, errno, "dup2");
 
1066
        _exit(EX_OSERR);
710
1067
      }
711
1068
      
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
 
        }
 
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);
727
1075
      }
728
1076
      /* no return */
729
1077
    }
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);
 
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)));
737
1087
      if(ret < 0){
738
 
        perror("sigprocmask");
 
1088
        error(0, errno, "sigprocmask");
739
1089
      }
740
 
      exitstatus = EXIT_FAILURE;
741
 
      goto end;
 
1090
      exitstatus = EX_OSERR;
 
1091
      free(direntries[i]);
 
1092
      goto fallback;
742
1093
    }
 
1094
    free(direntries[i]);
743
1095
    
744
 
    *new_process = (struct process){ .pid = pid,
745
 
                                     .fd = pipefd[0],
746
 
                                     .next = process_list };
747
 
    // List handling
748
 
    process_list = new_process;
 
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
 
749
1104
    /* Unblock SIGCHLD so signal handler can be run if this process
750
1105
       has already completed */
751
 
    ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
 
1106
    ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_UNBLOCK,
 
1107
                                              &sigchld_action.sa_mask,
 
1108
                                              NULL));
752
1109
    if(ret < 0){
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){
 
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){
787
1141
    fd_set rfds = rfds_all;
788
1142
    int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
789
 
    if (select_ret == -1){
790
 
      perror("select");
791
 
      exitstatus = EXIT_FAILURE;
792
 
      goto end;
 
1143
    if(select_ret == -1 and errno != EINTR){
 
1144
      error(0, errno, "select");
 
1145
      exitstatus = EX_OSERR;
 
1146
      goto fallback;
793
1147
    }
794
1148
    /* OK, now either a process completed, or something can be read
795
1149
       from one of them */
796
 
    for(process *proc = process_list; proc ; proc = proc->next){
 
1150
    for(plugin *proc = plugin_list; proc != NULL;){
797
1151
      /* Is this process completely done? */
798
 
      if(proc->eof and proc->completed){
 
1152
      if(proc->completed and proc->eof){
799
1153
        /* Only accept the plugin output if it exited cleanly */
800
1154
        if(not WIFEXITED(proc->status)
801
1155
           or WEXITSTATUS(proc->status) != 0){
802
1156
          /* Bad exit by plugin */
 
1157
          
803
1158
          if(debug){
804
1159
            if(WIFEXITED(proc->status)){
805
 
              fprintf(stderr, "Plugin %u exited with status %d\n",
806
 
                      (unsigned int) (proc->pid),
 
1160
              fprintf(stderr, "Plugin %s [%" PRIdMAX "] exited with"
 
1161
                      " status %d\n", proc->name,
 
1162
                      (intmax_t) (proc->pid),
807
1163
                      WEXITSTATUS(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));
 
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)));
815
1170
            }
816
1171
          }
 
1172
          
817
1173
          /* Remove the plugin */
818
1174
          FD_CLR(proc->fd, &rfds_all);
 
1175
          
819
1176
          /* Block signal while modifying process_list */
820
 
          ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
 
1177
          ret = (int)TEMP_FAILURE_RETRY(sigprocmask
 
1178
                                        (SIG_BLOCK,
 
1179
                                         &sigchld_action.sa_mask,
 
1180
                                         NULL));
821
1181
          if(ret < 0){
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
 
          }
 
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
          
839
1191
          /* We are done modifying process list, so unblock signal */
840
 
          ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,
841
 
                             NULL);
 
1192
          ret = (int)(TEMP_FAILURE_RETRY
 
1193
                      (sigprocmask(SIG_UNBLOCK,
 
1194
                                   &sigchld_action.sa_mask, NULL)));
842
1195
          if(ret < 0){
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;
 
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;
851
1206
        }
 
1207
        
852
1208
        /* This process exited nicely, so print its buffer */
853
 
 
854
 
        bool bret = print_out_password(proc->buffer, proc->buffer_length);
 
1209
        
 
1210
        bool bret = print_out_password(proc->buffer,
 
1211
                                       proc->buffer_length);
855
1212
        if(not bret){
856
 
          perror("print_out_password");
857
 
          exitstatus = EXIT_FAILURE;
 
1213
          error(0, errno, "print_out_password");
 
1214
          exitstatus = EX_IOERR;
858
1215
        }
859
 
        goto end;
 
1216
        goto fallback;
860
1217
      }
 
1218
      
861
1219
      /* This process has not completed.  Does it have any output? */
862
1220
      if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
863
1221
        /* This process had nothing to say at this time */
 
1222
        proc = proc->next;
864
1223
        continue;
865
1224
      }
866
1225
      /* Before reading, make the process' data buffer large enough */
867
1226
      if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
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;
 
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;
874
1233
        }
 
1234
        proc->buffer = new_buffer;
875
1235
        proc->buffer_size += BUFFER_SIZE;
876
1236
      }
877
1237
      /* Read from the process */
878
 
      ret = read(proc->fd, proc->buffer + proc->buffer_length,
879
 
                 BUFFER_SIZE);
880
 
      if(ret < 0){
 
1238
      sret = TEMP_FAILURE_RETRY(read(proc->fd,
 
1239
                                     proc->buffer
 
1240
                                     + proc->buffer_length,
 
1241
                                     BUFFER_SIZE));
 
1242
      if(sret < 0){
881
1243
        /* Read error from this process; ignore the error */
 
1244
        proc = proc->next;
882
1245
        continue;
883
1246
      }
884
 
      if(ret == 0){
 
1247
      if(sret == 0){
885
1248
        /* got EOF */
886
1249
        proc->eof = true;
887
1250
      } else {
888
 
        proc->buffer_length += (size_t) ret;
 
1251
        proc->buffer_length += (size_t) sret;
889
1252
      }
890
1253
    }
891
1254
  }
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 */
 
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 */
898
1263
    bool bret;
899
1264
    fprintf(stderr, "Going to fallback mode using getpass(3)\n");
900
1265
    char *passwordbuffer = getpass("Password: ");
901
 
    bret = print_out_password(passwordbuffer, strlen(passwordbuffer));
 
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);
902
1273
    if(not bret){
903
 
      perror("print_out_password");
904
 
      exitstatus = EXIT_FAILURE;
905
 
      goto end;
 
1274
      error(0, errno, "print_out_password");
 
1275
      exitstatus = EX_IOERR;
906
1276
    }
907
1277
  }
908
1278
  
909
1279
  /* Restore old signal handler */
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);
 
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");
921
1307
      }
922
1308
    }
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);
942
1309
  }
943
1310
  
944
1311
  /* Wait for any remaining child processes to terminate */
945
 
  do{
 
1312
  do {
946
1313
    ret = wait(NULL);
947
1314
  } while(ret >= 0);
948
1315
  if(errno != ECHILD){
949
 
    perror("wait");
 
1316
    error(0, errno, "wait");
950
1317
  }
951
1318
  
 
1319
  free_plugin_list();
 
1320
  
 
1321
  free(plugindir);
 
1322
  free(pluginhelperdir);
 
1323
  free(argfile);
 
1324
  
952
1325
  return exitstatus;
953
1326
}