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