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 a 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){
183
if(p == NULL){
184
return false;
185
}
186
return add_to_char_array(def, &(p->environ), &(p->envc));
187
}
188
189
/*
190
* Based on the example in the GNU LibC manual chapter 13.13 "File
191
* Descriptor Flags".
192
* *Note File Descriptor Flags:(libc)Descriptor Flags.
193
*/
194
static int set_cloexec_flag(int fd)
195
{
196
int ret = fcntl(fd, F_GETFD, 0);
197
/* If reading the flags failed, return error indication now. */
198
if(ret < 0){
199
return ret;
200
}
201
/* Store modified flag word in the descriptor. */
202
return fcntl(fd, F_SETFD, ret | FD_CLOEXEC);
203
}
204
205
206
/* Mark processes as completed when they exit, and save their exit
207
status. */
208
void handle_sigchld(__attribute__((unused)) int sig){
209
while(true){
210
plugin *proc = plugin_list;
211
int status;
212
pid_t pid = waitpid(-1, &status, WNOHANG);
213
if(pid == 0){
214
/* Only still running child processes */
215
break;
216
}
217
if(pid == -1){
218
if (errno != ECHILD){
219
perror("waitpid");
220
}
221
/* No child processes */
222
break;
223
}
224
225
/* A child exited, find it in process_list */
226
while(proc != NULL and proc->pid != pid){
227
proc = proc->next;
228
}
229
if(proc == NULL){
230
/* Process not found in process list */
231
continue;
232
}
233
proc->status = status;
234
proc->completed = true;
235
}
236
}
237
238
/* Prints out a password to stdout */
239
bool print_out_password(const char *buffer, size_t length){
240
ssize_t ret;
241
if(length>0 and buffer[length-1] == '\n'){
242
length--;
243
}
244
for(size_t written = 0; written < length; written += (size_t)ret){
245
ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO, buffer + written,
246
length - written));
247
if(ret < 0){
248
return false;
249
}
250
}
251
return true;
252
}
253
254
/* Removes and free a plugin from the plugin list */
255
static void free_plugin(plugin *plugin_node){
256
257
for(char **arg = plugin_node->argv; *arg != NULL; arg++){
258
free(*arg);
259
}
260
free(plugin_node->argv);
261
for(char **env = plugin_node->environ; *env != NULL; env++){
262
free(*env);
263
}
264
free(plugin_node->environ);
265
free(plugin_node->buffer);
266
267
/* Removes the plugin from the singly-linked list */
268
if(plugin_node == plugin_list){
269
/* First one - simple */
270
plugin_list = plugin_list->next;
271
} else {
272
/* Second one or later */
273
for(plugin *p = plugin_list; p != NULL; p = p->next){
274
if(p->next == plugin_node){
275
p->next = plugin_node->next;
276
break;
277
}
278
}
279
}
280
281
free(plugin_node);
282
}
283
284
static void free_plugin_list(void){
285
while(plugin_list != NULL){
286
free_plugin(plugin_list);
287
}
288
}
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[] = "";
289
113
290
114
int main(int argc, char *argv[]){
291
char *plugindir = NULL;
292
char *argfile = NULL;
293
FILE *conffp;
115
const char *plugindir = "plugins.d";
294
116
size_t d_name_len;
295
DIR *dir = NULL;
117
DIR *dir;
296
118
struct dirent *dirst;
297
119
struct stat st;
298
fd_set rfds_all;
120
fd_set rfds_orig;
299
121
int ret, maxfd = 0;
300
uid_t uid = 65534;
301
gid_t gid = 65534;
302
bool debug = false;
303
int exitstatus = EXIT_SUCCESS;
304
struct sigaction old_sigchld_action;
305
struct sigaction sigchld_action = { .sa_handler = handle_sigchld,
306
.sa_flags = SA_NOCLDSTOP };
307
char **custom_argv = NULL;
308
int custom_argc = 0;
309
310
/* Establish a signal handler */
311
sigemptyset(&sigchld_action.sa_mask);
312
ret = sigaddset(&sigchld_action.sa_mask, SIGCHLD);
313
if(ret == -1){
314
perror("sigaddset");
315
exitstatus = EXIT_FAILURE;
316
goto fallback;
317
}
318
ret = sigaction(SIGCHLD, &sigchld_action, &old_sigchld_action);
319
if(ret == -1){
320
perror("sigaction");
321
exitstatus = EXIT_FAILURE;
322
goto fallback;
323
}
324
122
process *process_list = NULL;
123
325
124
/* The options we understand. */
326
125
struct argp_option options[] = {
327
126
{ .name = "global-options", .key = 'g',
328
.arg = "OPTION[,OPTION[,...]]",
329
.doc = "Options passed to all plugins" },
330
{ .name = "global-envs", .key = 'e',
331
.arg = "VAR=value",
332
.doc = "Environment variable passed to all plugins" },
127
.arg = "option[,option[,...]]", .flags = 0,
128
.doc = "Options effecting all plugins" },
333
129
{ .name = "options-for", .key = 'o',
334
.arg = "PLUGIN:OPTION[,OPTION[,...]]",
335
.doc = "Options passed only to specified plugin" },
336
{ .name = "envs-for", .key = 'f',
337
.arg = "PLUGIN:ENV=value",
338
.doc = "Environment variable passed to specified plugin" },
339
{ .name = "disable", .key = 'd',
340
.arg = "PLUGIN",
341
.doc = "Disable 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" },
342
135
{ .name = "plugin-dir", .key = 128,
343
.arg = "DIRECTORY",
344
.doc = "Specify a different plugin directory", .group = 2 },
345
{ .name = "config-file", .key = 129,
346
.arg = "FILE",
347
.doc = "Specify a different configuration file", .group = 2 },
348
{ .name = "userid", .key = 130,
349
.arg = "ID", .flags = 0,
350
.doc = "User ID the plugins will run as", .group = 3 },
351
{ .name = "groupid", .key = 131,
352
.arg = "ID", .flags = 0,
353
.doc = "Group ID the plugins will run as", .group = 3 },
354
{ .name = "debug", .key = 132,
355
.doc = "Debug mode", .group = 4 },
136
.arg = "Directory", .flags = 0,
137
.doc = "Option to change directory to search for plugins" },
356
138
{ .name = NULL }
357
139
};
358
140
359
error_t parse_opt (int key, char *arg, __attribute__((unused)) struct argp_state *state) {
360
/* Get the INPUT argument from `argp_parse', which we know is a
361
pointer to our plugin list pointer. */
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;
362
145
switch (key) {
363
case 'g': /* --global-options */
364
if (arg != NULL){
365
char *p;
366
while((p = strsep(&arg, ",")) != NULL){
367
if(p[0] == '\0'){
368
continue;
369
}
370
if(not add_argument(getplugin(NULL), p)){
371
perror("add_argument");
372
return ARGP_ERR_UNKNOWN;
373
}
374
}
375
}
376
break;
377
case 'e': /* --global-envs */
378
if(arg == NULL){
379
break;
380
}
381
{
382
char *envdef = strdup(arg);
383
if(envdef == NULL){
384
break;
385
}
386
if(not add_environment(getplugin(NULL), envdef)){
387
perror("add_environment");
388
}
389
}
390
break;
391
case 'o': /* --options-for */
392
if (arg != NULL){
393
char *p_name = strsep(&arg, ":");
394
if(p_name[0] == '\0'){
395
break;
396
}
397
char *opt = strsep(&arg, ":");
398
if(opt[0] == '\0'){
399
break;
400
}
401
if(opt != NULL){
402
char *p;
403
while((p = strsep(&opt, ",")) != NULL){
404
if(p[0] == '\0'){
405
continue;
406
}
407
if(not add_argument(getplugin(p_name), p)){
408
perror("add_argument");
409
return ARGP_ERR_UNKNOWN;
410
}
411
}
412
}
413
}
414
break;
415
case 'f': /* --envs-for */
416
if(arg == NULL){
417
break;
418
}
419
{
420
char *envdef = strchr(arg, ':');
421
if(envdef == NULL){
422
break;
423
}
424
char *p_name = strndup(arg, (size_t) (envdef-arg));
425
if(p_name == NULL){
426
break;
427
}
428
envdef++;
429
if(not add_environment(getplugin(p_name), envdef)){
430
perror("add_environment");
431
}
432
}
433
break;
434
case 'd': /* --disable */
435
if (arg != NULL){
436
plugin *p = getplugin(arg);
437
if(p == NULL){
438
return ARGP_ERR_UNKNOWN;
439
}
440
p->disabled = true;
441
}
442
break;
443
case 128: /* --plugin-dir */
444
plugindir = strdup(arg);
445
if(plugindir == NULL){
446
perror("strdup");
447
}
448
break;
449
case 129: /* --config-file */
450
argfile = strdup(arg);
451
if(argfile == NULL){
452
perror("strdup");
453
}
454
break;
455
case 130: /* --userid */
456
uid = (uid_t)strtol(arg, NULL, 10);
457
break;
458
case 131: /* --groupid */
459
gid = (gid_t)strtol(arg, NULL, 10);
460
break;
461
case 132: /* --debug */
462
debug = true;
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;
463
177
break;
464
178
case ARGP_KEY_ARG:
465
fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);
179
argp_usage (state);
466
180
break;
467
181
case ARGP_KEY_END:
468
182
break;
471
185
}
472
186
return 0;
473
187
}
188
189
plugin *plugin_list = NULL;
474
190
475
191
struct argp argp = { .options = options, .parser = parse_opt,
476
.args_doc = "[+PLUS_SEPARATED_OPTIONS]",
477
.doc = "Mandos plugin runner -- Run plugins" };
478
479
ret = argp_parse (&argp, argc, argv, 0, 0, NULL);
480
if (ret == ARGP_ERR_UNKNOWN){
481
fprintf(stderr, "Unknown error while parsing arguments\n");
482
exitstatus = EXIT_FAILURE;
483
goto fallback;
484
}
485
486
/* Opens the configfile if aviable */
487
if (argfile == NULL){
488
conffp = fopen(AFILE, "r");
489
} else {
490
conffp = fopen(argfile, "r");
491
}
492
if(conffp != NULL){
493
char *org_line = NULL;
494
char *p, *arg, *new_arg, *line;
495
size_t size = 0;
496
ssize_t sret;
497
const char whitespace_delims[] = " \r\t\f\v\n";
498
const char comment_delim[] = "#";
499
500
custom_argc = 1;
501
custom_argv = malloc(sizeof(char*) * 2);
502
if(custom_argv == NULL){
503
perror("malloc");
504
exitstatus = EXIT_FAILURE;
505
goto fallback;
506
}
507
custom_argv[0] = argv[0];
508
custom_argv[1] = NULL;
509
510
/* for each line in the config file, strip whitespace and ignore commented text */
511
while(true){
512
sret = getline(&org_line, &size, conffp);
513
if(sret == -1){
514
break;
515
}
516
517
line = org_line;
518
arg = strsep(&line, comment_delim);
519
while((p = strsep(&arg, whitespace_delims)) != NULL){
520
if(p[0] == '\0'){
521
continue;
522
}
523
new_arg = strdup(p);
524
if(new_arg == NULL){
525
perror("strdup");
526
exitstatus = EXIT_FAILURE;
527
free(org_line);
528
goto fallback;
529
}
530
531
custom_argc += 1;
532
custom_argv = realloc(custom_argv, sizeof(char *)
533
* ((unsigned int) custom_argc + 1));
534
if(custom_argv == NULL){
535
perror("realloc");
536
exitstatus = EXIT_FAILURE;
537
free(org_line);
538
goto fallback;
539
}
540
custom_argv[custom_argc-1] = new_arg;
541
custom_argv[custom_argc] = NULL;
542
}
543
}
544
free(org_line);
545
} else{
546
/* Check for harmful errors and go to fallback. Other errors might
547
not affect opening plugins */
548
if (errno == EMFILE or errno == ENFILE or errno == ENOMEM){
549
perror("fopen");
550
exitstatus = EXIT_FAILURE;
551
goto fallback;
552
}
553
}
554
/* If there was any arguments from configuration file,
555
pass them to parser as command arguments */
556
if(custom_argv != NULL){
557
ret = argp_parse (&argp, custom_argc, custom_argv, 0, 0, NULL);
558
if (ret == ARGP_ERR_UNKNOWN){
559
fprintf(stderr, "Unknown error while parsing arguments\n");
560
exitstatus = EXIT_FAILURE;
561
goto fallback;
562
}
563
}
564
565
if(debug){
566
for(plugin *p = plugin_list; p != NULL; p=p->next){
567
fprintf(stderr, "Plugin: %s has %d arguments\n",
568
p->name ? p->name : "Global", p->argc - 1);
569
for(char **a = p->argv; *a != NULL; a++){
570
fprintf(stderr, "\tArg: %s\n", *a);
571
}
572
fprintf(stderr, "...and %u environment variables\n", p->envc);
573
for(char **a = p->environ; *a != NULL; a++){
574
fprintf(stderr, "\t%s\n", *a);
575
}
576
}
577
}
578
579
/* Strip permissions down to nobody */
580
ret = setuid(uid);
581
if (ret == -1){
582
perror("setuid");
583
}
584
setgid(gid);
585
if (ret == -1){
586
perror("setgid");
587
}
588
589
if (plugindir == NULL){
590
dir = opendir(PDIR);
591
} else {
592
dir = opendir(plugindir);
593
}
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);
594
206
595
207
if(dir == NULL){
596
perror("Could not open plugin dir");
597
exitstatus = EXIT_FAILURE;
598
goto fallback;
599
}
600
601
/* Set the FD_CLOEXEC flag on the directory, if possible */
602
{
603
int dir_fd = dirfd(dir);
604
if(dir_fd >= 0){
605
ret = set_cloexec_flag(dir_fd);
606
if(ret < 0){
607
perror("set_cloexec_flag");
608
exitstatus = EXIT_FAILURE;
609
goto fallback;
610
}
611
}
612
}
613
614
FD_ZERO(&rfds_all);
615
616
/* 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
617
214
while(true){
618
215
dirst = readdir(dir);
619
216
620
217
// All directory entries have been processed
621
218
if(dirst == NULL){
622
if (errno == EBADF){
623
perror("readdir");
624
exitstatus = EXIT_FAILURE;
625
goto fallback;
626
}
627
219
break;
628
220
}
629
221
630
222
d_name_len = strlen(dirst->d_name);
631
223
632
// Ignore dotfiles, backup files and other junk
633
{
634
bool bad_name = false;
635
636
const char const *bad_prefixes[] = { ".", "#", NULL };
637
638
const char const *bad_suffixes[] = { "~", "#", ".dpkg-new",
639
".dpkg-old",
640
".dpkg-divert", NULL };
641
for(const char **pre = bad_prefixes; *pre != NULL; pre++){
642
size_t pre_len = strlen(*pre);
643
if((d_name_len >= pre_len)
644
and strncmp((dirst->d_name), *pre, pre_len) == 0){
645
if(debug){
646
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
647
" with bad prefix %s\n", dirst->d_name, *pre);
648
}
649
bad_name = true;
650
break;
651
}
652
}
653
if(bad_name){
654
continue;
655
}
656
for(const char **suf = bad_suffixes; *suf != NULL; suf++){
657
size_t suf_len = strlen(*suf);
658
if((d_name_len >= suf_len)
659
and (strcmp((dirst->d_name)+d_name_len-suf_len, *suf)
660
== 0)){
661
if(debug){
662
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
663
" with bad suffix %s\n", dirst->d_name, *suf);
664
}
665
bad_name = true;
666
break;
667
}
668
}
669
670
if(bad_name){
671
continue;
672
}
673
}
674
675
char *filename;
676
ret = asprintf(&filename, "%s/%s", plugindir, dirst->d_name);
677
if(ret < 0){
678
perror("asprintf");
679
continue;
680
}
681
682
ret = stat(filename, &st);
683
if (ret == -1){
684
perror("stat");
685
free(filename);
686
continue;
687
}
688
689
/* Ignore non-executable files */
690
if (not S_ISREG(st.st_mode) or (access(filename, X_OK) != 0)){
691
if(debug){
692
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
693
" with bad type or mode\n", filename);
694
}
695
free(filename);
696
continue;
697
}
698
699
plugin *p = getplugin(dirst->d_name);
700
if(p == NULL){
701
perror("getplugin");
702
free(filename);
703
continue;
704
}
705
if(p->disabled){
706
if(debug){
707
fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
708
dirst->d_name);
709
}
710
free(filename);
711
continue;
712
}
713
{
714
/* Add global arguments to argument list for this plugin */
715
plugin *g = getplugin(NULL);
716
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);
717
257
for(char **a = g->argv + 1; *a != NULL; a++){
718
if(not add_argument(p, *a)){
719
perror("add_argument");
720
}
721
}
722
/* Add global environment variables */
723
for(char **e = g->environ; *e != NULL; e++){
724
if(not add_environment(p, *e)){
725
perror("add_environment");
726
}
727
}
728
}
729
}
730
/* If this plugin has any environment variables, we will call
731
using execve and need to duplicate the environment from this
732
process, too. */
733
if(p->environ[0] != NULL){
734
for(char **e = environ; *e != NULL; e++){
735
char *copy = strdup(*e);
736
if(copy == NULL){
737
perror("strdup");
738
continue;
739
}
740
if(not add_environment(p, copy)){
741
perror("add_environment");
742
}
743
}
744
}
745
746
int pipefd[2];
747
ret = pipe(pipefd);
748
if (ret == -1){
749
perror("pipe");
750
exitstatus = EXIT_FAILURE;
751
goto fallback;
752
}
753
/* Ask OS to automatic close the pipe on exec */
754
ret = set_cloexec_flag(pipefd[0]);
755
if(ret < 0){
756
perror("set_cloexec_flag");
757
exitstatus = EXIT_FAILURE;
758
goto fallback;
759
}
760
ret = set_cloexec_flag(pipefd[1]);
761
if(ret < 0){
762
perror("set_cloexec_flag");
763
exitstatus = EXIT_FAILURE;
764
goto fallback;
765
}
766
/* Block SIGCHLD until process is safely in process list */
767
ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);
768
if(ret < 0){
769
perror("sigprocmask");
770
exitstatus = EXIT_FAILURE;
771
goto fallback;
772
}
773
// Starting a new process to be watched
774
pid_t pid = fork();
775
if(pid == -1){
776
perror("fork");
777
exitstatus = EXIT_FAILURE;
778
goto fallback;
779
}
780
if(pid == 0){
781
/* this is the child process */
782
ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
783
if(ret < 0){
784
perror("sigaction");
785
_exit(EXIT_FAILURE);
786
}
787
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
788
if(ret < 0){
789
perror("sigprocmask");
790
_exit(EXIT_FAILURE);
791
}
792
793
ret = dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */
794
if(ret == -1){
795
perror("dup2");
796
_exit(EXIT_FAILURE);
797
}
798
799
if(dirfd(dir) < 0){
800
/* If dir has no file descriptor, we could not set FD_CLOEXEC
801
above and must now close it manually here. */
802
closedir(dir);
803
}
804
if(p->environ[0] == NULL){
258
addarguments(p, *a);
259
}
805
260
if(execv(filename, p->argv) < 0){
806
perror("execv");
807
_exit(EXIT_FAILURE);
808
}
809
} else {
810
if(execve(filename, p->argv, p->environ) < 0){
811
perror("execve");
812
_exit(EXIT_FAILURE);
813
}
814
}
815
/* no return */
816
}
817
/* Parent process */
818
close(pipefd[1]); /* Close unused write end of pipe */
819
free(filename);
820
plugin *new_plugin = getplugin(dirst->d_name);
821
if (new_plugin == NULL){
822
perror("getplugin");
823
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
824
if(ret < 0){
825
perror("sigprocmask");
826
}
827
exitstatus = EXIT_FAILURE;
828
goto fallback;
829
}
830
831
new_plugin->pid = pid;
832
new_plugin->fd = pipefd[0];
833
834
/* Unblock SIGCHLD so signal handler can be run if this process
835
has already completed */
836
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);
837
if(ret < 0){
838
perror("sigprocmask");
839
exitstatus = EXIT_FAILURE;
840
goto fallback;
841
}
842
843
FD_SET(new_plugin->fd, &rfds_all);
844
845
if (maxfd < new_plugin->fd){
846
maxfd = new_plugin->fd;
847
}
848
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
}
849
286
}
850
287
851
288
closedir(dir);
852
dir = NULL;
853
854
for(plugin *p = plugin_list; p != NULL; p = p->next){
855
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
}
856
362
break;
857
363
}
858
if(p->next == NULL){
859
fprintf(stderr, "No plugin processes started. Incorrect plugin"
860
" directory?\n");
861
free_plugin_list();
862
}
863
}
864
865
/* Main loop while running plugins exist */
866
while(plugin_list){
867
fd_set rfds = rfds_all;
868
int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);
869
if (select_ret == -1){
870
perror("select");
871
exitstatus = EXIT_FAILURE;
872
goto fallback;
873
}
874
/* OK, now either a process completed, or something can be read
875
from one of them */
876
for(plugin *proc = plugin_list; proc != NULL; proc = proc->next){
877
/* Is this process completely done? */
878
if(proc->eof and proc->completed){
879
/* Only accept the plugin output if it exited cleanly */
880
if(not WIFEXITED(proc->status)
881
or WEXITSTATUS(proc->status) != 0){
882
/* Bad exit by plugin */
883
884
if(debug){
885
if(WIFEXITED(proc->status)){
886
fprintf(stderr, "Plugin %u exited with status %d\n",
887
(unsigned int) (proc->pid),
888
WEXITSTATUS(proc->status));
889
} else if(WIFSIGNALED(proc->status)) {
890
fprintf(stderr, "Plugin %u killed by signal %d\n",
891
(unsigned int) (proc->pid),
892
WTERMSIG(proc->status));
893
} else if(WCOREDUMP(proc->status)){
894
fprintf(stderr, "Plugin %d dumped core\n",
895
(unsigned int) (proc->pid));
896
}
897
}
898
899
/* Remove the plugin */
900
FD_CLR(proc->fd, &rfds_all);
901
902
/* Block signal while modifying process_list */
903
ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);
904
if(ret < 0){
905
perror("sigprocmask");
906
exitstatus = EXIT_FAILURE;
907
goto fallback;
908
}
909
free_plugin(proc);
910
/* We are done modifying process list, so unblock signal */
911
ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,
912
NULL);
913
if(ret < 0){
914
perror("sigprocmask");
915
exitstatus = EXIT_FAILURE;
916
goto fallback;
917
}
918
919
if(plugin_list == NULL){
920
break;
921
}
922
continue;
923
}
924
925
/* This process exited nicely, so print its buffer */
926
927
bool bret = print_out_password(proc->buffer,
928
proc->buffer_length);
929
if(not bret){
930
perror("print_out_password");
931
exitstatus = EXIT_FAILURE;
932
}
933
goto fallback;
934
}
935
936
/* This process has not completed. Does it have any output? */
937
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
938
/* This process had nothing to say at this time */
939
continue;
940
}
941
/* Before reading, make the process' data buffer large enough */
942
if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
943
proc->buffer = realloc(proc->buffer, proc->buffer_size
944
+ (size_t) BUFFER_SIZE);
945
if (proc->buffer == NULL){
946
perror("malloc");
947
exitstatus = EXIT_FAILURE;
948
goto fallback;
949
}
950
proc->buffer_size += BUFFER_SIZE;
951
}
952
/* Read from the process */
953
ret = read(proc->fd, proc->buffer + proc->buffer_length,
954
BUFFER_SIZE);
955
if(ret < 0){
956
/* Read error from this process; ignore the error */
957
continue;
958
}
959
if(ret == 0){
960
/* got EOF */
961
proc->eof = true;
962
} else {
963
proc->buffer_length += (size_t) ret;
964
}
965
}
966
}
967
968
969
fallback:
970
971
if(plugin_list == NULL or exitstatus != EXIT_SUCCESS){
972
/* Fallback if all plugins failed, none are found or an error
973
occured */
974
bool bret;
975
fprintf(stderr, "Going to fallback mode using getpass(3)\n");
976
char *passwordbuffer = getpass("Password: ");
977
bret = print_out_password(passwordbuffer, strlen(passwordbuffer));
978
if(not bret){
979
perror("print_out_password");
980
exitstatus = EXIT_FAILURE;
981
}
982
}
983
984
/* Restore old signal handler */
985
ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);
986
if(ret == -1){
987
perror("sigaction");
988
exitstatus = EXIT_FAILURE;
989
}
990
991
if(custom_argv != NULL){
992
for(char **arg = custom_argv+1; *arg != NULL; arg++){
993
free(*arg);
994
}
995
free(custom_argv);
996
}
997
998
if(dir != NULL){
999
closedir(dir);
1000
}
1001
1002
/* Free the process list and kill the processes */
1003
for(plugin *p = plugin_list; p != NULL; p = p->next){
1004
if(p->pid != 0){
1005
close(p->fd);
1006
ret = kill(p->pid, SIGTERM);
1007
if(ret == -1 and errno != ESRCH){
1008
/* Set-uid proccesses might not get closed */
1009
perror("kill");
1010
}
1011
}
1012
}
1013
1014
/* Wait for any remaining child processes to terminate */
1015
do{
1016
ret = wait(NULL);
1017
} while(ret >= 0);
1018
if(errno != ECHILD){
1019
perror("wait");
1020
}
1021
1022
free_plugin_list();
1023
1024
free(plugindir);
1025
free(argfile);
1026
1027
return exitstatus;
364
}
365
return EXIT_SUCCESS;
1028
366
}
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0