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,
 
 
217
221
  /* namelen = length of name of environment variable */
 
218
222
  size_t namelen = (size_t)(strchrnul(def, '=') - def);
 
219
223
  /* Search for this environment variable */
 
220
 
  for(char **e = p->environ; *e != NULL; e++){
 
221
 
    if(strncmp(*e, def, namelen + 1) == 0){
 
 
224
  for(char **envdef = p->environ; *envdef != NULL; envdef++){
 
 
225
    if(strncmp(*envdef, def, namelen + 1) == 0){
 
222
226
      /* It already exists */
 
226
 
          new = realloc(*e, strlen(def) + 1);
 
227
 
        } 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);
 
 
811
860
    ret = set_cloexec_flag(dir_fd);
 
813
862
      error(0, errno, "set_cloexec_flag");
 
814
 
      TEMP_FAILURE_RETRY(close(dir_fd));
 
815
863
      exitstatus = EX_OSERR;
 
818
866
#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;
 
 
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;
 
829
903
  FD_ZERO(&rfds_all);
 
831
905
  /* 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
 
      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);
 
894
 
    if(plugindir == NULL){
 
895
 
      ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, PDIR "/%s",
 
898
 
      ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, "%s/%s",
 
903
 
      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");
 
907
 
    ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
 
 
914
    ret = (int)TEMP_FAILURE_RETRY(fstat(plugin_fd, &st));
 
909
916
      error(0, errno, "stat");
 
914
922
    /* Ignore non-executable files */
 
915
923
    if(not S_ISREG(st.st_mode)
 
916
 
       or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
 
 
924
       or (TEMP_FAILURE_RETRY(faccessat(dir_fd, direntries[i]->d_name,
 
918
 
        fprintf(stderr, "Ignoring plugin dir entry \"%s\""
 
919
 
                " 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);
 
925
 
    plugin *p = getplugin(dirst->d_name);
 
 
937
    plugin *p = getplugin(direntries[i]->d_name);
 
927
939
      error(0, errno, "getplugin");
 
933
946
        fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
 
 
947
                direntries[i]->d_name);
 
 
968
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 */
 
970
987
      error(0, errno, "pipe");
 
971
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;
 
974
1002
    /* Ask OS to automatic close the pipe on exec */
 
975
1003
    ret = set_cloexec_flag(pipefd[0]);
 
977
1005
      error(0, errno, "set_cloexec_flag");
 
978
1008
      exitstatus = EX_OSERR;
 
 
1009
      free(direntries[i]);
 
981
1012
    ret = set_cloexec_flag(pipefd[1]);
 
983
1014
      error(0, errno, "set_cloexec_flag");
 
984
1017
      exitstatus = EX_OSERR;
 
 
1018
      free(direntries[i]);
 
 
1021
#endif  /* not O_CLOEXEC */
 
987
1022
    /* Block SIGCHLD until process is safely in process list */
 
988
1023
    ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
 
989
1024
                                              &sigchld_action.sa_mask,
 
 
1022
1063
        _exit(EX_OSERR);
 
1026
 
        /* If dir has no file descriptor, we could not set FD_CLOEXEC
 
1027
 
           above and must now close it manually here. */
 
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);
 
 
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);
 
1041
1073
      /* no return */
 
1043
1075
    /* Parent process */
 
1044
 
    TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
 
1047
 
    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);
 
1048
1079
    if(new_plugin == NULL){
 
1049
1080
      error(0, errno, "getplugin");
 
1050
1081
      ret = (int)(TEMP_FAILURE_RETRY
 
 
1074
 
#if defined (__GNUC__) and defined (__GLIBC__)
 
1075
 
#if not __GLIBC_PREREQ(2, 16)
 
1076
 
#pragma GCC diagnostic push
 
1077
 
#pragma GCC diagnostic ignored "-Wsign-conversion"
 
1080
 
    FD_SET(new_plugin->fd, &rfds_all); /* Spurious warning from
 
1081
 
                                          -Wconversion in GNU libc
 
1083
 
#if defined (__GNUC__) and defined (__GLIBC__)
 
1084
 
#if not __GLIBC_PREREQ(2, 16)
 
1085
 
#pragma GCC diagnostic pop
 
 
1107
    FD_SET(new_plugin->fd, &rfds_all);
 
1089
1109
    if(maxfd < new_plugin->fd){
 
1090
1110
      maxfd = new_plugin->fd;
 
1094
 
  TEMP_FAILURE_RETRY(closedir(dir));
 
1096
1118
  free_plugin(getplugin(NULL));
 
1098
1120
  for(plugin *p = plugin_list; p != NULL; p = p->next){
 
 
1205
1214
      /* This process has not completed.  Does it have any output? */
 
1206
 
#if defined (__GNUC__) and defined (__GLIBC__)
 
1207
 
#if not __GLIBC_PREREQ(2, 16)
 
1208
 
#pragma GCC diagnostic push
 
1209
 
#pragma GCC diagnostic ignored "-Wsign-conversion"
 
1212
 
      if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious
 
1218
 
#if defined (__GNUC__) and defined (__GLIBC__)
 
1219
 
#if not __GLIBC_PREREQ(2, 16)
 
1220
 
#pragma GCC diagnostic pop
 
 
1215
      if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
 
1223
1216
        /* This process had nothing to say at this time */
 
1224
1217
        proc = proc->next;
 
1227
1220
      /* Before reading, make the process' data buffer large enough */
 
1228
1221
      if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
 
1229
 
        proc->buffer = realloc(proc->buffer, proc->buffer_size
 
1230
 
                               + (size_t) BUFFER_SIZE);
 
1231
 
        if(proc->buffer == NULL){
 
 
1222
        char *new_buffer = realloc(proc->buffer, proc->buffer_size
 
 
1223
                                   + (size_t) BUFFER_SIZE);
 
 
1224
        if(new_buffer == NULL){
 
1232
1225
          error(0, errno, "malloc");
 
1233
1226
          exitstatus = EX_OSERR;
 
 
1229
        proc->buffer = new_buffer;
 
1236
1230
        proc->buffer_size += BUFFER_SIZE;
 
1238
1232
      /* Read from the process */