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>          /* fstat(), struct stat, waitpid(),
 
34
 
                                   WIFEXITED(), WEXITSTATUS(), wait(),
 
35
 
                                   pid_t, uid_t, gid_t, getuid(),
 
 
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(),
 
37
38
#include <sys/select.h>         /* fd_set, select(), FD_ZERO(),
 
38
39
                                   FD_SET(), FD_ISSET(), FD_CLR */
 
39
40
#include <sys/wait.h>           /* wait(), waitpid(), WIFEXITED(),
 
40
 
                                   WEXITSTATUS(), WTERMSIG() */
 
41
 
#include <sys/stat.h>           /* struct stat, fstat(), S_ISREG() */
 
 
41
                                   WEXITSTATUS(), WTERMSIG(),
 
 
43
#include <sys/stat.h>           /* struct stat, stat(), S_ISREG() */
 
42
44
#include <iso646.h>             /* and, or, not */
 
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()
 
 
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(),
 
51
54
#include <fcntl.h>              /* fcntl(), F_GETFD, F_SETFD,
 
52
 
                                   FD_CLOEXEC, openat(), scandirat(),
 
54
 
#include <string.h>             /* strsep, strlen(), strsignal(),
 
55
 
                                   strcmp(), strncmp() */
 
 
56
#include <string.h>             /* strsep, strlen(), asprintf(),
 
 
57
                                   strsignal(), strcmp(), strncmp() */
 
56
58
#include <errno.h>              /* errno */
 
57
59
#include <argp.h>               /* struct argp_option, struct
 
58
60
                                   argp_state, struct argp,
 
 
859
811
    ret = set_cloexec_flag(dir_fd);
 
861
813
      error(0, errno, "set_cloexec_flag");
 
 
814
      TEMP_FAILURE_RETRY(close(dir_fd));
 
862
815
      exitstatus = EX_OSERR;
 
865
818
#endif  /* O_CLOEXEC */
 
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);
 
 
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;
 
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;
 
902
829
  FD_ZERO(&rfds_all);
 
904
831
  /* Read and execute any executable in the plugin directory*/
 
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");
 
 
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");
 
913
 
    ret = (int)TEMP_FAILURE_RETRY(fstat(plugin_fd, &st));
 
 
907
    ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));
 
915
909
      error(0, errno, "stat");
 
921
914
    /* Ignore non-executable files */
 
