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