/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
 
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY() */
25
 
 
 
25
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), getline(),
 
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>             /* strtok, strlen(), strcpy(),
55
 
                                   strcat() */
 
56
#include <string.h>             /* strsep, strlen(), asprintf(),
 
57
                                   strsignal(), strcmp(), strncmp() */
56
58
#include <errno.h>              /* errno */
57
59
#include <argp.h>               /* struct argp_option, struct
58
60
                                   argp_state, struct argp,
59
61
                                   argp_parse(), ARGP_ERR_UNKNOWN,
60
 
                                   ARGP_KEY_END, ARGP_KEY_ARG, error_t */
 
62
                                   ARGP_KEY_END, ARGP_KEY_ARG,
 
63
                                   error_t */
61
64
#include <signal.h>             /* struct sigaction, sigemptyset(),
62
65
                                   sigaddset(), sigaction(),
63
66
                                   sigprocmask(), SIG_BLOCK, SIGCHLD,
64
 
                                   SIG_UNBLOCK, kill() */
 
67
                                   SIG_UNBLOCK, kill(), sig_atomic_t
 
68
                                */
65
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() */
66
75
 
67
76
#define BUFFER_SIZE 256
68
77
 
69
 
const char *argp_program_version = "mandos-client 1.0";
70
 
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
71
 
 
72
 
struct process;
73
 
 
74
 
typedef struct process{
 
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>";
 
83
 
 
84
typedef struct plugin{
 
85
  char *name;                   /* can be NULL or any plugin name */
 
86
  char **argv;
 
87
  int argc;
 
88
  char **environ;
 
89
  int envc;
 
90
  bool disabled;
 
91
  
 
92
  /* Variables used for running processes*/
75
93
  pid_t pid;
76
94
  int fd;
77
95
  char *buffer;
78
96
  size_t buffer_size;
79
97
  size_t buffer_length;
80
98
  bool eof;
81
 
  bool completed;
 
99
  volatile sig_atomic_t completed;
82
100
  int status;
83
 
  struct process *next;
84
 
} process;
85
 
 
86
 
typedef struct plugin{
87
 
  char *name;                   /* can be NULL or any plugin name */
88
 
  char **argv;
89
 
  int argc;
90
 
  bool disabled;
91
101
  struct plugin *next;
92
102
} plugin;
93
103
 
94
 
static plugin *getplugin(char *name, plugin **plugin_list){
95
 
  for (plugin *p = *plugin_list; p != NULL; p = p->next){
96
 
    if ((p->name == name)
97
 
        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))){
98
113
      return p;
99
114
    }
100
115
  }
101
116
  /* Create a new plugin */
102
 
  plugin *new_plugin = malloc(sizeof(plugin));
103
 
  if (new_plugin == NULL){
104
 
    perror("malloc");
105
 
    exit(EXIT_FAILURE);
106
 
  }
107
 
  new_plugin->name = name;
108
 
  new_plugin->argv = malloc(sizeof(char *) * 2);
109
 
  if (new_plugin->argv == NULL){
110
 
    perror("malloc");
111
 
    exit(EXIT_FAILURE);
112
 
  }
113
 
  new_plugin->argv[0] = name;
 
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){
 
122
    return NULL;
 
123
  }
 
124
  char *copy_name = NULL;
 
125
  if(name != NULL){
 
126
    do {
 
127
      copy_name = strdup(name);
 
128
    } while(copy_name == NULL and errno == EINTR);
 
129
    if(copy_name == NULL){
 
130
      int e = errno;
 
131
      free(new_plugin);
 
132
      errno = e;
 
133
      return NULL;
 
134
    }
 
135
  }
 
136
  
 
137
  *new_plugin = (plugin){ .name = copy_name,
 
138
                          .argc = 1,
 
139
                          .disabled = false,
 
140
                          .next = plugin_list };
 
141
  
 
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;
 
147
    free(copy_name);
 
148
    free(new_plugin);
 
149
    errno = e;
 
150
    return NULL;
 
151
  }
 
152
  new_plugin->argv[0] = copy_name;
114
153
  new_plugin->argv[1] = NULL;
115
 
  new_plugin->argc = 1;
116
 
  new_plugin->disabled = false;
117
 
  new_plugin->next = *plugin_list;
 
154
  
 
155
  do {
 
156
    new_plugin->environ = malloc(sizeof(char *));
 
157
  } while(new_plugin->environ == NULL and errno == EINTR);
 
158
  if(new_plugin->environ == NULL){
 
159
    int e = errno;
 
160
    free(copy_name);
 
161
    free(new_plugin->argv);
 
162
    free(new_plugin);
 
163
    errno = e;
 
164
    return NULL;
 
165
  }
 
166
  new_plugin->environ[0] = NULL;
 
167
  
118
168
  /* Append the new plugin to the list */
119
 
  *plugin_list = new_plugin;
 
169
  plugin_list = new_plugin;
120
170
  return new_plugin;
121
171
}
122
172
 
123
 
