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, fclose() */
33
#include <sys/types.h> /* DIR, fdopendir(), fstat(), struct
32
stderr, STDOUT_FILENO */
33
#include <sys/types.h> /* DIR, fdopendir(), stat(), struct
34
34
stat, waitpid(), WIFEXITED(),
35
35
WEXITSTATUS(), wait(), pid_t,
36
36
uid_t, gid_t, getuid(), getgid(),
40
40
#include <sys/wait.h> /* wait(), waitpid(), WIFEXITED(),
41
41
WEXITSTATUS(), WTERMSIG(),
43
#include <sys/stat.h> /* struct stat, fstat(), S_ISREG() */
43
#include <sys/stat.h> /* struct stat, stat(), S_ISREG() */
44
44
#include <iso646.h> /* and, or, not */
45
45
#include <dirent.h> /* DIR, struct dirent, fdopendir(),
46
46
readdir(), closedir(), dirfd() */
47
#include <unistd.h> /* fcntl(), F_GETFD, F_SETFD,
48
FD_CLOEXEC, write(), STDOUT_FILENO,
49
struct stat, fstat(), close(),
50
setgid(), setuid(), S_ISREG(),
51
faccessat() pipe(), fork(),
52
_exit(), dup2(), fexecve(), read()
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(),
54
54
#include <fcntl.h> /* fcntl(), F_GETFD, F_SETFD,
55
FD_CLOEXEC, openat() */
56
#include <string.h> /* strsep, strlen(), strsignal(),
57
strcmp(), strncmp() */
56
#include <string.h> /* strsep, strlen(), asprintf(),
57
strsignal(), strcmp(), strncmp() */
58
58
#include <errno.h> /* errno */
59
59
#include <argp.h> /* struct argp_option, struct
60
60
argp_state, struct argp,
175
173
/* Helper function for add_argument and add_environment */
176
__attribute__((nonnull, warn_unused_result))
174
__attribute__((nonnull))
177
175
static bool add_to_char_array(const char *new, char ***array,
179
177
/* Resize the pointed-to array to hold one more pointer */
180
char **new_array = NULL;
182
new_array = realloc(*array, sizeof(char *)
183
* (size_t) ((*len) + 2));
184
} while(new_array == NULL and errno == EINTR);
179
*array = realloc(*array, sizeof(char *)
180
* (size_t) ((*len) + 2));
181
} while(*array == NULL and errno == EINTR);
185
182
/* Malloc check */
186
if(new_array == NULL){
190
186
/* Make a copy of the new string */
221
217
/* namelen = length of name of environment variable */
222
218
size_t namelen = (size_t)(strchrnul(def, '=') - def);
223
219
/* Search for this environment variable */
224
for(char **envdef = p->environ; *envdef != NULL; envdef++){
225
if(strncmp(*envdef, def, namelen + 1) == 0){
220
for(char **e = p->environ; *e != NULL; e++){
221
if(strncmp(*e, def, namelen + 1) == 0){
226
222
/* It already exists */
230
new_envdef = realloc(*envdef, strlen(def) + 1);
231
} while(new_envdef == NULL and errno == EINTR);
232
if(new_envdef == NULL){
226
new = realloc(*e, strlen(def) + 1);
227
} while(new == NULL and errno == EINTR);
235
*envdef = new_envdef;
236
strcpy(*envdef, def);
812
783
/* Open plugin directory with close_on_exec flag */
814
dir_fd = open(plugindir != NULL ? plugindir : PDIR, O_RDONLY |
817
#else /* not O_CLOEXEC */
819
#endif /* not O_CLOEXEC */
786
if(plugindir == NULL){
787
dir_fd = open(PDIR, O_RDONLY |
790
#else /* not O_CLOEXEC */
792
#endif /* not O_CLOEXEC */
795
dir_fd = open(plugindir, O_RDONLY |
798
#else /* not O_CLOEXEC */
800
#endif /* not O_CLOEXEC */
821
803
if(dir_fd == -1){
822
804
error(0, errno, "Could not open plugin dir");
823
805
exitstatus = EX_UNAVAILABLE;
847
d_name_len = strlen(dirst->d_name);
865
849
/* Ignore dotfiles, backup files and other junk */
867
851
bool bad_name = false;
868
const char * const patterns[] = { ".*", "#*#", "*~",
869
"*.dpkg-new", "*.dpkg-old",
870
"*.dpkg-bak", "*.dpkg-divert",
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, dirst->d_name,
882
FNM_FILE_NAME | FNM_PERIOD) != FNM_NOMATCH){
884
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
885
" matching pattern %s\n", dirst->d_name, *pat);
853
const char const *bad_prefixes[] = { ".", "#", NULL };
855
const char const *bad_suffixes[] = { "~", "#", ".dpkg-new",
858
".dpkg-divert", NULL };
859
for(const char **pre = bad_prefixes; *pre != NULL; pre++){
860
size_t pre_len = strlen(*pre);
861
if((d_name_len >= pre_len)
862
and strncmp((dirst->d_name), *pre, pre_len) == 0){
864
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
865
" with bad prefix %s\n", dirst->d_name, *pre);
874
for(const char **suf = bad_suffixes; *suf != NULL; suf++){
875
size_t suf_len = strlen(*suf);
876
if((d_name_len >= suf_len)
877
and (strcmp((dirst->d_name) + d_name_len-suf_len, *suf)
880
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
881
" with bad suffix %s\n", dirst->d_name, *suf);
896
int plugin_fd = openat(dir_fd, dirst->d_name, O_RDONLY |
899
#else /* not O_CLOEXEC */
901
#endif /* not O_CLOEXEC */
904
error(0, errno, "Could not open plugin");
894
if(plugindir == NULL){
895
ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, PDIR "/%s",
898
ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, "%s/%s",
908
/* Set the FD_CLOEXEC flag on the plugin FD */
909
ret = set_cloexec_flag(plugin_fd);
911
error(0, errno, "set_cloexec_flag");
912
TEMP_FAILURE_RETRY(close(plugin_fd));
903
error(0, errno, "asprintf");
915
#endif /* O_CLOEXEC */
916
ret = (int)TEMP_FAILURE_RETRY(fstat(plugin_fd, &st));
907
ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
918
909
error(0, errno, "stat");
919
TEMP_FAILURE_RETRY(close(plugin_fd));
923
914
/* Ignore non-executable files */
924
915
if(not S_ISREG(st.st_mode)
925
or (TEMP_FAILURE_RETRY(faccessat(dir_fd, dirst->d_name, X_OK,
916
or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
928
fprintf(stderr, "Ignoring plugin dir entry \"%s/%s\""
929
" with bad type or mode\n",
930
plugindir != NULL ? plugindir : PDIR, dirst->d_name);
918
fprintf(stderr, "Ignoring plugin dir entry \"%s\""
919
" with bad type or mode\n", filename);
932
TEMP_FAILURE_RETRY(close(plugin_fd));
936
925
plugin *p = getplugin(dirst->d_name);
938
927
error(0, errno, "getplugin");
939
TEMP_FAILURE_RETRY(close(plugin_fd));
981
971
exitstatus = EX_OSERR;
984
if(pipefd[0] >= FD_SETSIZE){
985
fprintf(stderr, "pipe()[0] (%d) >= FD_SETSIZE (%d)", pipefd[0],
987
TEMP_FAILURE_RETRY(close(pipefd[0]));
988
TEMP_FAILURE_RETRY(close(pipefd[1]));
989
exitstatus = EX_OSERR;
992
974
/* Ask OS to automatic close the pipe on exec */
993
975
ret = set_cloexec_flag(pipefd[0]);
995
977
error(0, errno, "set_cloexec_flag");
996
TEMP_FAILURE_RETRY(close(pipefd[0]));
997
TEMP_FAILURE_RETRY(close(pipefd[1]));
998
978
exitstatus = EX_OSERR;
1001
981
ret = set_cloexec_flag(pipefd[1]);
1003
983
error(0, errno, "set_cloexec_flag");
1004
TEMP_FAILURE_RETRY(close(pipefd[0]));
1005
TEMP_FAILURE_RETRY(close(pipefd[1]));
1006
984
exitstatus = EX_OSERR;
1053
1027
above and must now close it manually here. */
1056
if(fexecve(plugin_fd, p->argv,
1057
(p->environ[0] != NULL) ? p->environ : environ) < 0){
1058
error(0, errno, "fexecve for %s/%s",
1059
plugindir != NULL ? plugindir : PDIR, dirst->d_name);
1030
if(p->environ[0] == NULL){
1031
if(execv(filename, p->argv) < 0){
1032
error(0, errno, "execv for %s", filename);
1036
if(execve(filename, p->argv, p->environ) < 0){
1037
error(0, errno, "execve for %s", filename);
1062
1041
/* no return */
1064
1043
/* Parent process */
1065
1044
TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
1067
TEMP_FAILURE_RETRY(close(plugin_fd));
1068
1047
plugin *new_plugin = getplugin(dirst->d_name);
1069
1048
if(new_plugin == NULL){
1070
1049
error(0, errno, "getplugin");
1226
1181
/* This process has not completed. Does it have any output? */
1227
#if defined (__GNUC__) and defined (__GLIBC__)
1228
#if not __GLIBC_PREREQ(2, 16)
1229
#pragma GCC diagnostic push
1230
#pragma GCC diagnostic ignored "-Wsign-conversion"
1233
1182
if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious
1239
#if defined (__GNUC__) and defined (__GLIBC__)
1240
#if not __GLIBC_PREREQ(2, 16)
1241
#pragma GCC diagnostic pop
1244
1185
/* This process had nothing to say at this time */
1245
1186
proc = proc->next;
1248
1189
/* Before reading, make the process' data buffer large enough */
1249
1190
if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
1250
char *new_buffer = realloc(proc->buffer, proc->buffer_size
1251
+ (size_t) BUFFER_SIZE);
1252
if(new_buffer == NULL){
1191
proc->buffer = realloc(proc->buffer, proc->buffer_size
1192
+ (size_t) BUFFER_SIZE);
1193
if(proc->buffer == NULL){
1253
1194
error(0, errno, "malloc");
1254
1195
exitstatus = EX_OSERR;
1257
proc->buffer = new_buffer;
1258
1198
proc->buffer_size += BUFFER_SIZE;
1260
1200
/* Read from the process */