src/fdset.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU General Public License as published by
  10   the Free Software Foundation; either version 2 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <errno.h>
  23 #include <dirent.h>
  24 #include <fcntl.h>
  25 #include <unistd.h>
  26
  27 #include "set.h"
  28 #include "util.h"
  29 #include "macro.h"
  30 #include "fdset.h"
  31
  32 #define MAKE_SET(s) ((Set*) s)
  33 #define MAKE_FDSET(s) ((FDSet*) s)
  34
  35 /* Make sure we can distuingish fd 0 and NULL */
  36 #define FD_TO_PTR(fd) INT_TO_PTR((fd)+1)
  37 #define PTR_TO_FD(p) (PTR_TO_INT(p)-1)
  38
  39 FDSet *fdset_new(void) {
  40         return MAKE_FDSET(set_new(trivial_hash_func, trivial_compare_func));
  41 }
  42
  43 void fdset_free(FDSet *s) {
  44         void *p;
  45
  46         while ((p = set_steal_first(MAKE_SET(s)))) {
  47                 /* Valgrind's fd might have ended up in this set here,
  48                  * due to fdset_new_fill(). We'll ignore all failures
  49                  * here, so that the EBADFD that valgrind will return
  50                  * us on close() doesn't influence us */
  51
  52                 /* When reloading duplicates of the private bus
  53                  * connection fds and suchlike are closed here, which
  54                  * has no effect at all, since they are only
  55                  * duplicates. So don't be surprised about these log
  56                  * messages. */
  57
  58                 log_debug("Closing left-over fd %i", PTR_TO_FD(p));
  59                 close_nointr(PTR_TO_FD(p));
  60         }
  61
  62         set_free(MAKE_SET(s));
  63 }
  64
  65 int fdset_put(FDSet *s, int fd) {
  66         assert(s);
  67         assert(fd >= 0);
  68
  69         return set_put(MAKE_SET(s), FD_TO_PTR(fd));
  70 }
  71
  72 int fdset_put_dup(FDSet *s, int fd) {
  73         int copy, r;
  74
  75         assert(s);
  76         assert(fd >= 0);
  77
  78         if ((copy = fcntl(fd, F_DUPFD_CLOEXEC, 3)) < 0)
  79                 return -errno;
  80
  81         if ((r = fdset_put(s, copy)) < 0) {
  82                 close_nointr_nofail(copy);
  83                 return r;
  84         }
  85
  86         return copy;
  87 }
  88
  89 bool fdset_contains(FDSet *s, int fd) {
  90         assert(s);
  91         assert(fd >= 0);
  92
  93         return !!set_get(MAKE_SET(s), FD_TO_PTR(fd));
  94 }
  95
  96 int fdset_remove(FDSet *s, int fd) {
  97         assert(s);
  98         assert(fd >= 0);
  99
 100         return set_remove(MAKE_SET(s), FD_TO_PTR(fd)) ? fd : -ENOENT;
 101 }
 102
 103 int fdset_new_fill(FDSet **_s) {
 104         DIR *d;
 105         struct dirent *de;
 106         int r = 0;
 107         FDSet *s;
 108
 109         assert(_s);
 110
 111         /* Creates an fdsets and fills in all currently open file
 112          * descriptors. */
 113
 114         if (!(d = opendir("/proc/self/fd")))
 115                 return -errno;
 116
 117         if (!(s = fdset_new())) {
 118                 r = -ENOMEM;
 119                 goto finish;
 120         }
 121
 122         while ((de = readdir(d))) {
 123                 int fd = -1;
 124
 125                 if (ignore_file(de->d_name))
 126                         continue;
 127
 128                 if ((r = safe_atoi(de->d_name, &fd)) < 0)
 129                         goto finish;
 130
 131                 if (fd < 3)
 132                         continue;
 133
 134                 if (fd == dirfd(d))
 135                         continue;
 136
 137                 if ((r = fdset_put(s, fd)) < 0)
 138                         goto finish;
 139         }
 140
 141         r = 0;
 142         *_s = s;
 143         s = NULL;
 144
 145 finish:
 146         closedir(d);
 147
 148         /* We won't close the fds here! */
 149         if (s)
 150                 set_free(MAKE_SET(s));
 151
 152         return r;
 153 }
 154
 155 int fdset_cloexec(FDSet *fds, bool b) {
 156         Iterator i;
 157         void *p;
 158         int r;
 159
 160         assert(fds);
 161
 162         SET_FOREACH(p, MAKE_SET(fds), i)
 163                 if ((r = fd_cloexec(PTR_TO_FD(p), b)) < 0)
 164                         return r;
 165
 166         return 0;
 167 }