static void addargument(plugin *p, char *arg){
124
 
  p->argv[p->argc] = arg;
125
 
  p->argv = realloc(p->argv, sizeof(char *) * (size_t)(p->argc + 2));
126
 
  if (p->argv == NULL){
127
 
    perror("malloc");
128
 
    exit(EXIT_FAILURE);
129
 
  }
130
 
  p->argc++;
131
 
  p->argv[p->argc] = NULL;
 
173
/* Helper function for add_argument and add_environment */
 
174
__attribute__((nonnull))
 
175
static bool add_to_char_array(const char *new, char ***array,
 
176
                              int *len){
 
177
  /* Resize the pointed-to array to hold one more pointer */
 
178
  do {
 
179
    *array = realloc(*array, sizeof(char *)
 
180
                     * (size_t) ((*len) + 2));
 
181
  } while(*array == NULL and errno == EINTR);
 
182
  /* Malloc check */
 
183
  if(*array == NULL){
 
184
    return false;
 
185
  }
 
186
  /* Make a copy of the new string */
 
187
  char *copy;
 
188
  do {
 
189
    copy = strdup(new);
 
190
  } while(copy == NULL and errno == EINTR);
 
191
  if(copy == NULL){
 
192
    return false;
 
193
  }
 
194
  /* Insert the copy */
 
195
  (*array)[*len] = copy;
 
196
  (*len)++;
 
197
  /* Add a new terminating NULL pointer to the last element */
 
198
  (*array)[*len] = NULL;
 
199
  return true;
 
200
}
 
201
 
 
202
/* Add to a plugin's argument vector */
 
203
__attribute__((nonnull(2)))
 
204
static bool add_argument(plugin *p, const char *arg){
 
205
  if(p == NULL){
 
206
    return false;
 
207
  }
 
208
  return add_to_char_array(arg, &(p->argv), &(p->argc));
 
209
}
 
210
 
 
211
/* Add to a plugin's environment */
 
212
__attribute__((nonnull(2)))
 
213
static bool add_environment(plugin *p, const char *def, bool replace){
 
214
  if(p == NULL){
 
215
    return false;
 
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
  }
 
237
  return add_to_char_array(def, &(p->environ), &(p->envc));
132
238
}
133
239
 
134
240
/*
135
241
 * Based on the example in the GNU LibC manual chapter 13.13 "File
136
242
 * Descriptor Flags".
137
 
 * *Note File Descriptor Flags:(libc)Descriptor Flags.
 
243
 | [[info:libc:Descriptor%20Flags][File Descriptor Flags]] |
138
244
 */
139
 
static int set_cloexec_flag(int fd)
140
 
{
141
 
  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));
142
247
  /* If reading the flags failed, return error indication now. */
143
248
  if(ret < 0){
144
249
    return ret;
145
250
  }
146
251
  /* Store modified flag word in the descriptor. */
147
 
  return fcntl(fd, F_SETFD, ret | FD_CLOEXEC);
 
252
  return (int)TEMP_FAILURE_RETRY(fcntl(fd, F_SETFD,
 
253
                                       ret | FD_CLOEXEC));
148
254
}
149
255
 
150
 
process *process_list = NULL;
151
256
 
152
 
/* Mark a process as completed when it exits, and save its exit
 
257
/* Mark processes as completed when they exit, and save their exit
153
258
   status. */
154
 
void handle_sigchld(__attribute__((unused)) int sig){
155
 
  process *proc = process_list;
156
 
  int status;
157
 
  pid_t pid = wait(&status);
158
 
  if(pid == -1){
159
 
    perror("wait");
160
 
    return;
161
 
  }
162
 
  while(proc != NULL and proc->pid != pid){
163
 
    proc = proc->next;
164
 
  }
165
 
  if(proc == NULL){
166
 
    /* Process not found in process list */
167
 
    return;
168
 
  }
169
 
  proc->status = status;
170
 
  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;
171
289
}
172
290
 
173
 
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){
174
294
  ssize_t ret;
175
 
  if(length>0 and buffer[length-1] == '\n'){
176
 
    length--;
177
 
  }
178
295
  for(size_t written = 0; written < length; written += (size_t)ret){
179
296
    ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO, buffer + written,
180
297
                                   length - written));
185
302
  return true;
186
303
}
187
304
 
 
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
      }
 
330
    }
 
331
  }
 
332
  
 
333
  free(plugin_node);
 
334
}
 
335
 
 
336
static void free_plugin_list(void){
 
337
  while(plugin_list != NULL){
 
338
    free_plugin(plugin_list);
 
339
  }
 
340
}
 
341
 
