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
40
WEXITSTATUS(), WTERMSIG(),
43
#include <sys/stat.h> /* struct stat, stat(), S_ISREG() */
42
#include <sys/stat.h> /* struct stat, fstat(), S_ISREG() */
44
43
#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(),
44
#include <dirent.h> /* struct dirent, scandirat() */
45
#include <unistd.h> /* fcntl(), F_GETFD, F_SETFD,
46
FD_CLOEXEC, write(), STDOUT_FILENO,
47
struct stat, fstat(), close(),
48
setgid(), setuid(), S_ISREG(),
49
faccessat() pipe2(), fork(),
50
_exit(), dup2(), fexecve(), read()
54
52
#include <fcntl.h> /* fcntl(), F_GETFD, F_SETFD,
56
#include <string.h> /* strsep, strlen(), asprintf(),
57
strsignal(), strcmp(), strncmp() */
53
FD_CLOEXEC, openat(), scandirat(),
55
#include <string.h> /* strsep, strlen(), strsignal(),
56
strcmp(), strncmp() */
58
57
#include <errno.h> /* errno */
59
58
#include <argp.h> /* struct argp_option, struct
60
59
argp_state, struct argp,
214
221
/* namelen = length of name of environment variable */
215
222
size_t namelen = (size_t)(strchrnul(def, '=') - def);
216
223
/* Search for this environment variable */
217
for(char **e = p->environ; *e != NULL; e++){
218
if(strncmp(*e, def, namelen + 1) == 0){
224
for(char **envdef = p->environ; *envdef != NULL; envdef++){
225
if(strncmp(*envdef, def, namelen + 1) == 0){
219
226
/* It already exists */
223
new = realloc(*e, strlen(def) + 1);
224
} while(new == NULL and errno == EINTR);
230
new_envdef = realloc(*envdef, strlen(def) + 1);
231
} while(new_envdef == NULL and errno == EINTR);
232
if(new_envdef == NULL){
235
*envdef = new_envdef;
236
strcpy(*envdef, def);
488
512
plugin *p = getplugin(arg);
490
514
p->disabled = false;
494
519
case 128: /* --plugin-dir */
496
521
plugindir = strdup(arg);
522
if(plugindir != NULL){
498
526
case 129: /* --config-file */
499
527
/* This is already done by parse_opt_config_file() */
501
529
case 130: /* --userid */
502
tmpmax = strtoimax(arg, &tmp, 10);
530
tmp_id = strtoimax(arg, &tmp, 10);
503
531
if(errno != 0 or tmp == arg or *tmp != '\0'
504
or tmpmax != (uid_t)tmpmax){
532
or tmp_id != (uid_t)tmp_id){
505
533
argp_error(state, "Bad user ID number: \"%s\", using %"
506
534
PRIdMAX, arg, (intmax_t)uid);
511
540
case 131: /* --groupid */
512
tmpmax = strtoimax(arg, &tmp, 10);
541
tmp_id = strtoimax(arg, &tmp, 10);
513
542
if(errno != 0 or tmp == arg or *tmp != '\0'
514
or tmpmax != (gid_t)tmpmax){
543
or tmp_id != (gid_t)tmp_id){
515
544
argp_error(state, "Bad group ID number: \"%s\", using %"
516
545
PRIdMAX, arg, (intmax_t)gid);
521
551
case 132: /* --debug */
554
case 133: /* --plugin-helper-dir */
555
free(pluginhelperdir);
556
pluginhelperdir = strdup(arg);
557
if(pluginhelperdir != NULL){
525
562
* These reproduce what we would get without ARGP_NO_HELP
805
860
ret = set_cloexec_flag(dir_fd);
807
862
error(0, errno, "set_cloexec_flag");
808
TEMP_FAILURE_RETRY(close(dir_fd));
809
863
exitstatus = EX_OSERR;
812
866
#endif /* O_CLOEXEC */
814
dir = fdopendir(dir_fd);
816
error(0, errno, "Could not open plugin dir");
817
TEMP_FAILURE_RETRY(close(dir_fd));
818
exitstatus = EX_OSERR;
869
int good_name(const struct dirent * const dirent){
870
const char * const patterns[] = { ".*", "#*#", "*~", "*.dpkg-new",
871
"*.dpkg-old", "*.dpkg-bak",
872
"*.dpkg-divert", NULL };
874
#pragma GCC diagnostic push
875
#pragma GCC diagnostic ignored "-Wcast-qual"
877
for(const char **pat = (const char **)patterns;
878
*pat != NULL; pat++){
880
#pragma GCC diagnostic pop
882
if(fnmatch(*pat, dirent->d_name, FNM_FILE_NAME | FNM_PERIOD)
885
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
886
" matching pattern %s\n", dirent->d_name, *pat);
894
int numplugins = scandirat(dir_fd, ".", &direntries, good_name,
896
if(numplugins == -1){
897
error(0, errno, "Could not scan plugin dir");
899
exitstatus = EX_OSERR;
823
903
FD_ZERO(&rfds_all);
825
905
/* Read and execute any executable in the plugin directory*/
828
dirst = readdir(dir);
829
} while(dirst == NULL and errno == EINTR);
831
/* All directory entries have been processed */
834
error(0, errno, "readdir");
835
exitstatus = EX_IOERR;
841
d_name_len = strlen(dirst->d_name);
843
/* Ignore dotfiles, backup files and other junk */
845
bool bad_name = false;
847
const char const *bad_prefixes[] = { ".", "#", NULL };
849
const char const *bad_suffixes[] = { "~", "#", ".dpkg-new",
852
".dpkg-divert", NULL };
853
for(const char **pre = bad_prefixes; *pre != NULL; pre++){
854
size_t pre_len = strlen(*pre);
855
if((d_name_len >= pre_len)
856
and strncmp((dirst->d_name), *pre, pre_len) == 0){
858
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
859
" with bad prefix %s\n", dirst->d_name, *pre);
868
for(const char **suf = bad_suffixes; *suf != NULL; suf++){
869
size_t suf_len = strlen(*suf);
870
if((d_name_len >= suf_len)
871
and (strcmp((dirst->d_name) + d_name_len-suf_len, *suf)
874
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
875
" with bad suffix %s\n", dirst->d_name, *suf);
888
if(plugindir == NULL){
889
ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, PDIR "/%s",
892
ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, "%s/%s",
897
error(0, errno, "asprintf");
906
for(int i = 0; i < numplugins; i++){
908
int plugin_fd = openat(dir_fd, direntries[i]->d_name, O_RDONLY);
910
error(0, errno, "Could not open plugin");
901
ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
914
ret = (int)TEMP_FAILURE_RETRY(fstat(plugin_fd, &st));
903
916
error(0, errno, "stat");
908
922
/* Ignore non-executable files */
909
923
if(not S_ISREG(st.st_mode)
910
or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
924
or (TEMP_FAILURE_RETRY(faccessat(dir_fd, direntries[i]->d_name,
912
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
913
" with bad type or mode\n", filename);
927
fprintf(stderr, "Ignoring plugin dir entry \"%s/%s\""
928
" with bad type or mode\n",
929
plugindir != NULL ? plugindir : PDIR,
930
direntries[i]->d_name);
919
plugin *p = getplugin(dirst->d_name);
937
plugin *p = getplugin(direntries[i]->d_name);
921
939
error(0, errno, "getplugin");
927
946
fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
947
direntries[i]->d_name);
962
982
ret = (int)TEMP_FAILURE_RETRY(pipe(pipefd));
983
#else /* O_CLOEXEC */
984
ret = (int)TEMP_FAILURE_RETRY(pipe2(pipefd, O_CLOEXEC));
985
#endif /* O_CLOEXEC */
964
987
error(0, errno, "pipe");
965
988
exitstatus = EX_OSERR;
992
if(pipefd[0] >= FD_SETSIZE){
993
fprintf(stderr, "pipe()[0] (%d) >= FD_SETSIZE (%d)", pipefd[0],
997
exitstatus = EX_OSERR;
968
1002
/* Ask OS to automatic close the pipe on exec */
969
1003
ret = set_cloexec_flag(pipefd[0]);
971
1005
error(0, errno, "set_cloexec_flag");
972
1008
exitstatus = EX_OSERR;
1009
free(direntries[i]);
975
1012
ret = set_cloexec_flag(pipefd[1]);
977
1014
error(0, errno, "set_cloexec_flag");
978
1017
exitstatus = EX_OSERR;
1018
free(direntries[i]);
1021
#endif /* not O_CLOEXEC */
981
1022
/* Block SIGCHLD until process is safely in process list */
982
1023
ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
983
1024
&sigchld_action.sa_mask,
1016
1063
_exit(EX_OSERR);
1020
/* If dir has no file descriptor, we could not set FD_CLOEXEC
1021
above and must now close it manually here. */
1024
if(p->environ[0] == NULL){
1025
if(execv(filename, p->argv) < 0){
1026
error(0, errno, "execv for %s", filename);
1030
if(execve(filename, p->argv, p->environ) < 0){
1031
error(0, errno, "execve for %s", filename);
1066
if(fexecve(plugin_fd, p->argv,
1067
(p->environ[0] != NULL) ? p->environ : environ) < 0){
1068
error(0, errno, "fexecve for %s/%s",
1069
plugindir != NULL ? plugindir : PDIR,
1070
direntries[i]->d_name);
1035
1073
/* no return */
1037
1075
/* Parent process */
1038
TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
1041
plugin *new_plugin = getplugin(dirst->d_name);
1076
close(pipefd[1]); /* Close unused write end of pipe */
1078
plugin *new_plugin = getplugin(direntries[i]->d_name);
1042
1079
if(new_plugin == NULL){
1043
1080
error(0, errno, "getplugin");
1044
1081
ret = (int)(TEMP_FAILURE_RETRY
1175
1214
/* This process has not completed. Does it have any output? */
1176
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious
1215
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
1179
1216
/* This process had nothing to say at this time */
1180
1217
proc = proc->next;
1183
1220
/* Before reading, make the process' data buffer large enough */
1184
1221
if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
1185
proc->buffer = realloc(proc->buffer, proc->buffer_size
1186
+ (size_t) BUFFER_SIZE);
1187
if(proc->buffer == NULL){
1222
char *new_buffer = realloc(proc->buffer, proc->buffer_size
1223
+ (size_t) BUFFER_SIZE);
1224
if(new_buffer == NULL){
1188
1225
error(0, errno, "malloc");
1189
1226
exitstatus = EX_OSERR;
1229
proc->buffer = new_buffer;
1192
1230
proc->buffer_size += BUFFER_SIZE;
1194
1232
/* Read from the process */