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,
 
 
811
829
    ret = set_cloexec_flag(dir_fd);
 
813
831
      error(0, errno, "set_cloexec_flag");
 
814
 
      TEMP_FAILURE_RETRY(close(dir_fd));
 
815
832
      exitstatus = EX_OSERR;
 
818
835
#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;
 
 
838
  int good_name(const struct dirent * const dirent){
 
 
839
    const char * const patterns[] = { ".*", "#*#", "*~", "*.dpkg-new",
 
 
840
                                      "*.dpkg-old", "*.dpkg-bak",
 
 
841
                                      "*.dpkg-divert", NULL };
 
 
843
#pragma GCC diagnostic push
 
 
844
#pragma GCC diagnostic ignored "-Wcast-qual"
 
 
846
    for(const char **pat = (const char **)patterns;
 
 
847
        *pat != NULL; pat++){
 
 
849
#pragma GCC diagnostic pop
 
 
851
      if(fnmatch(*pat, dirent->d_name, FNM_FILE_NAME | FNM_PERIOD)
 
 
854
            fprintf(stderr, "Ignoring plugin dir entry \"%s\""
 
 
855
                    " matching pattern %s\n", dirent->d_name, *pat);
 
 
864
#if __GLIBC_PREREQ(2, 15)
 
 
865
  int numplugins = scandirat(dir_fd, ".", &direntries, good_name,
 
 
867
#else  /* not __GLIBC_PREREQ(2, 15) */
 
 
868
  int numplugins = scandir(plugindir != NULL ? plugindir : PDIR,
 
 
869
                           &direntries, good_name, alphasort);
 
 
870
#endif  /* not __GLIBC_PREREQ(2, 15) */
 
 
871
#else   /* not __GLIBC__ */
 
 
872
  int numplugins = scandir(plugindir != NULL ? plugindir : PDIR,
 
 
873
                           &direntries, good_name, alphasort);
 
 
874
#endif  /* not __GLIBC__ */
 
 
875
  if(numplugins == -1){
 
 
876
    error(0, errno, "Could not scan plugin dir");
 
 
878
    exitstatus = EX_OSERR;
 
829
882
  FD_ZERO(&rfds_all);
 
831
884
  /* 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");
 
 
885
  for(int i = 0; i < numplugins; i++){
 
 
887
    int plugin_fd = openat(dir_fd, direntries[i]->d_name, O_RDONLY);
 
 
889
      error(0, errno, "Could not open plugin");
 
907
 
    ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
 
 
893
    ret = (int)TEMP_FAILURE_RETRY(fstat(plugin_fd, &st));
 
909
895
      error(0, errno, "stat");
 
 
896
      TEMP_FAILURE_RETRY(close(plugin_fd));
 
914
901
    /* Ignore non-executable files */
 
915
902
    if(not S_ISREG(st.st_mode)
 
916
 
       or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
 
 
903
       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);
 
 
906
        fprintf(stderr, "Ignoring plugin dir entry \"%s/%s\""
 
 
907
                " with bad type or mode\n",
 
 
908
                plugindir != NULL ? plugindir : PDIR,
 
 
909
                direntries[i]->d_name);
 
 
911
      TEMP_FAILURE_RETRY(close(plugin_fd));
 
925
 
    plugin *p = getplugin(dirst->d_name);
 
 
916
    plugin *p = getplugin(direntries[i]->d_name);
 
927
918
      error(0, errno, "getplugin");
 
 
919
      TEMP_FAILURE_RETRY(close(plugin_fd));
 
933
925
        fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
 
 
926
                direntries[i]->d_name);
 
 
928
      TEMP_FAILURE_RETRY(close(plugin_fd));
 
 
968
961
    ret = (int)TEMP_FAILURE_RETRY(pipe(pipefd));
 
 
962
#else  /* O_CLOEXEC */
 
 
963
    ret = (int)TEMP_FAILURE_RETRY(pipe2(pipefd, O_CLOEXEC));
 
 
964
#endif  /* O_CLOEXEC */
 
970
966
      error(0, errno, "pipe");
 
971
967
      exitstatus = EX_OSERR;
 
 
971
    if(pipefd[0] >= FD_SETSIZE){
 
 
972
      fprintf(stderr, "pipe()[0] (%d) >= FD_SETSIZE (%d)", pipefd[0],
 
 
974
      TEMP_FAILURE_RETRY(close(pipefd[0]));
 
 
975
      TEMP_FAILURE_RETRY(close(pipefd[1]));
 
 
976
      exitstatus = EX_OSERR;
 
974
981
    /* Ask OS to automatic close the pipe on exec */
 
975
982
    ret = set_cloexec_flag(pipefd[0]);
 
977
984
      error(0, errno, "set_cloexec_flag");
 
 
985
      TEMP_FAILURE_RETRY(close(pipefd[0]));
 
 
986
      TEMP_FAILURE_RETRY(close(pipefd[1]));
 
978
987
      exitstatus = EX_OSERR;
 
981
991
    ret = set_cloexec_flag(pipefd[1]);
 
983
993
      error(0, errno, "set_cloexec_flag");
 
 
994
      TEMP_FAILURE_RETRY(close(pipefd[0]));
 
 
995
      TEMP_FAILURE_RETRY(close(pipefd[1]));
 
984
996
      exitstatus = EX_OSERR;
 
 
1000
#endif  /* not O_CLOEXEC */
 
987
1001
    /* Block SIGCHLD until process is safely in process list */
 
988
1002
    ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
 
989
1003
                                              &sigchld_action.sa_mask,
 
 
1022
1042
        _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);
 
 
1045
      if(fexecve(plugin_fd, p->argv,
 
 
1046
                (p->environ[0] != NULL) ? p->environ : environ) < 0){
 
 
1047
        error(0, errno, "fexecve for %s/%s",
 
 
1048
              plugindir != NULL ? plugindir : PDIR,
 
 
1049
              direntries[i]->d_name);
 
1041
1052
      /* no return */
 
1043
1054
    /* Parent process */
 
1044
1055
    TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
 
1047
 
    plugin *new_plugin = getplugin(dirst->d_name);
 
 
1057
    TEMP_FAILURE_RETRY(close(plugin_fd));
 
 
1058
    plugin *new_plugin = getplugin(direntries[i]->d_name);
 
1048
1059
    if(new_plugin == NULL){
 
1049
1060
      error(0, errno, "getplugin");
 
1050
1061
      ret = (int)(TEMP_FAILURE_RETRY
 
 
1181
1220
      /* This process has not completed.  Does it have any output? */
 
 
1221
#if defined (__GNUC__) and defined (__GLIBC__)
 
 
1222
#if not __GLIBC_PREREQ(2, 16)
 
 
1223
#pragma GCC diagnostic push
 
 
1224
#pragma GCC diagnostic ignored "-Wsign-conversion"
 
1182
1227
      if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious
 
 
1233
#if defined (__GNUC__) and defined (__GLIBC__)
 
 
1234
#if not __GLIBC_PREREQ(2, 16)
 
 
1235
#pragma GCC diagnostic pop
 
1185
1238
        /* This process had nothing to say at this time */
 
1186
1239
        proc = proc->next;
 
1189
1242
      /* Before reading, make the process' data buffer large enough */
 
1190
1243
      if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
 
1191
 
        proc->buffer = realloc(proc->buffer, proc->buffer_size
 
1192
 
                               + (size_t) BUFFER_SIZE);
 
1193
 
        if(proc->buffer == NULL){
 
 
1244
        char *new_buffer = realloc(proc->buffer, proc->buffer_size
 
 
1245
                                   + (size_t) BUFFER_SIZE);
 
 
1246
        if(new_buffer == NULL){
 
1194
1247
          error(0, errno, "malloc");
 
1195
1248
          exitstatus = EX_OSERR;
 
 
1251
        proc->buffer = new_buffer;
 
1198
1252
        proc->buffer_size += BUFFER_SIZE;
 
1200
1254
      /* Read from the process */