188
342
int main(int argc, char *argv[]){
189
 
  const char *plugindir = "/conf/conf.d/mandos/plugins.d";
 
343
  char *plugindir = NULL;
 
344
  char *argfile = NULL;
 
345
  FILE *conffp;
190
346
  size_t d_name_len;
191
347
  DIR *dir = NULL;
192
348
  struct dirent *dirst;
193
349
  struct stat st;
194
350
  fd_set rfds_all;
195
351
  int ret, maxfd = 0;
 
352
  ssize_t sret;
196
353
  uid_t uid = 65534;
197
354
  gid_t gid = 65534;
198
355
  bool debug = false;
200
357
  struct sigaction old_sigchld_action;
201
358
  struct sigaction sigchld_action = { .sa_handler = handle_sigchld,
202
359
                                      .sa_flags = SA_NOCLDSTOP };
203
 
  char *plus_options = NULL;
204
 
  char **plus_argv = NULL;
205
 
 
206
 
  errno = 0;
 
360
  char **custom_argv = NULL;
 
361
  int custom_argc = 0;
207
362
  
208
363
  /* Establish a signal handler */
209
364
  sigemptyset(&sigchld_action.sa_mask);
210
365
  ret = sigaddset(&sigchld_action.sa_mask, SIGCHLD);
211
 
  if(ret < 0){
212
 
    perror("sigaddset");
213
 
    exit(EXIT_FAILURE);
 
366
  if(ret == -1){
 
367
    error(0, errno, "sigaddset");
 
368
    exitstatus = EX_OSERR;
 
369
    goto fallback;
214
370
  }
215
371
  ret = sigaction(SIGCHLD, &sigchld_action, &old_sigchld_action);
216
 
  if(ret < 0){
217
 
    perror("sigaction");
218
 
    exit(EXIT_FAILURE);
 
372
  if(ret == -1){
 
373
    error(0, errno, "sigaction");
 
374
    exitstatus = EX_OSERR;
 
375
    goto fallback;
219
376
  }
220
377
  
221
378
  /* The options we understand. */
223
380
    { .name = "global-options", .key = 'g',
224
381
      .arg = "OPTION[,OPTION[,...]]",
225
382
      .doc = "Options passed to all plugins" },
 
383
    { .name = "global-env", .key = 'G',
 
384
      .arg = "VAR=value",
 
385
      .doc = "Environment variable passed to all plugins" },
226
386
    { .name = "options-for", .key = 'o',
227
387
      .arg = "PLUGIN:OPTION[,OPTION[,...]]",
228
388
      .doc = "Options passed only to specified plugin" },
 
389
    { .name = "env-for", .key = 'E',
 
390
      .arg = "PLUGIN:ENV=value",
 
391
      .doc = "Environment variable passed to specified plugin" },
229
392
    { .name = "disable", .key = 'd',
230
393
      .arg = "PLUGIN",
231
394
      .doc = "Disable a specific plugin", .group = 1 },
 
395
    { .name = "enable", .key = 'e',
 
396
      .arg = "PLUGIN",
 
397
      .doc = "Enable a specific plugin", .group = 1 },
232
398
    { .name = "plugin-dir", .key = 128,
233
399
      .arg = "DIRECTORY",
234
400
      .doc = "Specify a different plugin directory", .group = 2 },
235
 
    { .name = "userid", .key = 129,
236
 
      .arg = "ID", .flags = 0,
237
 
      .doc = "User ID the plugins will run as", .group = 2 },
238
 
    { .name = "groupid", .key = 130,
239
 
      .arg = "ID", .flags = 0,
240
 
      .doc = "Group ID the plugins will run as", .group = 2 },
241
 
    { .name = "debug", .key = 131,
242
 
      .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 },
243
421
    { .name = NULL }
244
422
  };
245
423
  
246
 
  error_t parse_opt (int key, char *arg, struct argp_state *state) {
247
 
    /* Get the INPUT argument from `argp_parse', which we know is a
248
 
       pointer to our plugin list pointer. */
249
 
    plugin **plugins = state->input;
250
 
    switch (key) {
251
 
    case 'g':
252
 
      if (arg != NULL){
253
 
        char *p = strtok(arg, ",");
254
 
        do{
255
 
          addargument(getplugin(NULL, plugins), p);
256
 
          p = strtok(NULL, ",");
257
 
        } while (p != NULL);
258
 
      }
259
 
      break;
260
 
    case 'o':
261
 
      if (arg != NULL){
262
 
        char *name = strtok(arg, ":");
263
 
        char *p = strtok(NULL, ":");
264
 
        if(p != NULL){
265
 
          p = strtok(p, ",");
266
 
          do{
267
 
            addargument(getplugin(name, plugins), p);
268
 
            p = strtok(NULL, ",");
269
 
          } while (p != NULL);
270
 
        }
271
 
      }
272
 
      break;
273
 
    case 'd':
274
 
      if (arg != NULL){
275
 
        getplugin(arg, plugins)->disabled = true;
276
 
      }
277
 
      break;
278
 
    case 128:
279
 
      plugindir = arg;
280
 
      break;
281
 
    case 129:
282
 
      uid = (uid_t)strtol(arg, NULL, 10);
283
 
      break;
284
 
    case 130:
285
 
      gid = (gid_t)strtol(arg, NULL, 10);
286
 
      break;
287
 
    case 131:
 
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 */
 
467
      {
 
468
        char *envdef = strchr(arg, ':');
 
469
        if(envdef == NULL){
 
470
          argp_error(state, "No colon in \"%s\"", arg);
 
471
          errno = EINVAL;
 
472
          break;
 
473
        }
 
474
        *envdef = '\0';
 
475
        envdef++;
 
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 */
288
528
      debug = true;
289
529
      break;
 
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
 */
290
548
    case ARGP_KEY_ARG:
291
 
      if(plus_options != NULL or arg == NULL or arg[0] != '+'){
292
 
        argp_usage (state);
 
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;
293
554
      }
294
 
      plus_options = arg;
295
 
      break;
296
 
    case ARGP_KEY_END:
297
 
      break;
298
 
    default:
299
 
      return ARGP_ERR_UNKNOWN;
300
 
    }
301
 
    return 0;
302
 
  }
303
 
  
304
 
  plugin *plugin_list = NULL;
305
 
  
306
 
  struct argp argp = { .options = options, .parser = parse_opt,
307
 
                       .args_doc = "[+PLUS_SEPARATED_OPTIONS]",
 
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 = "",
308
597
                       .doc = "Mandos plugin runner -- Run plugins" };
309
598
  
310
 
  ret = argp_parse (&argp, argc, argv, 0, 0, &plugin_list);
311
 
  if (ret == ARGP_ERR_UNKNOWN){
312
 
    fprintf(stderr, "Unknown error while parsing arguments\n");
313
 
    exitstatus = EXIT_FAILURE;
314
 
    goto end;
315
 
  }
316
 
  
317
 
  if(plus_options){
318
 
    /* This is a mangled argument in the form of
319
 
     "+--option+--other-option=parameter+--yet-another-option", etc */
320
 
    /* Make new argc and argv vars, and call argp_parse() again. */
321
 
    plus_options++;             /* skip the first '+' character */
322
 
    const char delims[] = "+";
323
 
    char *arg;
324
 
    int new_argc = 1;
325
 
    plus_argv = malloc(sizeof(char*) * 2);
326
 
    if(plus_argv == NULL){
327
 
      perror("malloc");
328
 
      exitstatus = EXIT_FAILURE;
329
 
      goto end;
330
 
    }
331
 
    plus_argv[0] = argv[0];
332
 
    plus_argv[1] = NULL;
333
 
    arg = strtok(plus_options, delims); /* Get first argument */
334
 
    while(arg != NULL){
335
 
      new_argc++;
336
 
      plus_argv = realloc(plus_argv, sizeof(char *)
337
 
                         * ((unsigned int) new_argc + 1));
338
 
      if(plus_argv == NULL){
339
 
        perror("realloc");
340
 
        exitstatus = EXIT_FAILURE;
341
 
        goto end;
342
 
      }
343
 
      plus_argv[new_argc-1] = arg;
344
 
      plus_argv[new_argc] = NULL;
345
 
      arg = strtok(NULL, delims); /* Get next argument */
346
 
    }
347
 
    ret = argp_parse (&argp, new_argc, plus_argv, 0, 0, &plugin_list);
348
 
    if (ret == ARGP_ERR_UNKNOWN){
349
 
      fprintf(stderr, "Unknown error while parsing arguments\n");
350
 
      exitstatus = EXIT_FAILURE;
351
 
      goto end;
352
 
    }
 
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
  }
 
627
  if(conffp != NULL){
 
628
    char *org_line = NULL;
 
629
    char *p, *arg, *new_arg, *line;
 
630
    size_t size = 0;
 
631
    const char whitespace_delims[] = " \r\t\f\v\n";
 
632
    const char comment_delim[] = "#";
 
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 */
 
646
    while(true){
 
647
      sret = getline(&org_line, &size, conffp);
 
648
      if(sret == -1){
 
649
        break;
 
650
      }
 
651
      
 
652
      line = org_line;
 
653
      arg = strsep(&line, comment_delim);
 
654
      while((p = strsep(&arg, whitespace_delims)) != NULL){
 
655
        if(p[0] == '\0'){
 
656
          continue;
 
657
        }
 
658
        new_arg = strdup(p);
 
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;
 
677
      }
 
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
    }
 
687
    free(org_line);
 
688
  } else {
 
689
    /* Check for harmful errors and go to fallback. Other errors might
 
690
       not affect opening plugins */
 
691
    if(errno == EMFILE or errno == ENFILE or errno == ENOMEM){
 
692
      error(0, errno, "fopen");
 
693
      exitstatus = EX_OSERR;
 
694
      goto fallback;
 
695
    }
 
696
  }
 
697
  /* If there were any arguments from the configuration file, pass
 
698
     them to parser as command line arguments */
 
699
  if(custom_argv != NULL){
 
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;
 
714
      goto fallback;
 
715
    }
 
716
  }
 
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;
353
735
  }
354
736
  
355
737
  if(debug){
359
741
      for(char **a = p->argv; *a != NULL; a++){
360
742
        fprintf(stderr, "\tArg: %s\n", *a);
361
743
      }
362
 
    }
363
 
  }