922
915
    if(not S_ISREG(st.st_mode)
 
923
 
       or (TEMP_FAILURE_RETRY(faccessat(dir_fd, direntries[i]->d_name,
 
 
916
       or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){
 
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);
 
 
918
        fprintf(stderr, "Ignoring plugin dir entry \"%s\""
 
 
919
                " with bad type or mode\n", filename);
 
936
 
    plugin *p = getplugin(direntries[i]->d_name);
 
 
925
    plugin *p = getplugin(dirst->d_name);
 
938
927
      error(0, errno, "getplugin");
 
945
933
        fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",
 
946
 
                direntries[i]->d_name);
 
 
981
968
    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 */
 
986
970
      error(0, errno, "pipe");
 
987
971
      exitstatus = EX_OSERR;
 
991
 
    if(pipefd[0] >= FD_SETSIZE){
 
992
 
      fprintf(stderr, "pipe()[0] (%d) >= FD_SETSIZE (%d)", pipefd[0],
 
996
 
      exitstatus = EX_OSERR;
 
1001
974
    /* Ask OS to automatic close the pipe on exec */
 
1002
975
    ret = set_cloexec_flag(pipefd[0]);
 
1004
977
      error(0, errno, "set_cloexec_flag");
 
1007
978
      exitstatus = EX_OSERR;
 
1008
 
      free(direntries[i]);
 
1011
981
    ret = set_cloexec_flag(pipefd[1]);
 
1013
983
      error(0, errno, "set_cloexec_flag");
 
1016
984
      exitstatus = EX_OSERR;
 
1017
 
      free(direntries[i]);
 
1020
 
#endif  /* not O_CLOEXEC */
 
1021
987
    /* Block SIGCHLD until process is safely in process list */
 
1022
988
    ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,
 
1023
989
                                              &sigchld_action.sa_mask,
 
 
1062
1022
        _exit(EX_OSERR);
 
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);
 
 
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);
 
1072
1041
      /* no return */
 
1074
1043
    /* Parent process */
 
1075
 
    close(pipefd[1]);           /* Close unused write end of pipe */
 
1077
 
    plugin *new_plugin = getplugin(direntries[i]->d_name);
 
 
1044
    TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of
 
 
1047
    plugin *new_plugin = getplugin(dirst->d_name);
 
1078
1048
    if(new_plugin == NULL){
 
1079
1049
      error(0, errno, "getplugin");
 
1080
1050
      ret = (int)(TEMP_FAILURE_RETRY
 
 
1106
 
    FD_SET(new_plugin->fd, &rfds_all);
 
 
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
 
1108
1089
    if(maxfd < new_plugin->fd){
 
1109
1090
      maxfd = new_plugin->fd;
 
 
1094
  TEMP_FAILURE_RETRY(closedir(dir));
 
1117
1096
  free_plugin(getplugin(NULL));
 
1119
1098
  for(plugin *p = plugin_list; p != NULL; p = p->next){
 
 
1158
1137
                      (intmax_t) (proc->pid),
 
1159
1138
                      WTERMSIG(proc->status),
 
1160
1139
                      strsignal(WTERMSIG(proc->status)));
 
 
1140
            } else if(WCOREDUMP(proc->status)){
 
 
1141
              fprintf(stderr, "Plugin %s [%" PRIdMAX "] dumped"
 
 
1142
                      " core\n", proc->name, (intmax_t) (proc->pid));
 
1164
1146
          /* Remove the plugin */
 
1165
 
          FD_CLR(proc->fd, &rfds_all);
 
 
1147
#if defined (__GNUC__) and defined (__GLIBC__)
 
 
1148
#if not __GLIBC_PREREQ(2, 16)
 
 
1149
#pragma GCC diagnostic push
 
 
1150
#pragma GCC diagnostic ignored "-Wsign-conversion"
 
 
1153
          FD_CLR(proc->fd, &rfds_all); /* Spurious warning from
 
 
1154
                                          -Wconversion in GNU libc
 
 
1156
#if defined (__GNUC__) and defined (__GLIBC__)
 
 
1157
#if not __GLIBC_PREREQ(2, 16)
 
 
1158
#pragma GCC diagnostic pop
 
1167
1162
          /* Block signal while modifying process_list */
 
1168
1163
          ret = (int)TEMP_FAILURE_RETRY(sigprocmask
 
 
1210
1205
      /* This process has not completed.  Does it have any output? */
 
1211
 
      if(proc->eof or not FD_ISSET(proc->fd, &rfds)){
 
 
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
 
1212
1223
        /* This process had nothing to say at this time */
 
1213
1224
        proc = proc->next;
 
1216
1227
      /* Before reading, make the process' data buffer large enough */
 
1217
1228
      if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){
 
1218
 
        char *new_buffer = realloc(proc->buffer, proc->buffer_size
 
1219
 
                                   + (size_t) BUFFER_SIZE);
 
1220
 
        if(new_buffer == NULL){
 
 
1229
        proc->buffer = realloc(proc->buffer, proc->buffer_size
 
 
1230
                               + (size_t) BUFFER_SIZE);
 
 
1231
        if(proc->buffer == NULL){
 
1221
1232
          error(0, errno, "malloc");
 
1222
1233
          exitstatus = EX_OSERR;
 
1225
 
        proc->buffer = new_buffer;
 
1226
1236
        proc->buffer_size += BUFFER_SIZE;
 
1228
1238
      /* Read from the process */