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