364
 
  
 
744
      fprintf(stderr, "...and %d environment variables\n", p->envc);
 
745
      for(char **a = p->environ; *a != NULL; a++){
 
746
        fprintf(stderr, "\t%s\n", *a);
 
747
      }
 
748
    }
 
749
  }
 
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
  }
365
778
  ret = setuid(uid);
366
 
  if (ret == -1){
367
 
    perror("setuid");
368
 
  }
369
 
  
370
 
  setgid(gid);
371
 
  if (ret == -1){
372
 
    perror("setgid");
373
 
  }
374
 
  
375
 
  dir = opendir(plugindir);
376
 
  if(dir == NULL){
377
 
    perror("Could not open plugin dir");
378
 
    exitstatus = EXIT_FAILURE;
379
 
    goto end;
380
 
  }
381
 
  
382
 
  /* 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 */
383
784
  {
384
 
    int dir_fd = dirfd(dir);
385
 
    if(dir_fd >= 0){
386
 
      ret = set_cloexec_flag(dir_fd);
387
 
      if(ret < 0){
388
 
        perror("set_cloexec_flag");
389
 
        exitstatus = EXIT_FAILURE;
390
 
        goto end;
391
 
      }
 
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;
392
826
    }
393
827
  }
394
828
  
395
829
  FD_ZERO(&rfds_all);
396
830
  
 
831
  /* Read and execute any executable in the plugin directory*/
