gnulib/import/fcntl.c

   1 /* Provide file descriptor control.
   2
   3    Copyright (C) 2009-2022 Free Software Foundation, Inc.
   4
   5    This file is free software: you can redistribute it and/or modify
   6    it under the terms of the GNU Lesser General Public License as
   7    published by the Free Software Foundation; either version 2.1 of the
   8    License, or (at your option) any later version.
   9
  10    This file is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public License
  16    along with this program.  If not, see <https://www.gnu.org/licenses/>.  */
  17
  18 /* Written by Eric Blake <ebb9@byu.net>.  */
  19
  20 #include <config.h>
  21
  22 /* Specification.  */
  23 #include <fcntl.h>
  24
  25 #include <errno.h>
  26 #include <limits.h>
  27 #include <stdarg.h>
  28 #include <stdlib.h>
  29 #include <unistd.h>
  30
  31 #ifdef __KLIBC__
  32 # define INCL_DOS
  33 # include <os2.h>
  34 #endif
  35
  36 #if defined _WIN32 && ! defined __CYGWIN__
  37 /* Get declarations of the native Windows API functions.  */
  38 # define WIN32_LEAN_AND_MEAN
  39 # include <windows.h>
  40
  41 /* Get _get_osfhandle.  */
  42 # if GNULIB_MSVC_NOTHROW
  43 #  include "msvc-nothrow.h"
  44 # else
  45 #  include <io.h>
  46 # endif
  47
  48 /* Upper bound on getdtablesize().  See lib/getdtablesize.c.  */
  49 # define OPEN_MAX_MAX 0x10000
  50
  51 /* Duplicate OLDFD into the first available slot of at least NEWFD,
  52    which must be positive, with FLAGS determining whether the duplicate
  53    will be inheritable.  */
  54 static int
  55 dupfd (int oldfd, int newfd, int flags)
  56 {
  57   /* Mingw has no way to create an arbitrary fd.  Iterate until all
  58      file descriptors less than newfd are filled up.  */
  59   HANDLE curr_process = GetCurrentProcess ();
  60   HANDLE old_handle = (HANDLE) _get_osfhandle (oldfd);
  61   unsigned char fds_to_close[OPEN_MAX_MAX / CHAR_BIT];
  62   unsigned int fds_to_close_bound = 0;
  63   int result;
  64   BOOL inherit = flags & O_CLOEXEC ? FALSE : TRUE;
  65   int mode;
  66
  67   if (newfd < 0 || getdtablesize () <= newfd)
  68     {
  69       errno = EINVAL;
  70       return -1;
  71     }
  72   if (old_handle == INVALID_HANDLE_VALUE
  73       || (mode = _setmode (oldfd, O_BINARY)) == -1)
  74     {
  75       /* oldfd is not open, or is an unassigned standard file
  76          descriptor.  */
  77       errno = EBADF;
  78       return -1;
  79     }
  80   _setmode (oldfd, mode);
  81   flags |= mode;
  82
  83   for (;;)
  84     {
  85       HANDLE new_handle;
  86       int duplicated_fd;
  87       unsigned int index;
  88
  89       if (!DuplicateHandle (curr_process,           /* SourceProcessHandle */
  90                             old_handle,             /* SourceHandle */
  91                             curr_process,           /* TargetProcessHandle */
  92                             (PHANDLE) &new_handle,  /* TargetHandle */
  93                             (DWORD) 0,              /* DesiredAccess */
  94                             inherit,                /* InheritHandle */
  95                             DUPLICATE_SAME_ACCESS)) /* Options */
  96         {
  97           switch (GetLastError ())
  98             {
  99               case ERROR_TOO_MANY_OPEN_FILES:
 100                 errno = EMFILE;
 101                 break;
 102               case ERROR_INVALID_HANDLE:
 103               case ERROR_INVALID_TARGET_HANDLE:
 104               case ERROR_DIRECT_ACCESS_HANDLE:
 105                 errno = EBADF;
 106                 break;
 107               case ERROR_INVALID_PARAMETER:
 108               case ERROR_INVALID_FUNCTION:
 109               case ERROR_INVALID_ACCESS:
 110                 errno = EINVAL;
 111                 break;
 112               default:
 113                 errno = EACCES;
 114                 break;
 115             }
 116           result = -1;
 117           break;
 118         }
 119       duplicated_fd = _open_osfhandle ((intptr_t) new_handle, flags);
 120       if (duplicated_fd < 0)
 121         {
 122           CloseHandle (new_handle);
 123           result = -1;
 124           break;
 125         }
 126       if (newfd <= duplicated_fd)
 127         {
 128           result = duplicated_fd;
 129           break;
 130         }
 131
 132       /* Set the bit duplicated_fd in fds_to_close[].  */
 133       index = (unsigned int) duplicated_fd / CHAR_BIT;
 134       if (fds_to_close_bound <= index)
 135         {
 136           if (sizeof fds_to_close <= index)
 137             /* Need to increase OPEN_MAX_MAX.  */
 138             abort ();
 139           memset (fds_to_close + fds_to_close_bound, '\0',
 140                   index + 1 - fds_to_close_bound);
 141           fds_to_close_bound = index + 1;
 142         }
 143       fds_to_close[index] |= 1 << ((unsigned int) duplicated_fd % CHAR_BIT);
 144     }
 145
 146   /* Close the previous fds that turned out to be too small.  */
 147   {
 148     int saved_errno = errno;
 149     unsigned int duplicated_fd;
 150
 151     for (duplicated_fd = 0;
 152          duplicated_fd < fds_to_close_bound * CHAR_BIT;
 153          duplicated_fd++)
 154       if ((fds_to_close[duplicated_fd / CHAR_BIT]
 155            >> (duplicated_fd % CHAR_BIT))
 156           & 1)
 157         close (duplicated_fd);
 158
 159     errno = saved_errno;
 160   }
 161
 162 # if REPLACE_FCHDIR
 163   if (0 <= result)
 164     result = _gl_register_dup (oldfd, result);
 165 # endif
 166   return result;
 167 }
 168 #endif /* W32 */
 169
 170 /* Forward declarations, because we '#undef fcntl' in the middle of this
 171    compilation unit.  */
 172 /* Our implementation of fcntl (fd, F_DUPFD, target).  */
 173 static int rpl_fcntl_DUPFD (int fd, int target);
 174 /* Our implementation of fcntl (fd, F_DUPFD_CLOEXEC, target).  */
 175 static int rpl_fcntl_DUPFD_CLOEXEC (int fd, int target);
 176 #ifdef __KLIBC__
 177 /* Adds support for fcntl on directories.  */
 178 static int klibc_fcntl (int fd, int action, /* arg */...);
 179 #endif
 180
 181
 182 /* Perform the specified ACTION on the file descriptor FD, possibly
 183    using the argument ARG further described below.  This replacement
 184    handles the following actions, and forwards all others on to the
 185    native fcntl.  An unrecognized ACTION returns -1 with errno set to
 186    EINVAL.
 187
 188    F_DUPFD - duplicate FD, with int ARG being the minimum target fd.
 189    If successful, return the duplicate, which will be inheritable;
 190    otherwise return -1 and set errno.
 191
 192    F_DUPFD_CLOEXEC - duplicate FD, with int ARG being the minimum
 193    target fd.  If successful, return the duplicate, which will not be
 194    inheritable; otherwise return -1 and set errno.
 195
 196    F_GETFD - ARG need not be present.  If successful, return a
 197    non-negative value containing the descriptor flags of FD (only
 198    FD_CLOEXEC is portable, but other flags may be present); otherwise
 199    return -1 and set errno.  */
 200
 201 int
 202 fcntl (int fd, int action, /* arg */...)
 203 #undef fcntl
 204 #ifdef __KLIBC__
 205 # define fcntl klibc_fcntl
 206 #endif
 207 {
 208   va_list arg;
 209   int result = -1;
 210   va_start (arg, action);
 211   switch (action)
 212     {
 213     case F_DUPFD:
 214       {
 215         int target = va_arg (arg, int);
 216         result = rpl_fcntl_DUPFD (fd, target);
 217         break;
 218       }
 219
 220     case F_DUPFD_CLOEXEC:
 221       {
 222         int target = va_arg (arg, int);
 223         result = rpl_fcntl_DUPFD_CLOEXEC (fd, target);
 224         break;
 225       }
 226
 227 #if !HAVE_FCNTL
 228     case F_GETFD:
 229       {
 230 # if defined _WIN32 && ! defined __CYGWIN__
 231         HANDLE handle = (HANDLE) _get_osfhandle (fd);
 232         DWORD flags;
 233         if (handle == INVALID_HANDLE_VALUE
 234             || GetHandleInformation (handle, &flags) == 0)
 235           errno = EBADF;
 236         else
 237           result = (flags & HANDLE_FLAG_INHERIT) ? 0 : FD_CLOEXEC;
 238 # else /* !W32 */
 239         /* Use dup2 to reject invalid file descriptors.  No way to
 240            access this information, so punt.  */
 241         if (0 <= dup2 (fd, fd))
 242           result = 0;
 243 # endif /* !W32 */
 244         break;
 245       } /* F_GETFD */
 246 #endif /* !HAVE_FCNTL */
 247
 248       /* Implementing F_SETFD on mingw is not trivial - there is no
 249          API for changing the O_NOINHERIT bit on an fd, and merely
 250          changing the HANDLE_FLAG_INHERIT bit on the underlying handle
 251          can lead to odd state.  It may be possible by duplicating the
 252          handle, using _open_osfhandle with the right flags, then
 253          using dup2 to move the duplicate onto the original, but that
 254          is not supported for now.  */
 255
 256     default:
 257       {
 258 #if HAVE_FCNTL
 259         switch (action)
 260           {
 261           #ifdef F_BARRIERFSYNC                  /* macOS */
 262           case F_BARRIERFSYNC:
 263           #endif
 264           #ifdef F_CHKCLEAN                      /* macOS */
 265           case F_CHKCLEAN:
 266           #endif
 267           #ifdef F_CLOSEM                        /* NetBSD, HP-UX */
 268           case F_CLOSEM:
 269           #endif
 270           #ifdef F_FLUSH_DATA                    /* macOS */
 271           case F_FLUSH_DATA:
 272           #endif
 273           #ifdef F_FREEZE_FS                     /* macOS */
 274           case F_FREEZE_FS:
 275           #endif
 276           #ifdef F_FULLFSYNC                     /* macOS */
 277           case F_FULLFSYNC:
 278           #endif
 279           #ifdef F_GETCONFINED                   /* macOS */
 280           case F_GETCONFINED:
 281           #endif
 282           #ifdef F_GETDEFAULTPROTLEVEL           /* macOS */
 283           case F_GETDEFAULTPROTLEVEL:
 284           #endif
 285           #ifdef F_GETFD                         /* POSIX */
 286           case F_GETFD:
 287           #endif
 288           #ifdef F_GETFL                         /* POSIX */
 289           case F_GETFL:
 290           #endif
 291           #ifdef F_GETLEASE                      /* Linux */
 292           case F_GETLEASE:
 293           #endif
 294           #ifdef F_GETNOSIGPIPE                  /* macOS */
 295           case F_GETNOSIGPIPE:
 296           #endif
 297           #ifdef F_GETOWN                        /* POSIX */
 298           case F_GETOWN:
 299           #endif
 300           #ifdef F_GETPIPE_SZ                    /* Linux */
 301           case F_GETPIPE_SZ:
 302           #endif
 303           #ifdef F_GETPROTECTIONCLASS            /* macOS */
 304           case F_GETPROTECTIONCLASS:
 305           #endif
 306           #ifdef F_GETPROTECTIONLEVEL            /* macOS */
 307           case F_GETPROTECTIONLEVEL:
 308           #endif
 309           #ifdef F_GET_SEALS                     /* Linux */
 310           case F_GET_SEALS:
 311           #endif
 312           #ifdef F_GETSIG                        /* Linux */
 313           case F_GETSIG:
 314           #endif
 315           #ifdef F_MAXFD                         /* NetBSD */
 316           case F_MAXFD:
 317           #endif
 318           #ifdef F_RECYCLE                       /* macOS */
 319           case F_RECYCLE:
 320           #endif
 321           #ifdef F_SETFIFOENH                    /* HP-UX */
 322           case F_SETFIFOENH:
 323           #endif
 324           #ifdef F_THAW_FS                       /* macOS */
 325           case F_THAW_FS:
 326           #endif
 327             /* These actions take no argument.  */
 328             result = fcntl (fd, action);
 329             break;
 330
 331           #ifdef F_ADD_SEALS                     /* Linux */
 332           case F_ADD_SEALS:
 333           #endif
 334           #ifdef F_BADFD                         /* Solaris */
 335           case F_BADFD:
 336           #endif
 337           #ifdef F_CHECK_OPENEVT                 /* macOS */
 338           case F_CHECK_OPENEVT:
 339           #endif
 340           #ifdef F_DUP2FD                        /* FreeBSD, AIX, Solaris */
 341           case F_DUP2FD:
 342           #endif
 343           #ifdef F_DUP2FD_CLOEXEC                /* FreeBSD, Solaris */
 344           case F_DUP2FD_CLOEXEC:
 345           #endif
 346           #ifdef F_DUP2FD_CLOFORK                /* Solaris */
 347           case F_DUP2FD_CLOFORK:
 348           #endif
 349           #ifdef F_DUPFD                         /* POSIX */
 350           case F_DUPFD:
 351           #endif
 352           #ifdef F_DUPFD_CLOEXEC                 /* POSIX */
 353           case F_DUPFD_CLOEXEC:
 354           #endif
 355           #ifdef F_DUPFD_CLOFORK                 /* Solaris */
 356           case F_DUPFD_CLOFORK:
 357           #endif
 358           #ifdef F_GETXFL                        /* Solaris */
 359           case F_GETXFL:
 360           #endif
 361           #ifdef F_GLOBAL_NOCACHE                /* macOS */
 362           case F_GLOBAL_NOCACHE:
 363           #endif
 364           #ifdef F_MAKECOMPRESSED                /* macOS */
 365           case F_MAKECOMPRESSED:
 366           #endif
 367           #ifdef F_MOVEDATAEXTENTS               /* macOS */
 368           case F_MOVEDATAEXTENTS:
 369           #endif
 370           #ifdef F_NOCACHE                       /* macOS */
 371           case F_NOCACHE:
 372           #endif
 373           #ifdef F_NODIRECT                      /* macOS */
 374           case F_NODIRECT:
 375           #endif
 376           #ifdef F_NOTIFY                        /* Linux */
 377           case F_NOTIFY:
 378           #endif
 379           #ifdef F_OPLKACK                       /* IRIX */
 380           case F_OPLKACK:
 381           #endif
 382           #ifdef F_OPLKREG                       /* IRIX */
 383           case F_OPLKREG:
 384           #endif
 385           #ifdef F_RDAHEAD                       /* macOS */
 386           case F_RDAHEAD:
 387           #endif
 388           #ifdef F_SETBACKINGSTORE               /* macOS */
 389           case F_SETBACKINGSTORE:
 390           #endif
 391           #ifdef F_SETCONFINED                   /* macOS */
 392           case F_SETCONFINED:
 393           #endif
 394           #ifdef F_SETFD                         /* POSIX */
 395           case F_SETFD:
 396           #endif
 397           #ifdef F_SETFL                         /* POSIX */
 398           case F_SETFL:
 399           #endif
 400           #ifdef F_SETLEASE                      /* Linux */
 401           case F_SETLEASE:
 402           #endif
 403           #ifdef F_SETNOSIGPIPE                  /* macOS */
 404           case F_SETNOSIGPIPE:
 405           #endif
 406           #ifdef F_SETOWN                        /* POSIX */
 407           case F_SETOWN:
 408           #endif
 409           #ifdef F_SETPIPE_SZ                    /* Linux */
 410           case F_SETPIPE_SZ:
 411           #endif
 412           #ifdef F_SETPROTECTIONCLASS            /* macOS */
 413           case F_SETPROTECTIONCLASS:
 414           #endif
 415           #ifdef F_SETSIG                        /* Linux */
 416           case F_SETSIG:
 417           #endif
 418           #ifdef F_SINGLE_WRITER                 /* macOS */
 419           case F_SINGLE_WRITER:
 420           #endif
 421             /* These actions take an 'int' argument.  */
 422             {
 423               int x = va_arg (arg, int);
 424               result = fcntl (fd, action, x);
 425             }
 426             break;
 427
 428           default:
 429             /* Other actions take a pointer argument.  */
 430             {
 431               void *p = va_arg (arg, void *);
 432               result = fcntl (fd, action, p);
 433             }
 434             break;
 435           }
 436 #else
 437         errno = EINVAL;
 438 #endif
 439         break;
 440       }
 441     }
 442   va_end (arg);
 443   return result;
 444 }
 445
 446 static int
 447 rpl_fcntl_DUPFD (int fd, int target)
 448 {
 449   int result;
 450 #if !HAVE_FCNTL
 451   result = dupfd (fd, target, 0);
 452 #elif FCNTL_DUPFD_BUGGY || REPLACE_FCHDIR
 453   /* Detect invalid target; needed for cygwin 1.5.x.  */
 454   if (target < 0 || getdtablesize () <= target)
 455     {
 456       result = -1;
 457       errno = EINVAL;
 458     }
 459   else
 460     {
 461       /* Haiku alpha 2 loses fd flags on original.  */
 462       int flags = fcntl (fd, F_GETFD);
 463       if (flags < 0)
 464         result = -1;
 465       else
 466         {
 467           result = fcntl (fd, F_DUPFD, target);
 468           if (0 <= result && fcntl (fd, F_SETFD, flags) == -1)
 469             {
 470               int saved_errno = errno;
 471               close (result);
 472               result = -1;
 473               errno = saved_errno;
 474             }
 475 # if REPLACE_FCHDIR
 476           if (0 <= result)
 477             result = _gl_register_dup (fd, result);
 478 # endif
 479         }
 480     }
 481 #else
 482   result = fcntl (fd, F_DUPFD, target);
 483 #endif
 484   return result;
 485 }
 486
 487 static int
 488 rpl_fcntl_DUPFD_CLOEXEC (int fd, int target)
 489 {
 490   int result;
 491 #if !HAVE_FCNTL
 492   result = dupfd (fd, target, O_CLOEXEC);
 493 #else /* HAVE_FCNTL */
 494 # if defined __NetBSD__ || defined __HAIKU__
 495   /* On NetBSD 9.0, the system fcntl (fd, F_DUPFD_CLOEXEC, target)
 496      has only the same effect as fcntl (fd, F_DUPFD, target).  */
 497   /* On Haiku, the system fcntl (fd, F_DUPFD_CLOEXEC, target) sets
 498      the FD_CLOEXEC flag on fd, not on target.  Therefore avoid the
 499      system fcntl in this case.  */
 500 #  define have_dupfd_cloexec -1
 501 # else
 502   /* Try the system call first, if the headers claim it exists
 503      (that is, if GNULIB_defined_F_DUPFD_CLOEXEC is 0), since we
 504      may be running with a glibc that has the macro but with an
 505      older kernel that does not support it.  Cache the
 506      information on whether the system call really works, but
 507      avoid caching failure if the corresponding F_DUPFD fails
 508      for any reason.  0 = unknown, 1 = yes, -1 = no.  */
 509   static int have_dupfd_cloexec = GNULIB_defined_F_DUPFD_CLOEXEC ? -1 : 0;
 510   if (0 <= have_dupfd_cloexec)
 511     {
 512       result = fcntl (fd, F_DUPFD_CLOEXEC, target);
 513       if (0 <= result || errno != EINVAL)
 514         {
 515           have_dupfd_cloexec = 1;
 516 #  if REPLACE_FCHDIR
 517           if (0 <= result)
 518             result = _gl_register_dup (fd, result);
 519 #  endif
 520         }
 521       else
 522         {
 523           result = rpl_fcntl_DUPFD (fd, target);
 524           if (result >= 0)
 525             have_dupfd_cloexec = -1;
 526         }
 527     }
 528   else
 529 # endif
 530     result = rpl_fcntl_DUPFD (fd, target);
 531   if (0 <= result && have_dupfd_cloexec == -1)
 532     {
 533       int flags = fcntl (result, F_GETFD);
 534       if (flags < 0 || fcntl (result, F_SETFD, flags | FD_CLOEXEC) == -1)
 535         {
 536           int saved_errno = errno;
 537           close (result);
 538           errno = saved_errno;
 539           result = -1;
 540         }
 541     }
 542 #endif /* HAVE_FCNTL */
 543   return result;
 544 }
 545
 546 #undef fcntl
 547
 548 #ifdef __KLIBC__
 549
 550 static int
 551 klibc_fcntl (int fd, int action, /* arg */...)
 552 {
 553   va_list arg_ptr;
 554   int arg;
 555   struct stat sbuf;
 556   int result;
 557
 558   va_start (arg_ptr, action);
 559   arg = va_arg (arg_ptr, int);
 560   result = fcntl (fd, action, arg);
 561   /* EPERM for F_DUPFD, ENOTSUP for others */
 562   if (result == -1 && (errno == EPERM || errno == ENOTSUP)
 563       && !fstat (fd, &sbuf) && S_ISDIR (sbuf.st_mode))
 564     {
 565       ULONG ulMode;
 566
 567       switch (action)
 568         {
 569         case F_DUPFD:
 570           /* Find available fd */
 571           while (fcntl (arg, F_GETFL) != -1 || errno != EBADF)
 572             arg++;
 573
 574           result = dup2 (fd, arg);
 575           break;
 576
 577         /* Using underlying APIs is right ? */
 578         case F_GETFD:
 579           if (DosQueryFHState (fd, &ulMode))
 580             break;
 581
 582           result = (ulMode & OPEN_FLAGS_NOINHERIT) ? FD_CLOEXEC : 0;
 583           break;
 584
 585         case F_SETFD:
 586           if (arg & ~FD_CLOEXEC)
 587             break;
 588
 589           if (DosQueryFHState (fd, &ulMode))
 590             break;
 591
 592           if (arg & FD_CLOEXEC)
 593             ulMode |= OPEN_FLAGS_NOINHERIT;
 594           else
 595             ulMode &= ~OPEN_FLAGS_NOINHERIT;
 596
 597           /* Filter supported flags.  */
 598           ulMode &= (OPEN_FLAGS_WRITE_THROUGH | OPEN_FLAGS_FAIL_ON_ERROR
 599                      | OPEN_FLAGS_NO_CACHE | OPEN_FLAGS_NOINHERIT);
 600
 601           if (DosSetFHState (fd, ulMode))
 602             break;
 603
 604           result = 0;
 605           break;
 606
 607         case F_GETFL:
 608           result = 0;
 609           break;
 610
 611         case F_SETFL:
 612           if (arg != 0)
 613             break;
 614
 615           result = 0;
 616           break;
 617
 618         default:
 619           errno = EINVAL;
 620           break;
 621         }
 622     }
 623
 624   va_end (arg_ptr);
 625
 626   return result;
 627 }
 628
 629 #endif