25
25
#define _GNU_SOURCE /* TEMP_FAILURE_RETRY(), getline(),
26
asprintf(), O_CLOEXEC */
27
27
#include <stddef.h> /* size_t, NULL */
28
28
#include <stdlib.h> /* malloc(), exit(), EXIT_SUCCESS,
30
30
#include <stdbool.h> /* bool, true, false */
31
31
#include <stdio.h> /* fileno(), fprintf(),
32
stderr, STDOUT_FILENO */
33
#include <sys/types.h> /* DIR, fdopendir(), stat(), struct
34
stat, waitpid(), WIFEXITED(),
35
WEXITSTATUS(), wait(), pid_t,
36
uid_t, gid_t, getuid(), getgid(),
32
stderr, STDOUT_FILENO, fclose() */
33
#include <sys/types.h> /* fstat(), struct stat, waitpid(),
34
WIFEXITED(), WEXITSTATUS(), wait(),
35
pid_t, uid_t, gid_t, getuid(),
38
37
#include <sys/select.h> /* fd_set, select(), FD_ZERO(),
39
38
FD_SET(), FD_ISSET(), FD_CLR */
40
39
#include <sys/wait.h> /* wait(), waitpid(), WIFEXITED(),
41
WEXITSTATUS(), WTERMSIG(),
43
#include <sys/stat.h> /* struct stat, stat(), S_ISREG() */
40
WEXITSTATUS(), WTERMSIG() */
41
#include <sys/stat.h> /* struct stat, fstat(), S_ISREG() */
44
42
#include <iso646.h> /* and, or, not */
45
#include <dirent.h> /* DIR, struct dirent, fdopendir(),
46
readdir(), closedir(), dirfd() */
47
#include <unistd.h> /* struct stat, stat(), S_ISREG(),
48
fcntl(), setuid(), setgid(),
49
F_GETFD, F_SETFD, FD_CLOEXEC,
50
access(), pipe(), fork(), close()
51
dup2(), STDOUT_FILENO, _exit(),
52
execv(), write(), read(),
43
#include <dirent.h> /* struct dirent, scandirat() */
44
#include <unistd.h> /* fcntl(), F_GETFD, F_SETFD,
45
FD_CLOEXEC, write(), STDOUT_FILENO,
46
struct stat, fstat(), close(),
47
setgid(), setuid(), S_ISREG(),
48
faccessat() pipe2(), fork(),
49
_exit(), dup2(), fexecve(), read()
54
51
#include <fcntl.h> /* fcntl(), F_GETFD, F_SETFD,
56
#include <string.h> /* strsep, strlen(), asprintf(),
57
strsignal(), strcmp(), strncmp() */
52
FD_CLOEXEC, openat(), scandirat(),
54
#include <string.h> /* strsep, strlen(), strsignal(),
55
strcmp(), strncmp() */
58
56
#include <errno.h> /* errno */
59
57
#include <argp.h> /* struct argp_option, struct
60
58
argp_state, struct argp,
777
char *pluginhelperenv;
779
ret = asprintf(&pluginhelperenv, "MANDOSPLUGINHELPERDIR=%s",
780
pluginhelperdir != NULL ? pluginhelperdir : PHDIR);
782
bret = add_environment(getplugin(NULL), pluginhelperenv, true);
784
if(ret == -1 or not bret){
785
error(0, errno, "Failed to set MANDOSPLUGINHELPERDIR"
786
" environment variable to \"%s\" for all plugins\n",
787
pluginhelperdir != NULL ? pluginhelperdir : PHDIR);
790
free(pluginhelperenv);
738
for(plugin *p = plugin_list; p != NULL; p=p->next){
795
for(plugin *p = plugin_list; p != NULL; p = p->next){
739
796
fprintf(stderr, "Plugin: %s has %d arguments\n",
740
797
p->name ? p->name : "Global", p->argc - 1);
741
798
for(char **a = p->argv; *a != NULL; a++){
811
859
ret = set_cloexec_flag(dir_fd);
813
861
error(0, errno, "set_cloexec_flag");
814
TEMP_FAILURE_RETRY(close(dir_fd));
815
862
exitstatus = EX_OSERR;
818
865
#endif /* O_CLOEXEC */
820
dir = fdopendir(dir_fd);
822
error(0, errno, "Could not open plugin dir");
823
TEMP_FAILURE_RETRY(close(dir_fd));
824
exitstatus = EX_OSERR;
868
int good_name(const struct dirent * const dirent){
869
const char * const patterns[] = { ".*", "#*#", "*~", "*.dpkg-new",
870
"*.dpkg-old", "*.dpkg-bak",
871
"*.dpkg-divert", NULL };
873
#pragma GCC diagnostic push
874
#pragma GCC diagnostic ignored "-Wcast-qual"
876
for(const char **pat = (const char **)patterns;
877
*pat != NULL; pat++){
879
#pragma GCC diagnostic pop
881
if(fnmatch(*pat, dirent->d_name, FNM_FILE_NAME | FNM_PERIOD)
884
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
885
" matching pattern %s\n", dirent->d_name, *pat);
893
int numplugins = scandirat(dir_fd, ".", &direntries, good_name,
895
if(numplugins == -1){
896
error(0, errno, "Could not scan plugin dir");
898
exitstatus = EX_OSERR;
829
902
FD_ZERO(&rfds_all);
831
904
/* Read and execute any executable in the plugin directory*/
834
dirst = readdir(dir);
835
} while(dirst == NULL and errno == EINTR);
837
/* All directory entries have been processed */
840
error(0, errno, "readdir");
841
exitstatus = EX_IOERR;
847
d_name_len = strlen(dirst->d_name);
849
/* Ignore dotfiles, backup files and other junk */
851
bool bad_name = false;
853
const char * const bad_prefixes[] = { ".", "#", NULL };
855
const char * const bad_suffixes[] = { "~", "#", ".dpkg-new",
858
".dpkg-divert", NULL };
859
#pragma GCC diagnostic push
860
#pragma GCC diagnostic ignored "-Wcast-qual"
861
for(const char **pre = (const char **)bad_prefixes;
862
*pre != NULL; pre++){
863
#pragma GCC diagnostic pop
864
size_t pre_len = strlen(*pre);
865
if((d_name_len >= pre_len)
866
and strncmp((dirst->d_name), *pre, pre_len) == 0){
868
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
869
" with bad prefix %s\n", dirst->d_name, *pre);
878
#pragma GCC diagnostic push
879
#pragma GCC diagnostic ignored "-Wcast-qual"
880
for(const char **suf = (const char **)bad_suffixes;
881
*suf != NULL; suf++){
882
#pragma GCC diagnostic pop
883
size_t suf_len = strlen(*suf);
884
if((d_name_len >= suf_len)
885
and (strcmp((dirst->d_name) + d_name_len-suf_len, *suf)
888
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
889
" with bad suffix %s\n", dirst->d_name, *suf);
902
if(plugindir == NULL){
903
ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, PDIR "/%s",
906
ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, "%s/%s",
911
error(0, errno, "asprintf");
905
for(int i = 0; i < numplugins; i++){
907
int plugin_fd = openat(dir_fd, direntries[i]->d_name, O_RDONLY);
909
error(0, errno, "Could not open plugin");
915
ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
913
ret = (int)TEMP_FAILURE_RETRY(fstat(plugin_fd, &st));
917
915
error(0, errno, "stat");
922
921
/* Ignore non-executable files */
923
922
if(not S_ISREG(st.st_mode)
924
or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
923
or (TEMP_FAILURE_RETRY(faccessat(dir_fd, direntries[i]->d_name,
926
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
927
" with bad type or mode\n", filename);
926
fprintf(stderr, "Ignoring plugin dir entry \"%s/%s\""
927
" with bad type or mode\n",
928
plugindir != NULL ? plugindir : PDIR,
929
direntries[i]->d_name);
933
plugin *p = getplugin(dirst->d_name);
936
plugin *p = getplugin(direntries[i]->d_name);
935
938
error(0, errno, "getplugin");
941
945
fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
946
direntries[i]->d_name);
976
981
ret = (int)TEMP_FAILURE_RETRY(pipe(pipefd));
982
#else /* O_CLOEXEC */
983
ret = (int)TEMP_FAILURE_RETRY(pipe2(pipefd, O_CLOEXEC));
984
#endif /* O_CLOEXEC */
978
986
error(0, errno, "pipe");
979
987
exitstatus = EX_OSERR;
991
if(pipefd[0] >= FD_SETSIZE){
992
fprintf(stderr, "pipe()[0] (%d) >= FD_SETSIZE (%d)", pipefd[0],
996
exitstatus = EX_OSERR;
982
1001
/* Ask OS to automatic close the pipe on exec */
983
1002
ret = set_cloexec_flag(pipefd[0]);
985
1004
error(0, errno, "set_cloexec_flag");
986
1007
exitstatus = EX_OSERR;
1008
free(direntries[i]);
989
1011
ret = set_cloexec_flag(pipefd[1]);
991
1013
error(0, errno, "set_cloexec_flag");
992
1016
exitstatus = EX_OSERR;
1017
free(direntries[i]);
1020
#endif /* not O_CLOEXEC */
995
1021
/* Block SIGCHLD until process is safely in process list */
996
1022
ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
997
1023
&sigchld_action.sa_mask,
1030
1062
_exit(EX_OSERR);
1034
/* If dir has no file descriptor, we could not set FD_CLOEXEC
1035
above and must now close it manually here. */
1038
if(p->environ[0] == NULL){
1039
if(execv(filename, p->argv) < 0){
1040
error(0, errno, "execv for %s", filename);
1044
if(execve(filename, p->argv, p->environ) < 0){
1045
error(0, errno, "execve for %s", filename);
1065
if(fexecve(plugin_fd, p->argv,
1066
(p->environ[0] != NULL) ? p->environ : environ) < 0){
1067
error(0, errno, "fexecve for %s/%s",
1068
plugindir != NULL ? plugindir : PDIR,
1069
direntries[i]->d_name);
1049
1072
/* no return */
1051
1074
/* Parent process */
1052
TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
1055
plugin *new_plugin = getplugin(dirst->d_name);
1075
close(pipefd[1]); /* Close unused write end of pipe */
1077
plugin *new_plugin = getplugin(direntries[i]->d_name);
1056
1078
if(new_plugin == NULL){
1057
1079
error(0, errno, "getplugin");
1058
1080
ret = (int)(TEMP_FAILURE_RETRY
1145
1158
(intmax_t) (proc->pid),
1146
1159
WTERMSIG(proc->status),
1147
1160
strsignal(WTERMSIG(proc->status)));
1148
} else if(WCOREDUMP(proc->status)){
1149
fprintf(stderr, "Plugin %s [%" PRIdMAX "] dumped"
1150
" core\n", proc->name, (intmax_t) (proc->pid));
1154
1164
/* Remove the plugin */
1155
#if defined (__GNUC__) and defined (__GLIBC__)
1156
#if not __GLIBC_PREREQ(2, 16)
1157
#pragma GCC diagnostic push
1158
#pragma GCC diagnostic ignored "-Wsign-conversion"
1161
FD_CLR(proc->fd, &rfds_all); /* Spurious warning from
1162
-Wconversion in GNU libc
1164
#if defined (__GNUC__) and defined (__GLIBC__)
1165
#if not __GLIBC_PREREQ(2, 16)
1166
#pragma GCC diagnostic pop
1165
FD_CLR(proc->fd, &rfds_all);
1170
1167
/* Block signal while modifying process_list */
1171
1168
ret = (int)TEMP_FAILURE_RETRY(sigprocmask
1213
1210
/* This process has not completed. Does it have any output? */
1214
#if defined (__GNUC__) and defined (__GLIBC__)
1215
#if not __GLIBC_PREREQ(2, 16)
1216
#pragma GCC diagnostic push
1217
#pragma GCC diagnostic ignored "-Wsign-conversion"
1220
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious
1226
#if defined (__GNUC__) and defined (__GLIBC__)
1227
#if not __GLIBC_PREREQ(2, 16)
1228
#pragma GCC diagnostic pop
1211
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
1231
1212
/* This process had nothing to say at this time */
1232
1213
proc = proc->next;
1235
1216
/* Before reading, make the process' data buffer large enough */
1236
1217
if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
1237
proc->buffer = realloc(proc->buffer, proc->buffer_size
1238
+ (size_t) BUFFER_SIZE);
1239
if(proc->buffer == NULL){
1218
char *new_buffer = realloc(proc->buffer, proc->buffer_size
1219
+ (size_t) BUFFER_SIZE);
1220
if(new_buffer == NULL){
1240
1221
error(0, errno, "malloc");
1241
1222
exitstatus = EX_OSERR;
1225
proc->buffer = new_buffer;
1244
1226
proc->buffer_size += BUFFER_SIZE;
1246
1228
/* Read from the process */