397
832
  while(true){
398
 
    dirst = readdir(dir);
 
833
    do {
 
834
      dirst = readdir(dir);
 
835
    } while(dirst == NULL and errno == EINTR);
399
836
    
400
 
    // All directory entries have been processed
 
837
    /* All directory entries have been processed */
401
838
    if(dirst == NULL){
402
 
      if (errno == EBADF){
403
 
        perror("readdir");
404
 
        exitstatus = EXIT_FAILURE;
405
 
        goto end;
 
839
      if(errno == EBADF){
 
840
        error(0, errno, "readdir");
 
841
        exitstatus = EX_IOERR;
 
842
        goto fallback;
406
843
      }
407
844
      break;
408
845
    }
409
846
    
410
847
    d_name_len = strlen(dirst->d_name);
411
848
    
412
 
    // Ignore dotfiles, backup files and other junk
 
849
    /* Ignore dotfiles, backup files and other junk */
413
850
    {
414
851
      bool bad_name = false;
415
852
      
417
854
      
418
855
      const char const *bad_suffixes[] = { "~", "#", ".dpkg-new",
419
856
                                           ".dpkg-old",
 
857
                                           ".dpkg-bak",
420
858
                                           ".dpkg-divert", NULL };
421
859
      for(const char **pre = bad_prefixes; *pre != NULL; pre++){
422
860
        size_t pre_len = strlen(*pre);
430
868
          break;
431
869
        }
432
870
      }
433
 
      
434
871
      if(bad_name){
435
872
        continue;
436
873
      }
437
 
      
438
874
      for(const char **suf = bad_suffixes; *suf != NULL; suf++){
439
875
        size_t suf_len = strlen(*suf);
440
876
        if((d_name_len >= suf_len)
441
 
           and (strcmp((dirst->d_name)+d_name_len-suf_len, *suf)
 
877
           and (strcmp((dirst->d_name) + d_name_len-suf_len, *suf)
442
878
                == 0)){
443
879
          if(debug){
444
880
            fprintf(stderr, "Ignoring plugin dir entry \"%s\""
454
890
      }
455
891
    }
456
892
    
457
 
    char *filename = malloc(d_name_len + strlen(plugindir) + 2);
458
 
    if (filename == NULL){
459
 
      perror("malloc");
460
 
      exitstatus = EXIT_FAILURE;
461
 
      goto end;
462
 
    }
463
 
    strcpy(filename, plugindir); /* Spurious warning */
464
 
    strcat(filename, "/");      /* Spurious warning */
465
 
    strcat(filename, dirst->d_name); /* Spurious warning */
466
 
    
467
 
    ret = stat(filename, &st);
468
 
    if (ret == -1){
469
 
      perror("stat");
470
 
      exitstatus = EXIT_FAILURE;
471
 
      goto end;
472
 
    }
473
 
    
474
 
    if (not S_ISREG(st.st_mode) or (access(filename, X_OK) != 0)){
 
893
    char *filename;
 
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
    }
 
902
    if(ret < 0){
 
903
      error(0, errno, "asprintf");
 
904
      continue;
 
905
    }
 
906
    
 
907
    ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
 
908
    if(ret == -1){
 
909
      error(0, errno, "stat");
 
910
      free(filename);
 
911
      continue;
 
912
    }
 
913
    
 
914
    /* Ignore non-executable files */
 
915
    if(not S_ISREG(st.st_mode)
 
916
       or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
475
917
      if(debug){
476
918
        fprintf(stderr, "Ignoring plugin dir entry \"%s\""
477
919
                " with bad type or mode\n", filename);
479
921
      free(filename);
480
922
      continue;
481
923
    }
482
 
    if(getplugin(dirst->d_name, &plugin_list)->disabled){
 
924
    
 
925
    plugin *p = getplugin(dirst->d_name);
 
926
    if(p == NULL){
 
927
      error(0, errno, "getplugin");
 
928
      free(filename);
 
929
      continue;
 
930
    }
 
931
    if(p->disabled){
483
932
      if(debug){
484
933
        fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
485
934
                dirst->d_name);
487
936
      free(filename);
488
937
      continue;
489
938
    }
490
 
    plugin *p = getplugin(dirst->d_name, &plugin_list);
491
939
    {
492
940
      /* Add global arguments to argument list for this plugin */
493
 
      plugin *g = getplugin(NULL, &plugin_list);
494
 
      for(char **a = g->argv + 1; *a != NULL; a++){
495
 
        addargument(p, *a);
496
 
      }
497
 
    }
498
 
    int pipefd[2]; 
499
 
    ret = pipe(pipefd);
500
 
    if (ret == -1){
501
 
      perror("pipe");
502
 
      exitstatus = EXIT_FAILURE;
503
 
      goto end;
504
 
    }
 
941
      plugin *g = getplugin(NULL);
 
942
      if(g != NULL){
 
943
        for(char **a = g->argv + 1; *a != NULL; a++){
 
944
          if(not add_argument(p, *a)){
 
945
            error(0, errno, "add_argument");
 
946
          }
 
947
        }
 
948
        /* Add global environment variables */
 
949
        for(char **e = g->environ; *e != NULL; e++){
 
950
          if(not add_environment(p, *e, false)){
 
951
            error(0, errno, "add_environment");
 
952
          }
 
953
        }
 
954
      }
 
955
    }
 
956
    /* If this plugin has any environment variables, we will call
 
957
       using execve and need to duplicate the environment from this
 
958
       process, too. */
 
959
    if(p->environ[0] != NULL){
 
960
      for(char **e = environ; *e != NULL; e++){
 
961
        if(not add_environment(p, *e, false)){
 
962
          error(0, errno, "add_environment");
 
963
        }
 
964
      }
 
965
    }
 
966
    
 
967
    int pipefd[2];
 
968
    ret = (int)TEMP_FAILURE_RETRY(pipe(pipefd));
 
969
    if(ret == -1){
 
970
      error(0, errno, "pipe");
 
971
      exitstatus = EX_OSERR;
 
972
      goto fallback;
 
973
    }
 
974
    /* Ask OS to automatic close the pipe on exec */
505
975
    ret = set_cloexec_flag(pipefd[0]);
506
976
    if(ret < 0){
507
 
      perror("set_cloexec_flag");
508
 
      exitstatus = EXIT_FAILURE;
509
 
      goto end;
 
977
      error(0, errno, "set_cloexec_flag");
 
978
      exitstatus = EX_OSERR;
 
979
      goto fallback;
510
980
    }
511
981
    ret = set_cloexec_flag(pipefd[1]);
512
982
    if(ret < 0){
513
 
      perror("set_cloexec_flag");
514
 
      exitstatus = EXIT_FAILURE;
515
 
      goto end;
 
983
      error(0, errno, "set_cloexec_flag");
 
984
      exitstatus = EX_OSERR;
 
985
      goto fallback;
516
986
    }
517
987
    /* Block SIGCHLD until process is safely in process list */
518
 
    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));
519
991
    if(ret < 0){
520
 
      perror("sigprocmask");
521
 
      exitstatus = EXIT_FAILURE;
522
 
      goto end;
523
 
    }
524
 
    // Starting a new process to be watched
525
 
    pid_t pid = fork();
 
992
      error(0, errno, "sigprocmask");
 
993
      exitstatus = EX_OSERR;
 
994
      goto fallback;
 
995
    }
 
996
    /* Starting a new process to be watched */
 
997
    pid_t pid;
 
998
    do {
 
999
      pid = fork();
 
1000
    } while(pid == -1 and errno == EINTR);
 
1001
    if(pid == -1){
 
1002
      error(0, errno, "fork");
 
1003
      exitstatus = EX_OSERR;
 
1004
      goto fallback;
 
1005
    }
526
1006
    if(pid == 0){
527
1007
      /* this is the child process */
528
1008
      ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
529
1009
      if(ret < 0){
530
 
        perror("sigaction");
531
 
        _exit(EXIT_FAILURE);
 
1010
        error(0, errno, "sigaction");
 
1011
        _exit(EX_OSERR);
532
1012
      }
533
 
      ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
 
1013
      ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
534
1014
      if(ret < 0){
535
 
        perror("sigprocmask");
536
 
        _exit(EXIT_FAILURE);
 
1015
        error(0, errno, "sigprocmask");
 
1016
        _exit(EX_OSERR);
537
1017
      }
538
 
 
 
1018
      
539
1019
      ret = dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */
540
1020
      if(ret == -1){
541
 
        perror("dup2");
542
 
        _exit(EXIT_FAILURE);
 
1021
        error(0, errno, "dup2");
 
1022
        _exit(EX_OSERR);
543
1023
      }
544
1024
      
545
1025
      if(dirfd(dir) < 0){
547
1027
           above and must now close it manually here. */
548
1028
        closedir(dir);
549
1029
      }
550
 
      if(execv(filename, p->argv) < 0){
551
 
        perror("execv");
552
 
        _exit(EXIT_FAILURE);
 
1030
      if(p->environ[0] == NULL){
 
1031
        if(execv(filename, p->argv) < 0){
 
1032
          error(0, errno, "execv for %s", filename);
 
1033
          _exit(EX_OSERR);
 
1034
        }
 
1035
      } else {
 
1036
        if(execve(filename, p->argv, p->environ) < 0){
 
1037
          error(0, errno, "execve for %s", filename);
 
1038
          _exit(EX_OSERR);
 
1039
        }
553
1040
      }
554
1041
      /* no return */
555
1042
    }
556
 
    /* parent process */
 
1043
    /* Parent process */
 
1044
    TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
 
1045
                                             pipe */
557
1046
    free(filename);
558
 
    close(pipefd[1]);           /* close unused write end of pipe */
559
 
    process *new_process = malloc(sizeof(process));
560
 
    if (new_process == NULL){
561
 
      perror("malloc");
562
 
      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)));
563
1053
      if(ret < 0){
564
 
        perror("sigprocmask");
 
1054
        error(0, errno, "sigprocmask");
565
1055
      }
566
 
      exitstatus = EXIT_FAILURE;
567
 
      goto end;
 
1056
      exitstatus = EX_OSERR;
 
1057
      goto fallback;
568
1058
    }
569
1059
    
570
 
    *new_process = (struct process){ .pid = pid,
571
 
                                     .fd = pipefd[0],
572
 
                                     .next = process_list };
573
 
    // List handling
574
 
    process_list = new_process;
 
1060
    new_plugin->pid = pid;
 
1061
    new_plugin->fd = pipefd[0];
 
1062
    
575
1063
    /* Unblock SIGCHLD so signal handler can be run if this process
576
1064
       has already completed */
577
 
    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));
