/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: 2011-12-31 20:07:11 UTC
  • mfrom: (535.1.9 wireless-network-hook)
  • Revision ID: teddy@recompile.se-20111231200711-6dli3r8drftem57r
Merge new wireless network hook.  Fix bridge network hook to use
hardware addresses instead of interface names.  Implement and document
new "CONNECT" environment variable for network hooks.

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