lib/dup-safer.c

   1 /* Invoke dup, but avoid some glitches.
   2
   3    Copyright (C) 2001, 2004, 2005, 2006, 2009 Free Software
   4    Foundation, Inc.
   5
   6    This program is free software: you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  18
  19 /* Written by Paul Eggert.  */
  20
  21 #include <config.h>
  22
  23 #include "unistd-safer.h"
  24
  25 #include <fcntl.h>
  26 #include <unistd.h>
  27
  28 /* Like dup, but do not return STDIN_FILENO, STDOUT_FILENO, or
  29    STDERR_FILENO.  */
  30
  31 int
  32 dup_safer (int fd)
  33 {
  34 #if defined F_DUPFD && !REPLACE_FCHDIR
  35   return fcntl (fd, F_DUPFD, STDERR_FILENO + 1);
  36 #else
  37   /* fd_safer calls us back, but eventually the recursion unwinds and
  38      does the right thing.  */
  39   return fd_safer (dup (fd));
  40 #endif
  41 }
  42
  43 #if GNULIB_CLOEXEC
  44
  45 # include "cloexec.h"
  46
  47 # ifndef O_CLOEXEC
  48 #  define O_CLOEXEC 0
  49 # endif
  50
  51 /* Like dup, but do not return STDIN_FILENO, STDOUT_FILENO, or
  52    STDERR_FILENO.  If FLAG contains O_CLOEXEC, behave like
  53    fcntl(F_DUPFD_CLOEXEC) rather than fcntl(F_DUPFD).  */
  54
  55 int
  56 dup_safer_flag (int fd, int flag)
  57 {
  58   if (flag & O_CLOEXEC)
  59     {
  60 # if defined F_DUPFD_CLOEXEC && !REPLACE_FCHDIR
  61       return fcntl (fd, F_DUPFD_CLOEXEC, STDERR_FILENO + 1);
  62 # else
  63       /* fd_safer_flag calls us back, but eventually the recursion
  64          unwinds and does the right thing.  */
  65       fd = dup_cloexec (fd);
  66       return fd_safer_flag (fd, flag);
  67 # endif
  68     }
  69   return dup_safer (fd);
  70 }
  71
  72 #endif