578
1068
    if(ret < 0){
579
 
      perror("sigprocmask");
580
 
      exitstatus = EXIT_FAILURE;
581
 
      goto end;
582
 
    }
583
 
    
584
 
    FD_SET(new_process->fd, &rfds_all);
585
 
    
586
 
    if (maxfd < new_process->fd){
587
 
      maxfd = new_process->fd;
588
 
    }
589
 
    
590
 
  }
591
 
  
592
 
  /* Free the plugin list */
593
 
  for(plugin *next; plugin_list != NULL; plugin_list = next){
594
 
    next = plugin_list->next;
595
 
    free(plugin_list->argv);
596
 
    free(plugin_list);
597
 
  }
598
 
  
599
 
  closedir(dir);
 
1069
      error(0, errno, "sigprocmask");
 
1070
      exitstatus = EX_OSERR;
 
1071
      goto fallback;
 
1072
    }
 
1073
    
 
1074
    FD_SET(new_plugin->fd, &rfds_all); /* Spurious warning from
 
1075
                                          -Wconversion */
 
1076
    
 
1077
    if(maxfd < new_plugin->fd){
 
1078
      maxfd = new_plugin->fd;
 
1079
    }
 
1080
  }
 
1081
  
 
1082
  TEMP_FAILURE_RETRY(closedir(dir));
600
1083
  dir = NULL;
601
 
    
602
 
  if (process_list == NULL){
603
 
    fprintf(stderr, "No plugin processes started. Incorrect plugin"
604
 
            " directory?\n");
605
 
    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
    }
606
1095
  }
607
 
  while(process_list){
 
1096
  
 
1097
  /* Main loop while running plugins exist */
 
1098
  while(plugin_list){
608
1099
    fd_set rfds = rfds_all;
609
1100
    int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
610
 
    if (select_ret == -1){
611
 
      perror("select");
612
 
      exitstatus = EXIT_FAILURE;
613
 
      goto end;
 
1101
    if(select_ret == -1 and errno != EINTR){
 
1102
      error(0, errno, "select");
 
1103
      exitstatus = EX_OSERR;
 
1104
      goto fallback;
614
1105
    }
615
1106
    /* OK, now either a process completed, or something can be read
616
1107
       from one of them */
617
 
    for(process *proc = process_list; proc ; proc = proc->next){
 
1108
    for(plugin *proc = plugin_list; proc != NULL;){
618
1109
      /* Is this process completely done? */
619
 
      if(proc->eof and proc->completed){
 
1110
      if(proc->completed and proc->eof){
620
1111
        /* Only accept the plugin output if it exited cleanly */
621
1112
        if(not WIFEXITED(proc->status)
622
1113
           or WEXITSTATUS(proc->status) != 0){
623
1114
          /* Bad exit by plugin */
 
1115
          
624
1116
          if(debug){
625
1117
            if(WIFEXITED(proc->status)){
626
 
              fprintf(stderr, "Plugin %u exited with status %d\n",
627
 
                      (unsigned int) (proc->pid),
 
1118
              fprintf(stderr, "Plugin %s [%" PRIdMAX "] exited with"
 
1119
                      " status %d\n", proc->name,
 
1120
                      (intmax_t) (proc->pid),
628
1121
                      WEXITSTATUS(proc->status));
629
 
            } else if(WIFSIGNALED(proc->status)) {
630
 
              fprintf(stderr, "Plugin %u killed by signal %d\n",
631
 
                      (unsigned int) (proc->pid),
632
 
                      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)));
633
1128
            } else if(WCOREDUMP(proc->status)){
634
 
              fprintf(stderr, "Plugin %d dumped core\n",
635
 
                      (unsigned int) (proc->pid));
 
1129
              fprintf(stderr, "Plugin %s [%" PRIdMAX "] dumped"
 
1130
                      " core\n", proc->name, (intmax_t) (proc->pid));
636
1131
            }
637
1132
          }
 
1133
          
638
1134
          /* Remove the plugin */
639
 
          FD_CLR(proc->fd, &rfds_all);
 
1135
          FD_CLR(proc->fd, &rfds_all); /* Spurious warning from
 
1136
                                          -Wconversion */
 
1137
          
640
1138
          /* Block signal while modifying process_list */
641
 
          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));
642
1143
          if(ret < 0){
643
 
            perror("sigprocmask");
644
 
            exitstatus = EXIT_FAILURE;
645
 
            goto end;
646
 
          }
647
 
          /* Delete this process entry from the list */
648
 
          if(process_list == proc){
649
 
            /* First one - simple */
650
 
            process_list = proc->next;
651
 
          } else {
652
 
            /* Second one or later */
653
 
            for(process *p = process_list; p != NULL; p = p->next){
654
 
              if(p->next == proc){
655
 
                p->next = proc->next;
656
 
                break;
657
 
              }
658
 
            }
659
 
          }
 
1144
            error(0, errno, "sigprocmask");
 
1145
            exitstatus = EX_OSERR;
 
1146
            goto fallback;
 
1147
          }
 
1148
          
 
1149
          plugin *next_plugin = proc->next;
 
1150
          free_plugin(proc);
 
1151
          proc = next_plugin;
 
1152
          
660
1153
          /* We are done modifying process list, so unblock signal */
661
 
          ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,
662
 
                             NULL);
 
1154
          ret = (int)(TEMP_FAILURE_RETRY
 
1155
                      (sigprocmask(SIG_UNBLOCK,
 
1156
                                   &sigchld_action.sa_mask, NULL)));
663
1157
          if(ret < 0){
664
 
            perror("sigprocmask");
665
 
          }
666
 
          free(proc->buffer);
667
 
          free(proc);
668
 
          /* We deleted this process from the list, so we can't go
669
 
             proc->next.  Therefore, start over from the beginning of
670
 
             the process list */
671
 
          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;
672
1168
        }
 
1169
        
673
1170
        /* This process exited nicely, so print its buffer */
674
 
 
675
 
        bool bret = print_out_password(proc->buffer, proc->buffer_length);
 
1171
        
 
1172
        bool bret = print_out_password(proc->buffer,
 
1173
                                       proc->buffer_length);
676
1174
        if(not bret){
677
 
          perror("print_out_password");
678
 
          exitstatus = EXIT_FAILURE;
 
1175
          error(0, errno, "print_out_password");
 
1176
          exitstatus = EX_IOERR;
679
1177
        }
680
 
        goto end;
 
1178
        goto fallback;
681
1179
      }
 
1180
      
682
1181
      /* This process has not completed.  Does it have any output? */
683
 
      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 */
684
1185
        /* This process had nothing to say at this time */
 
1186
        proc = proc->next;
685
1187
        continue;
686
1188
      }
687
1189
      /* Before reading, make the process' data buffer large enough */
688
1190
      if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
689
1191
        proc->buffer = realloc(proc->buffer, proc->buffer_size
690
1192
                               + (size_t) BUFFER_SIZE);
691
 
        if (proc->buffer == NULL){
692
 
          perror("malloc");
693
 
          exitstatus = EXIT_FAILURE;
694
 
          goto end;
 
1193
        if(proc->buffer == NULL){
 
1194
          error(0, errno, "malloc");
 
1195
          exitstatus = EX_OSERR;
 
1196
          goto fallback;
695
1197
        }
696
1198
        proc->buffer_size += BUFFER_SIZE;
697
1199
      }
698
1200
      /* Read from the process */
699
 
      ret = read(proc->fd, proc->buffer + proc->buffer_length,
700
 
                 BUFFER_SIZE);
701
 
      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){
702
1206
        /* Read error from this process; ignore the error */
 
1207
        proc = proc->next;
703
1208
        continue;
704
1209
      }
705
 
      if(ret == 0){
 
1210
      if(sret == 0){
706
1211
        /* got EOF */
707
1212
        proc->eof = true;
708
1213
      } else {
709
 
        proc->buffer_length += (size_t) ret;
 
1214
        proc->buffer_length += (size_t) sret;
710
1215
      }
711
1216
    }
712
1217
  }
713
 
 
714
 
 
715
 
 end:
716
 
  
717
 
  if(process_list == NULL or exitstatus != EXIT_SUCCESS){
718
 
    /* Fallback if all plugins failed or an error occured */
 
1218
  
 
1219
  
 
1220
 fallback:
 
1221
  
 
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 */
719
1226
    bool bret;
720
1227
    fprintf(stderr, "Going to fallback mode using getpass(3)\n");
721
1228
    char *passwordbuffer = getpass("Password: ");
722
 
    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);
723
1236
    if(not bret){
724
 
      perror("print_out_password");
725
 
      exitstatus = EXIT_FAILURE;
726
 
      goto end;
 
1237
      error(0, errno, "print_out_password");
 
1238
      exitstatus = EX_IOERR;
727
1239
    }
728
1240
  }
729
1241
  
730
1242
  /* Restore old signal handler */
731
 
  sigaction(SIGCHLD, &old_sigchld_action, NULL);
732
 
  
733
 
  free(plus_argv);
734
 
  
735
 
  /* Free the plugin list */
736
 
  for(plugin *next; plugin_list != NULL; plugin_list = next){
737
 
    next = plugin_list->next;
738
 
    free(plugin_list->argv);
739
 
    free(plugin_list);
 
1243
  ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
 
1244
  if(ret == -1){
 
1245
    error(0, errno, "sigaction");
 
1246
    exitstatus = EX_OSERR;
 
1247
  }
 
1248
  
 
1249
  if(custom_argv != NULL){
 
1250
    for(char **arg = custom_argv+1; *arg != NULL; arg++){
 
1251
      free(*arg);
 
1252
    }
 
1253
    free(custom_argv);
740
1254
  }
741
1255
  
742
1256
  if(dir != NULL){
743
1257
    closedir(dir);
744
1258
  }
745
1259
  
746
 
  /* Free the process list and kill the processes */
747
 
  for(process *next; process_list != NULL; process_list = next){
748
 
    next = process_list->next;
749
 
    close(process_list->fd);
750
 
    ret = kill(process_list->pid, SIGTERM);
751
 
    if(ret == -1 and errno != ESRCH){
752
 
      /* set-uid proccesses migth not get closed */
753
 
      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
      }
754
1269
    }
755
 
    free(process_list->buffer);
756
 
    free(process_list);
757
1270
  }
758
1271
  
759
1272
  /* Wait for any remaining child processes to terminate */
760
 
  do{
 
1273
  do {
761
1274
    ret = wait(NULL);
762
1275
  } while(ret >= 0);
763
1276
  if(errno != ECHILD){
764
 
    perror("wait");
 
1277
    error(0, errno, "wait");
765
1278
  }
766
1279
  
 
1280
  free_plugin_list();
 
1281
  
 
1282
  free(plugindir);
 
1283
  free(argfile);
 
1284
  
767
1285
  return exitstatus;
768
1286
}